diff options
Diffstat (limited to 'python/examples')
| -rw-r--r-- | python/examples/args_render_layer.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/python/examples/args_render_layer.py b/python/examples/args_render_layer.py index bac0889e..4941bbd1 100644 --- a/python/examples/args_render_layer.py +++ b/python/examples/args_render_layer.py @@ -97,7 +97,8 @@ def get_param_sites(mlil: MediumLevelILFunction) -> Mapping[LowLevelILInstructio if llil not in all_def_sites: all_def_sites[llil] = [] else: - print(f"got two at {llil.instr_index} @ {llil.address:#x} -> {call_site.address:#x}") + # print(f"got two at {llil.instr_index} @ {llil.address:#x} -> {call_site.address:#x}") + pass all_def_sites[llil].append((call_site, param_idx)) return all_def_sites @@ -196,6 +197,8 @@ class ArgumentsRenderLayer(RenderLayer): block: BasicBlock, lines: List['DisassemblyTextLine'] ): + if block.function.mlil_if_available is None: + return lines # Break this out into a helper so we don't have to write it twice renderer = DisassemblyTextRenderer(block.function) return apply_to_lines(lines, lambda line: block.function.get_llil_at(line.address), renderer) @@ -205,6 +208,8 @@ class ArgumentsRenderLayer(RenderLayer): block: BasicBlock, lines: List['DisassemblyTextLine'] ): + if block.function.mlil_if_available is None: + return lines # Break this out into a helper so we don't have to write it twice renderer = DisassemblyTextRenderer(block.function) return apply_to_lines(lines, lambda line: line.il_instruction, renderer) |