mirror of
https://github.com/RGBCube/serenity
synced 2025-10-31 21:32:44 +00:00
158 lines
4.6 KiB
C++
158 lines
4.6 KiB
C++
/*
|
|
* Copyright (c) 2021, Ben Wiederhake <BenWiederhake.GitHub@gmx.de>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/String.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibMain/Main.h>
|
|
#include <unistd.h>
|
|
|
|
#define ENUMERATE_TARGETS(T) \
|
|
T(BMPLoader) \
|
|
T(ELF) \
|
|
T(Gemini) \
|
|
T(GIFLoader) \
|
|
T(HttpRequest) \
|
|
T(ICOLoader) \
|
|
T(JPGLoader) \
|
|
T(Js) \
|
|
T(Markdown) \
|
|
T(PBMLoader) \
|
|
T(PGMLoader) \
|
|
T(PNGLoader) \
|
|
T(PPMLoader) \
|
|
T(QOILoader) \
|
|
T(RegexECMA262) \
|
|
T(RegexPosixExtended) \
|
|
T(Shell) \
|
|
T(TTF) \
|
|
T(URL)
|
|
|
|
#undef __ENUMERATE_TARGET
|
|
#define __ENUMERATE_TARGET(x) extern "C" int Test##x(uint8_t const*, size_t);
|
|
ENUMERATE_TARGETS(__ENUMERATE_TARGET)
|
|
#undef __ENUMERATE_TARGET
|
|
|
|
#define LLVMFuzzerTestOneInput TestBMPLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzBMPLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestELF
|
|
#include <Meta/Lagom/Fuzzers/FuzzELF.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestGemini
|
|
#include <Meta/Lagom/Fuzzers/FuzzGemini.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestGIFLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzGIFLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestHttpRequest
|
|
#include <Meta/Lagom/Fuzzers/FuzzHttpRequest.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestICOLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzICOLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestJPGLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzJPGLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestJs
|
|
#include <Meta/Lagom/Fuzzers/FuzzJs.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestMarkdown
|
|
#include <Meta/Lagom/Fuzzers/FuzzMarkdown.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestPBMLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzPBMLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestPGMLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzPGMLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestPNGLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzPNGLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestPPMLoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzPPMLoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestQOILoader
|
|
#include <Meta/Lagom/Fuzzers/FuzzQOILoader.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestRegexECMA262
|
|
#include <Meta/Lagom/Fuzzers/FuzzRegexECMA262.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestRegexPosixExtended
|
|
#include <Meta/Lagom/Fuzzers/FuzzRegexPosixExtended.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestShell
|
|
#include <Meta/Lagom/Fuzzers/FuzzShell.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestTTF
|
|
#include <Meta/Lagom/Fuzzers/FuzzTTF.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
#define LLVMFuzzerTestOneInput TestURL
|
|
#include <Meta/Lagom/Fuzzers/FuzzURL.cpp>
|
|
#undef LLVMFuzzerTestOneInput
|
|
|
|
static auto parse_target_name(StringView name)
|
|
{
|
|
if (name == "list"sv) {
|
|
outln("The following targets are included:");
|
|
#undef __ENUMERATE_TARGET
|
|
#define __ENUMERATE_TARGET(x) outln(#x);
|
|
ENUMERATE_TARGETS(__ENUMERATE_TARGET)
|
|
#undef __ENUMERATE_TARGET
|
|
exit(0);
|
|
}
|
|
|
|
#undef __ENUMERATE_TARGET
|
|
#define __ENUMERATE_TARGET(x) \
|
|
if (name == #x) \
|
|
return Test##x;
|
|
ENUMERATE_TARGETS(__ENUMERATE_TARGET)
|
|
#undef __ENUMERATE_TARGET
|
|
|
|
warnln("Unknown fuzzing target \"{}\". Try \"list\" to get a full list.", name);
|
|
exit(1);
|
|
}
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
StringView type;
|
|
StringView filename;
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_positional_argument(type, "Type of fuzzing target to run (use \"list\" to list all existing)", "target-kind");
|
|
args_parser.add_positional_argument(filename, "Input file", "filename", Core::ArgsParser::Required::No);
|
|
args_parser.parse(arguments);
|
|
|
|
if (arguments.strings.size() <= 2 && arguments.strings[1] != "list"sv) {
|
|
args_parser.print_usage_terminal(stderr, arguments.argv[0]);
|
|
return 0;
|
|
}
|
|
|
|
auto fn = parse_target_name(type);
|
|
|
|
auto file = TRY(Core::File::open(filename, Core::OpenMode::ReadOnly));
|
|
auto input = file->read_all();
|
|
|
|
return fn(input.data(), input.size());
|
|
}
|