diff options
| author | serpilliere <serpilliere@users.noreply.github.com> | 2020-04-16 00:18:08 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-04-16 00:18:08 +0200 |
| commit | 6b79e8c5ab820222e440d4b96d73dd1258630eb8 (patch) | |
| tree | 493a906042a0b98393b9818006f67fa29de4eb73 /example/ida/graph_ir.py | |
| parent | cdb64fcd6bb2db4b9e70812a66f01b321a74738c (diff) | |
| parent | 2ceadaf75ba55f33bbb6769860d0fd62d0e1020e (diff) | |
| download | focaccia-miasm-6b79e8c5ab820222e440d4b96d73dd1258630eb8.tar.gz focaccia-miasm-6b79e8c5ab820222e440d4b96d73dd1258630eb8.zip | |
Merge pull request #1189 from serpilliere/fix_ida_python
Fix ida graph_ir
Diffstat (limited to 'example/ida/graph_ir.py')
| -rw-r--r-- | example/ida/graph_ir.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/example/ida/graph_ir.py b/example/ida/graph_ir.py index c011b4c3..082fabd7 100644 --- a/example/ida/graph_ir.py +++ b/example/ida/graph_ir.py @@ -125,11 +125,11 @@ class GraphMiasmIR(idaapi.GraphViewer): def OnRefresh(self): self.Clear() addr_id = {} - for irblock in viewvalues(self.ircfg.blocks): + for (loc_key, irblock) in viewitems(self.ircfg.blocks): id_irblock = self.AddNode(color_irblock(irblock, self.ircfg)) - addr_id[irblock] = id_irblock + addr_id[loc_key] = id_irblock - for irblock in viewvalues(self.ircfg.blocks): + for (loc_key, irblock) in viewitems(self.ircfg.blocks): if not irblock: continue all_dst = self.ircfg.dst_trackback(irblock) @@ -138,9 +138,8 @@ class GraphMiasmIR(idaapi.GraphViewer): continue if not dst.loc_key in self.ircfg.blocks: continue - dst_block = self.ircfg.blocks[dst.loc_key] - node1 = addr_id[irblock] - node2 = addr_id[dst_block] + node1 = addr_id[loc_key] + node2 = addr_id[dst.loc_key] self.AddEdge(node1, node2) return True |