C语言中malloc()/calloc()内存分配失败的处理方案咨询

对于你开发的这种单线程简易文件读取提取应用，得看具体场景，但大部分情况优先选直接终止程序，下面给你掰扯清楚：

先说说重试分配的尴尬之处 #

• 现代系统里malloc/calloc失败，基本都是真的内存不够用了，重试十次八次大概率还是失败，除非刚好有其他进程释放了内存，但这种情况在单线程工具里太少见了——你处理文件的内存需求是可预期的，总不会突然暴涨到把系统内存吃光。
• 要是没做好重试逻辑，很容易陷入死循环，程序卡着不动，反而不如直接报错退出来得干脆。

直接用exit()终止的好处（适合你的场景） #

• 你的应用功能单一，核心就是读取文件提取信息，没内存存数据的话，根本干不了正事，继续运行也没意义。
• 退出前给用户输出明确的错误信息，比如用fprintf(stderr, "内存分配失败: %s\n", strerror(errno));，用户一眼就能知道是内存不够还是其他问题，比闷头重试靠谱多了。
• 代码写起来简单，符合简易应用的定位，不用折腾复杂的错误返回和处理逻辑，省事儿。

举个实际代码例子：

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

int main() {
    // 假设已经获取了文件大小file_size
    size_t file_size = 1024 * 1024;
    char* buffer = malloc(file_size);
    if (buffer == NULL) {
        fprintf(stderr, "分配内存失败，无法读取文件: %s\n", strerror(errno));
        exit(EXIT_FAILURE);
    }
    // 后续读取文件、提取信息的逻辑
    free(buffer);
    return 0;
}

什么时候考虑返回NULL重试？ #

只有当你的应用有可降级的处理逻辑时才值得这么做：比如你的应用同时处理多个文件，某个文件分配内存失败，还能跳过它继续处理其他文件；或者有非核心的功能模块，失败了不影响主流程。但对你的单线程简易工具来说，这种情况应该不存在。

如果真要做重试，也得谨慎，比如先尝试释放一些非必要的内存再重试，别盲目死循环：

char* try_allocate(size_t size) {
    char* buf = malloc(size);
    if (buf == NULL) {
        // 假设这里有之前分配的非必要内存，可以先释放
        // free(some_non_critical_buffer);
        fprintf(stderr, "第一次分配失败，尝试回收内存后重试...\n");
        buf = malloc(size);
    }
    return buf;
}

总的来说，对你的场景，直接用exit()输出错误后终止是最合理的选择，简单直接，用户体验也不差。

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