mirror of
https://github.com/nodejs/node.git
synced 2025-05-15 17:51:35 +00:00
cfbf9e0d14
PR-URL: https://github.com/nodejs/node/pull/52465 Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Rafael Gonzaga <rafael.nunu@hotmail.com> Reviewed-By: Michael Dawson <midawson@redhat.com>
60 lines
1.6 KiB
C++
60 lines
1.6 KiB
C++
// Copyright 2016 the V8 project authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <vector>
|
|
|
|
extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv);
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size);
|
|
|
|
int main(int argc, char* argv[]) {
|
|
if (LLVMFuzzerInitialize(&argc, &argv)) {
|
|
fprintf(stderr, "Failed to initialize fuzzer target\n");
|
|
return 1;
|
|
}
|
|
|
|
std::vector<uint8_t> input_data;
|
|
|
|
bool after_dash_dash = false;
|
|
for (int arg_idx = 1; arg_idx < argc; ++arg_idx) {
|
|
const char* const arg = argv[arg_idx];
|
|
// Ignore first '--' argument.
|
|
if (!after_dash_dash && arg[0] == '-' && arg[1] == '-' && arg[2] == '\0') {
|
|
after_dash_dash = true;
|
|
continue;
|
|
}
|
|
|
|
FILE* input = fopen(arg, "rb");
|
|
if (!input) {
|
|
fprintf(stderr, "Failed to open '%s'\n", arg);
|
|
return 1;
|
|
}
|
|
|
|
fseek(input, 0, SEEK_END);
|
|
size_t size = ftell(input);
|
|
fseek(input, 0, SEEK_SET);
|
|
|
|
size_t old_size = input_data.size();
|
|
input_data.resize(old_size + size);
|
|
|
|
size_t bytes_read = fread(input_data.data() + old_size, 1, size, input);
|
|
fclose(input);
|
|
|
|
if (bytes_read != size) {
|
|
fprintf(stderr, "Failed to read %zu bytes from %s\n", size, arg);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
// Ensure that {input_data.data()} is not {nullptr} to avoid having to handle
|
|
// this in specific fuzzers.
|
|
if (input_data.empty()) input_data.reserve(1);
|
|
|
|
return LLVMFuzzerTestOneInput(input_data.data(), input_data.size());
|
|
}
|