说起C/C++程序,一个老生常谈的问题便是内存分配,而这也是一个难点。在大型的软件开发项目中,常会遇到较大的常驻内存、内存泄露、内存破坏等问题,这些问题往往都是致命的。于是我们就借助于一些内存检测工具、API或者是某种技术手段,来动态地跟踪内存分配,查看内存使用细节,方法很多,各有利弊,GNU链接器就提供了一个好用的方法:
–wrap=symbol
函数名为“__wrap_symbol”,且称其为包装函数,“symbol”是一个函数名,大致执行过程是这样的:当调用“symbol”函数时,如果“symbol”函数未定义的话就会调用“__wrap_symbol”函数;“__real_symbol”也是个相关的函数,当其只声明不定义的话,我们对其的调用将调用真正的“symbol”函数;当然,我们还要添加编译链接参数:
-Wl,–wrap,symbol
下面以常用的malloc函数为例说明:
// wrap.c
#include <stdio.h>
#include <stdlib.h>
void* __real_malloc(size_t size); // 只声明不定义__real_malloc
void* __wrap_malloc(size_t size) // 定义__wrap_malloc
{
printf("__wrap_malloc called, size:%zd\n", size); // log输出
return __real_malloc(size); // 通过__real_malloc调用真正的malloc
}
下面是测试用例:
// test.c
#include <stdio.h>
#include <stdlib.h>
int main()
{
char* c = (char*)malloc(sizeof(char)); // 调用malloc
printf("c = %p\n", c);
free(c); // 调用free,防止内存泄漏
return 0;
}
下面是编译链接过程:
gcc -c wrap.c test.c
gcc -Wl,--wrap,malloc -o test wrap.o test.o // 链接参数-Wl,--wrap,malloc
结果查看:
./test
__wrap_malloc called, size:1
c = 0x60d010
从上面的结果中可以看出,测试用例中使用了malloc,首先调用的是__wrap_malloc函数,然后我们就可以在这个函数中添加我们想要的东西了。