ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-15 03:18:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed AFL_POST_PROCESS_KEEP_ORIGINAL for version 4.07a

Browse Source
This commit is contained in:
Maik Betka
2023-04-21 16:46:15 +02:00
parent 7101ffa1ae
commit 779a72ef8c
1 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/afl-fuzz-run.c
View File

@ -133,14 +133,22 @@ write_to_testcase(afl_state_t *afl, void **mem, u32 len, u32 fix) {
} }
if (new_mem != *mem && new_mem != NULL && new_size > 0 if (new_mem != *mem && new_mem != NULL && new_size > 0) {
&& !afl->afl_env.afl_post_process_keep_original) {
u8 *new_buf = afl_realloc(AFL_BUF_PARAM(out_scratch), new_size); new_buf = afl_realloc(AFL_BUF_PARAM(out_scratch), new_size);
if (unlikely(!new_buf)) { PFATAL("alloc"); } if (unlikely(!new_buf)) { PFATAL("alloc"); }
*mem = new_buf; memcpy(new_buf, new_mem, new_size);
memcpy(*mem, new_mem, new_size);
afl_swap_bufs(AFL_BUF_PARAM(out), AFL_BUF_PARAM(out_scratch)); /* if AFL_POST_PROCESS_KEEP_ORIGINAL is set then save the original memory
prior post-processing in new_mem to restore it later */
if (unlikely(afl->afl_env.afl_post_process_keep_original)) {
new_mem = *mem;
}
*mem = new_buf;
afl_swap_bufs(AFL_BUF_PARAM(out), AFL_BUF_PARAM(out_scratch));
} }
@ -162,17 +170,19 @@ write_to_testcase(afl_state_t *afl, void **mem, u32 len, u32 fix) {
if (likely(!sent)) { if (likely(!sent)) {
/* everything as planned. use the potentially new data. */ /* everything as planned. use the potentially new data. */
afl_fsrv_write_to_testcase(&afl->fsrv, *mem, new_size);
if (likely(!afl->afl_env.afl_post_process_keep_original)) { if (likely(!afl->afl_env.afl_post_process_keep_original)) {
afl_fsrv_write_to_testcase(&afl->fsrv, *mem, new_size); len = new_size;
} else { } else {
afl_fsrv_write_to_testcase(&afl->fsrv, new_mem, new_size); /* restore the original memory which was saved in new_mem */
*mem = new_mem;
afl_swap_bufs(AFL_BUF_PARAM(out), AFL_BUF_PARAM(out_scratch));
} }
len = new_size;
} }