使用宏或自定义函数检查malloc返回值的高效方式及场景对比

两种实现方式 #

1. 宏定义方式 #

#define ALLOC(p,n) do{ *p=malloc(n*sizeof **p); } while(0)

注：该宏仅简化了内存分配的写法，未包含返回值检查逻辑，需额外添加判断代码。

2. 静态内联函数方式 #

static inline void *MallocOrDie(size_t MemSize)
{
     void *AllocMem = malloc(MemSize);
     /* Some implementations return null on a 0 length alloc,
     * we may as well allow this as it increases compatibility
     * with very few side effects */
     if(!AllocMem && MemSize)
     {
          printf("Could not allocate memory!");
          exit(-1);
     }
     return AllocMem;
}

该函数自带malloc返回值检查逻辑，当分配失败且请求内存不为0时直接终止程序。

不同场景下的效率对比 #

场景1：分配失败直接终止程序 #

这种场景下，静态内联函数的效率和宏完全相当——编译器会将static inline函数的代码直接插入调用位置，不会产生函数调用的栈帧开销，和宏展开后的效果一致。

同时内联函数比宏更安全：它具备类型检查（比如传递的MemSize类型错误会触发编译报错），也不会出现宏常见的参数副作用陷阱（比如宏参数是带自增/自减的表达式时，可能被多次求值，内联函数则只会求值一次）。优先选内联函数，效率无损失还更可靠。

场景2：需要自定义分配失败处理逻辑 #

如果你的需求不是直接终止程序，而是要做重试、返回错误码等自定义处理，宏的灵活性更高——你可以在宏调用后自行添加检查逻辑，或者修改宏定义嵌入自定义检查，这种情况下宏的效率和手动写malloc+检查完全一致，没有额外开销。

而示例中的内联函数固定了失败就退出的逻辑，无法适配自定义处理场景；如果修改内联函数支持传入回调或错误处理逻辑，效率上和宏也不会有差异，但宏的写法更直接（不需要额外参数）。不过使用宏时要注意避免参数副作用问题，比如不要传入i++这类会改变值的表达式。

总结 #

• 固定失败终止场景：内联函数效率与宏持平，且更安全，优先使用内联函数。
• 自定义失败处理场景：宏更灵活，效率与适配后的内联函数相当，但需注意宏的使用陷阱。

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