存在一种在不使用全局变量的情况下释放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;
}