xref: /freebsd/sys/kern/kern_thread.c (revision d1ca25be49ab19232d650dd40f52e2dd7567b600)
19454b2d8SWarner Losh /*-
28a36da99SPedro F. Giffuni  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
38a36da99SPedro F. Giffuni  *
444990b8cSJulian Elischer  * Copyright (C) 2001 Julian Elischer <julian@freebsd.org>.
544990b8cSJulian Elischer  *  All rights reserved.
644990b8cSJulian Elischer  *
744990b8cSJulian Elischer  * Redistribution and use in source and binary forms, with or without
844990b8cSJulian Elischer  * modification, are permitted provided that the following conditions
944990b8cSJulian Elischer  * are met:
1044990b8cSJulian Elischer  * 1. Redistributions of source code must retain the above copyright
1144990b8cSJulian Elischer  *    notice(s), this list of conditions and the following disclaimer as
1244990b8cSJulian Elischer  *    the first lines of this file unmodified other than the possible
1344990b8cSJulian Elischer  *    addition of one or more copyright notices.
1444990b8cSJulian Elischer  * 2. Redistributions in binary form must reproduce the above copyright
1544990b8cSJulian Elischer  *    notice(s), this list of conditions and the following disclaimer in the
1644990b8cSJulian Elischer  *    documentation and/or other materials provided with the distribution.
1744990b8cSJulian Elischer  *
1844990b8cSJulian Elischer  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) ``AS IS'' AND ANY
1944990b8cSJulian Elischer  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
2044990b8cSJulian Elischer  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
2144990b8cSJulian Elischer  * DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) BE LIABLE FOR ANY
2244990b8cSJulian Elischer  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
2344990b8cSJulian Elischer  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
2444990b8cSJulian Elischer  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
2544990b8cSJulian Elischer  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
2644990b8cSJulian Elischer  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
2744990b8cSJulian Elischer  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
2844990b8cSJulian Elischer  * DAMAGE.
2944990b8cSJulian Elischer  */
3044990b8cSJulian Elischer 
313d06b4b3SAttilio Rao #include "opt_witness.h"
3216d95d4fSJoseph Koshy #include "opt_hwpmc_hooks.h"
333d06b4b3SAttilio Rao 
34677b542eSDavid E. O'Brien #include <sys/cdefs.h>
35677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
36677b542eSDavid E. O'Brien 
3744990b8cSJulian Elischer #include <sys/param.h>
3844990b8cSJulian Elischer #include <sys/systm.h>
3944990b8cSJulian Elischer #include <sys/kernel.h>
4044990b8cSJulian Elischer #include <sys/lock.h>
4144990b8cSJulian Elischer #include <sys/mutex.h>
4244990b8cSJulian Elischer #include <sys/proc.h>
4335bb59edSMateusz Guzik #include <sys/bitstring.h>
446febf180SGleb Smirnoff #include <sys/epoch.h>
458f0e9130SKonstantin Belousov #include <sys/rangelock.h>
46e170bfdaSDavid Xu #include <sys/resourcevar.h>
47b3e9e682SRyan Stone #include <sys/sdt.h>
4894e0a4cdSJulian Elischer #include <sys/smp.h>
49de028f5aSJeff Roberson #include <sys/sched.h>
5044f3b092SJohn Baldwin #include <sys/sleepqueue.h>
51ace8398dSJeff Roberson #include <sys/selinfo.h>
52d1e7a4a5SJohn Baldwin #include <sys/syscallsubr.h>
5391d1786fSDmitry Chagin #include <sys/sysent.h>
54961a7b24SJohn Baldwin #include <sys/turnstile.h>
5544990b8cSJulian Elischer #include <sys/ktr.h>
56cf7d9a8cSDavid Xu #include <sys/rwlock.h>
57bc8e6d81SDavid Xu #include <sys/umtx.h>
589ed01c32SGleb Smirnoff #include <sys/vmmeter.h>
59d7f687fcSJeff Roberson #include <sys/cpuset.h>
6016d95d4fSJoseph Koshy #ifdef	HWPMC_HOOKS
6116d95d4fSJoseph Koshy #include <sys/pmckern.h>
6216d95d4fSJoseph Koshy #endif
631bd3cf5dSMateusz Guzik #include <sys/priv.h>
6444990b8cSJulian Elischer 
65911b84b0SRobert Watson #include <security/audit/audit.h>
66911b84b0SRobert Watson 
6744990b8cSJulian Elischer #include <vm/vm.h>
6849a2507bSAlan Cox #include <vm/vm_extern.h>
6944990b8cSJulian Elischer #include <vm/uma.h>
70b209f889SRandall Stewart #include <sys/eventhandler.h>
7102fb42b0SPeter Wemm 
72acd9f517SKonstantin Belousov /*
73acd9f517SKonstantin Belousov  * Asserts below verify the stability of struct thread and struct proc
74acd9f517SKonstantin Belousov  * layout, as exposed by KBI to modules.  On head, the KBI is allowed
75acd9f517SKonstantin Belousov  * to drift, change to the structures must be accompanied by the
76acd9f517SKonstantin Belousov  * assert update.
77acd9f517SKonstantin Belousov  *
78acd9f517SKonstantin Belousov  * On the stable branches after KBI freeze, conditions must not be
79acd9f517SKonstantin Belousov  * violated.  Typically new fields are moved to the end of the
80acd9f517SKonstantin Belousov  * structures.
81acd9f517SKonstantin Belousov  */
82acd9f517SKonstantin Belousov #ifdef __amd64__
833f289c3fSJeff Roberson _Static_assert(offsetof(struct thread, td_flags) == 0xfc,
84acd9f517SKonstantin Belousov     "struct thread KBI td_flags");
853f289c3fSJeff Roberson _Static_assert(offsetof(struct thread, td_pflags) == 0x104,
86acd9f517SKonstantin Belousov     "struct thread KBI td_pflags");
871e2521ffSEdward Tomasz Napierala _Static_assert(offsetof(struct thread, td_frame) == 0x4a0,
88acd9f517SKonstantin Belousov     "struct thread KBI td_frame");
891724c563SMateusz Guzik _Static_assert(offsetof(struct thread, td_emuldata) == 0x6b0,
90acd9f517SKonstantin Belousov     "struct thread KBI td_emuldata");
91acd9f517SKonstantin Belousov _Static_assert(offsetof(struct proc, p_flag) == 0xb0,
92acd9f517SKonstantin Belousov     "struct proc KBI p_flag");
93acd9f517SKonstantin Belousov _Static_assert(offsetof(struct proc, p_pid) == 0xbc,
94acd9f517SKonstantin Belousov     "struct proc KBI p_pid");
958de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_filemon) == 0x3b8,
96acd9f517SKonstantin Belousov     "struct proc KBI p_filemon");
978de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_comm) == 0x3d0,
98acd9f517SKonstantin Belousov     "struct proc KBI p_comm");
998de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_emuldata) == 0x4b0,
100acd9f517SKonstantin Belousov     "struct proc KBI p_emuldata");
101acd9f517SKonstantin Belousov #endif
102acd9f517SKonstantin Belousov #ifdef __i386__
1033f289c3fSJeff Roberson _Static_assert(offsetof(struct thread, td_flags) == 0x98,
104acd9f517SKonstantin Belousov     "struct thread KBI td_flags");
1053f289c3fSJeff Roberson _Static_assert(offsetof(struct thread, td_pflags) == 0xa0,
106acd9f517SKonstantin Belousov     "struct thread KBI td_pflags");
1071e2521ffSEdward Tomasz Napierala _Static_assert(offsetof(struct thread, td_frame) == 0x300,
108acd9f517SKonstantin Belousov     "struct thread KBI td_frame");
1091e2521ffSEdward Tomasz Napierala _Static_assert(offsetof(struct thread, td_emuldata) == 0x344,
110acd9f517SKonstantin Belousov     "struct thread KBI td_emuldata");
111acd9f517SKonstantin Belousov _Static_assert(offsetof(struct proc, p_flag) == 0x68,
112acd9f517SKonstantin Belousov     "struct proc KBI p_flag");
113acd9f517SKonstantin Belousov _Static_assert(offsetof(struct proc, p_pid) == 0x74,
114acd9f517SKonstantin Belousov     "struct proc KBI p_pid");
1158de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_filemon) == 0x268,
116acd9f517SKonstantin Belousov     "struct proc KBI p_filemon");
1178de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_comm) == 0x27c,
118acd9f517SKonstantin Belousov     "struct proc KBI p_comm");
1198de97f39SRick Macklem _Static_assert(offsetof(struct proc, p_emuldata) == 0x308,
120acd9f517SKonstantin Belousov     "struct proc KBI p_emuldata");
121acd9f517SKonstantin Belousov #endif
122acd9f517SKonstantin Belousov 
123b3e9e682SRyan Stone SDT_PROVIDER_DECLARE(proc);
124d9fae5abSAndriy Gapon SDT_PROBE_DEFINE(proc, , , lwp__exit);
125b3e9e682SRyan Stone 
1268460a577SJohn Birrell /*
1278460a577SJohn Birrell  * thread related storage.
1288460a577SJohn Birrell  */
12944990b8cSJulian Elischer static uma_zone_t thread_zone;
13044990b8cSJulian Elischer 
131c5315f51SMateusz Guzik static __exclusive_cache_line struct thread *thread_zombies;
13244990b8cSJulian Elischer 
133ff8fbcffSJeff Roberson static void thread_zombie(struct thread *);
13484cdea97SKonstantin Belousov static int thread_unsuspend_one(struct thread *td, struct proc *p,
13584cdea97SKonstantin Belousov     bool boundary);
136755341dfSMateusz Guzik static void thread_free_batched(struct thread *td);
137ff8fbcffSJeff Roberson 
138*d1ca25beSMateusz Guzik static __exclusive_cache_line struct mtx tid_lock;
139934e7e5eSMateusz Guzik static bitstr_t *tid_bitmap;
14035bb59edSMateusz Guzik 
141cf7d9a8cSDavid Xu static MALLOC_DEFINE(M_TIDHASH, "tidhash", "thread hash");
142cf7d9a8cSDavid Xu 
1431bd3cf5dSMateusz Guzik static int maxthread;
1441bd3cf5dSMateusz Guzik SYSCTL_INT(_kern, OID_AUTO, maxthread, CTLFLAG_RDTUN,
1451bd3cf5dSMateusz Guzik     &maxthread, 0, "Maximum number of threads");
1461bd3cf5dSMateusz Guzik 
14762dbc992SMateusz Guzik static __exclusive_cache_line int nthreads;
1481bd3cf5dSMateusz Guzik 
149aae3547bSMateusz Guzik static LIST_HEAD(tidhashhead, thread) *tidhashtbl;
150aae3547bSMateusz Guzik static u_long	tidhash;
15126007fe3SMateusz Guzik static u_long	tidhashlock;
15226007fe3SMateusz Guzik static struct	rwlock *tidhashtbl_lock;
153aae3547bSMateusz Guzik #define	TIDHASH(tid)		(&tidhashtbl[(tid) & tidhash])
15426007fe3SMateusz Guzik #define	TIDHASHLOCK(tid)	(&tidhashtbl_lock[(tid) & tidhashlock])
155cf7d9a8cSDavid Xu 
1562ca45184SMatt Joras EVENTHANDLER_LIST_DEFINE(thread_ctor);
1572ca45184SMatt Joras EVENTHANDLER_LIST_DEFINE(thread_dtor);
1582ca45184SMatt Joras EVENTHANDLER_LIST_DEFINE(thread_init);
1592ca45184SMatt Joras EVENTHANDLER_LIST_DEFINE(thread_fini);
1602ca45184SMatt Joras 
16162dbc992SMateusz Guzik static bool
16262dbc992SMateusz Guzik thread_count_inc(void)
163ec6ea5e8SDavid Xu {
1641bd3cf5dSMateusz Guzik 	static struct timeval lastfail;
1651bd3cf5dSMateusz Guzik 	static int curfail;
16662dbc992SMateusz Guzik 	int nthreads_new;
167ec6ea5e8SDavid Xu 
16862dbc992SMateusz Guzik 	thread_reap();
16962dbc992SMateusz Guzik 
17062dbc992SMateusz Guzik 	nthreads_new = atomic_fetchadd_int(&nthreads, 1) + 1;
17162dbc992SMateusz Guzik 	if (nthreads_new >= maxthread - 100) {
1721bd3cf5dSMateusz Guzik 		if (priv_check_cred(curthread->td_ucred, PRIV_MAXPROC) != 0 ||
17362dbc992SMateusz Guzik 		    nthreads_new >= maxthread) {
17462dbc992SMateusz Guzik 			atomic_subtract_int(&nthreads, 1);
1751bd3cf5dSMateusz Guzik 			if (ppsratecheck(&lastfail, &curfail, 1)) {
1761bd3cf5dSMateusz Guzik 				printf("maxthread limit exceeded by uid %u "
1771bd3cf5dSMateusz Guzik 				"(pid %d); consider increasing kern.maxthread\n",
1781bd3cf5dSMateusz Guzik 				curthread->td_ucred->cr_ruid, curproc->p_pid);
1791bd3cf5dSMateusz Guzik 			}
18062dbc992SMateusz Guzik 			return (false);
1811bd3cf5dSMateusz Guzik 		}
1821bd3cf5dSMateusz Guzik 	}
18362dbc992SMateusz Guzik 	return (true);
18462dbc992SMateusz Guzik }
1851bd3cf5dSMateusz Guzik 
18662dbc992SMateusz Guzik static void
18762dbc992SMateusz Guzik thread_count_sub(int n)
18862dbc992SMateusz Guzik {
18962dbc992SMateusz Guzik 
19062dbc992SMateusz Guzik 	atomic_subtract_int(&nthreads, n);
19162dbc992SMateusz Guzik }
19262dbc992SMateusz Guzik 
19362dbc992SMateusz Guzik static void
19462dbc992SMateusz Guzik thread_count_dec(void)
19562dbc992SMateusz Guzik {
19662dbc992SMateusz Guzik 
19762dbc992SMateusz Guzik 	thread_count_sub(1);
19862dbc992SMateusz Guzik }
19962dbc992SMateusz Guzik 
20062dbc992SMateusz Guzik static lwpid_t
20162dbc992SMateusz Guzik tid_alloc(void)
20262dbc992SMateusz Guzik {
20362dbc992SMateusz Guzik 	static lwpid_t trytid;
20462dbc992SMateusz Guzik 	lwpid_t tid;
20562dbc992SMateusz Guzik 
20662dbc992SMateusz Guzik 	mtx_lock(&tid_lock);
20735bb59edSMateusz Guzik 	/*
20835bb59edSMateusz Guzik 	 * It is an invariant that the bitmap is big enough to hold maxthread
20935bb59edSMateusz Guzik 	 * IDs. If we got to this point there has to be at least one free.
21035bb59edSMateusz Guzik 	 */
21135bb59edSMateusz Guzik 	if (trytid >= maxthread)
21235bb59edSMateusz Guzik 		trytid = 0;
21335bb59edSMateusz Guzik 	bit_ffc_at(tid_bitmap, trytid, maxthread, &tid);
21435bb59edSMateusz Guzik 	if (tid == -1) {
21535bb59edSMateusz Guzik 		KASSERT(trytid != 0, ("unexpectedly ran out of IDs"));
21635bb59edSMateusz Guzik 		trytid = 0;
21735bb59edSMateusz Guzik 		bit_ffc_at(tid_bitmap, trytid, maxthread, &tid);
21835bb59edSMateusz Guzik 		KASSERT(tid != -1, ("unexpectedly ran out of IDs"));
219ec6ea5e8SDavid Xu 	}
22035bb59edSMateusz Guzik 	bit_set(tid_bitmap, tid);
221934e7e5eSMateusz Guzik 	trytid = tid + 1;
222ec6ea5e8SDavid Xu 	mtx_unlock(&tid_lock);
22335bb59edSMateusz Guzik 	return (tid + NO_PID);
224ec6ea5e8SDavid Xu }
225ec6ea5e8SDavid Xu 
226ec6ea5e8SDavid Xu static void
227755341dfSMateusz Guzik tid_free_locked(lwpid_t rtid)
228ec6ea5e8SDavid Xu {
22935bb59edSMateusz Guzik 	lwpid_t tid;
230ec6ea5e8SDavid Xu 
231755341dfSMateusz Guzik 	mtx_assert(&tid_lock, MA_OWNED);
23235bb59edSMateusz Guzik 	KASSERT(rtid >= NO_PID,
23335bb59edSMateusz Guzik 	    ("%s: invalid tid %d\n", __func__, rtid));
23435bb59edSMateusz Guzik 	tid = rtid - NO_PID;
23535bb59edSMateusz Guzik 	KASSERT(bit_test(tid_bitmap, tid) != 0,
23635bb59edSMateusz Guzik 	    ("thread ID %d not allocated\n", rtid));
23735bb59edSMateusz Guzik 	bit_clear(tid_bitmap, tid);
238755341dfSMateusz Guzik }
239755341dfSMateusz Guzik 
240755341dfSMateusz Guzik static void
241755341dfSMateusz Guzik tid_free(lwpid_t rtid)
242755341dfSMateusz Guzik {
243755341dfSMateusz Guzik 
244755341dfSMateusz Guzik 	mtx_lock(&tid_lock);
245755341dfSMateusz Guzik 	tid_free_locked(rtid);
246755341dfSMateusz Guzik 	mtx_unlock(&tid_lock);
247755341dfSMateusz Guzik }
248755341dfSMateusz Guzik 
249755341dfSMateusz Guzik static void
250755341dfSMateusz Guzik tid_free_batch(lwpid_t *batch, int n)
251755341dfSMateusz Guzik {
252755341dfSMateusz Guzik 	int i;
253755341dfSMateusz Guzik 
254755341dfSMateusz Guzik 	mtx_lock(&tid_lock);
255755341dfSMateusz Guzik 	for (i = 0; i < n; i++) {
256755341dfSMateusz Guzik 		tid_free_locked(batch[i]);
257755341dfSMateusz Guzik 	}
258ec6ea5e8SDavid Xu 	mtx_unlock(&tid_lock);
259ec6ea5e8SDavid Xu }
260ec6ea5e8SDavid Xu 
261fdcac928SMarcel Moolenaar /*
262696058c3SJulian Elischer  * Prepare a thread for use.
26344990b8cSJulian Elischer  */
264b23f72e9SBrian Feldman static int
265b23f72e9SBrian Feldman thread_ctor(void *mem, int size, void *arg, int flags)
26644990b8cSJulian Elischer {
26744990b8cSJulian Elischer 	struct thread	*td;
26844990b8cSJulian Elischer 
26944990b8cSJulian Elischer 	td = (struct thread *)mem;
27071fad9fdSJulian Elischer 	td->td_state = TDS_INACTIVE;
27194dd54b9SKonstantin Belousov 	td->td_lastcpu = td->td_oncpu = NOCPU;
2726c27c603SJuli Mallett 
2736c27c603SJuli Mallett 	/*
2746c27c603SJuli Mallett 	 * Note that td_critnest begins life as 1 because the thread is not
2756c27c603SJuli Mallett 	 * running and is thereby implicitly waiting to be on the receiving
276a54e85fdSJeff Roberson 	 * end of a context switch.
2776c27c603SJuli Mallett 	 */
278139b7550SJohn Baldwin 	td->td_critnest = 1;
279acbe332aSDavid Xu 	td->td_lend_user_pri = PRI_MAX;
280911b84b0SRobert Watson #ifdef AUDIT
281911b84b0SRobert Watson 	audit_thread_alloc(td);
282911b84b0SRobert Watson #endif
283d10183d9SDavid Xu 	umtx_thread_alloc(td);
284b23f72e9SBrian Feldman 	return (0);
28544990b8cSJulian Elischer }
28644990b8cSJulian Elischer 
28744990b8cSJulian Elischer /*
28844990b8cSJulian Elischer  * Reclaim a thread after use.
28944990b8cSJulian Elischer  */
29044990b8cSJulian Elischer static void
29144990b8cSJulian Elischer thread_dtor(void *mem, int size, void *arg)
29244990b8cSJulian Elischer {
29344990b8cSJulian Elischer 	struct thread *td;
29444990b8cSJulian Elischer 
29544990b8cSJulian Elischer 	td = (struct thread *)mem;
29644990b8cSJulian Elischer 
29744990b8cSJulian Elischer #ifdef INVARIANTS
29844990b8cSJulian Elischer 	/* Verify that this thread is in a safe state to free. */
29944990b8cSJulian Elischer 	switch (td->td_state) {
30071fad9fdSJulian Elischer 	case TDS_INHIBITED:
30171fad9fdSJulian Elischer 	case TDS_RUNNING:
30271fad9fdSJulian Elischer 	case TDS_CAN_RUN:
30344990b8cSJulian Elischer 	case TDS_RUNQ:
30444990b8cSJulian Elischer 		/*
30544990b8cSJulian Elischer 		 * We must never unlink a thread that is in one of
30644990b8cSJulian Elischer 		 * these states, because it is currently active.
30744990b8cSJulian Elischer 		 */
30844990b8cSJulian Elischer 		panic("bad state for thread unlinking");
30944990b8cSJulian Elischer 		/* NOTREACHED */
31071fad9fdSJulian Elischer 	case TDS_INACTIVE:
31144990b8cSJulian Elischer 		break;
31244990b8cSJulian Elischer 	default:
31344990b8cSJulian Elischer 		panic("bad thread state");
31444990b8cSJulian Elischer 		/* NOTREACHED */
31544990b8cSJulian Elischer 	}
31644990b8cSJulian Elischer #endif
3176e8525ceSRobert Watson #ifdef AUDIT
3186e8525ceSRobert Watson 	audit_thread_free(td);
3196e8525ceSRobert Watson #endif
3201ba4a712SPawel Jakub Dawidek 	/* Free all OSD associated to this thread. */
3211ba4a712SPawel Jakub Dawidek 	osd_thread_exit(td);
322aca4bb91SKonstantin Belousov 	td_softdep_cleanup(td);
323aca4bb91SKonstantin Belousov 	MPASS(td->td_su == NULL);
32444990b8cSJulian Elischer }
32544990b8cSJulian Elischer 
32644990b8cSJulian Elischer /*
32744990b8cSJulian Elischer  * Initialize type-stable parts of a thread (when newly created).
32844990b8cSJulian Elischer  */
329b23f72e9SBrian Feldman static int
330b23f72e9SBrian Feldman thread_init(void *mem, int size, int flags)
33144990b8cSJulian Elischer {
33244990b8cSJulian Elischer 	struct thread *td;
33344990b8cSJulian Elischer 
33444990b8cSJulian Elischer 	td = (struct thread *)mem;
335247aba24SMarcel Moolenaar 
33644f3b092SJohn Baldwin 	td->td_sleepqueue = sleepq_alloc();
337961a7b24SJohn Baldwin 	td->td_turnstile = turnstile_alloc();
3388f0e9130SKonstantin Belousov 	td->td_rlqe = NULL;
3392ca45184SMatt Joras 	EVENTHANDLER_DIRECT_INVOKE(thread_init, td);
340d10183d9SDavid Xu 	umtx_thread_init(td);
34189b57fcfSKonstantin Belousov 	td->td_kstack = 0;
342ad8b1d85SKonstantin Belousov 	td->td_sel = NULL;
343b23f72e9SBrian Feldman 	return (0);
34444990b8cSJulian Elischer }
34544990b8cSJulian Elischer 
34644990b8cSJulian Elischer /*
34744990b8cSJulian Elischer  * Tear down type-stable parts of a thread (just before being discarded).
34844990b8cSJulian Elischer  */
34944990b8cSJulian Elischer static void
35044990b8cSJulian Elischer thread_fini(void *mem, int size)
35144990b8cSJulian Elischer {
35244990b8cSJulian Elischer 	struct thread *td;
35344990b8cSJulian Elischer 
35444990b8cSJulian Elischer 	td = (struct thread *)mem;
3552ca45184SMatt Joras 	EVENTHANDLER_DIRECT_INVOKE(thread_fini, td);
3568f0e9130SKonstantin Belousov 	rlqentry_free(td->td_rlqe);
357961a7b24SJohn Baldwin 	turnstile_free(td->td_turnstile);
35844f3b092SJohn Baldwin 	sleepq_free(td->td_sleepqueue);
359d10183d9SDavid Xu 	umtx_thread_fini(td);
360ace8398dSJeff Roberson 	seltdfini(td);
36144990b8cSJulian Elischer }
3625215b187SJeff Roberson 
3635c8329edSJulian Elischer /*
3645215b187SJeff Roberson  * For a newly created process,
3655215b187SJeff Roberson  * link up all the structures and its initial threads etc.
366ed062c8dSJulian Elischer  * called from:
367e7d939bdSMarcel Moolenaar  * {arch}/{arch}/machdep.c   {arch}_init(), init386() etc.
368ed062c8dSJulian Elischer  * proc_dtor() (should go away)
369ed062c8dSJulian Elischer  * proc_init()
3705c8329edSJulian Elischer  */
3715c8329edSJulian Elischer void
37289b57fcfSKonstantin Belousov proc_linkup0(struct proc *p, struct thread *td)
37389b57fcfSKonstantin Belousov {
37489b57fcfSKonstantin Belousov 	TAILQ_INIT(&p->p_threads);	     /* all threads in proc */
37589b57fcfSKonstantin Belousov 	proc_linkup(p, td);
37689b57fcfSKonstantin Belousov }
37789b57fcfSKonstantin Belousov 
37889b57fcfSKonstantin Belousov void
3798460a577SJohn Birrell proc_linkup(struct proc *p, struct thread *td)
3805c8329edSJulian Elischer {
381a54e85fdSJeff Roberson 
3829104847fSDavid Xu 	sigqueue_init(&p->p_sigqueue, p);
383ebceaf6dSDavid Xu 	p->p_ksi = ksiginfo_alloc(1);
384ebceaf6dSDavid Xu 	if (p->p_ksi != NULL) {
3855c474517SDavid Xu 		/* XXX p_ksi may be null if ksiginfo zone is not ready */
386ebceaf6dSDavid Xu 		p->p_ksi->ksi_flags = KSI_EXT | KSI_INS;
387ebceaf6dSDavid Xu 	}
388b2f92ef9SDavid Xu 	LIST_INIT(&p->p_mqnotifier);
3895c8329edSJulian Elischer 	p->p_numthreads = 0;
3908460a577SJohn Birrell 	thread_link(td, p);
3915c8329edSJulian Elischer }
3925c8329edSJulian Elischer 
3931bd3cf5dSMateusz Guzik extern int max_threads_per_proc;
3941bd3cf5dSMateusz Guzik 
3955c8329edSJulian Elischer /*
39644990b8cSJulian Elischer  * Initialize global thread allocation resources.
39744990b8cSJulian Elischer  */
39844990b8cSJulian Elischer void
39944990b8cSJulian Elischer threadinit(void)
40044990b8cSJulian Elischer {
40126007fe3SMateusz Guzik 	u_long i;
402cf31cadeSMateusz Guzik 	lwpid_t tid0;
4035aa5420fSMark Johnston 	uint32_t flags;
40444990b8cSJulian Elischer 
4051bd3cf5dSMateusz Guzik 	/*
4061bd3cf5dSMateusz Guzik 	 * Place an upper limit on threads which can be allocated.
4071bd3cf5dSMateusz Guzik 	 *
4081bd3cf5dSMateusz Guzik 	 * Note that other factors may make the de facto limit much lower.
4091bd3cf5dSMateusz Guzik 	 *
4101bd3cf5dSMateusz Guzik 	 * Platform limits are somewhat arbitrary but deemed "more than good
4111bd3cf5dSMateusz Guzik 	 * enough" for the foreseable future.
4121bd3cf5dSMateusz Guzik 	 */
4131bd3cf5dSMateusz Guzik 	if (maxthread == 0) {
4141bd3cf5dSMateusz Guzik #ifdef _LP64
4151bd3cf5dSMateusz Guzik 		maxthread = MIN(maxproc * max_threads_per_proc, 1000000);
4161bd3cf5dSMateusz Guzik #else
4171bd3cf5dSMateusz Guzik 		maxthread = MIN(maxproc * max_threads_per_proc, 100000);
4181bd3cf5dSMateusz Guzik #endif
4191bd3cf5dSMateusz Guzik 	}
4201bd3cf5dSMateusz Guzik 
4211ea7a6f8SPoul-Henning Kamp 	mtx_init(&tid_lock, "TID lock", NULL, MTX_DEF);
42235bb59edSMateusz Guzik 	tid_bitmap = bit_alloc(maxthread, M_TIDHASH, M_WAITOK);
42362dbc992SMateusz Guzik 	/*
42462dbc992SMateusz Guzik 	 * Handle thread0.
42562dbc992SMateusz Guzik 	 */
42662dbc992SMateusz Guzik 	thread_count_inc();
427cf31cadeSMateusz Guzik 	tid0 = tid_alloc();
428cf31cadeSMateusz Guzik 	if (tid0 != THREAD0_TID)
429cf31cadeSMateusz Guzik 		panic("tid0 %d != %d\n", tid0, THREAD0_TID);
4301ea7a6f8SPoul-Henning Kamp 
4315aa5420fSMark Johnston 	flags = UMA_ZONE_NOFREE;
4325aa5420fSMark Johnston #ifdef __aarch64__
4335aa5420fSMark Johnston 	/*
4345aa5420fSMark Johnston 	 * Force thread structures to be allocated from the direct map.
4355aa5420fSMark Johnston 	 * Otherwise, superpage promotions and demotions may temporarily
4365aa5420fSMark Johnston 	 * invalidate thread structure mappings.  For most dynamically allocated
4375aa5420fSMark Johnston 	 * structures this is not a problem, but translation faults cannot be
4385aa5420fSMark Johnston 	 * handled without accessing curthread.
4395aa5420fSMark Johnston 	 */
4405aa5420fSMark Johnston 	flags |= UMA_ZONE_CONTIG;
4415aa5420fSMark Johnston #endif
442de028f5aSJeff Roberson 	thread_zone = uma_zcreate("THREAD", sched_sizeof_thread(),
44344990b8cSJulian Elischer 	    thread_ctor, thread_dtor, thread_init, thread_fini,
4445aa5420fSMark Johnston 	    32 - 1, flags);
445cf7d9a8cSDavid Xu 	tidhashtbl = hashinit(maxproc / 2, M_TIDHASH, &tidhash);
44626007fe3SMateusz Guzik 	tidhashlock = (tidhash + 1) / 64;
44726007fe3SMateusz Guzik 	if (tidhashlock > 0)
44826007fe3SMateusz Guzik 		tidhashlock--;
44926007fe3SMateusz Guzik 	tidhashtbl_lock = malloc(sizeof(*tidhashtbl_lock) * (tidhashlock + 1),
45026007fe3SMateusz Guzik 	    M_TIDHASH, M_WAITOK | M_ZERO);
45126007fe3SMateusz Guzik 	for (i = 0; i < tidhashlock + 1; i++)
45226007fe3SMateusz Guzik 		rw_init(&tidhashtbl_lock[i], "tidhash");
45344990b8cSJulian Elischer }
45444990b8cSJulian Elischer 
45544990b8cSJulian Elischer /*
456ff8fbcffSJeff Roberson  * Place an unused thread on the zombie list.
45744990b8cSJulian Elischer  */
45844990b8cSJulian Elischer void
459ff8fbcffSJeff Roberson thread_zombie(struct thread *td)
46044990b8cSJulian Elischer {
461c5315f51SMateusz Guzik 	struct thread *ztd;
462c5315f51SMateusz Guzik 
463c5315f51SMateusz Guzik 	ztd = atomic_load_ptr(&thread_zombies);
464c5315f51SMateusz Guzik 	for (;;) {
465c5315f51SMateusz Guzik 		td->td_zombie = ztd;
466c5315f51SMateusz Guzik 		if (atomic_fcmpset_rel_ptr((uintptr_t *)&thread_zombies,
467c5315f51SMateusz Guzik 		    (uintptr_t *)&ztd, (uintptr_t)td))
468c5315f51SMateusz Guzik 			break;
469c5315f51SMateusz Guzik 		continue;
470c5315f51SMateusz Guzik 	}
47144990b8cSJulian Elischer }
47244990b8cSJulian Elischer 
4735c8329edSJulian Elischer /*
474ff8fbcffSJeff Roberson  * Release a thread that has exited after cpu_throw().
475ff8fbcffSJeff Roberson  */
476ff8fbcffSJeff Roberson void
477ff8fbcffSJeff Roberson thread_stash(struct thread *td)
478ff8fbcffSJeff Roberson {
479ff8fbcffSJeff Roberson 	atomic_subtract_rel_int(&td->td_proc->p_exitthreads, 1);
480ff8fbcffSJeff Roberson 	thread_zombie(td);
481ff8fbcffSJeff Roberson }
482ff8fbcffSJeff Roberson 
483ff8fbcffSJeff Roberson /*
484c5315f51SMateusz Guzik  * Reap zombie threads.
48544990b8cSJulian Elischer  */
48644990b8cSJulian Elischer void
48744990b8cSJulian Elischer thread_reap(void)
48844990b8cSJulian Elischer {
489c5315f51SMateusz Guzik 	struct thread *itd, *ntd;
490755341dfSMateusz Guzik 	lwpid_t tidbatch[16];
491755341dfSMateusz Guzik 	int tidbatchn;
49244990b8cSJulian Elischer 
49344990b8cSJulian Elischer 	/*
494c5315f51SMateusz Guzik 	 * Reading upfront is pessimal if followed by concurrent atomic_swap,
495c5315f51SMateusz Guzik 	 * but most of the time the list is empty.
49644990b8cSJulian Elischer 	 */
497c5315f51SMateusz Guzik 	if (thread_zombies == NULL)
498c5315f51SMateusz Guzik 		return;
499c5315f51SMateusz Guzik 
500c5315f51SMateusz Guzik 	itd = (struct thread *)atomic_swap_ptr((uintptr_t *)&thread_zombies,
501c5315f51SMateusz Guzik 	    (uintptr_t)NULL);
502755341dfSMateusz Guzik 	tidbatchn = 0;
503c5315f51SMateusz Guzik 	while (itd != NULL) {
504c5315f51SMateusz Guzik 		ntd = itd->td_zombie;
505755341dfSMateusz Guzik 		tidbatch[tidbatchn] = itd->td_tid;
506755341dfSMateusz Guzik 		tidbatchn++;
507c5315f51SMateusz Guzik 		thread_cow_free(itd);
508755341dfSMateusz Guzik 		thread_free_batched(itd);
509755341dfSMateusz Guzik 		if (tidbatchn == nitems(tidbatch)) {
510755341dfSMateusz Guzik 			tid_free_batch(tidbatch, tidbatchn);
51162dbc992SMateusz Guzik 			thread_count_sub(tidbatchn);
512755341dfSMateusz Guzik 			tidbatchn = 0;
513755341dfSMateusz Guzik 		}
514c5315f51SMateusz Guzik 		itd = ntd;
51544990b8cSJulian Elischer 	}
516755341dfSMateusz Guzik 
517755341dfSMateusz Guzik 	if (tidbatchn != 0) {
518755341dfSMateusz Guzik 		tid_free_batch(tidbatch, tidbatchn);
51962dbc992SMateusz Guzik 		thread_count_sub(tidbatchn);
520755341dfSMateusz Guzik 	}
521ed062c8dSJulian Elischer }
52244990b8cSJulian Elischer 
5234f0db5e0SJulian Elischer /*
52444990b8cSJulian Elischer  * Allocate a thread.
52544990b8cSJulian Elischer  */
52644990b8cSJulian Elischer struct thread *
5278a945d10SKonstantin Belousov thread_alloc(int pages)
52844990b8cSJulian Elischer {
52989b57fcfSKonstantin Belousov 	struct thread *td;
5301bd3cf5dSMateusz Guzik 	lwpid_t tid;
5318460a577SJohn Birrell 
53262dbc992SMateusz Guzik 	if (!thread_count_inc()) {
5331bd3cf5dSMateusz Guzik 		return (NULL);
5341bd3cf5dSMateusz Guzik 	}
5351bd3cf5dSMateusz Guzik 
53662dbc992SMateusz Guzik 	tid = tid_alloc();
5371bd3cf5dSMateusz Guzik 	td = uma_zalloc(thread_zone, M_WAITOK);
53889b57fcfSKonstantin Belousov 	KASSERT(td->td_kstack == 0, ("thread_alloc got thread with kstack"));
5398a945d10SKonstantin Belousov 	if (!vm_thread_new(td, pages)) {
54089b57fcfSKonstantin Belousov 		uma_zfree(thread_zone, td);
5411bd3cf5dSMateusz Guzik 		tid_free(tid);
54262dbc992SMateusz Guzik 		thread_count_dec();
54389b57fcfSKonstantin Belousov 		return (NULL);
54489b57fcfSKonstantin Belousov 	}
5451bd3cf5dSMateusz Guzik 	td->td_tid = tid;
5460c3967e7SMarcel Moolenaar 	cpu_thread_alloc(td);
5471bd3cf5dSMateusz Guzik 	EVENTHANDLER_DIRECT_INVOKE(thread_ctor, td);
54889b57fcfSKonstantin Belousov 	return (td);
54944990b8cSJulian Elischer }
55044990b8cSJulian Elischer 
5518a945d10SKonstantin Belousov int
5528a945d10SKonstantin Belousov thread_alloc_stack(struct thread *td, int pages)
5538a945d10SKonstantin Belousov {
5548a945d10SKonstantin Belousov 
5558a945d10SKonstantin Belousov 	KASSERT(td->td_kstack == 0,
5568a945d10SKonstantin Belousov 	    ("thread_alloc_stack called on a thread with kstack"));
5578a945d10SKonstantin Belousov 	if (!vm_thread_new(td, pages))
5588a945d10SKonstantin Belousov 		return (0);
5598a945d10SKonstantin Belousov 	cpu_thread_alloc(td);
5608a945d10SKonstantin Belousov 	return (1);
5618a945d10SKonstantin Belousov }
5624f0db5e0SJulian Elischer 
5634f0db5e0SJulian Elischer /*
56444990b8cSJulian Elischer  * Deallocate a thread.
56544990b8cSJulian Elischer  */
566755341dfSMateusz Guzik static void
567755341dfSMateusz Guzik thread_free_batched(struct thread *td)
56844990b8cSJulian Elischer {
5692e6b8de4SJeff Roberson 
5701bd3cf5dSMateusz Guzik 	EVENTHANDLER_DIRECT_INVOKE(thread_dtor, td);
5712e6b8de4SJeff Roberson 	lock_profile_thread_exit(td);
57245aea8deSJeff Roberson 	if (td->td_cpuset)
573d7f687fcSJeff Roberson 		cpuset_rel(td->td_cpuset);
574d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
5750c3967e7SMarcel Moolenaar 	cpu_thread_free(td);
57689b57fcfSKonstantin Belousov 	if (td->td_kstack != 0)
57789b57fcfSKonstantin Belousov 		vm_thread_dispose(td);
5782d19b736SKonstantin Belousov 	callout_drain(&td->td_slpcallout);
579755341dfSMateusz Guzik 	/*
580755341dfSMateusz Guzik 	 * Freeing handled by the caller.
581755341dfSMateusz Guzik 	 */
5821bd3cf5dSMateusz Guzik 	td->td_tid = -1;
58344990b8cSJulian Elischer 	uma_zfree(thread_zone, td);
58444990b8cSJulian Elischer }
58544990b8cSJulian Elischer 
5864ea6a9a2SMateusz Guzik void
587755341dfSMateusz Guzik thread_free(struct thread *td)
588755341dfSMateusz Guzik {
589755341dfSMateusz Guzik 	lwpid_t tid;
590755341dfSMateusz Guzik 
591755341dfSMateusz Guzik 	tid = td->td_tid;
592755341dfSMateusz Guzik 	thread_free_batched(td);
593755341dfSMateusz Guzik 	tid_free(tid);
59462dbc992SMateusz Guzik 	thread_count_dec();
595755341dfSMateusz Guzik }
596755341dfSMateusz Guzik 
597755341dfSMateusz Guzik void
5984ea6a9a2SMateusz Guzik thread_cow_get_proc(struct thread *newtd, struct proc *p)
5994ea6a9a2SMateusz Guzik {
6004ea6a9a2SMateusz Guzik 
6014ea6a9a2SMateusz Guzik 	PROC_LOCK_ASSERT(p, MA_OWNED);
6021724c563SMateusz Guzik 	newtd->td_realucred = crcowget(p->p_ucred);
6031724c563SMateusz Guzik 	newtd->td_ucred = newtd->td_realucred;
604f6f6d240SMateusz Guzik 	newtd->td_limit = lim_hold(p->p_limit);
6054ea6a9a2SMateusz Guzik 	newtd->td_cowgen = p->p_cowgen;
6064ea6a9a2SMateusz Guzik }
6074ea6a9a2SMateusz Guzik 
6084ea6a9a2SMateusz Guzik void
6094ea6a9a2SMateusz Guzik thread_cow_get(struct thread *newtd, struct thread *td)
6104ea6a9a2SMateusz Guzik {
6114ea6a9a2SMateusz Guzik 
6121724c563SMateusz Guzik 	MPASS(td->td_realucred == td->td_ucred);
6131724c563SMateusz Guzik 	newtd->td_realucred = crcowget(td->td_realucred);
6141724c563SMateusz Guzik 	newtd->td_ucred = newtd->td_realucred;
615f6f6d240SMateusz Guzik 	newtd->td_limit = lim_hold(td->td_limit);
6164ea6a9a2SMateusz Guzik 	newtd->td_cowgen = td->td_cowgen;
6174ea6a9a2SMateusz Guzik }
6184ea6a9a2SMateusz Guzik 
6194ea6a9a2SMateusz Guzik void
6204ea6a9a2SMateusz Guzik thread_cow_free(struct thread *td)
6214ea6a9a2SMateusz Guzik {
6224ea6a9a2SMateusz Guzik 
6231724c563SMateusz Guzik 	if (td->td_realucred != NULL)
6241724c563SMateusz Guzik 		crcowfree(td);
625cd672ca6SMateusz Guzik 	if (td->td_limit != NULL)
626f6f6d240SMateusz Guzik 		lim_free(td->td_limit);
6274ea6a9a2SMateusz Guzik }
6284ea6a9a2SMateusz Guzik 
6294ea6a9a2SMateusz Guzik void
6304ea6a9a2SMateusz Guzik thread_cow_update(struct thread *td)
6314ea6a9a2SMateusz Guzik {
6324ea6a9a2SMateusz Guzik 	struct proc *p;
633cd672ca6SMateusz Guzik 	struct ucred *oldcred;
634cd672ca6SMateusz Guzik 	struct plimit *oldlimit;
6354ea6a9a2SMateusz Guzik 
6364ea6a9a2SMateusz Guzik 	p = td->td_proc;
637cd672ca6SMateusz Guzik 	oldlimit = NULL;
6384ea6a9a2SMateusz Guzik 	PROC_LOCK(p);
6391724c563SMateusz Guzik 	oldcred = crcowsync();
640cd672ca6SMateusz Guzik 	if (td->td_limit != p->p_limit) {
641cd672ca6SMateusz Guzik 		oldlimit = td->td_limit;
642cd672ca6SMateusz Guzik 		td->td_limit = lim_hold(p->p_limit);
643cd672ca6SMateusz Guzik 	}
6444ea6a9a2SMateusz Guzik 	td->td_cowgen = p->p_cowgen;
6454ea6a9a2SMateusz Guzik 	PROC_UNLOCK(p);
646cd672ca6SMateusz Guzik 	if (oldcred != NULL)
647cd672ca6SMateusz Guzik 		crfree(oldcred);
648cd672ca6SMateusz Guzik 	if (oldlimit != NULL)
649cd672ca6SMateusz Guzik 		lim_free(oldlimit);
6504ea6a9a2SMateusz Guzik }
6514ea6a9a2SMateusz Guzik 
65244990b8cSJulian Elischer /*
65344990b8cSJulian Elischer  * Discard the current thread and exit from its context.
65494e0a4cdSJulian Elischer  * Always called with scheduler locked.
65544990b8cSJulian Elischer  *
65644990b8cSJulian Elischer  * Because we can't free a thread while we're operating under its context,
657696058c3SJulian Elischer  * push the current thread into our CPU's deadthread holder. This means
658696058c3SJulian Elischer  * we needn't worry about someone else grabbing our context before we
6596617724cSJeff Roberson  * do a cpu_throw().
66044990b8cSJulian Elischer  */
66144990b8cSJulian Elischer void
66244990b8cSJulian Elischer thread_exit(void)
66344990b8cSJulian Elischer {
6647e3a96eaSJohn Baldwin 	uint64_t runtime, new_switchtime;
66544990b8cSJulian Elischer 	struct thread *td;
6661c4bcd05SJeff Roberson 	struct thread *td2;
66744990b8cSJulian Elischer 	struct proc *p;
6687847a9daSJohn Baldwin 	int wakeup_swapper;
66944990b8cSJulian Elischer 
67044990b8cSJulian Elischer 	td = curthread;
67144990b8cSJulian Elischer 	p = td->td_proc;
67244990b8cSJulian Elischer 
673a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
674ed062c8dSJulian Elischer 	mtx_assert(&Giant, MA_NOTOWNED);
675a54e85fdSJeff Roberson 
67644990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
677ed062c8dSJulian Elischer 	KASSERT(p != NULL, ("thread exiting without a process"));
678cc701b73SRobert Watson 	CTR3(KTR_PROC, "thread_exit: thread %p (pid %ld, %s)", td,
679e01eafefSJulian Elischer 	    (long)p->p_pid, td->td_name);
6806c9271a9SAndriy Gapon 	SDT_PROBE0(proc, , , lwp__exit);
6819104847fSDavid Xu 	KASSERT(TAILQ_EMPTY(&td->td_sigqueue.sq_list), ("signal pending"));
682936c24faSMateusz Guzik 	MPASS(td->td_realucred == td->td_ucred);
68344990b8cSJulian Elischer 
684ed062c8dSJulian Elischer 	/*
685ed062c8dSJulian Elischer 	 * drop FPU & debug register state storage, or any other
686ed062c8dSJulian Elischer 	 * architecture specific resources that
687ed062c8dSJulian Elischer 	 * would not be on a new untouched process.
688ed062c8dSJulian Elischer 	 */
689bd07998eSKonstantin Belousov 	cpu_thread_exit(td);
69044990b8cSJulian Elischer 
691ed062c8dSJulian Elischer 	/*
6921faf202eSJulian Elischer 	 * The last thread is left attached to the process
6931faf202eSJulian Elischer 	 * So that the whole bundle gets recycled. Skip
694ed062c8dSJulian Elischer 	 * all this stuff if we never had threads.
695ed062c8dSJulian Elischer 	 * EXIT clears all sign of other threads when
696ed062c8dSJulian Elischer 	 * it goes to single threading, so the last thread always
697ed062c8dSJulian Elischer 	 * takes the short path.
6981faf202eSJulian Elischer 	 */
699ed062c8dSJulian Elischer 	if (p->p_flag & P_HADTHREADS) {
7001faf202eSJulian Elischer 		if (p->p_numthreads > 1) {
701fd229b5bSKonstantin Belousov 			atomic_add_int(&td->td_proc->p_exitthreads, 1);
702d3a0bd78SJulian Elischer 			thread_unlink(td);
7031c4bcd05SJeff Roberson 			td2 = FIRST_THREAD_IN_PROC(p);
7041c4bcd05SJeff Roberson 			sched_exit_thread(td2, td);
705ed062c8dSJulian Elischer 
706ed062c8dSJulian Elischer 			/*
70744990b8cSJulian Elischer 			 * The test below is NOT true if we are the
7089182554aSKonstantin Belousov 			 * sole exiting thread. P_STOPPED_SINGLE is unset
70944990b8cSJulian Elischer 			 * in exit1() after it is the only survivor.
71044990b8cSJulian Elischer 			 */
7111279572aSDavid Xu 			if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
71244990b8cSJulian Elischer 				if (p->p_numthreads == p->p_suspcount) {
713a54e85fdSJeff Roberson 					thread_lock(p->p_singlethread);
7147847a9daSJohn Baldwin 					wakeup_swapper = thread_unsuspend_one(
71584cdea97SKonstantin Belousov 						p->p_singlethread, p, false);
7167847a9daSJohn Baldwin 					if (wakeup_swapper)
7177847a9daSJohn Baldwin 						kick_proc0();
71844990b8cSJulian Elischer 				}
71944990b8cSJulian Elischer 			}
72048bfcdddSJulian Elischer 
721696058c3SJulian Elischer 			PCPU_SET(deadthread, td);
7221faf202eSJulian Elischer 		} else {
723ed062c8dSJulian Elischer 			/*
724ed062c8dSJulian Elischer 			 * The last thread is exiting.. but not through exit()
725ed062c8dSJulian Elischer 			 */
726ed062c8dSJulian Elischer 			panic ("thread_exit: Last thread exiting on its own");
727ed062c8dSJulian Elischer 		}
7281faf202eSJulian Elischer 	}
72916d95d4fSJoseph Koshy #ifdef	HWPMC_HOOKS
73016d95d4fSJoseph Koshy 	/*
73116d95d4fSJoseph Koshy 	 * If this thread is part of a process that is being tracked by hwpmc(4),
73216d95d4fSJoseph Koshy 	 * inform the module of the thread's impending exit.
73316d95d4fSJoseph Koshy 	 */
7346161b98cSMatt Macy 	if (PMC_PROC_IS_USING_PMCS(td->td_proc)) {
73516d95d4fSJoseph Koshy 		PMC_SWITCH_CONTEXT(td, PMC_FN_CSW_OUT);
7366161b98cSMatt Macy 		PMC_CALL_HOOK_UNLOCKED(td, PMC_FN_THR_EXIT, NULL);
737ebfaf69cSMatt Macy 	} else if (PMC_SYSTEM_SAMPLING_ACTIVE())
738ebfaf69cSMatt Macy 		PMC_CALL_HOOK_UNLOCKED(td, PMC_FN_THR_EXIT_LOG, NULL);
73916d95d4fSJoseph Koshy #endif
740a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
7415c7bebf9SKonstantin Belousov 	PROC_STATLOCK(p);
7425c7bebf9SKonstantin Belousov 	thread_lock(td);
7435c7bebf9SKonstantin Belousov 	PROC_SUNLOCK(p);
7447e3a96eaSJohn Baldwin 
7457e3a96eaSJohn Baldwin 	/* Do the same timestamp bookkeeping that mi_switch() would do. */
7467e3a96eaSJohn Baldwin 	new_switchtime = cpu_ticks();
7477e3a96eaSJohn Baldwin 	runtime = new_switchtime - PCPU_GET(switchtime);
7487e3a96eaSJohn Baldwin 	td->td_runtime += runtime;
7497e3a96eaSJohn Baldwin 	td->td_incruntime += runtime;
7507e3a96eaSJohn Baldwin 	PCPU_SET(switchtime, new_switchtime);
7517e3a96eaSJohn Baldwin 	PCPU_SET(switchticks, ticks);
75283c9dea1SGleb Smirnoff 	VM_CNT_INC(v_swtch);
7537e3a96eaSJohn Baldwin 
7547e3a96eaSJohn Baldwin 	/* Save our resource usage in our process. */
7557e3a96eaSJohn Baldwin 	td->td_ru.ru_nvcsw++;
75661a74c5cSJeff Roberson 	ruxagg_locked(p, td);
7577e3a96eaSJohn Baldwin 	rucollect(&p->p_ru, &td->td_ru);
7585c7bebf9SKonstantin Belousov 	PROC_STATUNLOCK(p);
7597e3a96eaSJohn Baldwin 
760dcc9954eSJulian Elischer 	td->td_state = TDS_INACTIVE;
7613d06b4b3SAttilio Rao #ifdef WITNESS
7623d06b4b3SAttilio Rao 	witness_thread_exit(td);
7633d06b4b3SAttilio Rao #endif
764732d9528SJulian Elischer 	CTR1(KTR_PROC, "thread_exit: cpu_throw() thread %p", td);
765a54e85fdSJeff Roberson 	sched_throw(td);
766cc66ebe2SPeter Wemm 	panic("I'm a teapot!");
76744990b8cSJulian Elischer 	/* NOTREACHED */
76844990b8cSJulian Elischer }
76944990b8cSJulian Elischer 
77044990b8cSJulian Elischer /*
771696058c3SJulian Elischer  * Do any thread specific cleanups that may be needed in wait()
77237814395SPeter Wemm  * called with Giant, proc and schedlock not held.
773696058c3SJulian Elischer  */
774696058c3SJulian Elischer void
775696058c3SJulian Elischer thread_wait(struct proc *p)
776696058c3SJulian Elischer {
777696058c3SJulian Elischer 	struct thread *td;
778696058c3SJulian Elischer 
77937814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
780624bf9e1SKonstantin Belousov 	KASSERT(p->p_numthreads == 1, ("multiple threads in thread_wait()"));
781624bf9e1SKonstantin Belousov 	KASSERT(p->p_exitthreads == 0, ("p_exitthreads leaking"));
782ff8fbcffSJeff Roberson 	td = FIRST_THREAD_IN_PROC(p);
783ff8fbcffSJeff Roberson 	/* Lock the last thread so we spin until it exits cpu_throw(). */
784ff8fbcffSJeff Roberson 	thread_lock(td);
785ff8fbcffSJeff Roberson 	thread_unlock(td);
7862e6b8de4SJeff Roberson 	lock_profile_thread_exit(td);
787d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
788d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
789696058c3SJulian Elischer 	cpu_thread_clean(td);
7904ea6a9a2SMateusz Guzik 	thread_cow_free(td);
7912d19b736SKonstantin Belousov 	callout_drain(&td->td_slpcallout);
792696058c3SJulian Elischer 	thread_reap();	/* check for zombie threads etc. */
793696058c3SJulian Elischer }
794696058c3SJulian Elischer 
795696058c3SJulian Elischer /*
79644990b8cSJulian Elischer  * Link a thread to a process.
7971faf202eSJulian Elischer  * set up anything that needs to be initialized for it to
7981faf202eSJulian Elischer  * be used by the process.
79944990b8cSJulian Elischer  */
80044990b8cSJulian Elischer void
8018460a577SJohn Birrell thread_link(struct thread *td, struct proc *p)
80244990b8cSJulian Elischer {
80344990b8cSJulian Elischer 
804a54e85fdSJeff Roberson 	/*
805a54e85fdSJeff Roberson 	 * XXX This can't be enabled because it's called for proc0 before
806374ae2a3SJeff Roberson 	 * its lock has been created.
807374ae2a3SJeff Roberson 	 * PROC_LOCK_ASSERT(p, MA_OWNED);
808a54e85fdSJeff Roberson 	 */
80971fad9fdSJulian Elischer 	td->td_state    = TDS_INACTIVE;
81044990b8cSJulian Elischer 	td->td_proc     = p;
811b61ce5b0SJeff Roberson 	td->td_flags    = TDF_INMEM;
81244990b8cSJulian Elischer 
8131faf202eSJulian Elischer 	LIST_INIT(&td->td_contested);
814eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[0]);
815eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[1]);
816f6eccf96SGleb Smirnoff #ifdef EPOCH_TRACE
817dd902d01SGleb Smirnoff 	SLIST_INIT(&td->td_epochs);
818f6eccf96SGleb Smirnoff #endif
8199104847fSDavid Xu 	sigqueue_init(&td->td_sigqueue, p);
820fd90e2edSJung-uk Kim 	callout_init(&td->td_slpcallout, 1);
82166d8df9dSDaniel Eischen 	TAILQ_INSERT_TAIL(&p->p_threads, td, td_plist);
82244990b8cSJulian Elischer 	p->p_numthreads++;
82344990b8cSJulian Elischer }
82444990b8cSJulian Elischer 
825ed062c8dSJulian Elischer /*
826ed062c8dSJulian Elischer  * Called from:
827ed062c8dSJulian Elischer  *  thread_exit()
828ed062c8dSJulian Elischer  */
829d3a0bd78SJulian Elischer void
830d3a0bd78SJulian Elischer thread_unlink(struct thread *td)
831d3a0bd78SJulian Elischer {
832d3a0bd78SJulian Elischer 	struct proc *p = td->td_proc;
833d3a0bd78SJulian Elischer 
834374ae2a3SJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
835f6eccf96SGleb Smirnoff #ifdef EPOCH_TRACE
836dd902d01SGleb Smirnoff 	MPASS(SLIST_EMPTY(&td->td_epochs));
837f6eccf96SGleb Smirnoff #endif
838dd902d01SGleb Smirnoff 
839d3a0bd78SJulian Elischer 	TAILQ_REMOVE(&p->p_threads, td, td_plist);
840d3a0bd78SJulian Elischer 	p->p_numthreads--;
841d3a0bd78SJulian Elischer 	/* could clear a few other things here */
8428460a577SJohn Birrell 	/* Must  NOT clear links to proc! */
8435c8329edSJulian Elischer }
8445c8329edSJulian Elischer 
84579799053SKonstantin Belousov static int
84679799053SKonstantin Belousov calc_remaining(struct proc *p, int mode)
84779799053SKonstantin Belousov {
84879799053SKonstantin Belousov 	int remaining;
84979799053SKonstantin Belousov 
8507b519077SKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
8517b519077SKonstantin Belousov 	PROC_SLOCK_ASSERT(p, MA_OWNED);
85279799053SKonstantin Belousov 	if (mode == SINGLE_EXIT)
85379799053SKonstantin Belousov 		remaining = p->p_numthreads;
85479799053SKonstantin Belousov 	else if (mode == SINGLE_BOUNDARY)
85579799053SKonstantin Belousov 		remaining = p->p_numthreads - p->p_boundary_count;
8566ddcc233SKonstantin Belousov 	else if (mode == SINGLE_NO_EXIT || mode == SINGLE_ALLPROC)
85779799053SKonstantin Belousov 		remaining = p->p_numthreads - p->p_suspcount;
85879799053SKonstantin Belousov 	else
85979799053SKonstantin Belousov 		panic("calc_remaining: wrong mode %d", mode);
86079799053SKonstantin Belousov 	return (remaining);
86179799053SKonstantin Belousov }
86279799053SKonstantin Belousov 
86307a9368aSKonstantin Belousov static int
86407a9368aSKonstantin Belousov remain_for_mode(int mode)
86507a9368aSKonstantin Belousov {
86607a9368aSKonstantin Belousov 
8676ddcc233SKonstantin Belousov 	return (mode == SINGLE_ALLPROC ? 0 : 1);
86807a9368aSKonstantin Belousov }
86907a9368aSKonstantin Belousov 
87007a9368aSKonstantin Belousov static int
87107a9368aSKonstantin Belousov weed_inhib(int mode, struct thread *td2, struct proc *p)
87207a9368aSKonstantin Belousov {
87307a9368aSKonstantin Belousov 	int wakeup_swapper;
87407a9368aSKonstantin Belousov 
87507a9368aSKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
87607a9368aSKonstantin Belousov 	PROC_SLOCK_ASSERT(p, MA_OWNED);
87707a9368aSKonstantin Belousov 	THREAD_LOCK_ASSERT(td2, MA_OWNED);
87807a9368aSKonstantin Belousov 
87907a9368aSKonstantin Belousov 	wakeup_swapper = 0;
88061a74c5cSJeff Roberson 
88161a74c5cSJeff Roberson 	/*
88261a74c5cSJeff Roberson 	 * Since the thread lock is dropped by the scheduler we have
88361a74c5cSJeff Roberson 	 * to retry to check for races.
88461a74c5cSJeff Roberson 	 */
88561a74c5cSJeff Roberson restart:
88607a9368aSKonstantin Belousov 	switch (mode) {
88707a9368aSKonstantin Belousov 	case SINGLE_EXIT:
88861a74c5cSJeff Roberson 		if (TD_IS_SUSPENDED(td2)) {
88984cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, true);
89061a74c5cSJeff Roberson 			thread_lock(td2);
89161a74c5cSJeff Roberson 			goto restart;
89261a74c5cSJeff Roberson 		}
89361a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
89407a9368aSKonstantin Belousov 			wakeup_swapper |= sleepq_abort(td2, EINTR);
89561a74c5cSJeff Roberson 			return (wakeup_swapper);
89661a74c5cSJeff Roberson 		}
89707a9368aSKonstantin Belousov 		break;
89807a9368aSKonstantin Belousov 	case SINGLE_BOUNDARY:
89907a9368aSKonstantin Belousov 	case SINGLE_NO_EXIT:
90061a74c5cSJeff Roberson 		if (TD_IS_SUSPENDED(td2) &&
90161a74c5cSJeff Roberson 		    (td2->td_flags & TDF_BOUNDARY) == 0) {
90284cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, false);
90361a74c5cSJeff Roberson 			thread_lock(td2);
90461a74c5cSJeff Roberson 			goto restart;
90561a74c5cSJeff Roberson 		}
90661a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
90707a9368aSKonstantin Belousov 			wakeup_swapper |= sleepq_abort(td2, ERESTART);
90861a74c5cSJeff Roberson 			return (wakeup_swapper);
90961a74c5cSJeff Roberson 		}
910917dd390SKonstantin Belousov 		break;
9116ddcc233SKonstantin Belousov 	case SINGLE_ALLPROC:
9126ddcc233SKonstantin Belousov 		/*
9136ddcc233SKonstantin Belousov 		 * ALLPROC suspend tries to avoid spurious EINTR for
9146ddcc233SKonstantin Belousov 		 * threads sleeping interruptable, by suspending the
9156ddcc233SKonstantin Belousov 		 * thread directly, similarly to sig_suspend_threads().
9166ddcc233SKonstantin Belousov 		 * Since such sleep is not performed at the user
9176ddcc233SKonstantin Belousov 		 * boundary, TDF_BOUNDARY flag is not set, and TDF_ALLPROCSUSP
9186ddcc233SKonstantin Belousov 		 * is used to avoid immediate un-suspend.
9196ddcc233SKonstantin Belousov 		 */
9206ddcc233SKonstantin Belousov 		if (TD_IS_SUSPENDED(td2) && (td2->td_flags & (TDF_BOUNDARY |
92161a74c5cSJeff Roberson 		    TDF_ALLPROCSUSP)) == 0) {
92284cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, false);
92361a74c5cSJeff Roberson 			thread_lock(td2);
92461a74c5cSJeff Roberson 			goto restart;
92561a74c5cSJeff Roberson 		}
92661a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
9276ddcc233SKonstantin Belousov 			if ((td2->td_flags & TDF_SBDRY) == 0) {
9286ddcc233SKonstantin Belousov 				thread_suspend_one(td2);
9296ddcc233SKonstantin Belousov 				td2->td_flags |= TDF_ALLPROCSUSP;
9306ddcc233SKonstantin Belousov 			} else {
9316ddcc233SKonstantin Belousov 				wakeup_swapper |= sleepq_abort(td2, ERESTART);
93261a74c5cSJeff Roberson 				return (wakeup_swapper);
9336ddcc233SKonstantin Belousov 			}
9346ddcc233SKonstantin Belousov 		}
93507a9368aSKonstantin Belousov 		break;
93661a74c5cSJeff Roberson 	default:
93761a74c5cSJeff Roberson 		break;
93807a9368aSKonstantin Belousov 	}
93961a74c5cSJeff Roberson 	thread_unlock(td2);
94007a9368aSKonstantin Belousov 	return (wakeup_swapper);
94107a9368aSKonstantin Belousov }
94207a9368aSKonstantin Belousov 
9435215b187SJeff Roberson /*
94444990b8cSJulian Elischer  * Enforce single-threading.
94544990b8cSJulian Elischer  *
94644990b8cSJulian Elischer  * Returns 1 if the caller must abort (another thread is waiting to
94744990b8cSJulian Elischer  * exit the process or similar). Process is locked!
94844990b8cSJulian Elischer  * Returns 0 when you are successfully the only thread running.
94944990b8cSJulian Elischer  * A process has successfully single threaded in the suspend mode when
95044990b8cSJulian Elischer  * There are no threads in user mode. Threads in the kernel must be
95144990b8cSJulian Elischer  * allowed to continue until they get to the user boundary. They may even
95244990b8cSJulian Elischer  * copy out their return values and data before suspending. They may however be
953e2668f55SMaxim Konovalov  * accelerated in reaching the user boundary as we will wake up
95444990b8cSJulian Elischer  * any sleeping threads that are interruptable. (PCATCH).
95544990b8cSJulian Elischer  */
95644990b8cSJulian Elischer int
9576ddcc233SKonstantin Belousov thread_single(struct proc *p, int mode)
95844990b8cSJulian Elischer {
95944990b8cSJulian Elischer 	struct thread *td;
96044990b8cSJulian Elischer 	struct thread *td2;
961da7bbd2cSJohn Baldwin 	int remaining, wakeup_swapper;
96244990b8cSJulian Elischer 
96344990b8cSJulian Elischer 	td = curthread;
9646ddcc233SKonstantin Belousov 	KASSERT(mode == SINGLE_EXIT || mode == SINGLE_BOUNDARY ||
9656ddcc233SKonstantin Belousov 	    mode == SINGLE_ALLPROC || mode == SINGLE_NO_EXIT,
9666ddcc233SKonstantin Belousov 	    ("invalid mode %d", mode));
9676ddcc233SKonstantin Belousov 	/*
9686ddcc233SKonstantin Belousov 	 * If allowing non-ALLPROC singlethreading for non-curproc
9696ddcc233SKonstantin Belousov 	 * callers, calc_remaining() and remain_for_mode() should be
9706ddcc233SKonstantin Belousov 	 * adjusted to also account for td->td_proc != p.  For now
9716ddcc233SKonstantin Belousov 	 * this is not implemented because it is not used.
9726ddcc233SKonstantin Belousov 	 */
9736ddcc233SKonstantin Belousov 	KASSERT((mode == SINGLE_ALLPROC && td->td_proc != p) ||
9746ddcc233SKonstantin Belousov 	    (mode != SINGLE_ALLPROC && td->td_proc == p),
9756ddcc233SKonstantin Belousov 	    ("mode %d proc %p curproc %p", mode, p, td->td_proc));
97637814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
97744990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
97844990b8cSJulian Elischer 
9796ddcc233SKonstantin Belousov 	if ((p->p_flag & P_HADTHREADS) == 0 && mode != SINGLE_ALLPROC)
98044990b8cSJulian Elischer 		return (0);
98144990b8cSJulian Elischer 
982e3b9bf71SJulian Elischer 	/* Is someone already single threading? */
983906ac69dSDavid Xu 	if (p->p_singlethread != NULL && p->p_singlethread != td)
98444990b8cSJulian Elischer 		return (1);
98544990b8cSJulian Elischer 
986906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
987906ac69dSDavid Xu 		p->p_flag |= P_SINGLE_EXIT;
988906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_BOUNDARY;
989906ac69dSDavid Xu 	} else {
990906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_EXIT;
991906ac69dSDavid Xu 		if (mode == SINGLE_BOUNDARY)
992906ac69dSDavid Xu 			p->p_flag |= P_SINGLE_BOUNDARY;
993906ac69dSDavid Xu 		else
994906ac69dSDavid Xu 			p->p_flag &= ~P_SINGLE_BOUNDARY;
995906ac69dSDavid Xu 	}
9966ddcc233SKonstantin Belousov 	if (mode == SINGLE_ALLPROC)
9976ddcc233SKonstantin Belousov 		p->p_flag |= P_TOTAL_STOP;
9981279572aSDavid Xu 	p->p_flag |= P_STOPPED_SINGLE;
9997b4a950aSDavid Xu 	PROC_SLOCK(p);
1000112afcb2SJohn Baldwin 	p->p_singlethread = td;
100179799053SKonstantin Belousov 	remaining = calc_remaining(p, mode);
100207a9368aSKonstantin Belousov 	while (remaining != remain_for_mode(mode)) {
1003bf1a3220SDavid Xu 		if (P_SHOULDSTOP(p) != P_STOPPED_SINGLE)
1004bf1a3220SDavid Xu 			goto stopme;
1005da7bbd2cSJohn Baldwin 		wakeup_swapper = 0;
100644990b8cSJulian Elischer 		FOREACH_THREAD_IN_PROC(p, td2) {
100744990b8cSJulian Elischer 			if (td2 == td)
100844990b8cSJulian Elischer 				continue;
1009a54e85fdSJeff Roberson 			thread_lock(td2);
1010b7edba77SJeff Roberson 			td2->td_flags |= TDF_ASTPENDING | TDF_NEEDSUSPCHK;
10116ddcc233SKonstantin Belousov 			if (TD_IS_INHIBITED(td2)) {
101207a9368aSKonstantin Belousov 				wakeup_swapper |= weed_inhib(mode, td2, p);
1013d8267df7SDavid Xu #ifdef SMP
10146ddcc233SKonstantin Belousov 			} else if (TD_IS_RUNNING(td2) && td != td2) {
1015d8267df7SDavid Xu 				forward_signal(td2);
101661a74c5cSJeff Roberson 				thread_unlock(td2);
1017d8267df7SDavid Xu #endif
101861a74c5cSJeff Roberson 			} else
1019a54e85fdSJeff Roberson 				thread_unlock(td2);
10209d102777SJulian Elischer 		}
1021da7bbd2cSJohn Baldwin 		if (wakeup_swapper)
1022da7bbd2cSJohn Baldwin 			kick_proc0();
102379799053SKonstantin Belousov 		remaining = calc_remaining(p, mode);
1024ec008e96SDavid Xu 
10259d102777SJulian Elischer 		/*
10269d102777SJulian Elischer 		 * Maybe we suspended some threads.. was it enough?
10279d102777SJulian Elischer 		 */
102807a9368aSKonstantin Belousov 		if (remaining == remain_for_mode(mode))
10299d102777SJulian Elischer 			break;
10309d102777SJulian Elischer 
1031bf1a3220SDavid Xu stopme:
103244990b8cSJulian Elischer 		/*
103344990b8cSJulian Elischer 		 * Wake us up when everyone else has suspended.
1034e3b9bf71SJulian Elischer 		 * In the mean time we suspend as well.
103544990b8cSJulian Elischer 		 */
10366ddcc233SKonstantin Belousov 		thread_suspend_switch(td, p);
103779799053SKonstantin Belousov 		remaining = calc_remaining(p, mode);
103844990b8cSJulian Elischer 	}
1039906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
104091599697SJulian Elischer 		/*
10418626a0ddSKonstantin Belousov 		 * Convert the process to an unthreaded process.  The
10428626a0ddSKonstantin Belousov 		 * SINGLE_EXIT is called by exit1() or execve(), in
10438626a0ddSKonstantin Belousov 		 * both cases other threads must be retired.
104491599697SJulian Elischer 		 */
10458626a0ddSKonstantin Belousov 		KASSERT(p->p_numthreads == 1, ("Unthreading with >1 threads"));
1046ed062c8dSJulian Elischer 		p->p_singlethread = NULL;
10478626a0ddSKonstantin Belousov 		p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_HADTHREADS);
1048fd229b5bSKonstantin Belousov 
1049fd229b5bSKonstantin Belousov 		/*
1050fd229b5bSKonstantin Belousov 		 * Wait for any remaining threads to exit cpu_throw().
1051fd229b5bSKonstantin Belousov 		 */
1052fd229b5bSKonstantin Belousov 		while (p->p_exitthreads != 0) {
1053fd229b5bSKonstantin Belousov 			PROC_SUNLOCK(p);
1054fd229b5bSKonstantin Belousov 			PROC_UNLOCK(p);
1055fd229b5bSKonstantin Belousov 			sched_relinquish(td);
1056fd229b5bSKonstantin Belousov 			PROC_LOCK(p);
1057fd229b5bSKonstantin Belousov 			PROC_SLOCK(p);
1058fd229b5bSKonstantin Belousov 		}
1059ac437c07SKonstantin Belousov 	} else if (mode == SINGLE_BOUNDARY) {
1060ac437c07SKonstantin Belousov 		/*
1061ac437c07SKonstantin Belousov 		 * Wait until all suspended threads are removed from
1062ac437c07SKonstantin Belousov 		 * the processors.  The thread_suspend_check()
1063ac437c07SKonstantin Belousov 		 * increments p_boundary_count while it is still
1064ac437c07SKonstantin Belousov 		 * running, which makes it possible for the execve()
1065ac437c07SKonstantin Belousov 		 * to destroy vmspace while our other threads are
1066ac437c07SKonstantin Belousov 		 * still using the address space.
1067ac437c07SKonstantin Belousov 		 *
1068ac437c07SKonstantin Belousov 		 * We lock the thread, which is only allowed to
1069ac437c07SKonstantin Belousov 		 * succeed after context switch code finished using
1070ac437c07SKonstantin Belousov 		 * the address space.
1071ac437c07SKonstantin Belousov 		 */
1072ac437c07SKonstantin Belousov 		FOREACH_THREAD_IN_PROC(p, td2) {
1073ac437c07SKonstantin Belousov 			if (td2 == td)
1074ac437c07SKonstantin Belousov 				continue;
1075ac437c07SKonstantin Belousov 			thread_lock(td2);
1076ac437c07SKonstantin Belousov 			KASSERT((td2->td_flags & TDF_BOUNDARY) != 0,
1077ac437c07SKonstantin Belousov 			    ("td %p not on boundary", td2));
1078ac437c07SKonstantin Belousov 			KASSERT(TD_IS_SUSPENDED(td2),
1079ac437c07SKonstantin Belousov 			    ("td %p is not suspended", td2));
1080ac437c07SKonstantin Belousov 			thread_unlock(td2);
1081ac437c07SKonstantin Belousov 		}
108291599697SJulian Elischer 	}
10837b4a950aSDavid Xu 	PROC_SUNLOCK(p);
108444990b8cSJulian Elischer 	return (0);
108544990b8cSJulian Elischer }
108644990b8cSJulian Elischer 
10878638fe7bSKonstantin Belousov bool
10888638fe7bSKonstantin Belousov thread_suspend_check_needed(void)
10898638fe7bSKonstantin Belousov {
10908638fe7bSKonstantin Belousov 	struct proc *p;
10918638fe7bSKonstantin Belousov 	struct thread *td;
10928638fe7bSKonstantin Belousov 
10938638fe7bSKonstantin Belousov 	td = curthread;
10948638fe7bSKonstantin Belousov 	p = td->td_proc;
10958638fe7bSKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
10968638fe7bSKonstantin Belousov 	return (P_SHOULDSTOP(p) || ((p->p_flag & P_TRACED) != 0 &&
10978638fe7bSKonstantin Belousov 	    (td->td_dbgflags & TDB_SUSPEND) != 0));
10988638fe7bSKonstantin Belousov }
10998638fe7bSKonstantin Belousov 
110044990b8cSJulian Elischer /*
110144990b8cSJulian Elischer  * Called in from locations that can safely check to see
110244990b8cSJulian Elischer  * whether we have to suspend or at least throttle for a
110344990b8cSJulian Elischer  * single-thread event (e.g. fork).
110444990b8cSJulian Elischer  *
110544990b8cSJulian Elischer  * Such locations include userret().
110644990b8cSJulian Elischer  * If the "return_instead" argument is non zero, the thread must be able to
110744990b8cSJulian Elischer  * accept 0 (caller may continue), or 1 (caller must abort) as a result.
110844990b8cSJulian Elischer  *
110944990b8cSJulian Elischer  * The 'return_instead' argument tells the function if it may do a
111044990b8cSJulian Elischer  * thread_exit() or suspend, or whether the caller must abort and back
111144990b8cSJulian Elischer  * out instead.
111244990b8cSJulian Elischer  *
111344990b8cSJulian Elischer  * If the thread that set the single_threading request has set the
111444990b8cSJulian Elischer  * P_SINGLE_EXIT bit in the process flags then this call will never return
111544990b8cSJulian Elischer  * if 'return_instead' is false, but will exit.
111644990b8cSJulian Elischer  *
111744990b8cSJulian Elischer  * P_SINGLE_EXIT | return_instead == 0| return_instead != 0
111844990b8cSJulian Elischer  *---------------+--------------------+---------------------
111944990b8cSJulian Elischer  *       0       | returns 0          |   returns 0 or 1
1120353374b5SJohn Baldwin  *               | when ST ends       |   immediately
112144990b8cSJulian Elischer  *---------------+--------------------+---------------------
112244990b8cSJulian Elischer  *       1       | thread exits       |   returns 1
1123353374b5SJohn Baldwin  *               |                    |  immediately
112444990b8cSJulian Elischer  * 0 = thread_exit() or suspension ok,
112544990b8cSJulian Elischer  * other = return error instead of stopping the thread.
112644990b8cSJulian Elischer  *
112744990b8cSJulian Elischer  * While a full suspension is under effect, even a single threading
112844990b8cSJulian Elischer  * thread would be suspended if it made this call (but it shouldn't).
112944990b8cSJulian Elischer  * This call should only be made from places where
113044990b8cSJulian Elischer  * thread_exit() would be safe as that may be the outcome unless
113144990b8cSJulian Elischer  * return_instead is set.
113244990b8cSJulian Elischer  */
113344990b8cSJulian Elischer int
113444990b8cSJulian Elischer thread_suspend_check(int return_instead)
113544990b8cSJulian Elischer {
1136ecafb24bSJuli Mallett 	struct thread *td;
1137ecafb24bSJuli Mallett 	struct proc *p;
113846e47c4fSKonstantin Belousov 	int wakeup_swapper;
113944990b8cSJulian Elischer 
114044990b8cSJulian Elischer 	td = curthread;
114144990b8cSJulian Elischer 	p = td->td_proc;
114237814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
114344990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
11448638fe7bSKonstantin Belousov 	while (thread_suspend_check_needed()) {
11451279572aSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
114644990b8cSJulian Elischer 			KASSERT(p->p_singlethread != NULL,
114744990b8cSJulian Elischer 			    ("singlethread not set"));
114844990b8cSJulian Elischer 			/*
1149e3b9bf71SJulian Elischer 			 * The only suspension in action is a
1150e3b9bf71SJulian Elischer 			 * single-threading. Single threader need not stop.
1151bd07998eSKonstantin Belousov 			 * It is safe to access p->p_singlethread unlocked
1152bd07998eSKonstantin Belousov 			 * because it can only be set to our address by us.
115344990b8cSJulian Elischer 			 */
1154e3b9bf71SJulian Elischer 			if (p->p_singlethread == td)
115544990b8cSJulian Elischer 				return (0);	/* Exempt from stopping. */
115644990b8cSJulian Elischer 		}
115745a4bfa1SDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && return_instead)
115894f0972bSDavid Xu 			return (EINTR);
115944990b8cSJulian Elischer 
1160906ac69dSDavid Xu 		/* Should we goto user boundary if we didn't come from there? */
1161906ac69dSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
1162906ac69dSDavid Xu 		    (p->p_flag & P_SINGLE_BOUNDARY) && return_instead)
116394f0972bSDavid Xu 			return (ERESTART);
1164906ac69dSDavid Xu 
116544990b8cSJulian Elischer 		/*
11663077f938SKonstantin Belousov 		 * Ignore suspend requests if they are deferred.
1167d071a6faSJohn Baldwin 		 */
11683077f938SKonstantin Belousov 		if ((td->td_flags & TDF_SBDRY) != 0) {
1169d071a6faSJohn Baldwin 			KASSERT(return_instead,
1170d071a6faSJohn Baldwin 			    ("TDF_SBDRY set for unsafe thread_suspend_check"));
117146e47c4fSKonstantin Belousov 			KASSERT((td->td_flags & (TDF_SEINTR | TDF_SERESTART)) !=
117246e47c4fSKonstantin Belousov 			    (TDF_SEINTR | TDF_SERESTART),
117346e47c4fSKonstantin Belousov 			    ("both TDF_SEINTR and TDF_SERESTART"));
117446e47c4fSKonstantin Belousov 			return (TD_SBDRY_INTR(td) ? TD_SBDRY_ERRNO(td) : 0);
1175d071a6faSJohn Baldwin 		}
1176d071a6faSJohn Baldwin 
1177d071a6faSJohn Baldwin 		/*
117844990b8cSJulian Elischer 		 * If the process is waiting for us to exit,
117944990b8cSJulian Elischer 		 * this thread should just suicide.
11801279572aSDavid Xu 		 * Assumes that P_SINGLE_EXIT implies P_STOPPED_SINGLE.
118144990b8cSJulian Elischer 		 */
1182cf7d9a8cSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td)) {
1183cf7d9a8cSDavid Xu 			PROC_UNLOCK(p);
118491d1786fSDmitry Chagin 
118591d1786fSDmitry Chagin 			/*
118691d1786fSDmitry Chagin 			 * Allow Linux emulation layer to do some work
118791d1786fSDmitry Chagin 			 * before thread suicide.
118891d1786fSDmitry Chagin 			 */
118991d1786fSDmitry Chagin 			if (__predict_false(p->p_sysent->sv_thread_detach != NULL))
119091d1786fSDmitry Chagin 				(p->p_sysent->sv_thread_detach)(td);
11912a339d9eSKonstantin Belousov 			umtx_thread_exit(td);
1192d1e7a4a5SJohn Baldwin 			kern_thr_exit(td);
1193d1e7a4a5SJohn Baldwin 			panic("stopped thread did not exit");
1194cf7d9a8cSDavid Xu 		}
119521ecd1e9SDavid Xu 
119621ecd1e9SDavid Xu 		PROC_SLOCK(p);
119721ecd1e9SDavid Xu 		thread_stopped(p);
1198a54e85fdSJeff Roberson 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
1199a54e85fdSJeff Roberson 			if (p->p_numthreads == p->p_suspcount + 1) {
1200a54e85fdSJeff Roberson 				thread_lock(p->p_singlethread);
120184cdea97SKonstantin Belousov 				wakeup_swapper = thread_unsuspend_one(
120284cdea97SKonstantin Belousov 				    p->p_singlethread, p, false);
12037847a9daSJohn Baldwin 				if (wakeup_swapper)
12047847a9daSJohn Baldwin 					kick_proc0();
1205a54e85fdSJeff Roberson 			}
1206a54e85fdSJeff Roberson 		}
12073f9be10eSDavid Xu 		PROC_UNLOCK(p);
12087b4a950aSDavid Xu 		thread_lock(td);
120944990b8cSJulian Elischer 		/*
121044990b8cSJulian Elischer 		 * When a thread suspends, it just
1211ad1e7d28SJulian Elischer 		 * gets taken off all queues.
121244990b8cSJulian Elischer 		 */
121371fad9fdSJulian Elischer 		thread_suspend_one(td);
1214906ac69dSDavid Xu 		if (return_instead == 0) {
1215906ac69dSDavid Xu 			p->p_boundary_count++;
1216906ac69dSDavid Xu 			td->td_flags |= TDF_BOUNDARY;
1217cf19bf91SJulian Elischer 		}
12187b4a950aSDavid Xu 		PROC_SUNLOCK(p);
1219686bcb5cSJeff Roberson 		mi_switch(SW_INVOL | SWT_SUSPEND);
122044990b8cSJulian Elischer 		PROC_LOCK(p);
122144990b8cSJulian Elischer 	}
122244990b8cSJulian Elischer 	return (0);
122344990b8cSJulian Elischer }
122444990b8cSJulian Elischer 
1225478ca4b0SKonstantin Belousov /*
1226478ca4b0SKonstantin Belousov  * Check for possible stops and suspensions while executing a
1227478ca4b0SKonstantin Belousov  * casueword or similar transiently failing operation.
1228478ca4b0SKonstantin Belousov  *
1229478ca4b0SKonstantin Belousov  * The sleep argument controls whether the function can handle a stop
1230478ca4b0SKonstantin Belousov  * request itself or it should return ERESTART and the request is
1231478ca4b0SKonstantin Belousov  * proceed at the kernel/user boundary in ast.
1232478ca4b0SKonstantin Belousov  *
1233478ca4b0SKonstantin Belousov  * Typically, when retrying due to casueword(9) failure (rv == 1), we
1234478ca4b0SKonstantin Belousov  * should handle the stop requests there, with exception of cases when
1235478ca4b0SKonstantin Belousov  * the thread owns a kernel resource, for instance busied the umtx
1236300b525dSKonstantin Belousov  * key, or when functions return immediately if thread_check_susp()
1237478ca4b0SKonstantin Belousov  * returned non-zero.  On the other hand, retrying the whole lock
1238478ca4b0SKonstantin Belousov  * operation, we better not stop there but delegate the handling to
1239478ca4b0SKonstantin Belousov  * ast.
1240478ca4b0SKonstantin Belousov  *
1241478ca4b0SKonstantin Belousov  * If the request is for thread termination P_SINGLE_EXIT, we cannot
1242478ca4b0SKonstantin Belousov  * handle it at all, and simply return EINTR.
1243478ca4b0SKonstantin Belousov  */
1244478ca4b0SKonstantin Belousov int
1245478ca4b0SKonstantin Belousov thread_check_susp(struct thread *td, bool sleep)
1246478ca4b0SKonstantin Belousov {
1247478ca4b0SKonstantin Belousov 	struct proc *p;
1248478ca4b0SKonstantin Belousov 	int error;
1249478ca4b0SKonstantin Belousov 
1250478ca4b0SKonstantin Belousov 	/*
1251478ca4b0SKonstantin Belousov 	 * The check for TDF_NEEDSUSPCHK is racy, but it is enough to
1252478ca4b0SKonstantin Belousov 	 * eventually break the lockstep loop.
1253478ca4b0SKonstantin Belousov 	 */
1254478ca4b0SKonstantin Belousov 	if ((td->td_flags & TDF_NEEDSUSPCHK) == 0)
1255478ca4b0SKonstantin Belousov 		return (0);
1256478ca4b0SKonstantin Belousov 	error = 0;
1257478ca4b0SKonstantin Belousov 	p = td->td_proc;
1258478ca4b0SKonstantin Belousov 	PROC_LOCK(p);
1259478ca4b0SKonstantin Belousov 	if (p->p_flag & P_SINGLE_EXIT)
1260478ca4b0SKonstantin Belousov 		error = EINTR;
1261478ca4b0SKonstantin Belousov 	else if (P_SHOULDSTOP(p) ||
1262478ca4b0SKonstantin Belousov 	    ((p->p_flag & P_TRACED) && (td->td_dbgflags & TDB_SUSPEND)))
1263478ca4b0SKonstantin Belousov 		error = sleep ? thread_suspend_check(0) : ERESTART;
1264478ca4b0SKonstantin Belousov 	PROC_UNLOCK(p);
1265478ca4b0SKonstantin Belousov 	return (error);
1266478ca4b0SKonstantin Belousov }
1267478ca4b0SKonstantin Belousov 
126835c32a76SDavid Xu void
12696ddcc233SKonstantin Belousov thread_suspend_switch(struct thread *td, struct proc *p)
1270a54e85fdSJeff Roberson {
1271a54e85fdSJeff Roberson 
1272a54e85fdSJeff Roberson 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
1273a54e85fdSJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
12747b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
1275a54e85fdSJeff Roberson 	/*
1276a54e85fdSJeff Roberson 	 * We implement thread_suspend_one in stages here to avoid
1277a54e85fdSJeff Roberson 	 * dropping the proc lock while the thread lock is owned.
1278a54e85fdSJeff Roberson 	 */
12796ddcc233SKonstantin Belousov 	if (p == td->td_proc) {
1280a54e85fdSJeff Roberson 		thread_stopped(p);
1281a54e85fdSJeff Roberson 		p->p_suspcount++;
12826ddcc233SKonstantin Belousov 	}
12833f9be10eSDavid Xu 	PROC_UNLOCK(p);
12847b4a950aSDavid Xu 	thread_lock(td);
1285b7edba77SJeff Roberson 	td->td_flags &= ~TDF_NEEDSUSPCHK;
1286a54e85fdSJeff Roberson 	TD_SET_SUSPENDED(td);
1287c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
12887b4a950aSDavid Xu 	PROC_SUNLOCK(p);
1289a54e85fdSJeff Roberson 	DROP_GIANT();
1290686bcb5cSJeff Roberson 	mi_switch(SW_VOL | SWT_SUSPEND);
1291a54e85fdSJeff Roberson 	PICKUP_GIANT();
1292a54e85fdSJeff Roberson 	PROC_LOCK(p);
12937b4a950aSDavid Xu 	PROC_SLOCK(p);
1294a54e85fdSJeff Roberson }
1295a54e85fdSJeff Roberson 
1296a54e85fdSJeff Roberson void
129735c32a76SDavid Xu thread_suspend_one(struct thread *td)
129835c32a76SDavid Xu {
12996ddcc233SKonstantin Belousov 	struct proc *p;
130035c32a76SDavid Xu 
13016ddcc233SKonstantin Belousov 	p = td->td_proc;
13027b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
1303a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
1304e574e444SDavid Xu 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
130535c32a76SDavid Xu 	p->p_suspcount++;
1306b7edba77SJeff Roberson 	td->td_flags &= ~TDF_NEEDSUSPCHK;
130771fad9fdSJulian Elischer 	TD_SET_SUSPENDED(td);
1308c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
130935c32a76SDavid Xu }
131035c32a76SDavid Xu 
131184cdea97SKonstantin Belousov static int
131284cdea97SKonstantin Belousov thread_unsuspend_one(struct thread *td, struct proc *p, bool boundary)
131335c32a76SDavid Xu {
131435c32a76SDavid Xu 
1315a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
1316ad1e7d28SJulian Elischer 	KASSERT(TD_IS_SUSPENDED(td), ("Thread not suspended"));
131771fad9fdSJulian Elischer 	TD_CLR_SUSPENDED(td);
13186ddcc233SKonstantin Belousov 	td->td_flags &= ~TDF_ALLPROCSUSP;
13196ddcc233SKonstantin Belousov 	if (td->td_proc == p) {
13206ddcc233SKonstantin Belousov 		PROC_SLOCK_ASSERT(p, MA_OWNED);
132135c32a76SDavid Xu 		p->p_suspcount--;
132284cdea97SKonstantin Belousov 		if (boundary && (td->td_flags & TDF_BOUNDARY) != 0) {
132384cdea97SKonstantin Belousov 			td->td_flags &= ~TDF_BOUNDARY;
132484cdea97SKonstantin Belousov 			p->p_boundary_count--;
132584cdea97SKonstantin Belousov 		}
13266ddcc233SKonstantin Belousov 	}
132761a74c5cSJeff Roberson 	return (setrunnable(td, 0));
132835c32a76SDavid Xu }
132935c32a76SDavid Xu 
133044990b8cSJulian Elischer /*
133144990b8cSJulian Elischer  * Allow all threads blocked by single threading to continue running.
133244990b8cSJulian Elischer  */
133344990b8cSJulian Elischer void
133444990b8cSJulian Elischer thread_unsuspend(struct proc *p)
133544990b8cSJulian Elischer {
133644990b8cSJulian Elischer 	struct thread *td;
13377847a9daSJohn Baldwin 	int wakeup_swapper;
133844990b8cSJulian Elischer 
133944990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
13407b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
13417847a9daSJohn Baldwin 	wakeup_swapper = 0;
134244990b8cSJulian Elischer 	if (!P_SHOULDSTOP(p)) {
1343ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
1344a54e85fdSJeff Roberson 			thread_lock(td);
1345ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
134684cdea97SKonstantin Belousov 				wakeup_swapper |= thread_unsuspend_one(td, p,
134784cdea97SKonstantin Belousov 				    true);
134861a74c5cSJeff Roberson 			} else
1349a54e85fdSJeff Roberson 				thread_unlock(td);
1350ad1e7d28SJulian Elischer 		}
135184cdea97SKonstantin Belousov 	} else if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
135284cdea97SKonstantin Belousov 	    p->p_numthreads == p->p_suspcount) {
135344990b8cSJulian Elischer 		/*
135444990b8cSJulian Elischer 		 * Stopping everything also did the job for the single
135544990b8cSJulian Elischer 		 * threading request. Now we've downgraded to single-threaded,
135644990b8cSJulian Elischer 		 * let it continue.
135744990b8cSJulian Elischer 		 */
13586ddcc233SKonstantin Belousov 		if (p->p_singlethread->td_proc == p) {
1359a54e85fdSJeff Roberson 			thread_lock(p->p_singlethread);
13606ddcc233SKonstantin Belousov 			wakeup_swapper = thread_unsuspend_one(
136184cdea97SKonstantin Belousov 			    p->p_singlethread, p, false);
136244990b8cSJulian Elischer 		}
13636ddcc233SKonstantin Belousov 	}
13647847a9daSJohn Baldwin 	if (wakeup_swapper)
13657847a9daSJohn Baldwin 		kick_proc0();
136644990b8cSJulian Elischer }
136744990b8cSJulian Elischer 
1368ed062c8dSJulian Elischer /*
1369ed062c8dSJulian Elischer  * End the single threading mode..
1370ed062c8dSJulian Elischer  */
137144990b8cSJulian Elischer void
13726ddcc233SKonstantin Belousov thread_single_end(struct proc *p, int mode)
137344990b8cSJulian Elischer {
137444990b8cSJulian Elischer 	struct thread *td;
13757847a9daSJohn Baldwin 	int wakeup_swapper;
137644990b8cSJulian Elischer 
13776ddcc233SKonstantin Belousov 	KASSERT(mode == SINGLE_EXIT || mode == SINGLE_BOUNDARY ||
13786ddcc233SKonstantin Belousov 	    mode == SINGLE_ALLPROC || mode == SINGLE_NO_EXIT,
13796ddcc233SKonstantin Belousov 	    ("invalid mode %d", mode));
138044990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
13816ddcc233SKonstantin Belousov 	KASSERT((mode == SINGLE_ALLPROC && (p->p_flag & P_TOTAL_STOP) != 0) ||
13826ddcc233SKonstantin Belousov 	    (mode != SINGLE_ALLPROC && (p->p_flag & P_TOTAL_STOP) == 0),
13836ddcc233SKonstantin Belousov 	    ("mode %d does not match P_TOTAL_STOP", mode));
138484cdea97SKonstantin Belousov 	KASSERT(mode == SINGLE_ALLPROC || p->p_singlethread == curthread,
138584cdea97SKonstantin Belousov 	    ("thread_single_end from other thread %p %p",
138684cdea97SKonstantin Belousov 	    curthread, p->p_singlethread));
138784cdea97SKonstantin Belousov 	KASSERT(mode != SINGLE_BOUNDARY ||
138884cdea97SKonstantin Belousov 	    (p->p_flag & P_SINGLE_BOUNDARY) != 0,
138984cdea97SKonstantin Belousov 	    ("mis-matched SINGLE_BOUNDARY flags %x", p->p_flag));
13906ddcc233SKonstantin Belousov 	p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_SINGLE_BOUNDARY |
13916ddcc233SKonstantin Belousov 	    P_TOTAL_STOP);
13927b4a950aSDavid Xu 	PROC_SLOCK(p);
139344990b8cSJulian Elischer 	p->p_singlethread = NULL;
13947847a9daSJohn Baldwin 	wakeup_swapper = 0;
139549539972SJulian Elischer 	/*
13967847a9daSJohn Baldwin 	 * If there are other threads they may now run,
139749539972SJulian Elischer 	 * unless of course there is a blanket 'stop order'
139849539972SJulian Elischer 	 * on the process. The single threader must be allowed
139949539972SJulian Elischer 	 * to continue however as this is a bad place to stop.
140049539972SJulian Elischer 	 */
14016ddcc233SKonstantin Belousov 	if (p->p_numthreads != remain_for_mode(mode) && !P_SHOULDSTOP(p)) {
1402ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
1403a54e85fdSJeff Roberson 			thread_lock(td);
1404ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
140584cdea97SKonstantin Belousov 				wakeup_swapper |= thread_unsuspend_one(td, p,
140684cdea97SKonstantin Belousov 				    mode == SINGLE_BOUNDARY);
140761a74c5cSJeff Roberson 			} else
1408a54e85fdSJeff Roberson 				thread_unlock(td);
140949539972SJulian Elischer 		}
1410ad1e7d28SJulian Elischer 	}
141184cdea97SKonstantin Belousov 	KASSERT(mode != SINGLE_BOUNDARY || p->p_boundary_count == 0,
141284cdea97SKonstantin Belousov 	    ("inconsistent boundary count %d", p->p_boundary_count));
14137b4a950aSDavid Xu 	PROC_SUNLOCK(p);
14147847a9daSJohn Baldwin 	if (wakeup_swapper)
14157847a9daSJohn Baldwin 		kick_proc0();
141649539972SJulian Elischer }
14174fc21c09SDaniel Eischen 
1418aae3547bSMateusz Guzik /*
1419aae3547bSMateusz Guzik  * Locate a thread by number and return with proc lock held.
1420aae3547bSMateusz Guzik  *
1421aae3547bSMateusz Guzik  * thread exit establishes proc -> tidhash lock ordering, but lookup
1422aae3547bSMateusz Guzik  * takes tidhash first and needs to return locked proc.
1423aae3547bSMateusz Guzik  *
1424aae3547bSMateusz Guzik  * The problem is worked around by relying on type-safety of both
1425aae3547bSMateusz Guzik  * structures and doing the work in 2 steps:
1426aae3547bSMateusz Guzik  * - tidhash-locked lookup which saves both thread and proc pointers
1427aae3547bSMateusz Guzik  * - proc-locked verification that the found thread still matches
1428aae3547bSMateusz Guzik  */
1429aae3547bSMateusz Guzik static bool
1430aae3547bSMateusz Guzik tdfind_hash(lwpid_t tid, pid_t pid, struct proc **pp, struct thread **tdp)
1431cf7d9a8cSDavid Xu {
1432cf7d9a8cSDavid Xu #define RUN_THRESH	16
1433aae3547bSMateusz Guzik 	struct proc *p;
1434cf7d9a8cSDavid Xu 	struct thread *td;
1435aae3547bSMateusz Guzik 	int run;
1436aae3547bSMateusz Guzik 	bool locked;
1437cf7d9a8cSDavid Xu 
1438aae3547bSMateusz Guzik 	run = 0;
143926007fe3SMateusz Guzik 	rw_rlock(TIDHASHLOCK(tid));
1440aae3547bSMateusz Guzik 	locked = true;
1441cf7d9a8cSDavid Xu 	LIST_FOREACH(td, TIDHASH(tid), td_hash) {
1442aae3547bSMateusz Guzik 		if (td->td_tid != tid) {
1443aae3547bSMateusz Guzik 			run++;
1444aae3547bSMateusz Guzik 			continue;
1445cf7d9a8cSDavid Xu 		}
1446aae3547bSMateusz Guzik 		p = td->td_proc;
1447aae3547bSMateusz Guzik 		if (pid != -1 && p->p_pid != pid) {
1448cf7d9a8cSDavid Xu 			td = NULL;
1449cf7d9a8cSDavid Xu 			break;
1450cf7d9a8cSDavid Xu 		}
1451cf7d9a8cSDavid Xu 		if (run > RUN_THRESH) {
145226007fe3SMateusz Guzik 			if (rw_try_upgrade(TIDHASHLOCK(tid))) {
1453cf7d9a8cSDavid Xu 				LIST_REMOVE(td, td_hash);
1454cf7d9a8cSDavid Xu 				LIST_INSERT_HEAD(TIDHASH(td->td_tid),
1455cf7d9a8cSDavid Xu 					td, td_hash);
145626007fe3SMateusz Guzik 				rw_wunlock(TIDHASHLOCK(tid));
1457aae3547bSMateusz Guzik 				locked = false;
1458aae3547bSMateusz Guzik 				break;
1459cf7d9a8cSDavid Xu 			}
1460cf7d9a8cSDavid Xu 		}
1461cf7d9a8cSDavid Xu 		break;
1462cf7d9a8cSDavid Xu 	}
1463aae3547bSMateusz Guzik 	if (locked)
146426007fe3SMateusz Guzik 		rw_runlock(TIDHASHLOCK(tid));
1465aae3547bSMateusz Guzik 	if (td == NULL)
1466aae3547bSMateusz Guzik 		return (false);
1467aae3547bSMateusz Guzik 	*pp = p;
1468aae3547bSMateusz Guzik 	*tdp = td;
1469aae3547bSMateusz Guzik 	return (true);
1470aae3547bSMateusz Guzik }
1471aae3547bSMateusz Guzik 
1472aae3547bSMateusz Guzik struct thread *
1473aae3547bSMateusz Guzik tdfind(lwpid_t tid, pid_t pid)
1474aae3547bSMateusz Guzik {
1475aae3547bSMateusz Guzik 	struct proc *p;
1476aae3547bSMateusz Guzik 	struct thread *td;
1477aae3547bSMateusz Guzik 
1478aae3547bSMateusz Guzik 	td = curthread;
1479aae3547bSMateusz Guzik 	if (td->td_tid == tid) {
1480aae3547bSMateusz Guzik 		if (pid != -1 && td->td_proc->p_pid != pid)
1481aae3547bSMateusz Guzik 			return (NULL);
1482aae3547bSMateusz Guzik 		PROC_LOCK(td->td_proc);
1483cf7d9a8cSDavid Xu 		return (td);
1484cf7d9a8cSDavid Xu 	}
1485cf7d9a8cSDavid Xu 
1486aae3547bSMateusz Guzik 	for (;;) {
1487aae3547bSMateusz Guzik 		if (!tdfind_hash(tid, pid, &p, &td))
1488aae3547bSMateusz Guzik 			return (NULL);
1489aae3547bSMateusz Guzik 		PROC_LOCK(p);
1490aae3547bSMateusz Guzik 		if (td->td_tid != tid) {
1491aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1492aae3547bSMateusz Guzik 			continue;
1493aae3547bSMateusz Guzik 		}
1494aae3547bSMateusz Guzik 		if (td->td_proc != p) {
1495aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1496aae3547bSMateusz Guzik 			continue;
1497aae3547bSMateusz Guzik 		}
1498aae3547bSMateusz Guzik 		if (p->p_state == PRS_NEW) {
1499aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1500aae3547bSMateusz Guzik 			return (NULL);
1501aae3547bSMateusz Guzik 		}
1502aae3547bSMateusz Guzik 		return (td);
1503aae3547bSMateusz Guzik 	}
1504aae3547bSMateusz Guzik }
1505aae3547bSMateusz Guzik 
1506cf7d9a8cSDavid Xu void
1507cf7d9a8cSDavid Xu tidhash_add(struct thread *td)
1508cf7d9a8cSDavid Xu {
150926007fe3SMateusz Guzik 	rw_wlock(TIDHASHLOCK(td->td_tid));
1510cf7d9a8cSDavid Xu 	LIST_INSERT_HEAD(TIDHASH(td->td_tid), td, td_hash);
151126007fe3SMateusz Guzik 	rw_wunlock(TIDHASHLOCK(td->td_tid));
1512cf7d9a8cSDavid Xu }
1513cf7d9a8cSDavid Xu 
1514cf7d9a8cSDavid Xu void
1515cf7d9a8cSDavid Xu tidhash_remove(struct thread *td)
1516cf7d9a8cSDavid Xu {
151726007fe3SMateusz Guzik 
151826007fe3SMateusz Guzik 	rw_wlock(TIDHASHLOCK(td->td_tid));
1519cf7d9a8cSDavid Xu 	LIST_REMOVE(td, td_hash);
152026007fe3SMateusz Guzik 	rw_wunlock(TIDHASHLOCK(td->td_tid));
1521cf7d9a8cSDavid Xu }
1522