如何在C语言中测试malloc内存分配失败场景？

以下是针对你的场景，可快速触发malloc/realloc失败、验证错误处理逻辑的实用方法：

1. 用ulimit快速限制进程内存 #

Linux的ulimit命令可以临时限制当前shell会话中进程的虚拟内存大小，轻松触发内存分配失败：

• 先设置虚拟内存上限（比如设为1MB）：

  ulimit -v 1024
  

• 直接运行你的程序，当程序申请内存超过这个限制时，malloc就会返回NULL，触发你的错误处理分支。
• 测试完成后，执行ulimit -v unlimited即可取消限制。

2. 用LD_PRELOAD注入自定义malloc实现 #

通过动态库预加载替换系统原生malloc，可精准控制何时返回NULL，无需修改业务代码：

1. 编写测试用动态库代码（如test_malloc.c）：

   #include <stdio.h>
   #include <stdlib.h>
   #include <dlfcn.h>
   
   // 保存系统原生malloc的函数指针
   static void* (*real_malloc)(size_t) = NULL;
   
   void* malloc(size_t size) {
       if (!real_malloc) {
           real_malloc = dlsym(RTLD_NEXT, "malloc");
       }
   
       // 模拟你的场景：当申请1字节内存时返回NULL
       if (size == sizeof(char)) {
           printf("模拟malloc分配失败\n");
           return NULL;
       }
   
       // 其他情况调用原生malloc
       return real_malloc(size);
   }
   

2. 编译为动态库：

   gcc -shared -fPIC -o libtest_malloc.so test_malloc.c -ldl
   

3. 预加载库并运行程序：

   LD_PRELOAD=./libtest_malloc.so ./your_program
   

3. 编译时通过宏替换注入测试逻辑 #

如果不想用动态库，可通过编译宏替换malloc，仅在测试版本生效：

1. 编写测试头文件（如test_malloc.h）：

   #ifndef TEST_MALLOC_H
   #define TEST_MALLOC_H
   
   #include <stdlib.h>
   
   static int trigger_fail = 1;
   
   void* my_test_malloc(size_t size) {
       // 仅第一次malloc触发失败，后续正常
       if (trigger_fail) {
           trigger_fail = 0;
           return NULL;
       }
       return malloc(size);
   }
   
   #define malloc(size) my_test_malloc(size)
   
   #endif
   

2. 编译测试版本：

   gcc -o your_program_test your_program.c -include test_malloc.h
   

3. 运行测试版本即可触发失败逻辑，正常编译时去掉-include test_malloc.h即可恢复原逻辑。

4. 用cgroups精细控制内存（适合复杂场景） #

如果需要更精细的内存限制（比如区分物理内存/虚拟内存），可使用Linux cgroups（需root权限）：

1. 创建cgroup目录：

   mkdir /sys/fs/cgroup/memory/test_mem_limit
   

2. 设置内存上限为1MB：

   echo 1048576 > /sys/fs/cgroup/memory/test_mem_limit/memory.limit_in_bytes
   

3. 将进程加入cgroup并运行：

   echo $$ > /sys/fs/cgroup/memory/test_mem_limit/cgroup.procs
   ./your_program
   

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