是否存在不使用全局变量在 atexit 或类似中释放内存的方法？

Question

我正在用 C 语言开发一个项目，我需要释放分配的内存并在它退出之前关闭所有打开的文件。

我决定实现一个 clean 函数来完成所有这些工作并使用 atexit 调用它，因为有很多可能的退出场景。

问题是atexit不允许我设置带参数的函数，所以我无法将进程结束时需要释放的指针发送给clean。

所以我需要将每个可能需要释放的指针以及可能在程序中保持打开的每个文件都声明为全局变量？ （我已经这样做了，但看起来不太好）或者确实存在与atexit 类似的允许发送参数的函数？或者更可能还有另一种我想念的方式？

Answer 1

在函数内使用静态指针：

#include <stdio.h>
#include <stdlib.h>

void atexit_clean(void *data);

static void clean(void)
{
    atexit_clean(NULL);
}

void atexit_clean(void *data)
{
    static void *x;

    if (data) {
        x = data;
        atexit(clean);
    } else {
        free(x);
    }
}

int main(void)
{
    int *a = malloc(sizeof(int));

    atexit_clean(a);
    return 0;
}

使用单个全局变量的另一种方法：您可以将所有要释放的对象存储在指针数组或链表中，此示例使用realloc（为简洁起见，不检查 (m/re)alloc）：

#include <stdio.h>
#include <stdlib.h>

static void **vclean;
static size_t nclean;

void atexit_add(void *data)
{
    vclean = realloc(vclean, sizeof(void *) * (nclean + 1));
    vclean[nclean++] = data;
}

void clean(void)
{
    size_t i;

    for (i = 0; i < nclean; i++) {
        free(vclean[i]);
    }
    free(vclean);
}

int main(void)
{
    int *a, *b, *c;
    double *d;
    int e = 1;

    atexit(clean);
    a = &e;
    b = malloc(sizeof(int));
    atexit_add(b);
    c = malloc(sizeof(int));
    atexit_add(c);
    d = malloc(sizeof(double));
    atexit_add(d);
    return 0;
}

Answer 2

无法将任何参数传递给atexit()，因此您无法使用全局变量。

当您的程序通过exit() 或从main() 返回正常终止时，它将自动刷新和关闭所有打开的流并（在大多数操作系统下）释放分配的内存。但是，最好在程序终止之前明确清理资源，因为这通常会导致程序更加结构化。 有时编写程序最简洁的方法是退出并将清理工作留给实现。

但请注意，您应该始终检查fclose() 的返回值。请参阅“What are the reasons to check for error on close()?”，了解如果不这样做会发生什么的轶事。