62bc4884a7c5e9a72176bb9f861b58c4a1380fdf
[dragonfly.git] / sys / kern / kern_proc.c
1 /*
2  * (MPSAFE)
3  *
4  * Copyright (c) 1982, 1986, 1989, 1991, 1993
5  *      The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. All advertising materials mentioning features or use of this software
16  *    must display the following acknowledgement:
17  *      This product includes software developed by the University of
18  *      California, Berkeley and its contributors.
19  * 4. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  *
35  *      @(#)kern_proc.c 8.7 (Berkeley) 2/14/95
36  * $FreeBSD: src/sys/kern/kern_proc.c,v 1.63.2.9 2003/05/08 07:47:16 kbyanc Exp $
37  * $DragonFly: src/sys/kern/kern_proc.c,v 1.45 2008/06/12 23:25:02 dillon Exp $
38  */
39
40 #include <sys/param.h>
41 #include <sys/systm.h>
42 #include <sys/kernel.h>
43 #include <sys/sysctl.h>
44 #include <sys/malloc.h>
45 #include <sys/proc.h>
46 #include <sys/jail.h>
47 #include <sys/filedesc.h>
48 #include <sys/tty.h>
49 #include <sys/dsched.h>
50 #include <sys/signalvar.h>
51 #include <sys/spinlock.h>
52 #include <vm/vm.h>
53 #include <sys/lock.h>
54 #include <vm/pmap.h>
55 #include <vm/vm_map.h>
56 #include <sys/user.h>
57 #include <machine/smp.h>
58
59 #include <sys/spinlock2.h>
60 #include <sys/mplock2.h>
61
62 static MALLOC_DEFINE(M_PGRP, "pgrp", "process group header");
63 MALLOC_DEFINE(M_SESSION, "session", "session header");
64 MALLOC_DEFINE(M_PROC, "proc", "Proc structures");
65 MALLOC_DEFINE(M_LWP, "lwp", "lwp structures");
66 MALLOC_DEFINE(M_SUBPROC, "subproc", "Proc sub-structures");
67
68 int ps_showallprocs = 1;
69 static int ps_showallthreads = 1;
70 SYSCTL_INT(_security, OID_AUTO, ps_showallprocs, CTLFLAG_RW,
71     &ps_showallprocs, 0,
72     "Unprivileged processes can see proccesses with different UID/GID");
73 SYSCTL_INT(_security, OID_AUTO, ps_showallthreads, CTLFLAG_RW,
74     &ps_showallthreads, 0,
75     "Unprivileged processes can see kernel threads");
76
77 static void pgdelete(struct pgrp *);
78 static void orphanpg(struct pgrp *pg);
79 static pid_t proc_getnewpid_locked(int random_offset);
80
81 /*
82  * Other process lists
83  */
84 struct pidhashhead *pidhashtbl;
85 u_long pidhash;
86 struct pgrphashhead *pgrphashtbl;
87 u_long pgrphash;
88 struct proclist allproc;
89 struct proclist zombproc;
90
91 /*
92  * Random component to nextpid generation.  We mix in a random factor to make
93  * it a little harder to predict.  We sanity check the modulus value to avoid
94  * doing it in critical paths.  Don't let it be too small or we pointlessly
95  * waste randomness entropy, and don't let it be impossibly large.  Using a
96  * modulus that is too big causes a LOT more process table scans and slows
97  * down fork processing as the pidchecked caching is defeated.
98  */
99 static int randompid = 0;
100
101 /*
102  * No requirements.
103  */
104 static int
105 sysctl_kern_randompid(SYSCTL_HANDLER_ARGS)
106 {
107         int error, pid;
108
109         pid = randompid;
110         error = sysctl_handle_int(oidp, &pid, 0, req);
111         if (error || !req->newptr)
112                 return (error);
113         if (pid < 0 || pid > PID_MAX - 100)     /* out of range */
114                 pid = PID_MAX - 100;
115         else if (pid < 2)                       /* NOP */
116                 pid = 0;
117         else if (pid < 100)                     /* Make it reasonable */
118                 pid = 100;
119         randompid = pid;
120         return (error);
121 }
122
123 SYSCTL_PROC(_kern, OID_AUTO, randompid, CTLTYPE_INT|CTLFLAG_RW,
124             0, 0, sysctl_kern_randompid, "I", "Random PID modulus");
125
126 /*
127  * Initialize global process hashing structures.
128  *
129  * Called from the low level boot code only.
130  */
131 void
132 procinit(void)
133 {
134         LIST_INIT(&allproc);
135         LIST_INIT(&zombproc);
136         lwkt_init();
137         pidhashtbl = hashinit(maxproc / 4, M_PROC, &pidhash);
138         pgrphashtbl = hashinit(maxproc / 4, M_PROC, &pgrphash);
139         uihashinit();
140 }
141
142 /*
143  * Is p an inferior of the current process?
144  *
145  * No requirements.
146  * The caller must hold proc_token if the caller wishes a stable result.
147  */
148 int
149 inferior(struct proc *p)
150 {
151         lwkt_gettoken(&proc_token);
152         while (p != curproc) {
153                 if (p->p_pid == 0) {
154                         lwkt_reltoken(&proc_token);
155                         return (0);
156                 }
157                 p = p->p_pptr;
158         }
159         lwkt_reltoken(&proc_token);
160         return (1);
161 }
162
163 /*
164  * Locate a process by number
165  *
166  * XXX TODO - change API to PHOLD() the returned process ?
167  *
168  * No requirements.
169  * The caller must hold proc_token if the caller wishes a stable result.
170  */
171 struct proc *
172 pfind(pid_t pid)
173 {
174         struct proc *p;
175
176         lwkt_gettoken(&proc_token);
177         LIST_FOREACH(p, PIDHASH(pid), p_hash) {
178                 if (p->p_pid == pid) {
179                         lwkt_reltoken(&proc_token);
180                         return (p);
181                 }
182         }
183         lwkt_reltoken(&proc_token);
184         return (NULL);
185 }
186
187 /*
188  * Locate a process group by number
189  *
190  * No requirements.
191  * The caller must hold proc_token if the caller wishes a stable result.
192  */
193 struct pgrp *
194 pgfind(pid_t pgid)
195 {
196         struct pgrp *pgrp;
197
198         lwkt_gettoken(&proc_token);
199         LIST_FOREACH(pgrp, PGRPHASH(pgid), pg_hash) {
200                 if (pgrp->pg_id == pgid) {
201                         lwkt_reltoken(&proc_token);
202                         return (pgrp);
203                 }
204         }
205         lwkt_reltoken(&proc_token);
206         return (NULL);
207 }
208
209 /*
210  * Move p to a new or existing process group (and session)
211  *
212  * No requirements.
213  */
214 int
215 enterpgrp(struct proc *p, pid_t pgid, int mksess)
216 {
217         struct pgrp *pgrp;
218         int error;
219
220         lwkt_gettoken(&proc_token);
221         pgrp = pgfind(pgid);
222
223         KASSERT(pgrp == NULL || !mksess,
224                 ("enterpgrp: setsid into non-empty pgrp"));
225         KASSERT(!SESS_LEADER(p),
226                 ("enterpgrp: session leader attempted setpgrp"));
227
228         if (pgrp == NULL) {
229                 pid_t savepid = p->p_pid;
230                 struct proc *np;
231                 /*
232                  * new process group
233                  */
234                 KASSERT(p->p_pid == pgid,
235                         ("enterpgrp: new pgrp and pid != pgid"));
236                 if ((np = pfind(savepid)) == NULL || np != p) {
237                         error = ESRCH;
238                         goto fatal;
239                 }
240                 MALLOC(pgrp, struct pgrp *, sizeof(struct pgrp),
241                        M_PGRP, M_WAITOK);
242                 if (mksess) {
243                         struct session *sess;
244
245                         /*
246                          * new session
247                          */
248                         MALLOC(sess, struct session *, sizeof(struct session),
249                                M_SESSION, M_WAITOK);
250                         sess->s_leader = p;
251                         sess->s_sid = p->p_pid;
252                         sess->s_count = 1;
253                         sess->s_ttyvp = NULL;
254                         sess->s_ttyp = NULL;
255                         bcopy(p->p_session->s_login, sess->s_login,
256                               sizeof(sess->s_login));
257                         p->p_flag &= ~P_CONTROLT;
258                         pgrp->pg_session = sess;
259                         KASSERT(p == curproc,
260                                 ("enterpgrp: mksession and p != curproc"));
261                 } else {
262                         pgrp->pg_session = p->p_session;
263                         sess_hold(pgrp->pg_session);
264                 }
265                 pgrp->pg_id = pgid;
266                 LIST_INIT(&pgrp->pg_members);
267                 LIST_INSERT_HEAD(PGRPHASH(pgid), pgrp, pg_hash);
268                 pgrp->pg_jobc = 0;
269                 SLIST_INIT(&pgrp->pg_sigiolst);
270                 lockinit(&pgrp->pg_lock, "pgwt", 0, 0);
271         } else if (pgrp == p->p_pgrp) {
272                 goto done;
273         }
274
275         /*
276          * Adjust eligibility of affected pgrps to participate in job control.
277          * Increment eligibility counts before decrementing, otherwise we
278          * could reach 0 spuriously during the first call.
279          */
280         fixjobc(p, pgrp, 1);
281         fixjobc(p, p->p_pgrp, 0);
282
283         LIST_REMOVE(p, p_pglist);
284         if (LIST_EMPTY(&p->p_pgrp->pg_members))
285                 pgdelete(p->p_pgrp);
286         p->p_pgrp = pgrp;
287         LIST_INSERT_HEAD(&pgrp->pg_members, p, p_pglist);
288 done:
289         error = 0;
290 fatal:
291         lwkt_reltoken(&proc_token);
292         return (error);
293 }
294
295 /*
296  * Remove process from process group
297  *
298  * No requirements.
299  */
300 int
301 leavepgrp(struct proc *p)
302 {
303         lwkt_gettoken(&proc_token);
304         LIST_REMOVE(p, p_pglist);
305         if (LIST_EMPTY(&p->p_pgrp->pg_members))
306                 pgdelete(p->p_pgrp);
307         p->p_pgrp = NULL;
308         lwkt_reltoken(&proc_token);
309         return (0);
310 }
311
312 /*
313  * Delete a process group
314  *
315  * The caller must hold proc_token.
316  */
317 static void
318 pgdelete(struct pgrp *pgrp)
319 {
320         /*
321          * Reset any sigio structures pointing to us as a result of
322          * F_SETOWN with our pgid.
323          */
324         funsetownlst(&pgrp->pg_sigiolst);
325
326         if (pgrp->pg_session->s_ttyp != NULL &&
327             pgrp->pg_session->s_ttyp->t_pgrp == pgrp)
328                 pgrp->pg_session->s_ttyp->t_pgrp = NULL;
329         LIST_REMOVE(pgrp, pg_hash);
330         sess_rele(pgrp->pg_session);
331         kfree(pgrp, M_PGRP);
332 }
333
334 /*
335  * Adjust the ref count on a session structure.  When the ref count falls to
336  * zero the tty is disassociated from the session and the session structure
337  * is freed.  Note that tty assocation is not itself ref-counted.
338  *
339  * No requirements.
340  */
341 void
342 sess_hold(struct session *sp)
343 {
344         lwkt_gettoken(&tty_token);
345         ++sp->s_count;
346         lwkt_reltoken(&tty_token);
347 }
348
349 /*
350  * No requirements.
351  */
352 void
353 sess_rele(struct session *sp)
354 {
355         struct tty *tp;
356
357         KKASSERT(sp->s_count > 0);
358         lwkt_gettoken(&tty_token);
359         if (--sp->s_count == 0) {
360                 if (sp->s_ttyp && sp->s_ttyp->t_session) {
361 #ifdef TTY_DO_FULL_CLOSE
362                         /* FULL CLOSE, see ttyclearsession() */
363                         KKASSERT(sp->s_ttyp->t_session == sp);
364                         sp->s_ttyp->t_session = NULL;
365 #else
366                         /* HALF CLOSE, see ttyclearsession() */
367                         if (sp->s_ttyp->t_session == sp)
368                                 sp->s_ttyp->t_session = NULL;
369 #endif
370                 }
371                 if ((tp = sp->s_ttyp) != NULL) {
372                         sp->s_ttyp = NULL;
373                         ttyunhold(tp);
374                 }
375                 kfree(sp, M_SESSION);
376         }
377         lwkt_reltoken(&tty_token);
378 }
379
380 /*
381  * Adjust pgrp jobc counters when specified process changes process group.
382  * We count the number of processes in each process group that "qualify"
383  * the group for terminal job control (those with a parent in a different
384  * process group of the same session).  If that count reaches zero, the
385  * process group becomes orphaned.  Check both the specified process'
386  * process group and that of its children.
387  * entering == 0 => p is leaving specified group.
388  * entering == 1 => p is entering specified group.
389  *
390  * No requirements.
391  */
392 void
393 fixjobc(struct proc *p, struct pgrp *pgrp, int entering)
394 {
395         struct pgrp *hispgrp;
396         struct session *mysession;
397         struct proc *np;
398
399         /*
400          * Check p's parent to see whether p qualifies its own process
401          * group; if so, adjust count for p's process group.
402          */
403         lwkt_gettoken(&proc_token);
404         lwkt_gettoken(&p->p_token);     /* p_children scan */
405
406         mysession = pgrp->pg_session;
407         if ((hispgrp = p->p_pptr->p_pgrp) != pgrp &&
408             hispgrp->pg_session == mysession) {
409                 if (entering)
410                         pgrp->pg_jobc++;
411                 else if (--pgrp->pg_jobc == 0)
412                         orphanpg(pgrp);
413         }
414
415         /*
416          * Check this process' children to see whether they qualify
417          * their process groups; if so, adjust counts for children's
418          * process groups.
419          */
420         LIST_FOREACH(np, &p->p_children, p_sibling) {
421                 if ((hispgrp = np->p_pgrp) != pgrp &&
422                     hispgrp->pg_session == mysession &&
423                     np->p_stat != SZOMB) {
424                         if (entering)
425                                 hispgrp->pg_jobc++;
426                         else if (--hispgrp->pg_jobc == 0)
427                                 orphanpg(hispgrp);
428                 }
429         }
430         lwkt_reltoken(&p->p_token);
431         lwkt_reltoken(&proc_token);
432 }
433
434 /*
435  * A process group has become orphaned;
436  * if there are any stopped processes in the group,
437  * hang-up all process in that group.
438  *
439  * The caller must hold proc_token.
440  */
441 static void
442 orphanpg(struct pgrp *pg)
443 {
444         struct proc *p;
445
446         LIST_FOREACH(p, &pg->pg_members, p_pglist) {
447                 if (p->p_stat == SSTOP) {
448                         LIST_FOREACH(p, &pg->pg_members, p_pglist) {
449                                 ksignal(p, SIGHUP);
450                                 ksignal(p, SIGCONT);
451                         }
452                         return;
453                 }
454         }
455 }
456
457 /*
458  * Add a new process to the allproc list and the PID hash.  This
459  * also assigns a pid to the new process.
460  *
461  * No requirements.
462  */
463 void
464 proc_add_allproc(struct proc *p)
465 {
466         int random_offset;
467
468         if ((random_offset = randompid) != 0) {
469                 get_mplock();
470                 random_offset = karc4random() % random_offset;
471                 rel_mplock();
472         }
473
474         lwkt_gettoken(&proc_token);
475         p->p_pid = proc_getnewpid_locked(random_offset);
476         LIST_INSERT_HEAD(&allproc, p, p_list);
477         LIST_INSERT_HEAD(PIDHASH(p->p_pid), p, p_hash);
478         lwkt_reltoken(&proc_token);
479 }
480
481 /*
482  * Calculate a new process pid.  This function is integrated into
483  * proc_add_allproc() to guarentee that the new pid is not reused before
484  * the new process can be added to the allproc list.
485  *
486  * The caller must hold proc_token.
487  */
488 static
489 pid_t
490 proc_getnewpid_locked(int random_offset)
491 {
492         static pid_t nextpid;
493         static pid_t pidchecked;
494         struct proc *p;
495
496         /*
497          * Find an unused process ID.  We remember a range of unused IDs
498          * ready to use (from nextpid+1 through pidchecked-1).
499          */
500         nextpid = nextpid + 1 + random_offset;
501 retry:
502         /*
503          * If the process ID prototype has wrapped around,
504          * restart somewhat above 0, as the low-numbered procs
505          * tend to include daemons that don't exit.
506          */
507         if (nextpid >= PID_MAX) {
508                 nextpid = nextpid % PID_MAX;
509                 if (nextpid < 100)
510                         nextpid += 100;
511                 pidchecked = 0;
512         }
513         if (nextpid >= pidchecked) {
514                 int doingzomb = 0;
515
516                 pidchecked = PID_MAX;
517                 /*
518                  * Scan the active and zombie procs to check whether this pid
519                  * is in use.  Remember the lowest pid that's greater
520                  * than nextpid, so we can avoid checking for a while.
521                  */
522                 p = LIST_FIRST(&allproc);
523 again:
524                 for (; p != 0; p = LIST_NEXT(p, p_list)) {
525                         while (p->p_pid == nextpid ||
526                             p->p_pgrp->pg_id == nextpid ||
527                             p->p_session->s_sid == nextpid) {
528                                 nextpid++;
529                                 if (nextpid >= pidchecked)
530                                         goto retry;
531                         }
532                         if (p->p_pid > nextpid && pidchecked > p->p_pid)
533                                 pidchecked = p->p_pid;
534                         if (p->p_pgrp->pg_id > nextpid &&
535                             pidchecked > p->p_pgrp->pg_id)
536                                 pidchecked = p->p_pgrp->pg_id;
537                         if (p->p_session->s_sid > nextpid &&
538                             pidchecked > p->p_session->s_sid)
539                                 pidchecked = p->p_session->s_sid;
540                 }
541                 if (!doingzomb) {
542                         doingzomb = 1;
543                         p = LIST_FIRST(&zombproc);
544                         goto again;
545                 }
546         }
547         return(nextpid);
548 }
549
550 /*
551  * Called from exit1 to remove a process from the allproc
552  * list and move it to the zombie list.
553  *
554  * No requirements.
555  */
556 void
557 proc_move_allproc_zombie(struct proc *p)
558 {
559         lwkt_gettoken(&proc_token);
560         while (p->p_lock) {
561                 tsleep(p, 0, "reap1", hz / 10);
562         }
563         LIST_REMOVE(p, p_list);
564         LIST_INSERT_HEAD(&zombproc, p, p_list);
565         LIST_REMOVE(p, p_hash);
566         p->p_stat = SZOMB;
567         lwkt_reltoken(&proc_token);
568         dsched_exit_proc(p);
569 }
570
571 /*
572  * This routine is called from kern_wait() and will remove the process
573  * from the zombie list and the sibling list.  This routine will block
574  * if someone has a lock on the proces (p_lock).
575  *
576  * No requirements.
577  */
578 void
579 proc_remove_zombie(struct proc *p)
580 {
581         lwkt_gettoken(&proc_token);
582         while (p->p_lock) {
583                 tsleep(p, 0, "reap1", hz / 10);
584         }
585         LIST_REMOVE(p, p_list); /* off zombproc */
586         LIST_REMOVE(p, p_sibling);
587         lwkt_reltoken(&proc_token);
588 }
589
590 /*
591  * Scan all processes on the allproc list.  The process is automatically
592  * held for the callback.  A return value of -1 terminates the loop.
593  *
594  * No requirements.
595  * The callback is made with the process held and proc_token held.
596  */
597 void
598 allproc_scan(int (*callback)(struct proc *, void *), void *data)
599 {
600         struct proc *p;
601         int r;
602
603         lwkt_gettoken(&proc_token);
604         LIST_FOREACH(p, &allproc, p_list) {
605                 PHOLD(p);
606                 r = callback(p, data);
607                 PRELE(p);
608                 if (r < 0)
609                         break;
610         }
611         lwkt_reltoken(&proc_token);
612 }
613
614 /*
615  * Scan all lwps of processes on the allproc list.  The lwp is automatically
616  * held for the callback.  A return value of -1 terminates the loop.
617  *
618  * No requirements.
619  * The callback is made with the proces and lwp both held, and proc_token held.
620  */
621 void
622 alllwp_scan(int (*callback)(struct lwp *, void *), void *data)
623 {
624         struct proc *p;
625         struct lwp *lp;
626         int r = 0;
627
628         lwkt_gettoken(&proc_token);
629         LIST_FOREACH(p, &allproc, p_list) {
630                 PHOLD(p);
631                 FOREACH_LWP_IN_PROC(lp, p) {
632                         LWPHOLD(lp);
633                         r = callback(lp, data);
634                         LWPRELE(lp);
635                 }
636                 PRELE(p);
637                 if (r < 0)
638                         break;
639         }
640         lwkt_reltoken(&proc_token);
641 }
642
643 /*
644  * Scan all processes on the zombproc list.  The process is automatically
645  * held for the callback.  A return value of -1 terminates the loop.
646  *
647  * No requirements.
648  * The callback is made with the proces held and proc_token held.
649  */
650 void
651 zombproc_scan(int (*callback)(struct proc *, void *), void *data)
652 {
653         struct proc *p;
654         int r;
655
656         lwkt_gettoken(&proc_token);
657         LIST_FOREACH(p, &zombproc, p_list) {
658                 PHOLD(p);
659                 r = callback(p, data);
660                 PRELE(p);
661                 if (r < 0)
662                         break;
663         }
664         lwkt_reltoken(&proc_token);
665 }
666
667 #include "opt_ddb.h"
668 #ifdef DDB
669 #include <ddb/ddb.h>
670
671 /*
672  * Debugging only
673  */
674 DB_SHOW_COMMAND(pgrpdump, pgrpdump)
675 {
676         struct pgrp *pgrp;
677         struct proc *p;
678         int i;
679
680         for (i = 0; i <= pgrphash; i++) {
681                 if (!LIST_EMPTY(&pgrphashtbl[i])) {
682                         kprintf("\tindx %d\n", i);
683                         LIST_FOREACH(pgrp, &pgrphashtbl[i], pg_hash) {
684                                 kprintf(
685                         "\tpgrp %p, pgid %ld, sess %p, sesscnt %d, mem %p\n",
686                                     (void *)pgrp, (long)pgrp->pg_id,
687                                     (void *)pgrp->pg_session,
688                                     pgrp->pg_session->s_count,
689                                     (void *)LIST_FIRST(&pgrp->pg_members));
690                                 LIST_FOREACH(p, &pgrp->pg_members, p_pglist) {
691                                         kprintf("\t\tpid %ld addr %p pgrp %p\n", 
692                                             (long)p->p_pid, (void *)p,
693                                             (void *)p->p_pgrp);
694                                 }
695                         }
696                 }
697         }
698 }
699 #endif /* DDB */
700
701 /*
702  * Locate a process on the zombie list.  Return a process or NULL.
703  *
704  * The caller must hold proc_token if a stable result is desired.
705  * No other requirements.
706  */
707 struct proc *
708 zpfind(pid_t pid)
709 {
710         struct proc *p;
711
712         lwkt_gettoken(&proc_token);
713         LIST_FOREACH(p, &zombproc, p_list) {
714                 if (p->p_pid == pid) {
715                         lwkt_reltoken(&proc_token);
716                         return (p);
717                 }
718         }
719         lwkt_reltoken(&proc_token);
720         return (NULL);
721 }
722
723 /*
724  * The caller must hold proc_token.
725  */
726 static int
727 sysctl_out_proc(struct proc *p, struct sysctl_req *req, int flags)
728 {
729         struct kinfo_proc ki;
730         struct lwp *lp;
731         int skp = 0, had_output = 0;
732         int error;
733
734         bzero(&ki, sizeof(ki));
735         fill_kinfo_proc(p, &ki);
736         if ((flags & KERN_PROC_FLAG_LWP) == 0)
737                 skp = 1;
738         error = 0;
739         FOREACH_LWP_IN_PROC(lp, p) {
740                 LWPHOLD(lp);
741                 fill_kinfo_lwp(lp, &ki.kp_lwp);
742                 had_output = 1;
743                 error = SYSCTL_OUT(req, &ki, sizeof(ki));
744                 LWPRELE(lp);
745                 if (error)
746                         break;
747                 if (skp)
748                         break;
749         }
750         /* We need to output at least the proc, even if there is no lwp. */
751         if (had_output == 0) {
752                 error = SYSCTL_OUT(req, &ki, sizeof(ki));
753         }
754         return (error);
755 }
756
757 /*
758  * The caller must hold proc_token.
759  */
760 static int
761 sysctl_out_proc_kthread(struct thread *td, struct sysctl_req *req, int flags)
762 {
763         struct kinfo_proc ki;
764         int error;
765
766         fill_kinfo_proc_kthread(td, &ki);
767         error = SYSCTL_OUT(req, &ki, sizeof(ki));
768         if (error)
769                 return error;
770         return(0);
771 }
772
773 /*
774  * No requirements.
775  */
776 static int
777 sysctl_kern_proc(SYSCTL_HANDLER_ARGS)
778 {
779         int *name = (int*) arg1;
780         int oid = oidp->oid_number;
781         u_int namelen = arg2;
782         struct proc *p;
783         struct proclist *plist;
784         struct thread *td;
785         int doingzomb, flags = 0;
786         int error = 0;
787         int n;
788         int origcpu;
789         struct ucred *cr1 = curproc->p_ucred;
790
791         flags = oid & KERN_PROC_FLAGMASK;
792         oid &= ~KERN_PROC_FLAGMASK;
793
794         if ((oid == KERN_PROC_ALL && namelen != 0) ||
795             (oid != KERN_PROC_ALL && namelen != 1))
796                 return (EINVAL);
797
798         lwkt_gettoken(&proc_token);
799         if (oid == KERN_PROC_PID) {
800                 p = pfind((pid_t)name[0]);
801                 if (p == NULL)
802                         goto post_threads;
803                 if (!PRISON_CHECK(cr1, p->p_ucred))
804                         goto post_threads;
805                 PHOLD(p);
806                 error = sysctl_out_proc(p, req, flags);
807                 PRELE(p);
808                 goto post_threads;
809         }
810
811         if (!req->oldptr) {
812                 /* overestimate by 5 procs */
813                 error = SYSCTL_OUT(req, 0, sizeof (struct kinfo_proc) * 5);
814                 if (error)
815                         goto post_threads;
816         }
817         for (doingzomb = 0; doingzomb <= 1; doingzomb++) {
818                 if (doingzomb)
819                         plist = &zombproc;
820                 else
821                         plist = &allproc;
822                 LIST_FOREACH(p, plist, p_list) {
823                         /*
824                          * Show a user only their processes.
825                          */
826                         if ((!ps_showallprocs) && p_trespass(cr1, p->p_ucred))
827                                 continue;
828                         /*
829                          * Skip embryonic processes.
830                          */
831                         if (p->p_stat == SIDL)
832                                 continue;
833                         /*
834                          * TODO - make more efficient (see notes below).
835                          * do by session.
836                          */
837                         switch (oid) {
838                         case KERN_PROC_PGRP:
839                                 /* could do this by traversing pgrp */
840                                 if (p->p_pgrp == NULL || 
841                                     p->p_pgrp->pg_id != (pid_t)name[0])
842                                         continue;
843                                 break;
844
845                         case KERN_PROC_TTY:
846                                 if ((p->p_flag & P_CONTROLT) == 0 ||
847                                     p->p_session == NULL ||
848                                     p->p_session->s_ttyp == NULL ||
849                                     dev2udev(p->p_session->s_ttyp->t_dev) != 
850                                         (udev_t)name[0])
851                                         continue;
852                                 break;
853
854                         case KERN_PROC_UID:
855                                 if (p->p_ucred == NULL || 
856                                     p->p_ucred->cr_uid != (uid_t)name[0])
857                                         continue;
858                                 break;
859
860                         case KERN_PROC_RUID:
861                                 if (p->p_ucred == NULL || 
862                                     p->p_ucred->cr_ruid != (uid_t)name[0])
863                                         continue;
864                                 break;
865                         }
866
867                         if (!PRISON_CHECK(cr1, p->p_ucred))
868                                 continue;
869                         PHOLD(p);
870                         error = sysctl_out_proc(p, req, flags);
871                         PRELE(p);
872                         if (error)
873                                 goto post_threads;
874                 }
875         }
876
877         /*
878          * Iterate over all active cpus and scan their thread list.  Start
879          * with the next logical cpu and end with our original cpu.  We
880          * migrate our own thread to each target cpu in order to safely scan
881          * its thread list.  In the last loop we migrate back to our original
882          * cpu.
883          */
884         origcpu = mycpu->gd_cpuid;
885         if (!ps_showallthreads || jailed(cr1))
886                 goto post_threads;
887
888         for (n = 1; n <= ncpus; ++n) {
889                 globaldata_t rgd;
890                 int nid;
891
892                 nid = (origcpu + n) % ncpus;
893                 if ((smp_active_mask & CPUMASK(nid)) == 0)
894                         continue;
895                 rgd = globaldata_find(nid);
896                 lwkt_setcpu_self(rgd);
897
898                 TAILQ_FOREACH(td, &mycpu->gd_tdallq, td_allq) {
899                         if (td->td_proc)
900                                 continue;
901                         switch (oid) {
902                         case KERN_PROC_PGRP:
903                         case KERN_PROC_TTY:
904                         case KERN_PROC_UID:
905                         case KERN_PROC_RUID:
906                                 continue;
907                         default:
908                                 break;
909                         }
910                         lwkt_hold(td);
911                         error = sysctl_out_proc_kthread(td, req, doingzomb);
912                         lwkt_rele(td);
913                         if (error)
914                                 goto post_threads;
915                 }
916         }
917 post_threads:
918         lwkt_reltoken(&proc_token);
919         return (error);
920 }
921
922 /*
923  * This sysctl allows a process to retrieve the argument list or process
924  * title for another process without groping around in the address space
925  * of the other process.  It also allow a process to set its own "process 
926  * title to a string of its own choice.
927  *
928  * No requirements.
929  */
930 static int
931 sysctl_kern_proc_args(SYSCTL_HANDLER_ARGS)
932 {
933         int *name = (int*) arg1;
934         u_int namelen = arg2;
935         struct proc *p;
936         struct pargs *pa;
937         int error = 0;
938         struct ucred *cr1 = curproc->p_ucred;
939
940         if (namelen != 1) 
941                 return (EINVAL);
942
943         lwkt_gettoken(&proc_token);
944         p = pfind((pid_t)name[0]);
945         if (p == NULL)
946                 goto done;
947
948         if ((!ps_argsopen) && p_trespass(cr1, p->p_ucred))
949                 goto done;
950
951         if (req->newptr && curproc != p) {
952                 error = EPERM;
953                 goto done;
954         }
955
956         PHOLD(p);
957         if (req->oldptr && p->p_args != NULL) {
958                 error = SYSCTL_OUT(req, p->p_args->ar_args,
959                                    p->p_args->ar_length);
960         }
961         if (req->newptr == NULL) {
962                 PRELE(p);
963                 goto done;
964         }
965
966         if (p->p_args && --p->p_args->ar_ref == 0) 
967                 FREE(p->p_args, M_PARGS);
968         p->p_args = NULL;
969
970         if (req->newlen + sizeof(struct pargs) > ps_arg_cache_limit) {
971                 PRELE(p);
972                 goto done;
973         }
974
975         MALLOC(pa, struct pargs *, sizeof(struct pargs) + req->newlen, 
976                M_PARGS, M_WAITOK);
977         pa->ar_ref = 1;
978         pa->ar_length = req->newlen;
979         error = SYSCTL_IN(req, pa->ar_args, req->newlen);
980         if (!error)
981                 p->p_args = pa;
982         else
983                 FREE(pa, M_PARGS);
984         PRELE(p);
985 done:
986         lwkt_reltoken(&proc_token);
987         return (error);
988 }
989
990 static int
991 sysctl_kern_proc_cwd(SYSCTL_HANDLER_ARGS)
992 {
993         int *name = (int*) arg1;
994         u_int namelen = arg2;
995         struct proc *p;
996         int error = 0;
997         char *fullpath, *freepath;
998         struct ucred *cr1 = curproc->p_ucred;
999
1000         if (namelen != 1) 
1001                 return (EINVAL);
1002
1003         lwkt_gettoken(&proc_token);
1004         p = pfind((pid_t)name[0]);
1005         if (p == NULL)
1006                 goto done;
1007
1008         /*
1009          * If we are not allowed to see other args, we certainly shouldn't
1010          * get the cwd either. Also check the usual trespassing.
1011          */
1012         if ((!ps_argsopen) && p_trespass(cr1, p->p_ucred))
1013                 goto done;
1014
1015         PHOLD(p);
1016         if (req->oldptr && p->p_fd != NULL) {
1017                 error = cache_fullpath(p, &p->p_fd->fd_ncdir,
1018                     &fullpath, &freepath, 0);
1019                 if (error)
1020                         goto done;
1021                 error = SYSCTL_OUT(req, fullpath, strlen(fullpath) + 1);
1022                 kfree(freepath, M_TEMP);
1023         }
1024
1025         PRELE(p);
1026
1027 done:
1028         lwkt_reltoken(&proc_token);
1029         return (error);
1030 }
1031
1032 SYSCTL_NODE(_kern, KERN_PROC, proc, CTLFLAG_RD,  0, "Process table");
1033
1034 SYSCTL_PROC(_kern_proc, KERN_PROC_ALL, all, CTLFLAG_RD|CTLTYPE_STRUCT,
1035         0, 0, sysctl_kern_proc, "S,proc", "Return entire process table");
1036
1037 SYSCTL_NODE(_kern_proc, KERN_PROC_PGRP, pgrp, CTLFLAG_RD, 
1038         sysctl_kern_proc, "Process table");
1039
1040 SYSCTL_NODE(_kern_proc, KERN_PROC_TTY, tty, CTLFLAG_RD, 
1041         sysctl_kern_proc, "Process table");
1042
1043 SYSCTL_NODE(_kern_proc, KERN_PROC_UID, uid, CTLFLAG_RD, 
1044         sysctl_kern_proc, "Process table");
1045
1046 SYSCTL_NODE(_kern_proc, KERN_PROC_RUID, ruid, CTLFLAG_RD, 
1047         sysctl_kern_proc, "Process table");
1048
1049 SYSCTL_NODE(_kern_proc, KERN_PROC_PID, pid, CTLFLAG_RD, 
1050         sysctl_kern_proc, "Process table");
1051
1052 SYSCTL_NODE(_kern_proc, (KERN_PROC_ALL | KERN_PROC_FLAG_LWP), all_lwp, CTLFLAG_RD,
1053         sysctl_kern_proc, "Process table");
1054
1055 SYSCTL_NODE(_kern_proc, (KERN_PROC_PGRP | KERN_PROC_FLAG_LWP), pgrp_lwp, CTLFLAG_RD, 
1056         sysctl_kern_proc, "Process table");
1057
1058 SYSCTL_NODE(_kern_proc, (KERN_PROC_TTY | KERN_PROC_FLAG_LWP), tty_lwp, CTLFLAG_RD, 
1059         sysctl_kern_proc, "Process table");
1060
1061 SYSCTL_NODE(_kern_proc, (KERN_PROC_UID | KERN_PROC_FLAG_LWP), uid_lwp, CTLFLAG_RD, 
1062         sysctl_kern_proc, "Process table");
1063
1064 SYSCTL_NODE(_kern_proc, (KERN_PROC_RUID | KERN_PROC_FLAG_LWP), ruid_lwp, CTLFLAG_RD, 
1065         sysctl_kern_proc, "Process table");
1066
1067 SYSCTL_NODE(_kern_proc, (KERN_PROC_PID | KERN_PROC_FLAG_LWP), pid_lwp, CTLFLAG_RD, 
1068         sysctl_kern_proc, "Process table");
1069
1070 SYSCTL_NODE(_kern_proc, KERN_PROC_ARGS, args, CTLFLAG_RW | CTLFLAG_ANYBODY,
1071         sysctl_kern_proc_args, "Process argument list");
1072
1073 SYSCTL_NODE(_kern_proc, KERN_PROC_CWD, cwd, CTLFLAG_RD | CTLFLAG_ANYBODY,
1074         sysctl_kern_proc_cwd, "Process argument list");