diff options
Diffstat (limited to 'miasm2/analysis/cst_propag.py')
| -rw-r--r-- | miasm2/analysis/cst_propag.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/miasm2/analysis/cst_propag.py b/miasm2/analysis/cst_propag.py index d55d7e60..2a439ccc 100644 --- a/miasm2/analysis/cst_propag.py +++ b/miasm2/analysis/cst_propag.py @@ -139,9 +139,10 @@ def compute_cst_propagation_states(ir_arch, init_addr, init_infos): if (lbl, state) in done: continue done.add((lbl, state)) - symbexec_engine = SymbExecState(ir_arch, state) + if lbl not in ir_arch.blocks: + continue - assert lbl in ir_arch.blocks + symbexec_engine = SymbExecState(ir_arch, state) addr = symbexec_engine.emul_ir_block(lbl) symbexec_engine.del_mem_above_stack(ir_arch.sp) @@ -173,6 +174,8 @@ def propagate_cst_expr(ir_arch, addr, init_infos): states = compute_cst_propagation_states(ir_arch, addr, init_infos) cst_propag_link = {} for lbl, state in states.iteritems(): + if lbl not in ir_arch.blocks: + continue symbexec = SymbExecStateFix(ir_arch, state, cst_propag_link) symbexec.emulbloc(ir_arch.blocks[lbl]) return cst_propag_link |