安全起见,小心使用C语言realloc()函数

来源:互联网 发布:c语言返回值什么意思 编辑:程序博客网 时间:2024/05/20 02:52
在C语言中,良好的编程习惯要求一个函数只做一件事,如果一个函数实现了若干功能,可以说基本是一个糟糕的设计。

C语言 realloc() 函数位于 stdlib.h 头文件中,其原型为:
    void *realloc(void *ptr, size_t size); //
void *  意思是这个函数返回的是指针变量。 void * ptr传递进去的是指针变量

realloc() 会将 ptr 所指向的内存块的大小修改为 size,并将新的内存指针返回。

设之前内存块的大小为 n,如果 size < n,那么截取的内容不会发生变化,如果 size > n,那么新分配的内存不会被初始化。

如果 ptr = NULL,那么相当于调用 malloc(size);如果 size = 0,那么相当于调用 free(ptr)。虽然指针被free掉,但是指向NULL;

如果 ptr 不为 NULL,那么他肯定是由之前的内存分配函数返回的,例如 malloc()、calloc()或realloc()。

如果 ptr 所指的内存块被移动,那么会调用 free(ptr)。

看吧,一个简单的 realloc() 却赋予了好几个功能,这并不是良好的函数设计。估计也是为了兼容性,才容忍这个函数一直在C库中。虽然在编码中,realloc() 会提供一定的方便,但是也很容易引发Bug。

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

1) realloc() 第一种行为引发的Bug
  1. void *ptr = realloc(ptr, new_size); //用这个函数的时候不能返回值赋值给自己。要用第三者代替。
  2. if (!ptr) {
  3. // 错误处理
  4. }
这里就引出了一个内存泄露的问题,当realloc() 分配失败的时候,会返回NULL。但是参数中的 ptr 的内存是没有被释放的。如果直接将realloc()的返回值赋给ptr。那么当申请内存失败时,就会造成ptr原来指向的内存丢失,造成内存游离和泄露。

经常这样连贯着用:
  1. void *new_ptr = realloc(ptr, new_size);
  2. if (!new_ptr) {
  3. // 错误处理。
  4. }
  5. ptr = new_ptr

2) 第三种行为引发的Bug
实际上,malloc(0)是合法的语句,会返还一个合法的指针,且该指针可以通过free去释放。这就造成了很多人对realloc()的错误理解,认为当size为0时,实际上realloc()也会返回一个合法的指针,后面依然需要使用free去释放该内存。
纯文本复制
  1. void *new_ptr = realloc(old_ptr, new_size);
  2. //其它代码
  3. free(new_ptr); 用完还是要释放
由于错误的认识,不去检验new_size是否为0,还是按照new_size不为0的逻辑处理,最后并free(new_ptr)。这里就引入了double free的问题,造成程序崩溃。

所以,realloc() 这个设计并不怎么优良的函数陷阱还是不少的,一不小心就踩雷了,上面只是两个简单的小例子,大家在实际使用的时候还应该注意一些其他小问题。
如果你希望更加深入和透彻地学习编程,请了解VIP会员、C语言一对一辅导、C++一对一辅导。
0 0
原创粉丝点击