RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-14 08:34:57 +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

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

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