ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-19 04:58:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

tiny changes in custom mut api

Browse Source
This commit is contained in:
Dominik Maier
2020-03-28 11:55:10 +01:00
parent f370ef38c4
commit ab5e0703da
6 changed files with 39 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/afl-fuzz-mutators.c
View File

@ -229,7 +229,7 @@ u8 trim_case_custom(afl_state_t *afl, struct queue_entry *q, u8 *in_buf) {
size_t retlen = afl->mutator->afl_custom_trim(afl->mutator->data, &retbuf);
if (unlikely(retlen < 0 || !retbuf))
if (unlikely(!retbuf))
FATAL("custom_trim failed (ret %zd)", retlen);
else if (unlikely(retlen > orig_len))
FATAL(

6
src/afl-fuzz-one.c
View File

@ -1618,8 +1618,8 @@ custom_mutator_stage:
afl->mutator->data, out_buf, len, &mutated_buf, new_buf, target->len,
max_seed_size);
if (unlikely(mutated_size < 0))
FATAL("custom_fuzz returned %zd", mutated_size);
if (unlikely(!mutated_buf))
FATAL("Error in custom_fuzz. Size returned: %zd", mutated_size);
if (mutated_size > len) afl->out_size = mutated_size;
@ -1734,7 +1734,7 @@ havoc_stage:
u8 * custom_havoc_buf = NULL;
size_t new_len = afl->mutator->afl_custom_havoc_mutation(
afl->mutator->data, out_buf, temp_len, &custom_havoc_buf, MAX_FILE);
if (unlikely(new_len < 0))
if (unlikely(!custom_havoc_buf))
FATAL("Error in custom_havoc (return %zd)", new_len);
if (likely(new_len > 0 && custom_havoc_buf)) {

2
src/afl-fuzz-python.c
View File

@ -103,7 +103,7 @@ size_t fuzz_py(void *py_mutator, u8 *buf, size_t buf_size, u8 **out_buf,
} else {
PyErr_Print();
FATAL("Call failed");
FATAL("python custom fuzz: call failed");
}

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

@ -221,7 +221,7 @@ void write_to_testcase(afl_state_t *afl, void *mem, u32 len) {
size_t new_size = afl->mutator->afl_custom_pre_save(afl->mutator->data, mem,
len, &new_buf);
if (unlikely(new_size <= 0 || !new_buf))
if (unlikely(!new_buf))
FATAL("Custom_pre_save failed (ret: %lu)", (long unsigned)new_size);
/* everything as planned. use the new data. */