1
//===- LeaksContext.h - LeadDetector 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/ADT/SmallPtrSet.h"
21
static void print(const T* P) { errs() << P; }
25
struct PrinterTrait<Value> {
26
static void print(const Value* P) { errs() << *P; }
30
struct LeakDetectorImpl {
31
explicit LeakDetectorImpl(const char* const name = "") :
32
Cache(0), Name(name) { }
39
void setName(const char* n) {
43
// Because the most common usage pattern, by far, is to add a
44
// garbage object, then remove it immediately, we optimize this
45
// case. When an object is added, it is not added to the set
46
// immediately, it is added to the CachedValue Value. If it is
47
// immediately removed, no set search need be performed.
48
void addGarbage(const T* o) {
49
assert(Ts.count(o) == 0 && "Object already in set!");
51
assert(Cache != o && "Object already in set!");
57
void removeGarbage(const T* o) {
59
Cache = 0; // Cache hit
64
bool hasGarbage(const std::string& Message) {
65
addGarbage(0); // Flush the Cache
67
assert(Cache == 0 && "No value should be cached anymore!");
70
errs() << "Leaked " << Name << " objects found: " << Message << ":\n";
71
for (typename SmallPtrSet<const T*, 8>::iterator I = Ts.begin(),
72
E = Ts.end(); I != E; ++I) {
74
PrinterTrait<T>::print(*I);
86
SmallPtrSet<const T*, 8> Ts;