如果 Realloc() 失败，调用者会丢失前一次 malloc() 调用的数据吗？

Question

以下是考试引述（1% 顶尖大学）。

我失败了，因为我的答案与“批准”的答案不同。

我有一种预感，他（教授，著名的 C 语言专家）的答案不正确。

以下是“已批准”答案后跟的问题。

以下函数存在潜在错误。它是什么，我该如何解决？

提示：这与 realloc() 函数的使用有关。请确定您要更改的行号以及替换它们的内容。

BOOLEAN lengthen_string(char* string, const char newcontents[])
{
        int newlen = strlen(string) + strlen(newcontents) + 1;
        string = realloc(string, newlen);

        if (!string) {
                perror("malloc");
                return FALSE;
        }

        strcat(string, newcontents);
        return TRUE;
}

教授提供的“正确”答案是：

第 4 行：realloc 在分配失败时返回一个 NULL 指针。这意味着一旦失败，原始数据就会丢失。

要解决此问题，请将 realloc 的结果分配给一个临时变量并首先对其进行测试。

即：第 4 行：

char * temp=realloc(string, newlen);
if(!temp) ... (all remains the same)

在旧第 9 行之后，string = temp；

有什么想法吗？

顺便说一句，我的回答是 @string 是一个局部变量，函数的原型应该是 char **string，调用者将指针传递给它的字符串指针，然后被调用者分配任何 realloc()返回值到*string

有什么想法吗？

Answer 1

你们都是对的。

教授是正确的，因为realloc() 在失败时不会改变传入的内存，因此输入string 指针完好无损，但如果失败时的NULL 返回值立即分配给string 然后原始数据丢失和泄露。因此，在将新指针值分配给string 之前，需要先检查失败。

您是对的，因为string 需要通过指针传递，因此如果realloc() 返回不同的内存地址，它可以重新分配一个新值。

正确的解决方案应该是这样的：

BOOLEAN lengthen_string(char** string, const char newcontents[])
{
    if (!string)
    {
        errno = EINVAL;
        perror("bad input");
        return FALSE;
    }
    size_t newsize = strlen(*string) + strlen(newcontents) + 1;
    char *temp = realloc(*string, newsize);
    if (!temp)
    {
        perror("realloc failed");
        return FALSE;
    }
    strcat(temp, newcontents);
    *string = temp;
    return TRUE;
}

或者，还有一些优化空间，例如：

BOOLEAN lengthen_string(char** string, const char newcontents[])
{
    if (!string)
    {
        errno = EINVAL;
        perror("bad input");
        return FALSE;
    }
    char *temp;
    if (!*string)
    {
        temp = strdup(newcontents);
        if (!temp)
        {
            perror("strdup failed");
            return FALSE;
        }
    }
    else
    {
        size_t offset = strlen(*string);
        size_t size = strlen(newcontents) + 1;
        temp = realloc(*string, offset + size);
        if (!temp)
        {
            perror("realloc failed");
            return FALSE;
        }
        memcpy(temp + offset, newcontents, size);
    }
    *string = temp;
    return TRUE;
}