commit 111b680ee159380345472c047d4adaa6430af738
author Matt Morehouse <mascasa@google.com> Tue Jul 10 19:58:42 2018 +0000
committer Copybara-Service <copybara-worker@google.com> Thu Mar 25 23:17:32 2021 -0700
tree 8aa96af86447ab284322043928fa7aa91c9c41be
parent 13675a6fdceca4a1b4e75307300c37ccbede84c4

Revert "[Fuzzer] Afl driver changing iterations handling"

This reverts rL334510 due to breakage of afl_driver's command line
interface.

Patch By: Jonathan Metzman

Differential Revision: https://reviews.llvm.org/D49141

llvm-svn: 336719
GitOrigin-RevId: 0e904e8806fe18dbed895e74901cd25262a611a8

afl/afl_driver.cpp

diff --git a/afl/afl_driver.cpp b/afl/afl_driver.cpp
index f3435ff..fa494c0 100644
--- a/afl/afl_driver.cpp
+++ b/afl/afl_driver.cpp
@@ -59,7 +59,6 @@
 #include <sys/resource.h>
 #include <sys/time.h>
 #include <unistd.h>
-#include <limits.h>
 
 #include <fstream>
 #include <iostream>
@@ -306,18 +305,6 @@
   return 0;
 }
 
-static void set_iterations(int *N, const char *arg) {
-  char *next_char;
-  long NL = strtol(arg, &next_char, 10);
-  if (NL < 1 || NL > INT_MAX || *next_char != '\0') {
-    fprintf(stderr, "WARNING: iterations invalid `%s`\n",
-        arg);
-    ::exit(-1);
-  }
-
-  *N = static_cast<int>(NL);
-}
-
 int main(int argc, char **argv) {
   fprintf(stderr,
       "======================= INFO =========================\n"
@@ -344,12 +331,11 @@
 
   int N = 1000;
   if (argc == 2 && argv[1][0] == '-')
-    set_iterations(&N, argv[1] + 1);
-  else if(argc == 2) {
-    fprintf(stderr, "WARNING: using the deprecated call style `%s %d`\n",
-        argv[0], N);
-    set_iterations(&N, argv[1]);
-  } else if (argc > 1)
+      N = atoi(argv[1] + 1);
+  else if(argc == 2 && (N = atoi(argv[1])) > 0)
+      fprintf(stderr, "WARNING: using the deprecated call style `%s %d`\n",
+              argv[0], N);
+  else if (argc > 1)
     return ExecuteFilesOnyByOne(argc, argv);
 
   assert(N > 0);