Sử dụng Malloc Hooks

Question

Tôi đang cố gắng sử dụng một móc malloc để tạo ra một hàm tùy chỉnh my_malloc(). Trong chương trình chính của tôi khi tôi gọi malloc() Tôi muốn nó gọi my_malloc() có thể một người nào đó xin vui lòng cho tôi một ví dụ về cách để làm điều này trong C

Answer 1

#undef malloc 
#define malloc my_malloc 

Chỉ cần đặt rằng ở phía trên cùng của bất kỳ tập tin bạn cần phải làm điều đó.

Answer 2

Bạn có nhìn ở đây không? http://www.gnu.org/software/libtool/manual/libc/Hooks-for-Malloc.html

Answer 3

Nếu chức năng của bạn gọi sbrk trực tiếp, bạn có thể chỉ cần gọi nó là malloc và đảm bảo nó được liên kết trước malloc của thư viện. Điều này hoạt động trên tất cả các hệ điều hành kiểu Unix. Đối với các cửa sổ, hãy xem Is there a way to redefine malloc at link time on Windows?

Nếu hàm của bạn là một trình bao bọc xung quanh thư viện của thư viện, đề xuất #define của Alex sẽ hoạt động.

Answer 4

Theo http://www.gnu.org/software/libtool/manual/libc/Hooks-for-Malloc.html, dưới đây là cách thực hiện việc này với thư viện GCC.

/* Prototypes for __malloc_hook, __free_hook */ 
#include <malloc.h> 

/* Prototypes for our hooks. */ 
static void my_init_hook (void); 
static void *my_malloc_hook (size_t, const void *); 
static void my_free_hook (void*, const void *); 

/* Override initializing hook from the C library. */ 
void (*__malloc_initialize_hook) (void) = my_init_hook; 

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; 
} 

static void * 
my_malloc_hook (size_t size, const void *caller) 
{ 
    void *result; 
    /* 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) returns %p\n", (unsigned int) size, result); 
    /* Restore our own hooks */ 
    __malloc_hook = my_malloc_hook; 
    __free_hook = my_free_hook; 
    return result; 
} 

static void 
my_free_hook (void *ptr, const void *caller) 
{ 
    /* Restore all old hooks */ 
    __malloc_hook = old_malloc_hook; 
    __free_hook = old_free_hook; 
    /* Call recursively */ 
    free (ptr); 
    /* Save underlying hooks */ 
    old_malloc_hook = __malloc_hook; 
    old_free_hook = __free_hook; 
    /* printf might call free, so protect it too. */ 
    printf ("freed pointer %p\n", ptr); 
    /* Restore our own hooks */ 
    __malloc_hook = my_malloc_hook; 
    __free_hook = my_free_hook; 
} 

main() 
{ 
    ... 
} 

Answer 5

Chỉ cần lưu ý rằng giải pháp my_malloc_hook() không thực sự hoạt động trong chương trình bị cắt xén - xem http://www.phpman.info/index.php/man/malloc_hook/3.