内存泄露相信对C++程序员来说都不陌生。解决内存泄露的方案多种多样,大部分方案以追踪检测为主,这种方法实现起来容易,使用方便,也比较安全。
首先我们要确定这个模块的主要功能:
- 能追踪内存的分配和释放过程。
- 要能显示内存分配的相关信息,比如内存块大小,代码所在文件所在行等。
- 在发现内存泄露时及时给出相关信息。
- 能正确处理一些异常情况,比如内存不足,对象初始化失败等等。
- 是线程安全的。[*这个还没有实现]
有了一些基本功能需求,我们需要考虑每种功能怎么去实现。首先,我们可以通过重载的方式来追踪new,delete.malloc和free,C++给我提供了这样的特性。因为本文主要针对C++,所以主要讲重载new,delete的方法,malloc和free的重载实现于此类似,最终版本的程序中也实现了malloc和free的重载。
1.重载new和delete
首先我们要了解一下new和delete是怎么工作的。C++中的操作符最终都会被转换成函数形式,例如"new int"会变成"opetaor new(sizeof(int))",而"new double[10]"会变成"operator new(sizeof(double)*10)",同样“delete p”就变成了"operator delete(p)"。另外一个需要特别注意的地方是,new对于用户定义的数据类型(即你的自定义类)会自动调用该类型的构造函数,如果构造函数没有抛出异常,则正确分配,否则会中断分配操作,将异常传递给用户。默认情况下,new可以对象构造异常进行捕获。另外一个版本的new就是不带捕获异常功能的的了,所以C++系统提供的new和delete有:
1
2
3
4
5
6
7
8
9
|
void * operator new ( size_t size) throw (std::bad_alloc);
void * operator new []( size_t size) throw (std::bad_alloc);
void * operator new ( size_t ,std::nothrow_t&) throw ();
void * operator new []( size_t ,std::nothrow_t&) throw ();
void operator delete ( void * pointer);
void operator delete []( void * pointer);
void operator delete ( void * pointer,std::nothrow_t&);
void operator delete []( void * pointer,std::nothrow_t&);<br>
|
其中,nothrow_t是一个空结构体“struct nothrow_t{}",它的一个实例就是nothrow,C++用它来区分可以捕获异常的new和不可捕获异常的new。我们不能直接修改内部函数的行为,但是我们可以重载它们。为了实现提供内存分配信息的功能,我们给重载的函数加上几个参数。得到以下函数:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
void * operator new ( size_t size);
void * operator new []( size_t size);
void * operator new ( size_t ,std::nothrow_t&) throw ();
void * operator new []( size_t ,std::nothrow_t&) throw ();
void * operator new ( size_t size, const char * file, const char * func, const int line) throw (std::bad_alloc);
void * operator new []( size_t size, const char * file, const char * func, const int line) throw (std::bad_alloc);
void * operator delete ( void * pointer);
void * operator delete []( void * pointer);
void operator delete ( void * pointer, const char * file, const char * func, const int line);
void operator delete []( void * pointer, const char * file, const char * func, const int line);
void operator delete ( void * pointer,std::nothrow_t&);
void operator delete []( void * pointer,std::nothrow_t&);
|
中间的几个函数,就是我们主要需要重载的函数,模块的大部分工作也都由着几个函数完成。这些函数参数中,file表示分配代码所在的文件名,func表示代码所在的函数名,line表示代码行号。这几个参数信息我们可以通过编译器预定义好的几个宏来获得:__FILE__,__FUNCTION__,__LINE__。也就是说可以将"new ..."替换成"new(__FILE__,__FUNCTION__,__LINE__) ...",最终成为"operator new(sizeof(...),__FILE__,__FUNCTION__,__LINE__)"的形式,也就达到了我们的目的。关于 placement delete将在下面详细解释。
2.空间分配
接下来我们要考虑内存分配信息的组织问题了。我们先来了解一下编译器是怎么组织的。在大部分编译器中,new所分配的空间都要大于实际申请的空间,大出来的部分就是编译器定义的内存块的信息,包括了内存块的大小还有一些其他信息。如下图所示:
我们把包含内存分配信息的部分叫做cookie数据。为了方便,我们把cookie数据放在分配的内存的起始位置,之后紧接有效数据区。我们还需要把返回给调用者的指针和new分配的数据区联系起来,原本想用性能比较好的STL的map数据结构来储存这些数据,但是map内部同样也使用new来分配内存,所以如果直接使用map来储存,就会陷入死循环中。所以这里我们必须自己现实一个数据结构。我们可以对返回给调用者的地址进行Hash,得到hash 表中的地址,具有相同Hash值的数据我们用一个单向链表连接起来。最终的数据结构如下图所示:
2.1.构造函数中的异常
另外一个必须要注意的一点是,new操作符会先分配空间然后调用用户自定义类型的构造函数,如果构造函数抛出异常,需要用户手动释放已分配的内存。问题在于释放这样的内存不能用一般的delete操作符,可以用一个例子来说明:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#include <stdio.h>
#include <stdexcept>
void * operator new ( size_t size, int line) {
printf ( "Allocate %u bytes on line %d\\n" , size, line);
return operator new (size);
}
class UserClass {
public :
UserClass( int n) throw (){
if (n<=0){
throw std::runtime_error( "n must be positive" );
}
}
};
int main(){
try {
UserClass* myobj= new (__LINE__) UserClass(-10);
delete myobj;
} catch ( const std::runtime_error& e) {
fprintf (stderr, "Exception: %s\n" ,e.what());
}
return 0;
}<br>
|
这里,虽然在new过后试图使用delete释放已经分配的内存,但是实际上不会释放。也许你的编译器会给出这样一条消息:
“no matching operator delete found”
为了正确处理这种情况,并给用户提供相关的信息,我们需要定义placement delete操作符。placement delete是C++98标准中才有的一个特性,所以对于某些老的编译器(大致可以认为是指那些98年以前编写的编译器)不支持这个特性。这需要在模块中添加宏定义让用户可以关闭placement delete的定义,以便模块能在较老的编译器上编译。以下就是需要定义的placement delete操作符:
1
2
3
4
|
void operator delete ( void * pointer, const char * file, const char * func, const int line);
void operator delete []( void * pointer, const char * file, const char * func, const int line);
void operator delete ( void * pointer,std::nothrow_t&);
void operator delete []( void * pointer,std::nothrow_t&);<br>
|
3.检查内存泄露
有了上面的实现,我们可以方便的手动检测内存泄露。通过一个函数来实现,它会检索整个hash表,如果表不为空则存在内存泄露。
为了达到在最后程序退出时检查内存泄露的目的,我们需要在所有对象调用析构函数后进行内存泄露检测,这是因为某些用户类型在构造函数里调用new而在析构函数里调用delete。这样做能大大的减小误报的概率。而且因为对象的析构函数的调用往往在主函数main()执行结束之后进行,所以我们也不能直接在主函数里进行内存泄露检测。这里我们利用一个全局对象来实现这种检测。首先我们定义一个类:
1
2
3
4
5
6
7
8
9
10
11
|
class MemCheck{
public :
MemCheck(){
memset (pTable,0, sizeof (mc_block_node_t*) * MC_HASHTABLESIZE);
}
~MemCheck(){
if (mc_checkmem()){
abort ();
}
}
};
|
这里的构造函数初始化Hash表。析构函数检测内存泄露。然后定义一个MemCheck的全局静态对象,这样当程序运行之前会初始化hash表,程序退出时检测内存泄露。可是问题又来了,如果一个程序中有多个全局静态对象会怎样?不幸的是,对于全局静态对象的构造顺序和析构顺序是C++标准中的一个未定义问题,也就是说,这个顺序取决于编译器的具体实现。考虑,绝大多数平台使用VC和GCC编译器,我们可以针对这两种编译器来控制全局对象的构造和解析顺序。
1
2
3
4
5
6
7
8
9
|
#ifdef _MSC_VER
#pragma init_seg(lib)
#endif
static MemCheck mc_autocheck_object
#ifdef __GNUC__
__attribute__((init_priority (101)))
#endif
;
|
这里的宏定义部分都是编译器的选项。
posted on 2011-04-05 20:11
小果子 阅读(1603)
评论(1) 编辑 收藏 引用 所属分类:
C++