malloc 钩子

mem.h

[cpp] view plain copy print ?

1. #ifndef __MEM_H__   
2. #define __MEM_H__   
3.   
4. #include <malloc.h>   
5. #include <stdio.h>   
6.   
7. static void *(*old_malloc_hook)(size_t, const void*);  
8. static void *(*old_realloc_hook)(void *ptr, size_t size, const void *caller);  
9. static void (*old_free_hook)(void *, const void*);  
10.   
11. static inline void *mem_malloc_hook(size_t size, const void *caller)  
12. {  
13.     void *result;  
14.     __malloc_hook = old_malloc_hook;  
15.     result = malloc(size);  
16.     old_malloc_hook = __malloc_hook;  
17.     __malloc_hook = mem_malloc_hook;  
18.     printf("[malloc] size : [%u] at : [%p] caller : %p\n", (unsigned int)size, result, caller);  
19.     return result;  
20. }  
21.   
22. static inline void *mem_realloc_hook(void *ptr, size_t size, const void *caller)  
23. {  
24.     void *result;  
25.     __realloc_hook = old_realloc_hook;  
26.     result = realloc(ptr, size);  
27.     old_realloc_hook = __realloc_hook;  
28.     printf("[realloc] size : [%u] at : [%p] caller : %p\n", (unsigned int)size, result, caller);  
29.     __realloc_hook = mem_realloc_hook;  
30.     return result;  
31. }  
32.    
33. static inline void mem_free_hook(void *ptr, const void *caller)  
34. {  
35.     __free_hook   = old_free_hook;  
36.     free(ptr);  
37.     old_free_hook = __free_hook;  
38.     printf("[free] at : [%p] caller : %p\n", ptr, caller);  
39.     __free_hook   = mem_free_hook;  
40. }  
41.   
42. static inline void init_mem_hook(void)  
43. {  
44.     old_malloc_hook = __malloc_hook;  
45.     old_realloc_hook = __realloc_hook;  
46.     old_free_hook   = __free_hook;  
47.     __realloc_hook = mem_realloc_hook;  
48.     __malloc_hook = mem_malloc_hook;  
49.     __free_hook   = mem_free_hook;  
50. }  
51.   
52. #endif  

#ifndef __MEM_H__
#define __MEM_H__

#include <malloc.h>
#include <stdio.h>

static void *(*old_malloc_hook)(size_t, const void*);
static void *(*old_realloc_hook)(void *ptr, size_t size, const void *caller);
static void (*old_free_hook)(void *, const void*);

static inline void *mem_malloc_hook(size_t size, const void *caller)
{
    void *result;
    __malloc_hook = old_malloc_hook;
    result = malloc(size);
    old_malloc_hook = __malloc_hook;
    __malloc_hook = mem_malloc_hook;
    printf("[malloc] size : [%u] at : [%p] caller : %p\n", (unsigned int)size, result, caller);
    return result;
}

static inline void *mem_realloc_hook(void *ptr, size_t size, const void *caller)
{
    void *result;
    __realloc_hook = old_realloc_hook;
    result = realloc(ptr, size);
    old_realloc_hook = __realloc_hook;
    printf("[realloc] size : [%u] at : [%p] caller : %p\n", (unsigned int)size, result, caller);
    __realloc_hook = mem_realloc_hook;
    return result;
}
 
static inline void mem_free_hook(void *ptr, const void *caller)
{
    __free_hook   = old_free_hook;
    free(ptr);
    old_free_hook = __free_hook;
    printf("[free] at : [%p] caller : %p\n", ptr, caller);
    __free_hook   = mem_free_hook;
}

static inline void init_mem_hook(void)
{
    old_malloc_hook = __malloc_hook;
    old_realloc_hook = __realloc_hook;
    old_free_hook   = __free_hook;
    __realloc_hook = mem_realloc_hook;
    __malloc_hook = mem_malloc_hook;
    __free_hook   = mem_free_hook;
}

#endif

main.c

[cpp] view plain copy print ?

1. #include "mem.h"   
2.   
3. void (*__malloc_initialize_hook) (void) = init_mem_hook;  
4.   
5. int main(int argc, char **argv)  
6. {  
7.     char *ptr = malloc(1024);  
8.     ptr = realloc(ptr, 2048);  
9.     free(ptr);  
10.     return 0;  
11. }