From 47fb50d611b5ba8eb7e4dc827b522123642ca3b4 Mon Sep 17 00:00:00 2001
From: Fabrice Desclaux <fabrice.desclaux@cea.fr>
Date: Mon, 15 May 2017 21:25:57 +0200
Subject: Example: clean code

---
 example/expression/access_c.py | 116 ++++++++++++++++++++---------------------
 1 file changed, 57 insertions(+), 59 deletions(-)

(limited to 'example/expression/access_c.py')

diff --git a/example/expression/access_c.py b/example/expression/access_c.py
index 8a1e2962..9ba33822 100644
--- a/example/expression/access_c.py
+++ b/example/expression/access_c.py
@@ -127,62 +127,60 @@ class MyCHandler(CHandler):
     exprToAccessC_cls = MyExprToAccessC
 
 
-def test(data):
-    # Digest C informations
-    text = """
-    struct human {
-            unsigned short age;
-            unsigned int height;
-            char name[50];
-    };
-
-    struct ll_human {
-            struct ll_human* next;
-            struct human human;
-    };
-    """
-
-    base_types = CTypeAMD64_unk()
-    types_ast = CAstTypes()
-    types_ast.add_c_decl(text)
-
-    types_mngr = CTypesManagerNotPacked(types_ast, base_types)
-
-    # Analyze binary
-    cont = Container.fallback_container(data, None, addr=0)
-
-    machine = Machine("x86_64")
-    dis_engine, ira = machine.dis_engine, machine.ira
-
-    mdis = dis_engine(cont.bin_stream, symbol_pool=cont.symbol_pool)
-    addr_head = 0
-    blocks = mdis.dis_multibloc(addr_head)
-    lbl_head = mdis.symbol_pool.getby_offset(addr_head)
-
-    ir_arch_a = ira(mdis.symbol_pool)
-    for block in blocks:
-        ir_arch_a.add_bloc(block)
-
-    open('graph_irflow.dot', 'w').write(ir_arch_a.graph.dot())
-
-    # Main function's first argument's type is "struct ll_human*"
-    ptr_llhuman = types_mngr.get_objc(CTypePtr(CTypeStruct('ll_human')))
-    arg0 = ExprId('ptr', 64)
-    ctx = {ir_arch_a.arch.regs.RDI: arg0}
-    expr_types = {arg0.name: ptr_llhuman}
-
-    mychandler = MyCHandler(types_mngr, expr_types)
-
-    for expr in get_funcs_arg0(ctx, ir_arch_a, lbl_head):
-        print "Access:", expr
-        target_types = mychandler.expr_to_types(expr)
-        for target_type in target_types:
-            print '\tType:', target_type
-        c_strs = mychandler.expr_to_c(expr)
-        for c_str in c_strs:
-            print "\tC access:", c_str
-        print
-
-
-if __name__ == '__main__':
-    test(open(sys.argv[1]).read())
+
+data = open(sys.argv[1]).read()
+# Digest C informations
+text = """
+struct human {
+        unsigned short age;
+        unsigned int height;
+        char name[50];
+};
+
+struct ll_human {
+        struct ll_human* next;
+        struct human human;
+};
+"""
+
+base_types = CTypeAMD64_unk()
+types_ast = CAstTypes()
+types_ast.add_c_decl(text)
+
+types_mngr = CTypesManagerNotPacked(types_ast, base_types)
+
+# Analyze binary
+cont = Container.fallback_container(data, None, addr=0)
+
+machine = Machine("x86_64")
+dis_engine, ira = machine.dis_engine, machine.ira
+
+mdis = dis_engine(cont.bin_stream, symbol_pool=cont.symbol_pool)
+addr_head = 0
+blocks = mdis.dis_multibloc(addr_head)
+lbl_head = mdis.symbol_pool.getby_offset(addr_head)
+
+ir_arch_a = ira(mdis.symbol_pool)
+for block in blocks:
+    ir_arch_a.add_bloc(block)
+
+open('graph_irflow.dot', 'w').write(ir_arch_a.graph.dot())
+
+# Main function's first argument's type is "struct ll_human*"
+ptr_llhuman = types_mngr.get_objc(CTypePtr(CTypeStruct('ll_human')))
+arg0 = ExprId('ptr', 64)
+ctx = {ir_arch_a.arch.regs.RDI: arg0}
+expr_types = {arg0.name: ptr_llhuman}
+
+mychandler = MyCHandler(types_mngr, expr_types)
+
+for expr in get_funcs_arg0(ctx, ir_arch_a, lbl_head):
+    print "Access:", expr
+    target_types = mychandler.expr_to_types(expr)
+    for target_type in target_types:
+        print '\tType:', target_type
+    c_strs = mychandler.expr_to_c(expr)
+    for c_str in c_strs:
+        print "\tC access:", c_str
+    print
+
-- 
cgit 1.4.1