用单链表实现的内存管理

http://blog.csdn.net/csynyk/article/details/2861979

 

在C语言论坛中看了dgarc发表的一个贴子后，按要求写了一段代码，可以实现内存分配的管理，避免内存泄漏，欢迎各位测试，如遇bug，敬请指出，我将进行有针对性的改进，然后重新贴出来。

要求如下：

    题目：监控应用程序的内存申请和释放 
     描述：
         不考虑多线程，使用动态单链表的方式进行操作，在应用程序申请内存时，需要把该内存节点信息插入链表进行记录，应用程序释放内存，把相应内存节点的信息从链表中删除，在应用程序退出时，如果链表仍然存在节点，说明存在内存泄露，应用程序不存在内存泄露，则链表为空。
MLCmalloc（）  封装了malloc ，同时将信息插入链表操作
MLCfree（）    封装了free ，同时将信息从链表中删除

程序代码：
    下列代码在满足题目要求的同时，增加了对泄漏空间的检测和释放功能。

#include <stdio.h>#include <string.h>#include <windows.h>#define   _MC    1     // MLCmalloc()的标志#define   _FE    0     // MLCfree()的标志#define  _Check  -1    //检测内存的标志typedef struct _Memory         //单向链表结构{unsigned int address;struct _Memory *next;}Mem, *LinkMem;int MemoryContrl(size_t addr,char flag)  {static Mem head;        //保存链表头     LinkMem tmp=head.next, node=&head ; if( _MC==flag )         //增加链表节点{while(tmp)              {node=tmp;tmp=tmp->next;}node->next=(LinkMem)malloc(sizeof(Mem));if(node->next==NULL)return 1;   //分配失败!node=node->next;node->address=addr;node->next=NULL;}else if( _FE==flag)     //删除节点{while(tmp && (tmp->address!=addr)){node=tmp;tmp=tmp->next;}if(tmp==NULL)return -1;  //free地址出错!else{node->next=tmp->next;free(tmp);}}else                           //有泄漏返回地址，无则返回nullreturn (int)( head.next? head.next->address :0 );  return 0;}///////////////内存分配//////////////////////void *MLCmalloc(size_t size)   {void *pMalc=malloc(size);if(pMalc==NULL)return NULL;else if(MemoryContrl((size_t)pMalc, _MC)==1){free(pMalc);   //动态链建立失败时释放此次分配的空间return NULL;}       elsereturn pMalc;}///////////////内存释放//////////////////////void MLCfree(void *ptr)        {if(MemoryContrl((size_t)ptr, _FE)==-1){printf("%s/n","被释放的内存地址错误!");return ;}elsefree(ptr);}//////////检查泄漏//////////size_t CheckLeak()   {return (size_t)(MemoryContrl((size_t)0, _Check));}

 

#include "mem.h"int main(){char **p=NULL;char *str="1234567890098765432";char *tmp,i;p=(char **)MLCmalloc(10*sizeof(char*));printf("分配p=%#x/n",p);for( i=0;i<10; i++){p[i]=(char *)MLCmalloc(20*sizeof(char));sscanf(str,"%s",p[i]);}for(i=0; i<10; i++)printf("&p[%d]=%#x  *p[%d]=%s/n",i,p[i],i,p[i]);printf("释放p=%#x/n",p);   //不安正常顺序释放MLCfree(p);while((tmp=(char*)CheckLeak())!=0)  //检测是否有泄漏空间{MLCfree(tmp);               //释放检测到的泄漏空间printf("tmp=%#x/n",tmp);}return 0;}