Method UTModify

lib/generation/UTModify.cpp:20–63

const std::string UTModify::UTEntryFuncName = "enterAutofuzz";

UTModify::UTModify(const Fuzzer &F, const SourceAnalysisReport &SourceReport) {
  clang::IntrusiveRefCntPtr<llvm::vfs::OverlayFileSystem> OverlayFS =
      new llvm::vfs::OverlayFileSystem(llvm::vfs::getRealFileSystem());
  FManager = new clang::FileManager(clang::FileSystemOptions(), OverlayFS);

  clang::IntrusiveRefCntPtr<clang::DiagnosticOptions> DiagOpts =
      new clang::DiagnosticOptions();
  auto DiagnosticPrinter = std::make_unique<clang::TextDiagnosticPrinter>(
      llvm::outs(), DiagOpts.get());
  clang::IntrusiveRefCntPtr<clang::DiagnosticIDs> DiagID =
      new clang::DiagnosticIDs();
  auto Diagnostics = std::make_unique<clang::DiagnosticsEngine>(
      DiagID, DiagOpts.get(), DiagnosticPrinter.get(), false);
  SManager = std::make_unique<clang::SourceManager>(*Diagnostics, *FManager);

  std::map<std::string, std::string> FuzzVarDecls;
  std::map<std::string, std::vector<std::string>> FuzzVarFlagDecls;
  std::map<std::string, std::vector<UTModify::GlobalVarSetter>> Setters;
  auto UTPath = F.getUT().getFilePath();
  for (const auto &Iter : F.getFuzzInputMap()) {
    auto &Input = Iter.second;
    assert(Input && "Unexpected Program State");
    if (!Input->getCopyFrom()) {
      generateFuzzVarDeclarations(FuzzVarDecls, FuzzVarFlagDecls, *Input);
      generateFuzzVarGlobalSetters(Setters, *Input);
    }
    generateFuzzVarReplacements(*Input);
  }
  generateTop(FuzzVarDecls, FuzzVarFlagDecls, Setters, UTPath, SourceReport);
  generateBottom(FuzzVarFlagDecls, Setters, F.getUT(), SourceReport);
  if (!generateHeader(fs::path(UTPath).parent_path().string())) {
    clear();
    return;
  }
  generateMainDeletion(SourceReport);

  for (const auto &Iter : Replaces) {
    auto Replace = Iter.second;
    if (FileReplaceMap[Replace.getFilePath().str()].add(Replace)) {
      clear();
      return;
    }
  }
}

bool UTModify::addReplace(const clang::tooling::Replacement &Replace) {
  auto Key = std::make_pair(Replace.getFilePath().str(), Replace.getOffset());