内存泄露检测

    内存泄露是许多大规模程序中亟待解决的问题，下面是一点资料整理，帮我记忆吧。有VC自带的工具，也有免费的比较成熟的工具。控制台程序编译即可看到效果。

// = MemoryLeakDetect.cpp

/// 去掉注释观察VS2005自带内存检测工具效果

//#define DEFAULT_CRGDBG_MEM_CHECK
#ifndef DEFAULT_CRGDBG_MEM_CHECK
    // = vld内存泄露检测
    /// http://www.codeproject.com/tools/visualleakdetector.asp
    /// Visual Leak Detector - Enhanced Memory Leak Detection for Visual C++
    #include <vld.h>
#else
    // = VS2005自带内存检测设施
    /// http://www.vckbase.com/document/viewdoc/?id=1349
    /// VC++ 6.0 中如何使用 CRT 调试功能来检测内存泄漏
    #define _CRTDBG_MAP_ALLOC
    #include <stdlib.h>
    #include <crtdbg.h>
    /// 确定准确的内存泄露位置
    #include "debug_new.h"
#endif

// = UNICODE or ANSI
#include <tchar.h>
#include <strsafe.h>

void MemoryLeakDetectDemo()
{   
    const _TCHAR str[] = _T("Hello the world!  Hello the children! ");
    _TCHAR *p = new _TCHAR[128];    
    size_t size = 0;

    StringCchLength(str, 1024, &size);
    StringCchCopy(p, size, str);
}

int _tmain(int argc, _TCHAR* argv[])
{
#ifdef DEFAULT_CRGDBG_MEM_CHECK

    ///  程序无论从何种路径退出都会自动调用_CrtDumpMemoryLeaks
    ///_CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
#endif

    MemoryLeakDetectDemo();

#ifdef DEFAULT_CRGDBG_MEM_CHECK
    _CrtDumpMemoryLeaks();
#endif

    return 0;
}

 

// = debug_new.h

#ifndef _DEBUG_NEW_H_
#define _DEBUG_NEW_H_

#ifdef _DEBUG

#undef new
extern void _RegDebugNew( void );
extern void* __cdecl operator new( size_t, const char*, int );
extern void __cdecl operator delete( void*, const char*, int);
#define new new(__FILE__, __LINE__)

#define REG_DEBUG_NEW _RegDebugNew();

#else

#define REG_DEBUG_NEW

#endif // _DEBUG

#endif // _DEBUG_NEW_H_

 

/************************************************************************/
/* comment:  此文件与debug_new.h配合使用,用于在调试期发现内存泄漏       */
/*           仅在VC++编译器中适用(包括Intel C++,因为它使用了相同的库)   */
/* 作者:     周星星 http://blog.vckbase.com/bruceteen/                  */
/* 版权申明: 无,可任意 使用,修改 和 发布                                */
/************************************************************************/

//#include "debug_new.h"

#ifdef _DEBUG

#include <windows.h>
#include <crtdbg.h>

class _CriSec
{
    CRITICAL_SECTION criSection;
public:
    _CriSec()    { InitializeCriticalSection( &criSection ); }
    ~_CriSec()   { DeleteCriticalSection( &criSection );     }
    void Enter() { EnterCriticalSection( &criSection );      }
    void Leave() { LeaveCriticalSection( &criSection );      }
} _cs;

void _RegDebugNew( void )
{
    _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG | _CRTDBG_LEAK_CHECK_DF );
}
void* __cdecl operator new( size_t nSize, const char* lpszFileName, int nLine )
{
    // comment 1: MFC中提供的debug new虽然加了锁,但我在实际测试的时候发现多线程并发
    //            调用的时候还是抛出了系统错误,所以我在这里加了一个线程互斥量.
    // comment 2: debug new和debug delete之间需不需要互斥我并不知道,保险起见,我同样
    //            加了线程互斥量.
    // comment 3: 按照C++标准规定,在operator new失败后应当调用set_new_handler设置的
    //            函数,但是MSDN中却说"头文件new中的set_new_handler是stub的,而应该使
    //            用头文件new.h中的_set_new_handler",这简直是滑天下之大稽.
    //            以下是VC++6.0中的set_new_handler定义:
    //                new_handler __cdecl set_new_handler( new_handler new_p )
    //                {
    //                    assert( new_p == 0 ); // cannot use stub to register a new handler
    //                    _set_new_handler( 0 );
    //                    return 0;
    //                }
    //            所以我也无计可施,只能舍弃set_new_handler的作用.

    _cs.Enter();
    void* p = _malloc_dbg( nSize, _NORMAL_BLOCK, lpszFileName, nLine );
    _cs.Leave();
    return p;
}
void __cdecl operator delete( void* p, const char* /*lpszFileName*/, int /*nLine*/ )
{
    _cs.Enter();
    _free_dbg( p, _CLIENT_BLOCK );
    _cs.Leave();
}

#endif