微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决使用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 举报,一经查实,本站将立刻删除。