revng-revng/lib/Model/SerializeModelPass.cpp

/// \file SerializeModelPass.cpp
/// \brief Implementation of the pass taking care of serializing the
///        model in the Module as Metadata.

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

// LLVM includes
#include "llvm/IR/Metadata.h"
#include "llvm/IR/Module.h"
#include "llvm/PassSupport.h"

// Local libraries includes
#include "revng/Model/SerializeModelPass.h"

using namespace llvm;

char SerializeModelPass::ID;

template<typename T>
using RP = RegisterPass<T>;

static RP<SerializeModelPass>
  X("serialize-model", "Serialize the model", true, true);

void SerializeModelPass::writeModel(model::Binary &Model, llvm::Module &M) {

  NamedMDNode *NamedMD = M.getNamedMetadata(ModelMetadataName);
  revng_check(not NamedMD, "The model has alread been serialized");

  std::string Buffer;
  {
    llvm::raw_string_ostream Stream(Buffer);
    yaml::Output YAMLOutput(Stream);
    YAMLOutput << Model;
  }

  LLVMContext &Context = M.getContext();
  auto Tuple = MDTuple::get(Context, { MDString::get(Context, Buffer) });

  NamedMD = M.getOrInsertNamedMetadata(ModelMetadataName);
  NamedMD->addOperand(Tuple);
}

bool SerializeModelPass::runOnModule(Module &M) {

  writeModel(getAnalysis<LoadModelPass>().getWriteableModel(), M);

  return false;
}