linux可以用setrlimit进行资源限制,今天说的是setrlimit限制进程的cpu使用时间
代码如下:
#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
int main()
{
struct rlimit rlim;
rlim.rlim_cur = 10;
rlim.rlim_max = RLIM_INFINITY;
setrlimit(RLIMIT_CPU, &rlim);
//sleep(60);
while(1);
return 0;
}
此代码中,限制了进程的的cpu使用时间是10秒,当进程使用cpu的时间超过10秒时,就会收到SIGXCPU信号,该信号的默认处理是终结进程并产生core文件。
运行此程序,10秒后会运行完,现象如下:
可以找到对应的core文件,调试此core文件,结果如下:
可以看到,由于收到24信号,进程被终结,而SIGXCPU对应的数字就是24。
注意,sleep不占用cpu时间,所以如果将代码中的sleep(60)放开,则从运行到终结,需要70秒的时间。
咱们再看下多线程的进程,代码如下:
#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
#include <pthread.h>
void *th_fun(void *arg)
{
while(1);
}
int main()
{
pthread_t pid;//线程ID
void *tret;
int n=10;
struct rlimit rlim;
rlim.rlim_cur = 10;
rlim.rlim_max = RLIM_INFINITY;
setrlimit(RLIMIT_CPU, &rlim);
pthread_create(&pid,NULL,th_fun,(void*)&n);
while(1);
return 0;
}
该程序开启了一个线程,所以进程启动后,是两核拉满,然后经过5秒钟,进程被终结,注意,是5秒钟后被终结,不是10秒,所以setrlimit限制cpu时,是以单核为单位的。
rlim.rlim_cur = 10限制了以单核为单位,最多能运行10秒。现在两个线程在两个核同时跑,故5秒后,该进程的cpu占用时间(以单核为单位)就达到了10秒,从而进程被终结。