diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/aflpp_driver/aflpp_driver.c | 9 | ||||
| -rw-r--r-- | utils/libdislocator/libdislocator.so.c | 3 |
2 files changed, 8 insertions, 4 deletions
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; + + } } diff --git a/utils/libdislocator/libdislocator.so.c b/utils/libdislocator/libdislocator.so.c index f41491b1..b125f628 100644 --- a/utils/libdislocator/libdislocator.so.c +++ b/utils/libdislocator/libdislocator.so.c @@ -580,11 +580,10 @@ __attribute__((constructor)) void __dislocator_init(void) { __attribute__((destructor)) void __dislocator_fini(void) { - alloc_canary = ALLOC_CANARY; // restore to default canary value + alloc_canary = ALLOC_CANARY; // restore to default canary value } - /* NetBSD fault handler specific api subset */ void (*esetfunc(void (*fn)(int, const char *, ...)))(int, const char *, ...) { |