2018-11-27 05:57:47 +08:00
|
|
|
//===- StackSafetyAnalysis.cpp - Stack memory safety analysis -------------===//
|
|
|
|
//
|
2019-01-19 16:50:56 +08:00
|
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
2018-11-27 05:57:47 +08:00
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "llvm/Analysis/StackSafetyAnalysis.h"
|
|
|
|
#include "llvm/Analysis/ScalarEvolutionExpressions.h"
|
2018-11-27 05:57:59 +08:00
|
|
|
#include "llvm/IR/CallSite.h"
|
|
|
|
#include "llvm/IR/InstIterator.h"
|
|
|
|
#include "llvm/IR/IntrinsicInst.h"
|
2018-11-27 05:57:47 +08:00
|
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
|
|
|
|
using namespace llvm;
|
|
|
|
|
|
|
|
#define DEBUG_TYPE "stack-safety"
|
|
|
|
|
2018-11-27 07:05:58 +08:00
|
|
|
static cl::opt<int> StackSafetyMaxIterations("stack-safety-max-iterations",
|
|
|
|
cl::init(20), cl::Hidden);
|
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
/// Rewrite an SCEV expression for a memory access address to an expression that
|
|
|
|
/// represents offset from the given alloca.
|
|
|
|
class AllocaOffsetRewriter : public SCEVRewriteVisitor<AllocaOffsetRewriter> {
|
|
|
|
const Value *AllocaPtr;
|
|
|
|
|
|
|
|
public:
|
|
|
|
AllocaOffsetRewriter(ScalarEvolution &SE, const Value *AllocaPtr)
|
|
|
|
: SCEVRewriteVisitor(SE), AllocaPtr(AllocaPtr) {}
|
|
|
|
|
|
|
|
const SCEV *visit(const SCEV *Expr) {
|
|
|
|
// Only re-write the expression if the alloca is used in an addition
|
|
|
|
// expression (it can be used in other types of expressions if it's cast to
|
|
|
|
// an int and passed as an argument.)
|
|
|
|
if (!isa<SCEVAddRecExpr>(Expr) && !isa<SCEVAddExpr>(Expr) &&
|
|
|
|
!isa<SCEVUnknown>(Expr))
|
|
|
|
return Expr;
|
|
|
|
return SCEVRewriteVisitor<AllocaOffsetRewriter>::visit(Expr);
|
|
|
|
}
|
|
|
|
|
|
|
|
const SCEV *visitUnknown(const SCEVUnknown *Expr) {
|
|
|
|
// FIXME: look through one or several levels of definitions?
|
|
|
|
// This can be inttoptr(AllocaPtr) and SCEV would not unwrap
|
|
|
|
// it for us.
|
|
|
|
if (Expr->getValue() == AllocaPtr)
|
|
|
|
return SE.getZero(Expr->getType());
|
|
|
|
return Expr;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
/// Describes use of address in as a function call argument.
|
|
|
|
struct PassAsArgInfo {
|
|
|
|
/// Function being called.
|
|
|
|
const GlobalValue *Callee = nullptr;
|
|
|
|
/// Index of argument which pass address.
|
|
|
|
size_t ParamNo = 0;
|
|
|
|
// Offset range of address from base address (alloca or calling function
|
|
|
|
// argument).
|
|
|
|
// Range should never set to empty-set, that is an invalid access range
|
|
|
|
// that can cause empty-set to be propagated with ConstantRange::add
|
|
|
|
ConstantRange Offset;
|
|
|
|
PassAsArgInfo(const GlobalValue *Callee, size_t ParamNo, ConstantRange Offset)
|
|
|
|
: Callee(Callee), ParamNo(ParamNo), Offset(Offset) {}
|
|
|
|
|
|
|
|
StringRef getName() const { return Callee->getName(); }
|
|
|
|
};
|
|
|
|
|
|
|
|
raw_ostream &operator<<(raw_ostream &OS, const PassAsArgInfo &P) {
|
|
|
|
return OS << "@" << P.getName() << "(arg" << P.ParamNo << ", " << P.Offset
|
|
|
|
<< ")";
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Describe uses of address (alloca or parameter) inside of the function.
|
|
|
|
struct UseInfo {
|
|
|
|
// Access range if the address (alloca or parameters).
|
|
|
|
// It is allowed to be empty-set when there are no known accesses.
|
|
|
|
ConstantRange Range;
|
|
|
|
|
|
|
|
// List of calls which pass address as an argument.
|
|
|
|
SmallVector<PassAsArgInfo, 4> Calls;
|
|
|
|
|
|
|
|
explicit UseInfo(unsigned PointerSize) : Range{PointerSize, false} {}
|
|
|
|
|
|
|
|
void updateRange(ConstantRange R) { Range = Range.unionWith(R); }
|
|
|
|
};
|
|
|
|
|
|
|
|
raw_ostream &operator<<(raw_ostream &OS, const UseInfo &U) {
|
|
|
|
OS << U.Range;
|
|
|
|
for (auto &Call : U.Calls)
|
|
|
|
OS << ", " << Call;
|
|
|
|
return OS;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct AllocaInfo {
|
|
|
|
const AllocaInst *AI = nullptr;
|
|
|
|
uint64_t Size = 0;
|
|
|
|
UseInfo Use;
|
|
|
|
|
|
|
|
AllocaInfo(unsigned PointerSize, const AllocaInst *AI, uint64_t Size)
|
|
|
|
: AI(AI), Size(Size), Use(PointerSize) {}
|
|
|
|
|
|
|
|
StringRef getName() const { return AI->getName(); }
|
|
|
|
};
|
|
|
|
|
|
|
|
raw_ostream &operator<<(raw_ostream &OS, const AllocaInfo &A) {
|
|
|
|
return OS << A.getName() << "[" << A.Size << "]: " << A.Use;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct ParamInfo {
|
|
|
|
const Argument *Arg = nullptr;
|
|
|
|
UseInfo Use;
|
|
|
|
|
|
|
|
explicit ParamInfo(unsigned PointerSize, const Argument *Arg)
|
|
|
|
: Arg(Arg), Use(PointerSize) {}
|
|
|
|
|
|
|
|
StringRef getName() const { return Arg ? Arg->getName() : "<N/A>"; }
|
|
|
|
};
|
|
|
|
|
|
|
|
raw_ostream &operator<<(raw_ostream &OS, const ParamInfo &P) {
|
|
|
|
return OS << P.getName() << "[]: " << P.Use;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Calculate the allocation size of a given alloca. Returns 0 if the
|
|
|
|
/// size can not be statically determined.
|
|
|
|
uint64_t getStaticAllocaAllocationSize(const AllocaInst *AI) {
|
|
|
|
const DataLayout &DL = AI->getModule()->getDataLayout();
|
|
|
|
uint64_t Size = DL.getTypeAllocSize(AI->getAllocatedType());
|
|
|
|
if (AI->isArrayAllocation()) {
|
|
|
|
auto C = dyn_cast<ConstantInt>(AI->getArraySize());
|
|
|
|
if (!C)
|
|
|
|
return 0;
|
|
|
|
Size *= C->getZExtValue();
|
|
|
|
}
|
|
|
|
return Size;
|
|
|
|
}
|
|
|
|
|
|
|
|
} // end anonymous namespace
|
|
|
|
|
|
|
|
/// Describes uses of allocas and parameters inside of a single function.
|
|
|
|
struct StackSafetyInfo::FunctionInfo {
|
|
|
|
// May be a Function or a GlobalAlias
|
|
|
|
const GlobalValue *GV = nullptr;
|
|
|
|
// Informations about allocas uses.
|
|
|
|
SmallVector<AllocaInfo, 4> Allocas;
|
|
|
|
// Informations about parameters uses.
|
|
|
|
SmallVector<ParamInfo, 4> Params;
|
|
|
|
// TODO: describe return value as depending on one or more of its arguments.
|
|
|
|
|
2018-11-27 07:05:58 +08:00
|
|
|
// StackSafetyDataFlowAnalysis counter stored here for faster access.
|
|
|
|
int UpdateCount = 0;
|
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
FunctionInfo(const StackSafetyInfo &SSI) : FunctionInfo(*SSI.Info) {}
|
|
|
|
|
|
|
|
explicit FunctionInfo(const Function *F) : GV(F){};
|
2018-11-27 07:05:58 +08:00
|
|
|
// Creates FunctionInfo that forwards all the parameters to the aliasee.
|
|
|
|
explicit FunctionInfo(const GlobalAlias *A);
|
2018-11-27 05:57:59 +08:00
|
|
|
|
|
|
|
FunctionInfo(FunctionInfo &&) = default;
|
|
|
|
|
|
|
|
bool IsDSOLocal() const { return GV->isDSOLocal(); };
|
|
|
|
|
|
|
|
bool IsInterposable() const { return GV->isInterposable(); };
|
|
|
|
|
|
|
|
StringRef getName() const { return GV->getName(); }
|
|
|
|
|
|
|
|
void print(raw_ostream &O) const {
|
2018-11-27 07:05:58 +08:00
|
|
|
// TODO: Consider different printout format after
|
|
|
|
// StackSafetyDataFlowAnalysis. Calls and parameters are irrelevant then.
|
2018-11-27 05:57:59 +08:00
|
|
|
O << " @" << getName() << (IsDSOLocal() ? "" : " dso_preemptable")
|
|
|
|
<< (IsInterposable() ? " interposable" : "") << "\n";
|
|
|
|
O << " args uses:\n";
|
|
|
|
for (auto &P : Params)
|
|
|
|
O << " " << P << "\n";
|
|
|
|
O << " allocas uses:\n";
|
|
|
|
for (auto &AS : Allocas)
|
|
|
|
O << " " << AS << "\n";
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
FunctionInfo(const FunctionInfo &) = default;
|
|
|
|
};
|
|
|
|
|
2018-11-27 07:05:58 +08:00
|
|
|
StackSafetyInfo::FunctionInfo::FunctionInfo(const GlobalAlias *A) : GV(A) {
|
|
|
|
unsigned PointerSize = A->getParent()->getDataLayout().getPointerSizeInBits();
|
|
|
|
const GlobalObject *Aliasee = A->getBaseObject();
|
|
|
|
const FunctionType *Type = cast<FunctionType>(Aliasee->getValueType());
|
|
|
|
// 'Forward' all parameters to this alias to the aliasee
|
|
|
|
for (unsigned ArgNo = 0; ArgNo < Type->getNumParams(); ArgNo++) {
|
|
|
|
Params.emplace_back(PointerSize, nullptr);
|
|
|
|
UseInfo &US = Params.back().Use;
|
|
|
|
US.Calls.emplace_back(Aliasee, ArgNo, ConstantRange(APInt(PointerSize, 0)));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
class StackSafetyLocalAnalysis {
|
|
|
|
const Function &F;
|
|
|
|
const DataLayout &DL;
|
|
|
|
ScalarEvolution &SE;
|
|
|
|
unsigned PointerSize = 0;
|
|
|
|
|
|
|
|
const ConstantRange UnknownRange;
|
|
|
|
|
|
|
|
ConstantRange offsetFromAlloca(Value *Addr, const Value *AllocaPtr);
|
|
|
|
ConstantRange getAccessRange(Value *Addr, const Value *AllocaPtr,
|
|
|
|
uint64_t AccessSize);
|
|
|
|
ConstantRange getMemIntrinsicAccessRange(const MemIntrinsic *MI, const Use &U,
|
|
|
|
const Value *AllocaPtr);
|
2018-11-27 05:57:47 +08:00
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
bool analyzeAllUses(const Value *Ptr, UseInfo &AS);
|
|
|
|
|
|
|
|
ConstantRange getRange(uint64_t Lower, uint64_t Upper) const {
|
|
|
|
return ConstantRange(APInt(PointerSize, Lower), APInt(PointerSize, Upper));
|
|
|
|
}
|
|
|
|
|
|
|
|
public:
|
|
|
|
StackSafetyLocalAnalysis(const Function &F, ScalarEvolution &SE)
|
|
|
|
: F(F), DL(F.getParent()->getDataLayout()), SE(SE),
|
|
|
|
PointerSize(DL.getPointerSizeInBits()),
|
|
|
|
UnknownRange(PointerSize, true) {}
|
|
|
|
|
|
|
|
// Run the transformation on the associated function.
|
|
|
|
StackSafetyInfo run();
|
|
|
|
};
|
|
|
|
|
|
|
|
ConstantRange
|
|
|
|
StackSafetyLocalAnalysis::offsetFromAlloca(Value *Addr,
|
|
|
|
const Value *AllocaPtr) {
|
|
|
|
if (!SE.isSCEVable(Addr->getType()))
|
|
|
|
return UnknownRange;
|
|
|
|
|
|
|
|
AllocaOffsetRewriter Rewriter(SE, AllocaPtr);
|
|
|
|
const SCEV *Expr = Rewriter.visit(SE.getSCEV(Addr));
|
|
|
|
ConstantRange Offset = SE.getUnsignedRange(Expr).zextOrTrunc(PointerSize);
|
|
|
|
assert(!Offset.isEmptySet());
|
|
|
|
return Offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
ConstantRange StackSafetyLocalAnalysis::getAccessRange(Value *Addr,
|
|
|
|
const Value *AllocaPtr,
|
|
|
|
uint64_t AccessSize) {
|
|
|
|
if (!SE.isSCEVable(Addr->getType()))
|
|
|
|
return UnknownRange;
|
|
|
|
|
|
|
|
AllocaOffsetRewriter Rewriter(SE, AllocaPtr);
|
|
|
|
const SCEV *Expr = Rewriter.visit(SE.getSCEV(Addr));
|
|
|
|
|
|
|
|
ConstantRange AccessStartRange =
|
|
|
|
SE.getUnsignedRange(Expr).zextOrTrunc(PointerSize);
|
|
|
|
ConstantRange SizeRange = getRange(0, AccessSize);
|
|
|
|
ConstantRange AccessRange = AccessStartRange.add(SizeRange);
|
|
|
|
assert(!AccessRange.isEmptySet());
|
|
|
|
return AccessRange;
|
|
|
|
}
|
|
|
|
|
|
|
|
ConstantRange StackSafetyLocalAnalysis::getMemIntrinsicAccessRange(
|
|
|
|
const MemIntrinsic *MI, const Use &U, const Value *AllocaPtr) {
|
|
|
|
if (auto MTI = dyn_cast<MemTransferInst>(MI)) {
|
|
|
|
if (MTI->getRawSource() != U && MTI->getRawDest() != U)
|
|
|
|
return getRange(0, 1);
|
|
|
|
} else {
|
|
|
|
if (MI->getRawDest() != U)
|
|
|
|
return getRange(0, 1);
|
|
|
|
}
|
|
|
|
const auto *Len = dyn_cast<ConstantInt>(MI->getLength());
|
|
|
|
// Non-constant size => unsafe. FIXME: try SCEV getRange.
|
|
|
|
if (!Len)
|
|
|
|
return UnknownRange;
|
|
|
|
ConstantRange AccessRange = getAccessRange(U, AllocaPtr, Len->getZExtValue());
|
|
|
|
return AccessRange;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The function analyzes all local uses of Ptr (alloca or argument) and
|
|
|
|
/// calculates local access range and all function calls where it was used.
|
|
|
|
bool StackSafetyLocalAnalysis::analyzeAllUses(const Value *Ptr, UseInfo &US) {
|
|
|
|
SmallPtrSet<const Value *, 16> Visited;
|
|
|
|
SmallVector<const Value *, 8> WorkList;
|
|
|
|
WorkList.push_back(Ptr);
|
|
|
|
|
|
|
|
// A DFS search through all uses of the alloca in bitcasts/PHI/GEPs/etc.
|
|
|
|
while (!WorkList.empty()) {
|
|
|
|
const Value *V = WorkList.pop_back_val();
|
|
|
|
for (const Use &UI : V->uses()) {
|
|
|
|
auto I = cast<const Instruction>(UI.getUser());
|
|
|
|
assert(V == UI.get());
|
|
|
|
|
|
|
|
switch (I->getOpcode()) {
|
|
|
|
case Instruction::Load: {
|
|
|
|
US.updateRange(
|
|
|
|
getAccessRange(UI, Ptr, DL.getTypeStoreSize(I->getType())));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case Instruction::VAArg:
|
|
|
|
// "va-arg" from a pointer is safe.
|
|
|
|
break;
|
|
|
|
case Instruction::Store: {
|
|
|
|
if (V == I->getOperand(0)) {
|
|
|
|
// Stored the pointer - conservatively assume it may be unsafe.
|
|
|
|
US.updateRange(UnknownRange);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
US.updateRange(getAccessRange(
|
|
|
|
UI, Ptr, DL.getTypeStoreSize(I->getOperand(0)->getType())));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case Instruction::Ret:
|
|
|
|
// Information leak.
|
|
|
|
// FIXME: Process parameters correctly. This is a leak only if we return
|
|
|
|
// alloca.
|
|
|
|
US.updateRange(UnknownRange);
|
|
|
|
return false;
|
|
|
|
|
|
|
|
case Instruction::Call:
|
|
|
|
case Instruction::Invoke: {
|
|
|
|
ImmutableCallSite CS(I);
|
|
|
|
|
2018-12-22 05:49:40 +08:00
|
|
|
if (I->isLifetimeStartOrEnd())
|
|
|
|
break;
|
2018-11-27 05:57:59 +08:00
|
|
|
|
|
|
|
if (const MemIntrinsic *MI = dyn_cast<MemIntrinsic>(I)) {
|
|
|
|
US.updateRange(getMemIntrinsicAccessRange(MI, UI, Ptr));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
// FIXME: consult devirt?
|
|
|
|
// Do not follow aliases, otherwise we could inadvertently follow
|
|
|
|
// dso_preemptable aliases or aliases with interposable linkage.
|
|
|
|
const GlobalValue *Callee = dyn_cast<GlobalValue>(
|
|
|
|
CS.getCalledValue()->stripPointerCastsNoFollowAliases());
|
|
|
|
if (!Callee) {
|
|
|
|
US.updateRange(UnknownRange);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
assert(isa<Function>(Callee) || isa<GlobalAlias>(Callee));
|
|
|
|
|
|
|
|
ImmutableCallSite::arg_iterator B = CS.arg_begin(), E = CS.arg_end();
|
|
|
|
for (ImmutableCallSite::arg_iterator A = B; A != E; ++A) {
|
|
|
|
if (A->get() == V) {
|
|
|
|
ConstantRange Offset = offsetFromAlloca(UI, Ptr);
|
|
|
|
US.Calls.emplace_back(Callee, A - B, Offset);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
default:
|
|
|
|
if (Visited.insert(I).second)
|
|
|
|
WorkList.push_back(cast<const Instruction>(I));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
StackSafetyInfo StackSafetyLocalAnalysis::run() {
|
|
|
|
StackSafetyInfo::FunctionInfo Info(&F);
|
|
|
|
assert(!F.isDeclaration() &&
|
|
|
|
"Can't run StackSafety on a function declaration");
|
|
|
|
|
|
|
|
LLVM_DEBUG(dbgs() << "[StackSafety] " << F.getName() << "\n");
|
|
|
|
|
|
|
|
for (auto &I : instructions(F)) {
|
|
|
|
if (auto AI = dyn_cast<AllocaInst>(&I)) {
|
|
|
|
Info.Allocas.emplace_back(PointerSize, AI,
|
|
|
|
getStaticAllocaAllocationSize(AI));
|
|
|
|
AllocaInfo &AS = Info.Allocas.back();
|
|
|
|
analyzeAllUses(AI, AS.Use);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const Argument &A : make_range(F.arg_begin(), F.arg_end())) {
|
|
|
|
Info.Params.emplace_back(PointerSize, &A);
|
|
|
|
ParamInfo &PS = Info.Params.back();
|
|
|
|
analyzeAllUses(&A, PS.Use);
|
|
|
|
}
|
|
|
|
|
|
|
|
LLVM_DEBUG(dbgs() << "[StackSafety] done\n");
|
|
|
|
LLVM_DEBUG(Info.print(dbgs()));
|
|
|
|
return StackSafetyInfo(std::move(Info));
|
|
|
|
}
|
|
|
|
|
2018-11-27 07:05:58 +08:00
|
|
|
class StackSafetyDataFlowAnalysis {
|
|
|
|
using FunctionMap =
|
|
|
|
std::map<const GlobalValue *, StackSafetyInfo::FunctionInfo>;
|
|
|
|
|
|
|
|
FunctionMap Functions;
|
|
|
|
// Callee-to-Caller multimap.
|
|
|
|
DenseMap<const GlobalValue *, SmallVector<const GlobalValue *, 4>> Callers;
|
|
|
|
SetVector<const GlobalValue *> WorkList;
|
|
|
|
|
|
|
|
unsigned PointerSize = 0;
|
|
|
|
const ConstantRange UnknownRange;
|
|
|
|
|
|
|
|
ConstantRange getArgumentAccessRange(const GlobalValue *Callee,
|
|
|
|
unsigned ParamNo) const;
|
|
|
|
bool updateOneUse(UseInfo &US, bool UpdateToFullSet);
|
|
|
|
void updateOneNode(const GlobalValue *Callee,
|
|
|
|
StackSafetyInfo::FunctionInfo &FS);
|
|
|
|
void updateOneNode(const GlobalValue *Callee) {
|
|
|
|
updateOneNode(Callee, Functions.find(Callee)->second);
|
|
|
|
}
|
|
|
|
void updateAllNodes() {
|
|
|
|
for (auto &F : Functions)
|
|
|
|
updateOneNode(F.first, F.second);
|
|
|
|
}
|
|
|
|
void runDataFlow();
|
2019-03-02 01:15:21 +08:00
|
|
|
#ifndef NDEBUG
|
2018-11-27 07:05:58 +08:00
|
|
|
void verifyFixedPoint();
|
2019-03-02 01:15:21 +08:00
|
|
|
#endif
|
2018-11-27 07:05:58 +08:00
|
|
|
|
|
|
|
public:
|
|
|
|
StackSafetyDataFlowAnalysis(
|
|
|
|
Module &M, std::function<const StackSafetyInfo &(Function &)> FI);
|
|
|
|
StackSafetyGlobalInfo run();
|
|
|
|
};
|
|
|
|
|
|
|
|
StackSafetyDataFlowAnalysis::StackSafetyDataFlowAnalysis(
|
|
|
|
Module &M, std::function<const StackSafetyInfo &(Function &)> FI)
|
|
|
|
: PointerSize(M.getDataLayout().getPointerSizeInBits()),
|
|
|
|
UnknownRange(PointerSize, true) {
|
|
|
|
// Without ThinLTO, run the local analysis for every function in the TU and
|
2018-11-27 09:56:44 +08:00
|
|
|
// then run the DFA.
|
2018-11-27 07:05:58 +08:00
|
|
|
for (auto &F : M.functions())
|
|
|
|
if (!F.isDeclaration())
|
|
|
|
Functions.emplace(&F, FI(F));
|
|
|
|
for (auto &A : M.aliases())
|
|
|
|
if (isa<Function>(A.getBaseObject()))
|
2018-11-27 09:56:26 +08:00
|
|
|
Functions.emplace(&A, StackSafetyInfo::FunctionInfo(&A));
|
2018-11-27 07:05:58 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
ConstantRange
|
|
|
|
StackSafetyDataFlowAnalysis::getArgumentAccessRange(const GlobalValue *Callee,
|
|
|
|
unsigned ParamNo) const {
|
|
|
|
auto IT = Functions.find(Callee);
|
|
|
|
// Unknown callee (outside of LTO domain or an indirect call).
|
|
|
|
if (IT == Functions.end())
|
|
|
|
return UnknownRange;
|
|
|
|
const StackSafetyInfo::FunctionInfo &FS = IT->second;
|
|
|
|
// The definition of this symbol may not be the definition in this linkage
|
|
|
|
// unit.
|
|
|
|
if (!FS.IsDSOLocal() || FS.IsInterposable())
|
|
|
|
return UnknownRange;
|
|
|
|
if (ParamNo >= FS.Params.size()) // possibly vararg
|
|
|
|
return UnknownRange;
|
|
|
|
return FS.Params[ParamNo].Use.Range;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool StackSafetyDataFlowAnalysis::updateOneUse(UseInfo &US,
|
|
|
|
bool UpdateToFullSet) {
|
|
|
|
bool Changed = false;
|
|
|
|
for (auto &CS : US.Calls) {
|
2018-11-27 09:56:35 +08:00
|
|
|
assert(!CS.Offset.isEmptySet() &&
|
|
|
|
"Param range can't be empty-set, invalid offset range");
|
2018-11-27 07:05:58 +08:00
|
|
|
|
|
|
|
ConstantRange CalleeRange = getArgumentAccessRange(CS.Callee, CS.ParamNo);
|
|
|
|
CalleeRange = CalleeRange.add(CS.Offset);
|
|
|
|
if (!US.Range.contains(CalleeRange)) {
|
|
|
|
Changed = true;
|
|
|
|
if (UpdateToFullSet)
|
|
|
|
US.Range = UnknownRange;
|
|
|
|
else
|
|
|
|
US.Range = US.Range.unionWith(CalleeRange);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return Changed;
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyDataFlowAnalysis::updateOneNode(
|
|
|
|
const GlobalValue *Callee, StackSafetyInfo::FunctionInfo &FS) {
|
|
|
|
bool UpdateToFullSet = FS.UpdateCount > StackSafetyMaxIterations;
|
|
|
|
bool Changed = false;
|
|
|
|
for (auto &AS : FS.Allocas)
|
|
|
|
Changed |= updateOneUse(AS.Use, UpdateToFullSet);
|
|
|
|
for (auto &PS : FS.Params)
|
|
|
|
Changed |= updateOneUse(PS.Use, UpdateToFullSet);
|
|
|
|
|
|
|
|
if (Changed) {
|
|
|
|
LLVM_DEBUG(dbgs() << "=== update [" << FS.UpdateCount
|
|
|
|
<< (UpdateToFullSet ? ", full-set" : "") << "] "
|
|
|
|
<< FS.getName() << "\n");
|
|
|
|
// Callers of this function may need updating.
|
|
|
|
for (auto &CallerID : Callers[Callee])
|
|
|
|
WorkList.insert(CallerID);
|
|
|
|
|
|
|
|
++FS.UpdateCount;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyDataFlowAnalysis::runDataFlow() {
|
|
|
|
Callers.clear();
|
|
|
|
WorkList.clear();
|
|
|
|
|
|
|
|
SmallVector<const GlobalValue *, 16> Callees;
|
|
|
|
for (auto &F : Functions) {
|
|
|
|
Callees.clear();
|
|
|
|
StackSafetyInfo::FunctionInfo &FS = F.second;
|
|
|
|
for (auto &AS : FS.Allocas)
|
|
|
|
for (auto &CS : AS.Use.Calls)
|
|
|
|
Callees.push_back(CS.Callee);
|
|
|
|
for (auto &PS : FS.Params)
|
|
|
|
for (auto &CS : PS.Use.Calls)
|
|
|
|
Callees.push_back(CS.Callee);
|
|
|
|
|
|
|
|
llvm::sort(Callees);
|
|
|
|
Callees.erase(std::unique(Callees.begin(), Callees.end()), Callees.end());
|
|
|
|
|
|
|
|
for (auto &Callee : Callees)
|
|
|
|
Callers[Callee].push_back(F.first);
|
|
|
|
}
|
|
|
|
|
|
|
|
updateAllNodes();
|
|
|
|
|
|
|
|
while (!WorkList.empty()) {
|
|
|
|
const GlobalValue *Callee = WorkList.back();
|
|
|
|
WorkList.pop_back();
|
|
|
|
updateOneNode(Callee);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-02 01:15:21 +08:00
|
|
|
#ifndef NDEBUG
|
2018-11-27 07:05:58 +08:00
|
|
|
void StackSafetyDataFlowAnalysis::verifyFixedPoint() {
|
|
|
|
WorkList.clear();
|
|
|
|
updateAllNodes();
|
|
|
|
assert(WorkList.empty());
|
|
|
|
}
|
2019-03-02 01:15:21 +08:00
|
|
|
#endif
|
2018-11-27 07:05:58 +08:00
|
|
|
|
|
|
|
StackSafetyGlobalInfo StackSafetyDataFlowAnalysis::run() {
|
|
|
|
runDataFlow();
|
|
|
|
LLVM_DEBUG(verifyFixedPoint());
|
|
|
|
|
|
|
|
StackSafetyGlobalInfo SSI;
|
|
|
|
for (auto &F : Functions)
|
|
|
|
SSI.emplace(F.first, std::move(F.second));
|
|
|
|
return SSI;
|
|
|
|
}
|
|
|
|
|
2018-11-27 07:05:48 +08:00
|
|
|
void print(const StackSafetyGlobalInfo &SSI, raw_ostream &O, const Module &M) {
|
2018-11-27 07:05:58 +08:00
|
|
|
size_t Count = 0;
|
|
|
|
for (auto &F : M.functions())
|
|
|
|
if (!F.isDeclaration()) {
|
|
|
|
SSI.find(&F)->second.print(O);
|
|
|
|
O << "\n";
|
|
|
|
++Count;
|
|
|
|
}
|
|
|
|
for (auto &A : M.aliases()) {
|
|
|
|
SSI.find(&A)->second.print(O);
|
|
|
|
O << "\n";
|
|
|
|
++Count;
|
|
|
|
}
|
|
|
|
assert(Count == SSI.size() && "Unexpected functions in the result");
|
2018-11-27 07:05:48 +08:00
|
|
|
}
|
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
} // end anonymous namespace
|
|
|
|
|
|
|
|
StackSafetyInfo::StackSafetyInfo() = default;
|
|
|
|
StackSafetyInfo::StackSafetyInfo(StackSafetyInfo &&) = default;
|
|
|
|
StackSafetyInfo &StackSafetyInfo::operator=(StackSafetyInfo &&) = default;
|
|
|
|
|
|
|
|
StackSafetyInfo::StackSafetyInfo(FunctionInfo &&Info)
|
|
|
|
: Info(new FunctionInfo(std::move(Info))) {}
|
|
|
|
|
|
|
|
StackSafetyInfo::~StackSafetyInfo() = default;
|
|
|
|
|
|
|
|
void StackSafetyInfo::print(raw_ostream &O) const { Info->print(O); }
|
|
|
|
|
|
|
|
AnalysisKey StackSafetyAnalysis::Key;
|
2018-11-27 05:57:47 +08:00
|
|
|
|
|
|
|
StackSafetyInfo StackSafetyAnalysis::run(Function &F,
|
|
|
|
FunctionAnalysisManager &AM) {
|
2018-11-27 05:57:59 +08:00
|
|
|
StackSafetyLocalAnalysis SSLA(F, AM.getResult<ScalarEvolutionAnalysis>(F));
|
|
|
|
return SSLA.run();
|
2018-11-27 05:57:47 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PreservedAnalyses StackSafetyPrinterPass::run(Function &F,
|
|
|
|
FunctionAnalysisManager &AM) {
|
|
|
|
OS << "'Stack Safety Local Analysis' for function '" << F.getName() << "'\n";
|
|
|
|
AM.getResult<StackSafetyAnalysis>(F).print(OS);
|
|
|
|
return PreservedAnalyses::all();
|
|
|
|
}
|
|
|
|
|
|
|
|
char StackSafetyInfoWrapperPass::ID = 0;
|
|
|
|
|
|
|
|
StackSafetyInfoWrapperPass::StackSafetyInfoWrapperPass() : FunctionPass(ID) {
|
|
|
|
initializeStackSafetyInfoWrapperPassPass(*PassRegistry::getPassRegistry());
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyInfoWrapperPass::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
|
|
AU.addRequired<ScalarEvolutionWrapperPass>();
|
|
|
|
AU.setPreservesAll();
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyInfoWrapperPass::print(raw_ostream &O, const Module *M) const {
|
|
|
|
SSI.print(O);
|
|
|
|
}
|
|
|
|
|
2018-11-27 05:57:59 +08:00
|
|
|
bool StackSafetyInfoWrapperPass::runOnFunction(Function &F) {
|
|
|
|
StackSafetyLocalAnalysis SSLA(
|
|
|
|
F, getAnalysis<ScalarEvolutionWrapperPass>().getSE());
|
|
|
|
SSI = StackSafetyInfo(SSLA.run());
|
|
|
|
return false;
|
|
|
|
}
|
2018-11-27 05:57:47 +08:00
|
|
|
|
2018-11-27 07:05:48 +08:00
|
|
|
AnalysisKey StackSafetyGlobalAnalysis::Key;
|
|
|
|
|
|
|
|
StackSafetyGlobalInfo
|
|
|
|
StackSafetyGlobalAnalysis::run(Module &M, ModuleAnalysisManager &AM) {
|
2018-11-27 07:05:58 +08:00
|
|
|
FunctionAnalysisManager &FAM =
|
|
|
|
AM.getResult<FunctionAnalysisManagerModuleProxy>(M).getManager();
|
|
|
|
|
|
|
|
StackSafetyDataFlowAnalysis SSDFA(
|
|
|
|
M, [&FAM](Function &F) -> const StackSafetyInfo & {
|
|
|
|
return FAM.getResult<StackSafetyAnalysis>(F);
|
|
|
|
});
|
|
|
|
return SSDFA.run();
|
2018-11-27 07:05:48 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PreservedAnalyses StackSafetyGlobalPrinterPass::run(Module &M,
|
|
|
|
ModuleAnalysisManager &AM) {
|
|
|
|
OS << "'Stack Safety Analysis' for module '" << M.getName() << "'\n";
|
|
|
|
print(AM.getResult<StackSafetyGlobalAnalysis>(M), OS, M);
|
|
|
|
return PreservedAnalyses::all();
|
|
|
|
}
|
|
|
|
|
|
|
|
char StackSafetyGlobalInfoWrapperPass::ID = 0;
|
|
|
|
|
|
|
|
StackSafetyGlobalInfoWrapperPass::StackSafetyGlobalInfoWrapperPass()
|
|
|
|
: ModulePass(ID) {
|
|
|
|
initializeStackSafetyGlobalInfoWrapperPassPass(
|
|
|
|
*PassRegistry::getPassRegistry());
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyGlobalInfoWrapperPass::print(raw_ostream &O,
|
|
|
|
const Module *M) const {
|
|
|
|
::print(SSI, O, *M);
|
|
|
|
}
|
|
|
|
|
|
|
|
void StackSafetyGlobalInfoWrapperPass::getAnalysisUsage(
|
|
|
|
AnalysisUsage &AU) const {
|
|
|
|
AU.addRequired<StackSafetyInfoWrapperPass>();
|
|
|
|
}
|
|
|
|
|
2018-11-27 07:05:58 +08:00
|
|
|
bool StackSafetyGlobalInfoWrapperPass::runOnModule(Module &M) {
|
|
|
|
StackSafetyDataFlowAnalysis SSDFA(
|
|
|
|
M, [this](Function &F) -> const StackSafetyInfo & {
|
|
|
|
return getAnalysis<StackSafetyInfoWrapperPass>(F).getResult();
|
|
|
|
});
|
|
|
|
SSI = SSDFA.run();
|
|
|
|
return false;
|
|
|
|
}
|
2018-11-27 07:05:48 +08:00
|
|
|
|
2018-11-27 05:57:47 +08:00
|
|
|
static const char LocalPassArg[] = "stack-safety-local";
|
|
|
|
static const char LocalPassName[] = "Stack Safety Local Analysis";
|
|
|
|
INITIALIZE_PASS_BEGIN(StackSafetyInfoWrapperPass, LocalPassArg, LocalPassName,
|
|
|
|
false, true)
|
|
|
|
INITIALIZE_PASS_DEPENDENCY(ScalarEvolutionWrapperPass)
|
|
|
|
INITIALIZE_PASS_END(StackSafetyInfoWrapperPass, LocalPassArg, LocalPassName,
|
|
|
|
false, true)
|
2018-11-27 07:05:48 +08:00
|
|
|
|
|
|
|
static const char GlobalPassName[] = "Stack Safety Analysis";
|
|
|
|
INITIALIZE_PASS_BEGIN(StackSafetyGlobalInfoWrapperPass, DEBUG_TYPE,
|
|
|
|
GlobalPassName, false, false)
|
|
|
|
INITIALIZE_PASS_DEPENDENCY(StackSafetyInfoWrapperPass)
|
|
|
|
INITIALIZE_PASS_END(StackSafetyGlobalInfoWrapperPass, DEBUG_TYPE,
|
|
|
|
GlobalPassName, false, false)
|