Win32 下内存泄漏检测的一个方法

要使用纯Win32下的C++写程序，碰到第二个问题(第一个问题先不说了：）)，如何得知没有内存泄漏呢？

用MFC的话，很简单，有自带的。那Win32的程序呢？



Google了很多，找不到比较爽的



突然间的Idea：MFC不是开源的，何不把它那一套东西挖出来呢？



呵呵，效果还是有的，请大家往下看。



先写个很简单的程序：

#include "stdafx.h"

int _tmain(int argc, _TCHAR* argv[])
{
	int* p = new int[10];

	return 0;
}

很显然，结束程序后，不会有任何提示内存问题。



再看下面的：

#include "stdafx.h"
#include <crtdbg.h>

int _tmain(int argc, _TCHAR* argv[])
{
	int* p = new int[10];

	_CrtSetDbgFlag( _CRTDBG_LEAK_CHECK_DF );
	return 0;
}

程序结束后，提示：

Detected memory leaks!
Dumping objects ->
{117} normal block at 0x00397988, 40 bytes long.
Data: < > CD CD CD CD CD CD CD CD CD CD CD CD CD CD CD CD
Object dump complete.





大家注意到了 _CrtSetDbgFlag(...) 这个函数了，对的，它的说明如下：

Retrieves or modifies the state of the _crtDbgFlag flag to control the allocation behavior of the debug heap manager (debug version only).



_CRTDBG_LEAK_CHECK_DF 说明如下：

Perform automatic leak checking at program exit through a call to _CrtDumpMemoryLeaks and generate an error report if the application failed to free all the memory it allocated.

：） 具体我就不解释了(因为已经很清楚了啊！)，这个函数是在跟踪MFC的代码后发现的。有兴趣的朋友也可以自己试下





不过，还是不太完美！！！

MFC中的提示可以显示内存泄漏是哪一行等更具体的信息的。现在我们的实现还没有！



:)

原因是：我们提供把具体的内存分配信息给 heap Manager





请看下面的解决方法：

重写new,delete操作符，顺便把分配时的信息提供给heap manager



在 stdafx.h 中加入下面行

#define DEBUG_NEW new(__FILE__, __LINE__)

void * __cdecl operator new[](size_t);
void* __cdecl operator new[](size_t nSize, LPCSTR lpszFileName, int nLine);
void __cdecl operator delete[](void* p, LPCSTR lpszFileName, int nLine);
void __cdecl operator delete[](void *);

void * __cdecl operator new(size_t);
void* __cdecl operator new(size_t nSize, LPCSTR lpszFileName, int nLine);
void __cdecl operator delete(void* p, LPCSTR lpszFileName, int nLine);
void __cdecl operator delete(void *);

在stdafx.cpp 中加入：

#ifdef _DEBUG       // most of this file is for debugging

/* Memory block identification */
#define _FREE_BLOCK      0
#define _NORMAL_BLOCK    1
#define _CRT_BLOCK       2
#define _IGNORE_BLOCK    3
#define _CLIENT_BLOCK    4
#define _MAX_BLOCKS      5

void* __cdecl operator new(size_t nSize, int nType, LPCSTR lpszFileName, int nLine);
void* __cdecl operator new[](size_t nSize, int nType, LPCSTR lpszFileName, int nLine);

#ifndef _AFX_NO_DEBUG_CRT

void* __cdecl operator new(size_t nSize, LPCSTR lpszFileName, int nLine)
{
	return ::operator new(nSize, _NORMAL_BLOCK, lpszFileName, nLine);
}

void* __cdecl operator new[](size_t nSize, LPCSTR lpszFileName, int nLine)
{
	return ::operator new[](nSize, _NORMAL_BLOCK, lpszFileName, nLine);
}

void __cdecl operator delete(void* pData, LPCSTR /* lpszFileName */,
							int /* nLine */)
{
	::operator delete(pData);
}

void __cdecl operator delete[](void* pData, LPCSTR /* lpszFileName */,
	int /* nLine */)
{
	::operator delete(pData);
}

void* __cdecl operator new[](size_t nSize, int nType, LPCSTR lpszFileName, int nLine)
{
	return ::operator new(nSize, nType, lpszFileName, nLine);
}

#endif // _AFX_NO_DEBUG_CRT
#endif // _DEBUG

最后，我们的文件如下：

#include "stdafx.h"
#include <crtdbg.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

int _tmain(int argc, _TCHAR* argv[])
{
	int* p = new int[10];

	_CrtSetDbgFlag( _CRTDBG_LEAK_CHECK_DF );
	return 0;
}

运行后，结果如下：

Detected memory leaks!
Dumping objects ->
d:/anakin/temp/testmemleak/testmemleak/testmemleak.cpp(14) : {117} normal block at 0x00397988, 40 bytes long.
Data: < > CD CD CD CD CD CD CD CD CD CD CD CD CD CD CD CD
Object dump complete.





:) enjoy!