gcc -Wl,--wrap,malloc 替换系统函数

发布时间 2023-09-01 17:48:30作者: _小百

转载于:https://blog.csdn.net/weixin_41041321/article/details/83414286

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
}123456789

下面是测试用例:

 

// 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;
}12345678910

下面是编译链接过程:

 

gcc -c wrap.c test.c
gcc -Wl,--wrap,malloc -o test wrap.o test.o // 链接参数-Wl,--wrap,malloc12

结果查看:
./test

 

__wrap_malloc called, size:1
c = 0x60d010