kernel - Add callout debugging
[dragonfly.git] / sys / kern / kern_exit.c
1 /*
2  * Copyright (c) 1982, 1986, 1989, 1991, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  * (c) UNIX System Laboratories, Inc.
5  * All or some portions of this file are derived from material licensed
6  * to the University of California by American Telephone and Telegraph
7  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
8  * the permission of UNIX System Laboratories, Inc.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  *      @(#)kern_exit.c 8.7 (Berkeley) 2/12/94
35  * $FreeBSD: src/sys/kern/kern_exit.c,v 1.92.2.11 2003/01/13 22:51:16 dillon Exp $
36  */
37
38 #include "opt_ktrace.h"
39
40 #include <sys/param.h>
41 #include <sys/systm.h>
42 #include <sys/sysproto.h>
43 #include <sys/kernel.h>
44 #include <sys/malloc.h>
45 #include <sys/proc.h>
46 #include <sys/ktrace.h>
47 #include <sys/pioctl.h>
48 #include <sys/tty.h>
49 #include <sys/wait.h>
50 #include <sys/vnode.h>
51 #include <sys/resourcevar.h>
52 #include <sys/signalvar.h>
53 #include <sys/taskqueue.h>
54 #include <sys/ptrace.h>
55 #include <sys/acct.h>           /* for acct_process() function prototype */
56 #include <sys/filedesc.h>
57 #include <sys/shm.h>
58 #include <sys/sem.h>
59 #include <sys/jail.h>
60 #include <sys/kern_syscall.h>
61 #include <sys/unistd.h>
62 #include <sys/eventhandler.h>
63 #include <sys/dsched.h>
64
65 #include <vm/vm.h>
66 #include <vm/vm_param.h>
67 #include <sys/lock.h>
68 #include <vm/pmap.h>
69 #include <vm/vm_map.h>
70 #include <vm/vm_extern.h>
71 #include <sys/user.h>
72
73 #include <sys/refcount.h>
74 #include <sys/spinlock2.h>
75 #include <sys/mplock2.h>
76
77 #include <machine/vmm.h>
78
79 static void reaplwps(void *context, int dummy);
80 static void reaplwp(struct lwp *lp);
81 static void killlwps(struct lwp *lp);
82
83 static MALLOC_DEFINE(M_ATEXIT, "atexit", "atexit callback");
84
85 /*
86  * callout list for things to do at exit time
87  */
88 struct exitlist {
89         exitlist_fn function;
90         TAILQ_ENTRY(exitlist) next;
91 };
92
93 TAILQ_HEAD(exit_list_head, exitlist);
94 static struct exit_list_head exit_list = TAILQ_HEAD_INITIALIZER(exit_list);
95
96 /*
97  * LWP reaper data
98  */
99 static struct task *deadlwp_task[MAXCPU];
100 static struct lwplist deadlwp_list[MAXCPU];
101 static struct lwkt_token deadlwp_token[MAXCPU];
102
103 /*
104  * exit --
105  *      Death of process.
106  *
107  * SYS_EXIT_ARGS(int rval)
108  */
109 int
110 sys_exit(struct exit_args *uap)
111 {
112         exit1(W_EXITCODE(uap->rval, 0));
113         /* NOTREACHED */
114 }
115
116 /*
117  * Extended exit --
118  *      Death of a lwp or process with optional bells and whistles.
119  */
120 int
121 sys_extexit(struct extexit_args *uap)
122 {
123         struct proc *p = curproc;
124         int action, who;
125         int error;
126
127         action = EXTEXIT_ACTION(uap->how);
128         who = EXTEXIT_WHO(uap->how);
129
130         /* Check parameters before we might perform some action */
131         switch (who) {
132         case EXTEXIT_PROC:
133         case EXTEXIT_LWP:
134                 break;
135         default:
136                 return (EINVAL);
137         }
138
139         switch (action) {
140         case EXTEXIT_SIMPLE:
141                 break;
142         case EXTEXIT_SETINT:
143                 error = copyout(&uap->status, uap->addr, sizeof(uap->status));
144                 if (error)
145                         return (error);
146                 break;
147         default:
148                 return (EINVAL);
149         }
150
151         lwkt_gettoken(&p->p_token);
152
153         switch (who) {
154         case EXTEXIT_LWP:
155                 /*
156                  * Be sure only to perform a simple lwp exit if there is at
157                  * least one more lwp in the proc, which will call exit1()
158                  * later, otherwise the proc will be an UNDEAD and not even a
159                  * SZOMB!
160                  */
161                 if (p->p_nthreads > 1) {
162                         lwp_exit(0, NULL);      /* called w/ p_token held */
163                         /* NOT REACHED */
164                 }
165                 /* else last lwp in proc:  do the real thing */
166                 /* FALLTHROUGH */
167         default:        /* to help gcc */
168         case EXTEXIT_PROC:
169                 lwkt_reltoken(&p->p_token);
170                 exit1(W_EXITCODE(uap->status, 0));
171                 /* NOTREACHED */
172         }
173
174         /* NOTREACHED */
175         lwkt_reltoken(&p->p_token);     /* safety */
176 }
177
178 /*
179  * Kill all lwps associated with the current process except the
180  * current lwp.   Return an error if we race another thread trying to
181  * do the same thing and lose the race.
182  *
183  * If forexec is non-zero the current thread and process flags are
184  * cleaned up so they can be reused.
185  */
186 int
187 killalllwps(int forexec)
188 {
189         struct lwp *lp = curthread->td_lwp;
190         struct proc *p = lp->lwp_proc;
191         int fakestop;
192
193         /*
194          * Interlock against P_WEXIT.  Only one of the process's thread
195          * is allowed to do the master exit.
196          */
197         lwkt_gettoken(&p->p_token);
198         if (p->p_flags & P_WEXIT) {
199                 lwkt_reltoken(&p->p_token);
200                 return (EALREADY);
201         }
202         p->p_flags |= P_WEXIT;
203         lwkt_gettoken(&lp->lwp_token);
204
205         /*
206          * Set temporary stopped state in case we are racing a coredump.
207          * Otherwise the coredump may hang forever.
208          */
209         if (lp->lwp_mpflags & LWP_MP_WSTOP) {
210                 fakestop = 0;
211         } else {
212                 atomic_set_int(&lp->lwp_mpflags, LWP_MP_WSTOP);
213                 ++p->p_nstopped;
214                 fakestop = 1;
215                 wakeup(&p->p_nstopped);
216         }
217
218         /*
219          * Interlock with LWP_MP_WEXIT and kill any remaining LWPs
220          */
221         atomic_set_int(&lp->lwp_mpflags, LWP_MP_WEXIT);
222         if (p->p_nthreads > 1)
223                 killlwps(lp);
224
225         /*
226          * Undo temporary stopped state
227          */
228         if (fakestop && (lp->lwp_mpflags & LWP_MP_WSTOP)) {
229                 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_WSTOP);
230                 --p->p_nstopped;
231         }
232
233         /*
234          * If doing this for an exec, clean up the remaining thread
235          * (us) for continuing operation after all the other threads
236          * have been killed.
237          */
238         if (forexec) {
239                 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_WEXIT);
240                 p->p_flags &= ~P_WEXIT;
241         }
242         lwkt_reltoken(&lp->lwp_token);
243         lwkt_reltoken(&p->p_token);
244
245         return(0);
246 }
247
248 /*
249  * Kill all LWPs except the current one.  Do not try to signal
250  * LWPs which have exited on their own or have already been
251  * signaled.
252  */
253 static void
254 killlwps(struct lwp *lp)
255 {
256         struct proc *p = lp->lwp_proc;
257         struct lwp *tlp;
258
259         /*
260          * Kill the remaining LWPs.  We must send the signal before setting
261          * LWP_MP_WEXIT.  The setting of WEXIT is optional but helps reduce
262          * races.  tlp must be held across the call as it might block and
263          * allow the target lwp to rip itself out from under our loop.
264          */
265         FOREACH_LWP_IN_PROC(tlp, p) {
266                 LWPHOLD(tlp);
267                 lwkt_gettoken(&tlp->lwp_token);
268                 if ((tlp->lwp_mpflags & LWP_MP_WEXIT) == 0) {
269                         atomic_set_int(&tlp->lwp_mpflags, LWP_MP_WEXIT);
270                         lwpsignal(p, tlp, SIGKILL);
271                 }
272                 lwkt_reltoken(&tlp->lwp_token);
273                 LWPRELE(tlp);
274         }
275
276         /*
277          * Wait for everything to clear out.  Also make sure any tstop()s
278          * are signalled (we are holding p_token for the interlock).
279          */
280         wakeup(p);
281         while (p->p_nthreads > 1)
282                 tsleep(&p->p_nthreads, 0, "killlwps", 0);
283 }
284
285 /*
286  * Exit: deallocate address space and other resources, change proc state
287  * to zombie, and unlink proc from allproc and parent's lists.  Save exit
288  * status and rusage for wait().  Check for child processes and orphan them.
289  */
290 void
291 exit1(int rv)
292 {
293         struct thread *td = curthread;
294         struct proc *p = td->td_proc;
295         struct lwp *lp = td->td_lwp;
296         struct proc *q;
297         struct proc *pp;
298         struct proc *reproc;
299         struct sysreaper *reap;
300         struct vmspace *vm;
301         struct vnode *vtmp;
302         struct exitlist *ep;
303         int error;
304
305         lwkt_gettoken(&p->p_token);
306
307         if (p->p_pid == 1) {
308                 kprintf("init died (signal %d, exit %d)\n",
309                     WTERMSIG(rv), WEXITSTATUS(rv));
310                 panic("Going nowhere without my init!");
311         }
312         varsymset_clean(&p->p_varsymset);
313         lockuninit(&p->p_varsymset.vx_lock);
314
315         /*
316          * Kill all lwps associated with the current process, return an
317          * error if we race another thread trying to do the same thing
318          * and lose the race.
319          */
320         error = killalllwps(0);
321         if (error) {
322                 lwp_exit(0, NULL);
323                 /* NOT REACHED */
324         }
325
326         /* are we a task leader? */
327         if (p == p->p_leader) {
328                 struct kill_args killArgs;
329                 killArgs.signum = SIGKILL;
330                 q = p->p_peers;
331                 while(q) {
332                         killArgs.pid = q->p_pid;
333                         /*
334                          * The interface for kill is better
335                          * than the internal signal
336                          */
337                         sys_kill(&killArgs);
338                         q = q->p_peers;
339                 }
340                 while (p->p_peers) 
341                         tsleep((caddr_t)p, 0, "exit1", 0);
342         }
343
344 #ifdef PGINPROF
345         vmsizmon();
346 #endif
347         STOPEVENT(p, S_EXIT, rv);
348         p->p_flags |= P_POSTEXIT;       /* stop procfs stepping */
349
350         /* 
351          * Check if any loadable modules need anything done at process exit.
352          * e.g. SYSV IPC stuff
353          * XXX what if one of these generates an error?
354          */
355         p->p_xstat = rv;
356
357         /*
358          * XXX: imho, the eventhandler stuff is much cleaner than this.
359          *      Maybe we should move everything to use eventhandler.
360          */
361         TAILQ_FOREACH(ep, &exit_list, next) 
362                 (*ep->function)(td);
363
364         if (p->p_flags & P_PROFIL)
365                 stopprofclock(p);
366
367         SIGEMPTYSET(p->p_siglist);
368         SIGEMPTYSET(lp->lwp_siglist);
369         if (timevalisset(&p->p_realtimer.it_value))
370                 callout_stop_sync(&p->p_ithandle);
371
372         /*
373          * Reset any sigio structures pointing to us as a result of
374          * F_SETOWN with our pid.
375          */
376         funsetownlst(&p->p_sigiolst);
377
378         /*
379          * Close open files and release open-file table.
380          * This may block!
381          */
382         fdfree(p, NULL);
383
384         if (p->p_leader->p_peers) {
385                 q = p->p_leader;
386                 while(q->p_peers != p)
387                         q = q->p_peers;
388                 q->p_peers = p->p_peers;
389                 wakeup((caddr_t)p->p_leader);
390         }
391
392         /*
393          * XXX Shutdown SYSV semaphores
394          */
395         semexit(p);
396
397         /* The next two chunks should probably be moved to vmspace_exit. */
398         vm = p->p_vmspace;
399
400         /*
401          * Clean up data related to virtual kernel operation.  Clean up
402          * any vkernel context related to the current lwp now so we can
403          * destroy p_vkernel.
404          */
405         if (p->p_vkernel) {
406                 vkernel_lwp_exit(lp);
407                 vkernel_exit(p);
408         }
409
410         /*
411          * Release the user portion of address space.  The exitbump prevents
412          * the vmspace from being completely eradicated (using holdcnt).
413          * This releases references to vnodes, which could cause I/O if the
414          * file has been unlinked.  We need to do this early enough that
415          * we can still sleep.
416          *
417          * We can't free the entire vmspace as the kernel stack may be mapped
418          * within that space also.
419          *
420          * Processes sharing the same vmspace may exit in one order, and
421          * get cleaned up by vmspace_exit() in a different order.  The
422          * last exiting process to reach this point releases as much of
423          * the environment as it can, and the last process cleaned up
424          * by vmspace_exit() (which decrements exitingcnt) cleans up the
425          * remainder.
426          *
427          * NOTE: Releasing p_token around this call is helpful if the
428          *       vmspace had a huge RSS.  Otherwise some other process
429          *       trying to do an allproc or other scan (like 'ps') may
430          *       stall for a long time.
431          */
432         lwkt_reltoken(&p->p_token);
433         vmspace_relexit(vm);
434         lwkt_gettoken(&p->p_token);
435
436         if (SESS_LEADER(p)) {
437                 struct session *sp = p->p_session;
438
439                 if (sp->s_ttyvp) {
440                         /*
441                          * We are the controlling process.  Signal the 
442                          * foreground process group, drain the controlling
443                          * terminal, and revoke access to the controlling
444                          * terminal.
445                          *
446                          * NOTE: while waiting for the process group to exit
447                          * it is possible that one of the processes in the
448                          * group will revoke the tty, so the ttyclosesession()
449                          * function will re-check sp->s_ttyvp.
450                          */
451                         if (sp->s_ttyp && (sp->s_ttyp->t_session == sp)) {
452                                 if (sp->s_ttyp->t_pgrp)
453                                         pgsignal(sp->s_ttyp->t_pgrp, SIGHUP, 1);
454                                 ttywait(sp->s_ttyp);
455                                 ttyclosesession(sp, 1); /* also revoke */
456                         }
457                         /*
458                          * Release the tty.  If someone has it open via
459                          * /dev/tty then close it (since they no longer can
460                          * once we've NULL'd it out).
461                          */
462                         ttyclosesession(sp, 0);
463
464                         /*
465                          * s_ttyp is not zero'd; we use this to indicate
466                          * that the session once had a controlling terminal.
467                          * (for logging and informational purposes)
468                          */
469                 }
470                 sp->s_leader = NULL;
471         }
472         fixjobc(p, p->p_pgrp, 0);
473         (void)acct_process(p);
474 #ifdef KTRACE
475         /*
476          * release trace file
477          */
478         if (p->p_tracenode)
479                 ktrdestroy(&p->p_tracenode);
480         p->p_traceflag = 0;
481 #endif
482         /*
483          * Release reference to text vnode
484          */
485         if ((vtmp = p->p_textvp) != NULL) {
486                 p->p_textvp = NULL;
487                 vrele(vtmp);
488         }
489
490         /* Release namecache handle to text file */
491         if (p->p_textnch.ncp)
492                 cache_drop(&p->p_textnch);
493
494         /*
495          * We have to handle PPWAIT here or proc_move_allproc_zombie()
496          * will block on the PHOLD() the parent is doing.
497          *
498          * We are using the flag as an interlock so an atomic op is
499          * necessary to synchronize with the parent's cpu.
500          */
501         if (p->p_flags & P_PPWAIT) {
502                 if (p->p_pptr && p->p_pptr->p_upmap)
503                         atomic_add_int(&p->p_pptr->p_upmap->invfork, -1);
504                 atomic_clear_int(&p->p_flags, P_PPWAIT);
505                 wakeup(p->p_pptr);
506         }
507
508         /*
509          * Move the process to the zombie list.  This will block
510          * until the process p_lock count reaches 0.  The process will
511          * not be reaped until TDF_EXITING is set by cpu_thread_exit(),
512          * which is called from cpu_proc_exit().
513          *
514          * Interlock against waiters using p_waitgen.  We increment
515          * p_waitgen after completing the move of our process to the
516          * zombie list.
517          *
518          * WARNING: pp becomes stale when we block, clear it now as a
519          *          reminder.
520          */
521         proc_move_allproc_zombie(p);
522         pp = p->p_pptr;
523         atomic_add_long(&pp->p_waitgen, 1);
524         pp = NULL;
525
526         /*
527          * release controlled reaper for exit if we own it and return the
528          * remaining reaper (the one for us), which we will drop after we
529          * are done.
530          */
531         reap = reaper_exit(p);
532
533         /*
534          * Reparent all of this process's children to the init process or
535          * to the designated reaper.  We must hold the reaper's p_token in
536          * order to safely mess with p_children.
537          *
538          * We already hold p->p_token (to remove the children from our list).
539          */
540         reproc = NULL;
541         q = LIST_FIRST(&p->p_children);
542         if (q) {
543                 reproc = reaper_get(reap);
544                 lwkt_gettoken(&reproc->p_token);
545                 while ((q = LIST_FIRST(&p->p_children)) != NULL) {
546                         PHOLD(q);
547                         lwkt_gettoken(&q->p_token);
548                         if (q != LIST_FIRST(&p->p_children)) {
549                                 lwkt_reltoken(&q->p_token);
550                                 PRELE(q);
551                                 continue;
552                         }
553                         LIST_REMOVE(q, p_sibling);
554                         LIST_INSERT_HEAD(&reproc->p_children, q, p_sibling);
555                         q->p_pptr = reproc;
556                         q->p_ppid = reproc->p_pid;
557                         q->p_sigparent = SIGCHLD;
558
559                         /*
560                          * Traced processes are killed
561                          * since their existence means someone is screwing up.
562                          */
563                         if (q->p_flags & P_TRACED) {
564                                 q->p_flags &= ~P_TRACED;
565                                 ksignal(q, SIGKILL);
566                         }
567                         lwkt_reltoken(&q->p_token);
568                         PRELE(q);
569                 }
570                 lwkt_reltoken(&reproc->p_token);
571                 wakeup(reproc);
572         }
573
574         /*
575          * Save exit status and final rusage info.  We no longer add
576          * child rusage info into self times, wait4() and kern_wait()
577          * handles it in order to properly support wait6().
578          */
579         calcru_proc(p, &p->p_ru);
580         /*ruadd(&p->p_ru, &p->p_cru); REMOVED */
581
582         /*
583          * notify interested parties of our demise.
584          */
585         KNOTE(&p->p_klist, NOTE_EXIT);
586
587         /*
588          * Notify parent that we're gone.  If parent has the PS_NOCLDWAIT
589          * flag set, or if the handler is set to SIG_IGN, notify the reaper
590          * instead (it will handle this situation).
591          *
592          * NOTE: The reaper can still be the parent process.
593          *
594          * (must reload pp)
595          */
596         if (p->p_pptr->p_sigacts->ps_flag & (PS_NOCLDWAIT | PS_CLDSIGIGN)) {
597                 if (reproc == NULL)
598                         reproc = reaper_get(reap);
599                 proc_reparent(p, reproc);
600         }
601         if (reproc)
602                 PRELE(reproc);
603         if (reap)
604                 reaper_drop(reap);
605
606         /*
607          * Signal (possibly new) parent.
608          */
609         pp = p->p_pptr;
610         PHOLD(pp);
611         if (p->p_sigparent && pp != initproc) {
612                 int sig = p->p_sigparent;
613
614                 if (sig != SIGUSR1 && sig != SIGCHLD)
615                         sig = SIGCHLD;
616                 ksignal(pp, sig);
617         } else {
618                 ksignal(pp, SIGCHLD);
619         }
620         p->p_flags &= ~P_TRACED;
621         PRELE(pp);
622
623         /*
624          * cpu_exit is responsible for clearing curproc, since
625          * it is heavily integrated with the thread/switching sequence.
626          *
627          * Other substructures are freed from wait().
628          */
629         if (p->p_limit) {
630                 struct plimit *rlimit;
631
632                 rlimit = p->p_limit;
633                 p->p_limit = NULL;
634                 plimit_free(rlimit);
635         }
636
637         /*
638          * Finally, call machine-dependent code to release as many of the
639          * lwp's resources as we can and halt execution of this thread.
640          *
641          * pp is a wild pointer now but still the correct wakeup() target.
642          * lwp_exit() only uses it to send the wakeup() signal to the likely
643          * parent.  Any reparenting race that occurs will get a signal
644          * automatically and not be an issue.
645          */
646         lwp_exit(1, pp);
647 }
648
649 /*
650  * Eventually called by every exiting LWP
651  *
652  * p->p_token must be held.  mplock may be held and will be released.
653  */
654 void
655 lwp_exit(int masterexit, void *waddr)
656 {
657         struct thread *td = curthread;
658         struct lwp *lp = td->td_lwp;
659         struct proc *p = lp->lwp_proc;
660         int dowake = 0;
661
662         /*
663          * Release the current user process designation on the process so
664          * the userland scheduler can work in someone else.
665          */
666         p->p_usched->release_curproc(lp);
667
668         /*
669          * lwp_exit() may be called without setting LWP_MP_WEXIT, so
670          * make sure it is set here.
671          */
672         ASSERT_LWKT_TOKEN_HELD(&p->p_token);
673         atomic_set_int(&lp->lwp_mpflags, LWP_MP_WEXIT);
674
675         /*
676          * Clean up any virtualization
677          */
678         if (lp->lwp_vkernel)
679                 vkernel_lwp_exit(lp);
680
681         if (td->td_vmm)
682                 vmm_vmdestroy();
683
684         /*
685          * Clean up select/poll support
686          */
687         kqueue_terminate(&lp->lwp_kqueue);
688
689         /*
690          * Clean up any syscall-cached ucred or rlimit.
691          */
692         if (td->td_ucred) {
693                 crfree(td->td_ucred);
694                 td->td_ucred = NULL;
695         }
696         if (td->td_limit) {
697                 struct plimit *rlimit;
698
699                 rlimit = td->td_limit;
700                 td->td_limit = NULL;
701                 plimit_free(rlimit);
702         }
703
704         /*
705          * Cleanup any cached descriptors for this thread
706          */
707         if (p->p_fd)
708                 fexitcache(td);
709
710         /*
711          * Nobody actually wakes us when the lock
712          * count reaches zero, so just wait one tick.
713          */
714         while (lp->lwp_lock > 0)
715                 tsleep(lp, 0, "lwpexit", 1);
716
717         /* Hand down resource usage to our proc */
718         ruadd(&p->p_ru, &lp->lwp_ru);
719
720         /*
721          * If we don't hold the process until the LWP is reaped wait*()
722          * may try to dispose of its vmspace before all the LWPs have
723          * actually terminated.
724          */
725         PHOLD(p);
726
727         /*
728          * Do any remaining work that might block on us.  We should be
729          * coded such that further blocking is ok after decrementing
730          * p_nthreads but don't take the chance.
731          */
732         dsched_exit_thread(td);
733         biosched_done(curthread);
734
735         /*
736          * We have to use the reaper for all the LWPs except the one doing
737          * the master exit.  The LWP doing the master exit can just be
738          * left on p_lwps and the process reaper will deal with it
739          * synchronously, which is much faster.
740          *
741          * Wakeup anyone waiting on p_nthreads to drop to 1 or 0.
742          *
743          * The process is left held until the reaper calls lwp_dispose() on
744          * the lp (after calling lwp_wait()).
745          */
746         if (masterexit == 0) {
747                 int cpu = mycpuid;
748
749                 lwp_rb_tree_RB_REMOVE(&p->p_lwp_tree, lp);
750                 --p->p_nthreads;
751                 if ((p->p_flags & P_MAYBETHREADED) && p->p_nthreads <= 1)
752                         dowake = 1;
753                 lwkt_gettoken(&deadlwp_token[cpu]);
754                 LIST_INSERT_HEAD(&deadlwp_list[cpu], lp, u.lwp_reap_entry);
755                 taskqueue_enqueue(taskqueue_thread[cpu], deadlwp_task[cpu]);
756                 lwkt_reltoken(&deadlwp_token[cpu]);
757         } else {
758                 --p->p_nthreads;
759                 if ((p->p_flags & P_MAYBETHREADED) && p->p_nthreads <= 1)
760                         dowake = 1;
761         }
762
763         /*
764          * We no longer need p_token.
765          *
766          * Tell the userland scheduler that we are going away
767          */
768         lwkt_reltoken(&p->p_token);
769         p->p_usched->heuristic_exiting(lp, p);
770
771         /*
772          * Issue late wakeups after releasing our token to give us a chance
773          * to deschedule and switch away before another cpu in a wait*()
774          * reaps us.  This is done as late as possible to reduce contention.
775          */
776         if (dowake)
777                 wakeup(&p->p_nthreads);
778         if (waddr)
779                 wakeup(waddr);
780
781         cpu_lwp_exit();
782 }
783
784 /*
785  * Wait until a lwp is completely dead.  The final interlock in this drama
786  * is when TDF_EXITING is set in cpu_thread_exit() just before the final
787  * switchout.
788  *
789  * At the point TDF_EXITING is set a complete exit is accomplished when
790  * TDF_RUNNING and TDF_PREEMPT_LOCK are both clear.  td_mpflags has two
791  * post-switch interlock flags that can be used to wait for the TDF_
792  * flags to clear.
793  *
794  * Returns non-zero on success, and zero if the caller needs to retry
795  * the lwp_wait().
796  */
797 static int
798 lwp_wait(struct lwp *lp)
799 {
800         struct thread *td = lp->lwp_thread;
801         u_int mpflags;
802
803         KKASSERT(lwkt_preempted_proc() != lp);
804
805         /*
806          * This bit of code uses the thread destruction interlock
807          * managed by lwkt_switch_return() to wait for the lwp's
808          * thread to completely disengage.
809          *
810          * It is possible for us to race another cpu core so we
811          * have to do this correctly.
812          */
813         for (;;) {
814                 mpflags = td->td_mpflags;
815                 cpu_ccfence();
816                 if (mpflags & TDF_MP_EXITSIG)
817                         break;
818                 tsleep_interlock(td, 0);
819                 if (atomic_cmpset_int(&td->td_mpflags, mpflags,
820                                       mpflags | TDF_MP_EXITWAIT)) {
821                         tsleep(td, PINTERLOCKED, "lwpxt", 0);
822                 }
823         }
824
825         /*
826          * We've already waited for the core exit but there can still
827          * be other refs from e.g. process scans and such.
828          */
829         if (lp->lwp_lock > 0) {
830                 tsleep(lp, 0, "lwpwait1", 1);
831                 return(0);
832         }
833         if (td->td_refs) {
834                 tsleep(td, 0, "lwpwait2", 1);
835                 return(0);
836         }
837
838         /*
839          * Now that we have the thread destruction interlock these flags
840          * really should already be cleaned up, keep a check for safety.
841          *
842          * We can't rip its stack out from under it until TDF_EXITING is
843          * set and both TDF_RUNNING and TDF_PREEMPT_LOCK are clear.
844          * TDF_PREEMPT_LOCK must be checked because TDF_RUNNING
845          * will be cleared temporarily if a thread gets preempted.
846          */
847         while ((td->td_flags & (TDF_RUNNING |
848                                 TDF_RUNQ |
849                                 TDF_PREEMPT_LOCK |
850                                 TDF_EXITING)) != TDF_EXITING) {
851                 tsleep(lp, 0, "lwpwait3", 1);
852                 return (0);
853         }
854
855         KASSERT((td->td_flags & (TDF_RUNQ|TDF_TSLEEPQ)) == 0,
856                 ("lwp_wait: td %p (%s) still on run or sleep queue",
857                 td, td->td_comm));
858         return (1);
859 }
860
861 /*
862  * Release the resources associated with a lwp.
863  * The lwp must be completely dead.
864  */
865 void
866 lwp_dispose(struct lwp *lp)
867 {
868         struct thread *td = lp->lwp_thread;
869
870         KKASSERT(lwkt_preempted_proc() != lp);
871         KKASSERT(lp->lwp_lock == 0);
872         KKASSERT(td->td_refs == 0);
873         KKASSERT((td->td_flags & (TDF_RUNNING |
874                                   TDF_RUNQ |
875                                   TDF_PREEMPT_LOCK |
876                                   TDF_EXITING)) == TDF_EXITING);
877
878         PRELE(lp->lwp_proc);
879         lp->lwp_proc = NULL;
880         if (td != NULL) {
881                 td->td_proc = NULL;
882                 td->td_lwp = NULL;
883                 lp->lwp_thread = NULL;
884                 lwkt_free_thread(td);
885         }
886         kfree(lp, M_LWP);
887 }
888
889 int
890 sys_wait4(struct wait_args *uap)
891 {
892         struct __wrusage wrusage;
893         int error;
894         int status;
895         int options;
896         id_t id;
897         idtype_t idtype;
898
899         options = uap->options | WEXITED | WTRAPPED;
900         id = uap->pid;
901
902         if (id == WAIT_ANY) {
903                 idtype = P_ALL;
904         } else if (id == WAIT_MYPGRP) {
905                 idtype = P_PGID;
906                 id = curproc->p_pgid;
907         } else if (id < 0) {
908                 idtype = P_PGID;
909                 id = -id;
910         } else {
911                 idtype = P_PID;
912         }
913
914         error = kern_wait(idtype, id, &status, options, &wrusage,
915                           NULL, &uap->sysmsg_result);
916
917         if (error == 0 && uap->status)
918                 error = copyout(&status, uap->status, sizeof(*uap->status));
919         if (error == 0 && uap->rusage) {
920                 ruadd(&wrusage.wru_self, &wrusage.wru_children);
921                 error = copyout(&wrusage.wru_self, uap->rusage, sizeof(*uap->rusage));
922         }
923         return (error);
924 }
925
926 int
927 sys_wait6(struct wait6_args *uap)
928 {
929         struct __wrusage wrusage;
930         struct __siginfo info;
931         struct __siginfo *infop;
932         int error;
933         int status;
934         int options;
935         id_t id;
936         idtype_t idtype;
937
938         /*
939          * NOTE: wait6() requires WEXITED and WTRAPPED to be specified if
940          *       desired.
941          */
942         options = uap->options;
943         idtype = uap->idtype;
944         id = uap->id;
945         infop = uap->info ? &info : NULL;
946
947         switch(idtype) {
948         case P_PID:
949         case P_PGID:
950                 if (id == WAIT_MYPGRP) {
951                         idtype = P_PGID;
952                         id = curproc->p_pgid;
953                 }
954                 break;
955         default:
956                 /* let kern_wait deal with the remainder */
957                 break;
958         }
959
960         error = kern_wait(idtype, id, &status, options,
961                           &wrusage, infop, &uap->sysmsg_result);
962
963         if (error == 0 && uap->status)
964                 error = copyout(&status, uap->status, sizeof(*uap->status));
965         if (error == 0 && uap->wrusage)
966                 error = copyout(&wrusage, uap->wrusage, sizeof(*uap->wrusage));
967         if (error == 0 && uap->info)
968                 error = copyout(&info, uap->info, sizeof(*uap->info));
969         return (error);
970 }
971
972 /*
973  * kernel wait*() system call support
974  */
975 int
976 kern_wait(idtype_t idtype, id_t id, int *status, int options,
977           struct __wrusage *wrusage, struct __siginfo *info, int *res)
978 {
979         struct thread *td = curthread;
980         struct lwp *lp;
981         struct proc *q = td->td_proc;
982         struct proc *p, *t;
983         struct ucred *cr;
984         struct pargs *pa;
985         struct sigacts *ps;
986         int nfound, error;
987         long waitgen;
988
989         /*
990          * Must not have extraneous options.  Must have at least one
991          * matchable option.
992          */
993         if (options &~ (WUNTRACED|WNOHANG|WCONTINUED|WLINUXCLONE|WSTOPPED|
994                         WEXITED|WTRAPPED|WNOWAIT)) {
995                 return (EINVAL);
996         }
997         if ((options & (WEXITED | WUNTRACED | WCONTINUED | WTRAPPED)) == 0) {
998                 return (EINVAL);
999         }
1000
1001         /*
1002          * Protect the q->p_children list
1003          */
1004         lwkt_gettoken(&q->p_token);
1005 loop:
1006         /*
1007          * All sorts of things can change due to blocking so we have to loop
1008          * all the way back up here.
1009          *
1010          * The problem is that if a process group is stopped and the parent
1011          * is doing a wait*(..., WUNTRACED, ...), it will see the STOP
1012          * of the child and then stop itself when it tries to return from the
1013          * system call.  When the process group is resumed the parent will
1014          * then get the STOP status even though the child has now resumed
1015          * (a followup wait*() will get the CONT status).
1016          *
1017          * Previously the CONT would overwrite the STOP because the tstop
1018          * was handled within tsleep(), and the parent would only see
1019          * the CONT when both are stopped and continued together.  This little
1020          * two-line hack restores this effect.
1021          */
1022         if (STOPLWP(q, td->td_lwp))
1023             tstop();
1024
1025         nfound = 0;
1026
1027         /*
1028          * Loop on children.
1029          *
1030          * NOTE: We don't want to break q's p_token in the loop for the
1031          *       case where no children are found or we risk breaking the
1032          *       interlock between child and parent.
1033          */
1034         waitgen = atomic_fetchadd_long(&q->p_waitgen, 0x80000000);
1035         LIST_FOREACH(p, &q->p_children, p_sibling) {
1036                 /*
1037                  * Filter, (p) will be held on fall-through.  Try to optimize
1038                  * this to avoid the atomic op until we are pretty sure we
1039                  * want this process.
1040                  */
1041                 switch(idtype) {
1042                 case P_ALL:
1043                         PHOLD(p);
1044                         break;
1045                 case P_PID:
1046                         if (p->p_pid != (pid_t)id)
1047                                 continue;
1048                         PHOLD(p);
1049                         break;
1050                 case P_PGID:
1051                         if (p->p_pgid != (pid_t)id)
1052                                 continue;
1053                         PHOLD(p);
1054                         break;
1055                 case P_SID:
1056                         PHOLD(p);
1057                         if (p->p_session && p->p_session->s_sid != (pid_t)id) {
1058                                 PRELE(p);
1059                                 continue;
1060                         }
1061                         break;
1062                 case P_UID:
1063                         PHOLD(p);
1064                         if (p->p_ucred->cr_uid != (uid_t)id) {
1065                                 PRELE(p);
1066                                 continue;
1067                         }
1068                         break;
1069                 case P_GID:
1070                         PHOLD(p);
1071                         if (p->p_ucred->cr_gid != (gid_t)id) {
1072                                 PRELE(p);
1073                                 continue;
1074                         }
1075                         break;
1076                 case P_JAILID:
1077                         PHOLD(p);
1078                         if (p->p_ucred->cr_prison &&
1079                             p->p_ucred->cr_prison->pr_id != (int)id) {
1080                                 PRELE(p);
1081                                 continue;
1082                         }
1083                         break;
1084                 default:
1085                         /* unsupported filter */
1086                         continue;
1087                 }
1088                 /* (p) is held at this point */
1089
1090                 /*
1091                  * This special case handles a kthread spawned by linux_clone
1092                  * (see linux_misc.c).  The linux_wait4 and linux_waitpid 
1093                  * functions need to be able to distinguish between waiting
1094                  * on a process and waiting on a thread.  It is a thread if
1095                  * p_sigparent is not SIGCHLD, and the WLINUXCLONE option
1096                  * signifies we want to wait for threads and not processes.
1097                  */
1098                 if ((p->p_sigparent != SIGCHLD) ^ 
1099                     ((options & WLINUXCLONE) != 0)) {
1100                         PRELE(p);
1101                         continue;
1102                 }
1103
1104                 nfound++;
1105                 if (p->p_stat == SZOMB && (options & WEXITED)) {
1106                         /*
1107                          * We may go into SZOMB with threads still present.
1108                          * We must wait for them to exit before we can reap
1109                          * the master thread, otherwise we may race reaping
1110                          * non-master threads.
1111                          *
1112                          * Only this routine can remove a process from
1113                          * the zombie list and destroy it.
1114                          */
1115                         if (PHOLDZOMB(p)) {
1116                                 PRELE(p);
1117                                 goto loop;
1118                         }
1119                         lwkt_gettoken(&p->p_token);
1120                         if (p->p_pptr != q) {
1121                                 lwkt_reltoken(&p->p_token);
1122                                 PRELE(p);
1123                                 PRELEZOMB(p);
1124                                 goto loop;
1125                         }
1126                         while (p->p_nthreads > 0) {
1127                                 tsleep(&p->p_nthreads, 0, "lwpzomb", hz);
1128                         }
1129
1130                         /*
1131                          * Reap any LWPs left in p->p_lwps.  This is usually
1132                          * just the last LWP.  This must be done before
1133                          * we loop on p_lock since the lwps hold a ref on
1134                          * it as a vmspace interlock.
1135                          *
1136                          * Once that is accomplished p_nthreads had better
1137                          * be zero.
1138                          */
1139                         while ((lp = RB_ROOT(&p->p_lwp_tree)) != NULL) {
1140                                 /*
1141                                  * Make sure no one is using this lwp, before
1142                                  * it is removed from the tree.  If we didn't
1143                                  * wait it here, lwp tree iteration with
1144                                  * blocking operation would be broken.
1145                                  */
1146                                 while (lp->lwp_lock > 0)
1147                                         tsleep(lp, 0, "zomblwp", 1);
1148                                 lwp_rb_tree_RB_REMOVE(&p->p_lwp_tree, lp);
1149                                 reaplwp(lp);
1150                         }
1151                         KKASSERT(p->p_nthreads == 0);
1152
1153                         /*
1154                          * Don't do anything really bad until all references
1155                          * to the process go away.  This may include other
1156                          * LWPs which are still in the process of being
1157                          * reaped.  We can't just pull the rug out from under
1158                          * them because they may still be using the VM space.
1159                          *
1160                          * Certain kernel facilities such as /proc will also
1161                          * put a hold on the process for short periods of
1162                          * time.
1163                          */
1164                         PRELE(p);               /* from top of loop */
1165                         PSTALL(p, "reap3", 1);  /* 1 ref (for PZOMBHOLD) */
1166
1167                         /* Take care of our return values. */
1168                         *res = p->p_pid;
1169
1170                         *status = p->p_xstat;
1171                         wrusage->wru_self = p->p_ru;
1172                         wrusage->wru_children = p->p_cru;
1173
1174                         if (info) {
1175                                 bzero(info, sizeof(*info));
1176                                 info->si_errno = 0;
1177                                 info->si_signo = SIGCHLD;
1178                                 if (p->p_xstat)
1179                                         info->si_code = CLD_KILLED;
1180                                 else
1181                                         info->si_code = CLD_EXITED;
1182                                 info->si_status = p->p_xstat;
1183                                 info->si_pid = p->p_pid;
1184                                 info->si_uid = p->p_ucred->cr_uid;
1185                         }
1186
1187                         /*
1188                          * WNOWAIT shortcuts to done here, leaving the
1189                          * child on the zombie list.
1190                          */
1191                         if (options & WNOWAIT) {
1192                                 lwkt_reltoken(&p->p_token);
1193                                 PRELEZOMB(p);
1194                                 error = 0;
1195                                 goto done;
1196                         }
1197
1198                         /*
1199                          * If we got the child via a ptrace 'attach',
1200                          * we need to give it back to the old parent.
1201                          */
1202                         if (p->p_oppid && (t = pfind(p->p_oppid)) != NULL) {
1203                                 p->p_oppid = 0;
1204                                 proc_reparent(p, t);
1205                                 ksignal(t, SIGCHLD);
1206                                 wakeup((caddr_t)t);
1207                                 PRELE(t);
1208                                 lwkt_reltoken(&p->p_token);
1209                                 PRELEZOMB(p);
1210                                 error = 0;
1211                                 goto done;
1212                         }
1213
1214                         /*
1215                          * Unlink the proc from its process group so that
1216                          * the following operations won't lead to an
1217                          * inconsistent state for processes running down
1218                          * the zombie list.
1219                          */
1220                         proc_remove_zombie(p);
1221                         proc_userunmap(p);
1222                         lwkt_reltoken(&p->p_token);
1223                         leavepgrp(p);
1224
1225                         p->p_xstat = 0;
1226                         ruadd(&q->p_cru, &p->p_ru);
1227                         ruadd(&q->p_cru, &p->p_cru);
1228
1229                         /*
1230                          * Decrement the count of procs running with this uid.
1231                          */
1232                         chgproccnt(p->p_ucred->cr_ruidinfo, -1, 0);
1233
1234                         /*
1235                          * Free up credentials.  p_spin is required to
1236                          * avoid races against allproc scans.
1237                          */
1238                         spin_lock(&p->p_spin);
1239                         cr = p->p_ucred;
1240                         p->p_ucred = NULL;
1241                         spin_unlock(&p->p_spin);
1242                         crfree(cr);
1243
1244                         /*
1245                          * Remove unused arguments
1246                          */
1247                         pa = p->p_args;
1248                         p->p_args = NULL;
1249                         if (pa && refcount_release(&pa->ar_ref)) {
1250                                 kfree(pa, M_PARGS);
1251                                 pa = NULL;
1252                         }
1253
1254                         ps = p->p_sigacts;
1255                         p->p_sigacts = NULL;
1256                         if (ps && refcount_release(&ps->ps_refcnt)) {
1257                                 kfree(ps, M_SUBPROC);
1258                                 ps = NULL;
1259                         }
1260
1261                         /*
1262                          * Our exitingcount was incremented when the process
1263                          * became a zombie, now that the process has been
1264                          * removed from (almost) all lists we should be able
1265                          * to safely destroy its vmspace.  Wait for any current
1266                          * holders to go away (so the vmspace remains stable),
1267                          * then scrap it.
1268                          *
1269                          * NOTE: Releasing the parent process (q) p_token
1270                          *       across the vmspace_exitfree() call is
1271                          *       important here to reduce stalls on
1272                          *       interactions with (q) (such as
1273                          *       fork/exec/wait or 'ps').
1274                          */
1275                         PSTALL(p, "reap4", 1);
1276                         lwkt_reltoken(&q->p_token);
1277                         vmspace_exitfree(p);
1278                         lwkt_gettoken(&q->p_token);
1279                         PSTALL(p, "reap5", 1);
1280
1281                         /*
1282                          * NOTE: We have to officially release ZOMB in order
1283                          *       to ensure that a racing thread in kern_wait()
1284                          *       which blocked on ZOMB is woken up.
1285                          */
1286                         PRELEZOMB(p);
1287                         kfree(p->p_uidpcpu, M_SUBPROC);
1288                         kfree(p, M_PROC);
1289                         atomic_add_int(&nprocs, -1);
1290                         error = 0;
1291                         goto done;
1292                 }
1293
1294                 /*
1295                  * Process has not yet exited
1296                  */
1297                 if ((p->p_stat == SSTOP || p->p_stat == SCORE) &&
1298                     (p->p_flags & P_WAITED) == 0 &&
1299                     (((p->p_flags & P_TRACED) && (options & WTRAPPED)) ||
1300                      (options & WSTOPPED))) {
1301                         lwkt_gettoken(&p->p_token);
1302                         if (p->p_pptr != q) {
1303                                 lwkt_reltoken(&p->p_token);
1304                                 PRELE(p);
1305                                 goto loop;
1306                         }
1307                         if ((p->p_stat != SSTOP && p->p_stat != SCORE) ||
1308                             (p->p_flags & P_WAITED) != 0 ||
1309                             ((p->p_flags & P_TRACED) == 0 &&
1310                              (options & WUNTRACED) == 0)) {
1311                                 lwkt_reltoken(&p->p_token);
1312                                 PRELE(p);
1313                                 goto loop;
1314                         }
1315
1316                         /*
1317                          * Don't set P_WAITED if WNOWAIT specified, leaving
1318                          * the process in a waitable state.
1319                          */
1320                         if ((options & WNOWAIT) == 0)
1321                                 p->p_flags |= P_WAITED;
1322
1323                         *res = p->p_pid;
1324                         *status = W_STOPCODE(p->p_xstat);
1325                         /* Zero rusage so we get something consistent. */
1326                         bzero(wrusage, sizeof(*wrusage));
1327                         error = 0;
1328                         if (info) {
1329                                 bzero(info, sizeof(*info));
1330                                 if (p->p_flags & P_TRACED)
1331                                         info->si_code = CLD_TRAPPED;
1332                                 else
1333                                         info->si_code = CLD_STOPPED;
1334                                 info->si_status = p->p_xstat;
1335                         }
1336                         lwkt_reltoken(&p->p_token);
1337                         PRELE(p);
1338                         goto done;
1339                 }
1340                 if ((options & WCONTINUED) && (p->p_flags & P_CONTINUED)) {
1341                         lwkt_gettoken(&p->p_token);
1342                         if (p->p_pptr != q) {
1343                                 lwkt_reltoken(&p->p_token);
1344                                 PRELE(p);
1345                                 goto loop;
1346                         }
1347                         if ((p->p_flags & P_CONTINUED) == 0) {
1348                                 lwkt_reltoken(&p->p_token);
1349                                 PRELE(p);
1350                                 goto loop;
1351                         }
1352
1353                         *res = p->p_pid;
1354
1355                         /*
1356                          * Don't set P_WAITED if WNOWAIT specified, leaving
1357                          * the process in a waitable state.
1358                          */
1359                         if ((options & WNOWAIT) == 0)
1360                                 p->p_flags &= ~P_CONTINUED;
1361
1362                         *status = SIGCONT;
1363                         error = 0;
1364                         if (info) {
1365                                 bzero(info, sizeof(*info));
1366                                 info->si_code = CLD_CONTINUED;
1367                                 info->si_status = p->p_xstat;
1368                         }
1369                         lwkt_reltoken(&p->p_token);
1370                         PRELE(p);
1371                         goto done;
1372                 }
1373                 PRELE(p);
1374         }
1375         if (nfound == 0) {
1376                 error = ECHILD;
1377                 goto done;
1378         }
1379         if (options & WNOHANG) {
1380                 *res = 0;
1381                 error = 0;
1382                 goto done;
1383         }
1384
1385         /*
1386          * Wait for signal - interlocked using q->p_waitgen.
1387          */
1388         error = 0;
1389         while ((waitgen & 0x7FFFFFFF) == (q->p_waitgen & 0x7FFFFFFF)) {
1390                 tsleep_interlock(q, PCATCH);
1391                 waitgen = atomic_fetchadd_long(&q->p_waitgen, 0x80000000);
1392                 if ((waitgen & 0x7FFFFFFF) == (q->p_waitgen & 0x7FFFFFFF)) {
1393                         error = tsleep(q, PCATCH | PINTERLOCKED, "wait", 0);
1394                         break;
1395                 }
1396         }
1397         if (error) {
1398 done:
1399                 lwkt_reltoken(&q->p_token);
1400                 return (error);
1401         }
1402         goto loop;
1403 }
1404
1405 /*
1406  * Change child's parent process to parent.
1407  *
1408  * p_children/p_sibling requires the parent's token, and
1409  * changing pptr requires the child's token, so we have to
1410  * get three tokens to do this operation.  We also need to
1411  * hold pointers that might get ripped out from under us to
1412  * preserve structural integrity.
1413  *
1414  * It is possible to race another reparent or disconnect or other
1415  * similar operation.  We must retry when this situation occurs.
1416  * Once we successfully reparent the process we no longer care
1417  * about any races.
1418  */
1419 void
1420 proc_reparent(struct proc *child, struct proc *parent)
1421 {
1422         struct proc *opp;
1423
1424         PHOLD(parent);
1425         while ((opp = child->p_pptr) != parent) {
1426                 PHOLD(opp);
1427                 lwkt_gettoken(&opp->p_token);
1428                 lwkt_gettoken(&child->p_token);
1429                 lwkt_gettoken(&parent->p_token);
1430                 if (child->p_pptr != opp) {
1431                         lwkt_reltoken(&parent->p_token);
1432                         lwkt_reltoken(&child->p_token);
1433                         lwkt_reltoken(&opp->p_token);
1434                         PRELE(opp);
1435                         continue;
1436                 }
1437                 LIST_REMOVE(child, p_sibling);
1438                 LIST_INSERT_HEAD(&parent->p_children, child, p_sibling);
1439                 child->p_pptr = parent;
1440                 child->p_ppid = parent->p_pid;
1441                 lwkt_reltoken(&parent->p_token);
1442                 lwkt_reltoken(&child->p_token);
1443                 lwkt_reltoken(&opp->p_token);
1444                 if (LIST_EMPTY(&opp->p_children))
1445                         wakeup(opp);
1446                 PRELE(opp);
1447                 break;
1448         }
1449         PRELE(parent);
1450 }
1451
1452 /*
1453  * The next two functions are to handle adding/deleting items on the
1454  * exit callout list
1455  * 
1456  * at_exit():
1457  * Take the arguments given and put them onto the exit callout list,
1458  * However first make sure that it's not already there.
1459  * returns 0 on success.
1460  */
1461
1462 int
1463 at_exit(exitlist_fn function)
1464 {
1465         struct exitlist *ep;
1466
1467 #ifdef INVARIANTS
1468         /* Be noisy if the programmer has lost track of things */
1469         if (rm_at_exit(function)) 
1470                 kprintf("WARNING: exit callout entry (%p) already present\n",
1471                     function);
1472 #endif
1473         ep = kmalloc(sizeof(*ep), M_ATEXIT, M_NOWAIT);
1474         if (ep == NULL)
1475                 return (ENOMEM);
1476         ep->function = function;
1477         TAILQ_INSERT_TAIL(&exit_list, ep, next);
1478         return (0);
1479 }
1480
1481 /*
1482  * Scan the exit callout list for the given item and remove it.
1483  * Returns the number of items removed (0 or 1)
1484  */
1485 int
1486 rm_at_exit(exitlist_fn function)
1487 {
1488         struct exitlist *ep;
1489
1490         TAILQ_FOREACH(ep, &exit_list, next) {
1491                 if (ep->function == function) {
1492                         TAILQ_REMOVE(&exit_list, ep, next);
1493                         kfree(ep, M_ATEXIT);
1494                         return(1);
1495                 }
1496         }       
1497         return (0);
1498 }
1499
1500 /*
1501  * LWP reaper related code.
1502  */
1503 static void
1504 reaplwps(void *context, int dummy)
1505 {
1506         struct lwplist *lwplist = context;
1507         struct lwp *lp;
1508         int cpu = mycpuid;
1509
1510         lwkt_gettoken(&deadlwp_token[cpu]);
1511         while ((lp = LIST_FIRST(lwplist))) {
1512                 LIST_REMOVE(lp, u.lwp_reap_entry);
1513                 reaplwp(lp);
1514         }
1515         lwkt_reltoken(&deadlwp_token[cpu]);
1516 }
1517
1518 static void
1519 reaplwp(struct lwp *lp)
1520 {
1521         while (lwp_wait(lp) == 0)
1522                 ;
1523         lwp_dispose(lp);
1524 }
1525
1526 static void
1527 deadlwp_init(void)
1528 {
1529         int cpu;
1530
1531         for (cpu = 0; cpu < ncpus; cpu++) {
1532                 lwkt_token_init(&deadlwp_token[cpu], "deadlwpl");
1533                 LIST_INIT(&deadlwp_list[cpu]);
1534                 deadlwp_task[cpu] = kmalloc(sizeof(*deadlwp_task[cpu]),
1535                                             M_DEVBUF, M_WAITOK);
1536                 TASK_INIT(deadlwp_task[cpu], 0, reaplwps, &deadlwp_list[cpu]);
1537         }
1538 }
1539
1540 SYSINIT(deadlwpinit, SI_SUB_CONFIGURE, SI_ORDER_ANY, deadlwp_init, NULL);