da96299906
This patch adjusts tools/fuzzing/ in such a way that the relevant parts can be reused in the JS engine. Changes in detail include: * Various JS_STANDALONE checks to exclude parts that cannot be included in those builds. * Turn LibFuzzerRegistry and LibFuzzerRunner into generic FuzzerRegistry and FuzzerRunner classes and use them for AFL as well. Previously, AFL was piggy-backing on gtests which was kind of an ugly solution anyway (besides that it can't work in JS). Now more code like registry and harness is shared between the two and they follow almost the same call paths and entry points. AFL macros in FuzzingInterface have been rewritten accordingly. This also required name changes in various places. Furthermore, this unifies the way, the fuzzing target is selected, using the FUZZER environment variable rather than LIBFUZZER (using LIBFUZZER in browser builds will give you a deprecation warning because I know some people are using this already and need time to switch). Previously, AFL target had to be selected using GTEST_FILTER, so this is also much better now. * I had to split up FuzzingInterface* such that the STREAM parts are in a separate set of files FuzzingInterfaceStream* because they use nsStringStream which is not allowed to be included into the JS engine even in a full browser build (error: "Using XPCOM strings is limited to code linked into libxul."). I also had to pull FuzzingInterface.cpp (the RAW part only) into the header and make it static because otherwise, would have to make not only separate files but also separate libraries to statically link to the JS engine, which seemed overkill for a single small function. The streaming equivalent of the function is still in a cpp file. * LibFuzzerRegister functions are now unique by appending the module name to avoid redefinition errors. MozReview-Commit-ID: 44zWCdglnHr
74 lines
2.0 KiB
C++
74 lines
2.0 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
* * This Source Code Form is subject to the terms of the Mozilla Public
|
|
* * License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include <cstdlib>
|
|
|
|
#include "FuzzerRunner.h"
|
|
#include "mozilla/Attributes.h"
|
|
#include "prenv.h"
|
|
|
|
#include "FuzzerTestHarness.h"
|
|
|
|
namespace mozilla {
|
|
|
|
// We use a static var 'fuzzerRunner' defined in nsAppRunner.cpp.
|
|
// fuzzerRunner is initialized to nullptr but if this file is linked in,
|
|
// then fuzzerRunner will be set here indicating that
|
|
// we want to call into either LibFuzzer's main or the AFL entrypoint.
|
|
class _InitFuzzer {
|
|
public:
|
|
_InitFuzzer() {
|
|
fuzzerRunner = new FuzzerRunner();
|
|
}
|
|
} InitLibFuzzer;
|
|
|
|
int FuzzerRunner::Run(int* argc, char*** argv) {
|
|
ScopedXPCOM xpcom("Fuzzer");
|
|
const char* fuzzerEnv = getenv("FUZZER");
|
|
|
|
if (!fuzzerEnv) {
|
|
fuzzerEnv = getenv("LIBFUZZER");
|
|
if (fuzzerEnv) {
|
|
fprintf(stderr, "Fuzzer Interface: Warning: \
|
|
Using deprecated LIBFUZZER variable, use FUZZER instead\n");
|
|
} else {
|
|
fprintf(stderr, "Must specify fuzzing target in FUZZER environment variable\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
std::string moduleNameStr(fuzzerEnv);
|
|
FuzzerFunctions funcs = FuzzerRegistry::getInstance().getModuleFunctions(moduleNameStr);
|
|
FuzzerInitFunc initFunc = funcs.first;
|
|
FuzzerTestingFunc testingFunc = funcs.second;
|
|
if (initFunc) {
|
|
int ret = initFunc(argc, argv);
|
|
if (ret) {
|
|
fprintf(stderr, "Fuzzing Interface: Error: Initialize callback failed\n");
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
if (!testingFunc) {
|
|
fprintf(stderr, "Fuzzing Interface: Error: No testing callback found\n");
|
|
return 1;
|
|
}
|
|
|
|
#ifdef LIBFUZZER
|
|
return mFuzzerDriver(argc, argv, testingFunc);
|
|
#else
|
|
// For AFL, testingFunc points to the entry function we need.
|
|
return testingFunc(NULL, 0);
|
|
#endif
|
|
}
|
|
|
|
#ifdef LIBFUZZER
|
|
void FuzzerRunner::setParams(LibFuzzerDriver aDriver) {
|
|
mFuzzerDriver = aDriver;
|
|
}
|
|
#endif
|
|
|
|
} // namespace mozilla
|