本文主要向大家介绍了C++语言堆内存检查，通过具体的内容向大家展示，希望对大家学习C++语言有所帮助。

#include <cstdio>#include <map>

#define MEM_NEW(T,n)   mem_alloc<T>(n, __TIME__, __FILE__, __LINE__);#define MEM_DELETE(p)   if(p)\

           {\

           mem_free(p);\

           };#define MEM_CHECK()   mem_check();

struct AllocInfo

{

const char* time;

const char* file;

int         line;

};

std::map<void*,AllocInfo> g_UserMem;

/*

*使用模板函数为了new的时候明确类型从而调用构造函数 

*明确返回此类的对象指针不必类型转换

*/template<typename T>T* mem_alloc(int _size,

const char*  _time,

const char*  _file,

const int    _line){

void* p = new T[_size];

if(p)

{

AllocInfo _Info = {_time,_file,_line};

g_UserMem.insert(make_pair(p,_Info));

return static_cast<T*>(p);

}

else

{  

return NULL;

}

}

/*

*使用模板函数为了delete时明确类型从而调用析构函数 

*找到key对应的内存地址返回迭代器清除容器成员

*/template<typename T>int mem_free(T* _point){

map<void*,AllocInfo>::iterator it_map = g_UserMem.find(_point);

if(it_map != g_UserMem.end())//未找到返回尾部元素后一个地址

{

delete[] _point;

g_UserMem.erase(it_map);

return 0;

}

else

{

return -1;

}

}

void mem_check(){

printf("----------[Memory Check]----------\n");

if(!g_UserMem.size())

{

printf("Memory have all been released\n");

return;

}

printf("Exist memory hasn't be freed:\n");

std::map<void*,AllocInfo>::iterator it_map = g_UserMem.begin();

for(;it_map!=g_UserMem.end();++it_map)

{

if(it_map->first)

{

printf("Memory Allocation Time:[%s]\n\t%s - Line:%d\n",\

it_map->second.time, it_map->second.file, it_map->second.line);

}

}

}

由于C++的模版不支持头文件和CPP文件分离 所以以上函数定义在头文件中

下面是测试代码

将平时的new和delete操作替换成宏函数即可，在程序退出前检查

#include <iostream>#include "MemCheck.h"using namespace std;

class A

{public:

A()

{

cout<<"A"<<endl;

}

~A()

{

cout<<"~A"<<endl;

}

};

int main(){

A *p = MEM_NEW(A,1);

MEM_DELETE(p);

p = NULL;

int* pa=new int;

MEM_DELETE(pa);//不是用自定义的方式申请的内存释放无效

delete pa;

pa = NULL;

MEM_CHECK();

getchar();

return 0;

}

以上就介绍了C/C+的相关知识，希望对C/C+有兴趣的朋友有所帮助。了解更多内容，请关注职坐标编程语言C/C+频道！