mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
d578bc6cd2
In the new pipeline the root module is split off in its individual isolated modules at the end of `isolate`. Before splitting, there are a lot of global variables in the root module and only a small part is going to be needed after splitting for each module. To avoid excessive memory usage employ `ConservativeModuleCloner` in `Isolate` so that only the needed global variables are actually cloned when splitting off.
74 lines
2.2 KiB
C++
74 lines
2.2 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <memory>
|
|
|
|
#include "llvm/Pass.h"
|
|
|
|
#include "revng/BasicAnalyses/GeneratedCodeBasicInfo.h"
|
|
#include "revng/EarlyFunctionAnalysis/CollectCFG.h"
|
|
|
|
class IsolateFunctions : public llvm::ModulePass {
|
|
public:
|
|
static char ID;
|
|
|
|
public:
|
|
IsolateFunctions() : llvm::ModulePass(ID) {}
|
|
|
|
bool runOnModule(llvm::Module &M) override;
|
|
|
|
void getAnalysisUsage(llvm::AnalysisUsage &AU) const override;
|
|
};
|
|
|
|
class IsolateFunctionsImpl;
|
|
|
|
namespace revng::pypeline::piperuns {
|
|
|
|
class Isolate {
|
|
private:
|
|
const model::Binary &Binary;
|
|
std::unique_ptr<llvm::Module> ClonedModule;
|
|
LLVMFunctionContainer &Output;
|
|
std::optional<GeneratedCodeBasicInfo> GCBI;
|
|
// unique_ptr to the implementation. This is a temporary measure until the
|
|
// old pipeline is dropped and the body of the `Impl` class can be inlined in
|
|
// this one.
|
|
std::unique_ptr<IsolateFunctionsImpl> Impl;
|
|
std::vector<std::tuple<MetaAddress, llvm::Function *>> IsolatedFunctions;
|
|
|
|
public:
|
|
static constexpr llvm::StringRef Name = "isolate";
|
|
using Arguments = TypeList<
|
|
PipeRunArgument<const CFGMap, "CFG", "Function control flow data">,
|
|
PipeRunArgument<LLVMRootContainer,
|
|
"Input",
|
|
"Input LLVM module to be isolated",
|
|
// The root container is first modified in-place to be
|
|
// isolated, then, as part of the destructor, the individual
|
|
// functions are split and put in their respective module in
|
|
// the LLVMFunctionContainer.
|
|
Access::Read>,
|
|
PipeRunArgument<LLVMFunctionContainer,
|
|
"Output",
|
|
"Output LLVM modules with isolated functions",
|
|
Access::Write>>;
|
|
|
|
Isolate(const class Model &Model,
|
|
llvm::StringRef Config,
|
|
llvm::StringRef DynamicConfig,
|
|
const CFGMap &CFG,
|
|
LLVMRootContainer &Root,
|
|
LLVMFunctionContainer &Output);
|
|
~Isolate();
|
|
|
|
void runOnFunction(const model::Function &TheFunction);
|
|
|
|
private:
|
|
void splitIsolatedFunctionsToOutput();
|
|
};
|
|
|
|
} // namespace revng::pypeline::piperuns
|