最简单的包装 malloc 系列函数的方法

ammer

    在 libc(至少在2.3.2中) 中 malloc 等函数是定义为 weak
的,所以如果在自己的库中定义一个 global的 malloc 就可以覆盖掉 libc 中的 malloc, 同时 malloc
有个global 引出的符号: __libc_malloc ,它和 malloc 是同一个函数,只不过是 global
的,所以在自己实现的库中可以用 __libc_malloc 调用原 libc 中的 malloc.
    包装 malloc 等函数可用于开发阶段对内存泄漏的检查.
下面是一个简单的示例:
/* main.c */
#include
int testfunc(void);
int main(void)
{
  printf("Call malloc in main:\n");
  malloc(123);
  printf("Call testfunc in libtest.so:\n");
  testfunc();
  return 0;
}
/*  libmalloc.c */
#include
void *malloc(size_t size)
{
  void *rst = NULL;
  rst = __libc_malloc(size);
  printf("You Want %u bytes, and got %p.\n", size, rst);
  return rst;
}
/* libtest.c */
#include
int testfunc(void)
{
  void *p;
  printf("======= in a library call malloc =======\n");
  p = malloc(10);
  printf("===== in library call malloc got %p ====\n");
}
/* Makefile */
all : libmalloc.so libtest.so main
libmalloc.so : libmalloc.c
        gcc -shared -fPIC -o libmalloc.so libmalloc.c
libtest.so : libmalloc.so libtest.c
        gcc -shared -fPIC -o libtest.so libtest.c
main : main.c libmalloc.so libtest.c
        gcc -o main main.c -L. -ltest -lmalloc
clean :
        rm -dfr *.so *~ main
                                                              
               
               
               

本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u/15482/showart_89982.html