内存泄漏检测组件 -- 采用宏定义

C/C++Linux服务器开发/后台架构师【零声教育】-学习视频教程-腾讯课堂

# 宏 __FILE__,__FUNC__,__LINE__

        函数预编译的时候，会把宏下面的内容全部替换掉，那么是否可以定义malloc的宏呢？例如下面代码，main中的malloc被替换成了malloc_def，而malloc_def中的malloc却没有被替换。并且可以在宏定义的上下加个开关，如果代码想要进行内存泄漏检测就打开，不想就走原来的系统调用即可。

#include <stdio.h>
#include <dlfcn.h>
#include <stdlib.h>
#include <unistd.h>
#define check_mem_leak
void *malloc_def(size_t size, const char *file, const char *func, int line) {
    void *p = malloc(size);
	
    printf("[+%s:%s:%d] --> addr:%p, size:%ld\n", file, func, line, p, size);
	return p;
}
void free_def(void *p, const char *file, const char *func, int line) {
   	free(p);
    printf("[-%s:%s:%d] --> addr:%p\n", file, func, line, p);
}
#ifdef check_mem_leak
#define malloc(size) malloc_def(size,__FILE__,__FUNCTION__ ,__LINE__)
#define free(p) free_def(p,__FILE__,__FUNCTION__ ,__LINE__)
#endif
int main() {
    void *p1 = malloc(10);
    void *p2 = malloc(20);
    void *p3 = malloc(30);
    void *p4 = malloc(40);
    free(p1);
    free(p2);
    free(p4);
}