malloc_hook(3) - wariua/manpages-ko GitHub Wiki

#include <malloc.h>

void *(*__malloc_hook)(size_t size, const void *caller);

void *(*__realloc_hook)(void *ptr, size_t size, const void *caller);

void *(*__memalign_hook)(size_t alignment, size_t size,
                         const void *caller);

void (*__free_hook)(void *ptr, const void *caller);

void (*__malloc_initialize_hook)(void);

void (*__after_morecore_hook)(void);

void (*__malloc_initialize_hook)(void) = my_init_hook;

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

/* 훅 원형 */
static void my_init_hook(void);
static void *my_malloc_hook(size_t, const void *);

/* 원래 훅을 저장할 변수 */
static void *(*old_malloc_hook)(size_t, const void *);

/* C 라이브러리의 초기화 훅 오버라이드 */
void (*__malloc_initialize_hook) (void) = my_init_hook;

static void
my_init_hook(void)
{
    old_malloc_hook = __malloc_hook;
    __malloc_hook = my_malloc_hook;
}

static void *
my_malloc_hook(size_t size, const void *caller)
{
    void *result;

    /* 이전 훅 복원 */
    __malloc_hook = old_malloc_hook;

    /* 재귀적으로 호출 */
    result = malloc(size);

    /* 하위 훅 저장 */
    old_malloc_hook = __malloc_hook;

    /* printf()에서도 malloc()을 호출할 수도 있으므로 보호 필요 */
    printf("malloc(%u) called from %p returns %p\n",
            (unsigned int) size, caller, result);

    /* 우리 훅 복원 */
    __malloc_hook = my_malloc_hook;

    return result;
}