diff options
| author | Fabrice Desclaux <fabrice.desclaux@cea.fr> | 2020-04-15 15:50:04 +0200 |
|---|---|---|
| committer | Fabrice Desclaux <fabrice.desclaux@cea.fr> | 2020-04-15 15:50:04 +0200 |
| commit | 2ceadaf75ba55f33bbb6769860d0fd62d0e1020e (patch) | |
| tree | 4529a4ce79cbb7b3c0bff4e569e3c7627c58d30d /example | |
| parent | 5b33db2100c1c3cad3be9fbe12c2e9286d007323 (diff) | |
| download | focaccia-miasm-2ceadaf75ba55f33bbb6769860d0fd62d0e1020e.tar.gz focaccia-miasm-2ceadaf75ba55f33bbb6769860d0fd62d0e1020e.zip | |
Fix ida graph_ir
Diffstat (limited to 'example')
| -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 |