1 //===-- LeakDetector.cpp - Implement LeakDetector interface ---------------===//
3 // The LLVM Compiler Infrastructure
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file implements the LeakDetector class.
12 //===----------------------------------------------------------------------===//
14 #include "Support/LeakDetector.h"
15 #include "llvm/Value.h"
18 // Lazily allocate set so that release build doesn't have to do anything.
19 static std::set<const void*> *Objects = 0;
20 static std::set<const Value*> *LLVMObjects = 0;
22 // Because the most common usage pattern, by far, is to add a garbage object,
23 // then remove it immediately, we optimize this case. When an object is added,
24 // it is not added to the set immediately, it is added to the CachedValue Value.
25 // If it is immediately removed, no set search need be performed.
27 static const Value *CachedValue;
29 void LeakDetector::addGarbageObjectImpl(void *Object) {
31 Objects = new std::set<const void*>();
32 assert(Objects->count(Object) == 0 && "Object already in set!");
33 Objects->insert(Object);
36 void LeakDetector::removeGarbageObjectImpl(void *Object) {
38 Objects->erase(Object);
41 void LeakDetector::addGarbageObjectImpl(const Value *Object) {
44 LLVMObjects = new std::set<const Value*>();
45 assert(LLVMObjects->count(CachedValue) == 0 && "Object already in set!");
46 LLVMObjects->insert(CachedValue);
51 void LeakDetector::removeGarbageObjectImpl(const Value *Object) {
52 if (Object == CachedValue)
53 CachedValue = 0; // Cache hit!
55 LLVMObjects->erase(Object);
58 void LeakDetector::checkForGarbageImpl(const std::string &Message) {
59 if (CachedValue) // Flush the cache to the set...
60 addGarbageObjectImpl((Value*)0);
62 assert(CachedValue == 0 && "No value should be cached anymore!");
64 if ((Objects && !Objects->empty()) || (LLVMObjects && !LLVMObjects->empty())){
65 std::cerr << "Leaked objects found: " << Message << "\n";
67 if (Objects && !Objects->empty()) {
68 std::cerr << " Non-Value objects leaked:";
69 for (std::set<const void*>::iterator I = Objects->begin(),
70 E = Objects->end(); I != E; ++I)
71 std::cerr << " " << *I;
74 if (LLVMObjects && !LLVMObjects->empty()) {
75 std::cerr << " LLVM Value subclasses leaked:";
76 for (std::set<const Value*>::iterator I = LLVMObjects->begin(),
77 E = LLVMObjects->end(); I != E; ++I)
78 std::cerr << **I << "\n";
81 std::cerr << "This is probably because you removed an LLVM value "
82 << "(Instruction, BasicBlock, \netc), but didn't delete it. "
83 << "Please check your code for memory leaks.\n";
85 // Clear out results so we don't get duplicate warnings on next call...
86 delete Objects; delete LLVMObjects;
87 Objects = 0; LLVMObjects = 0;