RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-29 20:15:11 +00:00
Code Issues Activity Actions

test-fuzz: Port to LibMain

Browse source 
Also use StringView in place of raw C strings and String.
This commit is contained in:
Kenneth Myhra 2022-03-28 22:56:14 +02:00 • committed by Brian Gianforcaro
parent b47a9ab4dc
commit 4994718d8d
2 changed files with 10 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Userland/Utilities/CMakeLists.txt
View file

@ -197,7 +197,7 @@ target_link_libraries(tar LibMain LibArchive LibCompress)
target_link_libraries(tee LibMain) target_link_libraries(tee LibMain)
target_link_libraries(telws LibProtocol LibLine LibMain) target_link_libraries(telws LibProtocol LibLine LibMain)
target_link_libraries(test-bindtodevice LibMain) target_link_libraries(test-bindtodevice LibMain)
target_link_libraries(test-fuzz LibCore LibGemini LibGfx LibHTTP LibIPC LibJS LibMarkdown LibShell) target_link_libraries(test-fuzz LibCore LibGemini LibGfx LibHTTP LibIPC LibJS LibMarkdown LibShell LibMain)
target_link_libraries(test-imap LibIMAP LibMain) target_link_libraries(test-imap LibIMAP LibMain)
target_link_libraries(test-pthread LibThreading) target_link_libraries(test-pthread LibThreading)
target_link_libraries(timezone LibMain) target_link_libraries(timezone LibMain)

23
Userland/Utilities/test-fuzz.cpp
View file

@ -7,7 +7,7 @@
#include <AK/String.h> #include <AK/String.h>
#include <LibCore/ArgsParser.h> #include <LibCore/ArgsParser.h>
#include <LibCore/File.h> #include <LibCore/File.h>
#include <stdio.h> #include <LibMain/Main.h>
#include <unistd.h> #include <unistd.h>
#define ENUMERATE_TARGETS(T) \ #define ENUMERATE_TARGETS(T) \
@ -112,9 +112,9 @@ ENUMERATE_TARGETS(__ENUMERATE_TARGET)
#include <Meta/Lagom/Fuzzers/FuzzURL.cpp> #include <Meta/Lagom/Fuzzers/FuzzURL.cpp>
#undef LLVMFuzzerTestOneInput #undef LLVMFuzzerTestOneInput
static auto parse_target_name(const String& name) static auto parse_target_name(StringView name)
{ {
if (name == "list") { if (name == "list"sv) {
outln("The following targets are included:"); outln("The following targets are included:");
#undef __ENUMERATE_TARGET #undef __ENUMERATE_TARGET
#define __ENUMERATE_TARGET(x) outln(#x); #define __ENUMERATE_TARGET(x) outln(#x);
@ -134,25 +134,20 @@ static auto parse_target_name(const String& name)
exit(1); exit(1);
} }
int main(int argc, char** argv) ErrorOr<int> serenity_main(Main::Arguments arguments)
{ {
const char* type; StringView type;
const char* filename; StringView filename;
Core::ArgsParser args_parser; 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(type, "Type of fuzzing target to run (use \"list\" to list all existing)", "target-kind");
args_parser.add_positional_argument(filename, "Input file", "filename"); args_parser.add_positional_argument(filename, "Input file", "filename");
args_parser.parse(argc, argv); args_parser.parse(arguments);
auto fn = parse_target_name(type); auto fn = parse_target_name(type);
auto file = Core::File::open(filename, Core::OpenMode::ReadOnly); auto file = TRY(Core::File::open(filename, Core::OpenMode::ReadOnly));
if (file.is_error()) { auto input = file->read_all();
warnln("Cannot read from file: {}", file.error());
exit(1);
}
auto input = file.value()->read_all();
return fn(input.data(), input.size()); return fn(input.data(), input.size());
} }