版權(quán)申明
本文可以被自由轉(zhuǎn)載，但是必須遵循如下版權(quán)約定：
1、保留本約定，并保留在文章的開頭部分。
2、不能任意修改文章內(nèi)容，或者刪節(jié)，增加。如果認(rèn)為本文內(nèi)容有不當(dāng)之處需要修改，請與作者聯(lián)系。
3、不能摘抄本文的內(nèi)容，必須全文發(fā)表或者引用。
4、必須保留作者署名、注明文章出處。（本文授權(quán)給www.linuxaid.com.cn）
5、如不遵守本規(guī)定，則無權(quán)轉(zhuǎn)載本文。

作者
ariesram
電子郵件地址
ariesram@linuxaid.com.cn, 或 ariesram@may10.ca
本文及本人所有文章均收集在bambi.may10.ca/~ariesram/articles/中。
本文授權(quán)給www.linuxaid.com.cn。

正文：

??? 我曾經(jīng)參與過一個比較大的項目，在這個項目里面，我們沒有一個完全確定的設(shè)計文檔，所以程序的實現(xiàn)常常變動。雖然我們有一個比較靈活的框架，但是從程序的角度來講，它使我們的程序非常的混亂。直到發(fā)布的日期臨近，我們還沒有一個穩(wěn)定的可以用來做alpha測試的版本。所以我們必須盡快的刪除掉無用的代碼，讓這個版本足夠的穩(wěn)定。但是，在這個沒有足夠規(guī)范的軟件公司，我們沒有時間也沒有足夠的精力來做邊界測試之類的工作。所以我們只能采用變通的辦法。在軟件中最大的問題就是內(nèi)存泄漏。因為往往出現(xiàn)這樣的情況，我們在一段代碼中分配了內(nèi)存，但是卻沒有釋放它。這造成了很大的問題。我們需要一個簡單的解決方案，能夠簡單的編譯進這個項目，在運行的時候，它能夠產(chǎn)生一個沒有被釋放的內(nèi)存的列表，用這個列表，我們能夠改正程序的錯誤。這就是我們稱之為內(nèi)存跟蹤的方法。首先，我們需要一種代碼，能夠被加入到源代碼中去，而且這種代碼能夠被重用。代碼重用是一種很重要的特性，能夠節(jié)省大量的時間和金錢以及程序員的勞動。另外，我們的這種代碼必須簡單，因為我們當(dāng)時已經(jīng)沒有那么多的時間和精力去完全重看一遍所有的代碼來重新編寫以及改正錯誤從而使內(nèi)存跟蹤能夠起作用。

??? 好在，我們總能夠找到解決的辦法。首先，我們檢查了代碼，發(fā)現(xiàn)所有的代碼都是用new來分配內(nèi)存，用delete來釋放內(nèi)存。那么，我們能夠用一個全程替換，來替換掉所有的new和delete操作符嗎？不能。因為代碼的規(guī)模太大了，那樣做除了浪費時間沒有別的任何好處。好在我們的源代碼是用C++來寫成的，所以，這意味著沒有必要替換掉所有的new和delete,而只用重載這兩個操作符。對了，值用重載這兩個操作符，我們就能在分配和釋放內(nèi)存之前做點什么。這是一個絕對的好消息。我們也知道該如何去做。因為，MFC也是這么做的。我們需要做的是：跟蹤所有的內(nèi)存分配和交互引用以及內(nèi)存釋放。我們的源代碼使用Visual C++寫成，當(dāng)然這種解決方法也可以很輕松的使用在別的C++代碼里面。要做的第一件事情是重載new和delete操作符，它們將會在所有的代碼中被使用到。我們在stdafx.h中，加入：
????? #ifdef _DEBUG
????? inline void * __cdecl operator new(unsigned int size,
???????????????????????????????????????? const char *file, int line)
????? {
????? };

