From 537d8d99be3c25b74c1cd090c9acbf9bda2d97f6 Mon Sep 17 00:00:00 2001 From: Victor Hernandez Date: Fri, 18 Sep 2009 21:34:51 +0000 Subject: Enhance analysis passes so that they apply the same analysis to malloc calls as to MallocInst. Reviewed by Eli Friedman. llvm-svn: 82281 --- llvm/lib/Analysis/PointerTracking.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'llvm/lib/Analysis/PointerTracking.cpp') diff --git a/llvm/lib/Analysis/PointerTracking.cpp b/llvm/lib/Analysis/PointerTracking.cpp index e098647887b..22818369b8b 100644 --- a/llvm/lib/Analysis/PointerTracking.cpp +++ b/llvm/lib/Analysis/PointerTracking.cpp @@ -13,6 +13,7 @@ #include "llvm/Analysis/ConstantFolding.h" #include "llvm/Analysis/Dominators.h" #include "llvm/Analysis/LoopInfo.h" +#include "llvm/Analysis/MallocHelper.h" #include "llvm/Analysis/PointerTracking.h" #include "llvm/Analysis/ScalarEvolution.h" #include "llvm/Analysis/ScalarEvolutionExpressions.h" @@ -99,6 +100,14 @@ const SCEV *PointerTracking::computeAllocationCount(Value *P, return SE->getSCEV(arraySize); } + if (CallInst *CI = extractMallocCall(V)) { + Value *arraySize = getMallocArraySize(CI, P->getContext(), TD); + Ty = getMallocAllocatedType(CI); + if (!Ty || !arraySize) return SE->getCouldNotCompute(); + // arraySize elements of type Ty. + return SE->getSCEV(arraySize); + } + if (GlobalVariable *GV = dyn_cast(V)) { if (GV->hasDefinitiveInitializer()) { Constant *C = GV->getInitializer(); -- cgit v1.2.3