Realloc 内存拦截

Question

我想阅读几个文件并在内容中添加几行文本。我想将新更改的内容保存在缓冲区中。这些文件的大小可以从 0 字节到 16 MB。通过添加文本行，文件也可以更大，具体取决于内容。

一开始，我用 calloc 预留了 512 字节的内存。

wchar_t *buffer = (wchar_t*)calloc(512, sizeof(wchar_t));

每当将文本添加到内存时，我都会使用函数 realloc 检查保留内存的大小是否仍然足够。这仅适用于程序崩溃的较大文件的小文件。很遗憾，因为星座的关系，无法进行调试。

现在我的问题。 如果我用 realloc 重复扩展内存但它失败了，那么带有 buffer == NULL 的 if 语句是否正确？缓冲区理论上不能为零，因为它之前已经被填满了。

我怎样才能干净地捕捉错误或修复它？

size_t memoryallocated = 512;

wchar_t *buffer = (wchar_t*)calloc(memoryallocated , sizeof(wchar_t));          
while (memoryuse + contenlength >= memoryallocated)
{
 memoryallocated *= 2;
 buffer = (wchar_t *)realloc(buffer, memoryallocated* sizeof(wchar_t));     
 if ((buffer == NULL))
 {
   return NULL;
 }
}

wmemcpy(buffer + memoryuse, contentbuf,contenlength);
memoryuse += contenlength;    // thx @pm100

return buffer;

Answer 1

如果您阅读了realloc 的文档，您会看到如果函数无法调整缓冲区大小（由于内存不足或任何其他原因），它会返回NULL。它不返回旧指针。

因此将返回值与NULL 进行比较是绝对正确的。

除此之外你不应该这样做

buffer = realloc(buffer, ...);

因为如果realloc 返回NULL，则您丢失了未释放的旧指针（realloc 如果无法重新分配则不会释放）。这将导致内存泄漏。

你应该总是这样做 -

void * new_ptr = realloc(buffer, new_size); // new_size > 0 

if (new_ptr == NULL) {
    // Handle error and keep using buffer
} else {
    buffer = new_ptr;
}