diff options
Diffstat (limited to 'src/afl-fuzz-one.c')
-rw-r--r-- | src/afl-fuzz-one.c | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/afl-fuzz-one.c b/src/afl-fuzz-one.c index f1efe2df..24ea76f3 100644 --- a/src/afl-fuzz-one.c +++ b/src/afl-fuzz-one.c @@ -449,7 +449,7 @@ u8 fuzz_one_original(char** argv) { * TRIMMING * ************/ - if (!dumb_mode && !queue_cur->trim_done && !custom_mutator && !disable_trim) { + if (!dumb_mode && !queue_cur->trim_done && !disable_trim) { u8 res = trim_case(argv, queue_cur, in_buf); @@ -484,7 +484,7 @@ u8 fuzz_one_original(char** argv) { // custom_stage: // not used - yet - if (custom_mutator) { + if (mutator->afl_custom_fuzz) { stage_short = "custom"; stage_name = "custom mutator"; @@ -499,8 +499,9 @@ u8 fuzz_one_original(char** argv) { for (stage_cur = 0; stage_cur < stage_max; ++stage_cur) { size_t orig_size = (size_t)len; - size_t mutated_size = custom_mutator(in_buf, orig_size, mutated_buf, - max_seed_size, UR(UINT32_MAX)); + size_t mutated_size = mutator->afl_custom_fuzz(in_buf, orig_size, + mutated_buf, max_seed_size, + UR(UINT32_MAX)); if (mutated_size > 0) { out_buf = ck_realloc(out_buf, mutated_size); |