Add support for tracking array allocations


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@2328 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Analysis/DataStructure/NodeImpl.cpp b/lib/Analysis/DataStructure/NodeImpl.cpp
index 46c12e3..76b82ba 100644
--- a/lib/Analysis/DataStructure/NodeImpl.cpp
+++ b/lib/Analysis/DataStructure/NodeImpl.cpp
@@ -21,6 +21,12 @@
   return false;
 }
 
+void AllocDSNode::mergeInto(DSNode *Node) const {
+  // Make sure the merged node is variable size if this node is var size
+  AllocDSNode *N = cast<AllocDSNode>(Node);
+  N->isVarSize |= isVarSize;
+}
+
 bool GlobalDSNode::isEquivalentTo(DSNode *Node) const {
   if (GlobalDSNode *G = dyn_cast<GlobalDSNode>(Node)) {
     if (G->Val != Val) return false;
@@ -217,8 +223,12 @@
                     (ShadowDSNode*)NodeMap[Old->SynthNodes[i].second]));
 }
 
-AllocDSNode::AllocDSNode(AllocationInst *V)
+AllocDSNode::AllocDSNode(AllocationInst *V, bool isvarsize)
   : DSNode(NewNode, V->getType()->getElementType()), Allocation(V) {
+
+  // Is variable size if incoming flag says so, or if allocation is var size
+  // already.
+  isVarSize = isvarsize || !isa<Constant>(V->getArraySize());
 }
 
 bool AllocDSNode::isAllocaNode() const {
@@ -232,7 +242,7 @@
 
   WriteTypeSymbolic(OS, getType(),
                     Allocation->getParent()->getParent()->getParent());
-  if (Allocation->isArrayAllocation())
+  if (isVarSize)
     OS << "[ ]";
   return OS.str();
 }