使用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;
}