mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
f8bd4c3bac
* Make the following private headers public:
* Lift/CPUStateAccessAnalysisPass.h
* Lift/CSVOffsets.h
* Lift/PTCDump.h
* Lift/VariableManager.h
* Move from revngSupport to revngLift:
* IRAnnotators.{h,cpp}
* SelfReferencingDbgAnnotationWriter.{h,cpp}
* Move from revngSupport to revngModel:
* FunctionTags.{h,cpp}
* ProgramCounterHandler.{h,cpp}
* Move from revngSupport to revngRecompile:
* OriginalAssemblyAnnotationWriter.{h,cpp}
54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <iterator>
|
|
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/Passes/PassBuilder.h"
|
|
|
|
#include "revng/Model/FunctionTags.h"
|
|
#include "revng/Support/IRHelpers.h"
|
|
|
|
using namespace llvm;
|
|
|
|
class DropOpaqueReturnAddress : public ModulePass {
|
|
public:
|
|
static char ID;
|
|
DropOpaqueReturnAddress() : ModulePass(ID) {}
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {}
|
|
|
|
bool runOnModule(Module &M) override {
|
|
SmallVector<llvm::Function *, 16> FunctionsToPurge;
|
|
|
|
auto &OpaqueReturnAddress = FunctionTags::OpaqueReturnAddressFunction;
|
|
SmallVector<Function *, 16> Functions;
|
|
for (Function &F : OpaqueReturnAddress.functions(&M))
|
|
Functions.push_back(&F);
|
|
|
|
for (Function *F : Functions) {
|
|
auto *Undef = UndefValue::get(F->getReturnType());
|
|
|
|
SmallVector<CallBase *, 16> Callers;
|
|
llvm::copy(callers(F), std::back_inserter(Callers));
|
|
|
|
for (CallBase *Caller : Callers)
|
|
Caller->replaceAllUsesWith(Undef);
|
|
|
|
for (CallBase *Caller : Callers)
|
|
eraseFromParent(Caller);
|
|
|
|
eraseFromParent(F);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|
|
|
|
char DropOpaqueReturnAddress::ID;
|
|
|
|
using Register = RegisterPass<DropOpaqueReturnAddress>;
|
|
static Register R("drop-opaque-return-address", "", false, false);
|