第5章 进程环境（4）_进程资源限制

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

#define RLIMIT(name)   get_limit(#name, name);

void get_limit(char* name, int resource)
{
    struct rlimit limit;
    if(getrlimit(resource, &limit) < 0){
        perror("getrlimit error");
        return;
    }

    printf("%-15s", name);  //左对齐打印资源的名字
    //rlmit_cur：软资源限制
    if(limit.rlim_cur == RLIM_INFINITY){
        printf("    infinity");  //无限制
    }else{
        printf("%10ld", limit.rlim_cur);
    }
    //rlimit_max：硬资源限制
    if(limit.rlim_max == RLIM_INFINITY){
        printf("    infinity");  //无限制
    }else{
        printf("%10ld", limit.rlim_max);
    }
    printf("\n");
}

int main(int argc, char* argv[])
{
    RLIMIT(RLIMIT_AS);       //进程可用存储区大小
    RLIMIT(RLIMIT_CORE);     //core文件最大字节数
    RLIMIT(RLIMIT_CPU);      //CPU时间最大值
    RLIMIT(RLIMIT_DATA);     //数据段最大长度
    RLIMIT(RLIMIT_FSIZE);    //可创建文件的最大长度
    RLIMIT(RLIMIT_LOCKS);    //文件锁的最大数
    RLIMIT(RLIMIT_MEMLOCK);  //使用mlock能在存储器中锁定的最长字节数
    RLIMIT(RLIMIT_NOFILE);   //能打开的最大文件数
    RLIMIT(RLIMIT_NPROC);    //每个用户ID可拥有的最大子进程数
    RLIMIT(RLIMIT_RSS);      //最大驻内存集的字节长度
    RLIMIT(RLIMIT_STACK);    //栈的最大长度

    return 0;
}

#include <unistd.h>
#include <sys/resource.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>

#define MODE S_IRWXU | S_IRWXG | S_IRWXO

int open_file(const char* filename)
{
    int fd = open(filename, O_RDWR | O_CREAT | O_TRUNC, MODE);
    if(fd < 0){
        perror("open error");
        exit(1);
    }

    return fd;
}

int main(int argc, char* argv[])
{
    struct rlimit limit;
    limit.rlim_cur = 4;     //设置软资源限制
    limit.rlim_max = 1024;  //设置硬资源限制

    //设置进程打开文件的最大数
    //(注：0==>标准输入 1==标准输出 2==>标准错误 己打开)
    if(setrlimit(RLIMIT_NOFILE, &limit) < 0){
        perror("setrlimit error");
        exit(1);
    }

    //打开文件
    int fd1, fd2;
    fd1 = open_file("s1.txt"); //正常打开
    if(fd1 < 0){
        exit(1);
    }

    fd2 = open_file("s2.txt");//超过软资源限制指定的4个文件，则报错。
    if(fd2 < 0){
        exit(1);
    }

    close(fd1);
    close(fd2);

    return 0;
}