1 files changed, 6 insertions, 0 deletions
diff --git a/afl-cmin.bash b/afl-cmin.bash
index 9ac65199..db3e8ae5 100755
--- a/afl-cmin.bash
+++ b/afl-cmin.bash
@@ -215,6 +215,12 @@ if [ ! -f "$TARGET_BIN" -o ! -x "$TARGET_BIN" ]; then
 
 fi
 
+MAPSIZE=`AFL_DUMP_MAP_SIZE=1 "./$TARGET_BIN" 2>/dev/null`
+test -n "$MAPSIZE" && {
+  export AFL_MAP_SIZE=$MAPSIZE
+  echo "[+] Setting AFL_MAP_SIZE=$MAPSIZE"
+}
+
 if [ "$AFL_SKIP_BIN_CHECK" = "" -a "$QEMU_MODE" = "" -a "$FRIDA_MODE" = "" -a "$UNICORN_MODE" = "" ]; then
 
   if ! grep -qF "__AFL_SHM_ID" "$TARGET_BIN"; then