使用setrlimit控制进程使用系统内存

最新推荐文章于 2022-12-31 22:14:12 发布

原创最新推荐文章于 2022-12-31 22:14:12 发布 · 2.9k 阅读

0 ·

CC 4.0 BY-SA版权

C函数-OPQRST 专栏收录该内容

15 篇文章

订阅专栏

本文通过C语言实现了一个简单的程序来设置进程的最大内存使用限制，并通过不断分配内存直至失败来验证设置的有效性。该程序利用了POSIX的rlimit机制来控制进程的资源消耗。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/resource.h>
#include <errno.h>

int main(void)
{
	int nMemoryLimit = 1024 * 1024 * 5;//5M

	struct rlimit memoryL;

	getrlimit(RLIMIT_AS, &memoryL);
	printf("before set mem:%lu,%lu\r\n", (unsigned long)memoryL.rlim_cur, (unsigned long)memoryL.rlim_max);

	memoryL.rlim_cur = nMemoryLimit;
	setrlimit(RLIMIT_AS ,&memoryL);

	getrlimit(RLIMIT_AS, &memoryL);
	printf("after set mem:%lu,%lu\r\n", (unsigned long)memoryL.rlim_cur, (unsigned long)memoryL.rlim_max);

	int i = 0;
	while(1)
	{
		i++;
		//当进程内存达到最大值时，malloc calloc返回NULL,记录错误errno=ENOMEM
		errno  = 0;
		char *psString = (char *)calloc(1, 1024);
		if(NULL == psString)
		{
			//printf("errno=%d, ENOMEM=%d\r\n", errno, ENOMEM);
		}
		else
		{
			printf("success, %d\r\n", i);
		}
	}
	return 0;
}