如何通過代碼替換默認的malloc

Question

我想要替換默認的malloc並添加一些統計信息以及泄漏檢測和malloc函數的其他行爲。我已經看到了一些其他的如gperftool和jemlloc。他們可以通過鏈接靜態庫來替換默認的malloc。他們怎麼能這樣做？我想實現我的自定義malloc函數。

Answer 1

你可以環繞原始的malloc。

static void* (*r_malloc)(size_t) = NULL; 

void initialize() { 
    r_malloc = dlsym(RTLD_NEXT, "malloc"); 
} 
void* malloc(size_t size) { 
    //Do whatever you want 
    return r_malloc(bsize); 
} 

但是，不要忘記你也必須環繞calloc和realloc大概。在libc中也有不太常用的功能來分配內存。

要包裝calloc，你需要做一個骯髒的黑客攻擊，因爲dlsym嘗試使用calloc分配內存，但並不真正需要它。

static void* __temporary_calloc(size_t x __attribute__((unused)), size_t y __attribute__((unused))) { 
    return NULL; 
} 
static void* (*r_calloc)(size_t,size_t) = NULL; 

，並在初始化函數補充一點：

r_calloc = __temporary_calloc; 
r_calloc = dlsym(RTLD_NEXT, "calloc");