From d276208c8175932936bfdf59749caa310cce616d Mon Sep 17 00:00:00 2001
From: Darren Ldl <darrenldldev@gmail.com>
Date: Fri, 5 Feb 2021 15:29:06 +1100
Subject: [PATCH] Updated fuzz/run.sh with an upgraded version

---
 fuzz/run.sh | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/fuzz/run.sh b/fuzz/run.sh
index 8bdd3b88..a154a58e 100755
--- a/fuzz/run.sh
+++ b/fuzz/run.sh
@@ -2,16 +2,20 @@
 
 script_dir=$(dirname $(readlink -f "$0"))
 
-echo "Building"
+skip_build=$2
 
-dune build @all
+if [[ "$skip_build" != "skip_build" ]]; then
+  echo "Building"
+
+  dune build @all
+fi
 
 if [[ "$1" == "" ]]; then
   echo "Please enter a fuzzing test to run"
   exit 1
 fi
 
-name=$(echo "$1" | sed 's/\.exe$//')
+name=$(echo "$1" | sed 's/\.exe$//' | sed 's/\.ml$//')
 
 echo "Creating input directory"
 
@@ -23,6 +27,11 @@ mkdir -p "$input_dir"
 
 echo "abcd" > "$input_dir"/dummy
 
-mkdir -p "$output_dir"
+if [ -d "$output_dir" ]; then
+  afl-fuzz -t 1000 -i -            -o "$output_dir" "$script_dir"/../_build/default/fuzz/"$name".exe @@
+else
+  mkdir -p "$output_dir"
+
+  afl-fuzz -t 1000 -i "$input_dir" -o "$output_dir" "$script_dir"/../_build/default/fuzz/"$name".exe @@
+fi
 
-afl-fuzz -t 1000 -i "$input_dir" -o "$output_dir" "$script_dir"/../_build/default/fuzz/"$name".exe @@