xref: /freebsd/sys/kern/kern_thread.c (revision 6617724c5fb1fc40eecc58f541ffbdcd005ba5c8)
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
34889964dd2SRobert Watson 
349d10183d9SDavid Xu 	umtx_thread_exit(td);
350d10183d9SDavid Xu 
351ed062c8dSJulian Elischer 	/*
352ed062c8dSJulian Elischer 	 * drop FPU & debug register state storage, or any other
353ed062c8dSJulian Elischer 	 * architecture specific resources that
354ed062c8dSJulian Elischer 	 * would not be on a new untouched process.
355ed062c8dSJulian Elischer 	 */
35644990b8cSJulian Elischer 	cpu_thread_exit(td);	/* XXXSMP */
35744990b8cSJulian Elischer 
358e170bfdaSDavid Xu 	/* Do the same timestamp bookkeeping that mi_switch() would do. */
359e170bfdaSDavid Xu 	new_switchtime = cpu_ticks();
360e170bfdaSDavid Xu 	p->p_rux.rux_runtime += (new_switchtime - PCPU_GET(switchtime));
361e170bfdaSDavid Xu 	PCPU_SET(switchtime, new_switchtime);
362e170bfdaSDavid Xu 	PCPU_SET(switchticks, ticks);
363b4b70819SAttilio Rao 	PCPU_INC(cnt.v_swtch);
364a140976eSAttilio Rao 	/* Save our resource usage in our process. */
365a140976eSAttilio Rao 	td->td_ru.ru_nvcsw++;
366a140976eSAttilio Rao 	rucollect(&p->p_ru, &td->td_ru);
367ed062c8dSJulian Elischer 	/*
3681faf202eSJulian Elischer 	 * The last thread is left attached to the process
3691faf202eSJulian Elischer 	 * So that the whole bundle gets recycled. Skip
370ed062c8dSJulian Elischer 	 * all this stuff if we never had threads.
371ed062c8dSJulian Elischer 	 * EXIT clears all sign of other threads when
372ed062c8dSJulian Elischer 	 * it goes to single threading, so the last thread always
373ed062c8dSJulian Elischer 	 * takes the short path.
3741faf202eSJulian Elischer 	 */
375ed062c8dSJulian Elischer 	if (p->p_flag & P_HADTHREADS) {
3761faf202eSJulian Elischer 		if (p->p_numthreads > 1) {
377a54e85fdSJeff Roberson 			thread_lock(td);
378d3a0bd78SJulian Elischer 			thread_unlink(td);
379a54e85fdSJeff Roberson 			thread_unlock(td);
3801c4bcd05SJeff Roberson 			td2 = FIRST_THREAD_IN_PROC(p);
3811c4bcd05SJeff Roberson 			sched_exit_thread(td2, td);
382ed062c8dSJulian Elischer 
383ed062c8dSJulian Elischer 			/*
38444990b8cSJulian Elischer 			 * The test below is NOT true if we are the
3851faf202eSJulian Elischer 			 * sole exiting thread. P_STOPPED_SNGL is unset
38644990b8cSJulian Elischer 			 * in exit1() after it is the only survivor.
38744990b8cSJulian Elischer 			 */
3881279572aSDavid Xu 			if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
38944990b8cSJulian Elischer 				if (p->p_numthreads == p->p_suspcount) {
390a54e85fdSJeff Roberson 					thread_lock(p->p_singlethread);
39171fad9fdSJulian Elischer 					thread_unsuspend_one(p->p_singlethread);
392a54e85fdSJeff Roberson 					thread_unlock(p->p_singlethread);
39344990b8cSJulian Elischer 				}
39444990b8cSJulian Elischer 			}
39548bfcdddSJulian Elischer 
396ff8fbcffSJeff Roberson 			atomic_add_int(&td->td_proc->p_exitthreads, 1);
397696058c3SJulian Elischer 			PCPU_SET(deadthread, td);
3981faf202eSJulian Elischer 		} else {
399ed062c8dSJulian Elischer 			/*
400ed062c8dSJulian Elischer 			 * The last thread is exiting.. but not through exit()
401ed062c8dSJulian Elischer 			 */
402ed062c8dSJulian Elischer 			panic ("thread_exit: Last thread exiting on its own");
403ed062c8dSJulian Elischer 		}
4041faf202eSJulian Elischer 	}
405a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
406a54e85fdSJeff Roberson 	thread_lock(td);
407a140976eSAttilio Rao 	/* Save our tick information with both the thread and proc locked */
408a54e85fdSJeff Roberson 	ruxagg(&p->p_rux, td);
409a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
410dcc9954eSJulian Elischer 	td->td_state = TDS_INACTIVE;
411732d9528SJulian Elischer 	CTR1(KTR_PROC, "thread_exit: cpu_throw() thread %p", td);
412a54e85fdSJeff Roberson 	sched_throw(td);
413cc66ebe2SPeter Wemm 	panic("I'm a teapot!");
41444990b8cSJulian Elischer 	/* NOTREACHED */
41544990b8cSJulian Elischer }
41644990b8cSJulian Elischer 
41744990b8cSJulian Elischer /*
418696058c3SJulian Elischer  * Do any thread specific cleanups that may be needed in wait()
41937814395SPeter Wemm  * called with Giant, proc and schedlock not held.
420696058c3SJulian Elischer  */
421696058c3SJulian Elischer void
422696058c3SJulian Elischer thread_wait(struct proc *p)
423696058c3SJulian Elischer {
424696058c3SJulian Elischer 	struct thread *td;
425696058c3SJulian Elischer 
42637814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
42785495c72SJens Schweikhardt 	KASSERT((p->p_numthreads == 1), ("Multiple threads in wait1()"));
428ff8fbcffSJeff Roberson 	td = FIRST_THREAD_IN_PROC(p);
429ff8fbcffSJeff Roberson 	/* Lock the last thread so we spin until it exits cpu_throw(). */
430ff8fbcffSJeff Roberson 	thread_lock(td);
431ff8fbcffSJeff Roberson 	thread_unlock(td);
432ff8fbcffSJeff Roberson 	/* Wait for any remaining threads to exit cpu_throw(). */
433ff8fbcffSJeff Roberson 	while (p->p_exitthreads)
434ff8fbcffSJeff Roberson 		sched_relinquish(curthread);
435d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
436d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
437696058c3SJulian Elischer 	cpu_thread_clean(td);
438ed062c8dSJulian Elischer 	crfree(td->td_ucred);
439696058c3SJulian Elischer 	thread_reap();	/* check for zombie threads etc. */
440696058c3SJulian Elischer }
441696058c3SJulian Elischer 
442696058c3SJulian Elischer /*
44344990b8cSJulian Elischer  * Link a thread to a process.
4441faf202eSJulian Elischer  * set up anything that needs to be initialized for it to
4451faf202eSJulian Elischer  * be used by the process.
44644990b8cSJulian Elischer  */
44744990b8cSJulian Elischer void
4488460a577SJohn Birrell thread_link(struct thread *td, struct proc *p)
44944990b8cSJulian Elischer {
45044990b8cSJulian Elischer 
451a54e85fdSJeff Roberson 	/*
452a54e85fdSJeff Roberson 	 * XXX This can't be enabled because it's called for proc0 before
453a54e85fdSJeff Roberson 	 * it's spinlock has been created.
454a54e85fdSJeff Roberson 	 * PROC_SLOCK_ASSERT(p, MA_OWNED);
455a54e85fdSJeff Roberson 	 */
45671fad9fdSJulian Elischer 	td->td_state    = TDS_INACTIVE;
45744990b8cSJulian Elischer 	td->td_proc     = p;
458b61ce5b0SJeff Roberson 	td->td_flags    = TDF_INMEM;
45944990b8cSJulian Elischer 
4601faf202eSJulian Elischer 	LIST_INIT(&td->td_contested);
461eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[0]);
462eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[1]);
4639104847fSDavid Xu 	sigqueue_init(&td->td_sigqueue, p);
464c06eb4e2SSam Leffler 	callout_init(&td->td_slpcallout, CALLOUT_MPSAFE);
46544990b8cSJulian Elischer 	TAILQ_INSERT_HEAD(&p->p_threads, td, td_plist);
46644990b8cSJulian Elischer 	p->p_numthreads++;
46744990b8cSJulian Elischer }
46844990b8cSJulian Elischer 
469ed062c8dSJulian Elischer /*
470e5bedcefSJulian Elischer  * Convert a process with one thread to an unthreaded process.
471e5bedcefSJulian Elischer  */
472e5bedcefSJulian Elischer void
473e5bedcefSJulian Elischer thread_unthread(struct thread *td)
474e5bedcefSJulian Elischer {
475e5bedcefSJulian Elischer 	struct proc *p = td->td_proc;
476e5bedcefSJulian Elischer 
477e5bedcefSJulian Elischer 	KASSERT((p->p_numthreads == 1), ("Unthreading with >1 threads"));
4788460a577SJohn Birrell 	p->p_flag &= ~P_HADTHREADS;
479e5bedcefSJulian Elischer }
480e5bedcefSJulian Elischer 
481e5bedcefSJulian Elischer /*
482ed062c8dSJulian Elischer  * Called from:
483ed062c8dSJulian Elischer  *  thread_exit()
484ed062c8dSJulian Elischer  */
485d3a0bd78SJulian Elischer void
486d3a0bd78SJulian Elischer thread_unlink(struct thread *td)
487d3a0bd78SJulian Elischer {
488d3a0bd78SJulian Elischer 	struct proc *p = td->td_proc;
489d3a0bd78SJulian Elischer 
490a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
491d3a0bd78SJulian Elischer 	TAILQ_REMOVE(&p->p_threads, td, td_plist);
492d3a0bd78SJulian Elischer 	p->p_numthreads--;
493d3a0bd78SJulian Elischer 	/* could clear a few other things here */
4948460a577SJohn Birrell 	/* Must  NOT clear links to proc! */
4955c8329edSJulian Elischer }
4965c8329edSJulian Elischer 
4975215b187SJeff Roberson /*
49844990b8cSJulian Elischer  * Enforce single-threading.
49944990b8cSJulian Elischer  *
50044990b8cSJulian Elischer  * Returns 1 if the caller must abort (another thread is waiting to
50144990b8cSJulian Elischer  * exit the process or similar). Process is locked!
50244990b8cSJulian Elischer  * Returns 0 when you are successfully the only thread running.
50344990b8cSJulian Elischer  * A process has successfully single threaded in the suspend mode when
50444990b8cSJulian Elischer  * There are no threads in user mode. Threads in the kernel must be
50544990b8cSJulian Elischer  * allowed to continue until they get to the user boundary. They may even
50644990b8cSJulian Elischer  * copy out their return values and data before suspending. They may however be
507e2668f55SMaxim Konovalov  * accelerated in reaching the user boundary as we will wake up
50844990b8cSJulian Elischer  * any sleeping threads that are interruptable. (PCATCH).
50944990b8cSJulian Elischer  */
51044990b8cSJulian Elischer int
511906ac69dSDavid Xu thread_single(int mode)
51244990b8cSJulian Elischer {
51344990b8cSJulian Elischer 	struct thread *td;
51444990b8cSJulian Elischer 	struct thread *td2;
51544990b8cSJulian Elischer 	struct proc *p;
516ec008e96SDavid Xu 	int remaining;
51744990b8cSJulian Elischer 
51844990b8cSJulian Elischer 	td = curthread;
51944990b8cSJulian Elischer 	p = td->td_proc;
52037814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
52144990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
52244990b8cSJulian Elischer 	KASSERT((td != NULL), ("curthread is NULL"));
52344990b8cSJulian Elischer 
524ed062c8dSJulian Elischer 	if ((p->p_flag & P_HADTHREADS) == 0)
52544990b8cSJulian Elischer 		return (0);
52644990b8cSJulian Elischer 
527e3b9bf71SJulian Elischer 	/* Is someone already single threading? */
528906ac69dSDavid Xu 	if (p->p_singlethread != NULL && p->p_singlethread != td)
52944990b8cSJulian Elischer 		return (1);
53044990b8cSJulian Elischer 
531906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
532906ac69dSDavid Xu 		p->p_flag |= P_SINGLE_EXIT;
533906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_BOUNDARY;
534906ac69dSDavid Xu 	} else {
535906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_EXIT;
536906ac69dSDavid Xu 		if (mode == SINGLE_BOUNDARY)
537906ac69dSDavid Xu 			p->p_flag |= P_SINGLE_BOUNDARY;
538906ac69dSDavid Xu 		else
539906ac69dSDavid Xu 			p->p_flag &= ~P_SINGLE_BOUNDARY;
540906ac69dSDavid Xu 	}
5411279572aSDavid Xu 	p->p_flag |= P_STOPPED_SINGLE;
542a54e85fdSJeff Roberson 	PROC_SLOCK(p);
543112afcb2SJohn Baldwin 	p->p_singlethread = td;
544906ac69dSDavid Xu 	if (mode == SINGLE_EXIT)
545ec008e96SDavid Xu 		remaining = p->p_numthreads;
546906ac69dSDavid Xu 	else if (mode == SINGLE_BOUNDARY)
547906ac69dSDavid Xu 		remaining = p->p_numthreads - p->p_boundary_count;
548906ac69dSDavid Xu 	else
549ec008e96SDavid Xu 		remaining = p->p_numthreads - p->p_suspcount;
550ec008e96SDavid Xu 	while (remaining != 1) {
551bf1a3220SDavid Xu 		if (P_SHOULDSTOP(p) != P_STOPPED_SINGLE)
552bf1a3220SDavid Xu 			goto stopme;
55344990b8cSJulian Elischer 		FOREACH_THREAD_IN_PROC(p, td2) {
55444990b8cSJulian Elischer 			if (td2 == td)
55544990b8cSJulian Elischer 				continue;
556a54e85fdSJeff Roberson 			thread_lock(td2);
557588257e8SDavid Xu 			td2->td_flags |= TDF_ASTPENDING;
55871fad9fdSJulian Elischer 			if (TD_IS_INHIBITED(td2)) {
559906ac69dSDavid Xu 				switch (mode) {
560906ac69dSDavid Xu 				case SINGLE_EXIT:
561cbf4e354SDavid Xu 					if (td->td_flags & TDF_DBSUSPEND)
562cbf4e354SDavid Xu 						td->td_flags &= ~TDF_DBSUSPEND;
563906ac69dSDavid Xu 					if (TD_IS_SUSPENDED(td2))
56471fad9fdSJulian Elischer 						thread_unsuspend_one(td2);
56533862f40SDavid Xu 					if (TD_ON_SLEEPQ(td2) &&
566906ac69dSDavid Xu 					    (td2->td_flags & TDF_SINTR))
56794f0972bSDavid Xu 						sleepq_abort(td2, EINTR);
568906ac69dSDavid Xu 					break;
569906ac69dSDavid Xu 				case SINGLE_BOUNDARY:
570906ac69dSDavid Xu 					break;
571906ac69dSDavid Xu 				default:
572a54e85fdSJeff Roberson 					if (TD_IS_SUSPENDED(td2)) {
573a54e85fdSJeff Roberson 						thread_unlock(td2);
5749d102777SJulian Elischer 						continue;
575a54e85fdSJeff Roberson 					}
5765215b187SJeff Roberson 					/*
5772da78e38SRobert Watson 					 * maybe other inhibited states too?
5785215b187SJeff Roberson 					 */
5798acf6057SDavid Xu 					if ((td2->td_flags & TDF_SINTR) &&
5808acf6057SDavid Xu 					    (td2->td_inhibitors &
5818acf6057SDavid Xu 					    (TDI_SLEEPING | TDI_SWAPPED)))
5829d102777SJulian Elischer 						thread_suspend_one(td2);
583906ac69dSDavid Xu 					break;
58444990b8cSJulian Elischer 				}
58544990b8cSJulian Elischer 			}
586d8267df7SDavid Xu #ifdef SMP
587d8267df7SDavid Xu 			else if (TD_IS_RUNNING(td2) && td != td2) {
588d8267df7SDavid Xu 				forward_signal(td2);
589d8267df7SDavid Xu 			}
590d8267df7SDavid Xu #endif
591a54e85fdSJeff Roberson 			thread_unlock(td2);
5929d102777SJulian Elischer 		}
593906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
594ec008e96SDavid Xu 			remaining = p->p_numthreads;
595906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
596906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
597ec008e96SDavid Xu 		else
598ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
599ec008e96SDavid Xu 
6009d102777SJulian Elischer 		/*
6019d102777SJulian Elischer 		 * Maybe we suspended some threads.. was it enough?
6029d102777SJulian Elischer 		 */
603ec008e96SDavid Xu 		if (remaining == 1)
6049d102777SJulian Elischer 			break;
6059d102777SJulian Elischer 
606bf1a3220SDavid Xu stopme:
60744990b8cSJulian Elischer 		/*
60844990b8cSJulian Elischer 		 * Wake us up when everyone else has suspended.
609e3b9bf71SJulian Elischer 		 * In the mean time we suspend as well.
61044990b8cSJulian Elischer 		 */
611a54e85fdSJeff Roberson 		thread_suspend_switch(td);
612906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
613ec008e96SDavid Xu 			remaining = p->p_numthreads;
614906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
615906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
616ec008e96SDavid Xu 		else
617ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
61844990b8cSJulian Elischer 	}
619906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
62091599697SJulian Elischer 		/*
62191599697SJulian Elischer 		 * We have gotten rid of all the other threads and we
62291599697SJulian Elischer 		 * are about to either exit or exec. In either case,
62391599697SJulian Elischer 		 * we try our utmost  to revert to being a non-threaded
62491599697SJulian Elischer 		 * process.
62591599697SJulian Elischer 		 */
626ed062c8dSJulian Elischer 		p->p_singlethread = NULL;
62764895117SDavid Xu 		p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT);
628e5bedcefSJulian Elischer 		thread_unthread(td);
62991599697SJulian Elischer 	}
630a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
63144990b8cSJulian Elischer 	return (0);
63244990b8cSJulian Elischer }
63344990b8cSJulian Elischer 
63444990b8cSJulian Elischer /*
63544990b8cSJulian Elischer  * Called in from locations that can safely check to see
63644990b8cSJulian Elischer  * whether we have to suspend or at least throttle for a
63744990b8cSJulian Elischer  * single-thread event (e.g. fork).
63844990b8cSJulian Elischer  *
63944990b8cSJulian Elischer  * Such locations include userret().
64044990b8cSJulian Elischer  * If the "return_instead" argument is non zero, the thread must be able to
64144990b8cSJulian Elischer  * accept 0 (caller may continue), or 1 (caller must abort) as a result.
64244990b8cSJulian Elischer  *
64344990b8cSJulian Elischer  * The 'return_instead' argument tells the function if it may do a
64444990b8cSJulian Elischer  * thread_exit() or suspend, or whether the caller must abort and back
64544990b8cSJulian Elischer  * out instead.
64644990b8cSJulian Elischer  *
64744990b8cSJulian Elischer  * If the thread that set the single_threading request has set the
64844990b8cSJulian Elischer  * P_SINGLE_EXIT bit in the process flags then this call will never return
64944990b8cSJulian Elischer  * if 'return_instead' is false, but will exit.
65044990b8cSJulian Elischer  *
65144990b8cSJulian Elischer  * P_SINGLE_EXIT | return_instead == 0| return_instead != 0
65244990b8cSJulian Elischer  *---------------+--------------------+---------------------
65344990b8cSJulian Elischer  *       0       | returns 0          |   returns 0 or 1
65444990b8cSJulian Elischer  *               | when ST ends       |   immediatly
65544990b8cSJulian Elischer  *---------------+--------------------+---------------------
65644990b8cSJulian Elischer  *       1       | thread exits       |   returns 1
65744990b8cSJulian Elischer  *               |                    |  immediatly
65844990b8cSJulian Elischer  * 0 = thread_exit() or suspension ok,
65944990b8cSJulian Elischer  * other = return error instead of stopping the thread.
66044990b8cSJulian Elischer  *
66144990b8cSJulian Elischer  * While a full suspension is under effect, even a single threading
66244990b8cSJulian Elischer  * thread would be suspended if it made this call (but it shouldn't).
66344990b8cSJulian Elischer  * This call should only be made from places where
66444990b8cSJulian Elischer  * thread_exit() would be safe as that may be the outcome unless
66544990b8cSJulian Elischer  * return_instead is set.
66644990b8cSJulian Elischer  */
66744990b8cSJulian Elischer int
66844990b8cSJulian Elischer thread_suspend_check(int return_instead)
66944990b8cSJulian Elischer {
670ecafb24bSJuli Mallett 	struct thread *td;
671ecafb24bSJuli Mallett 	struct proc *p;
67244990b8cSJulian Elischer 
67344990b8cSJulian Elischer 	td = curthread;
67444990b8cSJulian Elischer 	p = td->td_proc;
67537814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
67644990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
677cbf4e354SDavid Xu 	while (P_SHOULDSTOP(p) ||
678cbf4e354SDavid Xu 	      ((p->p_flag & P_TRACED) && (td->td_flags & TDF_DBSUSPEND))) {
6791279572aSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
68044990b8cSJulian Elischer 			KASSERT(p->p_singlethread != NULL,
68144990b8cSJulian Elischer 			    ("singlethread not set"));
68244990b8cSJulian Elischer 			/*
683e3b9bf71SJulian Elischer 			 * The only suspension in action is a
684e3b9bf71SJulian Elischer 			 * single-threading. Single threader need not stop.
685b6d5995eSJulian Elischer 			 * XXX Should be safe to access unlocked
686b6d5995eSJulian Elischer 			 * as it can only be set to be true by us.
68744990b8cSJulian Elischer 			 */
688e3b9bf71SJulian Elischer 			if (p->p_singlethread == td)
68944990b8cSJulian Elischer 				return (0);	/* Exempt from stopping. */
69044990b8cSJulian Elischer 		}
69145a4bfa1SDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && return_instead)
69294f0972bSDavid Xu 			return (EINTR);
69344990b8cSJulian Elischer 
694906ac69dSDavid Xu 		/* Should we goto user boundary if we didn't come from there? */
695906ac69dSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
696906ac69dSDavid Xu 		    (p->p_flag & P_SINGLE_BOUNDARY) && return_instead)
69794f0972bSDavid Xu 			return (ERESTART);
698906ac69dSDavid Xu 
6999104847fSDavid Xu 		/* If thread will exit, flush its pending signals */
7009104847fSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
7019104847fSDavid Xu 			sigqueue_flush(&td->td_sigqueue);
7029104847fSDavid Xu 
703a54e85fdSJeff Roberson 		PROC_SLOCK(p);
704e574e444SDavid Xu 		thread_stopped(p);
70544990b8cSJulian Elischer 		/*
70644990b8cSJulian Elischer 		 * If the process is waiting for us to exit,
70744990b8cSJulian Elischer 		 * this thread should just suicide.
7081279572aSDavid Xu 		 * Assumes that P_SINGLE_EXIT implies P_STOPPED_SINGLE.
70944990b8cSJulian Elischer 		 */
710906ac69dSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
71144990b8cSJulian Elischer 			thread_exit();
712a54e85fdSJeff Roberson 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
713a54e85fdSJeff Roberson 			if (p->p_numthreads == p->p_suspcount + 1) {
714a54e85fdSJeff Roberson 				thread_lock(p->p_singlethread);
715a54e85fdSJeff Roberson 				thread_unsuspend_one(p->p_singlethread);
716a54e85fdSJeff Roberson 				thread_unlock(p->p_singlethread);
717a54e85fdSJeff Roberson 			}
718a54e85fdSJeff Roberson 		}
719a54e85fdSJeff Roberson 		PROC_UNLOCK(p);
720a54e85fdSJeff Roberson 		thread_lock(td);
72144990b8cSJulian Elischer 		/*
72244990b8cSJulian Elischer 		 * When a thread suspends, it just
723ad1e7d28SJulian Elischer 		 * gets taken off all queues.
72444990b8cSJulian Elischer 		 */
72571fad9fdSJulian Elischer 		thread_suspend_one(td);
726906ac69dSDavid Xu 		if (return_instead == 0) {
727906ac69dSDavid Xu 			p->p_boundary_count++;
728906ac69dSDavid Xu 			td->td_flags |= TDF_BOUNDARY;
729cf19bf91SJulian Elischer 		}
730a54e85fdSJeff Roberson 		PROC_SUNLOCK(p);
731bf0acc27SJohn Baldwin 		mi_switch(SW_INVOL, NULL);
732a54e85fdSJeff Roberson 		if (return_instead == 0)
733906ac69dSDavid Xu 			td->td_flags &= ~TDF_BOUNDARY;
734a54e85fdSJeff Roberson 		thread_unlock(td);
73544990b8cSJulian Elischer 		PROC_LOCK(p);
736a54e85fdSJeff Roberson 		if (return_instead == 0)
737a54e85fdSJeff Roberson 			p->p_boundary_count--;
73844990b8cSJulian Elischer 	}
73944990b8cSJulian Elischer 	return (0);
74044990b8cSJulian Elischer }
74144990b8cSJulian Elischer 
74235c32a76SDavid Xu void
743a54e85fdSJeff Roberson thread_suspend_switch(struct thread *td)
744a54e85fdSJeff Roberson {
745a54e85fdSJeff Roberson 	struct proc *p;
746a54e85fdSJeff Roberson 
747a54e85fdSJeff Roberson 	p = td->td_proc;
748a54e85fdSJeff Roberson 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
749a54e85fdSJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
750a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
751a54e85fdSJeff Roberson 	/*
752a54e85fdSJeff Roberson 	 * We implement thread_suspend_one in stages here to avoid
753a54e85fdSJeff Roberson 	 * dropping the proc lock while the thread lock is owned.
754a54e85fdSJeff Roberson 	 */
755a54e85fdSJeff Roberson 	thread_stopped(p);
756a54e85fdSJeff Roberson 	p->p_suspcount++;
757a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
758a54e85fdSJeff Roberson 	thread_lock(td);
759a54e85fdSJeff Roberson 	TD_SET_SUSPENDED(td);
760c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
761a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
762a54e85fdSJeff Roberson 	DROP_GIANT();
763a54e85fdSJeff Roberson 	mi_switch(SW_VOL, NULL);
764a54e85fdSJeff Roberson 	thread_unlock(td);
765a54e85fdSJeff Roberson 	PICKUP_GIANT();
766a54e85fdSJeff Roberson 	PROC_LOCK(p);
767a54e85fdSJeff Roberson 	PROC_SLOCK(p);
768a54e85fdSJeff Roberson }
769a54e85fdSJeff Roberson 
770a54e85fdSJeff Roberson void
77135c32a76SDavid Xu thread_suspend_one(struct thread *td)
77235c32a76SDavid Xu {
77335c32a76SDavid Xu 	struct proc *p = td->td_proc;
77435c32a76SDavid Xu 
775a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
776a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
777e574e444SDavid Xu 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
77835c32a76SDavid Xu 	p->p_suspcount++;
77971fad9fdSJulian Elischer 	TD_SET_SUSPENDED(td);
780c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
78135c32a76SDavid Xu }
78235c32a76SDavid Xu 
78335c32a76SDavid Xu void
78435c32a76SDavid Xu thread_unsuspend_one(struct thread *td)
78535c32a76SDavid Xu {
78635c32a76SDavid Xu 	struct proc *p = td->td_proc;
78735c32a76SDavid Xu 
788a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
789a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
790ad1e7d28SJulian Elischer 	KASSERT(TD_IS_SUSPENDED(td), ("Thread not suspended"));
79171fad9fdSJulian Elischer 	TD_CLR_SUSPENDED(td);
79235c32a76SDavid Xu 	p->p_suspcount--;
79371fad9fdSJulian Elischer 	setrunnable(td);
79435c32a76SDavid Xu }
79535c32a76SDavid Xu 
79644990b8cSJulian Elischer /*
79744990b8cSJulian Elischer  * Allow all threads blocked by single threading to continue running.
79844990b8cSJulian Elischer  */
79944990b8cSJulian Elischer void
80044990b8cSJulian Elischer thread_unsuspend(struct proc *p)
80144990b8cSJulian Elischer {
80244990b8cSJulian Elischer 	struct thread *td;
80344990b8cSJulian Elischer 
80444990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
805a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
80644990b8cSJulian Elischer 	if (!P_SHOULDSTOP(p)) {
807ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
808a54e85fdSJeff Roberson 			thread_lock(td);
809ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
81035c32a76SDavid Xu 				thread_unsuspend_one(td);
81144990b8cSJulian Elischer 			}
812a54e85fdSJeff Roberson 			thread_unlock(td);
813ad1e7d28SJulian Elischer 		}
8141279572aSDavid Xu 	} else if ((P_SHOULDSTOP(p) == P_STOPPED_SINGLE) &&
81544990b8cSJulian Elischer 	    (p->p_numthreads == p->p_suspcount)) {
81644990b8cSJulian Elischer 		/*
81744990b8cSJulian Elischer 		 * Stopping everything also did the job for the single
81844990b8cSJulian Elischer 		 * threading request. Now we've downgraded to single-threaded,
81944990b8cSJulian Elischer 		 * let it continue.
82044990b8cSJulian Elischer 		 */
821a54e85fdSJeff Roberson 		thread_lock(p->p_singlethread);
82235c32a76SDavid Xu 		thread_unsuspend_one(p->p_singlethread);
823a54e85fdSJeff Roberson 		thread_unlock(p->p_singlethread);
82444990b8cSJulian Elischer 	}
82544990b8cSJulian Elischer }
82644990b8cSJulian Elischer 
827ed062c8dSJulian Elischer /*
828ed062c8dSJulian Elischer  * End the single threading mode..
829ed062c8dSJulian Elischer  */
83044990b8cSJulian Elischer void
83144990b8cSJulian Elischer thread_single_end(void)
83244990b8cSJulian Elischer {
83344990b8cSJulian Elischer 	struct thread *td;
83444990b8cSJulian Elischer 	struct proc *p;
83544990b8cSJulian Elischer 
83644990b8cSJulian Elischer 	td = curthread;
83744990b8cSJulian Elischer 	p = td->td_proc;
83844990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
839906ac69dSDavid Xu 	p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_SINGLE_BOUNDARY);
840a54e85fdSJeff Roberson 	PROC_SLOCK(p);
84144990b8cSJulian Elischer 	p->p_singlethread = NULL;
84249539972SJulian Elischer 	/*
84349539972SJulian Elischer 	 * If there are other threads they mey now run,
84449539972SJulian Elischer 	 * unless of course there is a blanket 'stop order'
84549539972SJulian Elischer 	 * on the process. The single threader must be allowed
84649539972SJulian Elischer 	 * to continue however as this is a bad place to stop.
84749539972SJulian Elischer 	 */
84849539972SJulian Elischer 	if ((p->p_numthreads != 1) && (!P_SHOULDSTOP(p))) {
849ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
850a54e85fdSJeff Roberson 			thread_lock(td);
851ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
85271fad9fdSJulian Elischer 				thread_unsuspend_one(td);
85344990b8cSJulian Elischer 			}
854a54e85fdSJeff Roberson 			thread_unlock(td);
85549539972SJulian Elischer 		}
856ad1e7d28SJulian Elischer 	}
857a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
85849539972SJulian Elischer }
8594fc21c09SDaniel Eischen 
86044355392SDavid Xu struct thread *
86144355392SDavid Xu thread_find(struct proc *p, lwpid_t tid)
86244355392SDavid Xu {
86344355392SDavid Xu 	struct thread *td;
86444355392SDavid Xu 
86544355392SDavid Xu 	PROC_LOCK_ASSERT(p, MA_OWNED);
866a54e85fdSJeff Roberson 	PROC_SLOCK(p);
86744355392SDavid Xu 	FOREACH_THREAD_IN_PROC(p, td) {
86844355392SDavid Xu 		if (td->td_tid == tid)
86944355392SDavid Xu 			break;
87044355392SDavid Xu 	}
871a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
87244355392SDavid Xu 	return (td);
87344355392SDavid Xu }
874