blob: fb3233fc69e2ba1ff957aadb95aae2776d37c325 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#include <sys/types.h>
#include <sys/resource.h>
#include "ninitfeatures.h"
static char *arg;
static void doit(int resource)
{
unsigned long u;
struct rlimit r;
if (getrlimit(resource,&r) == -1) {
carp("g","etrlimit failed");
return;
}
if (arg[0]=='=' && arg[1]==0)
r.rlim_cur = r.rlim_max;
else {
if (arg[scan_ulong(arg,&u)]) {
carp("invalid line: ", arg-1);
return;
}
r.rlim_cur = u;
if (r.rlim_cur > r.rlim_max)
r.rlim_cur = r.rlim_max;
}
if (setrlimit(resource,&r) == -1)
carp("s","etrlimit failed");
}
void softlimit(char **argv) /*EXTRACT_INCL*/ {
while (*argv) {
arg = argv[0] + 1;
switch(argv[0][0]) {
case 0: break;
case 'a':
#ifdef RLIMIT_AS
doit(RLIMIT_AS);
#endif
#ifdef RLIMIT_VMEM
doit(RLIMIT_VMEM);
#endif
break;
case 'c':
#ifdef RLIMIT_CORE
doit(RLIMIT_CORE);
#endif
break;
case 'd':
#ifdef RLIMIT_DATA
doit(RLIMIT_DATA);
#endif
break;
case 'f':
#ifdef RLIMIT_FSIZE
doit(RLIMIT_FSIZE);
#endif
break;
case 'l':
#ifdef RLIMIT_MEMLOCK
doit(RLIMIT_MEMLOCK);
#endif
break;
case 'm':
#ifdef RLIMIT_DATA
doit(RLIMIT_DATA);
#endif
#ifdef RLIMIT_STACK
doit(RLIMIT_STACK);
#endif
#ifdef RLIMIT_MEMLOCK
doit(RLIMIT_MEMLOCK);
#endif
#ifdef RLIMIT_VMEM
doit(RLIMIT_VMEM);
#endif
#ifdef RLIMIT_AS
doit(RLIMIT_AS);
#endif
break;
case 'o':
#ifdef RLIMIT_NOFILE
doit(RLIMIT_NOFILE);
#endif
#ifdef RLIMIT_OFILE
doit(RLIMIT_OFILE);
#endif
break;
case 'p':
#ifdef RLIMIT_NPROC
doit(RLIMIT_NPROC);
#endif
break;
case 'r':
#ifdef RLIMIT_RSS
doit(RLIMIT_RSS);
#endif
break;
case 's':
#ifdef RLIMIT_STACK
doit(RLIMIT_STACK);
#endif
break;
case 't':
#ifdef RLIMIT_CPU
doit(RLIMIT_CPU);
#endif
break;
default:
carp("invalid line: ", argv[0]);
}
argv++;
}
}
|
