重载new和delete来检测内存泄漏

1. 简述

    内存泄漏属于资源泄漏的一种，百度百科将内存泄漏分为四种：常发性内存泄漏、偶发性内存泄漏、一次性内存泄漏和隐式内存泄漏。
    常发性指：内存泄漏的代码会被多次执行到。偶发性指：内存泄漏的代码只有在特定的条件下才会执行到。一次性指：内存泄漏的代码只会被执行到一次。隐式指：程序在运行中不断的开辟内存，知道程序结束时才释放内存，本质上虽然没有内存泄漏，但是如果这个程序在连续运行很长时间，会耗尽所有内存，导致系统崩溃。
    下面首先介绍内存检测的基本原理，然后给出代码样例，最后说明针对四种内存泄漏进行检测的想法。

2. 基本原理

    内存泄漏就是new出来的内存没有通过delete合理的释放掉。new和delete这两个函数就是关键点。可以重载new和delete，每次new中开辟一块内存就用链表把这个内存的信息保存下来，每次用delete删除一块内存就从链表中删除这块内存的记录。
3. 代码样例

  1  #include < iostream >
  2  using   namespace  std;
  3  // ---------------------------------------------------------------
  4  //  内存记录
  5  // ---------------------------------------------------------------
  6  class  MemInfo {
  7  private :
  8     void *  ptr;
  9     const   char *  file;
 10    unsigned  int  line;
 11    MemInfo *  link;
 12    friend  class  MemStack;
 13  };
 14  // ---------------------------------------------------------------
 15  //  内存记录栈 
 16  // ---------------------------------------------------------------
 17  class  MemStack {
 18  private :
 19    MemInfo *  head;
 20  public :
 21    MemStack():head(NULL) { }
 22     ~ MemStack() { 
 23      MemInfo *  tmp;
 24       while (head  !=  NULL) {
 25        free(head -> ptr);  //  释放泄漏的内存 
 26        tmp  =  head -> link;
 27        free(head);
 28        head  =  tmp;
 29      }
 30    }
 31     void  Insert( void *  ptr,  const   char *  file, unsigned  int  line) {
 32      MemInfo *  node  =  (MemInfo * )malloc( sizeof (MemInfo));
 33      node -> ptr  =  ptr; node -> file  =  file; node -> line = line;
 34      node -> link  =  head; head  =  node;    
 35    }
 36     void  Delete( void *  ptr) {
 37      MemInfo *  node  =  head;
 38      MemInfo *  pre  =  NULL;
 39       while (node  !=  NULL  &&  node -> ptr != ptr) {
 40        pre  =  node;
 41        node  =  node -> link;
 42      }
 43       if (node  ==  NULL)
 44        cout  <<   " 删除一个没有开辟的内存 "   <<  endl;
 45       else  {
 46         if (pre  ==  NULL)  //  删除的是head 
 47          head  =  node -> link;
 48         else  
 49          pre -> link  =  node -> link;
 50        free(node);
 51      }
 52    }
 53     void  Print() {
 54       if (head  ==  NULL) {
 55        cout  <<   " 内存都释放掉了 "   <<  endl; 
 56         return ;
 57      }
 58      cout  <<   " 有内存泄露出现 "   <<  endl; 
 59      MemInfo *  node  =  head;    
 60       while (node  !=  NULL) {
 61        cout  <<   " 文件名:  "   <<  node -> file  <<   "  ,  "   <<   " 行数:  "   <<  node -> line  <<   "  ,  "
 62           <<   " 地址:  "   <<  node -> ptr  <<  endl; 
 63        node  =  node -> link;
 64      }
 65    }
 66  };
 67  // ---------------------------------------------------------------
 68  //  全局对象 mem_stack记录开辟的内存 
 69  // ---------------------------------------------------------------
 70  MemStack mem_stack;
 71  // ---------------------------------------------------------------
 72  //  重载new,new[],delete,delete[] 
 73  // ---------------------------------------------------------------
 74  void *   operator   new (size_t size,  const   char *  file, unsigned  int  line) {
 75     void *  ptr  =  malloc(size);
 76    mem_stack.Insert(ptr, file, line);
 77     return  ptr;
 78  }
 79  void *   operator   new [](size_t size,  const   char *  file, unsigned  int  line) {
 80     return   operator   new (size, file, line);  //  不能用new 
 81  }
 82  void   operator  delete( void *  ptr) {
 83    free(ptr);
 84    mem_stack.Delete(ptr);
 85  }
 86  void   operator  delete[]( void *  ptr) {
 87     operator  delete(ptr);
 88  }
 89  // ---------------------------------------------------------------
 90  //  使用宏将带测试代码中的new和delte替换为重载的new和delete 
 91  // ---------------------------------------------------------------
 92  #define  new new(__FILE__,__LINE__)
 93  // ---------------------------------------------------------------
 94  //  待测试代码 
 95  // ---------------------------------------------------------------
 96  void  bad_code() {
 97     int   * p  =   new   int ;
 98     char   * q  =   new   char [ 5 ];
 99    delete []q;
100  } 
101 
102  void  good_code() {
103     int   * p  =   new   int ;
104     char   * q  =   new   char [ 5 ];
105    delete p;
106    delete []q;
107  } 
108  // ---------------------------------------------------------------
109  //  测试过程 
110  // ---------------------------------------------------------------
111  int  main() {
112    good_code();
113    bad_code();
114    mem_stack.Print();
115    system( " PAUSE " );
116     return   0 ;
117  }

    输出结果为：
    
    可见97行开辟的int，没有delete掉，输出结果也显示为97行。

4. 代码说明

4.1 关于new的参数问题。
    对于new int，编译器会解释为new(sizeof(int))，对于new int[5]，编译器会解释为new(sizeof(int)*5)。因此使用宏定义预编译后，new int就变为new (__FILE__,__LINE__) int，编译器会解释为new(sizeof(int), __FILE__,__LINE__)。

4.2 关于MemStack
    MemStack内部也是一个链表结构，注意内部实现不能使用new和delete，只能使用malloc和free来实现链表，因为待测代码中的重载new和delete中调用了MemStack的insert和delete函数，如果insert和delete函数也调用重载后的new和delete的话，会构成死循环的，所以直接使用free和malloc比较好。
    MemStack中的析构函数，会释放掉泄漏掉的内存。

5. 使用思考

    对于常发性和一次性的内存泄漏代码，直接放入测试就好了。对于偶发性的内存泄漏代码，只要满足特定条件，那么也就转化为常发性或者一次性的内存泄漏了。
    对于隐式内存泄漏，由于程序是在很长一段时间之后导致内存耗尽，我们需要长时间观察，每隔一段时间比较一下内存的使用量，如果在一个较长的时间内，内存使用量持续增加，那么可以考虑是内存泄漏。不过调试起来可能会比较麻烦，还是需要重新审视程序设计的。

6. 参考

    百度百科_内存泄漏：介绍内存泄漏的基本分类。
    http://baike.baidu.com/view/714962.htm
    如何检查内存泄漏-重载new和delete：十分生动的说明。
    http://www.cppblog.com/dawnbreak/articles/76223.html
    一个跨平台的C++内存泄漏检测器：十分专业化的讲解和实现。
    http://www.ibm.com/developerworks/cn/linux/l-mleak2/index.html