sys/kern/kern_proc.c

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