tests/oss-fuzz/repro_fuzz.cc

// Copyright 2023 Google LLC
// SPDX-License-Identifier: BSD-2-Clause

#include <chrono>
#include <cstddef>
#include <cstdint>
#include <fstream>
#include <iostream>
#include <vector>

using seconds = std::chrono::duration<double>;
using chrono = std::chrono::high_resolution_clock;

extern "C" int LLVMFuzzerTestOneInput(const uint8_t * data, size_t size);

namespace
{
std::vector<uint8_t> ReadFile(const char * path)
{
    std::vector<uint8_t> buffer;
    std::ifstream file(path, std::ios::binary);
    buffer.resize(file.seekg(0, std::ios::end).tellg());
    file.seekg(0, std::ios::beg).read(reinterpret_cast<char *>(buffer.data()), buffer.size());
    return buffer;
}
} // namespace

int main(int argc, char * argv[])
{
    if (argc < 2) {
        std::cerr << "Missing reproducer file" << std::endl;
        return 1;
    }

    for (int i = 1; i < argc; ++i) {
        const std::vector<uint8_t> buffer = ReadFile(argv[i]);
        const chrono::time_point start_time = chrono::now();
        LLVMFuzzerTestOneInput(buffer.data(), buffer.size());
        std::cout << "Reproducing " << argv[i] << " took " << seconds(chrono::now() - start_time).count() << " seconds." << std::endl;
    }
    return 0;
}