From 4b257d2d82c6323c6bc521017d0649b8cc7b3747 Mon Sep 17 00:00:00 2001 From: Fabrice Desclaux Date: Fri, 9 Feb 2018 10:01:53 +0100 Subject: IRBlock: irblock iterates on its assignblks --- example/ida/graph_ir.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'example/ida/graph_ir.py') diff --git a/example/ida/graph_ir.py b/example/ida/graph_ir.py index ca4ad55c..7e303aac 100644 --- a/example/ida/graph_ir.py +++ b/example/ida/graph_ir.py @@ -40,7 +40,7 @@ def color_irblock(irblock, ir_arch): out = [] lbl = idaapi.COLSTR(str(irblock.label), idaapi.SCOLOR_INSN) out.append(lbl) - for assignblk in irblock.assignblks: + for assignblk in irblock: for dst, src in sorted(assignblk.iteritems()): dst_f = expr2colorstr(ir_arch.arch.regs.all_regs_ids, dst) src_f = expr2colorstr(ir_arch.arch.regs.all_regs_ids, src) @@ -148,7 +148,7 @@ def build_graph(verbose=False, simplify=False): for irb in ir_arch.blocks.itervalues(): irs = [] - for assignblk in irb.assignblks: + for assignblk in irb: new_assignblk = { expr_simp(dst): expr_simp(src) for dst, src in assignblk.iteritems() -- cgit 1.4.1