mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-12-30 15:45:26 +00:00
83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
//===-- LeakDetector.cpp - Implement LeakDetector interface ---------------===//
|
|
//
|
|
// This file implements the LeakDetector class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Support/LeakDetector.h"
|
|
#include "llvm/Value.h"
|
|
#include <set>
|
|
|
|
// Lazily allocate set so that release build doesn't have to do anything.
|
|
static std::set<const void*> *Objects = 0;
|
|
static std::set<const Value*> *LLVMObjects = 0;
|
|
|
|
// Because the most common usage pattern, by far, is to add a garbage object,
|
|
// then remove it immediately, we optimize this case. When an object is added,
|
|
// it is not added to the set immediately, it is added to the CachedValue Value.
|
|
// If it is immediately removed, no set search need be performed.
|
|
//
|
|
static const Value *CachedValue;
|
|
|
|
void LeakDetector::addGarbageObjectImpl(void *Object) {
|
|
if (Objects == 0)
|
|
Objects = new std::set<const void*>();
|
|
assert(Objects->count(Object) == 0 && "Object already in set!");
|
|
Objects->insert(Object);
|
|
}
|
|
|
|
void LeakDetector::removeGarbageObjectImpl(void *Object) {
|
|
if (Objects)
|
|
Objects->erase(Object);
|
|
}
|
|
|
|
void LeakDetector::addGarbageObjectImpl(const Value *Object) {
|
|
if (CachedValue) {
|
|
if (LLVMObjects == 0)
|
|
LLVMObjects = new std::set<const Value*>();
|
|
assert(LLVMObjects->count(CachedValue) == 0 && "Object already in set!");
|
|
LLVMObjects->insert(CachedValue);
|
|
}
|
|
CachedValue = Object;
|
|
}
|
|
|
|
void LeakDetector::removeGarbageObjectImpl(const Value *Object) {
|
|
if (Object == CachedValue)
|
|
CachedValue = 0; // Cache hit!
|
|
else if (LLVMObjects)
|
|
LLVMObjects->erase(Object);
|
|
}
|
|
|
|
void LeakDetector::checkForGarbageImpl(const std::string &Message) {
|
|
if (CachedValue) // Flush the cache to the set...
|
|
addGarbageObjectImpl((Value*)0);
|
|
|
|
assert(CachedValue == 0 && "No value should be cached anymore!");
|
|
|
|
if ((Objects && !Objects->empty()) || (LLVMObjects && !LLVMObjects->empty())){
|
|
std::cerr << "Leaked objects found: " << Message << "\n";
|
|
|
|
if (Objects && !Objects->empty()) {
|
|
std::cerr << " Non-Value objects leaked:";
|
|
for (std::set<const void*>::iterator I = Objects->begin(),
|
|
E = Objects->end(); I != E; ++I)
|
|
std::cerr << " " << *I;
|
|
}
|
|
|
|
if (LLVMObjects && !LLVMObjects->empty()) {
|
|
std::cerr << " LLVM Value subclasses leaked:";
|
|
for (std::set<const Value*>::iterator I = LLVMObjects->begin(),
|
|
E = LLVMObjects->end(); I != E; ++I)
|
|
std::cerr << **I << "\n";
|
|
}
|
|
|
|
std::cerr << "This is probably because you removed an LLVM value "
|
|
<< "(Instruction, BasicBlock, \netc), but didn't delete it. "
|
|
<< "Please check your code for memory leaks.\n";
|
|
|
|
// Clear out results so we don't get duplicate warnings on next call...
|
|
delete Objects; delete LLVMObjects;
|
|
Objects = 0; LLVMObjects = 0;
|
|
}
|
|
}
|