about summary refs log tree commit diff
path: root/examples/aflpp_driver/aflpp_qemu_driver_hook.c
diff options
context:
space:
mode:
Diffstat (limited to 'examples/aflpp_driver/aflpp_qemu_driver_hook.c')
-rw-r--r--examples/aflpp_driver/aflpp_qemu_driver_hook.c21
1 files changed, 21 insertions, 0 deletions
diff --git a/examples/aflpp_driver/aflpp_qemu_driver_hook.c b/examples/aflpp_driver/aflpp_qemu_driver_hook.c
new file mode 100644
index 00000000..17a84668
--- /dev/null
+++ b/examples/aflpp_driver/aflpp_qemu_driver_hook.c
@@ -0,0 +1,21 @@
+#include <stdint.h>
+#include <string.h>
+
+#define g2h(x) ((void *)((unsigned long)(x) + guest_base))
+
+#define REGS_RDI 7
+#define REGS_RSI 6
+
+void afl_persistent_hook(uint64_t *regs, uint64_t guest_base,
+                         uint8_t* input_buf, uint32_t input_len) {
+
+  memcpy(g2h(regs[REGS_RDI]), input_buf, input_len); 
+  regs[REGS_RSI] = input_len;
+
+}
+
+int afl_persistent_hook_init(void) {
+
+  return 1;
+
+}