c - 如何避免在每个 "return "之前调用 free()

Question

我正面临 free() 的问题，而 free() 一直是 malloc()。有没有办法避免在下面的代码中多次写入 free(p)？

char *p = (char*) (malloc(100 * sizeof(char)));
if (A)
{
    free(p);
    p=NULL;
    return -1;
}
a++;
if (B)
{
    free(p);
    p=NULL;
    return -1;
}
b++;
if (C)
{
    free(p);
    p=NULL;
    return -1;
}
free(p);
p=NULL;
return 0; 

Answer 1

这可能不受欢迎，但您可以在函数末尾使用单个 goto 标签进行清理

    char *p = malloc(100);
    int ret = -1;
    if (A)
        goto cleanup;
    a++;
    if (B)
        goto cleanup;
    b++;
    if (C)
        goto cleanup;
    ret = 0; /* success */
cleanup:
    free(p);
    return ret; 

请注意，我对您的代码做了一些其他的小改动

• 从 malloc 的返回中删除了强制转换
• 从您的分配计算中删除了 sizeof(char) 的使用。这保证是 1
• 在函数末尾删除了 p 的 NULLing。它即将超出范围，因此它指向已释放的内存并不重要