xref: /freebsd/sys/kern/kern_thread.c (revision 374ae2a39338777f5f9ef0ed726fff2a5dee6485)
19454b2d8SWarner Losh /*-
244990b8cSJulian Elischer  * Copyright (C) 2001 Julian Elischer <julian@freebsd.org>.
344990b8cSJulian Elischer  *  All rights reserved.
444990b8cSJulian Elischer  *
544990b8cSJulian Elischer  * Redistribution and use in source and binary forms, with or without
644990b8cSJulian Elischer  * modification, are permitted provided that the following conditions
744990b8cSJulian Elischer  * are met:
844990b8cSJulian Elischer  * 1. Redistributions of source code must retain the above copyright
944990b8cSJulian Elischer  *    notice(s), this list of conditions and the following disclaimer as
1044990b8cSJulian Elischer  *    the first lines of this file unmodified other than the possible
1144990b8cSJulian Elischer  *    addition of one or more copyright notices.
1244990b8cSJulian Elischer  * 2. Redistributions in binary form must reproduce the above copyright
1344990b8cSJulian Elischer  *    notice(s), this list of conditions and the following disclaimer in the
1444990b8cSJulian Elischer  *    documentation and/or other materials provided with the distribution.
1544990b8cSJulian Elischer  *
1644990b8cSJulian Elischer  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) ``AS IS'' AND ANY
1744990b8cSJulian Elischer  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
1844990b8cSJulian Elischer  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
1944990b8cSJulian Elischer  * DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) BE LIABLE FOR ANY
2044990b8cSJulian Elischer  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
2144990b8cSJulian Elischer  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
2244990b8cSJulian Elischer  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
2344990b8cSJulian Elischer  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
2444990b8cSJulian Elischer  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
2544990b8cSJulian Elischer  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
2644990b8cSJulian Elischer  * DAMAGE.
2744990b8cSJulian Elischer  */
2844990b8cSJulian Elischer 
29677b542eSDavid E. O'Brien #include <sys/cdefs.h>
30677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
31677b542eSDavid E. O'Brien 
3244990b8cSJulian Elischer #include <sys/param.h>
3344990b8cSJulian Elischer #include <sys/systm.h>
3444990b8cSJulian Elischer #include <sys/kernel.h>
3544990b8cSJulian Elischer #include <sys/lock.h>
3644990b8cSJulian Elischer #include <sys/mutex.h>
3744990b8cSJulian Elischer #include <sys/proc.h>
38e170bfdaSDavid Xu #include <sys/resourcevar.h>
3994e0a4cdSJulian Elischer #include <sys/smp.h>
4044990b8cSJulian Elischer #include <sys/sysctl.h>
41de028f5aSJeff Roberson #include <sys/sched.h>
4244f3b092SJohn Baldwin #include <sys/sleepqueue.h>
43ace8398dSJeff Roberson #include <sys/selinfo.h>
44961a7b24SJohn Baldwin #include <sys/turnstile.h>
4544990b8cSJulian Elischer #include <sys/ktr.h>
46bc8e6d81SDavid Xu #include <sys/umtx.h>
47d7f687fcSJeff Roberson #include <sys/cpuset.h>
4844990b8cSJulian Elischer 
49911b84b0SRobert Watson #include <security/audit/audit.h>
50911b84b0SRobert Watson 
5144990b8cSJulian Elischer #include <vm/vm.h>
5249a2507bSAlan Cox #include <vm/vm_extern.h>
5344990b8cSJulian Elischer #include <vm/uma.h>
54b209f889SRandall Stewart #include <sys/eventhandler.h>
5502fb42b0SPeter Wemm 
568460a577SJohn Birrell /*
578460a577SJohn Birrell  * thread related storage.
588460a577SJohn Birrell  */
5944990b8cSJulian Elischer static uma_zone_t thread_zone;
6044990b8cSJulian Elischer 
6144990b8cSJulian Elischer SYSCTL_NODE(_kern, OID_AUTO, threads, CTLFLAG_RW, 0, "thread allocation");
62fdc5ecd2SDavid Xu 
63345ad866SJulian Elischer int max_threads_per_proc = 1500;
64fdc5ecd2SDavid Xu SYSCTL_INT(_kern_threads, OID_AUTO, max_threads_per_proc, CTLFLAG_RW,
654f0db5e0SJulian Elischer 	&max_threads_per_proc, 0, "Limit on threads per proc");
664f0db5e0SJulian Elischer 
67345ad866SJulian Elischer int max_threads_hits;
680252d203SDavid Xu SYSCTL_INT(_kern_threads, OID_AUTO, max_threads_hits, CTLFLAG_RD,
690252d203SDavid Xu 	&max_threads_hits, 0, "");
700252d203SDavid Xu 
715215b187SJeff Roberson TAILQ_HEAD(, thread) zombie_threads = TAILQ_HEAD_INITIALIZER(zombie_threads);
72c8790f5dSAttilio Rao static struct mtx zombie_lock;
73a54e85fdSJeff Roberson MTX_SYSINIT(zombie_lock, &zombie_lock, "zombie lock", MTX_SPIN);
7444990b8cSJulian Elischer 
75ff8fbcffSJeff Roberson static void thread_zombie(struct thread *);
76ff8fbcffSJeff Roberson 
77fdcac928SMarcel Moolenaar struct mtx tid_lock;
781ea7a6f8SPoul-Henning Kamp static struct unrhdr *tid_unrhdr;
79fdcac928SMarcel Moolenaar 
80fdcac928SMarcel Moolenaar /*
81696058c3SJulian Elischer  * Prepare a thread for use.
8244990b8cSJulian Elischer  */
83b23f72e9SBrian Feldman static int
84b23f72e9SBrian Feldman thread_ctor(void *mem, int size, void *arg, int flags)
8544990b8cSJulian Elischer {
8644990b8cSJulian Elischer 	struct thread	*td;
8744990b8cSJulian Elischer 
8844990b8cSJulian Elischer 	td = (struct thread *)mem;
8971fad9fdSJulian Elischer 	td->td_state = TDS_INACTIVE;
90060563ecSJulian Elischer 	td->td_oncpu = NOCPU;
916c27c603SJuli Mallett 
92773eff9dSPoul-Henning Kamp 	td->td_tid = alloc_unr(tid_unrhdr);
93f9bb7538SMohan Srinivasan 	td->td_syscalls = 0;
94773eff9dSPoul-Henning Kamp 
956c27c603SJuli Mallett 	/*
966c27c603SJuli Mallett 	 * Note that td_critnest begins life as 1 because the thread is not
976c27c603SJuli Mallett 	 * running and is thereby implicitly waiting to be on the receiving
98a54e85fdSJeff Roberson 	 * end of a context switch.
996c27c603SJuli Mallett 	 */
100139b7550SJohn Baldwin 	td->td_critnest = 1;
101b209f889SRandall Stewart 	EVENTHANDLER_INVOKE(thread_ctor, td);
102911b84b0SRobert Watson #ifdef AUDIT
103911b84b0SRobert Watson 	audit_thread_alloc(td);
104911b84b0SRobert Watson #endif
105d10183d9SDavid Xu 	umtx_thread_alloc(td);
106b23f72e9SBrian Feldman 	return (0);
10744990b8cSJulian Elischer }
10844990b8cSJulian Elischer 
10944990b8cSJulian Elischer /*
11044990b8cSJulian Elischer  * Reclaim a thread after use.
11144990b8cSJulian Elischer  */
11244990b8cSJulian Elischer static void
11344990b8cSJulian Elischer thread_dtor(void *mem, int size, void *arg)
11444990b8cSJulian Elischer {
11544990b8cSJulian Elischer 	struct thread *td;
11644990b8cSJulian Elischer 
11744990b8cSJulian Elischer 	td = (struct thread *)mem;
11844990b8cSJulian Elischer 
11944990b8cSJulian Elischer #ifdef INVARIANTS
12044990b8cSJulian Elischer 	/* Verify that this thread is in a safe state to free. */
12144990b8cSJulian Elischer 	switch (td->td_state) {
12271fad9fdSJulian Elischer 	case TDS_INHIBITED:
12371fad9fdSJulian Elischer 	case TDS_RUNNING:
12471fad9fdSJulian Elischer 	case TDS_CAN_RUN:
12544990b8cSJulian Elischer 	case TDS_RUNQ:
12644990b8cSJulian Elischer 		/*
12744990b8cSJulian Elischer 		 * We must never unlink a thread that is in one of
12844990b8cSJulian Elischer 		 * these states, because it is currently active.
12944990b8cSJulian Elischer 		 */
13044990b8cSJulian Elischer 		panic("bad state for thread unlinking");
13144990b8cSJulian Elischer 		/* NOTREACHED */
13271fad9fdSJulian Elischer 	case TDS_INACTIVE:
13344990b8cSJulian Elischer 		break;
13444990b8cSJulian Elischer 	default:
13544990b8cSJulian Elischer 		panic("bad thread state");
13644990b8cSJulian Elischer 		/* NOTREACHED */
13744990b8cSJulian Elischer 	}
13844990b8cSJulian Elischer #endif
1396e8525ceSRobert Watson #ifdef AUDIT
1406e8525ceSRobert Watson 	audit_thread_free(td);
1416e8525ceSRobert Watson #endif
142b209f889SRandall Stewart 	EVENTHANDLER_INVOKE(thread_dtor, td);
143773eff9dSPoul-Henning Kamp 	free_unr(tid_unrhdr, td->td_tid);
144ed062c8dSJulian Elischer 	sched_newthread(td);
14544990b8cSJulian Elischer }
14644990b8cSJulian Elischer 
14744990b8cSJulian Elischer /*
14844990b8cSJulian Elischer  * Initialize type-stable parts of a thread (when newly created).
14944990b8cSJulian Elischer  */
150b23f72e9SBrian Feldman static int
151b23f72e9SBrian Feldman thread_init(void *mem, int size, int flags)
15244990b8cSJulian Elischer {
15344990b8cSJulian Elischer 	struct thread *td;
15444990b8cSJulian Elischer 
15544990b8cSJulian Elischer 	td = (struct thread *)mem;
156247aba24SMarcel Moolenaar 
15744f3b092SJohn Baldwin 	td->td_sleepqueue = sleepq_alloc();
158961a7b24SJohn Baldwin 	td->td_turnstile = turnstile_alloc();
159b209f889SRandall Stewart 	EVENTHANDLER_INVOKE(thread_init, td);
160de028f5aSJeff Roberson 	td->td_sched = (struct td_sched *)&td[1];
161ed062c8dSJulian Elischer 	sched_newthread(td);
162d10183d9SDavid Xu 	umtx_thread_init(td);
16389b57fcfSKonstantin Belousov 	td->td_kstack = 0;
164b23f72e9SBrian Feldman 	return (0);
16544990b8cSJulian Elischer }
16644990b8cSJulian Elischer 
16744990b8cSJulian Elischer /*
16844990b8cSJulian Elischer  * Tear down type-stable parts of a thread (just before being discarded).
16944990b8cSJulian Elischer  */
17044990b8cSJulian Elischer static void
17144990b8cSJulian Elischer thread_fini(void *mem, int size)
17244990b8cSJulian Elischer {
17344990b8cSJulian Elischer 	struct thread *td;
17444990b8cSJulian Elischer 
17544990b8cSJulian Elischer 	td = (struct thread *)mem;
176b209f889SRandall Stewart 	EVENTHANDLER_INVOKE(thread_fini, td);
177961a7b24SJohn Baldwin 	turnstile_free(td->td_turnstile);
17844f3b092SJohn Baldwin 	sleepq_free(td->td_sleepqueue);
179d10183d9SDavid Xu 	umtx_thread_fini(td);
180ace8398dSJeff Roberson 	seltdfini(td);
18144990b8cSJulian Elischer }
1825215b187SJeff Roberson 
1835c8329edSJulian Elischer /*
1845215b187SJeff Roberson  * For a newly created process,
1855215b187SJeff Roberson  * link up all the structures and its initial threads etc.
186ed062c8dSJulian Elischer  * called from:
187ed062c8dSJulian Elischer  * {arch}/{arch}/machdep.c   ia64_init(), init386() etc.
188ed062c8dSJulian Elischer  * proc_dtor() (should go away)
189ed062c8dSJulian Elischer  * proc_init()
1905c8329edSJulian Elischer  */
1915c8329edSJulian Elischer void
19289b57fcfSKonstantin Belousov proc_linkup0(struct proc *p, struct thread *td)
19389b57fcfSKonstantin Belousov {
19489b57fcfSKonstantin Belousov 	TAILQ_INIT(&p->p_threads);	     /* all threads in proc */
19589b57fcfSKonstantin Belousov 	proc_linkup(p, td);
19689b57fcfSKonstantin Belousov }
19789b57fcfSKonstantin Belousov 
19889b57fcfSKonstantin Belousov void
1998460a577SJohn Birrell proc_linkup(struct proc *p, struct thread *td)
2005c8329edSJulian Elischer {
201a54e85fdSJeff Roberson 
2029104847fSDavid Xu 	sigqueue_init(&p->p_sigqueue, p);
203ebceaf6dSDavid Xu 	p->p_ksi = ksiginfo_alloc(1);
204ebceaf6dSDavid Xu 	if (p->p_ksi != NULL) {
2055c474517SDavid Xu 		/* XXX p_ksi may be null if ksiginfo zone is not ready */
206ebceaf6dSDavid Xu 		p->p_ksi->ksi_flags = KSI_EXT | KSI_INS;
207ebceaf6dSDavid Xu 	}
208b2f92ef9SDavid Xu 	LIST_INIT(&p->p_mqnotifier);
2095c8329edSJulian Elischer 	p->p_numthreads = 0;
2108460a577SJohn Birrell 	thread_link(td, p);
2115c8329edSJulian Elischer }
2125c8329edSJulian Elischer 
2135c8329edSJulian Elischer /*
21444990b8cSJulian Elischer  * Initialize global thread allocation resources.
21544990b8cSJulian Elischer  */
21644990b8cSJulian Elischer void
21744990b8cSJulian Elischer threadinit(void)
21844990b8cSJulian Elischer {
21944990b8cSJulian Elischer 
2201ea7a6f8SPoul-Henning Kamp 	mtx_init(&tid_lock, "TID lock", NULL, MTX_DEF);
2216829a5c5SJulian Elischer 	/* leave one number for thread0 */
2226829a5c5SJulian Elischer 	tid_unrhdr = new_unrhdr(PID_MAX + 2, INT_MAX, &tid_lock);
2231ea7a6f8SPoul-Henning Kamp 
224de028f5aSJeff Roberson 	thread_zone = uma_zcreate("THREAD", sched_sizeof_thread(),
22544990b8cSJulian Elischer 	    thread_ctor, thread_dtor, thread_init, thread_fini,
2264649e92bSJohn Baldwin 	    16 - 1, 0);
22744990b8cSJulian Elischer }
22844990b8cSJulian Elischer 
22944990b8cSJulian Elischer /*
230ff8fbcffSJeff Roberson  * Place an unused thread on the zombie list.
231ad1e7d28SJulian Elischer  * Use the slpq as that must be unused by now.
23244990b8cSJulian Elischer  */
23344990b8cSJulian Elischer void
234ff8fbcffSJeff Roberson thread_zombie(struct thread *td)
23544990b8cSJulian Elischer {
236a54e85fdSJeff Roberson 	mtx_lock_spin(&zombie_lock);
237ad1e7d28SJulian Elischer 	TAILQ_INSERT_HEAD(&zombie_threads, td, td_slpq);
238a54e85fdSJeff Roberson 	mtx_unlock_spin(&zombie_lock);
23944990b8cSJulian Elischer }
24044990b8cSJulian Elischer 
2415c8329edSJulian Elischer /*
242ff8fbcffSJeff Roberson  * Release a thread that has exited after cpu_throw().
243ff8fbcffSJeff Roberson  */
244ff8fbcffSJeff Roberson void
245ff8fbcffSJeff Roberson thread_stash(struct thread *td)
246ff8fbcffSJeff Roberson {
247ff8fbcffSJeff Roberson 	atomic_subtract_rel_int(&td->td_proc->p_exitthreads, 1);
248ff8fbcffSJeff Roberson 	thread_zombie(td);
249ff8fbcffSJeff Roberson }
250ff8fbcffSJeff Roberson 
251ff8fbcffSJeff Roberson /*
2526617724cSJeff Roberson  * Reap zombie resources.
25344990b8cSJulian Elischer  */
25444990b8cSJulian Elischer void
25544990b8cSJulian Elischer thread_reap(void)
25644990b8cSJulian Elischer {
2575c8329edSJulian Elischer 	struct thread *td_first, *td_next;
25844990b8cSJulian Elischer 
25944990b8cSJulian Elischer 	/*
2605215b187SJeff Roberson 	 * Don't even bother to lock if none at this instant,
2615215b187SJeff Roberson 	 * we really don't care about the next instant..
26244990b8cSJulian Elischer 	 */
2638460a577SJohn Birrell 	if (!TAILQ_EMPTY(&zombie_threads)) {
264a54e85fdSJeff Roberson 		mtx_lock_spin(&zombie_lock);
2655c8329edSJulian Elischer 		td_first = TAILQ_FIRST(&zombie_threads);
2665c8329edSJulian Elischer 		if (td_first)
2675c8329edSJulian Elischer 			TAILQ_INIT(&zombie_threads);
268a54e85fdSJeff Roberson 		mtx_unlock_spin(&zombie_lock);
2695c8329edSJulian Elischer 		while (td_first) {
270ad1e7d28SJulian Elischer 			td_next = TAILQ_NEXT(td_first, td_slpq);
2715215b187SJeff Roberson 			if (td_first->td_ucred)
2725215b187SJeff Roberson 				crfree(td_first->td_ucred);
2735c8329edSJulian Elischer 			thread_free(td_first);
2745c8329edSJulian Elischer 			td_first = td_next;
27544990b8cSJulian Elischer 		}
27644990b8cSJulian Elischer 	}
277ed062c8dSJulian Elischer }
27844990b8cSJulian Elischer 
2794f0db5e0SJulian Elischer /*
28044990b8cSJulian Elischer  * Allocate a thread.
28144990b8cSJulian Elischer  */
28244990b8cSJulian Elischer struct thread *
28344990b8cSJulian Elischer thread_alloc(void)
28444990b8cSJulian Elischer {
28589b57fcfSKonstantin Belousov 	struct thread *td;
2868460a577SJohn Birrell 
28744990b8cSJulian Elischer 	thread_reap(); /* check if any zombies to get */
28889b57fcfSKonstantin Belousov 
28989b57fcfSKonstantin Belousov 	td = (struct thread *)uma_zalloc(thread_zone, M_WAITOK);
29089b57fcfSKonstantin Belousov 	KASSERT(td->td_kstack == 0, ("thread_alloc got thread with kstack"));
29189b57fcfSKonstantin Belousov 	if (!vm_thread_new(td, 0)) {
29289b57fcfSKonstantin Belousov 		uma_zfree(thread_zone, td);
29389b57fcfSKonstantin Belousov 		return (NULL);
29489b57fcfSKonstantin Belousov 	}
2950c3967e7SMarcel Moolenaar 	cpu_thread_alloc(td);
29689b57fcfSKonstantin Belousov 	return (td);
29744990b8cSJulian Elischer }
29844990b8cSJulian Elischer 
2994f0db5e0SJulian Elischer 
3004f0db5e0SJulian Elischer /*
30144990b8cSJulian Elischer  * Deallocate a thread.
30244990b8cSJulian Elischer  */
30344990b8cSJulian Elischer void
30444990b8cSJulian Elischer thread_free(struct thread *td)
30544990b8cSJulian Elischer {
306d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
307d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
3080c3967e7SMarcel Moolenaar 	cpu_thread_free(td);
30989b57fcfSKonstantin Belousov 	if (td->td_altkstack != 0)
31089b57fcfSKonstantin Belousov 		vm_thread_dispose_altkstack(td);
31189b57fcfSKonstantin Belousov 	if (td->td_kstack != 0)
31289b57fcfSKonstantin Belousov 		vm_thread_dispose(td);
31344990b8cSJulian Elischer 	uma_zfree(thread_zone, td);
31444990b8cSJulian Elischer }
31544990b8cSJulian Elischer 
31644990b8cSJulian Elischer /*
31744990b8cSJulian Elischer  * Discard the current thread and exit from its context.
31894e0a4cdSJulian Elischer  * Always called with scheduler locked.
31944990b8cSJulian Elischer  *
32044990b8cSJulian Elischer  * Because we can't free a thread while we're operating under its context,
321696058c3SJulian Elischer  * push the current thread into our CPU's deadthread holder. This means
322696058c3SJulian Elischer  * we needn't worry about someone else grabbing our context before we
3236617724cSJeff Roberson  * do a cpu_throw().
32444990b8cSJulian Elischer  */
32544990b8cSJulian Elischer void
32644990b8cSJulian Elischer thread_exit(void)
32744990b8cSJulian Elischer {
328e170bfdaSDavid Xu 	uint64_t new_switchtime;
32944990b8cSJulian Elischer 	struct thread *td;
3301c4bcd05SJeff Roberson 	struct thread *td2;
33144990b8cSJulian Elischer 	struct proc *p;
33244990b8cSJulian Elischer 
33344990b8cSJulian Elischer 	td = curthread;
33444990b8cSJulian Elischer 	p = td->td_proc;
33544990b8cSJulian Elischer 
336a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
337ed062c8dSJulian Elischer 	mtx_assert(&Giant, MA_NOTOWNED);
338a54e85fdSJeff Roberson 
33944990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
340ed062c8dSJulian Elischer 	KASSERT(p != NULL, ("thread exiting without a process"));
341cc701b73SRobert Watson 	CTR3(KTR_PROC, "thread_exit: thread %p (pid %ld, %s)", td,
342e01eafefSJulian Elischer 	    (long)p->p_pid, td->td_name);
3439104847fSDavid Xu 	KASSERT(TAILQ_EMPTY(&td->td_sigqueue.sq_list), ("signal pending"));
34444990b8cSJulian Elischer 
34589964dd2SRobert Watson #ifdef AUDIT
34689964dd2SRobert Watson 	AUDIT_SYSCALL_EXIT(0, td);
34789964dd2SRobert Watson #endif
348d10183d9SDavid Xu 	umtx_thread_exit(td);
349ed062c8dSJulian Elischer 	/*
350ed062c8dSJulian Elischer 	 * drop FPU & debug register state storage, or any other
351ed062c8dSJulian Elischer 	 * architecture specific resources that
352ed062c8dSJulian Elischer 	 * would not be on a new untouched process.
353ed062c8dSJulian Elischer 	 */
35444990b8cSJulian Elischer 	cpu_thread_exit(td);	/* XXXSMP */
35544990b8cSJulian Elischer 
356e170bfdaSDavid Xu 	/* Do the same timestamp bookkeeping that mi_switch() would do. */
357e170bfdaSDavid Xu 	new_switchtime = cpu_ticks();
358e170bfdaSDavid Xu 	p->p_rux.rux_runtime += (new_switchtime - PCPU_GET(switchtime));
359e170bfdaSDavid Xu 	PCPU_SET(switchtime, new_switchtime);
360e170bfdaSDavid Xu 	PCPU_SET(switchticks, ticks);
361b4b70819SAttilio Rao 	PCPU_INC(cnt.v_swtch);
362a140976eSAttilio Rao 	/* Save our resource usage in our process. */
363a140976eSAttilio Rao 	td->td_ru.ru_nvcsw++;
364a140976eSAttilio Rao 	rucollect(&p->p_ru, &td->td_ru);
365ed062c8dSJulian Elischer 	/*
3661faf202eSJulian Elischer 	 * The last thread is left attached to the process
3671faf202eSJulian Elischer 	 * So that the whole bundle gets recycled. Skip
368ed062c8dSJulian Elischer 	 * all this stuff if we never had threads.
369ed062c8dSJulian Elischer 	 * EXIT clears all sign of other threads when
370ed062c8dSJulian Elischer 	 * it goes to single threading, so the last thread always
371ed062c8dSJulian Elischer 	 * takes the short path.
3721faf202eSJulian Elischer 	 */
373ed062c8dSJulian Elischer 	if (p->p_flag & P_HADTHREADS) {
3741faf202eSJulian Elischer 		if (p->p_numthreads > 1) {
375d3a0bd78SJulian Elischer 			thread_unlink(td);
3761c4bcd05SJeff Roberson 			td2 = FIRST_THREAD_IN_PROC(p);
3771c4bcd05SJeff Roberson 			sched_exit_thread(td2, td);
378ed062c8dSJulian Elischer 
379ed062c8dSJulian Elischer 			/*
38044990b8cSJulian Elischer 			 * The test below is NOT true if we are the
3811faf202eSJulian Elischer 			 * sole exiting thread. P_STOPPED_SNGL is unset
38244990b8cSJulian Elischer 			 * in exit1() after it is the only survivor.
38344990b8cSJulian Elischer 			 */
3841279572aSDavid Xu 			if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
38544990b8cSJulian Elischer 				if (p->p_numthreads == p->p_suspcount) {
386a54e85fdSJeff Roberson 					thread_lock(p->p_singlethread);
38771fad9fdSJulian Elischer 					thread_unsuspend_one(p->p_singlethread);
388a54e85fdSJeff Roberson 					thread_unlock(p->p_singlethread);
38944990b8cSJulian Elischer 				}
39044990b8cSJulian Elischer 			}
39148bfcdddSJulian Elischer 
392ff8fbcffSJeff Roberson 			atomic_add_int(&td->td_proc->p_exitthreads, 1);
393696058c3SJulian Elischer 			PCPU_SET(deadthread, td);
3941faf202eSJulian Elischer 		} else {
395ed062c8dSJulian Elischer 			/*
396ed062c8dSJulian Elischer 			 * The last thread is exiting.. but not through exit()
397ed062c8dSJulian Elischer 			 */
398ed062c8dSJulian Elischer 			panic ("thread_exit: Last thread exiting on its own");
399ed062c8dSJulian Elischer 		}
4001faf202eSJulian Elischer 	}
401a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
402a54e85fdSJeff Roberson 	thread_lock(td);
403a140976eSAttilio Rao 	/* Save our tick information with both the thread and proc locked */
404a54e85fdSJeff Roberson 	ruxagg(&p->p_rux, td);
405a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
406dcc9954eSJulian Elischer 	td->td_state = TDS_INACTIVE;
407732d9528SJulian Elischer 	CTR1(KTR_PROC, "thread_exit: cpu_throw() thread %p", td);
408a54e85fdSJeff Roberson 	sched_throw(td);
409cc66ebe2SPeter Wemm 	panic("I'm a teapot!");
41044990b8cSJulian Elischer 	/* NOTREACHED */
41144990b8cSJulian Elischer }
41244990b8cSJulian Elischer 
41344990b8cSJulian Elischer /*
414696058c3SJulian Elischer  * Do any thread specific cleanups that may be needed in wait()
41537814395SPeter Wemm  * called with Giant, proc and schedlock not held.
416696058c3SJulian Elischer  */
417696058c3SJulian Elischer void
418696058c3SJulian Elischer thread_wait(struct proc *p)
419696058c3SJulian Elischer {
420696058c3SJulian Elischer 	struct thread *td;
421696058c3SJulian Elischer 
42237814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
42385495c72SJens Schweikhardt 	KASSERT((p->p_numthreads == 1), ("Multiple threads in wait1()"));
424ff8fbcffSJeff Roberson 	td = FIRST_THREAD_IN_PROC(p);
425ff8fbcffSJeff Roberson 	/* Lock the last thread so we spin until it exits cpu_throw(). */
426ff8fbcffSJeff Roberson 	thread_lock(td);
427ff8fbcffSJeff Roberson 	thread_unlock(td);
428ff8fbcffSJeff Roberson 	/* Wait for any remaining threads to exit cpu_throw(). */
429ff8fbcffSJeff Roberson 	while (p->p_exitthreads)
430ff8fbcffSJeff Roberson 		sched_relinquish(curthread);
431d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
432d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
433696058c3SJulian Elischer 	cpu_thread_clean(td);
434ed062c8dSJulian Elischer 	crfree(td->td_ucred);
435696058c3SJulian Elischer 	thread_reap();	/* check for zombie threads etc. */
436696058c3SJulian Elischer }
437696058c3SJulian Elischer 
438696058c3SJulian Elischer /*
43944990b8cSJulian Elischer  * Link a thread to a process.
4401faf202eSJulian Elischer  * set up anything that needs to be initialized for it to
4411faf202eSJulian Elischer  * be used by the process.
44244990b8cSJulian Elischer  */
44344990b8cSJulian Elischer void
4448460a577SJohn Birrell thread_link(struct thread *td, struct proc *p)
44544990b8cSJulian Elischer {
44644990b8cSJulian Elischer 
447a54e85fdSJeff Roberson 	/*
448a54e85fdSJeff Roberson 	 * XXX This can't be enabled because it's called for proc0 before
449374ae2a3SJeff Roberson 	 * its lock has been created.
450374ae2a3SJeff Roberson 	 * PROC_LOCK_ASSERT(p, MA_OWNED);
451a54e85fdSJeff Roberson 	 */
45271fad9fdSJulian Elischer 	td->td_state    = TDS_INACTIVE;
45344990b8cSJulian Elischer 	td->td_proc     = p;
454b61ce5b0SJeff Roberson 	td->td_flags    = TDF_INMEM;
45544990b8cSJulian Elischer 
4561faf202eSJulian Elischer 	LIST_INIT(&td->td_contested);
457eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[0]);
458eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[1]);
4599104847fSDavid Xu 	sigqueue_init(&td->td_sigqueue, p);
460c06eb4e2SSam Leffler 	callout_init(&td->td_slpcallout, CALLOUT_MPSAFE);
46144990b8cSJulian Elischer 	TAILQ_INSERT_HEAD(&p->p_threads, td, td_plist);
46244990b8cSJulian Elischer 	p->p_numthreads++;
46344990b8cSJulian Elischer }
46444990b8cSJulian Elischer 
465ed062c8dSJulian Elischer /*
466e5bedcefSJulian Elischer  * Convert a process with one thread to an unthreaded process.
467e5bedcefSJulian Elischer  */
468e5bedcefSJulian Elischer void
469e5bedcefSJulian Elischer thread_unthread(struct thread *td)
470e5bedcefSJulian Elischer {
471e5bedcefSJulian Elischer 	struct proc *p = td->td_proc;
472e5bedcefSJulian Elischer 
473e5bedcefSJulian Elischer 	KASSERT((p->p_numthreads == 1), ("Unthreading with >1 threads"));
4748460a577SJohn Birrell 	p->p_flag &= ~P_HADTHREADS;
475e5bedcefSJulian Elischer }
476e5bedcefSJulian Elischer 
477e5bedcefSJulian Elischer /*
478ed062c8dSJulian Elischer  * Called from:
479ed062c8dSJulian Elischer  *  thread_exit()
480ed062c8dSJulian Elischer  */
481d3a0bd78SJulian Elischer void
482d3a0bd78SJulian Elischer thread_unlink(struct thread *td)
483d3a0bd78SJulian Elischer {
484d3a0bd78SJulian Elischer 	struct proc *p = td->td_proc;
485d3a0bd78SJulian Elischer 
486374ae2a3SJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
487d3a0bd78SJulian Elischer 	TAILQ_REMOVE(&p->p_threads, td, td_plist);
488d3a0bd78SJulian Elischer 	p->p_numthreads--;
489d3a0bd78SJulian Elischer 	/* could clear a few other things here */
4908460a577SJohn Birrell 	/* Must  NOT clear links to proc! */
4915c8329edSJulian Elischer }
4925c8329edSJulian Elischer 
4935215b187SJeff Roberson /*
49444990b8cSJulian Elischer  * Enforce single-threading.
49544990b8cSJulian Elischer  *
49644990b8cSJulian Elischer  * Returns 1 if the caller must abort (another thread is waiting to
49744990b8cSJulian Elischer  * exit the process or similar). Process is locked!
49844990b8cSJulian Elischer  * Returns 0 when you are successfully the only thread running.
49944990b8cSJulian Elischer  * A process has successfully single threaded in the suspend mode when
50044990b8cSJulian Elischer  * There are no threads in user mode. Threads in the kernel must be
50144990b8cSJulian Elischer  * allowed to continue until they get to the user boundary. They may even
50244990b8cSJulian Elischer  * copy out their return values and data before suspending. They may however be
503e2668f55SMaxim Konovalov  * accelerated in reaching the user boundary as we will wake up
50444990b8cSJulian Elischer  * any sleeping threads that are interruptable. (PCATCH).
50544990b8cSJulian Elischer  */
50644990b8cSJulian Elischer int
507906ac69dSDavid Xu thread_single(int mode)
50844990b8cSJulian Elischer {
50944990b8cSJulian Elischer 	struct thread *td;
51044990b8cSJulian Elischer 	struct thread *td2;
51144990b8cSJulian Elischer 	struct proc *p;
512ec008e96SDavid Xu 	int remaining;
51344990b8cSJulian Elischer 
51444990b8cSJulian Elischer 	td = curthread;
51544990b8cSJulian Elischer 	p = td->td_proc;
51637814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
51744990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
51844990b8cSJulian Elischer 	KASSERT((td != NULL), ("curthread is NULL"));
51944990b8cSJulian Elischer 
520ed062c8dSJulian Elischer 	if ((p->p_flag & P_HADTHREADS) == 0)
52144990b8cSJulian Elischer 		return (0);
52244990b8cSJulian Elischer 
523e3b9bf71SJulian Elischer 	/* Is someone already single threading? */
524906ac69dSDavid Xu 	if (p->p_singlethread != NULL && p->p_singlethread != td)
52544990b8cSJulian Elischer 		return (1);
52644990b8cSJulian Elischer 
527906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
528906ac69dSDavid Xu 		p->p_flag |= P_SINGLE_EXIT;
529906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_BOUNDARY;
530906ac69dSDavid Xu 	} else {
531906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_EXIT;
532906ac69dSDavid Xu 		if (mode == SINGLE_BOUNDARY)
533906ac69dSDavid Xu 			p->p_flag |= P_SINGLE_BOUNDARY;
534906ac69dSDavid Xu 		else
535906ac69dSDavid Xu 			p->p_flag &= ~P_SINGLE_BOUNDARY;
536906ac69dSDavid Xu 	}
5371279572aSDavid Xu 	p->p_flag |= P_STOPPED_SINGLE;
538a54e85fdSJeff Roberson 	PROC_SLOCK(p);
539112afcb2SJohn Baldwin 	p->p_singlethread = td;
540906ac69dSDavid Xu 	if (mode == SINGLE_EXIT)
541ec008e96SDavid Xu 		remaining = p->p_numthreads;
542906ac69dSDavid Xu 	else if (mode == SINGLE_BOUNDARY)
543906ac69dSDavid Xu 		remaining = p->p_numthreads - p->p_boundary_count;
544906ac69dSDavid Xu 	else
545ec008e96SDavid Xu 		remaining = p->p_numthreads - p->p_suspcount;
546ec008e96SDavid Xu 	while (remaining != 1) {
547bf1a3220SDavid Xu 		if (P_SHOULDSTOP(p) != P_STOPPED_SINGLE)
548bf1a3220SDavid Xu 			goto stopme;
54944990b8cSJulian Elischer 		FOREACH_THREAD_IN_PROC(p, td2) {
55044990b8cSJulian Elischer 			if (td2 == td)
55144990b8cSJulian Elischer 				continue;
552a54e85fdSJeff Roberson 			thread_lock(td2);
553588257e8SDavid Xu 			td2->td_flags |= TDF_ASTPENDING;
55471fad9fdSJulian Elischer 			if (TD_IS_INHIBITED(td2)) {
555906ac69dSDavid Xu 				switch (mode) {
556906ac69dSDavid Xu 				case SINGLE_EXIT:
557cbf4e354SDavid Xu 					if (td->td_flags & TDF_DBSUSPEND)
558cbf4e354SDavid Xu 						td->td_flags &= ~TDF_DBSUSPEND;
559906ac69dSDavid Xu 					if (TD_IS_SUSPENDED(td2))
56071fad9fdSJulian Elischer 						thread_unsuspend_one(td2);
56133862f40SDavid Xu 					if (TD_ON_SLEEPQ(td2) &&
562906ac69dSDavid Xu 					    (td2->td_flags & TDF_SINTR))
56394f0972bSDavid Xu 						sleepq_abort(td2, EINTR);
564906ac69dSDavid Xu 					break;
565906ac69dSDavid Xu 				case SINGLE_BOUNDARY:
566906ac69dSDavid Xu 					break;
567906ac69dSDavid Xu 				default:
568a54e85fdSJeff Roberson 					if (TD_IS_SUSPENDED(td2)) {
569a54e85fdSJeff Roberson 						thread_unlock(td2);
5709d102777SJulian Elischer 						continue;
571a54e85fdSJeff Roberson 					}
5725215b187SJeff Roberson 					/*
5732da78e38SRobert Watson 					 * maybe other inhibited states too?
5745215b187SJeff Roberson 					 */
5758acf6057SDavid Xu 					if ((td2->td_flags & TDF_SINTR) &&
5768acf6057SDavid Xu 					    (td2->td_inhibitors &
5778acf6057SDavid Xu 					    (TDI_SLEEPING | TDI_SWAPPED)))
5789d102777SJulian Elischer 						thread_suspend_one(td2);
579906ac69dSDavid Xu 					break;
58044990b8cSJulian Elischer 				}
58144990b8cSJulian Elischer 			}
582d8267df7SDavid Xu #ifdef SMP
583d8267df7SDavid Xu 			else if (TD_IS_RUNNING(td2) && td != td2) {
584d8267df7SDavid Xu 				forward_signal(td2);
585d8267df7SDavid Xu 			}
586d8267df7SDavid Xu #endif
587a54e85fdSJeff Roberson 			thread_unlock(td2);
5889d102777SJulian Elischer 		}
589906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
590ec008e96SDavid Xu 			remaining = p->p_numthreads;
591906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
592906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
593ec008e96SDavid Xu 		else
594ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
595ec008e96SDavid Xu 
5969d102777SJulian Elischer 		/*
5979d102777SJulian Elischer 		 * Maybe we suspended some threads.. was it enough?
5989d102777SJulian Elischer 		 */
599ec008e96SDavid Xu 		if (remaining == 1)
6009d102777SJulian Elischer 			break;
6019d102777SJulian Elischer 
602bf1a3220SDavid Xu stopme:
60344990b8cSJulian Elischer 		/*
60444990b8cSJulian Elischer 		 * Wake us up when everyone else has suspended.
605e3b9bf71SJulian Elischer 		 * In the mean time we suspend as well.
60644990b8cSJulian Elischer 		 */
607a54e85fdSJeff Roberson 		thread_suspend_switch(td);
608906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
609ec008e96SDavid Xu 			remaining = p->p_numthreads;
610906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
611906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
612ec008e96SDavid Xu 		else
613ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
61444990b8cSJulian Elischer 	}
615906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
61691599697SJulian Elischer 		/*
61791599697SJulian Elischer 		 * We have gotten rid of all the other threads and we
61891599697SJulian Elischer 		 * are about to either exit or exec. In either case,
61991599697SJulian Elischer 		 * we try our utmost  to revert to being a non-threaded
62091599697SJulian Elischer 		 * process.
62191599697SJulian Elischer 		 */
622ed062c8dSJulian Elischer 		p->p_singlethread = NULL;
62364895117SDavid Xu 		p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT);
624e5bedcefSJulian Elischer 		thread_unthread(td);
62591599697SJulian Elischer 	}
626a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
62744990b8cSJulian Elischer 	return (0);
62844990b8cSJulian Elischer }
62944990b8cSJulian Elischer 
63044990b8cSJulian Elischer /*
63144990b8cSJulian Elischer  * Called in from locations that can safely check to see
63244990b8cSJulian Elischer  * whether we have to suspend or at least throttle for a
63344990b8cSJulian Elischer  * single-thread event (e.g. fork).
63444990b8cSJulian Elischer  *
63544990b8cSJulian Elischer  * Such locations include userret().
63644990b8cSJulian Elischer  * If the "return_instead" argument is non zero, the thread must be able to
63744990b8cSJulian Elischer  * accept 0 (caller may continue), or 1 (caller must abort) as a result.
63844990b8cSJulian Elischer  *
63944990b8cSJulian Elischer  * The 'return_instead' argument tells the function if it may do a
64044990b8cSJulian Elischer  * thread_exit() or suspend, or whether the caller must abort and back
64144990b8cSJulian Elischer  * out instead.
64244990b8cSJulian Elischer  *
64344990b8cSJulian Elischer  * If the thread that set the single_threading request has set the
64444990b8cSJulian Elischer  * P_SINGLE_EXIT bit in the process flags then this call will never return
64544990b8cSJulian Elischer  * if 'return_instead' is false, but will exit.
64644990b8cSJulian Elischer  *
64744990b8cSJulian Elischer  * P_SINGLE_EXIT | return_instead == 0| return_instead != 0
64844990b8cSJulian Elischer  *---------------+--------------------+---------------------
64944990b8cSJulian Elischer  *       0       | returns 0          |   returns 0 or 1
65044990b8cSJulian Elischer  *               | when ST ends       |   immediatly
65144990b8cSJulian Elischer  *---------------+--------------------+---------------------
65244990b8cSJulian Elischer  *       1       | thread exits       |   returns 1
65344990b8cSJulian Elischer  *               |                    |  immediatly
65444990b8cSJulian Elischer  * 0 = thread_exit() or suspension ok,
65544990b8cSJulian Elischer  * other = return error instead of stopping the thread.
65644990b8cSJulian Elischer  *
65744990b8cSJulian Elischer  * While a full suspension is under effect, even a single threading
65844990b8cSJulian Elischer  * thread would be suspended if it made this call (but it shouldn't).
65944990b8cSJulian Elischer  * This call should only be made from places where
66044990b8cSJulian Elischer  * thread_exit() would be safe as that may be the outcome unless
66144990b8cSJulian Elischer  * return_instead is set.
66244990b8cSJulian Elischer  */
66344990b8cSJulian Elischer int
66444990b8cSJulian Elischer thread_suspend_check(int return_instead)
66544990b8cSJulian Elischer {
666ecafb24bSJuli Mallett 	struct thread *td;
667ecafb24bSJuli Mallett 	struct proc *p;
66844990b8cSJulian Elischer 
66944990b8cSJulian Elischer 	td = curthread;
67044990b8cSJulian Elischer 	p = td->td_proc;
67137814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
67244990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
673cbf4e354SDavid Xu 	while (P_SHOULDSTOP(p) ||
674cbf4e354SDavid Xu 	      ((p->p_flag & P_TRACED) && (td->td_flags & TDF_DBSUSPEND))) {
6751279572aSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
67644990b8cSJulian Elischer 			KASSERT(p->p_singlethread != NULL,
67744990b8cSJulian Elischer 			    ("singlethread not set"));
67844990b8cSJulian Elischer 			/*
679e3b9bf71SJulian Elischer 			 * The only suspension in action is a
680e3b9bf71SJulian Elischer 			 * single-threading. Single threader need not stop.
681b6d5995eSJulian Elischer 			 * XXX Should be safe to access unlocked
682b6d5995eSJulian Elischer 			 * as it can only be set to be true by us.
68344990b8cSJulian Elischer 			 */
684e3b9bf71SJulian Elischer 			if (p->p_singlethread == td)
68544990b8cSJulian Elischer 				return (0);	/* Exempt from stopping. */
68644990b8cSJulian Elischer 		}
68745a4bfa1SDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && return_instead)
68894f0972bSDavid Xu 			return (EINTR);
68944990b8cSJulian Elischer 
690906ac69dSDavid Xu 		/* Should we goto user boundary if we didn't come from there? */
691906ac69dSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
692906ac69dSDavid Xu 		    (p->p_flag & P_SINGLE_BOUNDARY) && return_instead)
69394f0972bSDavid Xu 			return (ERESTART);
694906ac69dSDavid Xu 
6959104847fSDavid Xu 		/* If thread will exit, flush its pending signals */
6969104847fSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
6979104847fSDavid Xu 			sigqueue_flush(&td->td_sigqueue);
6989104847fSDavid Xu 
699a54e85fdSJeff Roberson 		PROC_SLOCK(p);
700e574e444SDavid Xu 		thread_stopped(p);
70144990b8cSJulian Elischer 		/*
70244990b8cSJulian Elischer 		 * If the process is waiting for us to exit,
70344990b8cSJulian Elischer 		 * this thread should just suicide.
7041279572aSDavid Xu 		 * Assumes that P_SINGLE_EXIT implies P_STOPPED_SINGLE.
70544990b8cSJulian Elischer 		 */
706906ac69dSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
70744990b8cSJulian Elischer 			thread_exit();
708a54e85fdSJeff Roberson 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
709a54e85fdSJeff Roberson 			if (p->p_numthreads == p->p_suspcount + 1) {
710a54e85fdSJeff Roberson 				thread_lock(p->p_singlethread);
711a54e85fdSJeff Roberson 				thread_unsuspend_one(p->p_singlethread);
712a54e85fdSJeff Roberson 				thread_unlock(p->p_singlethread);
713a54e85fdSJeff Roberson 			}
714a54e85fdSJeff Roberson 		}
715a54e85fdSJeff Roberson 		PROC_UNLOCK(p);
716a54e85fdSJeff Roberson 		thread_lock(td);
71744990b8cSJulian Elischer 		/*
71844990b8cSJulian Elischer 		 * When a thread suspends, it just
719ad1e7d28SJulian Elischer 		 * gets taken off all queues.
72044990b8cSJulian Elischer 		 */
72171fad9fdSJulian Elischer 		thread_suspend_one(td);
722906ac69dSDavid Xu 		if (return_instead == 0) {
723906ac69dSDavid Xu 			p->p_boundary_count++;
724906ac69dSDavid Xu 			td->td_flags |= TDF_BOUNDARY;
725cf19bf91SJulian Elischer 		}
726a54e85fdSJeff Roberson 		PROC_SUNLOCK(p);
727bf0acc27SJohn Baldwin 		mi_switch(SW_INVOL, NULL);
728a54e85fdSJeff Roberson 		if (return_instead == 0)
729906ac69dSDavid Xu 			td->td_flags &= ~TDF_BOUNDARY;
730a54e85fdSJeff Roberson 		thread_unlock(td);
73144990b8cSJulian Elischer 		PROC_LOCK(p);
732a54e85fdSJeff Roberson 		if (return_instead == 0)
733a54e85fdSJeff Roberson 			p->p_boundary_count--;
73444990b8cSJulian Elischer 	}
73544990b8cSJulian Elischer 	return (0);
73644990b8cSJulian Elischer }
73744990b8cSJulian Elischer 
73835c32a76SDavid Xu void
739a54e85fdSJeff Roberson thread_suspend_switch(struct thread *td)
740a54e85fdSJeff Roberson {
741a54e85fdSJeff Roberson 	struct proc *p;
742a54e85fdSJeff Roberson 
743a54e85fdSJeff Roberson 	p = td->td_proc;
744a54e85fdSJeff Roberson 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
745a54e85fdSJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
746a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
747a54e85fdSJeff Roberson 	/*
748a54e85fdSJeff Roberson 	 * We implement thread_suspend_one in stages here to avoid
749a54e85fdSJeff Roberson 	 * dropping the proc lock while the thread lock is owned.
750a54e85fdSJeff Roberson 	 */
751a54e85fdSJeff Roberson 	thread_stopped(p);
752a54e85fdSJeff Roberson 	p->p_suspcount++;
753a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
754a54e85fdSJeff Roberson 	thread_lock(td);
755a54e85fdSJeff Roberson 	TD_SET_SUSPENDED(td);
756c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
757a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
758a54e85fdSJeff Roberson 	DROP_GIANT();
759a54e85fdSJeff Roberson 	mi_switch(SW_VOL, NULL);
760a54e85fdSJeff Roberson 	thread_unlock(td);
761a54e85fdSJeff Roberson 	PICKUP_GIANT();
762a54e85fdSJeff Roberson 	PROC_LOCK(p);
763a54e85fdSJeff Roberson 	PROC_SLOCK(p);
764a54e85fdSJeff Roberson }
765a54e85fdSJeff Roberson 
766a54e85fdSJeff Roberson void
76735c32a76SDavid Xu thread_suspend_one(struct thread *td)
76835c32a76SDavid Xu {
76935c32a76SDavid Xu 	struct proc *p = td->td_proc;
77035c32a76SDavid Xu 
771a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
772a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
773e574e444SDavid Xu 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
77435c32a76SDavid Xu 	p->p_suspcount++;
77571fad9fdSJulian Elischer 	TD_SET_SUSPENDED(td);
776c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
77735c32a76SDavid Xu }
77835c32a76SDavid Xu 
77935c32a76SDavid Xu void
78035c32a76SDavid Xu thread_unsuspend_one(struct thread *td)
78135c32a76SDavid Xu {
78235c32a76SDavid Xu 	struct proc *p = td->td_proc;
78335c32a76SDavid Xu 
784a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
785a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
786ad1e7d28SJulian Elischer 	KASSERT(TD_IS_SUSPENDED(td), ("Thread not suspended"));
78771fad9fdSJulian Elischer 	TD_CLR_SUSPENDED(td);
78835c32a76SDavid Xu 	p->p_suspcount--;
78971fad9fdSJulian Elischer 	setrunnable(td);
79035c32a76SDavid Xu }
79135c32a76SDavid Xu 
79244990b8cSJulian Elischer /*
79344990b8cSJulian Elischer  * Allow all threads blocked by single threading to continue running.
79444990b8cSJulian Elischer  */
79544990b8cSJulian Elischer void
79644990b8cSJulian Elischer thread_unsuspend(struct proc *p)
79744990b8cSJulian Elischer {
79844990b8cSJulian Elischer 	struct thread *td;
79944990b8cSJulian Elischer 
80044990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
801a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
80244990b8cSJulian Elischer 	if (!P_SHOULDSTOP(p)) {
803ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
804a54e85fdSJeff Roberson 			thread_lock(td);
805ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
80635c32a76SDavid Xu 				thread_unsuspend_one(td);
80744990b8cSJulian Elischer 			}
808a54e85fdSJeff Roberson 			thread_unlock(td);
809ad1e7d28SJulian Elischer 		}
8101279572aSDavid Xu 	} else if ((P_SHOULDSTOP(p) == P_STOPPED_SINGLE) &&
81144990b8cSJulian Elischer 	    (p->p_numthreads == p->p_suspcount)) {
81244990b8cSJulian Elischer 		/*
81344990b8cSJulian Elischer 		 * Stopping everything also did the job for the single
81444990b8cSJulian Elischer 		 * threading request. Now we've downgraded to single-threaded,
81544990b8cSJulian Elischer 		 * let it continue.
81644990b8cSJulian Elischer 		 */
817a54e85fdSJeff Roberson 		thread_lock(p->p_singlethread);
81835c32a76SDavid Xu 		thread_unsuspend_one(p->p_singlethread);
819a54e85fdSJeff Roberson 		thread_unlock(p->p_singlethread);
82044990b8cSJulian Elischer 	}
82144990b8cSJulian Elischer }
82244990b8cSJulian Elischer 
823ed062c8dSJulian Elischer /*
824ed062c8dSJulian Elischer  * End the single threading mode..
825ed062c8dSJulian Elischer  */
82644990b8cSJulian Elischer void
82744990b8cSJulian Elischer thread_single_end(void)
82844990b8cSJulian Elischer {
82944990b8cSJulian Elischer 	struct thread *td;
83044990b8cSJulian Elischer 	struct proc *p;
83144990b8cSJulian Elischer 
83244990b8cSJulian Elischer 	td = curthread;
83344990b8cSJulian Elischer 	p = td->td_proc;
83444990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
835906ac69dSDavid Xu 	p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_SINGLE_BOUNDARY);
836a54e85fdSJeff Roberson 	PROC_SLOCK(p);
83744990b8cSJulian Elischer 	p->p_singlethread = NULL;
83849539972SJulian Elischer 	/*
83949539972SJulian Elischer 	 * If there are other threads they mey now run,
84049539972SJulian Elischer 	 * unless of course there is a blanket 'stop order'
84149539972SJulian Elischer 	 * on the process. The single threader must be allowed
84249539972SJulian Elischer 	 * to continue however as this is a bad place to stop.
84349539972SJulian Elischer 	 */
84449539972SJulian Elischer 	if ((p->p_numthreads != 1) && (!P_SHOULDSTOP(p))) {
845ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
846a54e85fdSJeff Roberson 			thread_lock(td);
847ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
84871fad9fdSJulian Elischer 				thread_unsuspend_one(td);
84944990b8cSJulian Elischer 			}
850a54e85fdSJeff Roberson 			thread_unlock(td);
85149539972SJulian Elischer 		}
852ad1e7d28SJulian Elischer 	}
853a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
85449539972SJulian Elischer }
8554fc21c09SDaniel Eischen 
85644355392SDavid Xu struct thread *
85744355392SDavid Xu thread_find(struct proc *p, lwpid_t tid)
85844355392SDavid Xu {
85944355392SDavid Xu 	struct thread *td;
86044355392SDavid Xu 
86144355392SDavid Xu 	PROC_LOCK_ASSERT(p, MA_OWNED);
86244355392SDavid Xu 	FOREACH_THREAD_IN_PROC(p, td) {
86344355392SDavid Xu 		if (td->td_tid == tid)
86444355392SDavid Xu 			break;
86544355392SDavid Xu 	}
86644355392SDavid Xu 	return (td);
86744355392SDavid Xu }
868