1 39cff6e7 2004-10-17 devnull #include <u.h>
2 39cff6e7 2004-10-17 devnull #include <sys/time.h>
3 f7b74c17 2004-12-28 devnull #include <sys/stat.h>
4 39cff6e7 2004-10-17 devnull #include <sys/resource.h>
5 39cff6e7 2004-10-17 devnull #include <libc.h>
6 39cff6e7 2004-10-17 devnull #include "rc.h"
7 39cff6e7 2004-10-17 devnull #include "exec.h"
8 39cff6e7 2004-10-17 devnull #include "io.h"
9 39cff6e7 2004-10-17 devnull #include "fns.h"
10 39cff6e7 2004-10-17 devnull #include "getflags.h"
12 39cff6e7 2004-10-17 devnull extern char **mkargv(word*);
13 39cff6e7 2004-10-17 devnull extern int mapfd(int);
15 39cff6e7 2004-10-17 devnull static char *eargs = "cdflmnstuv";
16 39cff6e7 2004-10-17 devnull static int rlx[] = {
17 39cff6e7 2004-10-17 devnull RLIMIT_CORE,
18 39cff6e7 2004-10-17 devnull RLIMIT_DATA,
19 39cff6e7 2004-10-17 devnull RLIMIT_FSIZE,
20 005a85f3 2005-01-07 devnull #ifdef RLIMIT_MEMLOCK
21 39cff6e7 2004-10-17 devnull RLIMIT_MEMLOCK,
25 005a85f3 2005-01-07 devnull #ifdef RLIMIT_RSS
26 39cff6e7 2004-10-17 devnull RLIMIT_RSS,
30 39cff6e7 2004-10-17 devnull RLIMIT_NOFILE,
31 39cff6e7 2004-10-17 devnull RLIMIT_STACK,
32 39cff6e7 2004-10-17 devnull RLIMIT_CPU,
33 005a85f3 2005-01-07 devnull #ifdef RLIMIT_NPROC
34 39cff6e7 2004-10-17 devnull RLIMIT_NPROC,
38 005a85f3 2005-01-07 devnull #ifdef RLIMIT_RSS
39 39cff6e7 2004-10-17 devnull RLIMIT_RSS,
45 39cff6e7 2004-10-17 devnull static void
46 39cff6e7 2004-10-17 devnull eusage(void)
48 39cff6e7 2004-10-17 devnull fprint(mapfd(2), "usage: ulimit [-SHa%s [limit]]\n", eargs);
51 39cff6e7 2004-10-17 devnull #define Notset -4
52 39cff6e7 2004-10-17 devnull #define Unlimited -3
53 39cff6e7 2004-10-17 devnull #define Hard -2
54 39cff6e7 2004-10-17 devnull #define Soft -1
57 39cff6e7 2004-10-17 devnull execulimit(void)
59 39cff6e7 2004-10-17 devnull int fd, n, argc, sethard, setsoft, limit;
60 39cff6e7 2004-10-17 devnull int flag[256];
61 39cff6e7 2004-10-17 devnull char **argv, **oargv, *p;
62 39cff6e7 2004-10-17 devnull char *argv0;
63 39cff6e7 2004-10-17 devnull struct rlimit rl;
65 39cff6e7 2004-10-17 devnull argv0 = nil;
66 39cff6e7 2004-10-17 devnull setstatus("");
67 39cff6e7 2004-10-17 devnull oargv = mkargv(runq->argv->words);
68 39cff6e7 2004-10-17 devnull argv = oargv+1;
69 39cff6e7 2004-10-17 devnull for(argc=0; argv[argc]; argc++)
72 39cff6e7 2004-10-17 devnull memset(flag, 0, sizeof flag);
73 39cff6e7 2004-10-17 devnull ARGBEGIN{
75 39cff6e7 2004-10-17 devnull if(strchr(eargs, ARGC()) == nil){
76 39cff6e7 2004-10-17 devnull eusage();
79 39cff6e7 2004-10-17 devnull case 'S':
80 39cff6e7 2004-10-17 devnull case 'H':
81 39cff6e7 2004-10-17 devnull case 'a':
82 39cff6e7 2004-10-17 devnull flag[ARGC()] = 1;
86 39cff6e7 2004-10-17 devnull if(argc > 1){
87 39cff6e7 2004-10-17 devnull eusage();
88 39cff6e7 2004-10-17 devnull goto out;
91 39cff6e7 2004-10-17 devnull fd = mapfd(1);
93 39cff6e7 2004-10-17 devnull sethard = 1;
94 39cff6e7 2004-10-17 devnull setsoft = 1;
95 39cff6e7 2004-10-17 devnull if(flag['S'] && flag['H'])
97 39cff6e7 2004-10-17 devnull else if(flag['S'])
98 39cff6e7 2004-10-17 devnull sethard = 0;
99 39cff6e7 2004-10-17 devnull else if(flag['H'])
100 39cff6e7 2004-10-17 devnull setsoft = 0;
102 39cff6e7 2004-10-17 devnull limit = Notset;
103 39cff6e7 2004-10-17 devnull if(argc>0){
104 39cff6e7 2004-10-17 devnull if(strcmp(argv[0], "unlimited") == 0)
105 39cff6e7 2004-10-17 devnull limit = Unlimited;
106 39cff6e7 2004-10-17 devnull else if(strcmp(argv[0], "hard") == 0)
107 39cff6e7 2004-10-17 devnull limit = Hard;
108 39cff6e7 2004-10-17 devnull else if(strcmp(argv[0], "soft") == 0)
109 39cff6e7 2004-10-17 devnull limit = Soft;
110 39cff6e7 2004-10-17 devnull else if((limit = strtol(argv[0], &p, 0)) < 0 || *p != 0){
111 39cff6e7 2004-10-17 devnull eusage();
112 39cff6e7 2004-10-17 devnull goto out;
115 39cff6e7 2004-10-17 devnull if(flag['a']){
116 39cff6e7 2004-10-17 devnull for(p=eargs; *p; p++){
117 39cff6e7 2004-10-17 devnull getrlimit(rlx[p-eargs], &rl);
118 39cff6e7 2004-10-17 devnull n = flag['H'] ? rl.rlim_max : rl.rlim_cur;
119 39cff6e7 2004-10-17 devnull if(n == -1)
120 39cff6e7 2004-10-17 devnull fprint(fd, "ulimit -%c unlimited\n", *p);
122 39cff6e7 2004-10-17 devnull fprint(fd, "ulimit -%c %d\n", *p, n);
124 39cff6e7 2004-10-17 devnull goto out;
126 39cff6e7 2004-10-17 devnull for(p=eargs; *p; p++){
127 39cff6e7 2004-10-17 devnull if(flag[(uchar)*p]){
129 39cff6e7 2004-10-17 devnull getrlimit(rlx[p-eargs], &rl);
130 39cff6e7 2004-10-17 devnull switch(limit){
131 39cff6e7 2004-10-17 devnull case Notset:
132 39cff6e7 2004-10-17 devnull n = flag['H'] ? rl.rlim_max : rl.rlim_cur;
133 39cff6e7 2004-10-17 devnull if(n == -1)
134 39cff6e7 2004-10-17 devnull fprint(fd, "ulimit -%c unlimited\n", *p);
136 39cff6e7 2004-10-17 devnull fprint(fd, "ulimit -%c %d\n", *p, n);
138 39cff6e7 2004-10-17 devnull case Hard:
139 39cff6e7 2004-10-17 devnull n = rl.rlim_max;
140 39cff6e7 2004-10-17 devnull goto set;
141 39cff6e7 2004-10-17 devnull case Soft:
142 39cff6e7 2004-10-17 devnull n = rl.rlim_cur;
143 39cff6e7 2004-10-17 devnull goto set;
144 39cff6e7 2004-10-17 devnull case Unlimited:
146 39cff6e7 2004-10-17 devnull goto set;
147 39cff6e7 2004-10-17 devnull default:
148 39cff6e7 2004-10-17 devnull n = limit;
150 39cff6e7 2004-10-17 devnull if(setsoft)
151 39cff6e7 2004-10-17 devnull rl.rlim_cur = n;
152 39cff6e7 2004-10-17 devnull if(sethard)
153 39cff6e7 2004-10-17 devnull rl.rlim_max = n;
154 39cff6e7 2004-10-17 devnull if(setrlimit(rlx[p-eargs], &rl) < 0)
155 39cff6e7 2004-10-17 devnull fprint(mapfd(2), "setrlimit: %r\n");
161 39cff6e7 2004-10-17 devnull free(oargv);
162 39cff6e7 2004-10-17 devnull poplist();
163 39cff6e7 2004-10-17 devnull flush(err);
167 39cff6e7 2004-10-17 devnull execumask(void)
169 39cff6e7 2004-10-17 devnull int n, argc;
170 39cff6e7 2004-10-17 devnull char **argv, **oargv, *p;
171 39cff6e7 2004-10-17 devnull char *argv0;
173 39cff6e7 2004-10-17 devnull argv0 = nil;
174 39cff6e7 2004-10-17 devnull setstatus("");
175 39cff6e7 2004-10-17 devnull oargv = mkargv(runq->argv->words);
176 39cff6e7 2004-10-17 devnull argv = oargv+1;
177 39cff6e7 2004-10-17 devnull for(argc=0; argv[argc]; argc++)
180 39cff6e7 2004-10-17 devnull ARGBEGIN{
181 39cff6e7 2004-10-17 devnull default:
183 39cff6e7 2004-10-17 devnull fprint(mapfd(2), "usage: umask [mode]\n");
184 39cff6e7 2004-10-17 devnull goto out;
187 39cff6e7 2004-10-17 devnull if(argc > 1)
188 39cff6e7 2004-10-17 devnull goto usage;
190 39cff6e7 2004-10-17 devnull if(argc == 1){
191 39cff6e7 2004-10-17 devnull n = strtol(argv[0], &p, 8);
192 39cff6e7 2004-10-17 devnull if(*p != 0 || p == argv[0])
193 39cff6e7 2004-10-17 devnull goto usage;
194 f7b74c17 2004-12-28 devnull umask(n);
195 39cff6e7 2004-10-17 devnull goto out;
198 39cff6e7 2004-10-17 devnull n = umask(0);
199 39cff6e7 2004-10-17 devnull umask(n);
200 39cff6e7 2004-10-17 devnull if(n < 0){
201 39cff6e7 2004-10-17 devnull fprint(mapfd(2), "umask: %r\n");
202 39cff6e7 2004-10-17 devnull goto out;
205 39cff6e7 2004-10-17 devnull fprint(mapfd(1), "umask %03o\n", n);
208 39cff6e7 2004-10-17 devnull free(oargv);
209 39cff6e7 2004-10-17 devnull poplist();
210 39cff6e7 2004-10-17 devnull flush(err);