時(shí)間總是能讓一切改變，至上次發(fā)blog我已經(jīng)不在福州。現(xiàn)在就職于成都一家公司，剛來(lái)一直在閱讀代碼，發(fā)現(xiàn)服務(wù)器的邏輯代碼，有挺多讓直接new申請(qǐng)內(nèi)存，對(duì)這個(gè)我總覺(jué)得有些裸奔的感覺(jué)。啊哈，有可能被同事看到，我沒(méi)有其他意思哈，只是覺(jué)得沒(méi)有管理，一不小心就內(nèi)存泄露了，寫(xiě)的時(shí)候要小心又小心。如果要從堆中獲取一塊內(nèi)存，一般還是交給智能指針去管理更妥些。
以上這大段和下面寫(xiě)的都沒(méi)有任何關(guān)系。嗯，是的，確實(shí)是沒(méi)有關(guān)系，我只是在思考已經(jīng)這樣寫(xiě)了，能不能對(duì)內(nèi)存申請(qǐng)和釋放做個(gè)監(jiān)控。其實(shí)網(wǎng)上代碼一堆，我也自己按照自己的思路實(shí)現(xiàn)了一下，原理都是重載new和delete方式了。閑言少敘，話(huà)不多說(shuō)。
同樣，我也要重載new與delete關(guān)鍵字，但是我重載的是帶3參數(shù)的，具體的申請(qǐng)交給我的CMemLeakMonitorMgr管理器吧。以后再具體說(shuō)明，先看如下代碼：
void* operator new(size_t size, const char* funName, size_t line)
{
return g_pMemLeakMonitorMgr.Alloc(funName,line,size);
}
void* operator new[](size_t size, const char* file, int line)
{
return operator new(size, file, line);
}
void* operator new(size_t size)
{
return malloc(size);
}
void* operator new[](size_t size)
{
return malloc(size);
}
void* operator new(size_t size, const std::nothrow_t&) throw()
{
return ::operator new(size);
}
void* operator new[](size_t size, const std::nothrow_t&) throw()
{
return ::operator new[](size);
}
void operator delete(void* p)
{
return g_pMemLeakMonitorMgr.Free(p);
}
然后我定義以下宏，來(lái)替換new操作,請(qǐng)看下面：
#define DEBUG
#ifdef DEBUG
#define DEBUG_NEW new(__FUNCTION__, __LINE__)
#define DEBUG_DELETE(p) delete p
#else
#define DEBUG_NEW new
#define DEBUG_DELETE(p) delete p
#endif
上面的宏應(yīng)該不用多說(shuō)吧，設(shè)計(jì)的是只有在debug情況下，才會(huì)生效。這里用DEBUG_NEW與DEBUG_DELETE宏來(lái)申請(qǐng)和釋放內(nèi)存，其實(shí)完全可以直接替換new與delete，但是我不想這么做，因?yàn)槲蚁朐谑褂谜哂迷摵陼r(shí)明白它做了一些“見(jiàn)不得人”的事情。哈，
還是那句話(huà)閑言少敘，廢話(huà)不能多。看下怎么使用吧：
void fun1()
{
DEBUG_NEW int;
}
void fun2()
{
char *p = DEBUG_NEW char;
DEBUG_DELETE(p);
}
void fun3()
{
DEBUG_NEW int;
}
int main()
{
Test2 *p = DEBUG_NEW Test2();
DEBUG_DELETE(p);
fun1();
fun2();
fun3();
fun1();
fun1();
g_pMemLeakMonitorMgr.ReportMemLeak();
return 0;
}
嗯，是的，這個(gè)很好懂，通過(guò)宏來(lái)替換之前的new和delete，然后幾個(gè)函數(shù)中有些函數(shù)是有內(nèi)存泄露的。然后通過(guò)ReportMemLeak來(lái)報(bào)告結(jié)果。可以顯示的如下：
Test2 constructor.
Test2 destructor.
Test2 constructor.
Test2 destructor.
Memory leak FunName:fun1,line:182,totalsize:12 bytes.
Memory leak FunName:fun3,line:194,totalsize:4 bytes.