mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
43 lines
1.2 KiB
C++
43 lines
1.2 KiB
C++
/// \file LoadBinaryPass.cpp
|
|
/// \brief
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <cstdint>
|
|
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
#include "revng/ADT/STLExtras.h"
|
|
#include "revng/Lift/LoadBinaryPass.h"
|
|
#include "revng/Model/LoadModelPass.h"
|
|
#include "revng/Model/RawBinaryView.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
using namespace llvm::cl;
|
|
opt<std::string> RawBinaryPath("binary-path", desc("<raw binary path>"));
|
|
} // namespace
|
|
|
|
char LoadBinaryWrapperPass::ID;
|
|
|
|
using Register = llvm::RegisterPass<LoadBinaryWrapperPass>;
|
|
static Register X("load-binary", "Load Binary Pass", true, true);
|
|
|
|
LoadBinaryWrapperPass::LoadBinaryWrapperPass() : llvm::ModulePass(ID) {
|
|
revng_check(RawBinaryPath.getNumOccurrences() == 1);
|
|
auto Result = MemoryBuffer::getFileOrSTDIN(RawBinaryPath);
|
|
MaybeBuffer = cantFail(errorOrToExpected(std::move(Result)));
|
|
Data = toArrayRef(MaybeBuffer->getBuffer());
|
|
}
|
|
|
|
bool LoadBinaryWrapperPass::runOnModule(llvm::Module &M) {
|
|
auto &Model = getAnalysis<LoadModelWrapperPass>().get().getReadOnlyModel();
|
|
BinaryView.emplace(*Model, Data);
|
|
return false;
|
|
}
|