kern_resource.c revision 1.84 1 1.84 christos /* $NetBSD: kern_resource.c,v 1.84 2004/05/13 17:43:11 christos 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.84 christos __KERNEL_RCSID(0, "$NetBSD: kern_resource.c,v 1.84 2004/05/13 17:43:11 christos 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 rlim_t maxdmap = MAXDSIZ;
63 1.60 eeh rlim_t maxsmap = MAXSSIZ;
64 1.60 eeh
65 1.82 matt struct uihashhead *uihashtbl;
66 1.82 matt u_long uihash; /* size of hash table - 1 */
67 1.82 matt
68 1.79 christos static struct uidinfo *getuidinfo(uid_t);
69 1.79 christos static void freeuidinfo(struct uidinfo *);
70 1.79 christos static struct uidinfo *allocuidinfo(uid_t);
71 1.79 christos
72 1.60 eeh /*
73 1.17 cgd * Resource controls and accounting.
74 1.17 cgd */
75 1.17 cgd
76 1.25 cgd int
77 1.68 thorpej sys_getpriority(l, v, retval)
78 1.68 thorpej struct lwp *l;
79 1.30 thorpej void *v;
80 1.30 thorpej register_t *retval;
81 1.30 thorpej {
82 1.54 augustss struct sys_getpriority_args /* {
83 1.22 cgd syscallarg(int) which;
84 1.81 kleink syscallarg(id_t) who;
85 1.30 thorpej } */ *uap = v;
86 1.68 thorpej struct proc *curp = l->l_proc, *p;
87 1.54 augustss int low = NZERO + PRIO_MAX + 1;
88 1.17 cgd
89 1.22 cgd switch (SCARG(uap, which)) {
90 1.17 cgd
91 1.17 cgd case PRIO_PROCESS:
92 1.22 cgd if (SCARG(uap, who) == 0)
93 1.17 cgd p = curp;
94 1.17 cgd else
95 1.22 cgd p = pfind(SCARG(uap, who));
96 1.17 cgd if (p == 0)
97 1.17 cgd break;
98 1.17 cgd low = p->p_nice;
99 1.17 cgd break;
100 1.17 cgd
101 1.17 cgd case PRIO_PGRP: {
102 1.54 augustss struct pgrp *pg;
103 1.17 cgd
104 1.22 cgd if (SCARG(uap, who) == 0)
105 1.17 cgd pg = curp->p_pgrp;
106 1.22 cgd else if ((pg = pgfind(SCARG(uap, who))) == NULL)
107 1.17 cgd break;
108 1.64 matt LIST_FOREACH(p, &pg->pg_members, p_pglist) {
109 1.17 cgd if (p->p_nice < low)
110 1.17 cgd low = p->p_nice;
111 1.17 cgd }
112 1.17 cgd break;
113 1.17 cgd }
114 1.17 cgd
115 1.17 cgd case PRIO_USER:
116 1.22 cgd if (SCARG(uap, who) == 0)
117 1.22 cgd SCARG(uap, who) = curp->p_ucred->cr_uid;
118 1.52 thorpej proclist_lock_read();
119 1.64 matt LIST_FOREACH(p, &allproc, p_list) {
120 1.63 thorpej if (p->p_ucred->cr_uid == (uid_t) SCARG(uap, who) &&
121 1.22 cgd p->p_nice < low)
122 1.17 cgd low = p->p_nice;
123 1.64 matt }
124 1.51 thorpej proclist_unlock_read();
125 1.17 cgd break;
126 1.17 cgd
127 1.17 cgd default:
128 1.17 cgd return (EINVAL);
129 1.17 cgd }
130 1.37 ws if (low == NZERO + PRIO_MAX + 1)
131 1.17 cgd return (ESRCH);
132 1.37 ws *retval = low - NZERO;
133 1.17 cgd return (0);
134 1.17 cgd }
135 1.17 cgd
136 1.17 cgd /* ARGSUSED */
137 1.25 cgd int
138 1.68 thorpej sys_setpriority(l, v, retval)
139 1.68 thorpej struct lwp *l;
140 1.30 thorpej void *v;
141 1.30 thorpej register_t *retval;
142 1.30 thorpej {
143 1.54 augustss struct sys_setpriority_args /* {
144 1.22 cgd syscallarg(int) which;
145 1.81 kleink syscallarg(id_t) who;
146 1.22 cgd syscallarg(int) prio;
147 1.30 thorpej } */ *uap = v;
148 1.68 thorpej struct proc *curp = l->l_proc, *p;
149 1.17 cgd int found = 0, error = 0;
150 1.17 cgd
151 1.22 cgd switch (SCARG(uap, which)) {
152 1.17 cgd
153 1.17 cgd case PRIO_PROCESS:
154 1.22 cgd if (SCARG(uap, who) == 0)
155 1.17 cgd p = curp;
156 1.17 cgd else
157 1.22 cgd p = pfind(SCARG(uap, who));
158 1.17 cgd if (p == 0)
159 1.17 cgd break;
160 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
161 1.17 cgd found++;
162 1.17 cgd break;
163 1.17 cgd
164 1.17 cgd case PRIO_PGRP: {
165 1.54 augustss struct pgrp *pg;
166 1.17 cgd
167 1.22 cgd if (SCARG(uap, who) == 0)
168 1.17 cgd pg = curp->p_pgrp;
169 1.22 cgd else if ((pg = pgfind(SCARG(uap, who))) == NULL)
170 1.17 cgd break;
171 1.64 matt LIST_FOREACH(p, &pg->pg_members, p_pglist) {
172 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
173 1.17 cgd found++;
174 1.17 cgd }
175 1.17 cgd break;
176 1.17 cgd }
177 1.17 cgd
178 1.17 cgd case PRIO_USER:
179 1.22 cgd if (SCARG(uap, who) == 0)
180 1.22 cgd SCARG(uap, who) = curp->p_ucred->cr_uid;
181 1.52 thorpej proclist_lock_read();
182 1.64 matt LIST_FOREACH(p, &allproc, p_list) {
183 1.63 thorpej if (p->p_ucred->cr_uid == (uid_t) SCARG(uap, who)) {
184 1.22 cgd error = donice(curp, p, SCARG(uap, prio));
185 1.17 cgd found++;
186 1.17 cgd }
187 1.64 matt }
188 1.51 thorpej proclist_unlock_read();
189 1.17 cgd break;
190 1.17 cgd
191 1.17 cgd default:
192 1.17 cgd return (EINVAL);
193 1.17 cgd }
194 1.17 cgd if (found == 0)
195 1.17 cgd return (ESRCH);
196 1.17 cgd return (error);
197 1.17 cgd }
198 1.17 cgd
199 1.25 cgd int
200 1.17 cgd donice(curp, chgp, n)
201 1.54 augustss struct proc *curp, *chgp;
202 1.54 augustss int n;
203 1.17 cgd {
204 1.54 augustss struct pcred *pcred = curp->p_cred;
205 1.59 thorpej int s;
206 1.17 cgd
207 1.17 cgd if (pcred->pc_ucred->cr_uid && pcred->p_ruid &&
208 1.17 cgd pcred->pc_ucred->cr_uid != chgp->p_ucred->cr_uid &&
209 1.17 cgd pcred->p_ruid != chgp->p_ucred->cr_uid)
210 1.17 cgd return (EPERM);
211 1.17 cgd if (n > PRIO_MAX)
212 1.17 cgd n = PRIO_MAX;
213 1.17 cgd if (n < PRIO_MIN)
214 1.17 cgd n = PRIO_MIN;
215 1.37 ws n += NZERO;
216 1.17 cgd if (n < chgp->p_nice && suser(pcred->pc_ucred, &curp->p_acflag))
217 1.17 cgd return (EACCES);
218 1.17 cgd chgp->p_nice = n;
219 1.59 thorpej SCHED_LOCK(s);
220 1.68 thorpej (void)resetprocpriority(chgp);
221 1.59 thorpej SCHED_UNLOCK(s);
222 1.17 cgd return (0);
223 1.17 cgd }
224 1.17 cgd
225 1.17 cgd /* ARGSUSED */
226 1.25 cgd int
227 1.68 thorpej sys_setrlimit(l, v, retval)
228 1.68 thorpej struct lwp *l;
229 1.30 thorpej void *v;
230 1.30 thorpej register_t *retval;
231 1.30 thorpej {
232 1.54 augustss struct sys_setrlimit_args /* {
233 1.42 mycroft syscallarg(int) which;
234 1.39 cgd syscallarg(const struct rlimit *) rlp;
235 1.30 thorpej } */ *uap = v;
236 1.68 thorpej struct proc *p = l->l_proc;
237 1.42 mycroft int which = SCARG(uap, which);
238 1.19 cgd struct rlimit alim;
239 1.17 cgd int error;
240 1.17 cgd
241 1.46 perry error = copyin(SCARG(uap, rlp), &alim, sizeof(struct rlimit));
242 1.33 christos if (error)
243 1.17 cgd return (error);
244 1.53 bouyer return (dosetrlimit(p, p->p_cred, which, &alim));
245 1.17 cgd }
246 1.17 cgd
247 1.17 cgd int
248 1.53 bouyer dosetrlimit(p, cred, which, limp)
249 1.17 cgd struct proc *p;
250 1.53 bouyer struct pcred *cred;
251 1.42 mycroft int which;
252 1.17 cgd struct rlimit *limp;
253 1.17 cgd {
254 1.54 augustss struct rlimit *alimp;
255 1.83 pk struct plimit *oldplim;
256 1.17 cgd int error;
257 1.17 cgd
258 1.67 itojun if ((u_int)which >= RLIM_NLIMITS)
259 1.17 cgd return (EINVAL);
260 1.38 matthias
261 1.38 matthias if (limp->rlim_cur < 0 || limp->rlim_max < 0)
262 1.38 matthias return (EINVAL);
263 1.38 matthias
264 1.17 cgd alimp = &p->p_rlimit[which];
265 1.53 bouyer /* if we don't change the value, no need to limcopy() */
266 1.53 bouyer if (limp->rlim_cur == alimp->rlim_cur &&
267 1.53 bouyer limp->rlim_max == alimp->rlim_max)
268 1.53 bouyer return 0;
269 1.53 bouyer
270 1.62 jdolecek if (limp->rlim_cur > limp->rlim_max) {
271 1.62 jdolecek /*
272 1.62 jdolecek * This is programming error. According to SUSv2, we should
273 1.62 jdolecek * return error in this case.
274 1.62 jdolecek */
275 1.62 jdolecek return (EINVAL);
276 1.62 jdolecek }
277 1.62 jdolecek if (limp->rlim_max > alimp->rlim_max
278 1.62 jdolecek && (error = suser(cred->pc_ucred, &p->p_acflag)) != 0)
279 1.17 cgd return (error);
280 1.62 jdolecek
281 1.17 cgd if (p->p_limit->p_refcnt > 1 &&
282 1.17 cgd (p->p_limit->p_lflags & PL_SHAREMOD) == 0) {
283 1.83 pk p->p_limit = limcopy(oldplim = p->p_limit);
284 1.83 pk limfree(oldplim);
285 1.17 cgd alimp = &p->p_rlimit[which];
286 1.17 cgd }
287 1.17 cgd
288 1.17 cgd switch (which) {
289 1.17 cgd
290 1.17 cgd case RLIMIT_DATA:
291 1.19 cgd if (limp->rlim_cur > maxdmap)
292 1.19 cgd limp->rlim_cur = maxdmap;
293 1.19 cgd if (limp->rlim_max > maxdmap)
294 1.19 cgd limp->rlim_max = maxdmap;
295 1.17 cgd break;
296 1.17 cgd
297 1.17 cgd case RLIMIT_STACK:
298 1.19 cgd if (limp->rlim_cur > maxsmap)
299 1.19 cgd limp->rlim_cur = maxsmap;
300 1.19 cgd if (limp->rlim_max > maxsmap)
301 1.19 cgd limp->rlim_max = maxsmap;
302 1.62 jdolecek
303 1.62 jdolecek /*
304 1.62 jdolecek * Return EINVAL if the new stack size limit is lower than
305 1.62 jdolecek * current usage. Otherwise, the process would get SIGSEGV the
306 1.62 jdolecek * moment it would try to access anything on it's current stack.
307 1.62 jdolecek * This conforms to SUSv2.
308 1.62 jdolecek */
309 1.62 jdolecek if (limp->rlim_cur < p->p_vmspace->vm_ssize * PAGE_SIZE
310 1.62 jdolecek || limp->rlim_max < p->p_vmspace->vm_ssize * PAGE_SIZE)
311 1.62 jdolecek return (EINVAL);
312 1.40 enami
313 1.17 cgd /*
314 1.40 enami * Stack is allocated to the max at exec time with
315 1.40 enami * only "rlim_cur" bytes accessible (In other words,
316 1.40 enami * allocates stack dividing two contiguous regions at
317 1.40 enami * "rlim_cur" bytes boundary).
318 1.40 enami *
319 1.40 enami * Since allocation is done in terms of page, roundup
320 1.40 enami * "rlim_cur" (otherwise, contiguous regions
321 1.40 enami * overlap). If stack limit is going up make more
322 1.40 enami * accessible, if going down make inaccessible.
323 1.17 cgd */
324 1.40 enami limp->rlim_cur = round_page(limp->rlim_cur);
325 1.17 cgd if (limp->rlim_cur != alimp->rlim_cur) {
326 1.48 eeh vaddr_t addr;
327 1.48 eeh vsize_t size;
328 1.17 cgd vm_prot_t prot;
329 1.17 cgd
330 1.17 cgd if (limp->rlim_cur > alimp->rlim_cur) {
331 1.73 chs prot = VM_PROT_READ | VM_PROT_WRITE;
332 1.17 cgd size = limp->rlim_cur - alimp->rlim_cur;
333 1.17 cgd addr = USRSTACK - limp->rlim_cur;
334 1.17 cgd } else {
335 1.17 cgd prot = VM_PROT_NONE;
336 1.17 cgd size = alimp->rlim_cur - limp->rlim_cur;
337 1.17 cgd addr = USRSTACK - alimp->rlim_cur;
338 1.17 cgd }
339 1.43 mrg (void) uvm_map_protect(&p->p_vmspace->vm_map,
340 1.43 mrg addr, addr+size, prot, FALSE);
341 1.17 cgd }
342 1.17 cgd break;
343 1.19 cgd
344 1.19 cgd case RLIMIT_NOFILE:
345 1.19 cgd if (limp->rlim_cur > maxfiles)
346 1.19 cgd limp->rlim_cur = maxfiles;
347 1.19 cgd if (limp->rlim_max > maxfiles)
348 1.19 cgd limp->rlim_max = maxfiles;
349 1.19 cgd break;
350 1.19 cgd
351 1.19 cgd case RLIMIT_NPROC:
352 1.19 cgd if (limp->rlim_cur > maxproc)
353 1.19 cgd limp->rlim_cur = maxproc;
354 1.19 cgd if (limp->rlim_max > maxproc)
355 1.19 cgd limp->rlim_max = maxproc;
356 1.19 cgd break;
357 1.17 cgd }
358 1.17 cgd *alimp = *limp;
359 1.17 cgd return (0);
360 1.17 cgd }
361 1.17 cgd
362 1.17 cgd /* ARGSUSED */
363 1.25 cgd int
364 1.68 thorpej sys_getrlimit(l, v, retval)
365 1.68 thorpej struct lwp *l;
366 1.30 thorpej void *v;
367 1.30 thorpej register_t *retval;
368 1.30 thorpej {
369 1.54 augustss struct sys_getrlimit_args /* {
370 1.42 mycroft syscallarg(int) which;
371 1.22 cgd syscallarg(struct rlimit *) rlp;
372 1.30 thorpej } */ *uap = v;
373 1.68 thorpej struct proc *p = l->l_proc;
374 1.42 mycroft int which = SCARG(uap, which);
375 1.17 cgd
376 1.67 itojun if ((u_int)which >= RLIM_NLIMITS)
377 1.17 cgd return (EINVAL);
378 1.42 mycroft return (copyout(&p->p_rlimit[which], SCARG(uap, rlp),
379 1.46 perry sizeof(struct rlimit)));
380 1.17 cgd }
381 1.17 cgd
382 1.17 cgd /*
383 1.17 cgd * Transform the running time and tick information in proc p into user,
384 1.17 cgd * system, and interrupt time usage.
385 1.17 cgd */
386 1.25 cgd void
387 1.17 cgd calcru(p, up, sp, ip)
388 1.54 augustss struct proc *p;
389 1.54 augustss struct timeval *up;
390 1.54 augustss struct timeval *sp;
391 1.54 augustss struct timeval *ip;
392 1.17 cgd {
393 1.54 augustss u_quad_t u, st, ut, it, tot;
394 1.70 dsl unsigned long sec;
395 1.70 dsl long usec;
396 1.54 augustss int s;
397 1.17 cgd struct timeval tv;
398 1.68 thorpej struct lwp *l;
399 1.17 cgd
400 1.17 cgd s = splstatclock();
401 1.17 cgd st = p->p_sticks;
402 1.17 cgd ut = p->p_uticks;
403 1.17 cgd it = p->p_iticks;
404 1.17 cgd splx(s);
405 1.17 cgd
406 1.17 cgd sec = p->p_rtime.tv_sec;
407 1.17 cgd usec = p->p_rtime.tv_usec;
408 1.70 dsl LIST_FOREACH(l, &p->p_lwps, l_sibling) {
409 1.68 thorpej if (l->l_stat == LSONPROC) {
410 1.68 thorpej struct schedstate_percpu *spc;
411 1.68 thorpej
412 1.68 thorpej KDASSERT(l->l_cpu != NULL);
413 1.68 thorpej spc = &l->l_cpu->ci_schedstate;
414 1.68 thorpej
415 1.68 thorpej /*
416 1.68 thorpej * Adjust for the current time slice. This is
417 1.68 thorpej * actually fairly important since the error
418 1.68 thorpej * here is on the order of a time quantum,
419 1.68 thorpej * which is much greater than the sampling
420 1.68 thorpej * error.
421 1.68 thorpej */
422 1.68 thorpej microtime(&tv);
423 1.68 thorpej sec += tv.tv_sec - spc->spc_runtime.tv_sec;
424 1.68 thorpej usec += tv.tv_usec - spc->spc_runtime.tv_usec;
425 1.68 thorpej }
426 1.17 cgd }
427 1.69 dsl
428 1.69 dsl tot = st + ut + it;
429 1.70 dsl u = sec * 1000000ull + usec;
430 1.70 dsl
431 1.69 dsl if (tot == 0) {
432 1.69 dsl /* No ticks, so can't use to share time out, split 50-50 */
433 1.70 dsl st = ut = u / 2;
434 1.70 dsl } else {
435 1.70 dsl st = (u * st) / tot;
436 1.70 dsl ut = (u * ut) / tot;
437 1.69 dsl }
438 1.17 cgd sp->tv_sec = st / 1000000;
439 1.17 cgd sp->tv_usec = st % 1000000;
440 1.17 cgd up->tv_sec = ut / 1000000;
441 1.17 cgd up->tv_usec = ut % 1000000;
442 1.17 cgd if (ip != NULL) {
443 1.70 dsl if (it != 0)
444 1.70 dsl it = (u * it) / tot;
445 1.17 cgd ip->tv_sec = it / 1000000;
446 1.17 cgd ip->tv_usec = it % 1000000;
447 1.17 cgd }
448 1.17 cgd }
449 1.17 cgd
450 1.17 cgd /* ARGSUSED */
451 1.25 cgd int
452 1.68 thorpej sys_getrusage(l, v, retval)
453 1.68 thorpej struct lwp *l;
454 1.30 thorpej void *v;
455 1.30 thorpej register_t *retval;
456 1.30 thorpej {
457 1.54 augustss struct sys_getrusage_args /* {
458 1.22 cgd syscallarg(int) who;
459 1.22 cgd syscallarg(struct rusage *) rusage;
460 1.30 thorpej } */ *uap = v;
461 1.54 augustss struct rusage *rup;
462 1.68 thorpej struct proc *p = l->l_proc;
463 1.17 cgd
464 1.22 cgd switch (SCARG(uap, who)) {
465 1.17 cgd
466 1.19 cgd case RUSAGE_SELF:
467 1.17 cgd rup = &p->p_stats->p_ru;
468 1.17 cgd calcru(p, &rup->ru_utime, &rup->ru_stime, NULL);
469 1.17 cgd break;
470 1.17 cgd
471 1.17 cgd case RUSAGE_CHILDREN:
472 1.17 cgd rup = &p->p_stats->p_cru;
473 1.17 cgd break;
474 1.17 cgd
475 1.17 cgd default:
476 1.17 cgd return (EINVAL);
477 1.17 cgd }
478 1.46 perry return (copyout(rup, SCARG(uap, rusage), sizeof(struct rusage)));
479 1.17 cgd }
480 1.17 cgd
481 1.25 cgd void
482 1.17 cgd ruadd(ru, ru2)
483 1.54 augustss struct rusage *ru, *ru2;
484 1.17 cgd {
485 1.54 augustss long *ip, *ip2;
486 1.54 augustss int i;
487 1.17 cgd
488 1.27 mycroft timeradd(&ru->ru_utime, &ru2->ru_utime, &ru->ru_utime);
489 1.27 mycroft timeradd(&ru->ru_stime, &ru2->ru_stime, &ru->ru_stime);
490 1.17 cgd if (ru->ru_maxrss < ru2->ru_maxrss)
491 1.17 cgd ru->ru_maxrss = ru2->ru_maxrss;
492 1.17 cgd ip = &ru->ru_first; ip2 = &ru2->ru_first;
493 1.17 cgd for (i = &ru->ru_last - &ru->ru_first; i >= 0; i--)
494 1.17 cgd *ip++ += *ip2++;
495 1.17 cgd }
496 1.17 cgd
497 1.17 cgd /*
498 1.17 cgd * Make a copy of the plimit structure.
499 1.17 cgd * We share these structures copy-on-write after fork,
500 1.17 cgd * and copy when a limit is changed.
501 1.17 cgd */
502 1.17 cgd struct plimit *
503 1.17 cgd limcopy(lim)
504 1.17 cgd struct plimit *lim;
505 1.17 cgd {
506 1.54 augustss struct plimit *newlim;
507 1.83 pk size_t l = 0;
508 1.83 pk
509 1.83 pk simple_lock(&lim->p_slock);
510 1.83 pk if (lim->pl_corename != defcorename)
511 1.83 pk l = strlen(lim->pl_corename) + 1;
512 1.83 pk simple_unlock(&lim->p_slock);
513 1.17 cgd
514 1.49 thorpej newlim = pool_get(&plimit_pool, PR_WAITOK);
515 1.83 pk simple_lock_init(&newlim->p_slock);
516 1.83 pk newlim->p_lflags = 0;
517 1.83 pk newlim->p_refcnt = 1;
518 1.83 pk newlim->pl_corename = (l != 0)
519 1.83 pk ? malloc(l, M_TEMP, M_WAITOK)
520 1.83 pk : defcorename;
521 1.83 pk
522 1.83 pk simple_lock(&lim->p_slock);
523 1.47 perry memcpy(newlim->pl_rlimit, lim->pl_rlimit,
524 1.17 cgd sizeof(struct rlimit) * RLIM_NLIMITS);
525 1.83 pk
526 1.83 pk if (l != 0)
527 1.71 itojun strlcpy(newlim->pl_corename, lim->pl_corename, l);
528 1.83 pk simple_unlock(&lim->p_slock);
529 1.83 pk
530 1.32 mycroft return (newlim);
531 1.32 mycroft }
532 1.32 mycroft
533 1.32 mycroft void
534 1.32 mycroft limfree(lim)
535 1.32 mycroft struct plimit *lim;
536 1.32 mycroft {
537 1.84 christos int n;
538 1.83 pk simple_lock(&lim->p_slock);
539 1.84 christos n = --lim->p_refcnt;
540 1.83 pk simple_unlock(&lim->p_slock);
541 1.83 pk if (n > 0)
542 1.32 mycroft return;
543 1.53 bouyer #ifdef DIAGNOSTIC
544 1.83 pk if (n < 0)
545 1.53 bouyer panic("limfree");
546 1.53 bouyer #endif
547 1.53 bouyer if (lim->pl_corename != defcorename)
548 1.53 bouyer free(lim->pl_corename, M_TEMP);
549 1.49 thorpej pool_put(&plimit_pool, lim);
550 1.68 thorpej }
551 1.68 thorpej
552 1.68 thorpej struct pstats *
553 1.68 thorpej pstatscopy(ps)
554 1.68 thorpej struct pstats *ps;
555 1.68 thorpej {
556 1.68 thorpej
557 1.68 thorpej struct pstats *newps;
558 1.68 thorpej
559 1.68 thorpej newps = pool_get(&pstats_pool, PR_WAITOK);
560 1.68 thorpej
561 1.68 thorpej memset(&newps->pstat_startzero, 0,
562 1.68 thorpej (unsigned) ((caddr_t)&newps->pstat_endzero -
563 1.68 thorpej (caddr_t)&newps->pstat_startzero));
564 1.68 thorpej memcpy(&newps->pstat_startcopy, &ps->pstat_startcopy,
565 1.68 thorpej ((caddr_t)&newps->pstat_endcopy -
566 1.68 thorpej (caddr_t)&newps->pstat_startcopy));
567 1.68 thorpej
568 1.68 thorpej return (newps);
569 1.68 thorpej
570 1.68 thorpej }
571 1.68 thorpej
572 1.68 thorpej void
573 1.68 thorpej pstatsfree(ps)
574 1.68 thorpej struct pstats *ps;
575 1.68 thorpej {
576 1.68 thorpej
577 1.68 thorpej pool_put(&pstats_pool, ps);
578 1.74 atatat }
579 1.74 atatat
580 1.74 atatat /*
581 1.74 atatat * sysctl interface in five parts
582 1.74 atatat */
583 1.74 atatat
584 1.74 atatat /*
585 1.74 atatat * a routine for sysctl proc subtree helpers that need to pick a valid
586 1.74 atatat * process by pid.
587 1.74 atatat */
588 1.74 atatat static int
589 1.74 atatat sysctl_proc_findproc(struct proc *p, struct proc **p2, pid_t pid)
590 1.74 atatat {
591 1.74 atatat struct proc *ptmp;
592 1.74 atatat int i, error = 0;
593 1.74 atatat
594 1.74 atatat if (pid == PROC_CURPROC)
595 1.74 atatat ptmp = p;
596 1.74 atatat else if ((ptmp = pfind(pid)) == NULL)
597 1.74 atatat error = ESRCH;
598 1.74 atatat else {
599 1.74 atatat /*
600 1.74 atatat * suid proc of ours or proc not ours
601 1.74 atatat */
602 1.74 atatat if (p->p_cred->p_ruid != ptmp->p_cred->p_ruid ||
603 1.74 atatat p->p_cred->p_ruid != ptmp->p_cred->p_svuid)
604 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
605 1.74 atatat
606 1.74 atatat /*
607 1.74 atatat * sgid proc has sgid back to us temporarily
608 1.74 atatat */
609 1.74 atatat else if (ptmp->p_cred->p_rgid != ptmp->p_cred->p_svgid)
610 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
611 1.74 atatat
612 1.74 atatat /*
613 1.74 atatat * our rgid must be in target's group list (ie,
614 1.74 atatat * sub-processes started by a sgid process)
615 1.74 atatat */
616 1.74 atatat else {
617 1.74 atatat for (i = 0; i < p->p_ucred->cr_ngroups; i++) {
618 1.74 atatat if (p->p_ucred->cr_groups[i] ==
619 1.74 atatat ptmp->p_cred->p_rgid)
620 1.74 atatat break;
621 1.74 atatat }
622 1.74 atatat if (i == p->p_ucred->cr_ngroups)
623 1.74 atatat error = suser(p->p_ucred, &p->p_acflag);
624 1.74 atatat }
625 1.74 atatat }
626 1.74 atatat
627 1.74 atatat *p2 = ptmp;
628 1.74 atatat return (error);
629 1.74 atatat }
630 1.74 atatat
631 1.74 atatat /*
632 1.74 atatat * sysctl helper routine for setting a process's specific corefile
633 1.74 atatat * name. picks the process based on the given pid and checks the
634 1.74 atatat * correctness of the new value.
635 1.74 atatat */
636 1.74 atatat static int
637 1.74 atatat sysctl_proc_corename(SYSCTLFN_ARGS)
638 1.74 atatat {
639 1.74 atatat struct proc *ptmp, *p;
640 1.83 pk struct plimit *lim;
641 1.74 atatat int error = 0, len;
642 1.74 atatat char cname[MAXPATHLEN], *tmp;
643 1.74 atatat struct sysctlnode node;
644 1.74 atatat
645 1.74 atatat /*
646 1.74 atatat * is this all correct?
647 1.74 atatat */
648 1.74 atatat if (namelen != 0)
649 1.74 atatat return (EINVAL);
650 1.74 atatat if (name[-1] != PROC_PID_CORENAME)
651 1.74 atatat return (EINVAL);
652 1.74 atatat
653 1.74 atatat /*
654 1.74 atatat * whom are we tweaking?
655 1.74 atatat */
656 1.74 atatat p = l->l_proc;
657 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-2]);
658 1.74 atatat if (error)
659 1.74 atatat return (error);
660 1.74 atatat
661 1.74 atatat /*
662 1.74 atatat * let them modify a temporary copy of the core name
663 1.74 atatat */
664 1.74 atatat node = *rnode;
665 1.74 atatat strlcpy(cname, ptmp->p_limit->pl_corename, sizeof(cname));
666 1.74 atatat node.sysctl_data = cname;
667 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
668 1.74 atatat
669 1.74 atatat /*
670 1.74 atatat * if that failed, or they have nothing new to say, or we've
671 1.74 atatat * heard it before...
672 1.74 atatat */
673 1.74 atatat if (error || newp == NULL ||
674 1.74 atatat strcmp(cname, ptmp->p_limit->pl_corename) == 0)
675 1.74 atatat return (error);
676 1.74 atatat
677 1.74 atatat /*
678 1.74 atatat * no error yet and cname now has the new core name in it.
679 1.74 atatat * let's see if it looks acceptable. it must be either "core"
680 1.74 atatat * or end in ".core" or "/core".
681 1.74 atatat */
682 1.74 atatat len = strlen(cname);
683 1.74 atatat if (len < 4)
684 1.74 atatat return (EINVAL);
685 1.74 atatat if (strcmp(cname + len - 4, "core") != 0)
686 1.74 atatat return (EINVAL);
687 1.74 atatat if (len > 4 && cname[len - 5] != '/' && cname[len - 5] != '.')
688 1.74 atatat return (EINVAL);
689 1.74 atatat
690 1.74 atatat /*
691 1.74 atatat * hmm...looks good. now...where do we put it?
692 1.74 atatat */
693 1.74 atatat tmp = malloc(len + 1, M_TEMP, M_WAITOK|M_CANFAIL);
694 1.74 atatat if (tmp == NULL)
695 1.74 atatat return (ENOMEM);
696 1.74 atatat strlcpy(tmp, cname, len + 1);
697 1.74 atatat
698 1.83 pk lim = ptmp->p_limit;
699 1.83 pk if (lim->p_refcnt > 1 && (lim->p_lflags & PL_SHAREMOD) == 0) {
700 1.83 pk ptmp->p_limit = limcopy(lim);
701 1.83 pk limfree(lim);
702 1.83 pk lim = ptmp->p_limit;
703 1.83 pk }
704 1.83 pk if (lim->pl_corename != defcorename)
705 1.83 pk free(lim->pl_corename, M_TEMP);
706 1.83 pk lim->pl_corename = tmp;
707 1.74 atatat
708 1.74 atatat return (error);
709 1.74 atatat }
710 1.74 atatat
711 1.74 atatat /*
712 1.74 atatat * sysctl helper routine for checking/setting a process's stop flags,
713 1.74 atatat * one for fork and one for exec.
714 1.74 atatat */
715 1.74 atatat static int
716 1.74 atatat sysctl_proc_stop(SYSCTLFN_ARGS)
717 1.74 atatat {
718 1.74 atatat struct proc *p, *ptmp;
719 1.74 atatat int i, f, error = 0;
720 1.74 atatat struct sysctlnode node;
721 1.74 atatat
722 1.74 atatat if (namelen != 0)
723 1.74 atatat return (EINVAL);
724 1.74 atatat
725 1.74 atatat p = l->l_proc;
726 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-2]);
727 1.74 atatat if (error)
728 1.74 atatat return (error);
729 1.74 atatat
730 1.74 atatat switch (rnode->sysctl_num) {
731 1.74 atatat case PROC_PID_STOPFORK:
732 1.74 atatat f = P_STOPFORK;
733 1.74 atatat break;
734 1.74 atatat case PROC_PID_STOPEXEC:
735 1.74 atatat f = P_STOPEXEC;
736 1.74 atatat break;
737 1.74 atatat case PROC_PID_STOPEXIT:
738 1.74 atatat f = P_STOPEXIT;
739 1.74 atatat break;
740 1.74 atatat default:
741 1.74 atatat return (EINVAL);
742 1.74 atatat }
743 1.74 atatat
744 1.74 atatat i = (ptmp->p_flag & f) ? 1 : 0;
745 1.74 atatat node = *rnode;
746 1.74 atatat node.sysctl_data = &i;
747 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
748 1.74 atatat if (error || newp == NULL)
749 1.74 atatat return (error);
750 1.74 atatat
751 1.74 atatat if (i)
752 1.74 atatat ptmp->p_flag |= f;
753 1.74 atatat else
754 1.74 atatat ptmp->p_flag &= ~f;
755 1.74 atatat
756 1.74 atatat return (0);
757 1.74 atatat }
758 1.74 atatat
759 1.74 atatat /*
760 1.74 atatat * sysctl helper routine for a process's rlimits as exposed by sysctl.
761 1.74 atatat */
762 1.74 atatat static int
763 1.74 atatat sysctl_proc_plimit(SYSCTLFN_ARGS)
764 1.74 atatat {
765 1.74 atatat struct proc *ptmp, *p;
766 1.74 atatat u_int limitno;
767 1.74 atatat int which, error = 0;
768 1.74 atatat struct rlimit alim;
769 1.74 atatat struct sysctlnode node;
770 1.74 atatat
771 1.74 atatat if (namelen != 0)
772 1.74 atatat return (EINVAL);
773 1.74 atatat
774 1.74 atatat which = name[-1];
775 1.74 atatat if (which != PROC_PID_LIMIT_TYPE_SOFT &&
776 1.74 atatat which != PROC_PID_LIMIT_TYPE_HARD)
777 1.74 atatat return (EINVAL);
778 1.74 atatat
779 1.74 atatat limitno = name[-2] - 1;
780 1.74 atatat if (limitno >= RLIM_NLIMITS)
781 1.74 atatat return (EINVAL);
782 1.74 atatat
783 1.74 atatat if (name[-3] != PROC_PID_LIMIT)
784 1.74 atatat return (EINVAL);
785 1.74 atatat
786 1.74 atatat p = l->l_proc;
787 1.74 atatat error = sysctl_proc_findproc(p, &ptmp, (pid_t)name[-4]);
788 1.74 atatat if (error)
789 1.74 atatat return (error);
790 1.74 atatat
791 1.74 atatat node = *rnode;
792 1.74 atatat memcpy(&alim, &ptmp->p_rlimit[limitno], sizeof(alim));
793 1.74 atatat if (which == PROC_PID_LIMIT_TYPE_HARD)
794 1.74 atatat node.sysctl_data = &alim.rlim_max;
795 1.74 atatat else
796 1.74 atatat node.sysctl_data = &alim.rlim_cur;
797 1.74 atatat
798 1.74 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
799 1.74 atatat if (error || newp == NULL)
800 1.74 atatat return (error);
801 1.74 atatat
802 1.74 atatat return (dosetrlimit(ptmp, p->p_cred, limitno, &alim));
803 1.74 atatat }
804 1.74 atatat
805 1.74 atatat /*
806 1.74 atatat * and finally, the actually glue that sticks it to the tree
807 1.74 atatat */
808 1.74 atatat SYSCTL_SETUP(sysctl_proc_setup, "sysctl proc subtree setup")
809 1.74 atatat {
810 1.74 atatat
811 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
812 1.76 atatat CTLFLAG_PERMANENT,
813 1.74 atatat CTLTYPE_NODE, "proc", NULL,
814 1.74 atatat NULL, 0, NULL, 0,
815 1.74 atatat CTL_PROC, CTL_EOL);
816 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
817 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_ANYNUMBER,
818 1.78 atatat CTLTYPE_NODE, "curproc",
819 1.78 atatat SYSCTL_DESCR("Per-process settings"),
820 1.74 atatat NULL, 0, NULL, 0,
821 1.74 atatat CTL_PROC, PROC_CURPROC, CTL_EOL);
822 1.74 atatat
823 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
824 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READONLY2|CTLFLAG_ANYWRITE,
825 1.78 atatat CTLTYPE_STRING, "corename",
826 1.78 atatat SYSCTL_DESCR("Core file name"),
827 1.74 atatat sysctl_proc_corename, 0, NULL, MAXPATHLEN,
828 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_CORENAME, CTL_EOL);
829 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
830 1.76 atatat CTLFLAG_PERMANENT,
831 1.78 atatat CTLTYPE_NODE, "rlimit",
832 1.78 atatat SYSCTL_DESCR("Process limits"),
833 1.74 atatat NULL, 0, NULL, 0,
834 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, CTL_EOL);
835 1.74 atatat
836 1.74 atatat #define create_proc_plimit(s, n) do { \
837 1.76 atatat sysctl_createv(clog, 0, NULL, NULL, \
838 1.76 atatat CTLFLAG_PERMANENT, \
839 1.78 atatat CTLTYPE_NODE, s, \
840 1.78 atatat SYSCTL_DESCR("Process " s " limits"), \
841 1.74 atatat NULL, 0, NULL, 0, \
842 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
843 1.74 atatat CTL_EOL); \
844 1.76 atatat sysctl_createv(clog, 0, NULL, NULL, \
845 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE|CTLFLAG_ANYWRITE, \
846 1.78 atatat CTLTYPE_QUAD, "soft", \
847 1.78 atatat SYSCTL_DESCR("Process soft " s " limit"), \
848 1.74 atatat sysctl_proc_plimit, 0, NULL, 0, \
849 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
850 1.74 atatat PROC_PID_LIMIT_TYPE_SOFT, CTL_EOL); \
851 1.76 atatat sysctl_createv(clog, 0, NULL, NULL, \
852 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE|CTLFLAG_ANYWRITE, \
853 1.78 atatat CTLTYPE_QUAD, "hard", \
854 1.78 atatat SYSCTL_DESCR("Process hard " s " limit"), \
855 1.74 atatat sysctl_proc_plimit, 0, NULL, 0, \
856 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_LIMIT, n, \
857 1.74 atatat PROC_PID_LIMIT_TYPE_HARD, CTL_EOL); \
858 1.74 atatat } while (0/*CONSTCOND*/)
859 1.74 atatat
860 1.74 atatat create_proc_plimit("cputime", PROC_PID_LIMIT_CPU);
861 1.74 atatat create_proc_plimit("filesize", PROC_PID_LIMIT_FSIZE);
862 1.74 atatat create_proc_plimit("datasize", PROC_PID_LIMIT_DATA);
863 1.74 atatat create_proc_plimit("stacksize", PROC_PID_LIMIT_STACK);
864 1.74 atatat create_proc_plimit("coredumpsize", PROC_PID_LIMIT_CORE);
865 1.74 atatat create_proc_plimit("memoryuse", PROC_PID_LIMIT_RSS);
866 1.74 atatat create_proc_plimit("memorylocked", PROC_PID_LIMIT_MEMLOCK);
867 1.74 atatat create_proc_plimit("maxproc", PROC_PID_LIMIT_NPROC);
868 1.74 atatat create_proc_plimit("descriptors", PROC_PID_LIMIT_NOFILE);
869 1.79 christos create_proc_plimit("sbsize", PROC_PID_LIMIT_SBSIZE);
870 1.74 atatat
871 1.74 atatat #undef create_proc_plimit
872 1.74 atatat
873 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
874 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE|CTLFLAG_ANYWRITE,
875 1.78 atatat CTLTYPE_INT, "stopfork",
876 1.78 atatat SYSCTL_DESCR("Stop process at fork(2)"),
877 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
878 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPFORK, CTL_EOL);
879 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
880 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE|CTLFLAG_ANYWRITE,
881 1.78 atatat CTLTYPE_INT, "stopexec",
882 1.78 atatat SYSCTL_DESCR("Stop process at execve(2)"),
883 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
884 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPEXEC, CTL_EOL);
885 1.76 atatat sysctl_createv(clog, 0, NULL, NULL,
886 1.76 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE|CTLFLAG_ANYWRITE,
887 1.78 atatat CTLTYPE_INT, "stopexit",
888 1.78 atatat SYSCTL_DESCR("Stop process before completing exit"),
889 1.74 atatat sysctl_proc_stop, 0, NULL, 0,
890 1.74 atatat CTL_PROC, PROC_CURPROC, PROC_PID_STOPEXIT, CTL_EOL);
891 1.17 cgd }
892 1.79 christos
893 1.79 christos static struct uidinfo *
894 1.79 christos getuidinfo(uid_t uid)
895 1.79 christos {
896 1.79 christos struct uidinfo *uip;
897 1.79 christos struct uihashhead *uipp;
898 1.79 christos
899 1.79 christos uipp = UIHASH(uid);
900 1.79 christos
901 1.79 christos LIST_FOREACH(uip, uipp, ui_hash)
902 1.79 christos if (uip->ui_uid == uid)
903 1.79 christos return uip;
904 1.79 christos return NULL;
905 1.79 christos }
906 1.79 christos
907 1.79 christos static void
908 1.79 christos freeuidinfo(struct uidinfo *uip)
909 1.79 christos {
910 1.79 christos LIST_REMOVE(uip, ui_hash);
911 1.79 christos FREE(uip, M_PROC);
912 1.79 christos }
913 1.79 christos
914 1.79 christos static struct uidinfo *
915 1.79 christos allocuidinfo(uid_t uid)
916 1.79 christos {
917 1.79 christos struct uidinfo *uip;
918 1.79 christos struct uihashhead *uipp;
919 1.79 christos
920 1.79 christos uipp = UIHASH(uid);
921 1.79 christos MALLOC(uip, struct uidinfo *, sizeof(*uip), M_PROC, M_WAITOK);
922 1.79 christos LIST_INSERT_HEAD(uipp, uip, ui_hash);
923 1.79 christos uip->ui_uid = uid;
924 1.79 christos uip->ui_proccnt = 0;
925 1.79 christos uip->ui_sbsize = 0;
926 1.79 christos return uip;
927 1.79 christos }
928 1.79 christos
929 1.79 christos /*
930 1.79 christos * Change the count associated with number of processes
931 1.79 christos * a given user is using.
932 1.79 christos */
933 1.79 christos int
934 1.79 christos chgproccnt(uid_t uid, int diff)
935 1.79 christos {
936 1.79 christos struct uidinfo *uip;
937 1.79 christos
938 1.79 christos if (diff == 0)
939 1.79 christos return 0;
940 1.79 christos
941 1.79 christos if ((uip = getuidinfo(uid)) != NULL) {
942 1.79 christos uip->ui_proccnt += diff;
943 1.79 christos KASSERT(uip->ui_proccnt >= 0);
944 1.79 christos if (uip->ui_proccnt > 0)
945 1.79 christos return uip->ui_proccnt;
946 1.79 christos else {
947 1.79 christos if (uip->ui_sbsize == 0)
948 1.79 christos freeuidinfo(uip);
949 1.79 christos return 0;
950 1.79 christos }
951 1.79 christos } else {
952 1.79 christos if (diff < 0)
953 1.79 christos panic("chgproccnt: lost user %lu", (unsigned long)uid);
954 1.79 christos uip = allocuidinfo(uid);
955 1.79 christos uip->ui_proccnt = diff;
956 1.79 christos return uip->ui_proccnt;
957 1.79 christos }
958 1.79 christos }
959 1.79 christos
960 1.79 christos int
961 1.79 christos chgsbsize(uid_t uid, u_long *hiwat, u_long to, rlim_t max)
962 1.79 christos {
963 1.84 christos *hiwat = to;
964 1.84 christos return 1;
965 1.84 christos #ifdef notyet
966 1.79 christos struct uidinfo *uip;
967 1.79 christos rlim_t nsb;
968 1.79 christos int rv = 0;
969 1.79 christos
970 1.79 christos if ((uip = getuidinfo(uid)) == NULL)
971 1.79 christos uip = allocuidinfo(uid);
972 1.80 yamt nsb = uip->ui_sbsize + to - *hiwat;
973 1.80 yamt if (to > *hiwat && nsb > max)
974 1.79 christos goto done;
975 1.79 christos *hiwat = to;
976 1.79 christos uip->ui_sbsize = nsb;
977 1.79 christos rv = 1;
978 1.79 christos KASSERT(uip->ui_sbsize >= 0);
979 1.79 christos done:
980 1.79 christos if (uip->ui_sbsize == 0 && uip->ui_proccnt == 0)
981 1.79 christos freeuidinfo(uip);
982 1.79 christos return rv;
983 1.84 christos #endif
984 1.79 christos }
985