about summary refs log tree commit diff stats
path: root/example
diff options
context:
space:
mode:
Diffstat (limited to 'example')
-rwxr-xr-xexample/disas_and_graph.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/example/disas_and_graph.py b/example/disas_and_graph.py
index 07030ea4..c7ec26aa 100755
--- a/example/disas_and_graph.py
+++ b/example/disas_and_graph.py
@@ -162,17 +162,18 @@ for (n,f), ads in dll_dyn_funcs.items():
 
 def my_disasm_callback(ad):
     admode = opmode = u32
+    kargs = {}
     if options.machine_options:
         if options.machine_options in ['u16', 'u32']:
             admode = opmode = options.machine_options
+            kargs = {"admode":admode, "opmode":admode}
         else:
             raise ValueError('bad machine options')
     all_bloc = asmbloc.dis_bloc_all(mnemo, in_str, ad, set(),
                                     symbol_pool=symbol_pool,
-                                    opmode = admode,
-                                    admode = admode,
                                     dontdis_retcall = options.dontdiscallret,
-                                    follow_call = options.followcall)
+                                    follow_call = options.followcall,
+                                    **kargs)
     g = asmbloc.bloc2graph(all_bloc)
     open('graph.txt', 'w').write(g)
     if mnemo == ia32_arch.x86_mn: