Ben Kurtovic
12 years ago
1 changed files with 3 additions and 4 deletions
Unified View
Diff Options
-
+3 -4func_smash.py
+ 3
- 4
func_smash.py
View File
| @@ -11,14 +11,13 @@ MARKOV_END = -2 | |||||
| def make_chain(funcs): | def make_chain(funcs): | ||||
| chain = {} | chain = {} | ||||
| for func in funcs: | for func in funcs: | ||||
| co = func.func_code | |||||
| co = func.__code__ | |||||
| code = co.co_code | code = co.co_code | ||||
| n = len(code) | n = len(code) | ||||
| i = 0 | i = 0 | ||||
| lastop = MARKOV_START | lastop = MARKOV_START | ||||
| while i < n: | while i < n: | ||||
| c = code[i] | |||||
| op = ord(c) | |||||
| op = ord(code[i]) | |||||
| i += 1 | i += 1 | ||||
| if op >= opcode.HAVE_ARGUMENT: | if op >= opcode.HAVE_ARGUMENT: | ||||
| oparg = ord(code[i]) + ord(code[i + 1]) * 256 | oparg = ord(code[i]) + ord(code[i + 1]) * 256 | ||||
| @@ -111,7 +110,7 @@ if __name__ == "__main__": | |||||
| func = make_function(chain, "func") | func = make_function(chain, "func") | ||||
| print "SMASHED FUNCTION CODE:", [ord(c) for c in func.func_code.co_code] | |||||
| print "SMASHED FUNCTION CODE:", [ord(c) for c in func.__code__.co_code] | |||||
| print "FUNCTION DISASSEMBLY:" | print "FUNCTION DISASSEMBLY:" | ||||
| dis.dis(func) | dis.dis(func) | ||||