C语言中如何判断当前函数是否需要返回值以适配内存分配检查宏

标准C语言不存在你提到的类似argc的内置标识/预定义宏，能够在预处理或编译阶段自动判断当前函数是否为void返回类型、是否需要带返回值返回。
原因很简单：C语言的宏展开是预处理阶段的纯文本替换操作，这个阶段编译器还没有完成函数签名的语法解析，根本感知不到当前所在函数的返回值类型；即使到后续编译阶段，C标准也没有定义对应的内置接口来查询当前函数的返回类型。少数编译器的私有扩展可能提供相关能力，但兼容性极差，生产环境完全不建议使用。因此你设想的完全自适应、不需要额外传参的通用检查宏，靠标准C无法实现。

方案1：给宏增加错误返回值参数（兼容性最好，全C标准支持） #

这是工业界C项目最常用的实现方式，无编译器依赖，适配所有返回类型的函数：

// 用do{...}while(0)包裹宏，避免if/else嵌套场景下的语法错误
// 所有宏参数额外加括号，避免运算符优先级导致的逻辑异常
#define CHECK_CALLOC(ptr, err_ret)                                    \
    do {                                                              \
        if ((ptr) == NULL)                                            \
        {                                                             \
            printf("Error: failed to calloc in %s line %d of func %s.\n", \
                   __FILE__, __LINE__, __FUNCTION__);                 \
            return err_ret;                                           \
        }                                                             \
    } while(0)

调用方式非常灵活：

• 在void返回类型的函数中使用时，第二个参数留空即可：CHECK_CALLOC(p, );，宏展开后就是符合语法的return;
• 在返回指针的非void函数中使用时，传入空指针作为错误返回值：CHECK_CALLOC(p, NULL);
• 在返回int错误码的函数中使用时，传入对应错误码即可：CHECK_CALLOC(p, -ENOMEM);

方案2：统一跳转到错误清理标签（适合复杂资源管理场景） #

如果函数里有多处内存、文件句柄等资源申请，需要在错误退出时统一释放资源，不建议在宏里直接return，而是用goto跳转到统一的错误处理段，避免资源泄漏：

#define CHECK_CALLOC_GOTO(ptr, err_label)                             \
    do {                                                              \
        if ((ptr) == NULL)                                            \
        {                                                             \
            printf("Error: failed to calloc in %s line %d of func %s.\n", \
                   __FILE__, __LINE__, __FUNCTION__);                 \
            goto err_label;                                           \
        }                                                             \
    } while(0)

调用示例：

int foo() {
    int *a = NULL;
    char *b = NULL;
    a = calloc(10, sizeof(int));
    CHECK_CALLOC_GOTO(a, err_exit);
    b = calloc(20, sizeof(char));
    CHECK_CALLOC_GOTO(b, err_exit);

    // 正常业务逻辑
    // ...
    return 0;
err_exit:
    free(a);
    free(b);
    return -ENOMEM;
}

你最初写的宏存在两个C语言宏编写的常见问题：

1. 没有用do{...}while(0)结构包裹多语句，在无花括号的if/else分支中使用时，会出现else匹配错误，直接触发编译失败。
2. 宏参数ptr没有额外加括号，如果传入带运算符的表达式（比如CHECK_CALLOC(p + offset)），会因为运算符优先级问题出现逻辑错误。

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