Linux主动限制资源

限制cpu内存资源

他是一个linux接口

参数认识

设置一个进程占用资源的上限

第一个参数resource

重点讲述第一个和第三个

第一个表示是虚拟地址

第三个表示cpu时间

其他的就是针对一些数据的大小, 文件大小, core文件大小等

第二个参数

这是一个OS内部的结构体

两个参数分别要进行设置, 对于hard硬约束要设置为无穷大(他表示我们能设置的最大的资源限制上限), 主要是时间限制都与soft软约束有关,但是soft不能超过hard

RLIM_INFINITY表示当前资源的最大值

功能测试

超时溢出

#include <iostream>
#include <sys/time.h>
#include <sys/resource.h>

int main()
{
    struct rlimit r;
    r.rlim_cur = 1;
    r.rlim_max = RLIM_INFINITY;
    setrlimit(RLIMIT_CPU, &r);
    while(1);
    

    return 0;
}

运行结果:

那么OJ超时就是利用这个原理

内存溢出

#include <iostream>
#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>

int main()
{
    struct rlimit r;
    r.rlim_cur = 1024 * 1024 * 20;// 限制大小为20MB
    r.rlim_max = RLIM_INFINITY;
    setrlimit(RLIMIT_AS, &r);
    int count = 0;
    while(1)
    {
        int *p = new int[1024*1024];// 每次申请一个1M大小的内存
        std::cout << "size: " << ++count << std::endl;
        sleep(1);
    }
    

    return 0;
}

内存溢出

OJ的内存溢出也是这个原理

资源不足导致的进程终止是通过信号处理的

可以通过该代码查看是通过哪个信号进行的终止进程

信号捕捉

在main函数内将信号的处理机制设置为自定义函数

    for(int i = 1; i <= 31; ++i)
    {
        signal(i, Handler); // 将1 -- 31号信号的信号设置为Handler
    }

像一些特殊的, 如9号信号就不能进行设置, 他会自动跳过

那么设置完毕, 当收到某个信号就会执行这个自定义函数

void Handler(int signal)
{
    std::cout << "signal: " << signal << std::endl;// 收到某个信号就输出该信号的值
}

他的功能是打印对应信号的数值

整个验证内存限制下收到的信号流程就是:

#include <iostream>
#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
#include <signal.h>

void Handler(int signal)
{
    std::cout << "signal: " << signal << std::endl;// 收到某个信号就输出该信号的值
}
int main()
{

    for(int i = 1; i <= 31; ++i)
    {
        signal(i, Handler); // 将1 -- 31号信号的信号设置为Handler
    }
    struct rlimit r;
    r.rlim_cur = 1024 * 1024 * 20;// 限制大小为20MB
    r.rlim_max = RLIM_INFINITY;
    setrlimit(RLIMIT_AS, &r);
    int count = 0;
    while(1)
    {
        int *p = new int[1024*1024];// 每次申请一个1M大小的内存
        std::cout << "size: " << ++count << std::endl;
        sleep(1);
    }
    

    return 0;
}

结果演示

对于验证时间限制下收到的信号,可以自行尝试， 这边就不在进行赘述