persistent qemu retaddr offset

This commit is contained in:
Andrea Fioraldi 2019-09-22 19:38:53 +02:00
parent 63677bb1f9
commit 17bb51756f
3 changed files with 6 additions and 1 deletions

View File

@ -61,6 +61,7 @@ extern target_long persistent_stack_offset;
extern unsigned char persistent_first_pass;
extern unsigned char persistent_save_gpr;
extern target_ulong persistent_saved_gpr[CPU_NB_REGS];
extern int persisent_retaddr_offset;
extern __thread abi_ulong afl_prev_loc;

View File

@ -90,6 +90,7 @@ target_long persistent_stack_offset;
unsigned char persistent_first_pass;
unsigned char persistent_save_gpr;
target_ulong persistent_saved_gpr[CPU_NB_REGS];
int persisent_retaddr_offset;
/* Instrumentation ratio: */
@ -219,6 +220,9 @@ static void afl_setup(void) {
if (getenv("AFL_QEMU_PERSISTENT_GPR"))
persistent_save_gpr = 1;
if (getenv("AFL_QEMU_PERSISTENT_RETADDR_OFFSET"))
persisent_retaddr_offset = strtoll(getenv("AFL_QEMU_PERSISTENT_RETADDR_OFFSET"), NULL, 16);
if (getenv("AFL_QEMU_PERSISTENT_CNT"))
afl_persistent_cnt = strtoll(getenv("AFL_QEMU_PERSISTENT_CNT"), NULL, 16);
else

View File

@ -195,7 +195,7 @@ static void afl_gen_compcov(target_ulong cur_loc, TCGv_i64 arg1, TCGv_i64 arg2,
if (afl_persistent_ret_addr == 0) { \
\
TCGv_ptr paddr = tcg_const_ptr(afl_persistent_addr); \
tcg_gen_st_tl(paddr, cpu_regs[R_ESP], 0); \
tcg_gen_st_tl(paddr, cpu_regs[R_ESP], persisent_retaddr_offset); \
\
} \
tcg_gen_afl_call0(&afl_persistent_loop); \