xref: /freebsd/sys/kern/kern_thread.c (revision 45aea8de6e4418c47b21b2fd4082fc216ec6fc60)
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 {
30645aea8deSJeff Roberson 	if (td->td_cpuset)
307d7f687fcSJeff Roberson 		cpuset_rel(td->td_cpuset);
308d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
3090c3967e7SMarcel Moolenaar 	cpu_thread_free(td);
31089b57fcfSKonstantin Belousov 	if (td->td_altkstack != 0)
31189b57fcfSKonstantin Belousov 		vm_thread_dispose_altkstack(td);
31289b57fcfSKonstantin Belousov 	if (td->td_kstack != 0)
31389b57fcfSKonstantin Belousov 		vm_thread_dispose(td);
31444990b8cSJulian Elischer 	uma_zfree(thread_zone, td);
31544990b8cSJulian Elischer }
31644990b8cSJulian Elischer 
31744990b8cSJulian Elischer /*
31844990b8cSJulian Elischer  * Discard the current thread and exit from its context.
31994e0a4cdSJulian Elischer  * Always called with scheduler locked.
32044990b8cSJulian Elischer  *
32144990b8cSJulian Elischer  * Because we can't free a thread while we're operating under its context,
322696058c3SJulian Elischer  * push the current thread into our CPU's deadthread holder. This means
323696058c3SJulian Elischer  * we needn't worry about someone else grabbing our context before we
3246617724cSJeff Roberson  * do a cpu_throw().
32544990b8cSJulian Elischer  */
32644990b8cSJulian Elischer void
32744990b8cSJulian Elischer thread_exit(void)
32844990b8cSJulian Elischer {
329e170bfdaSDavid Xu 	uint64_t new_switchtime;
33044990b8cSJulian Elischer 	struct thread *td;
3311c4bcd05SJeff Roberson 	struct thread *td2;
33244990b8cSJulian Elischer 	struct proc *p;
33344990b8cSJulian Elischer 
33444990b8cSJulian Elischer 	td = curthread;
33544990b8cSJulian Elischer 	p = td->td_proc;
33644990b8cSJulian Elischer 
337a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
338ed062c8dSJulian Elischer 	mtx_assert(&Giant, MA_NOTOWNED);
339a54e85fdSJeff Roberson 
34044990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
341ed062c8dSJulian Elischer 	KASSERT(p != NULL, ("thread exiting without a process"));
342cc701b73SRobert Watson 	CTR3(KTR_PROC, "thread_exit: thread %p (pid %ld, %s)", td,
343e01eafefSJulian Elischer 	    (long)p->p_pid, td->td_name);
3449104847fSDavid Xu 	KASSERT(TAILQ_EMPTY(&td->td_sigqueue.sq_list), ("signal pending"));
34544990b8cSJulian Elischer 
34689964dd2SRobert Watson #ifdef AUDIT
34789964dd2SRobert Watson 	AUDIT_SYSCALL_EXIT(0, td);
34889964dd2SRobert Watson #endif
349d10183d9SDavid Xu 	umtx_thread_exit(td);
350ed062c8dSJulian Elischer 	/*
351ed062c8dSJulian Elischer 	 * drop FPU & debug register state storage, or any other
352ed062c8dSJulian Elischer 	 * architecture specific resources that
353ed062c8dSJulian Elischer 	 * would not be on a new untouched process.
354ed062c8dSJulian Elischer 	 */
35544990b8cSJulian Elischer 	cpu_thread_exit(td);	/* XXXSMP */
35644990b8cSJulian Elischer 
357e170bfdaSDavid Xu 	/* Do the same timestamp bookkeeping that mi_switch() would do. */
358e170bfdaSDavid Xu 	new_switchtime = cpu_ticks();
359e170bfdaSDavid Xu 	p->p_rux.rux_runtime += (new_switchtime - PCPU_GET(switchtime));
360e170bfdaSDavid Xu 	PCPU_SET(switchtime, new_switchtime);
361e170bfdaSDavid Xu 	PCPU_SET(switchticks, ticks);
362b4b70819SAttilio Rao 	PCPU_INC(cnt.v_swtch);
363a140976eSAttilio Rao 	/* Save our resource usage in our process. */
364a140976eSAttilio Rao 	td->td_ru.ru_nvcsw++;
365a140976eSAttilio Rao 	rucollect(&p->p_ru, &td->td_ru);
366ed062c8dSJulian Elischer 	/*
3671faf202eSJulian Elischer 	 * The last thread is left attached to the process
3681faf202eSJulian Elischer 	 * So that the whole bundle gets recycled. Skip
369ed062c8dSJulian Elischer 	 * all this stuff if we never had threads.
370ed062c8dSJulian Elischer 	 * EXIT clears all sign of other threads when
371ed062c8dSJulian Elischer 	 * it goes to single threading, so the last thread always
372ed062c8dSJulian Elischer 	 * takes the short path.
3731faf202eSJulian Elischer 	 */
374ed062c8dSJulian Elischer 	if (p->p_flag & P_HADTHREADS) {
3751faf202eSJulian Elischer 		if (p->p_numthreads > 1) {
376d3a0bd78SJulian Elischer 			thread_unlink(td);
3771c4bcd05SJeff Roberson 			td2 = FIRST_THREAD_IN_PROC(p);
3781c4bcd05SJeff Roberson 			sched_exit_thread(td2, td);
379ed062c8dSJulian Elischer 
380ed062c8dSJulian Elischer 			/*
38144990b8cSJulian Elischer 			 * The test below is NOT true if we are the
3821faf202eSJulian Elischer 			 * sole exiting thread. P_STOPPED_SNGL is unset
38344990b8cSJulian Elischer 			 * in exit1() after it is the only survivor.
38444990b8cSJulian Elischer 			 */
3851279572aSDavid Xu 			if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
38644990b8cSJulian Elischer 				if (p->p_numthreads == p->p_suspcount) {
387a54e85fdSJeff Roberson 					thread_lock(p->p_singlethread);
38871fad9fdSJulian Elischer 					thread_unsuspend_one(p->p_singlethread);
389a54e85fdSJeff Roberson 					thread_unlock(p->p_singlethread);
39044990b8cSJulian Elischer 				}
39144990b8cSJulian Elischer 			}
39248bfcdddSJulian Elischer 
393ff8fbcffSJeff Roberson 			atomic_add_int(&td->td_proc->p_exitthreads, 1);
394696058c3SJulian Elischer 			PCPU_SET(deadthread, td);
3951faf202eSJulian Elischer 		} else {
396ed062c8dSJulian Elischer 			/*
397ed062c8dSJulian Elischer 			 * The last thread is exiting.. but not through exit()
398ed062c8dSJulian Elischer 			 */
399ed062c8dSJulian Elischer 			panic ("thread_exit: Last thread exiting on its own");
400ed062c8dSJulian Elischer 		}
4011faf202eSJulian Elischer 	}
402a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
403a54e85fdSJeff Roberson 	thread_lock(td);
404a140976eSAttilio Rao 	/* Save our tick information with both the thread and proc locked */
405a54e85fdSJeff Roberson 	ruxagg(&p->p_rux, td);
406a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
407dcc9954eSJulian Elischer 	td->td_state = TDS_INACTIVE;
408732d9528SJulian Elischer 	CTR1(KTR_PROC, "thread_exit: cpu_throw() thread %p", td);
409a54e85fdSJeff Roberson 	sched_throw(td);
410cc66ebe2SPeter Wemm 	panic("I'm a teapot!");
41144990b8cSJulian Elischer 	/* NOTREACHED */
41244990b8cSJulian Elischer }
41344990b8cSJulian Elischer 
41444990b8cSJulian Elischer /*
415696058c3SJulian Elischer  * Do any thread specific cleanups that may be needed in wait()
41637814395SPeter Wemm  * called with Giant, proc and schedlock not held.
417696058c3SJulian Elischer  */
418696058c3SJulian Elischer void
419696058c3SJulian Elischer thread_wait(struct proc *p)
420696058c3SJulian Elischer {
421696058c3SJulian Elischer 	struct thread *td;
422696058c3SJulian Elischer 
42337814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
42485495c72SJens Schweikhardt 	KASSERT((p->p_numthreads == 1), ("Multiple threads in wait1()"));
425ff8fbcffSJeff Roberson 	td = FIRST_THREAD_IN_PROC(p);
426ff8fbcffSJeff Roberson 	/* Lock the last thread so we spin until it exits cpu_throw(). */
427ff8fbcffSJeff Roberson 	thread_lock(td);
428ff8fbcffSJeff Roberson 	thread_unlock(td);
429ff8fbcffSJeff Roberson 	/* Wait for any remaining threads to exit cpu_throw(). */
430ff8fbcffSJeff Roberson 	while (p->p_exitthreads)
431ff8fbcffSJeff Roberson 		sched_relinquish(curthread);
432d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
433d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
434696058c3SJulian Elischer 	cpu_thread_clean(td);
435ed062c8dSJulian Elischer 	crfree(td->td_ucred);
436696058c3SJulian Elischer 	thread_reap();	/* check for zombie threads etc. */
437696058c3SJulian Elischer }
438696058c3SJulian Elischer 
439696058c3SJulian Elischer /*
44044990b8cSJulian Elischer  * Link a thread to a process.
4411faf202eSJulian Elischer  * set up anything that needs to be initialized for it to
4421faf202eSJulian Elischer  * be used by the process.
44344990b8cSJulian Elischer  */
44444990b8cSJulian Elischer void
4458460a577SJohn Birrell thread_link(struct thread *td, struct proc *p)
44644990b8cSJulian Elischer {
44744990b8cSJulian Elischer 
448a54e85fdSJeff Roberson 	/*
449a54e85fdSJeff Roberson 	 * XXX This can't be enabled because it's called for proc0 before
450374ae2a3SJeff Roberson 	 * its lock has been created.
451374ae2a3SJeff Roberson 	 * PROC_LOCK_ASSERT(p, MA_OWNED);
452a54e85fdSJeff Roberson 	 */
45371fad9fdSJulian Elischer 	td->td_state    = TDS_INACTIVE;
45444990b8cSJulian Elischer 	td->td_proc     = p;
455b61ce5b0SJeff Roberson 	td->td_flags    = TDF_INMEM;
45644990b8cSJulian Elischer 
4571faf202eSJulian Elischer 	LIST_INIT(&td->td_contested);
458eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[0]);
459eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[1]);
4609104847fSDavid Xu 	sigqueue_init(&td->td_sigqueue, p);
461c06eb4e2SSam Leffler 	callout_init(&td->td_slpcallout, CALLOUT_MPSAFE);
46244990b8cSJulian Elischer 	TAILQ_INSERT_HEAD(&p->p_threads, td, td_plist);
46344990b8cSJulian Elischer 	p->p_numthreads++;
46444990b8cSJulian Elischer }
46544990b8cSJulian Elischer 
466ed062c8dSJulian Elischer /*
467e5bedcefSJulian Elischer  * Convert a process with one thread to an unthreaded process.
468e5bedcefSJulian Elischer  */
469e5bedcefSJulian Elischer void
470e5bedcefSJulian Elischer thread_unthread(struct thread *td)
471e5bedcefSJulian Elischer {
472e5bedcefSJulian Elischer 	struct proc *p = td->td_proc;
473e5bedcefSJulian Elischer 
474e5bedcefSJulian Elischer 	KASSERT((p->p_numthreads == 1), ("Unthreading with >1 threads"));
4758460a577SJohn Birrell 	p->p_flag &= ~P_HADTHREADS;
476e5bedcefSJulian Elischer }
477e5bedcefSJulian Elischer 
478e5bedcefSJulian Elischer /*
479ed062c8dSJulian Elischer  * Called from:
480ed062c8dSJulian Elischer  *  thread_exit()
481ed062c8dSJulian Elischer  */
482d3a0bd78SJulian Elischer void
483d3a0bd78SJulian Elischer thread_unlink(struct thread *td)
484d3a0bd78SJulian Elischer {
485d3a0bd78SJulian Elischer 	struct proc *p = td->td_proc;
486d3a0bd78SJulian Elischer 
487374ae2a3SJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
488d3a0bd78SJulian Elischer 	TAILQ_REMOVE(&p->p_threads, td, td_plist);
489d3a0bd78SJulian Elischer 	p->p_numthreads--;
490d3a0bd78SJulian Elischer 	/* could clear a few other things here */
4918460a577SJohn Birrell 	/* Must  NOT clear links to proc! */
4925c8329edSJulian Elischer }
4935c8329edSJulian Elischer 
4945215b187SJeff Roberson /*
49544990b8cSJulian Elischer  * Enforce single-threading.
49644990b8cSJulian Elischer  *
49744990b8cSJulian Elischer  * Returns 1 if the caller must abort (another thread is waiting to
49844990b8cSJulian Elischer  * exit the process or similar). Process is locked!
49944990b8cSJulian Elischer  * Returns 0 when you are successfully the only thread running.
50044990b8cSJulian Elischer  * A process has successfully single threaded in the suspend mode when
50144990b8cSJulian Elischer  * There are no threads in user mode. Threads in the kernel must be
50244990b8cSJulian Elischer  * allowed to continue until they get to the user boundary. They may even
50344990b8cSJulian Elischer  * copy out their return values and data before suspending. They may however be
504e2668f55SMaxim Konovalov  * accelerated in reaching the user boundary as we will wake up
50544990b8cSJulian Elischer  * any sleeping threads that are interruptable. (PCATCH).
50644990b8cSJulian Elischer  */
50744990b8cSJulian Elischer int
508906ac69dSDavid Xu thread_single(int mode)
50944990b8cSJulian Elischer {
51044990b8cSJulian Elischer 	struct thread *td;
51144990b8cSJulian Elischer 	struct thread *td2;
51244990b8cSJulian Elischer 	struct proc *p;
513ec008e96SDavid Xu 	int remaining;
51444990b8cSJulian Elischer 
51544990b8cSJulian Elischer 	td = curthread;
51644990b8cSJulian Elischer 	p = td->td_proc;
51737814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
51844990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
51944990b8cSJulian Elischer 	KASSERT((td != NULL), ("curthread is NULL"));
52044990b8cSJulian Elischer 
521ed062c8dSJulian Elischer 	if ((p->p_flag & P_HADTHREADS) == 0)
52244990b8cSJulian Elischer 		return (0);
52344990b8cSJulian Elischer 
524e3b9bf71SJulian Elischer 	/* Is someone already single threading? */
525906ac69dSDavid Xu 	if (p->p_singlethread != NULL && p->p_singlethread != td)
52644990b8cSJulian Elischer 		return (1);
52744990b8cSJulian Elischer 
528906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
529906ac69dSDavid Xu 		p->p_flag |= P_SINGLE_EXIT;
530906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_BOUNDARY;
531906ac69dSDavid Xu 	} else {
532906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_EXIT;
533906ac69dSDavid Xu 		if (mode == SINGLE_BOUNDARY)
534906ac69dSDavid Xu 			p->p_flag |= P_SINGLE_BOUNDARY;
535906ac69dSDavid Xu 		else
536906ac69dSDavid Xu 			p->p_flag &= ~P_SINGLE_BOUNDARY;
537906ac69dSDavid Xu 	}
5381279572aSDavid Xu 	p->p_flag |= P_STOPPED_SINGLE;
539a54e85fdSJeff Roberson 	PROC_SLOCK(p);
540112afcb2SJohn Baldwin 	p->p_singlethread = td;
541906ac69dSDavid Xu 	if (mode == SINGLE_EXIT)
542ec008e96SDavid Xu 		remaining = p->p_numthreads;
543906ac69dSDavid Xu 	else if (mode == SINGLE_BOUNDARY)
544906ac69dSDavid Xu 		remaining = p->p_numthreads - p->p_boundary_count;
545906ac69dSDavid Xu 	else
546ec008e96SDavid Xu 		remaining = p->p_numthreads - p->p_suspcount;
547ec008e96SDavid Xu 	while (remaining != 1) {
548bf1a3220SDavid Xu 		if (P_SHOULDSTOP(p) != P_STOPPED_SINGLE)
549bf1a3220SDavid Xu 			goto stopme;
55044990b8cSJulian Elischer 		FOREACH_THREAD_IN_PROC(p, td2) {
55144990b8cSJulian Elischer 			if (td2 == td)
55244990b8cSJulian Elischer 				continue;
553a54e85fdSJeff Roberson 			thread_lock(td2);
554588257e8SDavid Xu 			td2->td_flags |= TDF_ASTPENDING;
55571fad9fdSJulian Elischer 			if (TD_IS_INHIBITED(td2)) {
556906ac69dSDavid Xu 				switch (mode) {
557906ac69dSDavid Xu 				case SINGLE_EXIT:
558cbf4e354SDavid Xu 					if (td->td_flags & TDF_DBSUSPEND)
559cbf4e354SDavid Xu 						td->td_flags &= ~TDF_DBSUSPEND;
560906ac69dSDavid Xu 					if (TD_IS_SUSPENDED(td2))
56171fad9fdSJulian Elischer 						thread_unsuspend_one(td2);
56233862f40SDavid Xu 					if (TD_ON_SLEEPQ(td2) &&
563906ac69dSDavid Xu 					    (td2->td_flags & TDF_SINTR))
56494f0972bSDavid Xu 						sleepq_abort(td2, EINTR);
565906ac69dSDavid Xu 					break;
566906ac69dSDavid Xu 				case SINGLE_BOUNDARY:
567906ac69dSDavid Xu 					break;
568906ac69dSDavid Xu 				default:
569a54e85fdSJeff Roberson 					if (TD_IS_SUSPENDED(td2)) {
570a54e85fdSJeff Roberson 						thread_unlock(td2);
5719d102777SJulian Elischer 						continue;
572a54e85fdSJeff Roberson 					}
5735215b187SJeff Roberson 					/*
5742da78e38SRobert Watson 					 * maybe other inhibited states too?
5755215b187SJeff Roberson 					 */
5768acf6057SDavid Xu 					if ((td2->td_flags & TDF_SINTR) &&
5778acf6057SDavid Xu 					    (td2->td_inhibitors &
5788acf6057SDavid Xu 					    (TDI_SLEEPING | TDI_SWAPPED)))
5799d102777SJulian Elischer 						thread_suspend_one(td2);
580906ac69dSDavid Xu 					break;
58144990b8cSJulian Elischer 				}
58244990b8cSJulian Elischer 			}
583d8267df7SDavid Xu #ifdef SMP
584d8267df7SDavid Xu 			else if (TD_IS_RUNNING(td2) && td != td2) {
585d8267df7SDavid Xu 				forward_signal(td2);
586d8267df7SDavid Xu 			}
587d8267df7SDavid Xu #endif
588a54e85fdSJeff Roberson 			thread_unlock(td2);
5899d102777SJulian Elischer 		}
590906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
591ec008e96SDavid Xu 			remaining = p->p_numthreads;
592906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
593906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
594ec008e96SDavid Xu 		else
595ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
596ec008e96SDavid Xu 
5979d102777SJulian Elischer 		/*
5989d102777SJulian Elischer 		 * Maybe we suspended some threads.. was it enough?
5999d102777SJulian Elischer 		 */
600ec008e96SDavid Xu 		if (remaining == 1)
6019d102777SJulian Elischer 			break;
6029d102777SJulian Elischer 
603bf1a3220SDavid Xu stopme:
60444990b8cSJulian Elischer 		/*
60544990b8cSJulian Elischer 		 * Wake us up when everyone else has suspended.
606e3b9bf71SJulian Elischer 		 * In the mean time we suspend as well.
60744990b8cSJulian Elischer 		 */
608a54e85fdSJeff Roberson 		thread_suspend_switch(td);
609906ac69dSDavid Xu 		if (mode == SINGLE_EXIT)
610ec008e96SDavid Xu 			remaining = p->p_numthreads;
611906ac69dSDavid Xu 		else if (mode == SINGLE_BOUNDARY)
612906ac69dSDavid Xu 			remaining = p->p_numthreads - p->p_boundary_count;
613ec008e96SDavid Xu 		else
614ec008e96SDavid Xu 			remaining = p->p_numthreads - p->p_suspcount;
61544990b8cSJulian Elischer 	}
616906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
61791599697SJulian Elischer 		/*
61891599697SJulian Elischer 		 * We have gotten rid of all the other threads and we
61991599697SJulian Elischer 		 * are about to either exit or exec. In either case,
62091599697SJulian Elischer 		 * we try our utmost  to revert to being a non-threaded
62191599697SJulian Elischer 		 * process.
62291599697SJulian Elischer 		 */
623ed062c8dSJulian Elischer 		p->p_singlethread = NULL;
62464895117SDavid Xu 		p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT);
625e5bedcefSJulian Elischer 		thread_unthread(td);
62691599697SJulian Elischer 	}
627a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
62844990b8cSJulian Elischer 	return (0);
62944990b8cSJulian Elischer }
63044990b8cSJulian Elischer 
63144990b8cSJulian Elischer /*
63244990b8cSJulian Elischer  * Called in from locations that can safely check to see
63344990b8cSJulian Elischer  * whether we have to suspend or at least throttle for a
63444990b8cSJulian Elischer  * single-thread event (e.g. fork).
63544990b8cSJulian Elischer  *
63644990b8cSJulian Elischer  * Such locations include userret().
63744990b8cSJulian Elischer  * If the "return_instead" argument is non zero, the thread must be able to
63844990b8cSJulian Elischer  * accept 0 (caller may continue), or 1 (caller must abort) as a result.
63944990b8cSJulian Elischer  *
64044990b8cSJulian Elischer  * The 'return_instead' argument tells the function if it may do a
64144990b8cSJulian Elischer  * thread_exit() or suspend, or whether the caller must abort and back
64244990b8cSJulian Elischer  * out instead.
64344990b8cSJulian Elischer  *
64444990b8cSJulian Elischer  * If the thread that set the single_threading request has set the
64544990b8cSJulian Elischer  * P_SINGLE_EXIT bit in the process flags then this call will never return
64644990b8cSJulian Elischer  * if 'return_instead' is false, but will exit.
64744990b8cSJulian Elischer  *
64844990b8cSJulian Elischer  * P_SINGLE_EXIT | return_instead == 0| return_instead != 0
64944990b8cSJulian Elischer  *---------------+--------------------+---------------------
65044990b8cSJulian Elischer  *       0       | returns 0          |   returns 0 or 1
65144990b8cSJulian Elischer  *               | when ST ends       |   immediatly
65244990b8cSJulian Elischer  *---------------+--------------------+---------------------
65344990b8cSJulian Elischer  *       1       | thread exits       |   returns 1
65444990b8cSJulian Elischer  *               |                    |  immediatly
65544990b8cSJulian Elischer  * 0 = thread_exit() or suspension ok,
65644990b8cSJulian Elischer  * other = return error instead of stopping the thread.
65744990b8cSJulian Elischer  *
65844990b8cSJulian Elischer  * While a full suspension is under effect, even a single threading
65944990b8cSJulian Elischer  * thread would be suspended if it made this call (but it shouldn't).
66044990b8cSJulian Elischer  * This call should only be made from places where
66144990b8cSJulian Elischer  * thread_exit() would be safe as that may be the outcome unless
66244990b8cSJulian Elischer  * return_instead is set.
66344990b8cSJulian Elischer  */
66444990b8cSJulian Elischer int
66544990b8cSJulian Elischer thread_suspend_check(int return_instead)
66644990b8cSJulian Elischer {
667ecafb24bSJuli Mallett 	struct thread *td;
668ecafb24bSJuli Mallett 	struct proc *p;
66944990b8cSJulian Elischer 
67044990b8cSJulian Elischer 	td = curthread;
67144990b8cSJulian Elischer 	p = td->td_proc;
67237814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
67344990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
674cbf4e354SDavid Xu 	while (P_SHOULDSTOP(p) ||
675cbf4e354SDavid Xu 	      ((p->p_flag & P_TRACED) && (td->td_flags & TDF_DBSUSPEND))) {
6761279572aSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
67744990b8cSJulian Elischer 			KASSERT(p->p_singlethread != NULL,
67844990b8cSJulian Elischer 			    ("singlethread not set"));
67944990b8cSJulian Elischer 			/*
680e3b9bf71SJulian Elischer 			 * The only suspension in action is a
681e3b9bf71SJulian Elischer 			 * single-threading. Single threader need not stop.
682b6d5995eSJulian Elischer 			 * XXX Should be safe to access unlocked
683b6d5995eSJulian Elischer 			 * as it can only be set to be true by us.
68444990b8cSJulian Elischer 			 */
685e3b9bf71SJulian Elischer 			if (p->p_singlethread == td)
68644990b8cSJulian Elischer 				return (0);	/* Exempt from stopping. */
68744990b8cSJulian Elischer 		}
68845a4bfa1SDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && return_instead)
68994f0972bSDavid Xu 			return (EINTR);
69044990b8cSJulian Elischer 
691906ac69dSDavid Xu 		/* Should we goto user boundary if we didn't come from there? */
692906ac69dSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
693906ac69dSDavid Xu 		    (p->p_flag & P_SINGLE_BOUNDARY) && return_instead)
69494f0972bSDavid Xu 			return (ERESTART);
695906ac69dSDavid Xu 
6969104847fSDavid Xu 		/* If thread will exit, flush its pending signals */
6979104847fSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
6989104847fSDavid Xu 			sigqueue_flush(&td->td_sigqueue);
6999104847fSDavid Xu 
700a54e85fdSJeff Roberson 		PROC_SLOCK(p);
701e574e444SDavid Xu 		thread_stopped(p);
70244990b8cSJulian Elischer 		/*
70344990b8cSJulian Elischer 		 * If the process is waiting for us to exit,
70444990b8cSJulian Elischer 		 * this thread should just suicide.
7051279572aSDavid Xu 		 * Assumes that P_SINGLE_EXIT implies P_STOPPED_SINGLE.
70644990b8cSJulian Elischer 		 */
707906ac69dSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td))
70844990b8cSJulian Elischer 			thread_exit();
709a54e85fdSJeff Roberson 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
710a54e85fdSJeff Roberson 			if (p->p_numthreads == p->p_suspcount + 1) {
711a54e85fdSJeff Roberson 				thread_lock(p->p_singlethread);
712a54e85fdSJeff Roberson 				thread_unsuspend_one(p->p_singlethread);
713a54e85fdSJeff Roberson 				thread_unlock(p->p_singlethread);
714a54e85fdSJeff Roberson 			}
715a54e85fdSJeff Roberson 		}
716a54e85fdSJeff Roberson 		PROC_UNLOCK(p);
717a54e85fdSJeff Roberson 		thread_lock(td);
71844990b8cSJulian Elischer 		/*
71944990b8cSJulian Elischer 		 * When a thread suspends, it just
720ad1e7d28SJulian Elischer 		 * gets taken off all queues.
72144990b8cSJulian Elischer 		 */
72271fad9fdSJulian Elischer 		thread_suspend_one(td);
723906ac69dSDavid Xu 		if (return_instead == 0) {
724906ac69dSDavid Xu 			p->p_boundary_count++;
725906ac69dSDavid Xu 			td->td_flags |= TDF_BOUNDARY;
726cf19bf91SJulian Elischer 		}
727a54e85fdSJeff Roberson 		PROC_SUNLOCK(p);
728bf0acc27SJohn Baldwin 		mi_switch(SW_INVOL, NULL);
729a54e85fdSJeff Roberson 		if (return_instead == 0)
730906ac69dSDavid Xu 			td->td_flags &= ~TDF_BOUNDARY;
731a54e85fdSJeff Roberson 		thread_unlock(td);
73244990b8cSJulian Elischer 		PROC_LOCK(p);
733a54e85fdSJeff Roberson 		if (return_instead == 0)
734a54e85fdSJeff Roberson 			p->p_boundary_count--;
73544990b8cSJulian Elischer 	}
73644990b8cSJulian Elischer 	return (0);
73744990b8cSJulian Elischer }
73844990b8cSJulian Elischer 
73935c32a76SDavid Xu void
740a54e85fdSJeff Roberson thread_suspend_switch(struct thread *td)
741a54e85fdSJeff Roberson {
742a54e85fdSJeff Roberson 	struct proc *p;
743a54e85fdSJeff Roberson 
744a54e85fdSJeff Roberson 	p = td->td_proc;
745a54e85fdSJeff Roberson 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
746a54e85fdSJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
747a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
748a54e85fdSJeff Roberson 	/*
749a54e85fdSJeff Roberson 	 * We implement thread_suspend_one in stages here to avoid
750a54e85fdSJeff Roberson 	 * dropping the proc lock while the thread lock is owned.
751a54e85fdSJeff Roberson 	 */
752a54e85fdSJeff Roberson 	thread_stopped(p);
753a54e85fdSJeff Roberson 	p->p_suspcount++;
754a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
755a54e85fdSJeff Roberson 	thread_lock(td);
756a54e85fdSJeff Roberson 	TD_SET_SUSPENDED(td);
757c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
758a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
759a54e85fdSJeff Roberson 	DROP_GIANT();
760a54e85fdSJeff Roberson 	mi_switch(SW_VOL, NULL);
761a54e85fdSJeff Roberson 	thread_unlock(td);
762a54e85fdSJeff Roberson 	PICKUP_GIANT();
763a54e85fdSJeff Roberson 	PROC_LOCK(p);
764a54e85fdSJeff Roberson 	PROC_SLOCK(p);
765a54e85fdSJeff Roberson }
766a54e85fdSJeff Roberson 
767a54e85fdSJeff Roberson void
76835c32a76SDavid Xu thread_suspend_one(struct thread *td)
76935c32a76SDavid Xu {
77035c32a76SDavid Xu 	struct proc *p = td->td_proc;
77135c32a76SDavid Xu 
772a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
773a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
774e574e444SDavid Xu 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
77535c32a76SDavid Xu 	p->p_suspcount++;
77671fad9fdSJulian Elischer 	TD_SET_SUSPENDED(td);
777c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
77835c32a76SDavid Xu }
77935c32a76SDavid Xu 
78035c32a76SDavid Xu void
78135c32a76SDavid Xu thread_unsuspend_one(struct thread *td)
78235c32a76SDavid Xu {
78335c32a76SDavid Xu 	struct proc *p = td->td_proc;
78435c32a76SDavid Xu 
785a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
786a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
787ad1e7d28SJulian Elischer 	KASSERT(TD_IS_SUSPENDED(td), ("Thread not suspended"));
78871fad9fdSJulian Elischer 	TD_CLR_SUSPENDED(td);
78935c32a76SDavid Xu 	p->p_suspcount--;
79071fad9fdSJulian Elischer 	setrunnable(td);
79135c32a76SDavid Xu }
79235c32a76SDavid Xu 
79344990b8cSJulian Elischer /*
79444990b8cSJulian Elischer  * Allow all threads blocked by single threading to continue running.
79544990b8cSJulian Elischer  */
79644990b8cSJulian Elischer void
79744990b8cSJulian Elischer thread_unsuspend(struct proc *p)
79844990b8cSJulian Elischer {
79944990b8cSJulian Elischer 	struct thread *td;
80044990b8cSJulian Elischer 
80144990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
802a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
80344990b8cSJulian Elischer 	if (!P_SHOULDSTOP(p)) {
804ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
805a54e85fdSJeff Roberson 			thread_lock(td);
806ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
80735c32a76SDavid Xu 				thread_unsuspend_one(td);
80844990b8cSJulian Elischer 			}
809a54e85fdSJeff Roberson 			thread_unlock(td);
810ad1e7d28SJulian Elischer 		}
8111279572aSDavid Xu 	} else if ((P_SHOULDSTOP(p) == P_STOPPED_SINGLE) &&
81244990b8cSJulian Elischer 	    (p->p_numthreads == p->p_suspcount)) {
81344990b8cSJulian Elischer 		/*
81444990b8cSJulian Elischer 		 * Stopping everything also did the job for the single
81544990b8cSJulian Elischer 		 * threading request. Now we've downgraded to single-threaded,
81644990b8cSJulian Elischer 		 * let it continue.
81744990b8cSJulian Elischer 		 */
818a54e85fdSJeff Roberson 		thread_lock(p->p_singlethread);
81935c32a76SDavid Xu 		thread_unsuspend_one(p->p_singlethread);
820a54e85fdSJeff Roberson 		thread_unlock(p->p_singlethread);
82144990b8cSJulian Elischer 	}
82244990b8cSJulian Elischer }
82344990b8cSJulian Elischer 
824ed062c8dSJulian Elischer /*
825ed062c8dSJulian Elischer  * End the single threading mode..
826ed062c8dSJulian Elischer  */
82744990b8cSJulian Elischer void
82844990b8cSJulian Elischer thread_single_end(void)
82944990b8cSJulian Elischer {
83044990b8cSJulian Elischer 	struct thread *td;
83144990b8cSJulian Elischer 	struct proc *p;
83244990b8cSJulian Elischer 
83344990b8cSJulian Elischer 	td = curthread;
83444990b8cSJulian Elischer 	p = td->td_proc;
83544990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
836906ac69dSDavid Xu 	p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_SINGLE_BOUNDARY);
837a54e85fdSJeff Roberson 	PROC_SLOCK(p);
83844990b8cSJulian Elischer 	p->p_singlethread = NULL;
83949539972SJulian Elischer 	/*
84049539972SJulian Elischer 	 * If there are other threads they mey now run,
84149539972SJulian Elischer 	 * unless of course there is a blanket 'stop order'
84249539972SJulian Elischer 	 * on the process. The single threader must be allowed
84349539972SJulian Elischer 	 * to continue however as this is a bad place to stop.
84449539972SJulian Elischer 	 */
84549539972SJulian Elischer 	if ((p->p_numthreads != 1) && (!P_SHOULDSTOP(p))) {
846ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
847a54e85fdSJeff Roberson 			thread_lock(td);
848ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
84971fad9fdSJulian Elischer 				thread_unsuspend_one(td);
85044990b8cSJulian Elischer 			}
851a54e85fdSJeff Roberson 			thread_unlock(td);
85249539972SJulian Elischer 		}
853ad1e7d28SJulian Elischer 	}
854a54e85fdSJeff Roberson 	PROC_SUNLOCK(p);
85549539972SJulian Elischer }
8564fc21c09SDaniel Eischen 
85744355392SDavid Xu struct thread *
85844355392SDavid Xu thread_find(struct proc *p, lwpid_t tid)
85944355392SDavid Xu {
86044355392SDavid Xu 	struct thread *td;
86144355392SDavid Xu 
86244355392SDavid Xu 	PROC_LOCK_ASSERT(p, MA_OWNED);
86344355392SDavid Xu 	FOREACH_THREAD_IN_PROC(p, td) {
86444355392SDavid Xu 		if (td->td_tid == tid)
86544355392SDavid Xu 			break;
86644355392SDavid Xu 	}
86744355392SDavid Xu 	return (td);
86844355392SDavid Xu }
869