mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
ab125b35b0
Change company name to "rev.ng Labs Srl" in all license headers to reflect changed company name and legal status Add missing license headers to files that didn't have one
63 lines
2.2 KiB
C++
63 lines
2.2 KiB
C++
//
|
|
// Copyright (c) rev.ng Labs Srl. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "llvm/Analysis/LazyValueInfo.h"
|
|
#include "llvm/IR/ConstantRange.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
|
|
#include "revng/Model/IRHelpers.h"
|
|
|
|
#include "revng-c/PromoteStackPointer/ComputeStackAccessesBoundsPass.h"
|
|
#include "revng-c/PromoteStackPointer/InstrumentStackAccessesPass.h"
|
|
|
|
using namespace llvm;
|
|
|
|
bool ComputeStackAccessesBoundsPass::runOnModule(Module &M) {
|
|
for (Function &StackOffsetFunction : StackOffsetMarker.functions(&M)) {
|
|
auto *FunctionType = StackOffsetFunction.getFunctionType();
|
|
auto *DifferenceType = cast<IntegerType>(FunctionType->getParamType(1));
|
|
for (CallBase *Call : callers(&StackOffsetFunction)) {
|
|
Function &F = *Call->getParent()->getParent();
|
|
LazyValueInfo &LVI = getAnalysis<LazyValueInfoWrapperPass>(F).getLVI();
|
|
auto *Undef = UndefValue::get(DifferenceType);
|
|
|
|
// Identify lower bound of the lower bound
|
|
const auto &LowerBoundRange = LVI.getConstantRange(Call->getArgOperand(1),
|
|
cast<CallInst>(Call));
|
|
Value *LowerBound = nullptr;
|
|
if (not LowerBoundRange.isFullSet()) {
|
|
LowerBound = ConstantInt::get(DifferenceType,
|
|
LowerBoundRange.getLower());
|
|
} else {
|
|
LowerBound = Undef;
|
|
}
|
|
Call->setArgOperand(1, LowerBound);
|
|
|
|
// Identify upper bound of the upper bound
|
|
const auto &UpperBoundRange = LVI.getConstantRange(Call->getArgOperand(2),
|
|
cast<CallInst>(Call));
|
|
Value *UpperBound = nullptr;
|
|
if (not UpperBoundRange.isFullSet()) {
|
|
UpperBound = ConstantInt::get(DifferenceType,
|
|
UpperBoundRange.getUpper());
|
|
} else {
|
|
UpperBound = Undef;
|
|
}
|
|
Call->setArgOperand(2, UpperBound);
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void ComputeStackAccessesBoundsPass::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
AU.addRequired<LazyValueInfoWrapperPass>();
|
|
AU.setPreservesCFG();
|
|
}
|
|
|
|
char ComputeStackAccessesBoundsPass::ID = 0;
|
|
|
|
using RegisterCSAB = RegisterPass<ComputeStackAccessesBoundsPass>;
|
|
static RegisterCSAB
|
|
R("compute-stack-accesses-bounds", "Compute Stack Accesses Bounds Pass");
|