From f9a8b60b3b3eb508234e2303c208acdccc50333e Mon Sep 17 00:00:00 2001 From: vanhauser-thc Date: Mon, 7 Oct 2024 16:44:24 +0200 Subject: add LLVMFuzzerTestOneInput -1 support for non-asan --- utils/aflpp_driver/aflpp_driver.c | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'utils/aflpp_driver/aflpp_driver.c') diff --git a/utils/aflpp_driver/aflpp_driver.c b/utils/aflpp_driver/aflpp_driver.c index 6e992266..9b79ef9b 100644 --- a/utils/aflpp_driver/aflpp_driver.c +++ b/utils/aflpp_driver/aflpp_driver.c @@ -400,7 +400,7 @@ __attribute__((weak)) int LLVMFuzzerRunDriver( size_t prev_length = 0; // for speed only insert asan functions if the target is linked with asan - if (__asan_region_is_poisoned) { + if (unlikely(__asan_region_is_poisoned)) { while (__afl_persistent_loop(N)) { @@ -437,7 +437,12 @@ __attribute__((weak)) int LLVMFuzzerRunDriver( while (__afl_persistent_loop(N)) { - callback(__afl_fuzz_ptr, *__afl_fuzz_len); + if (unlikely(callback(__afl_fuzz_ptr, *__afl_fuzz_len) == -1)) { + + memset(__afl_area_ptr, 0, __afl_map_size); + __afl_area_ptr[0] = 1; + + } } -- cgit 1.4.1