about summary refs log tree commit diff
path: root/afl-cmin
diff options
context:
space:
mode:
authorh1994st <h1994st@gmail.com>2020-03-06 16:28:26 -0500
committerh1994st <h1994st@gmail.com>2020-03-06 16:28:26 -0500
commita10a3f2fa75f16bf7781a3c02cd23eab7164cff1 (patch)
tree653138ebd36b80fd9fd3faa31eedd8666e185135 /afl-cmin
parent9e5c4973eb8f9b2f007bb1fe10976a4634c0ea6a (diff)
parent2287534ec6dd68b06a5052caa4ab3305d15861ec (diff)
downloadafl++-a10a3f2fa75f16bf7781a3c02cd23eab7164cff1.tar.gz
Merge branch 'master' of https://github.com/vanhauser-thc/AFLplusplus
Diffstat (limited to 'afl-cmin')
-rwxr-xr-xafl-cmin13
1 files changed, 11 insertions, 2 deletions
diff --git a/afl-cmin b/afl-cmin
index 28d8c746..ea57015e 100755
--- a/afl-cmin
+++ b/afl-cmin
@@ -397,10 +397,19 @@ BEGIN {
   cur = 0;
   if (!stdin_file) {
     print "    Processing "in_count" files (forkserver mode)..."
-    system( "AFL_CMIN_ALLOW_ANY=1 \""showmap"\" -m "mem_limit" -t "timeout" -o \""trace_dir"\" -Z "extra_par" -i \""in_dir"\" -- \""target_bin"\" "prog_args_string)
+    retval = system( "AFL_CMIN_ALLOW_ANY=1 \""showmap"\" -m "mem_limit" -t "timeout" -o \""trace_dir"\" -Z "extra_par" -i \""in_dir"\" -- \""target_bin"\" "prog_args_string)
   } else {
     print "    Processing "in_count" files (forkserver mode)..."
-    system( "AFL_CMIN_ALLOW_ANY=1 \""showmap"\" -m "mem_limit" -t "timeout" -o \""trace_dir"\" -Z "extra_par" -i \""in_dir"\" -- \""target_bin"\" "prog_args_string" </dev/null")
+    retval = system( "AFL_CMIN_ALLOW_ANY=1 \""showmap"\" -m "mem_limit" -t "timeout" -o \""trace_dir"\" -Z "extra_par" -i \""in_dir"\" -- \""target_bin"\" "prog_args_string" </dev/null")
+  }
+
+  if (retval) {
+    print "[!]Exit code != 0 received from afl-showmap, terminating..."
+
+    if (!ENVIRON["AFL_KEEP_TRACES"]) {
+      system("rm -rf "trace_dir" 2>/dev/null")
+    }
+    exit retval
   }
 
   #######################################################