mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Pietro Fezzardi
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
// STL includes
|
|
#include <memory>
|
|
|
|
// LLVM includes
|
|
#include <llvm/CodeGen/Passes.h>
|
|
#include <llvm/IR/LegacyPassManager.h>
|
|
#include <llvm/IR/Module.h>
|
|
#include <llvm/Transforms/InstCombine/InstCombine.h>
|
|
#include <llvm/Transforms/Scalar.h>
|
|
#include <llvm/Transforms/Utils/Cloning.h>
|
|
|
|
// Local library includes
|
|
#include "revng-c/Decompiler/CDecompiler.h"
|
|
#include "revng-c/Decompiler/CDecompilerPass.h"
|
|
#include "revng-c/RemoveCpuLoopStore/RemoveCpuLoopStorePass.h"
|
|
#include "revng-c/RemoveExceptionCalls/RemoveExceptionCallsPass.h"
|
|
#include "revng-c/RemoveLLVMAssumeCalls/RemoveLLVMAssumeCallsPass.h"
|
|
#include "revng-c/RemoveNewPCCalls/RemoveNewPCCallsPass.h"
|
|
|
|
std::string
|
|
decompileFunction(const llvm::Module *M, const std::string &FunctionName) {
|
|
std::string ResultSourceCode;
|
|
std::unique_ptr<llvm::Module> ClonedModule = llvm::CloneModule(*M);
|
|
llvm::Function *TmpFunc = ClonedModule->getFunction(FunctionName);
|
|
|
|
std::unique_ptr<llvm::raw_ostream>
|
|
OS = std::make_unique<llvm::raw_string_ostream>(ResultSourceCode);
|
|
|
|
llvm::legacy::FunctionPassManager PM(&*ClonedModule);
|
|
|
|
// Remove revng's artifacts from IR
|
|
{
|
|
PM.add(new RemoveNewPCCallsPass());
|
|
PM.add(new RemoveExceptionCallsPass());
|
|
PM.add(new RemoveCpuLoopStorePass());
|
|
}
|
|
|
|
// Optimize IR with LLVM's passes
|
|
{
|
|
PM.add(llvm::createDeadCodeEliminationPass());
|
|
PM.add(llvm::createCFGSimplificationPass());
|
|
PM.add(llvm::createDeadStoreEliminationPass());
|
|
PM.add(llvm::createInstructionCombiningPass());
|
|
PM.add(llvm::createSROAPass());
|
|
PM.add(llvm::createConstantPropagationPass());
|
|
PM.add(llvm::createJumpThreadingPass());
|
|
PM.add(llvm::createLICMPass());
|
|
PM.add(llvm::createUnreachableBlockEliminationPass());
|
|
PM.add(llvm::createInstructionCombiningPass());
|
|
PM.add(llvm::createEarlyCSEPass());
|
|
PM.add(llvm::createCFGSimplificationPass());
|
|
}
|
|
|
|
// Remove LLVM's artifacts from IR
|
|
{
|
|
PM.add(new RemoveLLVMAssumeCallsPass());
|
|
PM.add(llvm::createDeadCodeEliminationPass());
|
|
}
|
|
|
|
// Decompile!
|
|
PM.add(new CDecompilerPass(std::move(OS)));
|
|
PM.doInitialization();
|
|
PM.run(*TmpFunc);
|
|
|
|
return ResultSourceCode;
|
|
}
|