bounds checking: add support for byval arguments
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@157498 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Scalar/BoundsChecking.cpp b/lib/Transforms/Scalar/BoundsChecking.cpp
index 4c3dea2..7b1a417 100644
--- a/lib/Transforms/Scalar/BoundsChecking.cpp
+++ b/lib/Transforms/Scalar/BoundsChecking.cpp
@@ -104,7 +104,6 @@
BasicBlock *Cont = OldBB->splitBasicBlock(Inst);
OldBB->getTerminator()->eraseFromParent();
- // FIXME: add unlikely branch taken metadata?
if (Cmp)
BranchInst::Create(getTrapBB(), Cont, Cmp, OldBB);
else
@@ -152,6 +151,15 @@
SizeValue = Builder->CreateMul(SizeValue, ArraySize);
return NotConst;
+ // function arguments
+ } else if (Argument *A = dyn_cast<Argument>(Alloc)) {
+ if (!A->hasByValAttr())
+ return Dunno;
+
+ PointerType *PT = cast<PointerType>(A->getType());
+ Size = TD->getTypeAllocSize(PT->getElementType());
+ return Const;
+
// ptr = select(ptr1, ptr2)
} else if (SelectInst *SI = dyn_cast<SelectInst>(Alloc)) {
uint64_t SizeFalse;