From e05f66ef2e37bcbd88acaeef2bf14ddb1f04488f Mon Sep 17 00:00:00 2001
From: Victor Hernandez <vhernandez@apple.com>
Date: Thu, 14 Jan 2010 20:12:34 +0000
Subject: [PATCH] In debug builds, assert that function-local metadata has only
 1 parent function

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@93449 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/VMCore/Metadata.cpp | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/lib/VMCore/Metadata.cpp b/lib/VMCore/Metadata.cpp
index 1e767ffba69..7988b446fb6 100644
--- a/lib/VMCore/Metadata.cpp
+++ b/lib/VMCore/Metadata.cpp
@@ -121,9 +121,39 @@ MDNode::~MDNode() {
     Op->~MDNodeOperand();
 }
 
+#ifndef NDEBUG
+static Function *assertLocalFunction(const MDNode *N,
+                                     SmallPtrSet<const MDNode *, 32> &Visited) {
+  Function *F = NULL;
+  // Only visit each MDNode once.
+  if (!Visited.insert(N)) return F;
+  
+  for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
+    Value *V = N->getOperand(i);
+    Function *NewF = NULL;
+    if (!V) continue;
+    if (Instruction *I = dyn_cast<Instruction>(V))
+      NewF = I->getParent()->getParent();
+    else if (BasicBlock *BB = dyn_cast<BasicBlock>(V))
+      NewF = BB->getParent();
+    else if (Argument *A = dyn_cast<Argument>(V))
+      NewF = A->getParent();
+    else if (MDNode *MD = dyn_cast<MDNode>(V))
+      if (MD->isFunctionLocal())
+        NewF = assertLocalFunction(MD, Visited);
+    if (F && NewF) assert(F == NewF && "inconsistent function-local metadata");
+    if (!F) F = NewF;
+  }
+  return F;
+}
+#endif
+
 static Function *getFunctionHelper(const MDNode *N,
                                    SmallPtrSet<const MDNode *, 32> &Visited) {
   assert(N->isFunctionLocal() && "Should only be called on function-local MD");
+#ifndef NDEBUG
+  return assertLocalFunction(N, Visited);
+#endif
   Function *F = NULL;
   // Only visit each MDNode once.
   if (!Visited.insert(N)) return F;
@@ -142,7 +172,6 @@ static Function *getFunctionHelper(const MDNode *N,
         F = getFunctionHelper(MD, Visited);
     if (F) break;
   }
-  
   return F;
 }
 
-- 
2.34.1