1 //===---------------- ----LeaksContext.h - Implementation ------*- C++ -*--===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines various helper methods and classes used by
11 // LLVMContextImpl for leaks detectors.
13 //===----------------------------------------------------------------------===//
15 #include "llvm/Value.h"
16 #include "llvm/Support/Streams.h"
17 #include "llvm/ADT/SmallPtrSet.h"
23 static void print(const T* P) { cerr << P; }
27 struct PrinterTrait<Value> {
28 static void print(const Value* P) { cerr << *P; }
32 struct LeakDetectorImpl {
33 explicit LeakDetectorImpl(const char* const name = "") :
34 Cache(0), Name(name) { }
41 void setName(const char* n) {
45 // Because the most common usage pattern, by far, is to add a
46 // garbage object, then remove it immediately, we optimize this
47 // case. When an object is added, it is not added to the set
48 // immediately, it is added to the CachedValue Value. If it is
49 // immediately removed, no set search need be performed.
50 void addGarbage(const T* o) {
52 assert(Ts.count(Cache) == 0 && "Object already in set!");
58 void removeGarbage(const T* o) {
60 Cache = 0; // Cache hit
65 bool hasGarbage(const std::string& Message) {
66 addGarbage(0); // Flush the Cache
68 assert(Cache == 0 && "No value should be cached anymore!");
71 cerr << "Leaked " << Name << " objects found: " << Message << ":\n";
72 for (typename SmallPtrSet<const T*, 8>::iterator I = Ts.begin(),
73 E = Ts.end(); I != E; ++I) {
75 PrinterTrait<T>::print(*I);
87 SmallPtrSet<const T*, 8> Ts;