c - 覆盖 malloc 以记录分配大小

Question

我正在尝试使用一点代码来为我的项目在 tcpdump 中记录 malloc，我使用的代码如下:

#include <stdlib.h>


unsigned int memCount = 0;

void *my_malloc(size_t size) {
void *p = malloc(size);
memCount = memCount + size;
printf("Memory Allocated :%u \n", size
return p;
}

#define malloc(size) my_malloc(size)

在网上查看了许多类似的问题后，这似乎应该有效，但是我的自定义 malloc 似乎没有被调用。任何帮助将不胜感激，谢谢!

Answer 1

您必须在每个源文件中再次#define malloc 宏 - 因此可以通过包含定义该宏的头文件来实现。