about summary refs log tree commit diff stats
path: root/example/ida/graph_ir.py
diff options
context:
space:
mode:
authorCamille Mougey <commial@gmail.com>2018-06-19 17:53:45 +0200
committerGitHub <noreply@github.com>2018-06-19 17:53:45 +0200
commit0adddb12a7410c4bbdd6ee79252e9bf05ca22818 (patch)
tree92facadca71fc07d7760c60d2ca877774c73579b /example/ida/graph_ir.py
parent8f5ca332780cf9e08761060e9903bc085dbc8430 (diff)
parentf1b32ba2b8db22a5f92a2734af55e9a5dba2271e (diff)
downloadmiasm-0adddb12a7410c4bbdd6ee79252e9bf05ca22818.tar.gz
miasm-0adddb12a7410c4bbdd6ee79252e9bf05ca22818.zip
Merge pull request #770 from serpilliere/ida_guess_machine_addr
Example/ida: use addr to guess arch
Diffstat (limited to 'example/ida/graph_ir.py')
-rw-r--r--example/ida/graph_ir.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/example/ida/graph_ir.py b/example/ida/graph_ir.py
index 370500e5..97d30851 100644
--- a/example/ida/graph_ir.py
+++ b/example/ida/graph_ir.py
@@ -98,7 +98,9 @@ class GraphMiasmIR(idaapi.GraphViewer):
 
 
 def build_graph(verbose=False, simplify=False):
-    machine = guess_machine()
+    start_addr = idc.ScreenEA()
+
+    machine = guess_machine(addr=start_addr)
     mn, dis_engine, ira = machine.mn, machine.dis_engine, machine.ira
 
     if verbose:
@@ -125,17 +127,16 @@ def build_graph(verbose=False, simplify=False):
 
     if verbose:
         print "start disasm"
-    addr = idc.ScreenEA()
     if verbose:
         print hex(addr)
 
-    asmcfg = mdis.dis_multiblock(addr)
+    asmcfg = mdis.dis_multiblock(start_addr)
 
     if verbose:
         print "generating graph"
         open('asm_flow.dot', 'w').write(asmcfg.dot())
 
-        print "generating IR... %x" % addr
+        print "generating IR... %x" % start_addr
 
     for block in asmcfg.blocks:
         if verbose:
@@ -144,7 +145,7 @@ def build_graph(verbose=False, simplify=False):
         ir_arch.add_block(block)
 
     if verbose:
-        print "IR ok... %x" % addr
+        print "IR ok... %x" % start_addr
 
     for irb in ir_arch.blocks.itervalues():
         irs = []