mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
ee0b8f44c1
This is a big step to split revng-lift in two parts: one that only writes the model and one that actually lifts to LLVM IR. * Introduce `revng import binary` * Split off `BinaryFile.h` * Drop `revng.h` * `GeneratedCodeBasicInfo`: use model * Reduce role of `GeneratedCodeBasicInfo` in favor of `model::Architecture` and `model::Register` methods * `CodeGenerator`: adopt `RawBinaryView` and model * `JumpTargetManager`: adopt `RawBinaryView` and model * `ExternalJumpsHandler`: adopt model * `InstructionTranslator`: discard `Architecture` in favor of `EndianessMismatch` * Many other changes
82 lines
3.0 KiB
C++
82 lines
3.0 KiB
C++
/// \file RemoveHelperCalls.cpp
|
|
/// \brief Remove calls to helpers in a function and replaces them
|
|
/// with stores of an opaque value onto the CSVs clobbered by the helper.
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
#include "revng/BasicAnalyses/RemoveHelperCalls.h"
|
|
#include "revng/Support/IRHelpers.h"
|
|
|
|
llvm::PreservedAnalyses
|
|
RemoveHelperCallsPass::run(llvm::Function &F,
|
|
llvm::FunctionAnalysisManager &FAM) {
|
|
using namespace llvm;
|
|
|
|
// Get the result of the GCBI analysis
|
|
GCBI = &(FAM.getResult<GeneratedCodeBasicInfoAnalysis>(F));
|
|
revng_assert(GCBI != nullptr);
|
|
|
|
SmallVector<Instruction *, 16> ToReplace;
|
|
for (auto &BB : F)
|
|
for (auto &I : BB)
|
|
if (isCallToHelper(&I))
|
|
ToReplace.push_back(&I);
|
|
|
|
bool Changed = not ToReplace.empty();
|
|
if (!Changed)
|
|
return PreservedAnalyses::all();
|
|
|
|
OpaqueFunctionsPool<Type *> OFPOriginalHelper(F.getParent(), false);
|
|
OpaqueFunctionsPool<Type *> OFPRegsClobberedHelper(F.getParent(), false);
|
|
|
|
OFPOriginalHelper.addFnAttribute(Attribute::ReadOnly);
|
|
OFPOriginalHelper.addFnAttribute(Attribute::NoUnwind);
|
|
OFPOriginalHelper.addFnAttribute(Attribute::WillReturn);
|
|
|
|
OFPRegsClobberedHelper.addFnAttribute(Attribute::ReadOnly);
|
|
OFPRegsClobberedHelper.addFnAttribute(Attribute::NoUnwind);
|
|
OFPRegsClobberedHelper.addFnAttribute(Attribute::WillReturn);
|
|
|
|
IRBuilder<> Builder(F.getContext());
|
|
for (auto *I : ToReplace) {
|
|
Builder.SetInsertPoint(I);
|
|
|
|
// Assumption: helpers do not leave the stack altered, thus we can save the
|
|
// stack pointer and restore it back later.
|
|
auto *SP = Builder.CreateLoad(GCBI->spReg());
|
|
|
|
auto *RetTy = cast<CallInst>(I)->getFunctionType()->getReturnType();
|
|
auto *OriginalHelperMarker = OFPOriginalHelper.get(RetTy,
|
|
RetTy,
|
|
{},
|
|
"original_helper");
|
|
|
|
// Create opaque helper for the original helper and taint the registers
|
|
// originally clobbered.
|
|
CallInst *NewHelper = Builder.CreateCall(OriginalHelperMarker);
|
|
|
|
for (auto *CSV : getCSVUsedByHelperCall(I).Written) {
|
|
auto *CSVTy = CSV->getType()->getPointerElementType();
|
|
auto *RegisterClobberedMarker = OFPRegsClobberedHelper.get(CSVTy,
|
|
CSVTy,
|
|
{},
|
|
"regs_"
|
|
"clobbered_"
|
|
"helper");
|
|
Builder.CreateStore(Builder.CreateCall(RegisterClobberedMarker), CSV);
|
|
}
|
|
|
|
// Restore stack pointer back.
|
|
Builder.CreateStore(SP, GCBI->spReg());
|
|
|
|
I->replaceAllUsesWith(NewHelper);
|
|
I->eraseFromParent();
|
|
}
|
|
|
|
return PreservedAnalyses::none();
|
|
}
|