我曾經(jīng)參與過一個(gè)比較大的項(xiàng)目，在這個(gè)項(xiàng)目里面，我們沒有一個(gè)完全確定的設(shè)計(jì)文檔，所以程序的實(shí)現(xiàn)常常變動(dòng)。雖然我們有一個(gè)比較靈活的框架，但是從程序的角度來講，它使我們的程序非常的混亂。直到發(fā)布的日期臨近，我們還沒有一個(gè)穩(wěn)定的可以用來做alpha測試的版本。所以我們必須盡快的刪除掉無用的代碼，讓這個(gè)版本足夠的穩(wěn)定。但是，在這個(gè)沒有足夠規(guī)范的軟件公司，我們沒有時(shí)間也沒有足夠的精力來做邊界測試之類的工作。所以我們只能采用變通的辦法。在軟件中大的問題是內(nèi)存泄漏。因?yàn)橥霈F(xiàn)這樣的情況，我們在一段代碼中分配了內(nèi)存，但是卻沒有釋放它。這造成了很大的問題。我們需要一個(gè)簡單的解決方案，能夠簡單的編譯進(jìn)這個(gè)項(xiàng)目，在運(yùn)行的時(shí)候，它能夠產(chǎn)生一個(gè)沒有被釋放的內(nèi)存的列表，用這個(gè)列表，我們能夠改正程序的錯(cuò)誤。這是我們稱之為內(nèi)存跟蹤的方法。首先，我們需要一種代碼，能夠被加入到源代碼中去，而且這種代碼能夠被重用。代碼重用是一種很重要的特性，能夠節(jié)省大量的時(shí)間和金錢以及程序員的勞動(dòng)。另外，我們的這種代碼必須簡單，因?yàn)槲覀儺?dāng)時(shí)已經(jīng)沒有那么多的時(shí)間和精力去完全重看一遍所有的代碼來重新編寫以及改正錯(cuò)誤從而使內(nèi)存跟蹤能夠起作用。
好在，我們總能夠找到解決的辦法。首先，我們檢查了代碼，發(fā)現(xiàn)所有的代碼都是用new來分配內(nèi)存，用delete來釋放內(nèi)存。那么，我們能夠用一個(gè)全程替換，來替換掉所有的new和delete操作符嗎？不能。因?yàn)榇a的規(guī)模太大了，那樣做除了浪費(fèi)時(shí)間沒有別的任何好處。好在我們的源代碼是用C++來寫成的，所以，這意味著沒有必要替換掉所有的new和delete，而只用重載這兩個(gè)操作符。對(duì)了，值用重載這兩個(gè)操作符，我們能在分配和釋放內(nèi)存之前做點(diǎn)什么。這是一個(gè)的好消息。我們也知道該如何去做。因?yàn)�，MFC也是這么做的。我們需要做的是：跟蹤所有的內(nèi)存分配和交互引用以及內(nèi)存釋放。我們的源代碼使用Visual C++寫成，當(dāng)然這種解決方法也可以很輕松的使用在別的C++代碼里面。要做的第一件事情是重載new和delete操作符，它們將會(huì)在所有的代碼中被使用到。我們在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來包住這兩個(gè)重載操作符，這樣，這兩個(gè)操作符不會(huì)在發(fā)布版本中出現(xiàn)�？匆豢催@段代碼，會(huì)發(fā)現(xiàn)，new操作符有三個(gè)參數(shù)，它們是，分配的內(nèi)存大小，出現(xiàn)的文件名，和行號(hào)。這對(duì)于尋找內(nèi)存泄漏是必需的和重要的。否則，會(huì)需要很多時(shí)間去尋找它們出現(xiàn)的確切地方。加入了這段代碼，我們的調(diào)用new()的代碼仍然是指向只接受一個(gè)參數(shù)的new操作符，而不是這個(gè)接受三個(gè)參數(shù)的操作符。另外，我們也不想記錄所有的new操作符的語句去包含__FILE__和__LINE__參數(shù)。我們需要做的是自動(dòng)的讓所有的接受一個(gè)參數(shù)的new操作符調(diào)用接受三個(gè)參數(shù)的new操作符。這一點(diǎn)可以用一點(diǎn)點(diǎn)小的技巧去做，例如下面的這一段宏定義，
#ifdef _DEBUG
#define DEBUG_NEW new(__FILE__， __LINE__)
#else
#define DEBUG_NEW new
#endif
#define new DEBUG_NEW
現(xiàn)在我們所有的接受一個(gè)參數(shù)的new操作符都成為了接受三個(gè)參數(shù)的new操作符號(hào)，__FILE__和__LINE__被預(yù)編譯器自動(dòng)的插入到其中了。然后，是作實(shí)際的跟蹤了。我們需要加入一些例程到我們的重載的函數(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來維護(hù)存儲(chǔ)內(nèi)存分配記錄的連接表。
這兩個(gè)函數(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存儲(chǔ)了沒有被釋放的內(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)在我們有了一個(gè)可以重用的代碼，用來監(jiān)測跟蹤所有的內(nèi)存泄漏了。這段代碼可以用來加入到所有的項(xiàng)目中去。雖然它不會(huì)讓你的程序看起來更好，但是起碼它能夠幫助你檢查錯(cuò)誤，讓程序更加的穩(wěn)定。
