c6f6125bd1eea11ecd1a03bb46817d6cf96cd56d
[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. All advertising materials mentioning features or use of this software
19  *    must display the following acknowledgement:
20  *      This product includes software developed by the University of
21  *      California, Berkeley and its contributors.
22  * 4. Neither the name of the University nor the names of its contributors
23  *    may be used to endorse or promote products derived from this software
24  *    without specific prior written permission.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36  * SUCH DAMAGE.
37  *
38  *      @(#)kern_exit.c 8.7 (Berkeley) 2/12/94
39  * $FreeBSD: src/sys/kern/kern_exit.c,v 1.92.2.11 2003/01/13 22:51:16 dillon Exp $
40  * $DragonFly: src/sys/kern/kern_exit.c,v 1.91 2008/05/18 20:02:02 nth Exp $
41  */
42
43 #include "opt_compat.h"
44 #include "opt_ktrace.h"
45
46 #include <sys/param.h>
47 #include <sys/systm.h>
48 #include <sys/sysproto.h>
49 #include <sys/kernel.h>
50 #include <sys/malloc.h>
51 #include <sys/proc.h>
52 #include <sys/ktrace.h>
53 #include <sys/pioctl.h>
54 #include <sys/tty.h>
55 #include <sys/wait.h>
56 #include <sys/vnode.h>
57 #include <sys/resourcevar.h>
58 #include <sys/signalvar.h>
59 #include <sys/taskqueue.h>
60 #include <sys/ptrace.h>
61 #include <sys/acct.h>           /* for acct_process() function prototype */
62 #include <sys/filedesc.h>
63 #include <sys/shm.h>
64 #include <sys/sem.h>
65 #include <sys/aio.h>
66 #include <sys/jail.h>
67 #include <sys/kern_syscall.h>
68 #include <sys/upcall.h>
69 #include <sys/caps.h>
70 #include <sys/unistd.h>
71 #include <sys/eventhandler.h>
72 #include <sys/dsched.h>
73
74 #include <vm/vm.h>
75 #include <vm/vm_param.h>
76 #include <sys/lock.h>
77 #include <vm/pmap.h>
78 #include <vm/vm_map.h>
79 #include <vm/vm_extern.h>
80 #include <sys/user.h>
81
82 #include <sys/thread2.h>
83 #include <sys/sysref2.h>
84 #include <sys/mplock2.h>
85
86 static void reaplwps(void *context, int dummy);
87 static void reaplwp(struct lwp *lp);
88 static void killlwps(struct lwp *lp);
89
90 static MALLOC_DEFINE(M_ATEXIT, "atexit", "atexit callback");
91 static MALLOC_DEFINE(M_ZOMBIE, "zombie", "zombie proc status");
92
93 static struct lwkt_token deadlwp_token = LWKT_TOKEN_INITIALIZER(deadlwp_token);
94
95 /*
96  * callout list for things to do at exit time
97  */
98 struct exitlist {
99         exitlist_fn function;
100         TAILQ_ENTRY(exitlist) next;
101 };
102
103 TAILQ_HEAD(exit_list_head, exitlist);
104 static struct exit_list_head exit_list = TAILQ_HEAD_INITIALIZER(exit_list);
105
106 /*
107  * LWP reaper data
108  */
109 struct task *deadlwp_task[MAXCPU];
110 struct lwplist deadlwp_list[MAXCPU];
111
112 /*
113  * exit --
114  *      Death of process.
115  *
116  * SYS_EXIT_ARGS(int rval)
117  */
118 int
119 sys_exit(struct exit_args *uap)
120 {
121         exit1(W_EXITCODE(uap->rval, 0));
122         /* NOTREACHED */
123 }
124
125 /*
126  * Extended exit --
127  *      Death of a lwp or process with optional bells and whistles.
128  *
129  * MPALMOSTSAFE
130  */
131 int
132 sys_extexit(struct extexit_args *uap)
133 {
134         struct proc *p = curproc;
135         int action, who;
136         int error;
137
138         action = EXTEXIT_ACTION(uap->how);
139         who = EXTEXIT_WHO(uap->how);
140
141         /* Check parameters before we might perform some action */
142         switch (who) {
143         case EXTEXIT_PROC:
144         case EXTEXIT_LWP:
145                 break;
146         default:
147                 return (EINVAL);
148         }
149
150         switch (action) {
151         case EXTEXIT_SIMPLE:
152                 break;
153         case EXTEXIT_SETINT:
154                 error = copyout(&uap->status, uap->addr, sizeof(uap->status));
155                 if (error)
156                         return (error);
157                 break;
158         default:
159                 return (EINVAL);
160         }
161
162         lwkt_gettoken(&p->p_token);
163
164         switch (who) {
165         case EXTEXIT_LWP:
166                 /*
167                  * Be sure only to perform a simple lwp exit if there is at
168                  * least one more lwp in the proc, which will call exit1()
169                  * later, otherwise the proc will be an UNDEAD and not even a
170                  * SZOMB!
171                  */
172                 if (p->p_nthreads > 1) {
173                         lwp_exit(0);    /* called w/ p_token held */
174                         /* NOT REACHED */
175                 }
176                 /* else last lwp in proc:  do the real thing */
177                 /* FALLTHROUGH */
178         default:        /* to help gcc */
179         case EXTEXIT_PROC:
180                 lwkt_reltoken(&p->p_token);
181                 exit1(W_EXITCODE(uap->status, 0));
182                 /* NOTREACHED */
183         }
184
185         /* NOTREACHED */
186         lwkt_reltoken(&p->p_token);     /* safety */
187 }
188
189 /*
190  * Kill all lwps associated with the current process except the
191  * current lwp.   Return an error if we race another thread trying to
192  * do the same thing and lose the race.
193  *
194  * If forexec is non-zero the current thread and process flags are
195  * cleaned up so they can be reused.
196  */
197 int
198 killalllwps(int forexec)
199 {
200         struct lwp *lp = curthread->td_lwp;
201         struct proc *p = lp->lwp_proc;
202
203         /*
204          * Interlock against P_WEXIT.  Only one of the process's thread
205          * is allowed to do the master exit.
206          */
207         if (p->p_flag & P_WEXIT)
208                 return (EALREADY);
209         p->p_flag |= P_WEXIT;
210
211         /*
212          * Interlock with LWP_WEXIT and kill any remaining LWPs
213          */
214         lp->lwp_flag |= LWP_WEXIT;
215         if (p->p_nthreads > 1)
216                 killlwps(lp);
217
218         /*
219          * If doing this for an exec, clean up the remaining thread
220          * (us) for continuing operation after all the other threads
221          * have been killed.
222          */
223         if (forexec) {
224                 lp->lwp_flag &= ~LWP_WEXIT;
225                 p->p_flag &= ~P_WEXIT;
226         }
227         return(0);
228 }
229
230 /*
231  * Kill all LWPs except the current one.  Do not try to signal
232  * LWPs which have exited on their own or have already been
233  * signaled.
234  */
235 static void
236 killlwps(struct lwp *lp)
237 {
238         struct proc *p = lp->lwp_proc;
239         struct lwp *tlp;
240
241         /*
242          * Kill the remaining LWPs.  We must send the signal before setting
243          * LWP_WEXIT.  The setting of WEXIT is optional but helps reduce
244          * races.  tlp must be held across the call as it might block and
245          * allow the target lwp to rip itself out from under our loop.
246          */
247         FOREACH_LWP_IN_PROC(tlp, p) {
248                 LWPHOLD(tlp);
249                 if ((tlp->lwp_flag & LWP_WEXIT) == 0) {
250                         lwpsignal(p, tlp, SIGKILL);
251                         tlp->lwp_flag |= LWP_WEXIT;
252                 }
253                 LWPRELE(tlp);
254         }
255
256         /*
257          * Wait for everything to clear out.
258          */
259         while (p->p_nthreads > 1) {
260                 tsleep(&p->p_nthreads, 0, "killlwps", 0);
261         }
262 }
263
264 /*
265  * Exit: deallocate address space and other resources, change proc state
266  * to zombie, and unlink proc from allproc and parent's lists.  Save exit
267  * status and rusage for wait().  Check for child processes and orphan them.
268  */
269 void
270 exit1(int rv)
271 {
272         struct thread *td = curthread;
273         struct proc *p = td->td_proc;
274         struct lwp *lp = td->td_lwp;
275         struct proc *q, *nq;
276         struct vmspace *vm;
277         struct vnode *vtmp;
278         struct exitlist *ep;
279         int error;
280
281         lwkt_gettoken(&p->p_token);
282
283         if (p->p_pid == 1) {
284                 kprintf("init died (signal %d, exit %d)\n",
285                     WTERMSIG(rv), WEXITSTATUS(rv));
286                 panic("Going nowhere without my init!");
287         }
288         varsymset_clean(&p->p_varsymset);
289         lockuninit(&p->p_varsymset.vx_lock);
290         /*
291          * Kill all lwps associated with the current process, return an
292          * error if we race another thread trying to do the same thing
293          * and lose the race.
294          */
295         error = killalllwps(0);
296         if (error) {
297                 lwp_exit(0);
298                 /* NOT REACHED */
299         }
300
301         caps_exit(lp->lwp_thread);
302         aio_proc_rundown(p);
303
304         /* are we a task leader? */
305         if (p == p->p_leader) {
306                 struct kill_args killArgs;
307                 killArgs.signum = SIGKILL;
308                 q = p->p_peers;
309                 while(q) {
310                         killArgs.pid = q->p_pid;
311                         /*
312                          * The interface for kill is better
313                          * than the internal signal
314                          */
315                         sys_kill(&killArgs);
316                         nq = q;
317                         q = q->p_peers;
318                 }
319                 while (p->p_peers) 
320                         tsleep((caddr_t)p, 0, "exit1", 0);
321         }
322
323 #ifdef PGINPROF
324         vmsizmon();
325 #endif
326         STOPEVENT(p, S_EXIT, rv);
327         wakeup(&p->p_stype);    /* Wakeup anyone in procfs' PIOCWAIT */
328
329         /* 
330          * Check if any loadable modules need anything done at process exit.
331          * e.g. SYSV IPC stuff
332          * XXX what if one of these generates an error?
333          */
334         p->p_xstat = rv;
335         EVENTHANDLER_INVOKE(process_exit, p);
336
337         /*
338          * XXX: imho, the eventhandler stuff is much cleaner than this.
339          *      Maybe we should move everything to use eventhandler.
340          */
341         TAILQ_FOREACH(ep, &exit_list, next) 
342                 (*ep->function)(td);
343
344         if (p->p_flag & P_PROFIL)
345                 stopprofclock(p);
346         /*
347          * If parent is waiting for us to exit or exec,
348          * P_PPWAIT is set; we will wakeup the parent below.
349          */
350         p->p_flag &= ~(P_TRACED | P_PPWAIT);
351         SIGEMPTYSET(p->p_siglist);
352         SIGEMPTYSET(lp->lwp_siglist);
353         if (timevalisset(&p->p_realtimer.it_value))
354                 callout_stop(&p->p_ithandle);
355
356         /*
357          * Reset any sigio structures pointing to us as a result of
358          * F_SETOWN with our pid.
359          */
360         funsetownlst(&p->p_sigiolst);
361
362         /*
363          * Close open files and release open-file table.
364          * This may block!
365          */
366         fdfree(p, NULL);
367
368         if(p->p_leader->p_peers) {
369                 q = p->p_leader;
370                 while(q->p_peers != p)
371                         q = q->p_peers;
372                 q->p_peers = p->p_peers;
373                 wakeup((caddr_t)p->p_leader);
374         }
375
376         /*
377          * XXX Shutdown SYSV semaphores
378          */
379         semexit(p);
380
381         KKASSERT(p->p_numposixlocks == 0);
382
383         /* The next two chunks should probably be moved to vmspace_exit. */
384         vm = p->p_vmspace;
385
386         /*
387          * Release upcalls associated with this process
388          */
389         if (vm->vm_upcalls)
390                 upc_release(vm, lp);
391
392         /*
393          * Clean up data related to virtual kernel operation.  Clean up
394          * any vkernel context related to the current lwp now so we can
395          * destroy p_vkernel.
396          */
397         if (p->p_vkernel) {
398                 vkernel_lwp_exit(lp);
399                 vkernel_exit(p);
400         }
401
402         /*
403          * Release user portion of address space.
404          * This releases references to vnodes,
405          * which could cause I/O if the file has been unlinked.
406          * Need to do this early enough that we can still sleep.
407          * Can't free the entire vmspace as the kernel stack
408          * may be mapped within that space also.
409          *
410          * Processes sharing the same vmspace may exit in one order, and
411          * get cleaned up by vmspace_exit() in a different order.  The
412          * last exiting process to reach this point releases as much of
413          * the environment as it can, and the last process cleaned up
414          * by vmspace_exit() (which decrements exitingcnt) cleans up the
415          * remainder.
416          */
417         vmspace_exitbump(vm);
418         sysref_put(&vm->vm_sysref);
419
420         if (SESS_LEADER(p)) {
421                 struct session *sp = p->p_session;
422
423                 if (sp->s_ttyvp) {
424                         /*
425                          * We are the controlling process.  Signal the 
426                          * foreground process group, drain the controlling
427                          * terminal, and revoke access to the controlling
428                          * terminal.
429                          *
430                          * NOTE: while waiting for the process group to exit
431                          * it is possible that one of the processes in the
432                          * group will revoke the tty, so the ttyclosesession()
433                          * function will re-check sp->s_ttyvp.
434                          */
435                         if (sp->s_ttyp && (sp->s_ttyp->t_session == sp)) {
436                                 if (sp->s_ttyp->t_pgrp)
437                                         pgsignal(sp->s_ttyp->t_pgrp, SIGHUP, 1);
438                                 ttywait(sp->s_ttyp);
439                                 ttyclosesession(sp, 1); /* also revoke */
440                         }
441                         /*
442                          * Release the tty.  If someone has it open via
443                          * /dev/tty then close it (since they no longer can
444                          * once we've NULL'd it out).
445                          */
446                         ttyclosesession(sp, 0);
447
448                         /*
449                          * s_ttyp is not zero'd; we use this to indicate
450                          * that the session once had a controlling terminal.
451                          * (for logging and informational purposes)
452                          */
453                 }
454                 sp->s_leader = NULL;
455         }
456         fixjobc(p, p->p_pgrp, 0);
457         (void)acct_process(p);
458 #ifdef KTRACE
459         /*
460          * release trace file
461          */
462         if (p->p_tracenode)
463                 ktrdestroy(&p->p_tracenode);
464         p->p_traceflag = 0;
465 #endif
466         /*
467          * Release reference to text vnode
468          */
469         if ((vtmp = p->p_textvp) != NULL) {
470                 p->p_textvp = NULL;
471                 vrele(vtmp);
472         }
473
474         /* Release namecache handle to text file */
475         if (p->p_textnch.ncp)
476                 cache_drop(&p->p_textnch);
477
478         /*
479          * Move the process to the zombie list.  This will block
480          * until the process p_lock count reaches 0.  The process will
481          * not be reaped until TDF_EXITING is set by cpu_thread_exit(),
482          * which is called from cpu_proc_exit().
483          */
484         proc_move_allproc_zombie(p);
485
486         q = LIST_FIRST(&p->p_children);
487         if (q)          /* only need this if any child is S_ZOMB */
488                 wakeup((caddr_t) initproc);
489         for (; q != 0; q = nq) {
490                 nq = LIST_NEXT(q, p_sibling);
491                 LIST_REMOVE(q, p_sibling);
492                 LIST_INSERT_HEAD(&initproc->p_children, q, p_sibling);
493                 q->p_pptr = initproc;
494                 q->p_sigparent = SIGCHLD;
495                 /*
496                  * Traced processes are killed
497                  * since their existence means someone is screwing up.
498                  */
499                 if (q->p_flag & P_TRACED) {
500                         q->p_flag &= ~P_TRACED;
501                         ksignal(q, SIGKILL);
502                 }
503         }
504
505         /*
506          * Save exit status and final rusage info, adding in child rusage
507          * info and self times.
508          */
509         calcru_proc(p, &p->p_ru);
510         ruadd(&p->p_ru, &p->p_cru);
511
512         /*
513          * notify interested parties of our demise.
514          */
515         KNOTE(&p->p_klist, NOTE_EXIT);
516
517         /*
518          * Notify parent that we're gone.  If parent has the PS_NOCLDWAIT
519          * flag set, notify process 1 instead (and hope it will handle
520          * this situation).
521          */
522         if (p->p_pptr->p_sigacts->ps_flag & PS_NOCLDWAIT) {
523                 struct proc *pp = p->p_pptr;
524                 proc_reparent(p, initproc);
525                 /*
526                  * If this was the last child of our parent, notify
527                  * parent, so in case he was wait(2)ing, he will
528                  * continue.
529                  */
530                 if (LIST_EMPTY(&pp->p_children))
531                         wakeup((caddr_t)pp);
532         }
533
534         /* lwkt_gettoken(&proc_token); */
535         q = p->p_pptr;
536         if (p->p_sigparent && q != initproc) {
537                 PHOLD(q);
538                 ksignal(q, p->p_sigparent);
539                 PRELE(q);
540         } else {
541                 ksignal(q, SIGCHLD);
542         }
543         /* lwkt_reltoken(&proc_token); */
544         /* NOTE: p->p_pptr can get ripped out */
545
546         wakeup(p->p_pptr);
547         /*
548          * cpu_exit is responsible for clearing curproc, since
549          * it is heavily integrated with the thread/switching sequence.
550          *
551          * Other substructures are freed from wait().
552          */
553         plimit_free(p);
554
555         /*
556          * Release the current user process designation on the process so
557          * the userland scheduler can work in someone else.
558          */
559         p->p_usched->release_curproc(lp);
560
561         /*
562          * Finally, call machine-dependent code to release as many of the
563          * lwp's resources as we can and halt execution of this thread.
564          */
565         lwp_exit(1);
566 }
567
568 /*
569  * Eventually called by every exiting LWP
570  *
571  * p->p_token must be held.  mplock may be held and will be released.
572  */
573 void
574 lwp_exit(int masterexit)
575 {
576         struct thread *td = curthread;
577         struct lwp *lp = td->td_lwp;
578         struct proc *p = lp->lwp_proc;
579
580         /*
581          * lwp_exit() may be called without setting LWP_WEXIT, so
582          * make sure it is set here.
583          */
584         ASSERT_LWKT_TOKEN_HELD(&p->p_token);
585         lp->lwp_flag |= LWP_WEXIT;
586
587         /*
588          * Clean up any virtualization
589          */
590         if (lp->lwp_vkernel)
591                 vkernel_lwp_exit(lp);
592
593         /*
594          * Clean up select/poll support
595          */
596         kqueue_terminate(&lp->lwp_kqueue);
597
598         /*
599          * Clean up any syscall-cached ucred
600          */
601         if (td->td_ucred) {
602                 crfree(td->td_ucred);
603                 td->td_ucred = NULL;
604         }
605
606         /*
607          * Nobody actually wakes us when the lock
608          * count reaches zero, so just wait one tick.
609          */
610         while (lp->lwp_lock > 0)
611                 tsleep(lp, 0, "lwpexit", 1);
612
613         /* Hand down resource usage to our proc */
614         ruadd(&p->p_ru, &lp->lwp_ru);
615
616         /*
617          * If we don't hold the process until the LWP is reaped wait*()
618          * may try to dispose of its vmspace before all the LWPs have
619          * actually terminated.
620          */
621         PHOLD(p);
622
623         /*
624          * Do any remaining work that might block on us.  We should be
625          * coded such that further blocking is ok after decrementing
626          * p_nthreads but don't take the chance.
627          */
628         dsched_exit_thread(td);
629         biosched_done(curthread);
630
631         /*
632          * We have to use the reaper for all the LWPs except the one doing
633          * the master exit.  The LWP doing the master exit can just be
634          * left on p_lwps and the process reaper will deal with it
635          * synchronously, which is much faster.
636          *
637          * Wakeup anyone waiting on p_nthreads to drop to 1 or 0.
638          */
639         if (masterexit == 0) {
640                 lwp_rb_tree_RB_REMOVE(&p->p_lwp_tree, lp);
641                 --p->p_nthreads;
642                 if (p->p_nthreads <= 1)
643                         wakeup(&p->p_nthreads);
644                 lwkt_gettoken(&deadlwp_token);
645                 LIST_INSERT_HEAD(&deadlwp_list[mycpuid], lp, u.lwp_reap_entry);
646                 taskqueue_enqueue(taskqueue_thread[mycpuid],
647                                   deadlwp_task[mycpuid]);
648                 lwkt_reltoken(&deadlwp_token);
649         } else {
650                 --p->p_nthreads;
651                 if (p->p_nthreads <= 1)
652                         wakeup(&p->p_nthreads);
653         }
654
655         /*
656          * Release p_token.  The mp_token may also be held and we depend on
657          * the lwkt_switch() code to clean it up.
658          */
659         lwkt_reltoken(&p->p_token);
660         cpu_lwp_exit();
661 }
662
663 /*
664  * Wait until a lwp is completely dead.
665  *
666  * If the thread is still executing, which can't be waited upon,
667  * return failure.  The caller is responsible of waiting a little
668  * bit and checking again.
669  *
670  * Suggested use:
671  * while (!lwp_wait(lp))
672  *      tsleep(lp, 0, "lwpwait", 1);
673  */
674 static int
675 lwp_wait(struct lwp *lp)
676 {
677         struct thread *td = lp->lwp_thread;;
678
679         KKASSERT(lwkt_preempted_proc() != lp);
680
681         while (lp->lwp_lock > 0)
682                 tsleep(lp, 0, "lwpwait1", 1);
683
684         lwkt_wait_free(td);
685
686         /*
687          * The lwp's thread may still be in the middle
688          * of switching away, we can't rip its stack out from
689          * under it until TDF_EXITING is set and both
690          * TDF_RUNNING and TDF_PREEMPT_LOCK are clear.
691          * TDF_PREEMPT_LOCK must be checked because TDF_RUNNING
692          * will be cleared temporarily if a thread gets
693          * preempted.
694          *
695          * YYY no wakeup occurs, so we simply return failure
696          * and let the caller deal with sleeping and calling
697          * us again.
698          */
699         if ((td->td_flags & (TDF_RUNNING|TDF_PREEMPT_LOCK|
700                              TDF_EXITING|TDF_RUNQ)) != TDF_EXITING) {
701                 return (0);
702         }
703         KASSERT((td->td_flags & TDF_TSLEEPQ) == 0,
704                 ("lwp_wait: td %p (%s) still on sleep queue", td, td->td_comm));
705         return (1);
706 }
707
708 /*
709  * Release the resources associated with a lwp.
710  * The lwp must be completely dead.
711  */
712 void
713 lwp_dispose(struct lwp *lp)
714 {
715         struct thread *td = lp->lwp_thread;;
716
717         KKASSERT(lwkt_preempted_proc() != lp);
718         KKASSERT(td->td_refs == 0);
719         KKASSERT((td->td_flags & (TDF_RUNNING|TDF_PREEMPT_LOCK|TDF_EXITING)) ==
720                  TDF_EXITING);
721
722         PRELE(lp->lwp_proc);
723         lp->lwp_proc = NULL;
724         if (td != NULL) {
725                 td->td_proc = NULL;
726                 td->td_lwp = NULL;
727                 lp->lwp_thread = NULL;
728                 lwkt_free_thread(td);
729         }
730         kfree(lp, M_LWP);
731 }
732
733 /*
734  * MPSAFE
735  */
736 int
737 sys_wait4(struct wait_args *uap)
738 {
739         struct rusage rusage;
740         int error, status;
741
742         error = kern_wait(uap->pid, (uap->status ? &status : NULL),
743                           uap->options, (uap->rusage ? &rusage : NULL),
744                           &uap->sysmsg_result);
745
746         if (error == 0 && uap->status)
747                 error = copyout(&status, uap->status, sizeof(*uap->status));
748         if (error == 0 && uap->rusage)
749                 error = copyout(&rusage, uap->rusage, sizeof(*uap->rusage));
750         return (error);
751 }
752
753 /*
754  * wait1()
755  *
756  * wait_args(int pid, int *status, int options, struct rusage *rusage)
757  *
758  * MPALMOSTSAFE
759  */
760 int
761 kern_wait(pid_t pid, int *status, int options, struct rusage *rusage, int *res)
762 {
763         struct thread *td = curthread;
764         struct lwp *lp;
765         struct proc *q = td->td_proc;
766         struct proc *p, *t;
767         int nfound, error;
768
769         if (pid == 0)
770                 pid = -q->p_pgid;
771         if (options &~ (WUNTRACED|WNOHANG|WCONTINUED|WLINUXCLONE))
772                 return (EINVAL);
773
774         lwkt_gettoken(&q->p_token);
775 loop:
776         /*
777          * All sorts of things can change due to blocking so we have to loop
778          * all the way back up here.
779          *
780          * The problem is that if a process group is stopped and the parent
781          * is doing a wait*(..., WUNTRACED, ...), it will see the STOP
782          * of the child and then stop itself when it tries to return from the
783          * system call.  When the process group is resumed the parent will
784          * then get the STOP status even though the child has now resumed
785          * (a followup wait*() will get the CONT status).
786          *
787          * Previously the CONT would overwrite the STOP because the tstop
788          * was handled within tsleep(), and the parent would only see
789          * the CONT when both are stopped and continued together.  This little
790          * two-line hack restores this effect.
791          */
792         while (q->p_stat == SSTOP)
793             tstop();
794
795         nfound = 0;
796
797         LIST_FOREACH(p, &q->p_children, p_sibling) {
798                 if (pid != WAIT_ANY &&
799                     p->p_pid != pid && p->p_pgid != -pid) {
800                         continue;
801                 }
802
803                 /*
804                  * This special case handles a kthread spawned by linux_clone
805                  * (see linux_misc.c).  The linux_wait4 and linux_waitpid 
806                  * functions need to be able to distinguish between waiting
807                  * on a process and waiting on a thread.  It is a thread if
808                  * p_sigparent is not SIGCHLD, and the WLINUXCLONE option
809                  * signifies we want to wait for threads and not processes.
810                  */
811                 if ((p->p_sigparent != SIGCHLD) ^ 
812                     ((options & WLINUXCLONE) != 0)) {
813                         continue;
814                 }
815
816                 nfound++;
817                 if (p->p_stat == SZOMB) {
818                         /*
819                          * We may go into SZOMB with threads still present.
820                          * We must wait for them to exit before we can reap
821                          * the master thread, otherwise we may race reaping
822                          * non-master threads.
823                          */
824                         lwkt_gettoken(&p->p_token);
825                         while (p->p_nthreads > 0) {
826                                 tsleep(&p->p_nthreads, 0, "lwpzomb", hz);
827                         }
828
829                         /*
830                          * Reap any LWPs left in p->p_lwps.  This is usually
831                          * just the last LWP.  This must be done before
832                          * we loop on p_lock since the lwps hold a ref on
833                          * it as a vmspace interlock.
834                          *
835                          * Once that is accomplished p_nthreads had better
836                          * be zero.
837                          */
838                         while ((lp = RB_ROOT(&p->p_lwp_tree)) != NULL) {
839                                 lwp_rb_tree_RB_REMOVE(&p->p_lwp_tree, lp);
840                                 reaplwp(lp);
841                         }
842                         KKASSERT(p->p_nthreads == 0);
843                         lwkt_reltoken(&p->p_token);
844
845                         /*
846                          * Don't do anything really bad until all references
847                          * to the process go away.  This may include other
848                          * LWPs which are still in the process of being
849                          * reaped.  We can't just pull the rug out from under
850                          * them because they may still be using the VM space.
851                          *
852                          * Certain kernel facilities such as /proc will also
853                          * put a hold on the process for short periods of
854                          * time.
855                          */
856                         while (p->p_lock)
857                                 tsleep(p, 0, "reap3", hz);
858
859                         /* Take care of our return values. */
860                         *res = p->p_pid;
861                         if (status)
862                                 *status = p->p_xstat;
863                         if (rusage)
864                                 *rusage = p->p_ru;
865                         /*
866                          * If we got the child via a ptrace 'attach',
867                          * we need to give it back to the old parent.
868                          */
869                         if (p->p_oppid && (t = pfind(p->p_oppid))) {
870                                 p->p_oppid = 0;
871                                 proc_reparent(p, t);
872                                 ksignal(t, SIGCHLD);
873                                 wakeup((caddr_t)t);
874                                 error = 0;
875                                 goto done;
876                         }
877
878                         /*
879                          * Unlink the proc from its process group so that
880                          * the following operations won't lead to an
881                          * inconsistent state for processes running down
882                          * the zombie list.
883                          */
884                         KKASSERT(p->p_lock == 0);
885                         proc_remove_zombie(p);
886                         leavepgrp(p);
887
888                         p->p_xstat = 0;
889                         ruadd(&q->p_cru, &p->p_ru);
890
891                         /*
892                          * Decrement the count of procs running with this uid.
893                          */
894                         chgproccnt(p->p_ucred->cr_ruidinfo, -1, 0);
895
896                         /*
897                          * Free up credentials.
898                          */
899                         crfree(p->p_ucred);
900                         p->p_ucred = NULL;
901
902                         /*
903                          * Remove unused arguments
904                          */
905                         if (p->p_args && --p->p_args->ar_ref == 0)
906                                 FREE(p->p_args, M_PARGS);
907
908                         if (--p->p_sigacts->ps_refcnt == 0) {
909                                 kfree(p->p_sigacts, M_SUBPROC);
910                                 p->p_sigacts = NULL;
911                         }
912
913                         vm_waitproc(p);
914                         kfree(p, M_PROC);
915                         nprocs--;
916                         error = 0;
917                         goto done;
918                 }
919                 if (p->p_stat == SSTOP && (p->p_flag & P_WAITED) == 0 &&
920                     (p->p_flag & P_TRACED || options & WUNTRACED)) {
921                         p->p_flag |= P_WAITED;
922
923                         *res = p->p_pid;
924                         if (status)
925                                 *status = W_STOPCODE(p->p_xstat);
926                         /* Zero rusage so we get something consistent. */
927                         if (rusage)
928                                 bzero(rusage, sizeof(rusage));
929                         error = 0;
930                         goto done;
931                 }
932                 if (options & WCONTINUED && (p->p_flag & P_CONTINUED)) {
933                         *res = p->p_pid;
934                         p->p_flag &= ~P_CONTINUED;
935
936                         if (status)
937                                 *status = SIGCONT;
938                         error = 0;
939                         goto done;
940                 }
941         }
942         if (nfound == 0) {
943                 error = ECHILD;
944                 goto done;
945         }
946         if (options & WNOHANG) {
947                 *res = 0;
948                 error = 0;
949                 goto done;
950         }
951         error = tsleep((caddr_t)q, PCATCH, "wait", 0);
952         if (error) {
953 done:
954                 lwkt_reltoken(&q->p_token);
955                 return (error);
956         }
957         goto loop;
958 }
959
960 /*
961  * Make process 'parent' the new parent of process 'child'.
962  */
963 void
964 proc_reparent(struct proc *child, struct proc *parent)
965 {
966         if (child->p_pptr == parent)
967                 return;
968         PHOLD(parent);
969         lwkt_gettoken(&child->p_token);
970         lwkt_gettoken(&parent->p_token);
971         LIST_REMOVE(child, p_sibling);
972         LIST_INSERT_HEAD(&parent->p_children, child, p_sibling);
973         child->p_pptr = parent;
974         lwkt_reltoken(&parent->p_token);
975         lwkt_reltoken(&child->p_token);
976         PRELE(parent);
977 }
978
979 /*
980  * The next two functions are to handle adding/deleting items on the
981  * exit callout list
982  * 
983  * at_exit():
984  * Take the arguments given and put them onto the exit callout list,
985  * However first make sure that it's not already there.
986  * returns 0 on success.
987  */
988
989 int
990 at_exit(exitlist_fn function)
991 {
992         struct exitlist *ep;
993
994 #ifdef INVARIANTS
995         /* Be noisy if the programmer has lost track of things */
996         if (rm_at_exit(function)) 
997                 kprintf("WARNING: exit callout entry (%p) already present\n",
998                     function);
999 #endif
1000         ep = kmalloc(sizeof(*ep), M_ATEXIT, M_NOWAIT);
1001         if (ep == NULL)
1002                 return (ENOMEM);
1003         ep->function = function;
1004         TAILQ_INSERT_TAIL(&exit_list, ep, next);
1005         return (0);
1006 }
1007
1008 /*
1009  * Scan the exit callout list for the given item and remove it.
1010  * Returns the number of items removed (0 or 1)
1011  */
1012 int
1013 rm_at_exit(exitlist_fn function)
1014 {
1015         struct exitlist *ep;
1016
1017         TAILQ_FOREACH(ep, &exit_list, next) {
1018                 if (ep->function == function) {
1019                         TAILQ_REMOVE(&exit_list, ep, next);
1020                         kfree(ep, M_ATEXIT);
1021                         return(1);
1022                 }
1023         }       
1024         return (0);
1025 }
1026
1027 /*
1028  * LWP reaper related code.
1029  */
1030 static void
1031 reaplwps(void *context, int dummy)
1032 {
1033         struct lwplist *lwplist = context;
1034         struct lwp *lp;
1035
1036         lwkt_gettoken(&deadlwp_token);
1037         while ((lp = LIST_FIRST(lwplist))) {
1038                 LIST_REMOVE(lp, u.lwp_reap_entry);
1039                 reaplwp(lp);
1040         }
1041         lwkt_reltoken(&deadlwp_token);
1042 }
1043
1044 static void
1045 reaplwp(struct lwp *lp)
1046 {
1047         while (lwp_wait(lp) == 0)
1048                 tsleep(lp, 0, "lwpreap", 1);
1049         lwp_dispose(lp);
1050 }
1051
1052 static void
1053 deadlwp_init(void)
1054 {
1055         int cpu;
1056
1057         for (cpu = 0; cpu < ncpus; cpu++) {
1058                 LIST_INIT(&deadlwp_list[cpu]);
1059                 deadlwp_task[cpu] = kmalloc(sizeof(*deadlwp_task[cpu]), M_DEVBUF, M_WAITOK);
1060                 TASK_INIT(deadlwp_task[cpu], 0, reaplwps, &deadlwp_list[cpu]);
1061         }
1062 }
1063
1064 SYSINIT(deadlwpinit, SI_SUB_CONFIGURE, SI_ORDER_ANY, deadlwp_init, NULL);