在 C 中处理错误和释放内存的正确方法

Question

在某些函数中，我需要使用 malloc() 分配内存并有多个 if..else 语句，如伪代码所示：

allocate memory
if condition_1
    do_stuff
    if condition_2
        do_more_stuff
    else
        error
else
    error
free allocated memory
return

所以我一开始就分配内存，如果一切正常，它就会被释放。但目前错误功能仅打印错误消息并退出程序。但是正如我经常读到的那样，尽管当程序退出并且 OS 句柄通常处理之后的释放时，但不释放内存，这不是好的风格。怎么偷懒把钱放出来？我是否必须编写一个错误函数，将每个指针都指向我分配的必须释放的内存，指针可能具有不同的数据类型？或者我应该在调用错误函数之前放置 free(ptr) 吗？一个错误函数，它接受一个数据类型为 void 的指针数组，并且比所有都释放，会达到这个目的吗？

Answer 1

据我了解，您关心的是在因错误退出程序时释放所有内存，但又不想手动处理所有指针。

这里有一个有趣的想法，写一个函数allocMemory，returns malloc 的结果同时也把指针放入链表，然后freeMemory 从链表中删除它，最后释放所有函数遍历列表并释放所有指针。使用allocMemory和freeMomory函数代替malloc和free，出错时调用freeMemory函数

Answer 2

我有两个解决方案。

您可以在调用 free 和 error 的地方放置标签：

void function(void)
{
    Memory *p = malloc(sizeof(*p));
    if (condition_1) {
        do_stuff();
        if (condition_2) {
            do_more_stuff();
        } else {
            goto err;
        }
    } else {
        goto err;
    }
    free(p);
    return;
err:
    free(p);
    error();
}

您也可以使用标志来标记错误：

void function(void)
{
    Memory *p = malloc(sizeof(*p));
    bool err = false;
    if (condition_1) {
        do_stuff();
        if (condition_2) {
            do_more_stuff();
        } else {
            err = true;
        }
    } else {
        err = true;
    }
    free(p);
    if (err)
        error();
}

我认为第二种解决方案在这种情况下看起来最好，但它们都同样有效。

Answer 3

不要使用 goto。一次性使用。此外，如果您需要错误标志，请将其默认为 true 而不是 false 以保存代码：

...malloc...
err = 1;
do {
    ...
    if <condition> break;
    ...
    if <condition> break;
    ...
    if <condition> break;
    ...
    err = 0;
} while (0);
...free...
if (err) ...

在 C 中处理错误和释放内存的正确方法

Right way to handle errors and free memory in C

c

memory

error-handling

malloc