实用小代码之内存泄漏检测(二)

实用小代码之内存泄漏检测(二)

继上次 实用小代码之内存泄漏检测 一文。这次继续介绍另一个我平时用于动态监测内存泄漏的代码。

我这里所说的 动态监测内存泄漏 是指在程序执行时，以替换 malloc 和 free 函数来达到检测内存泄漏的目的，因为内存泄漏的主要来源就是 malloc 和 free 这组函数。相比那些分析源代码的检测工具，动态监测能够很好的检测循环中分配的内存是否被泄露。下面我不想多说，一切以代码来说话。

这里的代码只是给各位一个抛砖引玉的借鉴。其中还有很多不完善的地方。比如没有多线程锁之类的问题。

/**//*
PROGRAMER:WayneHuang
DATE:2008.04.18
LICENCE:GPLv2.0
*/

#defineMMALLOC(s)mm_malloc(s,__FILE__,__LINE__)
#defineMFREE(p)mm_free(p,__FILE__,__LINE__)
#defineMOUTWE"[%s:%d]{%c}0x%x "

typedefstructmmbsmmbs;
structmmbs...{
mmbs*next;
char*file;
intline;
void*mm;
};

staticmmbsmhandle=...{NULL,0,0,0};
mmbs*handle=&mhandle;

void*mm_malloc(unsignedints,char*file,intline);
voidmm_free(void*p,char*file,intline);
voidmm_exit(intrel);

void*mm_malloc(unsignedints,char*file,intline)...{
mmbs*p=malloc(sizeof(mmbs)+s);
mmbs*i=NULL;

if(p)...{
p->next=NULL;
p->file=file;p->line=line;
p->mm=&p[1];

for(i=handle;i->next;i=i->next);

i->next=p;
}
return(p->mm);
}

voidmm_exit(intrel)...{
mmbs*t=NULL;

for(t=handle->next;t;t=t->next)...{
printf(MOUTWE,t->file,t->line,'E',(int)t->mm);
}
if(rel)...{
t=handle->next;
while(t)...{
handle->next=t->next;
free(t);
t=handle->next;
}
}

return;
}

voidmm_free(void*p,char*file,intline)...{
mmbs*now=handle,*pre=handle;
intnf=1;
if(p)...{
while(now)...{
if(now->mm==p)...{
pre->next=now->next;
free(now);
nf=0;
break;
}else...{
pre=now;
now=now->next;
}
}
if(nf)...{
printf(MOUTWE,file,line,'W',(int)p);
}
}
return;
}

用 MMALLOC(size); 替代你原先代码中的 malloc(size); MFREE(p); 替代你原来的 free(p); 然后在程序退出以前 执行函数 mm_exit(1); 就可以查看报告，并将泄漏的内存一并释放。如果需要在代码执行过程中查看内存的分配情况，可以执行 mm_exit(0); 来查看。

两个函数的不同之处在于，第一个会将未释放的内存释放掉。而第二个只是显示通过函数分配的内存情况。

对于程序执行过程中的输出 {W} 表示释放了一个没有被申请的内存。而 {E} 表示有内存泄漏的情况。在报告中会指明代码的具体位置等相关信息。


文外音:
原本在第一篇文章发表后，我曾记得有个人在我的文章留言里用C++写了一长串的代码，大概是我代码的两倍吧，好像还没贴完。来实现类似于我今天写的这点代码实现的功能，我粗略得看了下。其中用了符号的重载之类的。的确，我没有办法在不改变人们编程习惯的情况下用C语言实现更好的动态监测泄漏的代码。

我个人的观点是，能用10行代码解决的事情，绝对不要用100行代码来解决。

北斗星君(黄庠魁)