tiny changes in custom mut api

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

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)) {