mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Pietro Fezzardi
8fa8ca2d03
The previous design returned a model::Binary &, which did not allow users writing the model to properly initialize cross references between different parts of the model.
55 lines
1.5 KiB
C++
55 lines
1.5 KiB
C++
/// \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"
|
|
|
|
// Local libraries includes
|
|
#include "revng/Model/SerializeModelPass.h"
|
|
|
|
using namespace llvm;
|
|
|
|
char SerializeModelWrapperPass::ID;
|
|
|
|
template<typename T>
|
|
using RP = RegisterPass<T>;
|
|
|
|
static RP<SerializeModelWrapperPass>
|
|
X("serialize-model", "Serialize the model", true, true);
|
|
|
|
void 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);
|
|
serialize(Stream, Model);
|
|
}
|
|
|
|
LLVMContext &Context = M.getContext();
|
|
auto Tuple = MDTuple::get(Context, { MDString::get(Context, Buffer) });
|
|
|
|
NamedMD = M.getOrInsertNamedMetadata(ModelMetadataName);
|
|
NamedMD->addOperand(Tuple);
|
|
}
|
|
|
|
bool SerializeModelWrapperPass::runOnModule(Module &M) {
|
|
writeModel(*getAnalysis<LoadModelWrapperPass>().get().getWriteableModel(), M);
|
|
return false;
|
|
}
|
|
|
|
llvm::PreservedAnalyses
|
|
SerializeModelPass::run(llvm::Module &M, llvm::ModuleAnalysisManager &MAM) {
|
|
auto &ModelWrapper = MAM.getResult<LoadModelAnalysis>(M);
|
|
writeModel(*ModelWrapper.getWriteableModel(), M);
|
|
return PreservedAnalyses::all();
|
|
}
|