realloc在第二次调用时失败。

Question

我正在尝试将一堆WCHAR添加到缓冲区中。这个函数就是将其添加到我的缓冲区的方法。

DWORD add_to_buffer(BYTE *databuffer, WCHAR *path, WCHAR *value_name, DWORD type, BYTE *data, DWORD data_size, DWORD already_added) {
    DWORD path_size = wcslen(path) * 2;
    DWORD value_name_size = wcslen(value_name) * 2;

    WCHAR *type_name = reg_type_to_wchar(type);
    DWORD type_size = wcslen(type_name) * 2;


    DWORD total_length = already_added + path_size + value_name_size + type_size + data_size;

    *databuffer = realloc(databuffer, total_length);
    
    CopyMemory(databuffer, path, path_size);
    CopyMemory(databuffer + path_size, value_name, value_name_size);
    CopyMemory(databuffer + path_size + value_name_size, type_name, type_size);
    CopyMemory(databuffer + path_size + value_name_size + type_size, data, data_size);
        
    return total_length;
}

在第二次调用add_to_buffer()时，realloc()失败了。 我基本上一遍又一遍地调用这个函数，同时添加信息并根据需要使其变得更大。 我不确定如何解决这个问题，因为在进入函数时，VS中的一切看起来都是正确的。 databuffer没有在此函数之外释放。

- Stack Overflow

Answer 1

您混淆了指针间接级别。例如，在您的这行代码中：*databuffer = realloc(databuffer, total_length);，在左侧您对databuffer变量进行了取消引用操作，但在函数调用内部，您未这样做。

如果您想使您的函数修改指针（这是它所做的），那么您需要传递一个指向该指针的指针，以便调用模块可以使用修改后的值（新地址）。就像这样：

DWORD add_to_buffer(BYTE **databuffer, WCHAR *path, WCHAR *value_name, DWORD type, BYTE
 *data, DWORD data_size, DWORD already_added) { // Pass a "databuffer" as a DOUBLE pointer
    DWORD path_size = wcslen(path) * 2;
    DWORD value_name_size = wcslen(value_name) * 2;

    WCHAR *type_name = reg_type_to_wchar(type);
    DWORD type_size = wcslen(type_name) * 2;

    DWORD total_length = already_added + path_size + value_name_size + type_size + data_size;

    // It is also bad practice to overwrite the argument in "realloc" calls; save to a
    // temp, so that you can check for failure ...
    BYTE *temp = realloc(*databuffer, total_length); // Not the "*" before databuffer!
    if (temp == NULL) { // Allocation failure ...
        // Here, place code to handle/signal the error
        // But note that we STILL HAVE THE ORIGINAL POINTER!
        return 0; // and return a value that indicates failure
    }
    *databuffer = temp; // Succeeded: we can now safely reassign the passed pointer.
    
    // We now need to also dereference the double pointer in the following calls ...
    CopyMemory(*databuffer, path, path_size);
    CopyMemory(*databuffer + path_size, value_name, value_name_size);
    CopyMemory(*databuffer + path_size + value_name_size, type_name, type_size);
    CopyMemory(*databuffer + path_size + value_name_size + type_size, data, data_size);
        
    return total_length;
}

- Adrian Mole

---

当然，您还需要对函数的任何调用进行适当修改（通常在第一个参数之前添加&运算符）。 - Adrian Mole

不要猜测WCHAR的大小为2，使用sizeof(WCHAR)。 - koder

Answer 2

realloc返回值: "此函数返回指向新分配内存的指针，如果请求失败则返回NULL。" 在第二次调用时，应该没有更多可用的空间来重新分配... 我认为最简单的方法是将其更改为: *databuffer = realloc(databuffer, total_length); if(databuffer == NULL) { *databuffer = malloc(total_length); }

- Devi Khositashvili

Answer 3

*databuffer = realloc(databuffer, total_length);

在这里，你将realloc返回的值分配给databuffer的第一个字节。你不应该对databuffer进行解引用。

- Skult

---

但是，调用模块将永远不会知道重新分配指针的新值。需要一个双指针参数。 - Adrian Mole

好的观点，我关注了代码的问题，而没有专注于如何正确修复它。 - Skult