重载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