我正在尝试编写一个简单的共享库,用于记录对 stderr 的 malloc 调用(如果您愿意的话,是一种“mtrace”)。
但是,这是行不通的。 这是我的做法:
/* mtrace.c */
#include <dlfcn.h>
#include <stdio.h>
static void* (*real_malloc)(size_t);
void *malloc(size_t size)
{
void *p = NULL;
fprintf(stderr, "malloc(%d) = ", size);
p = real_malloc(size);
fprintf(stderr, "%p\n", p);
return p;
}
static void __mtrace_init(void) __attribute__((constructor));
static void __mtrace_init(void)
{
void *handle = NULL;
handle = dlopen("libc.so.6", RTLD_LAZY);
if (NULL == handle) {
fprintf(stderr, "Error in `dlopen`: %s\n", dlerror());
return;
}
real_malloc = dlsym(handle, "malloc");
if (NULL == real_malloc) {
fprintf(stderr, "Error in `dlsym`: %s\n", dlerror());
return;
}
}
我编译它:
gcc -shared -fPIC -o mtrace.so mtrace.c
然后当我尝试执行 ls
时:
$ LD_PRELOAD=./mtrace.so ls
malloc(352) = Segmentation fault
现在,我怀疑 dlopen 需要 malloc,并且当我在共享库中重新定义它时,它使用该版本和仍未分配的 real_malloc
。
问题是……我该如何让它发挥作用?
附言很抱歉标签太少,我找不到合适的标签,而且我仍然没有足够的声誉来创建新标签。
最佳答案
我总是这样做:
#define _GNU_SOURCE
#include <stdio.h>
#include <dlfcn.h>
static void* (*real_malloc)(size_t)=NULL;
static void mtrace_init(void)
{
real_malloc = dlsym(RTLD_NEXT, "malloc");
if (NULL == real_malloc) {
fprintf(stderr, "Error in `dlsym`: %s\n", dlerror());
}
}
void *malloc(size_t size)
{
if(real_malloc==NULL) {
mtrace_init();
}
void *p = NULL;
fprintf(stderr, "malloc(%d) = ", size);
p = real_malloc(size);
fprintf(stderr, "%p\n", p);
return p;
}
不要使用构造函数,只需在第一次调用 malloc
时进行初始化。使用 RTLD_NEXT
来避免 dlopen
。你也可以试试 malloc hooks .请注意,所有这些都是 GNU 扩展,可能无法在其他地方使用。
关于c - 使用 LD_PRELOAD 机制覆盖 'malloc',我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57897343/