c++ - 如何创建随机内存分配失败？

Question

我最近发现了PageHeap ，一个 Windows 调试工具。它的选项 /fault 允许偶尔注入(inject)内存分配失败。就像，我希望每 100 次或 1000 次分配失败一次，由您决定。我们在 Linux 上有类似的东西吗？

我可以想到 ulimit，但这不是一回事，一旦达到内存，它就会使后续的每一次分配都失败，而不仅仅是一次。

Answer 1

是的，您可以使用 LD_PRELOAD 做类似的事情；你需要编译如下内容:

#include <dlcfn.h>

void *malloc(size_t size) {
  static unsigned count = 0;
  static void *(*real_malloc)(size_t) = NULL;

  if (!real_malloc)
    real_malloc = dlsym(RTLD_NEXT, "malloc");

  if (++count == 100) {
    count = 0;
    return NULL;
  }

  return real_malloc (size);
}

作为共享对象，例如与

gcc -shared -ldl -fPIC my_malloc.c -o libmy_malloc.so

完成后，您可以像这样运行您的程序:

LD_PRELOAD=/path/to/libmy_malloc.so ./my_program

它会调用替换 malloc() 例程。

我想上面的预构建版本具有比这更多的调试功能，但你明白了。

哦，顺便说一句，上面的代码没有经过测试(写在 Stack Overflow 中)。它还不是线程安全的，如果替换 malloc()，您需要线程安全，因此您需要使用 pthread_once()，以及用于修复此问题的 pthread_mutex API。