上篇文章我们大概介绍了一下DEBUG_NEW的原理和实现。
上篇的TracerFileLn,我们用一个hash_map来提供了Trace功能。
这个中间可能会存在一个潜在的陷阱,在进入下章前,我们需要把这个潜在的陷阱给灭了。
如果重载C++原始的operator new,也加上Trace会如何?
void* operator new(size_t InSize)
{void* pPtr = ::malloc(InSize);
TracerFileLn::singleton().traceMalloc(pPtr, "<Null>", -1);
return pPtr;
}
会stack overflow!
为什么呢?
一旦重载了原生operator new,C++在调用::operator new的时候会优先使用用户提供的版本,结果就是:
hash_map的insert里,分配内存也用到了::operator new,记得C++ 默认allocator的实现吗?allocator::allocate的实现调用的就是::operator new!
而这个operator new被我们重载了,里面又调用了traceMalloc,于是:
new -> operator new -> traceMalloc -> insert -> operator new -> traceMalloc -> insert -> …………………………
所以,为了要解决这个问题,必须实现一个不再调用::operator new版本的allocator。
具体的实现方法就不多说了,请参考侯捷翻的那本STL书,上面写的再清楚不过了。
把operator new系调用改成malloc / free系调用即可。
同样的,如果有哪些STL的容器不想进行trace的话,只要用自己这个版本的allocator替代掉即可。
另外就是,如果自己写的类不想进行Trace呢?
也很简单,C++类,如果自己实现了operator new,则new会优先调用类自己提供的operator new,所以,让这个类从下面这个类派生、或者自己实现着几个方法即可:
class UseSystemMallocForNew
{
public:
void* operator new( size_t Size )
{
return ::malloc( Size );
}
void operator delete( void* Ptr )
{
::free( Ptr );
}
void* operator new[]( size_t Size )
{
return ::malloc( Size );
}
void operator delete[]( void* Ptr )
{
::free( Ptr );
}
};