????? inline void __cdecl operator delete(void *p)
????? {
????? };
????? #endif
??? 這樣，我們就重載了new和delete操作符。我們用$ifdef和#endif來包住這兩個重載操作符，這樣，這兩個操作符就不會在發(fā)布版本中出現(xiàn)。看一看這段代碼，會發(fā)現(xiàn)，new操作符有三個參數(shù)，它們是，分配的內(nèi)存大小，出現(xiàn)的文件名，和行號。這對于尋找內(nèi)存泄漏是必需的和重要的。否則，就會需要很多時間去尋找它們出現(xiàn)的確切地方。加入了這段代碼，我們的調(diào)用new()的代碼仍然是指向只接受一個參數(shù)的new操作符，而不是這個接受三個參數(shù)的操作符。另外，我們也不想記錄所有的new操作符的語句去包含__FILE__和__LINE__參數(shù)。我們需要做的是自動的讓所有的接受一個參數(shù)的new操作符調(diào)用接受三個參數(shù)的new操作符。這一點可以用一點點小的技巧去做，例如下面的這一段宏定義，
????? #ifdef _DEBUG
????? #define DEBUG_NEW new(__FILE__, __LINE__)
????? #else
????? #define DEBUG_NEW new
????? #endif
????? #define new DEBUG_NEW
??? 現(xiàn)在我們所有的接受一個參數(shù)的new操作符都成為了接受三個參數(shù)的new操作符號，__FILE__和__LINE__被預(yù)編譯器自動的插入到其中了。然后，就是作實際的跟蹤了。我們需要加入一些例程到我們的重載的函數(shù)中去，讓它們能夠完成分配內(nèi)存和釋放內(nèi)存的工作。這樣來做， #ifdef _DEBUG
????? inline void * __cdecl operator new(unsigned int size,
???????????????????????????????????????? const char *file, int line)
????? {
void *ptr = (void *)malloc(size);
AddTrack((DWORD)ptr, size, file, line);
return(ptr);
????? };
????? inline void __cdecl operator delete(void *p)
????? {
RemoveTrack((DWORD)p);
free(p);
????? };
????? #endif
??? 另外，還需要用相同的方法來重載new[]和delete[]操作符。這里就省略掉它們了。
??? 最后，我們需要提供一套函數(shù)AddTrack()和RemoveTrack()。我用STL來維護存儲內(nèi)存分配記錄的連接表。
這兩個函數(shù)如下：
????? typedef struct {
DWORD address;
DWORD size;
char file[64];
DWORD line;
????? } ALLOC_INFO;

????? typedef list〈ALLOC_INFO*〉 AllocList;

????? AllocList *allocList;

????? void AddTrack(DWORD addr, DWORD asize, const char *fname, DWORD lnum)
????? {
ALLOC_INFO *info;

if(!allocList) {
allocList = new(AllocList);
}

info = new(ALLOC_INFO);
info-〉address = addr;
strncpy(info-〉file, fname, 63);
info-〉line = lnum;
info-〉size = asize;
allocList-〉insert(allocList-〉begin(), info);
????? };

????? void RemoveTrack(DWORD addr)
????? {
AllocList::iterator i;

if(!allocList)
return;
for(i = allocList-〉begin(); i != allocList-〉end(); i++)
{
if((*i)-〉address == addr)
{
allocList-〉remove((*i));
break;
}
}
????? };
??? 現(xiàn)在，在我們的程序退出之前，allocList存儲了沒有被釋放的內(nèi)存分配。為了看到它們是什么和在哪里被分配的，我們需要打印出allocList中的數(shù)據(jù)。我使用了Visual C++中的Output窗口來做這件事情。
????? void DumpUnfreed()
????? {
AllocList::iterator i;
DWORD totalSize = 0;
char buf[1024];

if(!allocList)
return;

for(i = allocList-〉begin(); i != allocList-〉end(); i++) {
sprintf(buf, "%-50s: LINE %d, ADDRESS %d %d unfreed ",
(*i)-〉file, (*i)-〉line, (*i)-〉address, (*i)-〉size);
OutputDebugString(buf);
totalSize += (*i)-〉size;
}
sprintf(buf, "----------------------------------------------------------- ");
OutputDebugString(buf);
sprintf(buf, "Total Unfreed: %d bytes ", totalSize);
OutputDebugString(buf);
????? };
??? 現(xiàn)在我們就有了一個可以重用的代碼，用來監(jiān)測跟蹤所有的內(nèi)存泄漏了。這段代碼可以用來加入到所有的項目中去。雖然它不會讓你的程序看起來更好，但是起碼它能夠幫助你檢查錯誤，讓程序更加的穩(wěn)定。