commit 184b298edcf86613b6d5ba8efc4c6055351a7fcc
author Chris Lattner <sabre@nondot.org> Sun Sep 08 18:59:35 2002 +0000
committer Chris Lattner <sabre@nondot.org> Sun Sep 08 18:59:35 2002 +0000
tree ee5a136b781090cdc326ace1473ea839ef1c2796
parent bdb147c37312365ed70d16c954a3ff066eb0bf78

Enable "garbage detection" of LLVM objects.  Now users should be obnoxious
warnings.  If they accidentally leak LLVM Value's.

llvm-svn: 3620

llvm/lib/VMCore/Function.cpp

diff --git a/llvm/lib/VMCore/Function.cpp b/llvm/lib/VMCore/Function.cpp
index 5901717..fb940bf 100644
--- a/llvm/lib/VMCore/Function.cpp
+++ b/llvm/lib/VMCore/Function.cpp
@@ -8,10 +8,14 @@
 #include "llvm/Module.h"
 #include "llvm/DerivedTypes.h"
 #include "llvm/iOther.h"
+#include "Support/LeakDetector.h"
 #include "SymbolTableListTraitsImpl.h"
 
 BasicBlock *ilist_traits<BasicBlock>::createNode() {
-  return new BasicBlock();
+  BasicBlock *Ret = new BasicBlock();
+  // This should not be garbage monitored.
+  LeakDetector::removeGarbageObject(Ret);
+  return Ret;
 }
 
 iplist<BasicBlock> &ilist_traits<BasicBlock>::getList(Function *F) {
@@ -19,7 +23,10 @@
 }
 
 Argument *ilist_traits<Argument>::createNode() {
-  return new Argument(Type::IntTy);
+  Argument *Ret = new Argument(Type::IntTy);
+  // This should not be garbage monitored.
+  LeakDetector::removeGarbageObject(Ret);
+  return Ret;
 }
 
 iplist<Argument> &ilist_traits<Argument>::getList(Function *F) {
@@ -38,6 +45,10 @@
 Argument::Argument(const Type *Ty, const std::string &Name = "", Function *Par) 
   : Value(Ty, Value::ArgumentVal, Name) {
   Parent = 0;
+
+  // Make sure that we get added to a function
+  LeakDetector::addGarbageObject(this);
+
   if (Par)
     Par->getArgumentList().push_back(this);
 }
@@ -54,7 +65,11 @@
 }
 
 void Argument::setParent(Function *parent) {
+  if (getParent())
+    LeakDetector::addGarbageObject(this);
   Parent = parent;
+  if (getParent())
+    LeakDetector::removeGarbageObject(this);
 }
 
 
@@ -71,6 +86,9 @@
   ArgumentList.setParent(this);
   ParentSymTab = SymTab = 0;
 
+  // Make sure that we get added to a function
+  LeakDetector::addGarbageObject(this);
+
   if (ParentModule)
     ParentModule->getFunctionList().push_back(this);
 }
@@ -97,7 +115,11 @@
 }
 
 void Function::setParent(Module *parent) {
+  if (getParent())
+    LeakDetector::addGarbageObject(this);
   Parent = parent;
+  if (getParent())
+    LeakDetector::removeGarbageObject(this);
 
   // Relink symbol tables together...
   ParentSymTab = Parent ? Parent->getSymbolTableSure() : 0;
@@ -157,12 +179,18 @@
     isConstantGlobal(constant) {
   if (Initializer) Operands.push_back(Use((Value*)Initializer, this));
 
+  LeakDetector::addGarbageObject(this);
+
   if (ParentModule)
     ParentModule->getGlobalList().push_back(this);
 }
 
 void GlobalVariable::setParent(Module *parent) {
+  if (getParent())
+    LeakDetector::addGarbageObject(this);
   Parent = parent;
+  if (getParent())
+    LeakDetector::removeGarbageObject(this);
 }
 
 // Specialize setName to take care of symbol table majik