xref: /netbsd-src/sys/external/bsd/compiler_rt/dist/lib/fuzzer/FuzzerDriver.cpp (revision a7c257b03e4462df2b1020128fb82716512d7856)
1*a7c257b0Skamil //===- FuzzerDriver.cpp - FuzzerDriver function and flags -----------------===//
2*a7c257b0Skamil //
3*a7c257b0Skamil //                     The LLVM Compiler Infrastructure
4*a7c257b0Skamil //
5*a7c257b0Skamil // This file is distributed under the University of Illinois Open Source
6*a7c257b0Skamil // License. See LICENSE.TXT for details.
7*a7c257b0Skamil //
8*a7c257b0Skamil //===----------------------------------------------------------------------===//
9*a7c257b0Skamil // FuzzerDriver and flag parsing.
10*a7c257b0Skamil //===----------------------------------------------------------------------===//
11*a7c257b0Skamil 
12*a7c257b0Skamil #include "FuzzerCommand.h"
13*a7c257b0Skamil #include "FuzzerCorpus.h"
14*a7c257b0Skamil #include "FuzzerIO.h"
15*a7c257b0Skamil #include "FuzzerInterface.h"
16*a7c257b0Skamil #include "FuzzerInternal.h"
17*a7c257b0Skamil #include "FuzzerMutate.h"
18*a7c257b0Skamil #include "FuzzerRandom.h"
19*a7c257b0Skamil #include "FuzzerShmem.h"
20*a7c257b0Skamil #include "FuzzerTracePC.h"
21*a7c257b0Skamil #include <algorithm>
22*a7c257b0Skamil #include <atomic>
23*a7c257b0Skamil #include <chrono>
24*a7c257b0Skamil #include <cstdlib>
25*a7c257b0Skamil #include <cstring>
26*a7c257b0Skamil #include <mutex>
27*a7c257b0Skamil #include <string>
28*a7c257b0Skamil #include <thread>
29*a7c257b0Skamil 
30*a7c257b0Skamil // This function should be present in the libFuzzer so that the client
31*a7c257b0Skamil // binary can test for its existence.
__libfuzzer_is_present()32*a7c257b0Skamil extern "C" __attribute__((used)) void __libfuzzer_is_present() {}
33*a7c257b0Skamil 
34*a7c257b0Skamil namespace fuzzer {
35*a7c257b0Skamil 
36*a7c257b0Skamil // Program arguments.
37*a7c257b0Skamil struct FlagDescription {
38*a7c257b0Skamil   const char *Name;
39*a7c257b0Skamil   const char *Description;
40*a7c257b0Skamil   int   Default;
41*a7c257b0Skamil   int   *IntFlag;
42*a7c257b0Skamil   const char **StrFlag;
43*a7c257b0Skamil   unsigned int *UIntFlag;
44*a7c257b0Skamil };
45*a7c257b0Skamil 
46*a7c257b0Skamil struct {
47*a7c257b0Skamil #define FUZZER_DEPRECATED_FLAG(Name)
48*a7c257b0Skamil #define FUZZER_FLAG_INT(Name, Default, Description) int Name;
49*a7c257b0Skamil #define FUZZER_FLAG_UNSIGNED(Name, Default, Description) unsigned int Name;
50*a7c257b0Skamil #define FUZZER_FLAG_STRING(Name, Description) const char *Name;
51*a7c257b0Skamil #include "FuzzerFlags.def"
52*a7c257b0Skamil #undef FUZZER_DEPRECATED_FLAG
53*a7c257b0Skamil #undef FUZZER_FLAG_INT
54*a7c257b0Skamil #undef FUZZER_FLAG_UNSIGNED
55*a7c257b0Skamil #undef FUZZER_FLAG_STRING
56*a7c257b0Skamil } Flags;
57*a7c257b0Skamil 
58*a7c257b0Skamil static const FlagDescription FlagDescriptions [] {
59*a7c257b0Skamil #define FUZZER_DEPRECATED_FLAG(Name)                                           \
60*a7c257b0Skamil   {#Name, "Deprecated; don't use", 0, nullptr, nullptr, nullptr},
61*a7c257b0Skamil #define FUZZER_FLAG_INT(Name, Default, Description)                            \
62*a7c257b0Skamil   {#Name, Description, Default, &Flags.Name, nullptr, nullptr},
63*a7c257b0Skamil #define FUZZER_FLAG_UNSIGNED(Name, Default, Description)                       \
64*a7c257b0Skamil   {#Name,   Description, static_cast<int>(Default),                            \
65*a7c257b0Skamil    nullptr, nullptr, &Flags.Name},
66*a7c257b0Skamil #define FUZZER_FLAG_STRING(Name, Description)                                  \
67*a7c257b0Skamil   {#Name, Description, 0, nullptr, &Flags.Name, nullptr},
68*a7c257b0Skamil #include "FuzzerFlags.def"
69*a7c257b0Skamil #undef FUZZER_DEPRECATED_FLAG
70*a7c257b0Skamil #undef FUZZER_FLAG_INT
71*a7c257b0Skamil #undef FUZZER_FLAG_UNSIGNED
72*a7c257b0Skamil #undef FUZZER_FLAG_STRING
73*a7c257b0Skamil };
74*a7c257b0Skamil 
75*a7c257b0Skamil static const size_t kNumFlags =
76*a7c257b0Skamil     sizeof(FlagDescriptions) / sizeof(FlagDescriptions[0]);
77*a7c257b0Skamil 
78*a7c257b0Skamil static Vector<std::string> *Inputs;
79*a7c257b0Skamil static std::string *ProgName;
80*a7c257b0Skamil 
PrintHelp()81*a7c257b0Skamil static void PrintHelp() {
82*a7c257b0Skamil   Printf("Usage:\n");
83*a7c257b0Skamil   auto Prog = ProgName->c_str();
84*a7c257b0Skamil   Printf("\nTo run fuzzing pass 0 or more directories.\n");
85*a7c257b0Skamil   Printf("%s [-flag1=val1 [-flag2=val2 ...] ] [dir1 [dir2 ...] ]\n", Prog);
86*a7c257b0Skamil 
87*a7c257b0Skamil   Printf("\nTo run individual tests without fuzzing pass 1 or more files:\n");
88*a7c257b0Skamil   Printf("%s [-flag1=val1 [-flag2=val2 ...] ] file1 [file2 ...]\n", Prog);
89*a7c257b0Skamil 
90*a7c257b0Skamil   Printf("\nFlags: (strictly in form -flag=value)\n");
91*a7c257b0Skamil   size_t MaxFlagLen = 0;
92*a7c257b0Skamil   for (size_t F = 0; F < kNumFlags; F++)
93*a7c257b0Skamil     MaxFlagLen = std::max(strlen(FlagDescriptions[F].Name), MaxFlagLen);
94*a7c257b0Skamil 
95*a7c257b0Skamil   for (size_t F = 0; F < kNumFlags; F++) {
96*a7c257b0Skamil     const auto &D = FlagDescriptions[F];
97*a7c257b0Skamil     if (strstr(D.Description, "internal flag") == D.Description) continue;
98*a7c257b0Skamil     Printf(" %s", D.Name);
99*a7c257b0Skamil     for (size_t i = 0, n = MaxFlagLen - strlen(D.Name); i < n; i++)
100*a7c257b0Skamil       Printf(" ");
101*a7c257b0Skamil     Printf("\t");
102*a7c257b0Skamil     Printf("%d\t%s\n", D.Default, D.Description);
103*a7c257b0Skamil   }
104*a7c257b0Skamil   Printf("\nFlags starting with '--' will be ignored and "
105*a7c257b0Skamil             "will be passed verbatim to subprocesses.\n");
106*a7c257b0Skamil }
107*a7c257b0Skamil 
FlagValue(const char * Param,const char * Name)108*a7c257b0Skamil static const char *FlagValue(const char *Param, const char *Name) {
109*a7c257b0Skamil   size_t Len = strlen(Name);
110*a7c257b0Skamil   if (Param[0] == '-' && strstr(Param + 1, Name) == Param + 1 &&
111*a7c257b0Skamil       Param[Len + 1] == '=')
112*a7c257b0Skamil       return &Param[Len + 2];
113*a7c257b0Skamil   return nullptr;
114*a7c257b0Skamil }
115*a7c257b0Skamil 
116*a7c257b0Skamil // Avoid calling stol as it triggers a bug in clang/glibc build.
MyStol(const char * Str)117*a7c257b0Skamil static long MyStol(const char *Str) {
118*a7c257b0Skamil   long Res = 0;
119*a7c257b0Skamil   long Sign = 1;
120*a7c257b0Skamil   if (*Str == '-') {
121*a7c257b0Skamil     Str++;
122*a7c257b0Skamil     Sign = -1;
123*a7c257b0Skamil   }
124*a7c257b0Skamil   for (size_t i = 0; Str[i]; i++) {
125*a7c257b0Skamil     char Ch = Str[i];
126*a7c257b0Skamil     if (Ch < '0' || Ch > '9')
127*a7c257b0Skamil       return Res;
128*a7c257b0Skamil     Res = Res * 10 + (Ch - '0');
129*a7c257b0Skamil   }
130*a7c257b0Skamil   return Res * Sign;
131*a7c257b0Skamil }
132*a7c257b0Skamil 
ParseOneFlag(const char * Param)133*a7c257b0Skamil static bool ParseOneFlag(const char *Param) {
134*a7c257b0Skamil   if (Param[0] != '-') return false;
135*a7c257b0Skamil   if (Param[1] == '-') {
136*a7c257b0Skamil     static bool PrintedWarning = false;
137*a7c257b0Skamil     if (!PrintedWarning) {
138*a7c257b0Skamil       PrintedWarning = true;
139*a7c257b0Skamil       Printf("INFO: libFuzzer ignores flags that start with '--'\n");
140*a7c257b0Skamil     }
141*a7c257b0Skamil     for (size_t F = 0; F < kNumFlags; F++)
142*a7c257b0Skamil       if (FlagValue(Param + 1, FlagDescriptions[F].Name))
143*a7c257b0Skamil         Printf("WARNING: did you mean '%s' (single dash)?\n", Param + 1);
144*a7c257b0Skamil     return true;
145*a7c257b0Skamil   }
146*a7c257b0Skamil   for (size_t F = 0; F < kNumFlags; F++) {
147*a7c257b0Skamil     const char *Name = FlagDescriptions[F].Name;
148*a7c257b0Skamil     const char *Str = FlagValue(Param, Name);
149*a7c257b0Skamil     if (Str)  {
150*a7c257b0Skamil       if (FlagDescriptions[F].IntFlag) {
151*a7c257b0Skamil         int Val = MyStol(Str);
152*a7c257b0Skamil         *FlagDescriptions[F].IntFlag = Val;
153*a7c257b0Skamil         if (Flags.verbosity >= 2)
154*a7c257b0Skamil           Printf("Flag: %s %d\n", Name, Val);
155*a7c257b0Skamil         return true;
156*a7c257b0Skamil       } else if (FlagDescriptions[F].UIntFlag) {
157*a7c257b0Skamil         unsigned int Val = std::stoul(Str);
158*a7c257b0Skamil         *FlagDescriptions[F].UIntFlag = Val;
159*a7c257b0Skamil         if (Flags.verbosity >= 2)
160*a7c257b0Skamil           Printf("Flag: %s %u\n", Name, Val);
161*a7c257b0Skamil         return true;
162*a7c257b0Skamil       } else if (FlagDescriptions[F].StrFlag) {
163*a7c257b0Skamil         *FlagDescriptions[F].StrFlag = Str;
164*a7c257b0Skamil         if (Flags.verbosity >= 2)
165*a7c257b0Skamil           Printf("Flag: %s %s\n", Name, Str);
166*a7c257b0Skamil         return true;
167*a7c257b0Skamil       } else {  // Deprecated flag.
168*a7c257b0Skamil         Printf("Flag: %s: deprecated, don't use\n", Name);
169*a7c257b0Skamil         return true;
170*a7c257b0Skamil       }
171*a7c257b0Skamil     }
172*a7c257b0Skamil   }
173*a7c257b0Skamil   Printf("\n\nWARNING: unrecognized flag '%s'; "
174*a7c257b0Skamil          "use -help=1 to list all flags\n\n", Param);
175*a7c257b0Skamil   return true;
176*a7c257b0Skamil }
177*a7c257b0Skamil 
178*a7c257b0Skamil // We don't use any library to minimize dependencies.
ParseFlags(const Vector<std::string> & Args)179*a7c257b0Skamil static void ParseFlags(const Vector<std::string> &Args) {
180*a7c257b0Skamil   for (size_t F = 0; F < kNumFlags; F++) {
181*a7c257b0Skamil     if (FlagDescriptions[F].IntFlag)
182*a7c257b0Skamil       *FlagDescriptions[F].IntFlag = FlagDescriptions[F].Default;
183*a7c257b0Skamil     if (FlagDescriptions[F].UIntFlag)
184*a7c257b0Skamil       *FlagDescriptions[F].UIntFlag =
185*a7c257b0Skamil           static_cast<unsigned int>(FlagDescriptions[F].Default);
186*a7c257b0Skamil     if (FlagDescriptions[F].StrFlag)
187*a7c257b0Skamil       *FlagDescriptions[F].StrFlag = nullptr;
188*a7c257b0Skamil   }
189*a7c257b0Skamil   Inputs = new Vector<std::string>;
190*a7c257b0Skamil   for (size_t A = 1; A < Args.size(); A++) {
191*a7c257b0Skamil     if (ParseOneFlag(Args[A].c_str())) {
192*a7c257b0Skamil       if (Flags.ignore_remaining_args)
193*a7c257b0Skamil         break;
194*a7c257b0Skamil       continue;
195*a7c257b0Skamil     }
196*a7c257b0Skamil     Inputs->push_back(Args[A]);
197*a7c257b0Skamil   }
198*a7c257b0Skamil }
199*a7c257b0Skamil 
200*a7c257b0Skamil static std::mutex Mu;
201*a7c257b0Skamil 
PulseThread()202*a7c257b0Skamil static void PulseThread() {
203*a7c257b0Skamil   while (true) {
204*a7c257b0Skamil     SleepSeconds(600);
205*a7c257b0Skamil     std::lock_guard<std::mutex> Lock(Mu);
206*a7c257b0Skamil     Printf("pulse...\n");
207*a7c257b0Skamil   }
208*a7c257b0Skamil }
209*a7c257b0Skamil 
WorkerThread(const Command & BaseCmd,std::atomic<unsigned> * Counter,unsigned NumJobs,std::atomic<bool> * HasErrors)210*a7c257b0Skamil static void WorkerThread(const Command &BaseCmd, std::atomic<unsigned> *Counter,
211*a7c257b0Skamil                          unsigned NumJobs, std::atomic<bool> *HasErrors) {
212*a7c257b0Skamil   while (true) {
213*a7c257b0Skamil     unsigned C = (*Counter)++;
214*a7c257b0Skamil     if (C >= NumJobs) break;
215*a7c257b0Skamil     std::string Log = "fuzz-" + std::to_string(C) + ".log";
216*a7c257b0Skamil     Command Cmd(BaseCmd);
217*a7c257b0Skamil     Cmd.setOutputFile(Log);
218*a7c257b0Skamil     Cmd.combineOutAndErr();
219*a7c257b0Skamil     if (Flags.verbosity) {
220*a7c257b0Skamil       std::string CommandLine = Cmd.toString();
221*a7c257b0Skamil       Printf("%s\n", CommandLine.c_str());
222*a7c257b0Skamil     }
223*a7c257b0Skamil     int ExitCode = ExecuteCommand(Cmd);
224*a7c257b0Skamil     if (ExitCode != 0)
225*a7c257b0Skamil       *HasErrors = true;
226*a7c257b0Skamil     std::lock_guard<std::mutex> Lock(Mu);
227*a7c257b0Skamil     Printf("================== Job %u exited with exit code %d ============\n",
228*a7c257b0Skamil            C, ExitCode);
229*a7c257b0Skamil     fuzzer::CopyFileToErr(Log);
230*a7c257b0Skamil   }
231*a7c257b0Skamil }
232*a7c257b0Skamil 
CloneArgsWithoutX(const Vector<std::string> & Args,const char * X1,const char * X2)233*a7c257b0Skamil std::string CloneArgsWithoutX(const Vector<std::string> &Args,
234*a7c257b0Skamil                               const char *X1, const char *X2) {
235*a7c257b0Skamil   std::string Cmd;
236*a7c257b0Skamil   for (auto &S : Args) {
237*a7c257b0Skamil     if (FlagValue(S.c_str(), X1) || FlagValue(S.c_str(), X2))
238*a7c257b0Skamil       continue;
239*a7c257b0Skamil     Cmd += S + " ";
240*a7c257b0Skamil   }
241*a7c257b0Skamil   return Cmd;
242*a7c257b0Skamil }
243*a7c257b0Skamil 
RunInMultipleProcesses(const Vector<std::string> & Args,unsigned NumWorkers,unsigned NumJobs)244*a7c257b0Skamil static int RunInMultipleProcesses(const Vector<std::string> &Args,
245*a7c257b0Skamil                                   unsigned NumWorkers, unsigned NumJobs) {
246*a7c257b0Skamil   std::atomic<unsigned> Counter(0);
247*a7c257b0Skamil   std::atomic<bool> HasErrors(false);
248*a7c257b0Skamil   Command Cmd(Args);
249*a7c257b0Skamil   Cmd.removeFlag("jobs");
250*a7c257b0Skamil   Cmd.removeFlag("workers");
251*a7c257b0Skamil   Vector<std::thread> V;
252*a7c257b0Skamil   std::thread Pulse(PulseThread);
253*a7c257b0Skamil   Pulse.detach();
254*a7c257b0Skamil   for (unsigned i = 0; i < NumWorkers; i++)
255*a7c257b0Skamil     V.push_back(std::thread(WorkerThread, std::ref(Cmd), &Counter, NumJobs, &HasErrors));
256*a7c257b0Skamil   for (auto &T : V)
257*a7c257b0Skamil     T.join();
258*a7c257b0Skamil   return HasErrors ? 1 : 0;
259*a7c257b0Skamil }
260*a7c257b0Skamil 
RssThread(Fuzzer * F,size_t RssLimitMb)261*a7c257b0Skamil static void RssThread(Fuzzer *F, size_t RssLimitMb) {
262*a7c257b0Skamil   while (true) {
263*a7c257b0Skamil     SleepSeconds(1);
264*a7c257b0Skamil     size_t Peak = GetPeakRSSMb();
265*a7c257b0Skamil     if (Peak > RssLimitMb)
266*a7c257b0Skamil       F->RssLimitCallback();
267*a7c257b0Skamil   }
268*a7c257b0Skamil }
269*a7c257b0Skamil 
StartRssThread(Fuzzer * F,size_t RssLimitMb)270*a7c257b0Skamil static void StartRssThread(Fuzzer *F, size_t RssLimitMb) {
271*a7c257b0Skamil   if (!RssLimitMb) return;
272*a7c257b0Skamil   std::thread T(RssThread, F, RssLimitMb);
273*a7c257b0Skamil   T.detach();
274*a7c257b0Skamil }
275*a7c257b0Skamil 
RunOneTest(Fuzzer * F,const char * InputFilePath,size_t MaxLen)276*a7c257b0Skamil int RunOneTest(Fuzzer *F, const char *InputFilePath, size_t MaxLen) {
277*a7c257b0Skamil   Unit U = FileToVector(InputFilePath);
278*a7c257b0Skamil   if (MaxLen && MaxLen < U.size())
279*a7c257b0Skamil     U.resize(MaxLen);
280*a7c257b0Skamil   F->ExecuteCallback(U.data(), U.size());
281*a7c257b0Skamil   F->TryDetectingAMemoryLeak(U.data(), U.size(), true);
282*a7c257b0Skamil   return 0;
283*a7c257b0Skamil }
284*a7c257b0Skamil 
AllInputsAreFiles()285*a7c257b0Skamil static bool AllInputsAreFiles() {
286*a7c257b0Skamil   if (Inputs->empty()) return false;
287*a7c257b0Skamil   for (auto &Path : *Inputs)
288*a7c257b0Skamil     if (!IsFile(Path))
289*a7c257b0Skamil       return false;
290*a7c257b0Skamil   return true;
291*a7c257b0Skamil }
292*a7c257b0Skamil 
GetDedupTokenFromFile(const std::string & Path)293*a7c257b0Skamil static std::string GetDedupTokenFromFile(const std::string &Path) {
294*a7c257b0Skamil   auto S = FileToString(Path);
295*a7c257b0Skamil   auto Beg = S.find("DEDUP_TOKEN:");
296*a7c257b0Skamil   if (Beg == std::string::npos)
297*a7c257b0Skamil     return "";
298*a7c257b0Skamil   auto End = S.find('\n', Beg);
299*a7c257b0Skamil   if (End == std::string::npos)
300*a7c257b0Skamil     return "";
301*a7c257b0Skamil   return S.substr(Beg, End - Beg);
302*a7c257b0Skamil }
303*a7c257b0Skamil 
CleanseCrashInput(const Vector<std::string> & Args,const FuzzingOptions & Options)304*a7c257b0Skamil int CleanseCrashInput(const Vector<std::string> &Args,
305*a7c257b0Skamil                        const FuzzingOptions &Options) {
306*a7c257b0Skamil   if (Inputs->size() != 1 || !Flags.exact_artifact_path) {
307*a7c257b0Skamil     Printf("ERROR: -cleanse_crash should be given one input file and"
308*a7c257b0Skamil           " -exact_artifact_path\n");
309*a7c257b0Skamil     exit(1);
310*a7c257b0Skamil   }
311*a7c257b0Skamil   std::string InputFilePath = Inputs->at(0);
312*a7c257b0Skamil   std::string OutputFilePath = Flags.exact_artifact_path;
313*a7c257b0Skamil   Command Cmd(Args);
314*a7c257b0Skamil   Cmd.removeFlag("cleanse_crash");
315*a7c257b0Skamil 
316*a7c257b0Skamil   assert(Cmd.hasArgument(InputFilePath));
317*a7c257b0Skamil   Cmd.removeArgument(InputFilePath);
318*a7c257b0Skamil 
319*a7c257b0Skamil   auto LogFilePath = DirPlusFile(
320*a7c257b0Skamil       TmpDir(), "libFuzzerTemp." + std::to_string(GetPid()) + ".txt");
321*a7c257b0Skamil   auto TmpFilePath = DirPlusFile(
322*a7c257b0Skamil       TmpDir(), "libFuzzerTemp." + std::to_string(GetPid()) + ".repro");
323*a7c257b0Skamil   Cmd.addArgument(TmpFilePath);
324*a7c257b0Skamil   Cmd.setOutputFile(LogFilePath);
325*a7c257b0Skamil   Cmd.combineOutAndErr();
326*a7c257b0Skamil 
327*a7c257b0Skamil   std::string CurrentFilePath = InputFilePath;
328*a7c257b0Skamil   auto U = FileToVector(CurrentFilePath);
329*a7c257b0Skamil   size_t Size = U.size();
330*a7c257b0Skamil 
331*a7c257b0Skamil   const Vector<uint8_t> ReplacementBytes = {' ', 0xff};
332*a7c257b0Skamil   for (int NumAttempts = 0; NumAttempts < 5; NumAttempts++) {
333*a7c257b0Skamil     bool Changed = false;
334*a7c257b0Skamil     for (size_t Idx = 0; Idx < Size; Idx++) {
335*a7c257b0Skamil       Printf("CLEANSE[%d]: Trying to replace byte %zd of %zd\n", NumAttempts,
336*a7c257b0Skamil              Idx, Size);
337*a7c257b0Skamil       uint8_t OriginalByte = U[Idx];
338*a7c257b0Skamil       if (ReplacementBytes.end() != std::find(ReplacementBytes.begin(),
339*a7c257b0Skamil                                               ReplacementBytes.end(),
340*a7c257b0Skamil                                               OriginalByte))
341*a7c257b0Skamil         continue;
342*a7c257b0Skamil       for (auto NewByte : ReplacementBytes) {
343*a7c257b0Skamil         U[Idx] = NewByte;
344*a7c257b0Skamil         WriteToFile(U, TmpFilePath);
345*a7c257b0Skamil         auto ExitCode = ExecuteCommand(Cmd);
346*a7c257b0Skamil         RemoveFile(TmpFilePath);
347*a7c257b0Skamil         if (!ExitCode) {
348*a7c257b0Skamil           U[Idx] = OriginalByte;
349*a7c257b0Skamil         } else {
350*a7c257b0Skamil           Changed = true;
351*a7c257b0Skamil           Printf("CLEANSE: Replaced byte %zd with 0x%x\n", Idx, NewByte);
352*a7c257b0Skamil           WriteToFile(U, OutputFilePath);
353*a7c257b0Skamil           break;
354*a7c257b0Skamil         }
355*a7c257b0Skamil       }
356*a7c257b0Skamil     }
357*a7c257b0Skamil     if (!Changed) break;
358*a7c257b0Skamil   }
359*a7c257b0Skamil   RemoveFile(LogFilePath);
360*a7c257b0Skamil   return 0;
361*a7c257b0Skamil }
362*a7c257b0Skamil 
MinimizeCrashInput(const Vector<std::string> & Args,const FuzzingOptions & Options)363*a7c257b0Skamil int MinimizeCrashInput(const Vector<std::string> &Args,
364*a7c257b0Skamil                        const FuzzingOptions &Options) {
365*a7c257b0Skamil   if (Inputs->size() != 1) {
366*a7c257b0Skamil     Printf("ERROR: -minimize_crash should be given one input file\n");
367*a7c257b0Skamil     exit(1);
368*a7c257b0Skamil   }
369*a7c257b0Skamil   std::string InputFilePath = Inputs->at(0);
370*a7c257b0Skamil   Command BaseCmd(Args);
371*a7c257b0Skamil   BaseCmd.removeFlag("minimize_crash");
372*a7c257b0Skamil   BaseCmd.removeFlag("exact_artifact_path");
373*a7c257b0Skamil   assert(BaseCmd.hasArgument(InputFilePath));
374*a7c257b0Skamil   BaseCmd.removeArgument(InputFilePath);
375*a7c257b0Skamil   if (Flags.runs <= 0 && Flags.max_total_time == 0) {
376*a7c257b0Skamil     Printf("INFO: you need to specify -runs=N or "
377*a7c257b0Skamil            "-max_total_time=N with -minimize_crash=1\n"
378*a7c257b0Skamil            "INFO: defaulting to -max_total_time=600\n");
379*a7c257b0Skamil     BaseCmd.addFlag("max_total_time", "600");
380*a7c257b0Skamil   }
381*a7c257b0Skamil 
382*a7c257b0Skamil   auto LogFilePath = DirPlusFile(
383*a7c257b0Skamil       TmpDir(), "libFuzzerTemp." + std::to_string(GetPid()) + ".txt");
384*a7c257b0Skamil   BaseCmd.setOutputFile(LogFilePath);
385*a7c257b0Skamil   BaseCmd.combineOutAndErr();
386*a7c257b0Skamil 
387*a7c257b0Skamil   std::string CurrentFilePath = InputFilePath;
388*a7c257b0Skamil   while (true) {
389*a7c257b0Skamil     Unit U = FileToVector(CurrentFilePath);
390*a7c257b0Skamil     Printf("CRASH_MIN: minimizing crash input: '%s' (%zd bytes)\n",
391*a7c257b0Skamil            CurrentFilePath.c_str(), U.size());
392*a7c257b0Skamil 
393*a7c257b0Skamil     Command Cmd(BaseCmd);
394*a7c257b0Skamil     Cmd.addArgument(CurrentFilePath);
395*a7c257b0Skamil 
396*a7c257b0Skamil     std::string CommandLine = Cmd.toString();
397*a7c257b0Skamil     Printf("CRASH_MIN: executing: %s\n", CommandLine.c_str());
398*a7c257b0Skamil     int ExitCode = ExecuteCommand(Cmd);
399*a7c257b0Skamil     if (ExitCode == 0) {
400*a7c257b0Skamil       Printf("ERROR: the input %s did not crash\n", CurrentFilePath.c_str());
401*a7c257b0Skamil       exit(1);
402*a7c257b0Skamil     }
403*a7c257b0Skamil     Printf("CRASH_MIN: '%s' (%zd bytes) caused a crash. Will try to minimize "
404*a7c257b0Skamil            "it further\n",
405*a7c257b0Skamil            CurrentFilePath.c_str(), U.size());
406*a7c257b0Skamil     auto DedupToken1 = GetDedupTokenFromFile(LogFilePath);
407*a7c257b0Skamil     if (!DedupToken1.empty())
408*a7c257b0Skamil       Printf("CRASH_MIN: DedupToken1: %s\n", DedupToken1.c_str());
409*a7c257b0Skamil 
410*a7c257b0Skamil     std::string ArtifactPath =
411*a7c257b0Skamil         Flags.exact_artifact_path
412*a7c257b0Skamil             ? Flags.exact_artifact_path
413*a7c257b0Skamil             : Options.ArtifactPrefix + "minimized-from-" + Hash(U);
414*a7c257b0Skamil     Cmd.addFlag("minimize_crash_internal_step", "1");
415*a7c257b0Skamil     Cmd.addFlag("exact_artifact_path", ArtifactPath);
416*a7c257b0Skamil     CommandLine = Cmd.toString();
417*a7c257b0Skamil     Printf("CRASH_MIN: executing: %s\n", CommandLine.c_str());
418*a7c257b0Skamil     ExitCode = ExecuteCommand(Cmd);
419*a7c257b0Skamil     CopyFileToErr(LogFilePath);
420*a7c257b0Skamil     if (ExitCode == 0) {
421*a7c257b0Skamil       if (Flags.exact_artifact_path) {
422*a7c257b0Skamil         CurrentFilePath = Flags.exact_artifact_path;
423*a7c257b0Skamil         WriteToFile(U, CurrentFilePath);
424*a7c257b0Skamil       }
425*a7c257b0Skamil       Printf("CRASH_MIN: failed to minimize beyond %s (%d bytes), exiting\n",
426*a7c257b0Skamil              CurrentFilePath.c_str(), U.size());
427*a7c257b0Skamil       break;
428*a7c257b0Skamil     }
429*a7c257b0Skamil     auto DedupToken2 = GetDedupTokenFromFile(LogFilePath);
430*a7c257b0Skamil     if (!DedupToken2.empty())
431*a7c257b0Skamil       Printf("CRASH_MIN: DedupToken2: %s\n", DedupToken2.c_str());
432*a7c257b0Skamil 
433*a7c257b0Skamil     if (DedupToken1 != DedupToken2) {
434*a7c257b0Skamil       if (Flags.exact_artifact_path) {
435*a7c257b0Skamil         CurrentFilePath = Flags.exact_artifact_path;
436*a7c257b0Skamil         WriteToFile(U, CurrentFilePath);
437*a7c257b0Skamil       }
438*a7c257b0Skamil       Printf("CRASH_MIN: mismatch in dedup tokens"
439*a7c257b0Skamil              " (looks like a different bug). Won't minimize further\n");
440*a7c257b0Skamil       break;
441*a7c257b0Skamil     }
442*a7c257b0Skamil 
443*a7c257b0Skamil     CurrentFilePath = ArtifactPath;
444*a7c257b0Skamil     Printf("*********************************\n");
445*a7c257b0Skamil   }
446*a7c257b0Skamil   RemoveFile(LogFilePath);
447*a7c257b0Skamil   return 0;
448*a7c257b0Skamil }
449*a7c257b0Skamil 
MinimizeCrashInputInternalStep(Fuzzer * F,InputCorpus * Corpus)450*a7c257b0Skamil int MinimizeCrashInputInternalStep(Fuzzer *F, InputCorpus *Corpus) {
451*a7c257b0Skamil   assert(Inputs->size() == 1);
452*a7c257b0Skamil   std::string InputFilePath = Inputs->at(0);
453*a7c257b0Skamil   Unit U = FileToVector(InputFilePath);
454*a7c257b0Skamil   Printf("INFO: Starting MinimizeCrashInputInternalStep: %zd\n", U.size());
455*a7c257b0Skamil   if (U.size() < 2) {
456*a7c257b0Skamil     Printf("INFO: The input is small enough, exiting\n");
457*a7c257b0Skamil     exit(0);
458*a7c257b0Skamil   }
459*a7c257b0Skamil   F->SetMaxInputLen(U.size());
460*a7c257b0Skamil   F->SetMaxMutationLen(U.size() - 1);
461*a7c257b0Skamil   F->MinimizeCrashLoop(U);
462*a7c257b0Skamil   Printf("INFO: Done MinimizeCrashInputInternalStep, no crashes found\n");
463*a7c257b0Skamil   exit(0);
464*a7c257b0Skamil   return 0;
465*a7c257b0Skamil }
466*a7c257b0Skamil 
AnalyzeDictionary(Fuzzer * F,const Vector<Unit> & Dict,UnitVector & Corpus)467*a7c257b0Skamil int AnalyzeDictionary(Fuzzer *F, const Vector<Unit>& Dict,
468*a7c257b0Skamil                       UnitVector& Corpus) {
469*a7c257b0Skamil   Printf("Started dictionary minimization (up to %d tests)\n",
470*a7c257b0Skamil          Dict.size() * Corpus.size() * 2);
471*a7c257b0Skamil 
472*a7c257b0Skamil   // Scores and usage count for each dictionary unit.
473*a7c257b0Skamil   Vector<int> Scores(Dict.size());
474*a7c257b0Skamil   Vector<int> Usages(Dict.size());
475*a7c257b0Skamil 
476*a7c257b0Skamil   Vector<size_t> InitialFeatures;
477*a7c257b0Skamil   Vector<size_t> ModifiedFeatures;
478*a7c257b0Skamil   for (auto &C : Corpus) {
479*a7c257b0Skamil     // Get coverage for the testcase without modifications.
480*a7c257b0Skamil     F->ExecuteCallback(C.data(), C.size());
481*a7c257b0Skamil     InitialFeatures.clear();
482*a7c257b0Skamil     TPC.CollectFeatures([&](size_t Feature) {
483*a7c257b0Skamil       InitialFeatures.push_back(Feature);
484*a7c257b0Skamil     });
485*a7c257b0Skamil 
486*a7c257b0Skamil     for (size_t i = 0; i < Dict.size(); ++i) {
487*a7c257b0Skamil       Vector<uint8_t> Data = C;
488*a7c257b0Skamil       auto StartPos = std::search(Data.begin(), Data.end(),
489*a7c257b0Skamil                                   Dict[i].begin(), Dict[i].end());
490*a7c257b0Skamil       // Skip dictionary unit, if the testcase does not contain it.
491*a7c257b0Skamil       if (StartPos == Data.end())
492*a7c257b0Skamil         continue;
493*a7c257b0Skamil 
494*a7c257b0Skamil       ++Usages[i];
495*a7c257b0Skamil       while (StartPos != Data.end()) {
496*a7c257b0Skamil         // Replace all occurrences of dictionary unit in the testcase.
497*a7c257b0Skamil         auto EndPos = StartPos + Dict[i].size();
498*a7c257b0Skamil         for (auto It = StartPos; It != EndPos; ++It)
499*a7c257b0Skamil           *It ^= 0xFF;
500*a7c257b0Skamil 
501*a7c257b0Skamil         StartPos = std::search(EndPos, Data.end(),
502*a7c257b0Skamil                                Dict[i].begin(), Dict[i].end());
503*a7c257b0Skamil       }
504*a7c257b0Skamil 
505*a7c257b0Skamil       // Get coverage for testcase with masked occurrences of dictionary unit.
506*a7c257b0Skamil       F->ExecuteCallback(Data.data(), Data.size());
507*a7c257b0Skamil       ModifiedFeatures.clear();
508*a7c257b0Skamil       TPC.CollectFeatures([&](size_t Feature) {
509*a7c257b0Skamil         ModifiedFeatures.push_back(Feature);
510*a7c257b0Skamil       });
511*a7c257b0Skamil 
512*a7c257b0Skamil       if (InitialFeatures == ModifiedFeatures)
513*a7c257b0Skamil         --Scores[i];
514*a7c257b0Skamil       else
515*a7c257b0Skamil         Scores[i] += 2;
516*a7c257b0Skamil     }
517*a7c257b0Skamil   }
518*a7c257b0Skamil 
519*a7c257b0Skamil   Printf("###### Useless dictionary elements. ######\n");
520*a7c257b0Skamil   for (size_t i = 0; i < Dict.size(); ++i) {
521*a7c257b0Skamil     // Dictionary units with positive score are treated as useful ones.
522*a7c257b0Skamil     if (Scores[i] > 0)
523*a7c257b0Skamil        continue;
524*a7c257b0Skamil 
525*a7c257b0Skamil     Printf("\"");
526*a7c257b0Skamil     PrintASCII(Dict[i].data(), Dict[i].size(), "\"");
527*a7c257b0Skamil     Printf(" # Score: %d, Used: %d\n", Scores[i], Usages[i]);
528*a7c257b0Skamil   }
529*a7c257b0Skamil   Printf("###### End of useless dictionary elements. ######\n");
530*a7c257b0Skamil   return 0;
531*a7c257b0Skamil }
532*a7c257b0Skamil 
FuzzerDriver(int * argc,char *** argv,UserCallback Callback)533*a7c257b0Skamil int FuzzerDriver(int *argc, char ***argv, UserCallback Callback) {
534*a7c257b0Skamil   using namespace fuzzer;
535*a7c257b0Skamil   assert(argc && argv && "Argument pointers cannot be nullptr");
536*a7c257b0Skamil   std::string Argv0((*argv)[0]);
537*a7c257b0Skamil   EF = new ExternalFunctions();
538*a7c257b0Skamil   if (EF->LLVMFuzzerInitialize)
539*a7c257b0Skamil     EF->LLVMFuzzerInitialize(argc, argv);
540*a7c257b0Skamil   if (EF->__msan_scoped_disable_interceptor_checks)
541*a7c257b0Skamil     EF->__msan_scoped_disable_interceptor_checks();
542*a7c257b0Skamil   const Vector<std::string> Args(*argv, *argv + *argc);
543*a7c257b0Skamil   assert(!Args.empty());
544*a7c257b0Skamil   ProgName = new std::string(Args[0]);
545*a7c257b0Skamil   if (Argv0 != *ProgName) {
546*a7c257b0Skamil     Printf("ERROR: argv[0] has been modified in LLVMFuzzerInitialize\n");
547*a7c257b0Skamil     exit(1);
548*a7c257b0Skamil   }
549*a7c257b0Skamil   ParseFlags(Args);
550*a7c257b0Skamil   if (Flags.help) {
551*a7c257b0Skamil     PrintHelp();
552*a7c257b0Skamil     return 0;
553*a7c257b0Skamil   }
554*a7c257b0Skamil 
555*a7c257b0Skamil   if (Flags.close_fd_mask & 2)
556*a7c257b0Skamil     DupAndCloseStderr();
557*a7c257b0Skamil   if (Flags.close_fd_mask & 1)
558*a7c257b0Skamil     CloseStdout();
559*a7c257b0Skamil 
560*a7c257b0Skamil   if (Flags.jobs > 0 && Flags.workers == 0) {
561*a7c257b0Skamil     Flags.workers = std::min(NumberOfCpuCores() / 2, Flags.jobs);
562*a7c257b0Skamil     if (Flags.workers > 1)
563*a7c257b0Skamil       Printf("Running %u workers\n", Flags.workers);
564*a7c257b0Skamil   }
565*a7c257b0Skamil 
566*a7c257b0Skamil   if (Flags.workers > 0 && Flags.jobs > 0)
567*a7c257b0Skamil     return RunInMultipleProcesses(Args, Flags.workers, Flags.jobs);
568*a7c257b0Skamil 
569*a7c257b0Skamil   FuzzingOptions Options;
570*a7c257b0Skamil   Options.Verbosity = Flags.verbosity;
571*a7c257b0Skamil   Options.MaxLen = Flags.max_len;
572*a7c257b0Skamil   Options.LenControl = Flags.len_control;
573*a7c257b0Skamil   Options.UnitTimeoutSec = Flags.timeout;
574*a7c257b0Skamil   Options.ErrorExitCode = Flags.error_exitcode;
575*a7c257b0Skamil   Options.TimeoutExitCode = Flags.timeout_exitcode;
576*a7c257b0Skamil   Options.MaxTotalTimeSec = Flags.max_total_time;
577*a7c257b0Skamil   Options.DoCrossOver = Flags.cross_over;
578*a7c257b0Skamil   Options.MutateDepth = Flags.mutate_depth;
579*a7c257b0Skamil   Options.ReduceDepth = Flags.reduce_depth;
580*a7c257b0Skamil   Options.UseCounters = Flags.use_counters;
581*a7c257b0Skamil   Options.UseMemmem = Flags.use_memmem;
582*a7c257b0Skamil   Options.UseCmp = Flags.use_cmp;
583*a7c257b0Skamil   Options.UseValueProfile = Flags.use_value_profile;
584*a7c257b0Skamil   Options.Shrink = Flags.shrink;
585*a7c257b0Skamil   Options.ReduceInputs = Flags.reduce_inputs;
586*a7c257b0Skamil   Options.ShuffleAtStartUp = Flags.shuffle;
587*a7c257b0Skamil   Options.PreferSmall = Flags.prefer_small;
588*a7c257b0Skamil   Options.ReloadIntervalSec = Flags.reload;
589*a7c257b0Skamil   Options.OnlyASCII = Flags.only_ascii;
590*a7c257b0Skamil   Options.DetectLeaks = Flags.detect_leaks;
591*a7c257b0Skamil   Options.PurgeAllocatorIntervalSec = Flags.purge_allocator_interval;
592*a7c257b0Skamil   Options.TraceMalloc = Flags.trace_malloc;
593*a7c257b0Skamil   Options.RssLimitMb = Flags.rss_limit_mb;
594*a7c257b0Skamil   Options.MallocLimitMb = Flags.malloc_limit_mb;
595*a7c257b0Skamil   if (!Options.MallocLimitMb)
596*a7c257b0Skamil     Options.MallocLimitMb = Options.RssLimitMb;
597*a7c257b0Skamil   if (Flags.runs >= 0)
598*a7c257b0Skamil     Options.MaxNumberOfRuns = Flags.runs;
599*a7c257b0Skamil   if (!Inputs->empty() && !Flags.minimize_crash_internal_step)
600*a7c257b0Skamil     Options.OutputCorpus = (*Inputs)[0];
601*a7c257b0Skamil   Options.ReportSlowUnits = Flags.report_slow_units;
602*a7c257b0Skamil   if (Flags.artifact_prefix)
603*a7c257b0Skamil     Options.ArtifactPrefix = Flags.artifact_prefix;
604*a7c257b0Skamil   if (Flags.exact_artifact_path)
605*a7c257b0Skamil     Options.ExactArtifactPath = Flags.exact_artifact_path;
606*a7c257b0Skamil   Vector<Unit> Dictionary;
607*a7c257b0Skamil   if (Flags.dict)
608*a7c257b0Skamil     if (!ParseDictionaryFile(FileToString(Flags.dict), &Dictionary))
609*a7c257b0Skamil       return 1;
610*a7c257b0Skamil   if (Flags.verbosity > 0 && !Dictionary.empty())
611*a7c257b0Skamil     Printf("Dictionary: %zd entries\n", Dictionary.size());
612*a7c257b0Skamil   bool DoPlainRun = AllInputsAreFiles();
613*a7c257b0Skamil   Options.SaveArtifacts =
614*a7c257b0Skamil       !DoPlainRun || Flags.minimize_crash_internal_step;
615*a7c257b0Skamil   Options.PrintNewCovPcs = Flags.print_pcs;
616*a7c257b0Skamil   Options.PrintNewCovFuncs = Flags.print_funcs;
617*a7c257b0Skamil   Options.PrintFinalStats = Flags.print_final_stats;
618*a7c257b0Skamil   Options.PrintCorpusStats = Flags.print_corpus_stats;
619*a7c257b0Skamil   Options.PrintCoverage = Flags.print_coverage;
620*a7c257b0Skamil   Options.PrintUnstableStats = Flags.print_unstable_stats;
621*a7c257b0Skamil   if (Flags.handle_unstable == TracePC::MinUnstable ||
622*a7c257b0Skamil       Flags.handle_unstable == TracePC::ZeroUnstable)
623*a7c257b0Skamil     Options.HandleUnstable = Flags.handle_unstable;
624*a7c257b0Skamil   Options.DumpCoverage = Flags.dump_coverage;
625*a7c257b0Skamil   if (Flags.exit_on_src_pos)
626*a7c257b0Skamil     Options.ExitOnSrcPos = Flags.exit_on_src_pos;
627*a7c257b0Skamil   if (Flags.exit_on_item)
628*a7c257b0Skamil     Options.ExitOnItem = Flags.exit_on_item;
629*a7c257b0Skamil   if (Flags.focus_function)
630*a7c257b0Skamil     Options.FocusFunction = Flags.focus_function;
631*a7c257b0Skamil   if (Flags.data_flow_trace)
632*a7c257b0Skamil     Options.DataFlowTrace = Flags.data_flow_trace;
633*a7c257b0Skamil 
634*a7c257b0Skamil   unsigned Seed = Flags.seed;
635*a7c257b0Skamil   // Initialize Seed.
636*a7c257b0Skamil   if (Seed == 0)
637*a7c257b0Skamil     Seed =
638*a7c257b0Skamil         std::chrono::system_clock::now().time_since_epoch().count() + GetPid();
639*a7c257b0Skamil   if (Flags.verbosity)
640*a7c257b0Skamil     Printf("INFO: Seed: %u\n", Seed);
641*a7c257b0Skamil 
642*a7c257b0Skamil   Random Rand(Seed);
643*a7c257b0Skamil   auto *MD = new MutationDispatcher(Rand, Options);
644*a7c257b0Skamil   auto *Corpus = new InputCorpus(Options.OutputCorpus);
645*a7c257b0Skamil   auto *F = new Fuzzer(Callback, *Corpus, *MD, Options);
646*a7c257b0Skamil 
647*a7c257b0Skamil   for (auto &U: Dictionary)
648*a7c257b0Skamil     if (U.size() <= Word::GetMaxSize())
649*a7c257b0Skamil       MD->AddWordToManualDictionary(Word(U.data(), U.size()));
650*a7c257b0Skamil 
651*a7c257b0Skamil   StartRssThread(F, Flags.rss_limit_mb);
652*a7c257b0Skamil 
653*a7c257b0Skamil   Options.HandleAbrt = Flags.handle_abrt;
654*a7c257b0Skamil   Options.HandleBus = Flags.handle_bus;
655*a7c257b0Skamil   Options.HandleFpe = Flags.handle_fpe;
656*a7c257b0Skamil   Options.HandleIll = Flags.handle_ill;
657*a7c257b0Skamil   Options.HandleInt = Flags.handle_int;
658*a7c257b0Skamil   Options.HandleSegv = Flags.handle_segv;
659*a7c257b0Skamil   Options.HandleTerm = Flags.handle_term;
660*a7c257b0Skamil   Options.HandleXfsz = Flags.handle_xfsz;
661*a7c257b0Skamil   Options.HandleUsr1 = Flags.handle_usr1;
662*a7c257b0Skamil   Options.HandleUsr2 = Flags.handle_usr2;
663*a7c257b0Skamil   SetSignalHandler(Options);
664*a7c257b0Skamil 
665*a7c257b0Skamil   std::atexit(Fuzzer::StaticExitCallback);
666*a7c257b0Skamil 
667*a7c257b0Skamil   if (Flags.minimize_crash)
668*a7c257b0Skamil     return MinimizeCrashInput(Args, Options);
669*a7c257b0Skamil 
670*a7c257b0Skamil   if (Flags.minimize_crash_internal_step)
671*a7c257b0Skamil     return MinimizeCrashInputInternalStep(F, Corpus);
672*a7c257b0Skamil 
673*a7c257b0Skamil   if (Flags.cleanse_crash)
674*a7c257b0Skamil     return CleanseCrashInput(Args, Options);
675*a7c257b0Skamil 
676*a7c257b0Skamil #if 0  // deprecated, to be removed.
677*a7c257b0Skamil   if (auto Name = Flags.run_equivalence_server) {
678*a7c257b0Skamil     SMR.Destroy(Name);
679*a7c257b0Skamil     if (!SMR.Create(Name)) {
680*a7c257b0Skamil        Printf("ERROR: can't create shared memory region\n");
681*a7c257b0Skamil       return 1;
682*a7c257b0Skamil     }
683*a7c257b0Skamil     Printf("INFO: EQUIVALENCE SERVER UP\n");
684*a7c257b0Skamil     while (true) {
685*a7c257b0Skamil       SMR.WaitClient();
686*a7c257b0Skamil       size_t Size = SMR.ReadByteArraySize();
687*a7c257b0Skamil       SMR.WriteByteArray(nullptr, 0);
688*a7c257b0Skamil       const Unit tmp(SMR.GetByteArray(), SMR.GetByteArray() + Size);
689*a7c257b0Skamil       F->ExecuteCallback(tmp.data(), tmp.size());
690*a7c257b0Skamil       SMR.PostServer();
691*a7c257b0Skamil     }
692*a7c257b0Skamil     return 0;
693*a7c257b0Skamil   }
694*a7c257b0Skamil 
695*a7c257b0Skamil   if (auto Name = Flags.use_equivalence_server) {
696*a7c257b0Skamil     if (!SMR.Open(Name)) {
697*a7c257b0Skamil       Printf("ERROR: can't open shared memory region\n");
698*a7c257b0Skamil       return 1;
699*a7c257b0Skamil     }
700*a7c257b0Skamil     Printf("INFO: EQUIVALENCE CLIENT UP\n");
701*a7c257b0Skamil   }
702*a7c257b0Skamil #endif
703*a7c257b0Skamil 
704*a7c257b0Skamil   if (DoPlainRun) {
705*a7c257b0Skamil     Options.SaveArtifacts = false;
706*a7c257b0Skamil     int Runs = std::max(1, Flags.runs);
707*a7c257b0Skamil     Printf("%s: Running %zd inputs %d time(s) each.\n", ProgName->c_str(),
708*a7c257b0Skamil            Inputs->size(), Runs);
709*a7c257b0Skamil     for (auto &Path : *Inputs) {
710*a7c257b0Skamil       auto StartTime = system_clock::now();
711*a7c257b0Skamil       Printf("Running: %s\n", Path.c_str());
712*a7c257b0Skamil       for (int Iter = 0; Iter < Runs; Iter++)
713*a7c257b0Skamil         RunOneTest(F, Path.c_str(), Options.MaxLen);
714*a7c257b0Skamil       auto StopTime = system_clock::now();
715*a7c257b0Skamil       auto MS = duration_cast<milliseconds>(StopTime - StartTime).count();
716*a7c257b0Skamil       Printf("Executed %s in %zd ms\n", Path.c_str(), (long)MS);
717*a7c257b0Skamil     }
718*a7c257b0Skamil     Printf("***\n"
719*a7c257b0Skamil            "*** NOTE: fuzzing was not performed, you have only\n"
720*a7c257b0Skamil            "***       executed the target code on a fixed set of inputs.\n"
721*a7c257b0Skamil            "***\n");
722*a7c257b0Skamil     F->PrintFinalStats();
723*a7c257b0Skamil     exit(0);
724*a7c257b0Skamil   }
725*a7c257b0Skamil 
726*a7c257b0Skamil   if (Flags.merge) {
727*a7c257b0Skamil     F->CrashResistantMerge(Args, *Inputs,
728*a7c257b0Skamil                            Flags.load_coverage_summary,
729*a7c257b0Skamil                            Flags.save_coverage_summary,
730*a7c257b0Skamil                            Flags.merge_control_file);
731*a7c257b0Skamil     exit(0);
732*a7c257b0Skamil   }
733*a7c257b0Skamil 
734*a7c257b0Skamil   if (Flags.merge_inner) {
735*a7c257b0Skamil     const size_t kDefaultMaxMergeLen = 1 << 20;
736*a7c257b0Skamil     if (Options.MaxLen == 0)
737*a7c257b0Skamil       F->SetMaxInputLen(kDefaultMaxMergeLen);
738*a7c257b0Skamil     assert(Flags.merge_control_file);
739*a7c257b0Skamil     F->CrashResistantMergeInternalStep(Flags.merge_control_file);
740*a7c257b0Skamil     exit(0);
741*a7c257b0Skamil   }
742*a7c257b0Skamil 
743*a7c257b0Skamil   if (Flags.analyze_dict) {
744*a7c257b0Skamil     size_t MaxLen = INT_MAX;  // Large max length.
745*a7c257b0Skamil     UnitVector InitialCorpus;
746*a7c257b0Skamil     for (auto &Inp : *Inputs) {
747*a7c257b0Skamil       Printf("Loading corpus dir: %s\n", Inp.c_str());
748*a7c257b0Skamil       ReadDirToVectorOfUnits(Inp.c_str(), &InitialCorpus, nullptr,
749*a7c257b0Skamil                              MaxLen, /*ExitOnError=*/false);
750*a7c257b0Skamil     }
751*a7c257b0Skamil 
752*a7c257b0Skamil     if (Dictionary.empty() || Inputs->empty()) {
753*a7c257b0Skamil       Printf("ERROR: can't analyze dict without dict and corpus provided\n");
754*a7c257b0Skamil       return 1;
755*a7c257b0Skamil     }
756*a7c257b0Skamil     if (AnalyzeDictionary(F, Dictionary, InitialCorpus)) {
757*a7c257b0Skamil       Printf("Dictionary analysis failed\n");
758*a7c257b0Skamil       exit(1);
759*a7c257b0Skamil     }
760*a7c257b0Skamil     Printf("Dictionary analysis succeeded\n");
761*a7c257b0Skamil     exit(0);
762*a7c257b0Skamil   }
763*a7c257b0Skamil 
764*a7c257b0Skamil   F->Loop(*Inputs);
765*a7c257b0Skamil 
766*a7c257b0Skamil   if (Flags.verbosity)
767*a7c257b0Skamil     Printf("Done %zd runs in %zd second(s)\n", F->getTotalNumberOfRuns(),
768*a7c257b0Skamil            F->secondsSinceProcessStartUp());
769*a7c257b0Skamil   F->PrintFinalStats();
770*a7c257b0Skamil 
771*a7c257b0Skamil   exit(0);  // Don't let F destroy itself.
772*a7c257b0Skamil }
773*a7c257b0Skamil 
774*a7c257b0Skamil // Storage for global ExternalFunctions object.
775*a7c257b0Skamil ExternalFunctions *EF = nullptr;
776*a7c257b0Skamil 
777*a7c257b0Skamil }  // namespace fuzzer
778