about summary refs log tree commit diff
path: root/src/afl-fuzz-one.c
diff options
context:
space:
mode:
authorDominik Maier <domenukk@gmail.com>2020-03-28 11:55:10 +0100
committerDominik Maier <domenukk@gmail.com>2020-04-01 13:10:06 +0200
commitab5e0703dab404a77cd4726f720db99bed1a0c71 (patch)
treeaf02917d3857f49d804ecd5a1d29374ff62d4209 /src/afl-fuzz-one.c
parentf370ef38c47eb9243c5ca06b98948e33cf5347b3 (diff)
downloadafl++-ab5e0703dab404a77cd4726f720db99bed1a0c71.tar.gz
tiny changes in custom mut api
Diffstat (limited to 'src/afl-fuzz-one.c')
-rw-r--r--src/afl-fuzz-one.c6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/afl-fuzz-one.c b/src/afl-fuzz-one.c
index 8dfafb7b..b1bbad0a 100644
--- a/src/afl-fuzz-one.c
+++ b/src/afl-fuzz-one.c
@@ -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)) {