如何查找内存泄露

 

这两天，在查找内存泄露的问题。因为内存都是放在memory pool里，所以不能通过valgrind等工具抓到那个地方分配的内存没有手动被释放。

使用gcc，有一个方法去打包内存分配函数，而且不需要编辑已有的code并且不需要修改目标文件。linker ld 提供了一个内建的选项去替换函数符号。

 

–wrap（一个横线） 表示把函数 func 替换为（两个下划线前缀）__wrap_func 。可以通过这个选项传给gcc去做恰当的链接。

举个例子来说明吧。  如果最后的free函数被注释掉，那么p就会出现内存泄漏，如果calloc已经被打包把内存放到memory pool里，程序退出激活释放memory pool的话。

那么在程序为退出之前，那些mem还是存在的，因此当前进程会暂用很多内存（如果很多分配没有free的话）

 

view plain copy to clipboard print ?

1. #include<iostream>   
2. using namespace std;  
3.   
4. int main()  
5. {  
6.   int allocTimes = 100;  
7.   while(allocTimes--) {  
8.     int *p = (int*)calloc(5, sizeof(int));  
9.     //free(p);   
10.   }  
11. }  

#include<iostream> using namespace std; int main() { int allocTimes = 100; while(allocTimes--) { int *p = (int*)calloc(5, sizeof(int)); //free(p); } }

怎样来检查这些分配没释放的内存呢。就可以使用上面说的，定义一些函数，然后通过编译器传给连接器去替换函数。

把这些函数定义在一个文件中（test.h）：

view plain copy to clipboard print ?

1. #ifndef  TEST_INC   
2. #define  TEST_INC   
3.   
4. #include<iostream>   
5. using namespace std;  
6.   
7. #ifdef __cplusplus   
8. extern "C" {  
9. #endif   
10.   
11. extern void *__real_calloc(size_t nmemb, size_t size);  
12. extern void *__real_realloc(void *ptr, size_t size);  
13. extern void *__real_malloc(size_t size);  
14. extern void __real_free(void *ptr);  
15.   
16. void *__wrap_calloc(size_t numemb, size_t size);  
17. void *__wrap_realloc(void *ptr, size_t size);  
18. void *__wrap_malloc(size_t size);  
19. void __wrap_free(void *ptr);  
20.   
21. /* wrap calloc */  
22. void * __wrap_calloc(size_t numemb, size_t size)  
23. {  
24.   cout << "wrap alloc" << endl;  
25.   return __real_calloc(numemb, size);  
26. }  
27.   
28. /* wrap realloc */  
29. void *  
30. __wrap_realloc(void *ptr, size_t size)  
31. {  
32.   cout << "wrap realloc" << endl;  
33.   
34.   return (void*)__real_realloc(ptr, size);  
35. }  
36.   
37. /* wrap malloc */  
38. void *  
39. __wrap_malloc(size_t size)  
40. {  
41.   cout << "wrap malloc" << endl;  
42.   return (void*)__real_malloc(size);  
43. }  
44.   
45. /* wrap malloc */  
46. void __wrap_free(void *p)  
47. {  
48.   cout << "wrap free" << endl;  
49.   __real_free(p);  
50. }  
51. #ifdef __cplusplus   
52.    }; /* end of extern "C" */  
53. #endif   
54.   
55. #endif   /* ----- #ifndef TEST_INC  ----- */  

#ifndef TEST_INC #define TEST_INC #include<iostream> using namespace std; #ifdef __cplusplus extern "C" { #endif extern void *__real_calloc(size_t nmemb, size_t size); extern void *__real_realloc(void *ptr, size_t size); extern void *__real_malloc(size_t size); extern void __real_free(void *ptr); void *__wrap_calloc(size_t numemb, size_t size); void *__wrap_realloc(void *ptr, size_t size); void *__wrap_malloc(size_t size); void __wrap_free(void *ptr); /* wrap calloc */ void * __wrap_calloc(size_t numemb, size_t size) { cout << "wrap alloc" << endl; return __real_calloc(numemb, size); } /* wrap realloc */ void * __wrap_realloc(void *ptr, size_t size) { cout << "wrap realloc" << endl; return (void*)__real_realloc(ptr, size); } /* wrap malloc */ void * __wrap_malloc(size_t size) { cout << "wrap malloc" << endl; return (void*)__real_malloc(size); } /* wrap malloc */ void __wrap_free(void *p) { cout << "wrap free" << endl; __real_free(p); } #ifdef __cplusplus }; /* end of extern "C" */ #endif #endif /* ----- #ifndef TEST_INC ----- */

.cpp只需要加上这个头文件

view plain copy to clipboard print ?

1. #include<iostream>   
2. using namespace std;  
3.   
4. <span style="color:#ff0000;">#include "test.h"</span>  
5.   
6. int main()  
7. {  
8.   int allocTimes = 100;  
9.   while(allocTimes--) {  
10.     int *p = (int*)calloc(5, sizeof(int));  
11.     //free(p);   
12.   }  
13. }  

#include<iostream> using namespace std; <span style="color:#ff0000;">#include "test.h"</span> int main() { int allocTimes = 100; while(allocTimes--) { int *p = (int*)calloc(5, sizeof(int)); //free(p); } }

然后在Makefile加上编译选项

CFLAGS = -g -O0 -Wall -Wl,--wrap,malloc -Wl,--wrap,free -Wl,--wrap,calloc -Wl,--wrap,realloc

 

或者g++  test.cc -o test  -g  -O0 -Wall -Wl,--wrap,calloc -Wl,--wrap,free -Wl,-wrap,malloc -Wl,-wrap,realloc 去编译一个文件

执行 ./test 即可发现alloc 和 free不匹配。

此方法只是抛砖引玉。可以对包装函数里加更多的东西，比如以分配内存地址建立map，first = (long)分配地址，second初始化为0。free掉了就赋值1.最后遍历map来查找未free的。当然更好的还是要记录下分配函数的调用文件以及位置，这是很简单的实现了。.......