/*

* simplecpp - A simple and high-fidelity C/C++ preprocessor library

* Copyright (C) 2016-2024 simplecpp team

*/

#include "simplecpp.h"

#include <cstdint>

#ifdef NO_FUZZ

#include <cstdlib>

#include <fstream>

#include <sstream>

#include <string>

#endif

/*

0 - store in corpus

-1 - omit from corpus

*/

static int doProcess(const uint8_t *data, size_t dataSize)

{

simplecpp::DUI dui;

dui.defines.emplace_back(std::string(reinterpret_cast<const char*>(data), dataSize));

simplecpp::preprocess(dui);

return 0;

}

#ifndef NO_FUZZ

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

int LLVMFuzzerTestOneInput(const uint8_t *data, size_t dataSize)

{

return doProcess(data, dataSize);

}

#else

int main(int argc, char * argv[])

{

if (argc < 2 || argc > 3)

return EXIT_FAILURE;

std::ifstream f(argv[1]);

if (!f.is_open())

return EXIT_FAILURE;

std::ostringstream oss;

oss << f.rdbuf();

if (!f.good())

return EXIT_FAILURE;

const int cnt = (argc == 3) ? std::stoi(argv[2]) : 1;

const std::string code = oss.str();

for (int i = 0; i < cnt; ++i)

doProcess(reinterpret_cast<const uint8_t*>(code.data()), code.size());

return EXIT_SUCCESS;

}

#endif