在C99中实现Go风格defer关键字的跨平台方案问询

针对你需要在C99中实现无容量限制、跨平台的Go风格defer功能，以下是基于编译器栈清理属性的解决方案，既规避了固定数组的限制，又兼容主流编译器：

核心思路 #

利用GCC/clang的__attribute__((cleanup))和MSVC的__declspec(cleanup)扩展，结合栈上链表结构实现：

1. 每个函数调用栈帧内维护一个defer任务链表，无需预分配固定大小数组
2. 编译器自动在函数返回时触发链表遍历，执行所有延迟任务
3. 链表节点直接分配在栈上，避免动态内存分配的性能开销

完整实现代码 #

#include <stddef.h>

// 定义defer任务节点，存储要延迟执行的函数指针
typedef void (*defer_func_t)(void*);
typedef struct defer_node {
    defer_func_t func;
    void* arg;
    struct defer_node* next;
} defer_node_t;

// 遍历并执行当前栈帧所有defer任务（逆序执行，匹配Go的defer语义）
static void defer_run_all(defer_node_t** head_ptr) {
    defer_node_t* current = *head_ptr;
    defer_node_t* reversed = NULL;

    // 先反转链表，保证先defer的任务后执行
    while (current != NULL) {
        defer_node_t* next = current->next;
        current->next = reversed;
        reversed = current;
        current = next;
    }

    // 执行所有任务
    current = reversed;
    while (current != NULL) {
        current->func(current->arg);
        current = current->next;
    }
}

// 跨平台栈清理属性宏
#if defined(__GNUC__) || defined(__clang__)
#define DEFER_CLEANUP_ATTR(func) __attribute__((cleanup(func)))
#elif defined(_MSC_VER)
#define DEFER_CLEANUP_ATTR(func) __declspec(cleanup(func))
#else
#error "Unsupported compiler: only GCC/clang/MSVC are supported"
#endif

// 初始化当前函数的defer链表（必须在函数开头调用）
#define DEFER_INIT() \
    defer_node_t* defer_head = NULL; \
    DEFER_CLEANUP_ATTR(defer_run_all) defer_node_t** defer_head_ptr = &defer_head;

// 定义延迟执行的代码块（与Go的defer用法一致）
#define DEFER(code) \
    do { \
        static void defer_lambda(void* unused) { \
            (void)unused; \
            code; \
        } \
        defer_node_t defer_node = { \
            .func = defer_lambda, \
            .arg = NULL, \
            .next = defer_head \
        }; \
        defer_head = &defer_node; \
    } while(0)

使用示例 #

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

int main() {
    DEFER_INIT();

    // 示例1：延迟释放内存
    int* num = malloc(sizeof(int));
    *num = 42;
    DEFER({
        free(num);
        printf("Freed int pointer\n");
    });

    // 示例2：延迟打印信息
    DEFER(printf("Exiting main function\n"));

    printf("Current num value: %d\n", *num);

    return 0;
}

运行输出：

Current num value: 42
Exiting main function
Freed int pointer

方案优势 #

• 无容量限制：基于栈上链表，支持任意数量的DEFER调用，无需手动指定容量
• 跨平台兼容：通过条件编译适配GCC/clang和MSVC，无需修改业务代码
• 语义匹配Go：逆序执行延迟任务，与Go的defer行为完全一致
• 低性能开销：所有节点分配在栈上，无动态内存分配或多次alloca调用的损耗

注意事项 #

• 必须在每个需要使用DEFER的函数开头调用DEFER_INIT()
• 不支持在函数外或静态代码块中使用DEFER
• 极端场景下大量DEFER调用会占用栈空间，需根据实际栈大小调整使用量

内容的提问来源于stack exchange，提问作者CodeSurgeon