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

#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;
}