main.cpp

#include <bitcoinfuzz/module_loader.h>

static std::shared_ptr<bitcoinfuzz::Driver> driver = nullptr;
static bitcoinfuzz::ModuleLogger module_logger;
static std::string g_target;

extern "C" int LLVMFuzzerInitialize([[maybe_unused]] int *argc,
                                    [[maybe_unused]] char ***argv) {
  bitcoinfuzz::InitializeRegistry();
  const char *target = std::getenv("FUZZ");
  if (target == nullptr || target[0] == '\0') {
    std::cerr << "FUZZ environment variable must be set" << std::endl;
    std::exit(1);
  }
  g_target = target;
  const char *log_outputs_env = std::getenv("LOG_OUTPUTS");
  const bool log_outputs = log_outputs_env != nullptr &&
                           log_outputs_env[0] != '\0' &&
                           std::string_view(log_outputs_env) != "0";
  driver = std::make_shared<bitcoinfuzz::Driver>(module_logger, log_outputs);
  bitcoinfuzz::LoadModules(driver, module_logger);
  return 0;
}

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
  driver->Run(Data, Size, g_target);
  return 0;
}