1 files changed, 67 insertions, 0 deletions
diff --git a/example/disasm/callback.py b/example/disasm/callback.py
new file mode 100644
index 00000000..6c77023e
--- /dev/null
+++ b/example/disasm/callback.py
@@ -0,0 +1,67 @@
+from miasm2.core.bin_stream import bin_stream_str
+from miasm2.core.asmbloc import asm_label, asm_constraint, expr_is_label
+from miasm2.arch.x86.disasm import dis_x86_32, cb_x86_funcs
+
+
+def cb_x86_callpop(cur_bloc, symbol_pool, *args, **kwargs):
+    """
+    1000: call 1005
+    1005: pop
+
+    Will give:
+
+    1000: push 1005
+    1005: pop
+
+    """
+    # Pattern matching
+    if len(cur_bloc.lines) < 1:
+        return
+    ## We want to match a CALL, always the last line of a basic block
+    last_instr = cur_bloc.lines[-1]
+    if last_instr.name != 'CALL':
+        return
+    ## The destination must be a label
+    dst = last_instr.args[0]
+    if not expr_is_label(dst):
+        return
+    ## The destination must be the next instruction
+    if dst.name.offset != last_instr.offset + last_instr.l:
+        return
+
+    # Update instruction instance
+    last_instr.name = 'PUSH'
+
+    # Update next blocks to process in the disassembly engine
+    cur_bloc.bto.clear()
+    cur_bloc.add_cst(dst.name.offset, asm_constraint.c_next, symbol_pool)
+
+
+# Prepare a tiny shellcode
+shellcode = ''.join(["\xe8\x00\x00\x00\x00", # CALL $
+                     "X",                    # POP EAX
+                     "\xc3",                 # RET
+                     ])
+bin_stream = bin_stream_str(shellcode)
+mdis = dis_x86_32(bin_stream)
+
+print "Without callback:\n"
+blocks = mdis.dis_multibloc(0)
+print "\n".join(str(block) for block in blocks)
+
+# Enable callback
+cb_x86_funcs.append(cb_x86_callpop)
+## Other method:
+## mdis.dis_bloc_callback = cb_x86_callpop
+
+# Clean disassembly cache
+mdis.job_done.clear()
+
+print "=" * 40
+print "With callback:\n"
+blocks_after = mdis.dis_multibloc(0)
+print "\n".join(str(block) for block in blocks_after)
+
+# Ensure the callback has been called
+assert blocks[0].lines[0].name == "CALL"
+assert blocks_after[0].lines[0].name == "PUSH"