如何限制可用内存使`malloc()`失败？

2024年1月4日 213次阅读

我想通过限制可用内存使malloc()失败.

$ulimit -v 1000
$./main.exe 10000000
0x102bfb000

但即使使用ulimit,以下程序仍然可以正确完成.有人知道怎么让malloc()失败吗？谢谢.

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

int main(int argc, char *argv[]) {
    size_t size = atoi(argv[1]);
    void *ptr = NULL;

    if((ptr = malloc(size)) == NULL) {
        perror("malloc()");
        exit(1);
    }

    printf("%p\n", ptr);
    free(ptr);
    return 0;
}

编辑：以上是在Mac OS X上.

在Linux上,我得到了分段错误.为什么malloc()会导致分段错误？如何让malloc()返回一个NULL指针？

最佳答案 根据文件：

在ptr是空指针的情况下,该函数的行为类似于malloc,分配一个大小字节的新块并返回指向其开头的指针

如果要限制程序可以分配的内存,可以使用：

#include <sys/time.h>
#include <sys/resource.h>
rlimit l;
getrlimit(RLIMIT_AS, &l);
l.rlim_cur = 1000;
setrlimit(RLIMIT_AS, &l);

http://man7.org/linux/man-pages/man2/setrlimit.2.html

点赞