编写安全代码：小心使用realloc

本文的copyleft归[e-mail]所有，使用GPL发布，可以自由拷贝，转载。但转载请保持文档的完整性，注明原作者及原链接，严禁用于任何商业用途。
作者：[e-mail]
博客：linuxfocus.blog.chinaunix.net    

在良好的代码风格中，其中有一项要求就是，一个函数只做一件事情。如果该函数实现了多个功能，那基本上可以说这不是一个设计良好的函数。

今天看C库中的函数realloc。其原型是void *realloc(void *ptr, size_t size);函数说明如下：

realloc() changes the size of the memory block pointed to by ptr to size bytes. The contents will be unchanged to the minimum of the old and new sizes; newly allocated memory will be uninitialized. If ptr is NULL, the call is equivalent to malloc(size); if sizeis equal to zero, the call is equivalent to free(ptr). Unless ptr is NULL, it must have been returned by an earlier call to malloc(), calloc() or realloc(). If the area pointed to was moved, a free(ptr) is done.

总结一下，有以下几种行为：

1. 与名字相符，真正的realloc，参数ptr和size均不为NULL，重新调整内存大小，并将新的内存指针返回，并保证最小的size的内容不变；

2. 参数ptr为NULL，但size不为0，那么行为就等于malloc(size)；

3. 参数size为0，则realloc的行为为free(ptr)；这时原有的指针已经被free掉，不能继续使用。而此时realloc的返回值为NULL。这意味着不检查realloc的返回值，直接使用，会导致crash。

看，一个简单C库函数，却赋予了三种行为，所以这个realloc并不是设计良好的库函数。估计也是为了兼容性，才容忍这个函数一直在C库中。虽然在编码中，realloc会提供一定的方便，但是也很容易引发bug。

下面就举两个例子，来说明一下。

1. realloc第一种行为引发的bug

void *ptr = realloc(ptr, new_size);
if (!ptr) {
    错误处理
}

这里就引出了一个内存泄露的问题，当realloc分配失败的时候，会返回NULL。但是参数中的ptr的内存是没有被释放的。如果直接将realloc的返回值赋给ptr。那么当申请内存失败时，就会造成ptr原来指向的内存丢失，造成泄露。

正确的处理应该是这样

void *new_ptr = realloc(ptr, new_size);
if (!new_ptr) {
    错误处理。
}
ptr = new_ptr

2. 第三种行为引发的bug

这种bug由一种不好的编程习惯引发的。即认为申请内存始终可以成功，因此并不检查malloc的返回值。这在一般情况下，不会引发问题。但是对于realloc来说，当new_size为0时，realloc返回NULL。而在后面的代码上，继续使用new_ptr，比如会导致程序crash。

void *new_ptr = realloc(old_ptr, new_size);

 //其它代码

 ...... ......

从上面可以看出，在面对这个设计并非良好的API时，我们需要小心小心再小心。上面只是举了两个例子，其实还有一些其它的小问题。