malloc hook初探

http://www.jianshu.com/p/0d7aa3166eec

在程序中设置钩子，用来在malloc， realloc，free的时候，对其进行检查，可以看到对应的函数调用后的地址是什么。这个有时候可以用在debug的时候使用。

几个变量

• __malloc_hook
  这是一个函数指针变量，指向的是：

    void * function(size_t size, void * caller)

其中caller是表示在栈中调用malloc的函数的时候的函数地址，%p可以打印出他的地址。

• __free_hook
  同样这个也是一个函数指针变量，指向的是：

    void function(void * ptr, void * caller)

• __realloc_hook
  这也是指向一个函数指针变量，指向的是:

    void * function(void * ptr ,size_t size, void * caller)

• __memalign_hook
  这个函数是指向的一个函数指针，不过是：
  aligned_alloc, memalign, posix_memalign and valloc

• __malloc_initialize_hook

    void (*__malloc_initialize_hook) (void) = my_init_hook;

这是一个弱类型，只是在初始化的时候使用一次。

使用原理

hook函数只是对malloc realloc以及free等函数有一个包装，即每当调用了这些函数后，利用hook函数，可以得到我们想要的结果，比如当前调用的动态分配内存函数的地址，hook变量的值等等

这样在debug的时候，可以知道内存泄漏的地点，当然使用valgrind也是可以的。

例子

    /*prototype define for us*/
    static void my_init_hook(void);
    static void *my_malloc_hook(size_t, const void *);
    static void *my_free_hook(void *,const void *);

    /*some temp variable*/
    static void *(*old_malloc_hook)(size_t, const void *);
    static void (*old_free_hook)(void *, const void *)

    /*initialize hook*/
    void (*__malloc_initialize_hook) (void) = my_init_hook;

    /*init function*/
    static void
    my_init_hook(void){
        old_malloc_hook = __malloc_hook;
        old_free_hook = __free_hook;
        __malloc_hook = my_malloc_hook;
        __free_hook = my_free_hook;
    }

    /*my alloc hook*/
    static void *
    my_malloc_hook(size_t size, const void * caller)
    {
        void *result; /*malloc's return*/
        /*restore all old hooks*/
        __malloc_hook = old_malloc_hook;
        __free_hook = old_free_hook;
        /*call recursively*/
        result = malloc(size);
        /*save underlying hooks*/
        old_malloc_hook = __malloc_hook;
        old_free_hook = __free_hook;
        /*printf might call malloc, so protect it too*/
        printf("malloc(%u) call from %p, return %p\n",(unsigned int)size, caller, result);
        /*restore our own hooks*/
        __malloc_hook = my_malloc_hook;
        __free_hook = my_free_hook;

        return result;
    }
    /*free hook is same like malloc*/
    static void
    my_free_hook(void *ptr,const void *caller)
    {
        .....
    }
    /*main*/
    int main(void)
    {
        char *p;

        p = malloc(10);
        free(p);

        return 0;
    }

执行后会输出如下的：

1:__malloc_hook =0 old_malloc_hook = 0
2:__malloc_hook =0 old_malloc_hook = 0
3:__malloc_hook =0 old_malloc_hook = 0
malloc(10) caled from 0x4007b0 returns 0x1d253010
4:__malloc_hook =400620 old_malloc_hook = 0
freed pointer 0x1d253010

输出内容是可以自定制的，也就是hook的用途所在了。更多的内容可以参阅GCC的相关文档。

作者：JesseChen
链接：http://www.jianshu.com/p/0d7aa3166eec
來源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。