使用DLL注入记录内存分配功能

问题描述

我有一个简单的程序,我们称它为受害者

#include <stdio.h>
#include <stdlib.h>

int
main(void)
{
    int *a = malloc(200);
    printf("%p\n",a);
}

我想向该程序中注入一个共享对象,该对象将记录对malloc函数的所有调用。因此它将像内存记录器一样工作:

#define _GNU_SOURCE

#include <dlfcn.h>
#include <stdio.h>
#include <stdlib.h>

void *
malloc(size_t n)
{
    static void * (*real_malloc)(size_t) = NULL;

    if (real_malloc == NULL) {
        real_malloc = dlsym(RTLD_NEXT,"malloc");

        if (real_malloc == NULL) {
#if 0
            printf("MEMORY FILTER: can't find the real malloc() function: %s\n",dlerror());
#endif
            exit(EXIT_FAILURE);
        }
    }

    void *result = real_malloc(n);

#if 0
    if (result != NULL)
        printf("MEMORY FILTER: allocated %zu bytes at address %p.\n",n,result);
    else
        printf("MEMORY FILTER: Failed to allocate %zu bytes.\n",n);
#endif

    return result;
}

然后我按如下方式运行受害者:

$ LD_LIBRARY_PATH=. LD_PRELOAD=libmemlog.so ./victim
0x55e0747792a0

它工作正常,但是当我取消注释#if 0 ... #endif并在调用printf函数时阻塞时,就会出现分段错误。之所以发生这种情况,是因为printf本身内部调用malloc,而我们得到了无穷无尽的递归,最终导致崩溃。

在这种情况下如何获取日志记录?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)