lib/Support/LeakDetector.cpp

   1 //===-- LeakDetector.cpp - Implement LeakDetector interface ---------------===//
   2 //
   3 // This file implements the LeakDetector class.
   4 //
   5 //===----------------------------------------------------------------------===//
   6
   7 #include "Support/LeakDetector.h"
   8 #include "llvm/Value.h"
   9 #include <set>
  10 #include <iostream>
  11
  12 // Lazily allocate set so that release build doesn't have to do anything.
  13 static std::set<const void*> *Objects = 0;
  14 static std::set<const Value*> *LLVMObjects = 0;
  15
  16 void LeakDetector::addGarbageObjectImpl(void *Object) {
  17   if (Objects == 0)
  18     Objects = new std::set<const void*>();
  19   assert(Objects->count(Object) == 0 && "Object already in set!");
  20   Objects->insert(Object);
  21 }
  22
  23 void LeakDetector::removeGarbageObjectImpl(void *Object) {
  24   if (Objects)
  25     Objects->erase(Object);
  26 }
  27
  28 void LeakDetector::addGarbageObjectImpl(const Value *Object) {
  29   if (LLVMObjects == 0)
  30     LLVMObjects = new std::set<const Value*>();
  31   assert(LLVMObjects->count(Object) == 0 && "Object already in set!");
  32   LLVMObjects->insert(Object);
  33 }
  34
  35 void LeakDetector::removeGarbageObjectImpl(const Value *Object) {
  36   if (LLVMObjects)
  37     LLVMObjects->erase(Object);
  38 }
  39
  40 void LeakDetector::checkForGarbageImpl(const std::string &Message) {
  41   if ((Objects && !Objects->empty()) || (LLVMObjects && !LLVMObjects->empty())){
  42     std::cerr << "Leaked objects found: " << Message << "\n";
  43
  44     if (Objects && !Objects->empty()) {
  45       std::cerr << "  Non-Value objects leaked:";
  46       for (std::set<const void*>::iterator I = Objects->begin(),
  47              E = Objects->end(); I != E; ++I)
  48         std::cerr << " " << *I;
  49     }
  50
  51     if (LLVMObjects && !LLVMObjects->empty()) {
  52       std::cerr << "  LLVM Value subclasses leaked:";
  53       for (std::set<const Value*>::iterator I = LLVMObjects->begin(),
  54              E = LLVMObjects->end(); I != E; ++I)
  55         std::cerr << **I << "\n";
  56     }
  57
  58     std::cerr << "This is probably because you removed an LLVM value "
  59               << "(Instruction, BasicBlock, \netc), but didn't delete it.  "
  60               << "Please check your code for memory leaks.\n";
  61
  62     // Clear out results so we don't get duplicate warnings on next call...
  63     delete Objects; delete LLVMObjects;
  64     Objects = 0; LLVMObjects = 0;
  65   }
  66 }