kern_resource.c revision 1.74 1 1.74 atatat /* $NetBSD: kern_resource.c,v 1.74 2003/12/04 19:38:23 atatat Exp $ */
2 1.20 cgd
3 1.17 cgd /*-
4 1.19 cgd * Copyright (c) 1982, 1986, 1991, 1993
5 1.19 cgd * The Regents of the University of California. All rights reserved.
6 1.17 cgd * (c) UNIX System Laboratories, Inc.
7 1.17 cgd * All or some portions of this file are derived from material licensed
8 1.17 cgd * to the University of California by American Telephone and Telegraph
9 1.17 cgd * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10 1.17 cgd * the permission of UNIX System Laboratories, Inc.
11 1.17 cgd *
12 1.17 cgd * Redistribution and use in source and binary forms, with or without
13 1.17 cgd * modification, are permitted provided that the following conditions
14 1.17 cgd * are met:
15 1.17 cgd * 1. Redistributions of source code must retain the above copyright
16 1.17 cgd * notice, this list of conditions and the following disclaimer.
17 1.17 cgd * 2. Redistributions in binary form must reproduce the above copyright
18 1.17 cgd * notice, this list of conditions and the following disclaimer in the
19 1.17 cgd * documentation and/or other materials provided with the distribution.
20 1.72 agc * 3. Neither the name of the University nor the names of its contributors
21 1.17 cgd * may be used to endorse or promote products derived from this software
22 1.17 cgd * without specific prior written permission.
23 1.17 cgd *
24 1.17 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 1.17 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 1.17 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 1.17 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 1.17 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 1.17 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 1.17 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 1.17 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 1.17 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 1.17 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 1.17 cgd * SUCH DAMAGE.
35 1.17 cgd *
36 1.45 fvdl * @(#)kern_resource.c 8.8 (Berkeley) 2/14/95
37 1.17 cgd */
38 1.61 lukem
39 1.61 lukem #include <sys/cdefs.h>
40 1.74 atatat __KERNEL_RCSID(0, "$NetBSD: kern_resource.c,v 1.74 2003/12/04 19:38:23 atatat Exp $");
41 1.44 mrg
42 1.17 cgd #include <sys/param.h>
43 1.22 cgd #include <sys/systm.h>
44 1.17 cgd #include <sys/kernel.h>
45 1.19 cgd #include <sys/file.h>
46 1.17 cgd #include <sys/resourcevar.h>
47 1.17 cgd #include <sys/malloc.h>
48 1.49 thorpej #include <sys/pool.h>
49 1.17 cgd #include <sys/proc.h>
50 1.74 atatat #include <sys/sysctl.h>
51 1.17 cgd
52 1.22 cgd #include <sys/mount.h>
53 1.68 thorpej #include <sys/sa.h>
54 1.22 cgd #include <sys/syscallargs.h>
55 1.17 cgd
56 1.43 mrg #include <uvm/uvm_extern.h>
57 1.43 mrg
58 1.17 cgd /*
59 1.60 eeh * Maximum process data and stack limits.
60 1.60 eeh * They are variables so they are patchable.
61 1.60 eeh *
62 1.60 eeh * XXXX Do we really need them to be patchable?
63 1.60 eeh */
64 1.60 eeh rlim_t maxdmap = MAXDSIZ;
65 1.60 eeh rlim_t maxsmap = MAXSSIZ;
66 1.60 eeh
67 1.60 eeh /*
68 1.17 cgd * Resource controls and accounting.
69 1.17 cgd */
70 1.17 cgd
71 1.25 cgd int
72 1.68 thorpej sys_getpriority(l, v, retval)
73 1.68 thorpej struct lwp *l;
74 1.30 thorpej void *v;
75 1.30 thorpej register_t *retval;
76 1.30 thorpej {
77 1.54 augustss struct sys_getpriority_args /* {
78 1.22 cgd syscallarg(int) which;
79 1.22 cgd syscallarg(int) who;
80 1.30 thorpej } */ *uap = v;
81 1.68 thorpej struct proc *curp = l->l_proc, *p;
82 1.54 augustss int low = NZERO + PRIO_MAX + 1;
83 1.17 cgd
84 1.22 cgd switch (SCARG(uap, which)) {
85 1.17 cgd
86 1.17 cgd case PRIO_PROCESS:
87 1.22 cgd if (SCARG(uap, who) == 0)
88 1.17 cgd p = curp;
89 1.17 cgd else
90 1.22 cgd p = pfind(SCARG(uap, who));
91 1.17 cgd if (p == 0)
92 1.17 cgd break;
93 1.17 cgd low = p->p_nice;
94 1.17 cgd break;
95 1.17 cgd
96 1.17 cgd case PRIO_PGRP: {
97 1.54 augustss struct pgrp *pg;
98 1.17 cgd
99 1.22 cgd if (SCARG(uap, who) == 0)
100 1.17 cgd pg = curp->p_pgrp;
101 1.22 cgd else if ((pg = pgfind(SCARG(uap, who))) == NULL)
102 1.17 cgd break;
103 1.64 matt LIST_FOREACH(p, &pg->pg_members, p_pglist) {
104 1.17 cgd if (p->p_nice < low)
105 1.17 cgd low = p->p_nice;
106 1.17 cgd }
107 1.17 cgd break;
108 1.17 cgd }
109 1.17 cgd
110 1.17 cgd case PRIO_USER:
111 1.22 cgd if (SCARG(uap, who) == 0)
112 1.22 cgd SCARG(uap, who) = curp->p_ucred->cr_uid;
113 1.52 thorpej proclist_lock_read();
114 1.64 matt LIST_FOREACH(p, &allproc, p_list) {
115 1.63 thorpej if (p->p_ucred->cr_uid == (uid_t) SCARG(uap, who) &&
116 1.22 cgd p->p_nice < low)
117 1.17 cgd low = p->p_nice;
118 1.64 matt }
119 1.51 thorpej proclist_unlock_read();
120 1.17 cgd break;
121 1.17 cgd
122 1.17 cgd default:
123 1.17 cgd return (EINVAL);
124 1.17 cgd }
125 1.37 ws if (low == NZERO + PRIO_MAX + 1)
126 1.17 cgd return (ESRCH);
127 1.37 ws *retval = low - NZERO;
128 1.17 cgd return (0);
129 1.17 cgd }
130 1.17 cgd
131 1.17 cgd /* ARGSUSED */
132 1.25 cgd int
133 1.68 thorpej sys_setpriority(l, v, retval)
134 1.68 thorpej struct lwp *l;
135 1.30 thorpej void *v;
136 1.30 thorpej register_t *retval;
137 1.30 thorpej {
138 1.54 augustss struct sys_setpriority_args /* {
139 1.22 cgd syscallarg(int) which;
140 1.22 cgd syscallarg(int) who;
141 1.22 cgd syscallarg(int) prio;
142 1.30 thorpej } */ *uap = v;
143 1.68 thorpej struct proc *curp = l->l_proc, *p;
144 1.17 cgd int found = 0, error = 0;
145 1.17 cgd
146 1.22 cgd switch (SCARG(uap, which)) {
147 1.17 cgd
148 1.17 cgd case PRIO_PROCESS:
149 1.22 cgd if (SCARG(uap, who) == 0)
150 1.17 cgd p = curp;
151 1.17 cgd else
152 1.22 cgd p = pfind(SCARG(uap, who));
153 1.17 cgd if (p == 0)
154 1.17 cgd break;
155 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
156 1.17 cgd found++;
157 1.17 cgd break;
158 1.17 cgd
159 1.17 cgd case PRIO_PGRP: {
160 1.54 augustss struct pgrp *pg;
161 1.17 cgd
162 1.22 cgd if (SCARG(uap, who) == 0)
163 1.17 cgd pg = curp->p_pgrp;
164 1.22 cgd else if ((pg = pgfind(SCARG(uap, who))) == NULL)
165 1.17 cgd break;
166 1.64 matt LIST_FOREACH(p, &pg->pg_members, p_pglist) {
167 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
168 1.17 cgd found++;
169 1.17 cgd }
170 1.17 cgd break;
171 1.17 cgd }
172 1.17 cgd
173 1.17 cgd case PRIO_USER:
174 1.22 cgd if (SCARG(uap, who) == 0)
175 1.22 cgd SCARG(uap, who) = curp->p_ucred->cr_uid;
176 1.52 thorpej proclist_lock_read();
177 1.64 matt LIST_FOREACH(p, &allproc, p_list) {
178 1.63 thorpej if (p->p_ucred->cr_uid == (uid_t) SCARG(uap, who)) {
179 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
180 1.17 cgd found++;
181 1.17 cgd }
182 1.64 matt }
183 1.51 thorpej proclist_unlock_read();
184 1.17 cgd break;
185 1.17 cgd
186 1.17 cgd default:
187 1.17 cgd return (EINVAL);
188 1.17 cgd }
189 1.17 cgd if (found == 0)
190 1.17 cgd return (ESRCH);
191 1.17 cgd return (error);
192 1.17 cgd }
193 1.17 cgd
194 1.25 cgd int
195 1.17 cgd donice(curp, chgp, n)
196 1.54 augustss struct proc *curp, *chgp;
197 1.54 augustss int n;
198 1.17 cgd {
199 1.54 augustss struct pcred *pcred = curp->p_cred;
200 1.59 thorpej int s;
201 1.17 cgd
202 1.17 cgd if (pcred->pc_ucred->cr_uid && pcred->p_ruid &&
203 1.17 cgd pcred->pc_ucred->cr_uid != chgp->p_ucred->cr_uid &&
204 1.17 cgd pcred->p_ruid != chgp->p_ucred->cr_uid)
205 1.17 cgd return (EPERM);
206 1.17 cgd if (n > PRIO_MAX)
207 1.17 cgd n = PRIO_MAX;
208 1.17 cgd if (n < PRIO_MIN)
209 1.17 cgd n = PRIO_MIN;
210 1.37 ws n += NZERO;
211 1.17 cgd if (n < chgp->p_nice && suser(pcred->pc_ucred, &curp->p_acflag))
212 1.17 cgd return (EACCES);
213 1.17 cgd chgp->p_nice = n;
214 1.59 thorpej SCHED_LOCK(s);
215 1.68 thorpej (void)resetprocpriority(chgp);
216 1.59 thorpej SCHED_UNLOCK(s);
217 1.17 cgd return (0);
218 1.17 cgd }
219 1.17 cgd
220 1.17 cgd /* ARGSUSED */
221 1.25 cgd int
222 1.68 thorpej sys_setrlimit(l, v, retval)
223 1.68 thorpej struct lwp *l;
224 1.30 thorpej void *v;
225 1.30 thorpej register_t *retval;
226 1.30 thorpej {
227 1.54 augustss struct sys_setrlimit_args /* {
228 1.42 mycroft syscallarg(int) which;
229 1.39 cgd syscallarg(const struct rlimit *) rlp;
230 1.30 thorpej } */ *uap = v;
231 1.68 thorpej struct proc *p = l->l_proc;
232 1.42 mycroft int which = SCARG(uap, which);
233 1.19 cgd struct rlimit alim;
234 1.17 cgd int error;
235 1.17 cgd
236 1.46 perry error = copyin(SCARG(uap, rlp), &alim, sizeof(struct rlimit));
237 1.33 christos if (error)
238 1.17 cgd return (error);
239 1.53 bouyer return (dosetrlimit(p, p->p_cred, which, &alim));
240 1.17 cgd }
241 1.17 cgd
242 1.17 cgd int
243 1.53 bouyer dosetrlimit(p, cred, which, limp)
244 1.17 cgd struct proc *p;
245 1.53 bouyer struct pcred *cred;
246 1.42 mycroft int which;
247 1.17 cgd struct rlimit *limp;
248 1.17 cgd {
249 1.54 augustss struct rlimit *alimp;
250 1.53 bouyer struct plimit *newplim;
251 1.17 cgd int error;
252 1.17 cgd
253 1.67 itojun if ((u_int)which >= RLIM_NLIMITS)
254 1.17 cgd return (EINVAL);
255 1.38 matthias
256 1.38 matthias if (limp->rlim_cur < 0 || limp->rlim_max < 0)
257 1.38 matthias return (EINVAL);
258 1.38 matthias
259 1.17 cgd alimp = &p->p_rlimit[which];
260 1.53 bouyer /* if we don't change the value, no need to limcopy() */
261 1.53 bouyer if (limp->rlim_cur == alimp->rlim_cur &&
262 1.53 bouyer limp->rlim_max == alimp->rlim_max)
263 1.53 bouyer return 0;
264 1.53 bouyer
265 1.62 jdolecek if (limp->rlim_cur > limp->rlim_max) {
266 1.62 jdolecek /*
267 1.62 jdolecek * This is programming error. According to SUSv2, we should
268 1.62 jdolecek * return error in this case.
269 1.62 jdolecek */
270 1.62 jdolecek return (EINVAL);
271 1.62 jdolecek }
272 1.62 jdolecek if (limp->rlim_max > alimp->rlim_max
273 1.62 jdolecek && (error = suser(cred->pc_ucred, &p->p_acflag)) != 0)
274 1.17 cgd return (error);
275 1.62 jdolecek
276 1.17 cgd if (p->p_limit->p_refcnt > 1 &&
277 1.17 cgd (p->p_limit->p_lflags & PL_SHAREMOD) == 0) {
278 1.53 bouyer newplim = limcopy(p->p_limit);
279 1.53 bouyer limfree(p->p_limit);
280 1.53 bouyer p->p_limit = newplim;
281 1.17 cgd alimp = &p->p_rlimit[which];
282 1.17 cgd }
283 1.17 cgd
284 1.17 cgd switch (which) {
285 1.17 cgd
286 1.17 cgd case RLIMIT_DATA:
287 1.19 cgd if (limp->rlim_cur > maxdmap)
288 1.19 cgd limp->rlim_cur = maxdmap;
289 1.19 cgd if (limp->rlim_max > maxdmap)
290 1.19 cgd limp->rlim_max = maxdmap;
291 1.17 cgd break;
292 1.17 cgd
293 1.17 cgd case RLIMIT_STACK:
294 1.19 cgd if (limp->rlim_cur > maxsmap)
295 1.19 cgd limp->rlim_cur = maxsmap;
296 1.19 cgd if (limp->rlim_max > maxsmap)
297 1.19 cgd limp->rlim_max = maxsmap;
298 1.62 jdolecek
299 1.62 jdolecek /*
300 1.62 jdolecek * Return EINVAL if the new stack size limit is lower than
301 1.62 jdolecek * current usage. Otherwise, the process would get SIGSEGV the
302 1.62 jdolecek * moment it would try to access anything on it's current stack.
303 1.62 jdolecek * This conforms to SUSv2.
304 1.62 jdolecek */
305 1.62 jdolecek if (limp->rlim_cur < p->p_vmspace->vm_ssize * PAGE_SIZE
306 1.62 jdolecek || limp->rlim_max < p->p_vmspace->vm_ssize * PAGE_SIZE)
307 1.62 jdolecek return (EINVAL);
308 1.40 enami
309 1.17 cgd /*
310 1.40 enami * Stack is allocated to the max at exec time with
311 1.40 enami * only "rlim_cur" bytes accessible (In other words,
312 1.40 enami * allocates stack dividing two contiguous regions at
313 1.40 enami * "rlim_cur" bytes boundary).
314 1.40 enami *
315 1.40 enami * Since allocation is done in terms of page, roundup
316 1.40 enami * "rlim_cur" (otherwise, contiguous regions
317 1.40 enami * overlap). If stack limit is going up make more
318 1.40 enami * accessible, if going down make inaccessible.
319 1.17 cgd */
320 1.40 enami limp->rlim_cur = round_page(limp->rlim_cur);
321 1.17 cgd if (limp->rlim_cur != alimp->rlim_cur) {
322 1.48 eeh vaddr_t addr;
323 1.48 eeh vsize_t size;
324 1.17 cgd vm_prot_t prot;
325 1.17 cgd
326 1.17 cgd if (limp->rlim_cur > alimp->rlim_cur) {
327 1.73 chs prot = VM_PROT_READ | VM_PROT_WRITE;
328 1.17 cgd size = limp->rlim_cur - alimp->rlim_cur;
329 1.17 cgd addr = USRSTACK - limp->rlim_cur;
330 1.17 cgd } else {
331 1.17 cgd prot = VM_PROT_NONE;
332 1.17 cgd size = alimp->rlim_cur - limp->rlim_cur;
333 1.17 cgd addr = USRSTACK - alimp->rlim_cur;
334 1.17 cgd }
335 1.43 mrg (void) uvm_map_protect(&p->p_vmspace->vm_map,
336 1.43 mrg addr, addr+size, prot, FALSE);
337 1.17 cgd }
338 1.17 cgd break;
339 1.19 cgd
340 1.19 cgd case RLIMIT_NOFILE:
341 1.19 cgd if (limp->rlim_cur > maxfiles)
342 1.19 cgd limp->rlim_cur = maxfiles;
343 1.19 cgd if (limp->rlim_max > maxfiles)
344 1.19 cgd limp->rlim_max = maxfiles;
345 1.19 cgd break;
346 1.19 cgd
347 1.19 cgd case RLIMIT_NPROC:
348 1.19 cgd if (limp->rlim_cur > maxproc)
349 1.19 cgd limp->rlim_cur = maxproc;
350 1.19 cgd if (limp->rlim_max > maxproc)
351 1.19 cgd limp->rlim_max = maxproc;
352 1.19 cgd break;
353 1.17 cgd }
354 1.17 cgd *alimp = *limp;
355 1.17 cgd return (0);
356 1.17 cgd }
357 1.17 cgd
358 1.17 cgd /* ARGSUSED */
359 1.25 cgd int
360 1.68 thorpej sys_getrlimit(l, v, retval)
361 1.68 thorpej struct lwp *l;
362 1.30 thorpej void *v;
363 1.30 thorpej register_t *retval;
364 1.30 thorpej {
365 1.54 augustss struct sys_getrlimit_args /* {
366 1.42 mycroft syscallarg(int) which;
367 1.22 cgd syscallarg(struct rlimit *) rlp;
368 1.30 thorpej } */ *uap = v;
369 1.68 thorpej struct proc *p = l->l_proc;
370 1.42 mycroft int which = SCARG(uap, which);
371 1.17 cgd
372 1.67 itojun if ((u_int)which >= RLIM_NLIMITS)
373 1.17 cgd return (EINVAL);
374 1.42 mycroft return (copyout(&p->p_rlimit[which], SCARG(uap, rlp),
375 1.46 perry sizeof(struct rlimit)));
376 1.17 cgd }
377 1.17 cgd
378 1.17 cgd /*
379 1.17 cgd * Transform the running time and tick information in proc p into user,
380 1.17 cgd * system, and interrupt time usage.
381 1.17 cgd */
382 1.25 cgd void
383 1.17 cgd calcru(p, up, sp, ip)
384 1.54 augustss struct proc *p;
385 1.54 augustss struct timeval *up;
386 1.54 augustss struct timeval *sp;
387 1.54 augustss struct timeval *ip;
388 1.17 cgd {
389 1.54 augustss u_quad_t u, st, ut, it, tot;
390 1.70 dsl unsigned long sec;
391 1.70 dsl long usec;
392 1.54 augustss int s;
393 1.17 cgd struct timeval tv;
394 1.68 thorpej struct lwp *l;
395 1.17 cgd
396 1.17 cgd s = splstatclock();
397 1.17 cgd st = p->p_sticks;
398 1.17 cgd ut = p->p_uticks;
399 1.17 cgd it = p->p_iticks;
400 1.17 cgd splx(s);
401 1.17 cgd
402 1.17 cgd sec = p->p_rtime.tv_sec;
403 1.17 cgd usec = p->p_rtime.tv_usec;
404 1.70 dsl LIST_FOREACH(l, &p->p_lwps, l_sibling) {
405 1.68 thorpej if (l->l_stat == LSONPROC) {
406 1.68 thorpej struct schedstate_percpu *spc;
407 1.68 thorpej
408 1.68 thorpej KDASSERT(l->l_cpu != NULL);
409 1.68 thorpej spc = &l->l_cpu->ci_schedstate;
410 1.68 thorpej
411 1.68 thorpej /*
412 1.68 thorpej * Adjust for the current time slice. This is
413 1.68 thorpej * actually fairly important since the error
414 1.68 thorpej * here is on the order of a time quantum,
415 1.68 thorpej * which is much greater than the sampling
416 1.68 thorpej * error.
417 1.68 thorpej */
418 1.68 thorpej microtime(&tv);
419 1.68 thorpej sec += tv.tv_sec - spc->spc_runtime.tv_sec;
420 1.68 thorpej usec += tv.tv_usec - spc->spc_runtime.tv_usec;
421 1.68 thorpej }
422 1.17 cgd }
423 1.69 dsl
424 1.69 dsl tot = st + ut + it;
425 1.70 dsl u = sec * 1000000ull + usec;
426 1.70 dsl
427 1.69 dsl if (tot == 0) {
428 1.69 dsl /* No ticks, so can't use to share time out, split 50-50 */
429 1.70 dsl st = ut = u / 2;
430 1.70 dsl } else {
431 1.70 dsl st = (u * st) / tot;
432 1.70 dsl ut = (u * ut) / tot;
433 1.69 dsl }
434 1.17 cgd sp->tv_sec = st / 1000000;
435 1.17 cgd sp->tv_usec = st % 1000000;
436 1.17 cgd up->tv_sec = ut / 1000000;
437 1.17 cgd up->tv_usec = ut % 1000000;
438 1.17 cgd if (ip != NULL) {
439 1.70 dsl if (it != 0)
440 1.70 dsl it = (u * it) / tot;
441 1.17 cgd ip->tv_sec = it / 1000000;
442 1.17 cgd ip->tv_usec = it % 1000000;
443 1.17 cgd }
444 1.17 cgd }
445 1.17 cgd
446 1.17 cgd /* ARGSUSED */
447 1.25 cgd int
448 1.68 thorpej sys_getrusage(l, v, retval)
449 1.68 thorpej struct lwp *l;
450 1.30 thorpej void *v;
451 1.30 thorpej register_t *retval;
452 1.30 thorpej {
453 1.54 augustss struct sys_getrusage_args /* {
454 1.22 cgd syscallarg(int) who;
455 1.22 cgd syscallarg(struct rusage *) rusage;
456 1.30 thorpej } */ *uap = v;
457 1.54 augustss struct rusage *rup;
458 1.68 thorpej struct proc *p = l->l_proc;
459 1.17 cgd
460 1.22 cgd switch (SCARG(uap, who)) {
461 1.17 cgd
462 1.19 cgd case RUSAGE_SELF:
463 1.17 cgd rup = &p->p_stats->p_ru;
464 1.17 cgd calcru(p, &rup->ru_utime, &rup->ru_stime, NULL);
465 1.17 cgd break;
466 1.17 cgd
467 1.17 cgd case RUSAGE_CHILDREN:
468 1.17 cgd rup = &p->p_stats->p_cru;
469 1.17 cgd break;
470 1.17 cgd
471 1.17 cgd default:
472 1.17 cgd return (EINVAL);
473 1.17 cgd }
474 1.46 perry return (copyout(rup, SCARG(uap, rusage), sizeof(struct rusage)));
475 1.17 cgd }
476 1.17 cgd
477 1.25 cgd void
478 1.17 cgd ruadd(ru, ru2)
479 1.54 augustss struct rusage *ru, *ru2;
480 1.17 cgd {
481 1.54 augustss long *ip, *ip2;
482 1.54 augustss int i;
483 1.17 cgd
484 1.27 mycroft timeradd(&ru->ru_utime, &ru2->ru_utime, &ru->ru_utime);
485 1.27 mycroft timeradd(&ru->ru_stime, &ru2->ru_stime, &ru->ru_stime);
486 1.17 cgd if (ru->ru_maxrss < ru2->ru_maxrss)
487 1.17 cgd ru->ru_maxrss = ru2->ru_maxrss;
488 1.17 cgd ip = &ru->ru_first; ip2 = &ru2->ru_first;
489 1.17 cgd for (i = &ru->ru_last - &ru->ru_first; i >= 0; i--)
490 1.17 cgd *ip++ += *ip2++;
491 1.17 cgd }
492 1.17 cgd
493 1.17 cgd /*
494 1.17 cgd * Make a copy of the plimit structure.
495 1.17 cgd * We share these structures copy-on-write after fork,
496 1.17 cgd * and copy when a limit is changed.
497 1.17 cgd */
498 1.17 cgd struct plimit *
499 1.17 cgd limcopy(lim)
500 1.17 cgd struct plimit *lim;
501 1.17 cgd {
502 1.54 augustss struct plimit *newlim;
503 1.71 itojun size_t l;
504 1.17 cgd
505 1.49 thorpej newlim = pool_get(&plimit_pool, PR_WAITOK);
506 1.47 perry memcpy(newlim->pl_rlimit, lim->pl_rlimit,
507 1.17 cgd sizeof(struct rlimit) * RLIM_NLIMITS);
508 1.53 bouyer if (lim->pl_corename == defcorename) {
509 1.53 bouyer newlim->pl_corename = defcorename;
510 1.53 bouyer } else {
511 1.71 itojun l = strlen(lim->pl_corename) + 1;
512 1.71 itojun newlim->pl_corename = malloc(l, M_TEMP, M_WAITOK);
513 1.71 itojun strlcpy(newlim->pl_corename, lim->pl_corename, l);
514 1.53 bouyer }
515 1.32 mycroft newlim->p_lflags = 0;
516 1.32 mycroft newlim->p_refcnt = 1;
517 1.32 mycroft return (newlim);
518 1.32 mycroft }
519 1.32 mycroft
520 1.32 mycroft void
521 1.32 mycroft limfree(lim)
522 1.32 mycroft struct plimit *lim;
523 1.32 mycroft {
524 1.32 mycroft
525 1.32 mycroft if (--lim->p_refcnt > 0)
526 1.32 mycroft return;
527 1.53 bouyer #ifdef DIAGNOSTIC
528 1.53 bouyer if (lim->p_refcnt < 0)
529 1.53 bouyer panic("limfree");
530 1.53 bouyer #endif
531 1.53 bouyer if (lim->pl_corename != defcorename)
532 1.53 bouyer free(lim->pl_corename, M_TEMP);
533 1.49 thorpej pool_put(&plimit_pool, lim);
534 1.68 thorpej }
535 1.68 thorpej
536 1.68 thorpej struct pstats *
537 1.68 thorpej pstatscopy(ps)
538 1.68 thorpej struct pstats *ps;
539 1.68 thorpej {
540 1.68 thorpej
541 1.68 thorpej struct pstats *newps;
542 1.68 thorpej
543 1.68 thorpej newps = pool_get(&pstats_pool, PR_WAITOK);
544 1.68 thorpej
545 1.68 thorpej memset(&newps->pstat_startzero, 0,
546 1.68 thorpej (unsigned) ((caddr_t)&newps->pstat_endzero -
547 1.68 thorpej (caddr_t)&newps->pstat_startzero));
548 1.68 thorpej memcpy(&newps->pstat_startcopy, &ps->pstat_startcopy,
549 1.68 thorpej ((caddr_t)&newps->pstat_endcopy -
550 1.68 thorpej (caddr_t)&newps->pstat_startcopy));
551 1.68 thorpej
552 1.68 thorpej return (newps);
553 1.68 thorpej
554 1.68 thorpej }
555 1.68 thorpej
556 1.68 thorpej void
557 1.68 thorpej pstatsfree(ps)
558 1.68 thorpej struct pstats *ps;
559 1.68 thorpej {
560 1.68 thorpej
561 1.68 thorpej pool_put(&pstats_pool, ps);
562 1.74 atatat }
563 1.74 atatat
564 1.74 atatat /*
565 1.74 atatat * sysctl interface in five parts
566 1.74 atatat */
567 1.74 atatat
568 1.74 atatat /*
569 1.74 atatat * a routine for sysctl proc subtree helpers that need to pick a valid
570 1.74 atatat * process by pid.
571 1.74 atatat */
572 1.74 atatat static int
573 1.74 atatat sysctl_proc_findproc(struct proc *p, struct proc **p2, pid_t pid)
574 1.74 atatat {
575 1.74 atatat struct proc *ptmp;
576 1.74 atatat int i, error = 0;
577 1.74 atatat
578 1.74 atatat if (pid == PROC_CURPROC)
579 1.74 atatat ptmp = p;
580 1.74 atatat else if ((ptmp = pfind(pid)) == NULL)
581 1.74 atatat error = ESRCH;
582 1.74 atatat else {
583 1.74 atatat /*
584 1.74 atatat * suid proc of ours or proc not ours
585 1.74 atatat */
586 1.74 atatat if (p->p_cred->p_ruid != ptmp->p_cred->p_ruid ||
587 1.74 atatat p->p_cred->p_ruid != ptmp->p_cred->p_svuid)
588 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
589 1.74 atatat
590 1.74 atatat /*
591 1.74 atatat * sgid proc has sgid back to us temporarily
592 1.74 atatat */
593 1.74 atatat else if (ptmp->p_cred->p_rgid != ptmp->p_cred->p_svgid)
594 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
595 1.74 atatat
596 1.74 atatat /*
597 1.74 atatat * our rgid must be in target's group list (ie,
598 1.74 atatat * sub-processes started by a sgid process)
599 1.74 atatat */
600 1.74 atatat else {
601 1.74 atatat for (i = 0; i < p->p_ucred->cr_ngroups; i++) {
602 1.74 atatat if (p->p_ucred->cr_groups[i] ==
603 1.74 atatat ptmp->p_cred->p_rgid)
604 1.74 atatat break;
605 1.74 atatat }
606 1.74 atatat if (i == p->p_ucred->cr_ngroups)
607 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
608 1.74 atatat }
609 1.74 atatat }
610 1.74 atatat
611 1.74 atatat *p2 = ptmp;
612 1.74 atatat return (error);
613 1.74 atatat }
614 1.74 atatat
615 1.74 atatat /*
616 1.74 atatat * sysctl helper routine for setting a process's specific corefile
617 1.74 atatat * name. picks the process based on the given pid and checks the
618 1.74 atatat * correctness of the new value.
619 1.74 atatat */
620 1.74 atatat static int
621 1.74 atatat sysctl_proc_corename(SYSCTLFN_ARGS)
622 1.74 atatat {
623 1.74 atatat struct proc *ptmp, *p;
624 1.74 atatat struct plimit *newplim;
625 1.74 atatat int error = 0, len;
626 1.74 atatat char cname[MAXPATHLEN], *tmp;
627 1.74 atatat struct sysctlnode node;
628 1.74 atatat
629 1.74 atatat /*
630 1.74 atatat * is this all correct?
631 1.74 atatat */
632 1.74 atatat if (namelen != 0)
633 1.74 atatat return (EINVAL);
634 1.74 atatat if (name[-1] != PROC_PID_CORENAME)
635 1.74 atatat return (EINVAL);
636 1.74 atatat
637 1.74 atatat /*
638 1.74 atatat * whom are we tweaking?
639 1.74 atatat */
640 1.74 atatat p = l->l_proc;
641 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-2]);
642 1.74 atatat if (error)
643 1.74 atatat return (error);
644 1.74 atatat
645 1.74 atatat /*
646 1.74 atatat * let them modify a temporary copy of the core name
647 1.74 atatat */
648 1.74 atatat node = *rnode;
649 1.74 atatat strlcpy(cname, ptmp->p_limit->pl_corename, sizeof(cname));
650 1.74 atatat node.sysctl_data = cname;
651 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
652 1.74 atatat
653 1.74 atatat /*
654 1.74 atatat * if that failed, or they have nothing new to say, or we've
655 1.74 atatat * heard it before...
656 1.74 atatat */
657 1.74 atatat if (error || newp == NULL ||
658 1.74 atatat strcmp(cname, ptmp->p_limit->pl_corename) == 0)
659 1.74 atatat return (error);
660 1.74 atatat
661 1.74 atatat /*
662 1.74 atatat * no error yet and cname now has the new core name in it.
663 1.74 atatat * let's see if it looks acceptable. it must be either "core"
664 1.74 atatat * or end in ".core" or "/core".
665 1.74 atatat */
666 1.74 atatat len = strlen(cname);
667 1.74 atatat if (len < 4)
668 1.74 atatat return (EINVAL);
669 1.74 atatat if (strcmp(cname + len - 4, "core") != 0)
670 1.74 atatat return (EINVAL);
671 1.74 atatat if (len > 4 && cname[len - 5] != '/' && cname[len - 5] != '.')
672 1.74 atatat return (EINVAL);
673 1.74 atatat
674 1.74 atatat /*
675 1.74 atatat * hmm...looks good. now...where do we put it?
676 1.74 atatat */
677 1.74 atatat tmp = malloc(len + 1, M_TEMP, M_WAITOK|M_CANFAIL);
678 1.74 atatat if (tmp == NULL)
679 1.74 atatat return (ENOMEM);
680 1.74 atatat strlcpy(tmp, cname, len + 1);
681 1.74 atatat
682 1.74 atatat if (ptmp->p_limit->p_refcnt > 1 &&
683 1.74 atatat (ptmp->p_limit->p_lflags & PL_SHAREMOD) == 0) {
684 1.74 atatat newplim = limcopy(ptmp->p_limit);
685 1.74 atatat limfree(ptmp->p_limit);
686 1.74 atatat ptmp->p_limit = newplim;
687 1.74 atatat }
688 1.74 atatat if (ptmp->p_limit->pl_corename != defcorename)
689 1.74 atatat FREE(ptmp->p_limit->pl_corename, M_SYSCTLDATA);
690 1.74 atatat ptmp->p_limit->pl_corename = tmp;
691 1.74 atatat
692 1.74 atatat return (error);
693 1.74 atatat }
694 1.74 atatat
695 1.74 atatat /*
696 1.74 atatat * sysctl helper routine for checking/setting a process's stop flags,
697 1.74 atatat * one for fork and one for exec.
698 1.74 atatat */
699 1.74 atatat static int
700 1.74 atatat sysctl_proc_stop(SYSCTLFN_ARGS)
701 1.74 atatat {
702 1.74 atatat struct proc *p, *ptmp;
703 1.74 atatat int i, f, error = 0;
704 1.74 atatat struct sysctlnode node;
705 1.74 atatat
706 1.74 atatat if (namelen != 0)
707 1.74 atatat return (EINVAL);
708 1.74 atatat
709 1.74 atatat p = l->l_proc;
710 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-2]);
711 1.74 atatat if (error)
712 1.74 atatat return (error);
713 1.74 atatat
714 1.74 atatat switch (rnode->sysctl_num) {
715 1.74 atatat case PROC_PID_STOPFORK:
716 1.74 atatat f = P_STOPFORK;
717 1.74 atatat break;
718 1.74 atatat case PROC_PID_STOPEXEC:
719 1.74 atatat f = P_STOPEXEC;
720 1.74 atatat break;
721 1.74 atatat #ifdef PROC_PID_STOPEXIT
722 1.74 atatat case PROC_PID_STOPEXIT:
723 1.74 atatat f = P_STOPEXIT;
724 1.74 atatat break;
725 1.74 atatat #endif /* PROC_PID_STOPEXIT */
726 1.74 atatat default:
727 1.74 atatat return (EINVAL);
728 1.74 atatat }
729 1.74 atatat
730 1.74 atatat i = (ptmp->p_flag & f) ? 1 : 0;
731 1.74 atatat node = *rnode;
732 1.74 atatat node.sysctl_data = &i;
733 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
734 1.74 atatat if (error || newp == NULL)
735 1.74 atatat return (error);
736 1.74 atatat
737 1.74 atatat if (i)
738 1.74 atatat ptmp->p_flag |= f;
739 1.74 atatat else
740 1.74 atatat ptmp->p_flag &= ~f;
741 1.74 atatat
742 1.74 atatat return (0);
743 1.74 atatat }
744 1.74 atatat
745 1.74 atatat /*
746 1.74 atatat * sysctl helper routine for a process's rlimits as exposed by sysctl.
747 1.74 atatat */
748 1.74 atatat static int
749 1.74 atatat sysctl_proc_plimit(SYSCTLFN_ARGS)
750 1.74 atatat {
751 1.74 atatat struct proc *ptmp, *p;
752 1.74 atatat u_int limitno;
753 1.74 atatat int which, error = 0;
754 1.74 atatat struct rlimit alim;
755 1.74 atatat struct sysctlnode node;
756 1.74 atatat
757 1.74 atatat if (namelen != 0)
758 1.74 atatat return (EINVAL);
759 1.74 atatat
760 1.74 atatat which = name[-1];
761 1.74 atatat if (which != PROC_PID_LIMIT_TYPE_SOFT &&
762 1.74 atatat which != PROC_PID_LIMIT_TYPE_HARD)
763 1.74 atatat return (EINVAL);
764 1.74 atatat
765 1.74 atatat limitno = name[-2] - 1;
766 1.74 atatat if (limitno >= RLIM_NLIMITS)
767 1.74 atatat return (EINVAL);
768 1.74 atatat
769 1.74 atatat if (name[-3] != PROC_PID_LIMIT)
770 1.74 atatat return (EINVAL);
771 1.74 atatat
772 1.74 atatat p = l->l_proc;
773 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-4]);
774 1.74 atatat if (error)
775 1.74 atatat return (error);
776 1.74 atatat
777 1.74 atatat node = *rnode;
778 1.74 atatat memcpy(&alim, &ptmp->p_rlimit[limitno], sizeof(alim));
779 1.74 atatat if (which == PROC_PID_LIMIT_TYPE_HARD)
780 1.74 atatat node.sysctl_data = &alim.rlim_max;
781 1.74 atatat else
782 1.74 atatat node.sysctl_data = &alim.rlim_cur;
783 1.74 atatat
784 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
785 1.74 atatat if (error || newp == NULL)
786 1.74 atatat return (error);
787 1.74 atatat
788 1.74 atatat return (dosetrlimit(ptmp, p->p_cred, limitno, &alim));
789 1.74 atatat }
790 1.74 atatat
791 1.74 atatat /*
792 1.74 atatat * and finally, the actually glue that sticks it to the tree
793 1.74 atatat */
794 1.74 atatat SYSCTL_SETUP(sysctl_proc_setup, "sysctl proc subtree setup")
795 1.74 atatat {
796 1.74 atatat
797 1.74 atatat sysctl_createv(SYSCTL_PERMANENT,
798 1.74 atatat CTLTYPE_NODE, "proc", NULL,
799 1.74 atatat NULL, 0, NULL, 0,
800 1.74 atatat CTL_PROC, CTL_EOL);
801 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_ANYNUMBER,
802 1.74 atatat CTLTYPE_NODE, "curproc", NULL,
803 1.74 atatat NULL, 0, NULL, 0,
804 1.74 atatat CTL_PROC, PROC_CURPROC, CTL_EOL);
805 1.74 atatat
806 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READONLY2|SYSCTL_ANYWRITE,
807 1.74 atatat CTLTYPE_STRING, "corename", NULL,
808 1.74 atatat sysctl_proc_corename, 0, NULL, MAXPATHLEN,
809 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_CORENAME, CTL_EOL);
810 1.74 atatat sysctl_createv(SYSCTL_PERMANENT,
811 1.74 atatat CTLTYPE_NODE, "rlimit", NULL,
812 1.74 atatat NULL, 0, NULL, 0,
813 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, CTL_EOL);
814 1.74 atatat
815 1.74 atatat #define create_proc_plimit(s, n) do { \
816 1.74 atatat sysctl_createv(SYSCTL_PERMANENT, \
817 1.74 atatat CTLTYPE_NODE, s, NULL, \
818 1.74 atatat NULL, 0, NULL, 0, \
819 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
820 1.74 atatat CTL_EOL); \
821 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READWRITE|SYSCTL_ANYWRITE, \
822 1.74 atatat CTLTYPE_QUAD, "soft", NULL, \
823 1.74 atatat sysctl_proc_plimit, 0, NULL, 0, \
824 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
825 1.74 atatat PROC_PID_LIMIT_TYPE_SOFT, CTL_EOL); \
826 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READWRITE|SYSCTL_ANYWRITE, \
827 1.74 atatat CTLTYPE_QUAD, "hard", NULL, \
828 1.74 atatat sysctl_proc_plimit, 0, NULL, 0, \
829 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
830 1.74 atatat PROC_PID_LIMIT_TYPE_HARD, CTL_EOL); \
831 1.74 atatat } while (0/*CONSTCOND*/)
832 1.74 atatat
833 1.74 atatat create_proc_plimit("cputime", PROC_PID_LIMIT_CPU);
834 1.74 atatat create_proc_plimit("filesize", PROC_PID_LIMIT_FSIZE);
835 1.74 atatat create_proc_plimit("datasize", PROC_PID_LIMIT_DATA);
836 1.74 atatat create_proc_plimit("stacksize", PROC_PID_LIMIT_STACK);
837 1.74 atatat create_proc_plimit("coredumpsize", PROC_PID_LIMIT_CORE);
838 1.74 atatat create_proc_plimit("memoryuse", PROC_PID_LIMIT_RSS);
839 1.74 atatat create_proc_plimit("memorylocked", PROC_PID_LIMIT_MEMLOCK);
840 1.74 atatat create_proc_plimit("maxproc", PROC_PID_LIMIT_NPROC);
841 1.74 atatat create_proc_plimit("descriptors", PROC_PID_LIMIT_NOFILE);
842 1.74 atatat
843 1.74 atatat #undef create_proc_plimit
844 1.74 atatat
845 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READWRITE|SYSCTL_ANYWRITE,
846 1.74 atatat CTLTYPE_INT, "stopfork", NULL,
847 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
848 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPFORK, CTL_EOL);
849 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READWRITE|SYSCTL_ANYWRITE,
850 1.74 atatat CTLTYPE_INT, "stopexec", NULL,
851 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
852 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPEXEC, CTL_EOL);
853 1.74 atatat #ifdef PROC_PID_STOPEXIT
854 1.74 atatat sysctl_createv(SYSCTL_PERMANENT|SYSCTL_READWRITE|SYSCTL_ANYWRITE,
855 1.74 atatat CTLTYPE_INT, "stopexit", NULL,
856 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
857 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPEXIT, CTL_EOL);
858 1.74 atatat #endif /* PROC_PID_STOPEXIT */
859 1.17 cgd }
860