1 //===-- FuzzerCLI.cpp -----------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/FuzzMutate/FuzzerCLI.h"
11 #include "llvm/ADT/Triple.h"
12 #include "llvm/Support/CommandLine.h"
13 #include "llvm/Support/Compiler.h"
14 #include "llvm/Support/Error.h"
15 #include "llvm/Support/MemoryBuffer.h"
16 #include "llvm/Support/raw_ostream.h"
17 
18 using namespace llvm;
19 
20 void llvm::parseFuzzerCLOpts(int ArgC, char *ArgV[]) {
21   std::vector<const char *> CLArgs;
22   CLArgs.push_back(ArgV[0]);
23 
24   int I = 1;
25   while (I < ArgC)
26     if (StringRef(ArgV[I++]).equals("-ignore_remaining_args=1"))
27       break;
28   while (I < ArgC)
29     CLArgs.push_back(ArgV[I++]);
30 
31   cl::ParseCommandLineOptions(CLArgs.size(), CLArgs.data());
32 }
33 
34 void llvm::handleExecNameEncodedBEOpts(StringRef ExecName) {
35   std::vector<std::string> Args{ExecName};
36 
37   auto NameAndArgs = ExecName.split("--");
38   if (NameAndArgs.second.empty())
39     return;
40 
41   SmallVector<StringRef, 4> Opts;
42   NameAndArgs.second.split(Opts, '-');
43   for (StringRef Opt : Opts) {
44     if (Opt.equals("gisel")) {
45       Args.push_back("-global-isel");
46       // For now we default GlobalISel to -O0
47       Args.push_back("-O0");
48     } else if (Opt.startswith("O")) {
49       Args.push_back("-" + Opt.str());
50     } else if (Triple(Opt).getArch()) {
51       Args.push_back("-mtriple=" + Opt.str());
52     } else {
53       errs() << ExecName << ": Unknown option: " << Opt << ".\n";
54       exit(1);
55     }
56   }
57   errs() << NameAndArgs.first << ": Injected args:";
58   for (int I = 1, E = Args.size(); I < E; ++I)
59     errs() << " " << Args[I];
60   errs() << "\n";
61 
62   std::vector<const char *> CLArgs;
63   CLArgs.reserve(Args.size());
64   for (std::string &S : Args)
65     CLArgs.push_back(S.c_str());
66 
67   cl::ParseCommandLineOptions(CLArgs.size(), CLArgs.data());
68 }
69 
70 void llvm::handleExecNameEncodedOptimizerOpts(StringRef ExecName) {
71   // TODO: Refactor parts common with the 'handleExecNameEncodedBEOpts'
72   std::vector<std::string> Args{ExecName};
73 
74   auto NameAndArgs = ExecName.split("--");
75   if (NameAndArgs.second.empty())
76     return;
77 
78   SmallVector<StringRef, 4> Opts;
79   NameAndArgs.second.split(Opts, '-');
80   for (StringRef Opt : Opts) {
81     if (Opt.startswith("instcombine")) {
82       Args.push_back("-passes=instcombine");
83     } else if (Triple(Opt).getArch()) {
84       Args.push_back("-mtriple=" + Opt.str());
85     } else {
86       errs() << ExecName << ": Unknown option: " << Opt << ".\n";
87       exit(1);
88     }
89   }
90 
91   errs() << NameAndArgs.first << ": Injected args:";
92   for (int I = 1, E = Args.size(); I < E; ++I)
93     errs() << " " << Args[I];
94   errs() << "\n";
95 
96   std::vector<const char *> CLArgs;
97   CLArgs.reserve(Args.size());
98   for (std::string &S : Args)
99     CLArgs.push_back(S.c_str());
100 
101   cl::ParseCommandLineOptions(CLArgs.size(), CLArgs.data());
102 }
103 
104 int llvm::runFuzzerOnInputs(int ArgC, char *ArgV[], FuzzerTestFun TestOne,
105                             FuzzerInitFun Init) {
106   errs() << "*** This tool was not linked to libFuzzer.\n"
107          << "*** No fuzzing will be performed.\n";
108   if (int RC = Init(&ArgC, &ArgV)) {
109     errs() << "Initialization failed\n";
110     return RC;
111   }
112 
113   for (int I = 1; I < ArgC; ++I) {
114     StringRef Arg(ArgV[I]);
115     if (Arg.startswith("-")) {
116       if (Arg.equals("-ignore_remaining_args=1"))
117         break;
118       continue;
119     }
120 
121     auto BufOrErr = MemoryBuffer::getFile(Arg, /*FileSize-*/ -1,
122                                           /*RequiresNullTerminator=*/false);
123     if (std::error_code EC = BufOrErr.getError()) {
124       errs() << "Error reading file: " << Arg << ": " << EC.message() << "\n";
125       return 1;
126     }
127     std::unique_ptr<MemoryBuffer> Buf = std::move(BufOrErr.get());
128     errs() << "Running: " << Arg << " (" << Buf->getBufferSize() << " bytes)\n";
129     TestOne(reinterpret_cast<const uint8_t *>(Buf->getBufferStart()),
130             Buf->getBufferSize());
131   }
132   return 0;
133 }
134