From e51c650b9644c244f64ad39081c0393747a3c489 Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@verizon.net>
Date: Mon, 16 Apr 2012 21:55:08 -0400
Subject: [PATCH] Minor cleanup

---
 func_smash.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/func_smash.py b/func_smash.py
index 61323e7..9159de1 100644
--- a/func_smash.py
+++ b/func_smash.py
@@ -11,14 +11,13 @@ MARKOV_END = -2
 def make_chain(funcs):
     chain = {}
     for func in funcs:
-        co = func.func_code
+        co = func.__code__
         code = co.co_code
         n = len(code)
         i = 0
         lastop = MARKOV_START
         while i < n:
-            c = code[i]
-            op = ord(c)
+            op = ord(code[i])
             i += 1
             if op >= opcode.HAVE_ARGUMENT:
                 oparg = ord(code[i]) + ord(code[i + 1]) * 256
@@ -111,7 +110,7 @@ if __name__ == "__main__":
     print
 
     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:"
     dis.dis(func)
     print