xref: /freebsd/sys/kern/kern_thread.c (revision 5ef7b7a0f3b9da88e336822003de8b88b445b4ad)
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 
138d1ca25beSMateusz 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 /*
262*5ef7b7a0SMateusz Guzik  * Batching for thread reapping.
263*5ef7b7a0SMateusz Guzik  */
264*5ef7b7a0SMateusz Guzik struct tidbatch {
265*5ef7b7a0SMateusz Guzik 	lwpid_t tab[16];
266*5ef7b7a0SMateusz Guzik 	int n;
267*5ef7b7a0SMateusz Guzik };
268*5ef7b7a0SMateusz Guzik 
269*5ef7b7a0SMateusz Guzik static void
270*5ef7b7a0SMateusz Guzik tidbatch_prep(struct tidbatch *tb)
271*5ef7b7a0SMateusz Guzik {
272*5ef7b7a0SMateusz Guzik 
273*5ef7b7a0SMateusz Guzik 	tb->n = 0;
274*5ef7b7a0SMateusz Guzik }
275*5ef7b7a0SMateusz Guzik 
276*5ef7b7a0SMateusz Guzik static void
277*5ef7b7a0SMateusz Guzik tidbatch_add(struct tidbatch *tb, struct thread *td)
278*5ef7b7a0SMateusz Guzik {
279*5ef7b7a0SMateusz Guzik 
280*5ef7b7a0SMateusz Guzik 	KASSERT(tb->n < nitems(tb->tab),
281*5ef7b7a0SMateusz Guzik 	    ("%s: count too high %d", __func__, tb->n));
282*5ef7b7a0SMateusz Guzik 	tb->tab[tb->n] = td->td_tid;
283*5ef7b7a0SMateusz Guzik 	tb->n++;
284*5ef7b7a0SMateusz Guzik }
285*5ef7b7a0SMateusz Guzik 
286*5ef7b7a0SMateusz Guzik static void
287*5ef7b7a0SMateusz Guzik tidbatch_process(struct tidbatch *tb)
288*5ef7b7a0SMateusz Guzik {
289*5ef7b7a0SMateusz Guzik 
290*5ef7b7a0SMateusz Guzik 	KASSERT(tb->n <= nitems(tb->tab),
291*5ef7b7a0SMateusz Guzik 	    ("%s: count too high %d", __func__, tb->n));
292*5ef7b7a0SMateusz Guzik 	if (tb->n == nitems(tb->tab)) {
293*5ef7b7a0SMateusz Guzik 		tid_free_batch(tb->tab, tb->n);
294*5ef7b7a0SMateusz Guzik 		tb->n = 0;
295*5ef7b7a0SMateusz Guzik 	}
296*5ef7b7a0SMateusz Guzik }
297*5ef7b7a0SMateusz Guzik 
298*5ef7b7a0SMateusz Guzik static void
299*5ef7b7a0SMateusz Guzik tidbatch_final(struct tidbatch *tb)
300*5ef7b7a0SMateusz Guzik {
301*5ef7b7a0SMateusz Guzik 
302*5ef7b7a0SMateusz Guzik 	KASSERT(tb->n <= nitems(tb->tab),
303*5ef7b7a0SMateusz Guzik 	    ("%s: count too high %d", __func__, tb->n));
304*5ef7b7a0SMateusz Guzik 	if (tb->n != 0) {
305*5ef7b7a0SMateusz Guzik 		tid_free_batch(tb->tab, tb->n);
306*5ef7b7a0SMateusz Guzik 	}
307*5ef7b7a0SMateusz Guzik }
308*5ef7b7a0SMateusz Guzik 
309*5ef7b7a0SMateusz Guzik /*
310696058c3SJulian Elischer  * Prepare a thread for use.
31144990b8cSJulian Elischer  */
312b23f72e9SBrian Feldman static int
313b23f72e9SBrian Feldman thread_ctor(void *mem, int size, void *arg, int flags)
31444990b8cSJulian Elischer {
31544990b8cSJulian Elischer 	struct thread	*td;
31644990b8cSJulian Elischer 
31744990b8cSJulian Elischer 	td = (struct thread *)mem;
31871fad9fdSJulian Elischer 	td->td_state = TDS_INACTIVE;
31994dd54b9SKonstantin Belousov 	td->td_lastcpu = td->td_oncpu = NOCPU;
3206c27c603SJuli Mallett 
3216c27c603SJuli Mallett 	/*
3226c27c603SJuli Mallett 	 * Note that td_critnest begins life as 1 because the thread is not
3236c27c603SJuli Mallett 	 * running and is thereby implicitly waiting to be on the receiving
324a54e85fdSJeff Roberson 	 * end of a context switch.
3256c27c603SJuli Mallett 	 */
326139b7550SJohn Baldwin 	td->td_critnest = 1;
327acbe332aSDavid Xu 	td->td_lend_user_pri = PRI_MAX;
328911b84b0SRobert Watson #ifdef AUDIT
329911b84b0SRobert Watson 	audit_thread_alloc(td);
330911b84b0SRobert Watson #endif
331d10183d9SDavid Xu 	umtx_thread_alloc(td);
332b23f72e9SBrian Feldman 	return (0);
33344990b8cSJulian Elischer }
33444990b8cSJulian Elischer 
33544990b8cSJulian Elischer /*
33644990b8cSJulian Elischer  * Reclaim a thread after use.
33744990b8cSJulian Elischer  */
33844990b8cSJulian Elischer static void
33944990b8cSJulian Elischer thread_dtor(void *mem, int size, void *arg)
34044990b8cSJulian Elischer {
34144990b8cSJulian Elischer 	struct thread *td;
34244990b8cSJulian Elischer 
34344990b8cSJulian Elischer 	td = (struct thread *)mem;
34444990b8cSJulian Elischer 
34544990b8cSJulian Elischer #ifdef INVARIANTS
34644990b8cSJulian Elischer 	/* Verify that this thread is in a safe state to free. */
34744990b8cSJulian Elischer 	switch (td->td_state) {
34871fad9fdSJulian Elischer 	case TDS_INHIBITED:
34971fad9fdSJulian Elischer 	case TDS_RUNNING:
35071fad9fdSJulian Elischer 	case TDS_CAN_RUN:
35144990b8cSJulian Elischer 	case TDS_RUNQ:
35244990b8cSJulian Elischer 		/*
35344990b8cSJulian Elischer 		 * We must never unlink a thread that is in one of
35444990b8cSJulian Elischer 		 * these states, because it is currently active.
35544990b8cSJulian Elischer 		 */
35644990b8cSJulian Elischer 		panic("bad state for thread unlinking");
35744990b8cSJulian Elischer 		/* NOTREACHED */
35871fad9fdSJulian Elischer 	case TDS_INACTIVE:
35944990b8cSJulian Elischer 		break;
36044990b8cSJulian Elischer 	default:
36144990b8cSJulian Elischer 		panic("bad thread state");
36244990b8cSJulian Elischer 		/* NOTREACHED */
36344990b8cSJulian Elischer 	}
36444990b8cSJulian Elischer #endif
3656e8525ceSRobert Watson #ifdef AUDIT
3666e8525ceSRobert Watson 	audit_thread_free(td);
3676e8525ceSRobert Watson #endif
3681ba4a712SPawel Jakub Dawidek 	/* Free all OSD associated to this thread. */
3691ba4a712SPawel Jakub Dawidek 	osd_thread_exit(td);
370aca4bb91SKonstantin Belousov 	td_softdep_cleanup(td);
371aca4bb91SKonstantin Belousov 	MPASS(td->td_su == NULL);
37244990b8cSJulian Elischer }
37344990b8cSJulian Elischer 
37444990b8cSJulian Elischer /*
37544990b8cSJulian Elischer  * Initialize type-stable parts of a thread (when newly created).
37644990b8cSJulian Elischer  */
377b23f72e9SBrian Feldman static int
378b23f72e9SBrian Feldman thread_init(void *mem, int size, int flags)
37944990b8cSJulian Elischer {
38044990b8cSJulian Elischer 	struct thread *td;
38144990b8cSJulian Elischer 
38244990b8cSJulian Elischer 	td = (struct thread *)mem;
383247aba24SMarcel Moolenaar 
38444f3b092SJohn Baldwin 	td->td_sleepqueue = sleepq_alloc();
385961a7b24SJohn Baldwin 	td->td_turnstile = turnstile_alloc();
3868f0e9130SKonstantin Belousov 	td->td_rlqe = NULL;
3872ca45184SMatt Joras 	EVENTHANDLER_DIRECT_INVOKE(thread_init, td);
388d10183d9SDavid Xu 	umtx_thread_init(td);
38989b57fcfSKonstantin Belousov 	td->td_kstack = 0;
390ad8b1d85SKonstantin Belousov 	td->td_sel = NULL;
391b23f72e9SBrian Feldman 	return (0);
39244990b8cSJulian Elischer }
39344990b8cSJulian Elischer 
39444990b8cSJulian Elischer /*
39544990b8cSJulian Elischer  * Tear down type-stable parts of a thread (just before being discarded).
39644990b8cSJulian Elischer  */
39744990b8cSJulian Elischer static void
39844990b8cSJulian Elischer thread_fini(void *mem, int size)
39944990b8cSJulian Elischer {
40044990b8cSJulian Elischer 	struct thread *td;
40144990b8cSJulian Elischer 
40244990b8cSJulian Elischer 	td = (struct thread *)mem;
4032ca45184SMatt Joras 	EVENTHANDLER_DIRECT_INVOKE(thread_fini, td);
4048f0e9130SKonstantin Belousov 	rlqentry_free(td->td_rlqe);
405961a7b24SJohn Baldwin 	turnstile_free(td->td_turnstile);
40644f3b092SJohn Baldwin 	sleepq_free(td->td_sleepqueue);
407d10183d9SDavid Xu 	umtx_thread_fini(td);
408ace8398dSJeff Roberson 	seltdfini(td);
40944990b8cSJulian Elischer }
4105215b187SJeff Roberson 
4115c8329edSJulian Elischer /*
4125215b187SJeff Roberson  * For a newly created process,
4135215b187SJeff Roberson  * link up all the structures and its initial threads etc.
414ed062c8dSJulian Elischer  * called from:
415e7d939bdSMarcel Moolenaar  * {arch}/{arch}/machdep.c   {arch}_init(), init386() etc.
416ed062c8dSJulian Elischer  * proc_dtor() (should go away)
417ed062c8dSJulian Elischer  * proc_init()
4185c8329edSJulian Elischer  */
4195c8329edSJulian Elischer void
42089b57fcfSKonstantin Belousov proc_linkup0(struct proc *p, struct thread *td)
42189b57fcfSKonstantin Belousov {
42289b57fcfSKonstantin Belousov 	TAILQ_INIT(&p->p_threads);	     /* all threads in proc */
42389b57fcfSKonstantin Belousov 	proc_linkup(p, td);
42489b57fcfSKonstantin Belousov }
42589b57fcfSKonstantin Belousov 
42689b57fcfSKonstantin Belousov void
4278460a577SJohn Birrell proc_linkup(struct proc *p, struct thread *td)
4285c8329edSJulian Elischer {
429a54e85fdSJeff Roberson 
4309104847fSDavid Xu 	sigqueue_init(&p->p_sigqueue, p);
431ebceaf6dSDavid Xu 	p->p_ksi = ksiginfo_alloc(1);
432ebceaf6dSDavid Xu 	if (p->p_ksi != NULL) {
4335c474517SDavid Xu 		/* XXX p_ksi may be null if ksiginfo zone is not ready */
434ebceaf6dSDavid Xu 		p->p_ksi->ksi_flags = KSI_EXT | KSI_INS;
435ebceaf6dSDavid Xu 	}
436b2f92ef9SDavid Xu 	LIST_INIT(&p->p_mqnotifier);
4375c8329edSJulian Elischer 	p->p_numthreads = 0;
4388460a577SJohn Birrell 	thread_link(td, p);
4395c8329edSJulian Elischer }
4405c8329edSJulian Elischer 
4411bd3cf5dSMateusz Guzik extern int max_threads_per_proc;
4421bd3cf5dSMateusz Guzik 
4435c8329edSJulian Elischer /*
44444990b8cSJulian Elischer  * Initialize global thread allocation resources.
44544990b8cSJulian Elischer  */
44644990b8cSJulian Elischer void
44744990b8cSJulian Elischer threadinit(void)
44844990b8cSJulian Elischer {
44926007fe3SMateusz Guzik 	u_long i;
450cf31cadeSMateusz Guzik 	lwpid_t tid0;
4515aa5420fSMark Johnston 	uint32_t flags;
45244990b8cSJulian Elischer 
4531bd3cf5dSMateusz Guzik 	/*
4541bd3cf5dSMateusz Guzik 	 * Place an upper limit on threads which can be allocated.
4551bd3cf5dSMateusz Guzik 	 *
4561bd3cf5dSMateusz Guzik 	 * Note that other factors may make the de facto limit much lower.
4571bd3cf5dSMateusz Guzik 	 *
4581bd3cf5dSMateusz Guzik 	 * Platform limits are somewhat arbitrary but deemed "more than good
4591bd3cf5dSMateusz Guzik 	 * enough" for the foreseable future.
4601bd3cf5dSMateusz Guzik 	 */
4611bd3cf5dSMateusz Guzik 	if (maxthread == 0) {
4621bd3cf5dSMateusz Guzik #ifdef _LP64
4631bd3cf5dSMateusz Guzik 		maxthread = MIN(maxproc * max_threads_per_proc, 1000000);
4641bd3cf5dSMateusz Guzik #else
4651bd3cf5dSMateusz Guzik 		maxthread = MIN(maxproc * max_threads_per_proc, 100000);
4661bd3cf5dSMateusz Guzik #endif
4671bd3cf5dSMateusz Guzik 	}
4681bd3cf5dSMateusz Guzik 
4691ea7a6f8SPoul-Henning Kamp 	mtx_init(&tid_lock, "TID lock", NULL, MTX_DEF);
47035bb59edSMateusz Guzik 	tid_bitmap = bit_alloc(maxthread, M_TIDHASH, M_WAITOK);
47162dbc992SMateusz Guzik 	/*
47262dbc992SMateusz Guzik 	 * Handle thread0.
47362dbc992SMateusz Guzik 	 */
47462dbc992SMateusz Guzik 	thread_count_inc();
475cf31cadeSMateusz Guzik 	tid0 = tid_alloc();
476cf31cadeSMateusz Guzik 	if (tid0 != THREAD0_TID)
477cf31cadeSMateusz Guzik 		panic("tid0 %d != %d\n", tid0, THREAD0_TID);
4781ea7a6f8SPoul-Henning Kamp 
4795aa5420fSMark Johnston 	flags = UMA_ZONE_NOFREE;
4805aa5420fSMark Johnston #ifdef __aarch64__
4815aa5420fSMark Johnston 	/*
4825aa5420fSMark Johnston 	 * Force thread structures to be allocated from the direct map.
4835aa5420fSMark Johnston 	 * Otherwise, superpage promotions and demotions may temporarily
4845aa5420fSMark Johnston 	 * invalidate thread structure mappings.  For most dynamically allocated
4855aa5420fSMark Johnston 	 * structures this is not a problem, but translation faults cannot be
4865aa5420fSMark Johnston 	 * handled without accessing curthread.
4875aa5420fSMark Johnston 	 */
4885aa5420fSMark Johnston 	flags |= UMA_ZONE_CONTIG;
4895aa5420fSMark Johnston #endif
490de028f5aSJeff Roberson 	thread_zone = uma_zcreate("THREAD", sched_sizeof_thread(),
49144990b8cSJulian Elischer 	    thread_ctor, thread_dtor, thread_init, thread_fini,
4925aa5420fSMark Johnston 	    32 - 1, flags);
493cf7d9a8cSDavid Xu 	tidhashtbl = hashinit(maxproc / 2, M_TIDHASH, &tidhash);
49426007fe3SMateusz Guzik 	tidhashlock = (tidhash + 1) / 64;
49526007fe3SMateusz Guzik 	if (tidhashlock > 0)
49626007fe3SMateusz Guzik 		tidhashlock--;
49726007fe3SMateusz Guzik 	tidhashtbl_lock = malloc(sizeof(*tidhashtbl_lock) * (tidhashlock + 1),
49826007fe3SMateusz Guzik 	    M_TIDHASH, M_WAITOK | M_ZERO);
49926007fe3SMateusz Guzik 	for (i = 0; i < tidhashlock + 1; i++)
50026007fe3SMateusz Guzik 		rw_init(&tidhashtbl_lock[i], "tidhash");
50144990b8cSJulian Elischer }
50244990b8cSJulian Elischer 
50344990b8cSJulian Elischer /*
504ff8fbcffSJeff Roberson  * Place an unused thread on the zombie list.
50544990b8cSJulian Elischer  */
50644990b8cSJulian Elischer void
507ff8fbcffSJeff Roberson thread_zombie(struct thread *td)
50844990b8cSJulian Elischer {
509c5315f51SMateusz Guzik 	struct thread *ztd;
510c5315f51SMateusz Guzik 
511c5315f51SMateusz Guzik 	ztd = atomic_load_ptr(&thread_zombies);
512c5315f51SMateusz Guzik 	for (;;) {
513c5315f51SMateusz Guzik 		td->td_zombie = ztd;
514c5315f51SMateusz Guzik 		if (atomic_fcmpset_rel_ptr((uintptr_t *)&thread_zombies,
515c5315f51SMateusz Guzik 		    (uintptr_t *)&ztd, (uintptr_t)td))
516c5315f51SMateusz Guzik 			break;
517c5315f51SMateusz Guzik 		continue;
518c5315f51SMateusz Guzik 	}
51944990b8cSJulian Elischer }
52044990b8cSJulian Elischer 
5215c8329edSJulian Elischer /*
522ff8fbcffSJeff Roberson  * Release a thread that has exited after cpu_throw().
523ff8fbcffSJeff Roberson  */
524ff8fbcffSJeff Roberson void
525ff8fbcffSJeff Roberson thread_stash(struct thread *td)
526ff8fbcffSJeff Roberson {
527ff8fbcffSJeff Roberson 	atomic_subtract_rel_int(&td->td_proc->p_exitthreads, 1);
528ff8fbcffSJeff Roberson 	thread_zombie(td);
529ff8fbcffSJeff Roberson }
530ff8fbcffSJeff Roberson 
531ff8fbcffSJeff Roberson /*
532c5315f51SMateusz Guzik  * Reap zombie threads.
53344990b8cSJulian Elischer  */
53444990b8cSJulian Elischer void
53544990b8cSJulian Elischer thread_reap(void)
53644990b8cSJulian Elischer {
537c5315f51SMateusz Guzik 	struct thread *itd, *ntd;
538*5ef7b7a0SMateusz Guzik 	struct tidbatch tidbatch;
539*5ef7b7a0SMateusz Guzik 	int tdcount;
54044990b8cSJulian Elischer 
54144990b8cSJulian Elischer 	/*
542c5315f51SMateusz Guzik 	 * Reading upfront is pessimal if followed by concurrent atomic_swap,
543c5315f51SMateusz Guzik 	 * but most of the time the list is empty.
54444990b8cSJulian Elischer 	 */
545c5315f51SMateusz Guzik 	if (thread_zombies == NULL)
546c5315f51SMateusz Guzik 		return;
547c5315f51SMateusz Guzik 
548c5315f51SMateusz Guzik 	itd = (struct thread *)atomic_swap_ptr((uintptr_t *)&thread_zombies,
549c5315f51SMateusz Guzik 	    (uintptr_t)NULL);
550*5ef7b7a0SMateusz Guzik 	if (itd == NULL)
551*5ef7b7a0SMateusz Guzik 		return;
552*5ef7b7a0SMateusz Guzik 
553*5ef7b7a0SMateusz Guzik 	tidbatch_prep(&tidbatch);
554*5ef7b7a0SMateusz Guzik 	tdcount = 0;
555c5315f51SMateusz Guzik 	while (itd != NULL) {
556c5315f51SMateusz Guzik 		ntd = itd->td_zombie;
557*5ef7b7a0SMateusz Guzik 		EVENTHANDLER_DIRECT_INVOKE(thread_dtor, itd);
558*5ef7b7a0SMateusz Guzik 		tidbatch_add(&tidbatch, itd);
559c5315f51SMateusz Guzik 		thread_cow_free(itd);
560755341dfSMateusz Guzik 		thread_free_batched(itd);
561*5ef7b7a0SMateusz Guzik 		tidbatch_process(&tidbatch);
562*5ef7b7a0SMateusz Guzik 		tdcount++;
563*5ef7b7a0SMateusz Guzik 		if (tdcount == 32) {
564*5ef7b7a0SMateusz Guzik 			thread_count_sub(tdcount);
565*5ef7b7a0SMateusz Guzik 			tdcount = 0;
566755341dfSMateusz Guzik 		}
567c5315f51SMateusz Guzik 		itd = ntd;
56844990b8cSJulian Elischer 	}
569755341dfSMateusz Guzik 
570*5ef7b7a0SMateusz Guzik 	tidbatch_final(&tidbatch);
571*5ef7b7a0SMateusz Guzik 	if (tdcount != 0) {
572*5ef7b7a0SMateusz Guzik 		thread_count_sub(tdcount);
573755341dfSMateusz Guzik 	}
574ed062c8dSJulian Elischer }
57544990b8cSJulian Elischer 
5764f0db5e0SJulian Elischer /*
57744990b8cSJulian Elischer  * Allocate a thread.
57844990b8cSJulian Elischer  */
57944990b8cSJulian Elischer struct thread *
5808a945d10SKonstantin Belousov thread_alloc(int pages)
58144990b8cSJulian Elischer {
58289b57fcfSKonstantin Belousov 	struct thread *td;
5831bd3cf5dSMateusz Guzik 	lwpid_t tid;
5848460a577SJohn Birrell 
58562dbc992SMateusz Guzik 	if (!thread_count_inc()) {
5861bd3cf5dSMateusz Guzik 		return (NULL);
5871bd3cf5dSMateusz Guzik 	}
5881bd3cf5dSMateusz Guzik 
58962dbc992SMateusz Guzik 	tid = tid_alloc();
5901bd3cf5dSMateusz Guzik 	td = uma_zalloc(thread_zone, M_WAITOK);
59189b57fcfSKonstantin Belousov 	KASSERT(td->td_kstack == 0, ("thread_alloc got thread with kstack"));
5928a945d10SKonstantin Belousov 	if (!vm_thread_new(td, pages)) {
59389b57fcfSKonstantin Belousov 		uma_zfree(thread_zone, td);
5941bd3cf5dSMateusz Guzik 		tid_free(tid);
59562dbc992SMateusz Guzik 		thread_count_dec();
59689b57fcfSKonstantin Belousov 		return (NULL);
59789b57fcfSKonstantin Belousov 	}
5981bd3cf5dSMateusz Guzik 	td->td_tid = tid;
5990c3967e7SMarcel Moolenaar 	cpu_thread_alloc(td);
6001bd3cf5dSMateusz Guzik 	EVENTHANDLER_DIRECT_INVOKE(thread_ctor, td);
60189b57fcfSKonstantin Belousov 	return (td);
60244990b8cSJulian Elischer }
60344990b8cSJulian Elischer 
6048a945d10SKonstantin Belousov int
6058a945d10SKonstantin Belousov thread_alloc_stack(struct thread *td, int pages)
6068a945d10SKonstantin Belousov {
6078a945d10SKonstantin Belousov 
6088a945d10SKonstantin Belousov 	KASSERT(td->td_kstack == 0,
6098a945d10SKonstantin Belousov 	    ("thread_alloc_stack called on a thread with kstack"));
6108a945d10SKonstantin Belousov 	if (!vm_thread_new(td, pages))
6118a945d10SKonstantin Belousov 		return (0);
6128a945d10SKonstantin Belousov 	cpu_thread_alloc(td);
6138a945d10SKonstantin Belousov 	return (1);
6148a945d10SKonstantin Belousov }
6154f0db5e0SJulian Elischer 
6164f0db5e0SJulian Elischer /*
61744990b8cSJulian Elischer  * Deallocate a thread.
61844990b8cSJulian Elischer  */
619755341dfSMateusz Guzik static void
620755341dfSMateusz Guzik thread_free_batched(struct thread *td)
62144990b8cSJulian Elischer {
6222e6b8de4SJeff Roberson 
6232e6b8de4SJeff Roberson 	lock_profile_thread_exit(td);
62445aea8deSJeff Roberson 	if (td->td_cpuset)
625d7f687fcSJeff Roberson 		cpuset_rel(td->td_cpuset);
626d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
6270c3967e7SMarcel Moolenaar 	cpu_thread_free(td);
62889b57fcfSKonstantin Belousov 	if (td->td_kstack != 0)
62989b57fcfSKonstantin Belousov 		vm_thread_dispose(td);
6302d19b736SKonstantin Belousov 	callout_drain(&td->td_slpcallout);
631755341dfSMateusz Guzik 	/*
632755341dfSMateusz Guzik 	 * Freeing handled by the caller.
633755341dfSMateusz Guzik 	 */
6341bd3cf5dSMateusz Guzik 	td->td_tid = -1;
63544990b8cSJulian Elischer 	uma_zfree(thread_zone, td);
63644990b8cSJulian Elischer }
63744990b8cSJulian Elischer 
6384ea6a9a2SMateusz Guzik void
639755341dfSMateusz Guzik thread_free(struct thread *td)
640755341dfSMateusz Guzik {
641755341dfSMateusz Guzik 	lwpid_t tid;
642755341dfSMateusz Guzik 
643*5ef7b7a0SMateusz Guzik 	EVENTHANDLER_DIRECT_INVOKE(thread_dtor, td);
644755341dfSMateusz Guzik 	tid = td->td_tid;
645755341dfSMateusz Guzik 	thread_free_batched(td);
646755341dfSMateusz Guzik 	tid_free(tid);
64762dbc992SMateusz Guzik 	thread_count_dec();
648755341dfSMateusz Guzik }
649755341dfSMateusz Guzik 
650755341dfSMateusz Guzik void
6514ea6a9a2SMateusz Guzik thread_cow_get_proc(struct thread *newtd, struct proc *p)
6524ea6a9a2SMateusz Guzik {
6534ea6a9a2SMateusz Guzik 
6544ea6a9a2SMateusz Guzik 	PROC_LOCK_ASSERT(p, MA_OWNED);
6551724c563SMateusz Guzik 	newtd->td_realucred = crcowget(p->p_ucred);
6561724c563SMateusz Guzik 	newtd->td_ucred = newtd->td_realucred;
657f6f6d240SMateusz Guzik 	newtd->td_limit = lim_hold(p->p_limit);
6584ea6a9a2SMateusz Guzik 	newtd->td_cowgen = p->p_cowgen;
6594ea6a9a2SMateusz Guzik }
6604ea6a9a2SMateusz Guzik 
6614ea6a9a2SMateusz Guzik void
6624ea6a9a2SMateusz Guzik thread_cow_get(struct thread *newtd, struct thread *td)
6634ea6a9a2SMateusz Guzik {
6644ea6a9a2SMateusz Guzik 
6651724c563SMateusz Guzik 	MPASS(td->td_realucred == td->td_ucred);
6661724c563SMateusz Guzik 	newtd->td_realucred = crcowget(td->td_realucred);
6671724c563SMateusz Guzik 	newtd->td_ucred = newtd->td_realucred;
668f6f6d240SMateusz Guzik 	newtd->td_limit = lim_hold(td->td_limit);
6694ea6a9a2SMateusz Guzik 	newtd->td_cowgen = td->td_cowgen;
6704ea6a9a2SMateusz Guzik }
6714ea6a9a2SMateusz Guzik 
6724ea6a9a2SMateusz Guzik void
6734ea6a9a2SMateusz Guzik thread_cow_free(struct thread *td)
6744ea6a9a2SMateusz Guzik {
6754ea6a9a2SMateusz Guzik 
6761724c563SMateusz Guzik 	if (td->td_realucred != NULL)
6771724c563SMateusz Guzik 		crcowfree(td);
678cd672ca6SMateusz Guzik 	if (td->td_limit != NULL)
679f6f6d240SMateusz Guzik 		lim_free(td->td_limit);
6804ea6a9a2SMateusz Guzik }
6814ea6a9a2SMateusz Guzik 
6824ea6a9a2SMateusz Guzik void
6834ea6a9a2SMateusz Guzik thread_cow_update(struct thread *td)
6844ea6a9a2SMateusz Guzik {
6854ea6a9a2SMateusz Guzik 	struct proc *p;
686cd672ca6SMateusz Guzik 	struct ucred *oldcred;
687cd672ca6SMateusz Guzik 	struct plimit *oldlimit;
6884ea6a9a2SMateusz Guzik 
6894ea6a9a2SMateusz Guzik 	p = td->td_proc;
690cd672ca6SMateusz Guzik 	oldlimit = NULL;
6914ea6a9a2SMateusz Guzik 	PROC_LOCK(p);
6921724c563SMateusz Guzik 	oldcred = crcowsync();
693cd672ca6SMateusz Guzik 	if (td->td_limit != p->p_limit) {
694cd672ca6SMateusz Guzik 		oldlimit = td->td_limit;
695cd672ca6SMateusz Guzik 		td->td_limit = lim_hold(p->p_limit);
696cd672ca6SMateusz Guzik 	}
6974ea6a9a2SMateusz Guzik 	td->td_cowgen = p->p_cowgen;
6984ea6a9a2SMateusz Guzik 	PROC_UNLOCK(p);
699cd672ca6SMateusz Guzik 	if (oldcred != NULL)
700cd672ca6SMateusz Guzik 		crfree(oldcred);
701cd672ca6SMateusz Guzik 	if (oldlimit != NULL)
702cd672ca6SMateusz Guzik 		lim_free(oldlimit);
7034ea6a9a2SMateusz Guzik }
7044ea6a9a2SMateusz Guzik 
70544990b8cSJulian Elischer /*
70644990b8cSJulian Elischer  * Discard the current thread and exit from its context.
70794e0a4cdSJulian Elischer  * Always called with scheduler locked.
70844990b8cSJulian Elischer  *
70944990b8cSJulian Elischer  * Because we can't free a thread while we're operating under its context,
710696058c3SJulian Elischer  * push the current thread into our CPU's deadthread holder. This means
711696058c3SJulian Elischer  * we needn't worry about someone else grabbing our context before we
7126617724cSJeff Roberson  * do a cpu_throw().
71344990b8cSJulian Elischer  */
71444990b8cSJulian Elischer void
71544990b8cSJulian Elischer thread_exit(void)
71644990b8cSJulian Elischer {
7177e3a96eaSJohn Baldwin 	uint64_t runtime, new_switchtime;
71844990b8cSJulian Elischer 	struct thread *td;
7191c4bcd05SJeff Roberson 	struct thread *td2;
72044990b8cSJulian Elischer 	struct proc *p;
7217847a9daSJohn Baldwin 	int wakeup_swapper;
72244990b8cSJulian Elischer 
72344990b8cSJulian Elischer 	td = curthread;
72444990b8cSJulian Elischer 	p = td->td_proc;
72544990b8cSJulian Elischer 
726a54e85fdSJeff Roberson 	PROC_SLOCK_ASSERT(p, MA_OWNED);
727ed062c8dSJulian Elischer 	mtx_assert(&Giant, MA_NOTOWNED);
728a54e85fdSJeff Roberson 
72944990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
730ed062c8dSJulian Elischer 	KASSERT(p != NULL, ("thread exiting without a process"));
731cc701b73SRobert Watson 	CTR3(KTR_PROC, "thread_exit: thread %p (pid %ld, %s)", td,
732e01eafefSJulian Elischer 	    (long)p->p_pid, td->td_name);
7336c9271a9SAndriy Gapon 	SDT_PROBE0(proc, , , lwp__exit);
7349104847fSDavid Xu 	KASSERT(TAILQ_EMPTY(&td->td_sigqueue.sq_list), ("signal pending"));
735936c24faSMateusz Guzik 	MPASS(td->td_realucred == td->td_ucred);
73644990b8cSJulian Elischer 
737ed062c8dSJulian Elischer 	/*
738ed062c8dSJulian Elischer 	 * drop FPU & debug register state storage, or any other
739ed062c8dSJulian Elischer 	 * architecture specific resources that
740ed062c8dSJulian Elischer 	 * would not be on a new untouched process.
741ed062c8dSJulian Elischer 	 */
742bd07998eSKonstantin Belousov 	cpu_thread_exit(td);
74344990b8cSJulian Elischer 
744ed062c8dSJulian Elischer 	/*
7451faf202eSJulian Elischer 	 * The last thread is left attached to the process
7461faf202eSJulian Elischer 	 * So that the whole bundle gets recycled. Skip
747ed062c8dSJulian Elischer 	 * all this stuff if we never had threads.
748ed062c8dSJulian Elischer 	 * EXIT clears all sign of other threads when
749ed062c8dSJulian Elischer 	 * it goes to single threading, so the last thread always
750ed062c8dSJulian Elischer 	 * takes the short path.
7511faf202eSJulian Elischer 	 */
752ed062c8dSJulian Elischer 	if (p->p_flag & P_HADTHREADS) {
7531faf202eSJulian Elischer 		if (p->p_numthreads > 1) {
754fd229b5bSKonstantin Belousov 			atomic_add_int(&td->td_proc->p_exitthreads, 1);
755d3a0bd78SJulian Elischer 			thread_unlink(td);
7561c4bcd05SJeff Roberson 			td2 = FIRST_THREAD_IN_PROC(p);
7571c4bcd05SJeff Roberson 			sched_exit_thread(td2, td);
758ed062c8dSJulian Elischer 
759ed062c8dSJulian Elischer 			/*
76044990b8cSJulian Elischer 			 * The test below is NOT true if we are the
7619182554aSKonstantin Belousov 			 * sole exiting thread. P_STOPPED_SINGLE is unset
76244990b8cSJulian Elischer 			 * in exit1() after it is the only survivor.
76344990b8cSJulian Elischer 			 */
7641279572aSDavid Xu 			if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
76544990b8cSJulian Elischer 				if (p->p_numthreads == p->p_suspcount) {
766a54e85fdSJeff Roberson 					thread_lock(p->p_singlethread);
7677847a9daSJohn Baldwin 					wakeup_swapper = thread_unsuspend_one(
76884cdea97SKonstantin Belousov 						p->p_singlethread, p, false);
7697847a9daSJohn Baldwin 					if (wakeup_swapper)
7707847a9daSJohn Baldwin 						kick_proc0();
77144990b8cSJulian Elischer 				}
77244990b8cSJulian Elischer 			}
77348bfcdddSJulian Elischer 
774696058c3SJulian Elischer 			PCPU_SET(deadthread, td);
7751faf202eSJulian Elischer 		} else {
776ed062c8dSJulian Elischer 			/*
777ed062c8dSJulian Elischer 			 * The last thread is exiting.. but not through exit()
778ed062c8dSJulian Elischer 			 */
779ed062c8dSJulian Elischer 			panic ("thread_exit: Last thread exiting on its own");
780ed062c8dSJulian Elischer 		}
7811faf202eSJulian Elischer 	}
78216d95d4fSJoseph Koshy #ifdef	HWPMC_HOOKS
78316d95d4fSJoseph Koshy 	/*
78416d95d4fSJoseph Koshy 	 * If this thread is part of a process that is being tracked by hwpmc(4),
78516d95d4fSJoseph Koshy 	 * inform the module of the thread's impending exit.
78616d95d4fSJoseph Koshy 	 */
7876161b98cSMatt Macy 	if (PMC_PROC_IS_USING_PMCS(td->td_proc)) {
78816d95d4fSJoseph Koshy 		PMC_SWITCH_CONTEXT(td, PMC_FN_CSW_OUT);
7896161b98cSMatt Macy 		PMC_CALL_HOOK_UNLOCKED(td, PMC_FN_THR_EXIT, NULL);
790ebfaf69cSMatt Macy 	} else if (PMC_SYSTEM_SAMPLING_ACTIVE())
791ebfaf69cSMatt Macy 		PMC_CALL_HOOK_UNLOCKED(td, PMC_FN_THR_EXIT_LOG, NULL);
79216d95d4fSJoseph Koshy #endif
793a54e85fdSJeff Roberson 	PROC_UNLOCK(p);
7945c7bebf9SKonstantin Belousov 	PROC_STATLOCK(p);
7955c7bebf9SKonstantin Belousov 	thread_lock(td);
7965c7bebf9SKonstantin Belousov 	PROC_SUNLOCK(p);
7977e3a96eaSJohn Baldwin 
7987e3a96eaSJohn Baldwin 	/* Do the same timestamp bookkeeping that mi_switch() would do. */
7997e3a96eaSJohn Baldwin 	new_switchtime = cpu_ticks();
8007e3a96eaSJohn Baldwin 	runtime = new_switchtime - PCPU_GET(switchtime);
8017e3a96eaSJohn Baldwin 	td->td_runtime += runtime;
8027e3a96eaSJohn Baldwin 	td->td_incruntime += runtime;
8037e3a96eaSJohn Baldwin 	PCPU_SET(switchtime, new_switchtime);
8047e3a96eaSJohn Baldwin 	PCPU_SET(switchticks, ticks);
80583c9dea1SGleb Smirnoff 	VM_CNT_INC(v_swtch);
8067e3a96eaSJohn Baldwin 
8077e3a96eaSJohn Baldwin 	/* Save our resource usage in our process. */
8087e3a96eaSJohn Baldwin 	td->td_ru.ru_nvcsw++;
80961a74c5cSJeff Roberson 	ruxagg_locked(p, td);
8107e3a96eaSJohn Baldwin 	rucollect(&p->p_ru, &td->td_ru);
8115c7bebf9SKonstantin Belousov 	PROC_STATUNLOCK(p);
8127e3a96eaSJohn Baldwin 
813dcc9954eSJulian Elischer 	td->td_state = TDS_INACTIVE;
8143d06b4b3SAttilio Rao #ifdef WITNESS
8153d06b4b3SAttilio Rao 	witness_thread_exit(td);
8163d06b4b3SAttilio Rao #endif
817732d9528SJulian Elischer 	CTR1(KTR_PROC, "thread_exit: cpu_throw() thread %p", td);
818a54e85fdSJeff Roberson 	sched_throw(td);
819cc66ebe2SPeter Wemm 	panic("I'm a teapot!");
82044990b8cSJulian Elischer 	/* NOTREACHED */
82144990b8cSJulian Elischer }
82244990b8cSJulian Elischer 
82344990b8cSJulian Elischer /*
824696058c3SJulian Elischer  * Do any thread specific cleanups that may be needed in wait()
82537814395SPeter Wemm  * called with Giant, proc and schedlock not held.
826696058c3SJulian Elischer  */
827696058c3SJulian Elischer void
828696058c3SJulian Elischer thread_wait(struct proc *p)
829696058c3SJulian Elischer {
830696058c3SJulian Elischer 	struct thread *td;
831696058c3SJulian Elischer 
83237814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
833624bf9e1SKonstantin Belousov 	KASSERT(p->p_numthreads == 1, ("multiple threads in thread_wait()"));
834624bf9e1SKonstantin Belousov 	KASSERT(p->p_exitthreads == 0, ("p_exitthreads leaking"));
835ff8fbcffSJeff Roberson 	td = FIRST_THREAD_IN_PROC(p);
836ff8fbcffSJeff Roberson 	/* Lock the last thread so we spin until it exits cpu_throw(). */
837ff8fbcffSJeff Roberson 	thread_lock(td);
838ff8fbcffSJeff Roberson 	thread_unlock(td);
8392e6b8de4SJeff Roberson 	lock_profile_thread_exit(td);
840d7f687fcSJeff Roberson 	cpuset_rel(td->td_cpuset);
841d7f687fcSJeff Roberson 	td->td_cpuset = NULL;
842696058c3SJulian Elischer 	cpu_thread_clean(td);
8434ea6a9a2SMateusz Guzik 	thread_cow_free(td);
8442d19b736SKonstantin Belousov 	callout_drain(&td->td_slpcallout);
845696058c3SJulian Elischer 	thread_reap();	/* check for zombie threads etc. */
846696058c3SJulian Elischer }
847696058c3SJulian Elischer 
848696058c3SJulian Elischer /*
84944990b8cSJulian Elischer  * Link a thread to a process.
8501faf202eSJulian Elischer  * set up anything that needs to be initialized for it to
8511faf202eSJulian Elischer  * be used by the process.
85244990b8cSJulian Elischer  */
85344990b8cSJulian Elischer void
8548460a577SJohn Birrell thread_link(struct thread *td, struct proc *p)
85544990b8cSJulian Elischer {
85644990b8cSJulian Elischer 
857a54e85fdSJeff Roberson 	/*
858a54e85fdSJeff Roberson 	 * XXX This can't be enabled because it's called for proc0 before
859374ae2a3SJeff Roberson 	 * its lock has been created.
860374ae2a3SJeff Roberson 	 * PROC_LOCK_ASSERT(p, MA_OWNED);
861a54e85fdSJeff Roberson 	 */
86271fad9fdSJulian Elischer 	td->td_state    = TDS_INACTIVE;
86344990b8cSJulian Elischer 	td->td_proc     = p;
864b61ce5b0SJeff Roberson 	td->td_flags    = TDF_INMEM;
86544990b8cSJulian Elischer 
8661faf202eSJulian Elischer 	LIST_INIT(&td->td_contested);
867eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[0]);
868eea4f254SJeff Roberson 	LIST_INIT(&td->td_lprof[1]);
869f6eccf96SGleb Smirnoff #ifdef EPOCH_TRACE
870dd902d01SGleb Smirnoff 	SLIST_INIT(&td->td_epochs);
871f6eccf96SGleb Smirnoff #endif
8729104847fSDavid Xu 	sigqueue_init(&td->td_sigqueue, p);
873fd90e2edSJung-uk Kim 	callout_init(&td->td_slpcallout, 1);
87466d8df9dSDaniel Eischen 	TAILQ_INSERT_TAIL(&p->p_threads, td, td_plist);
87544990b8cSJulian Elischer 	p->p_numthreads++;
87644990b8cSJulian Elischer }
87744990b8cSJulian Elischer 
878ed062c8dSJulian Elischer /*
879ed062c8dSJulian Elischer  * Called from:
880ed062c8dSJulian Elischer  *  thread_exit()
881ed062c8dSJulian Elischer  */
882d3a0bd78SJulian Elischer void
883d3a0bd78SJulian Elischer thread_unlink(struct thread *td)
884d3a0bd78SJulian Elischer {
885d3a0bd78SJulian Elischer 	struct proc *p = td->td_proc;
886d3a0bd78SJulian Elischer 
887374ae2a3SJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
888f6eccf96SGleb Smirnoff #ifdef EPOCH_TRACE
889dd902d01SGleb Smirnoff 	MPASS(SLIST_EMPTY(&td->td_epochs));
890f6eccf96SGleb Smirnoff #endif
891dd902d01SGleb Smirnoff 
892d3a0bd78SJulian Elischer 	TAILQ_REMOVE(&p->p_threads, td, td_plist);
893d3a0bd78SJulian Elischer 	p->p_numthreads--;
894d3a0bd78SJulian Elischer 	/* could clear a few other things here */
8958460a577SJohn Birrell 	/* Must  NOT clear links to proc! */
8965c8329edSJulian Elischer }
8975c8329edSJulian Elischer 
89879799053SKonstantin Belousov static int
89979799053SKonstantin Belousov calc_remaining(struct proc *p, int mode)
90079799053SKonstantin Belousov {
90179799053SKonstantin Belousov 	int remaining;
90279799053SKonstantin Belousov 
9037b519077SKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
9047b519077SKonstantin Belousov 	PROC_SLOCK_ASSERT(p, MA_OWNED);
90579799053SKonstantin Belousov 	if (mode == SINGLE_EXIT)
90679799053SKonstantin Belousov 		remaining = p->p_numthreads;
90779799053SKonstantin Belousov 	else if (mode == SINGLE_BOUNDARY)
90879799053SKonstantin Belousov 		remaining = p->p_numthreads - p->p_boundary_count;
9096ddcc233SKonstantin Belousov 	else if (mode == SINGLE_NO_EXIT || mode == SINGLE_ALLPROC)
91079799053SKonstantin Belousov 		remaining = p->p_numthreads - p->p_suspcount;
91179799053SKonstantin Belousov 	else
91279799053SKonstantin Belousov 		panic("calc_remaining: wrong mode %d", mode);
91379799053SKonstantin Belousov 	return (remaining);
91479799053SKonstantin Belousov }
91579799053SKonstantin Belousov 
91607a9368aSKonstantin Belousov static int
91707a9368aSKonstantin Belousov remain_for_mode(int mode)
91807a9368aSKonstantin Belousov {
91907a9368aSKonstantin Belousov 
9206ddcc233SKonstantin Belousov 	return (mode == SINGLE_ALLPROC ? 0 : 1);
92107a9368aSKonstantin Belousov }
92207a9368aSKonstantin Belousov 
92307a9368aSKonstantin Belousov static int
92407a9368aSKonstantin Belousov weed_inhib(int mode, struct thread *td2, struct proc *p)
92507a9368aSKonstantin Belousov {
92607a9368aSKonstantin Belousov 	int wakeup_swapper;
92707a9368aSKonstantin Belousov 
92807a9368aSKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
92907a9368aSKonstantin Belousov 	PROC_SLOCK_ASSERT(p, MA_OWNED);
93007a9368aSKonstantin Belousov 	THREAD_LOCK_ASSERT(td2, MA_OWNED);
93107a9368aSKonstantin Belousov 
93207a9368aSKonstantin Belousov 	wakeup_swapper = 0;
93361a74c5cSJeff Roberson 
93461a74c5cSJeff Roberson 	/*
93561a74c5cSJeff Roberson 	 * Since the thread lock is dropped by the scheduler we have
93661a74c5cSJeff Roberson 	 * to retry to check for races.
93761a74c5cSJeff Roberson 	 */
93861a74c5cSJeff Roberson restart:
93907a9368aSKonstantin Belousov 	switch (mode) {
94007a9368aSKonstantin Belousov 	case SINGLE_EXIT:
94161a74c5cSJeff Roberson 		if (TD_IS_SUSPENDED(td2)) {
94284cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, true);
94361a74c5cSJeff Roberson 			thread_lock(td2);
94461a74c5cSJeff Roberson 			goto restart;
94561a74c5cSJeff Roberson 		}
94661a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
94707a9368aSKonstantin Belousov 			wakeup_swapper |= sleepq_abort(td2, EINTR);
94861a74c5cSJeff Roberson 			return (wakeup_swapper);
94961a74c5cSJeff Roberson 		}
95007a9368aSKonstantin Belousov 		break;
95107a9368aSKonstantin Belousov 	case SINGLE_BOUNDARY:
95207a9368aSKonstantin Belousov 	case SINGLE_NO_EXIT:
95361a74c5cSJeff Roberson 		if (TD_IS_SUSPENDED(td2) &&
95461a74c5cSJeff Roberson 		    (td2->td_flags & TDF_BOUNDARY) == 0) {
95584cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, false);
95661a74c5cSJeff Roberson 			thread_lock(td2);
95761a74c5cSJeff Roberson 			goto restart;
95861a74c5cSJeff Roberson 		}
95961a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
96007a9368aSKonstantin Belousov 			wakeup_swapper |= sleepq_abort(td2, ERESTART);
96161a74c5cSJeff Roberson 			return (wakeup_swapper);
96261a74c5cSJeff Roberson 		}
963917dd390SKonstantin Belousov 		break;
9646ddcc233SKonstantin Belousov 	case SINGLE_ALLPROC:
9656ddcc233SKonstantin Belousov 		/*
9666ddcc233SKonstantin Belousov 		 * ALLPROC suspend tries to avoid spurious EINTR for
9676ddcc233SKonstantin Belousov 		 * threads sleeping interruptable, by suspending the
9686ddcc233SKonstantin Belousov 		 * thread directly, similarly to sig_suspend_threads().
9696ddcc233SKonstantin Belousov 		 * Since such sleep is not performed at the user
9706ddcc233SKonstantin Belousov 		 * boundary, TDF_BOUNDARY flag is not set, and TDF_ALLPROCSUSP
9716ddcc233SKonstantin Belousov 		 * is used to avoid immediate un-suspend.
9726ddcc233SKonstantin Belousov 		 */
9736ddcc233SKonstantin Belousov 		if (TD_IS_SUSPENDED(td2) && (td2->td_flags & (TDF_BOUNDARY |
97461a74c5cSJeff Roberson 		    TDF_ALLPROCSUSP)) == 0) {
97584cdea97SKonstantin Belousov 			wakeup_swapper |= thread_unsuspend_one(td2, p, false);
97661a74c5cSJeff Roberson 			thread_lock(td2);
97761a74c5cSJeff Roberson 			goto restart;
97861a74c5cSJeff Roberson 		}
97961a74c5cSJeff Roberson 		if (TD_CAN_ABORT(td2)) {
9806ddcc233SKonstantin Belousov 			if ((td2->td_flags & TDF_SBDRY) == 0) {
9816ddcc233SKonstantin Belousov 				thread_suspend_one(td2);
9826ddcc233SKonstantin Belousov 				td2->td_flags |= TDF_ALLPROCSUSP;
9836ddcc233SKonstantin Belousov 			} else {
9846ddcc233SKonstantin Belousov 				wakeup_swapper |= sleepq_abort(td2, ERESTART);
98561a74c5cSJeff Roberson 				return (wakeup_swapper);
9866ddcc233SKonstantin Belousov 			}
9876ddcc233SKonstantin Belousov 		}
98807a9368aSKonstantin Belousov 		break;
98961a74c5cSJeff Roberson 	default:
99061a74c5cSJeff Roberson 		break;
99107a9368aSKonstantin Belousov 	}
99261a74c5cSJeff Roberson 	thread_unlock(td2);
99307a9368aSKonstantin Belousov 	return (wakeup_swapper);
99407a9368aSKonstantin Belousov }
99507a9368aSKonstantin Belousov 
9965215b187SJeff Roberson /*
99744990b8cSJulian Elischer  * Enforce single-threading.
99844990b8cSJulian Elischer  *
99944990b8cSJulian Elischer  * Returns 1 if the caller must abort (another thread is waiting to
100044990b8cSJulian Elischer  * exit the process or similar). Process is locked!
100144990b8cSJulian Elischer  * Returns 0 when you are successfully the only thread running.
100244990b8cSJulian Elischer  * A process has successfully single threaded in the suspend mode when
100344990b8cSJulian Elischer  * There are no threads in user mode. Threads in the kernel must be
100444990b8cSJulian Elischer  * allowed to continue until they get to the user boundary. They may even
100544990b8cSJulian Elischer  * copy out their return values and data before suspending. They may however be
1006e2668f55SMaxim Konovalov  * accelerated in reaching the user boundary as we will wake up
100744990b8cSJulian Elischer  * any sleeping threads that are interruptable. (PCATCH).
100844990b8cSJulian Elischer  */
100944990b8cSJulian Elischer int
10106ddcc233SKonstantin Belousov thread_single(struct proc *p, int mode)
101144990b8cSJulian Elischer {
101244990b8cSJulian Elischer 	struct thread *td;
101344990b8cSJulian Elischer 	struct thread *td2;
1014da7bbd2cSJohn Baldwin 	int remaining, wakeup_swapper;
101544990b8cSJulian Elischer 
101644990b8cSJulian Elischer 	td = curthread;
10176ddcc233SKonstantin Belousov 	KASSERT(mode == SINGLE_EXIT || mode == SINGLE_BOUNDARY ||
10186ddcc233SKonstantin Belousov 	    mode == SINGLE_ALLPROC || mode == SINGLE_NO_EXIT,
10196ddcc233SKonstantin Belousov 	    ("invalid mode %d", mode));
10206ddcc233SKonstantin Belousov 	/*
10216ddcc233SKonstantin Belousov 	 * If allowing non-ALLPROC singlethreading for non-curproc
10226ddcc233SKonstantin Belousov 	 * callers, calc_remaining() and remain_for_mode() should be
10236ddcc233SKonstantin Belousov 	 * adjusted to also account for td->td_proc != p.  For now
10246ddcc233SKonstantin Belousov 	 * this is not implemented because it is not used.
10256ddcc233SKonstantin Belousov 	 */
10266ddcc233SKonstantin Belousov 	KASSERT((mode == SINGLE_ALLPROC && td->td_proc != p) ||
10276ddcc233SKonstantin Belousov 	    (mode != SINGLE_ALLPROC && td->td_proc == p),
10286ddcc233SKonstantin Belousov 	    ("mode %d proc %p curproc %p", mode, p, td->td_proc));
102937814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
103044990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
103144990b8cSJulian Elischer 
10326ddcc233SKonstantin Belousov 	if ((p->p_flag & P_HADTHREADS) == 0 && mode != SINGLE_ALLPROC)
103344990b8cSJulian Elischer 		return (0);
103444990b8cSJulian Elischer 
1035e3b9bf71SJulian Elischer 	/* Is someone already single threading? */
1036906ac69dSDavid Xu 	if (p->p_singlethread != NULL && p->p_singlethread != td)
103744990b8cSJulian Elischer 		return (1);
103844990b8cSJulian Elischer 
1039906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
1040906ac69dSDavid Xu 		p->p_flag |= P_SINGLE_EXIT;
1041906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_BOUNDARY;
1042906ac69dSDavid Xu 	} else {
1043906ac69dSDavid Xu 		p->p_flag &= ~P_SINGLE_EXIT;
1044906ac69dSDavid Xu 		if (mode == SINGLE_BOUNDARY)
1045906ac69dSDavid Xu 			p->p_flag |= P_SINGLE_BOUNDARY;
1046906ac69dSDavid Xu 		else
1047906ac69dSDavid Xu 			p->p_flag &= ~P_SINGLE_BOUNDARY;
1048906ac69dSDavid Xu 	}
10496ddcc233SKonstantin Belousov 	if (mode == SINGLE_ALLPROC)
10506ddcc233SKonstantin Belousov 		p->p_flag |= P_TOTAL_STOP;
10511279572aSDavid Xu 	p->p_flag |= P_STOPPED_SINGLE;
10527b4a950aSDavid Xu 	PROC_SLOCK(p);
1053112afcb2SJohn Baldwin 	p->p_singlethread = td;
105479799053SKonstantin Belousov 	remaining = calc_remaining(p, mode);
105507a9368aSKonstantin Belousov 	while (remaining != remain_for_mode(mode)) {
1056bf1a3220SDavid Xu 		if (P_SHOULDSTOP(p) != P_STOPPED_SINGLE)
1057bf1a3220SDavid Xu 			goto stopme;
1058da7bbd2cSJohn Baldwin 		wakeup_swapper = 0;
105944990b8cSJulian Elischer 		FOREACH_THREAD_IN_PROC(p, td2) {
106044990b8cSJulian Elischer 			if (td2 == td)
106144990b8cSJulian Elischer 				continue;
1062a54e85fdSJeff Roberson 			thread_lock(td2);
1063b7edba77SJeff Roberson 			td2->td_flags |= TDF_ASTPENDING | TDF_NEEDSUSPCHK;
10646ddcc233SKonstantin Belousov 			if (TD_IS_INHIBITED(td2)) {
106507a9368aSKonstantin Belousov 				wakeup_swapper |= weed_inhib(mode, td2, p);
1066d8267df7SDavid Xu #ifdef SMP
10676ddcc233SKonstantin Belousov 			} else if (TD_IS_RUNNING(td2) && td != td2) {
1068d8267df7SDavid Xu 				forward_signal(td2);
106961a74c5cSJeff Roberson 				thread_unlock(td2);
1070d8267df7SDavid Xu #endif
107161a74c5cSJeff Roberson 			} else
1072a54e85fdSJeff Roberson 				thread_unlock(td2);
10739d102777SJulian Elischer 		}
1074da7bbd2cSJohn Baldwin 		if (wakeup_swapper)
1075da7bbd2cSJohn Baldwin 			kick_proc0();
107679799053SKonstantin Belousov 		remaining = calc_remaining(p, mode);
1077ec008e96SDavid Xu 
10789d102777SJulian Elischer 		/*
10799d102777SJulian Elischer 		 * Maybe we suspended some threads.. was it enough?
10809d102777SJulian Elischer 		 */
108107a9368aSKonstantin Belousov 		if (remaining == remain_for_mode(mode))
10829d102777SJulian Elischer 			break;
10839d102777SJulian Elischer 
1084bf1a3220SDavid Xu stopme:
108544990b8cSJulian Elischer 		/*
108644990b8cSJulian Elischer 		 * Wake us up when everyone else has suspended.
1087e3b9bf71SJulian Elischer 		 * In the mean time we suspend as well.
108844990b8cSJulian Elischer 		 */
10896ddcc233SKonstantin Belousov 		thread_suspend_switch(td, p);
109079799053SKonstantin Belousov 		remaining = calc_remaining(p, mode);
109144990b8cSJulian Elischer 	}
1092906ac69dSDavid Xu 	if (mode == SINGLE_EXIT) {
109391599697SJulian Elischer 		/*
10948626a0ddSKonstantin Belousov 		 * Convert the process to an unthreaded process.  The
10958626a0ddSKonstantin Belousov 		 * SINGLE_EXIT is called by exit1() or execve(), in
10968626a0ddSKonstantin Belousov 		 * both cases other threads must be retired.
109791599697SJulian Elischer 		 */
10988626a0ddSKonstantin Belousov 		KASSERT(p->p_numthreads == 1, ("Unthreading with >1 threads"));
1099ed062c8dSJulian Elischer 		p->p_singlethread = NULL;
11008626a0ddSKonstantin Belousov 		p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_HADTHREADS);
1101fd229b5bSKonstantin Belousov 
1102fd229b5bSKonstantin Belousov 		/*
1103fd229b5bSKonstantin Belousov 		 * Wait for any remaining threads to exit cpu_throw().
1104fd229b5bSKonstantin Belousov 		 */
1105fd229b5bSKonstantin Belousov 		while (p->p_exitthreads != 0) {
1106fd229b5bSKonstantin Belousov 			PROC_SUNLOCK(p);
1107fd229b5bSKonstantin Belousov 			PROC_UNLOCK(p);
1108fd229b5bSKonstantin Belousov 			sched_relinquish(td);
1109fd229b5bSKonstantin Belousov 			PROC_LOCK(p);
1110fd229b5bSKonstantin Belousov 			PROC_SLOCK(p);
1111fd229b5bSKonstantin Belousov 		}
1112ac437c07SKonstantin Belousov 	} else if (mode == SINGLE_BOUNDARY) {
1113ac437c07SKonstantin Belousov 		/*
1114ac437c07SKonstantin Belousov 		 * Wait until all suspended threads are removed from
1115ac437c07SKonstantin Belousov 		 * the processors.  The thread_suspend_check()
1116ac437c07SKonstantin Belousov 		 * increments p_boundary_count while it is still
1117ac437c07SKonstantin Belousov 		 * running, which makes it possible for the execve()
1118ac437c07SKonstantin Belousov 		 * to destroy vmspace while our other threads are
1119ac437c07SKonstantin Belousov 		 * still using the address space.
1120ac437c07SKonstantin Belousov 		 *
1121ac437c07SKonstantin Belousov 		 * We lock the thread, which is only allowed to
1122ac437c07SKonstantin Belousov 		 * succeed after context switch code finished using
1123ac437c07SKonstantin Belousov 		 * the address space.
1124ac437c07SKonstantin Belousov 		 */
1125ac437c07SKonstantin Belousov 		FOREACH_THREAD_IN_PROC(p, td2) {
1126ac437c07SKonstantin Belousov 			if (td2 == td)
1127ac437c07SKonstantin Belousov 				continue;
1128ac437c07SKonstantin Belousov 			thread_lock(td2);
1129ac437c07SKonstantin Belousov 			KASSERT((td2->td_flags & TDF_BOUNDARY) != 0,
1130ac437c07SKonstantin Belousov 			    ("td %p not on boundary", td2));
1131ac437c07SKonstantin Belousov 			KASSERT(TD_IS_SUSPENDED(td2),
1132ac437c07SKonstantin Belousov 			    ("td %p is not suspended", td2));
1133ac437c07SKonstantin Belousov 			thread_unlock(td2);
1134ac437c07SKonstantin Belousov 		}
113591599697SJulian Elischer 	}
11367b4a950aSDavid Xu 	PROC_SUNLOCK(p);
113744990b8cSJulian Elischer 	return (0);
113844990b8cSJulian Elischer }
113944990b8cSJulian Elischer 
11408638fe7bSKonstantin Belousov bool
11418638fe7bSKonstantin Belousov thread_suspend_check_needed(void)
11428638fe7bSKonstantin Belousov {
11438638fe7bSKonstantin Belousov 	struct proc *p;
11448638fe7bSKonstantin Belousov 	struct thread *td;
11458638fe7bSKonstantin Belousov 
11468638fe7bSKonstantin Belousov 	td = curthread;
11478638fe7bSKonstantin Belousov 	p = td->td_proc;
11488638fe7bSKonstantin Belousov 	PROC_LOCK_ASSERT(p, MA_OWNED);
11498638fe7bSKonstantin Belousov 	return (P_SHOULDSTOP(p) || ((p->p_flag & P_TRACED) != 0 &&
11508638fe7bSKonstantin Belousov 	    (td->td_dbgflags & TDB_SUSPEND) != 0));
11518638fe7bSKonstantin Belousov }
11528638fe7bSKonstantin Belousov 
115344990b8cSJulian Elischer /*
115444990b8cSJulian Elischer  * Called in from locations that can safely check to see
115544990b8cSJulian Elischer  * whether we have to suspend or at least throttle for a
115644990b8cSJulian Elischer  * single-thread event (e.g. fork).
115744990b8cSJulian Elischer  *
115844990b8cSJulian Elischer  * Such locations include userret().
115944990b8cSJulian Elischer  * If the "return_instead" argument is non zero, the thread must be able to
116044990b8cSJulian Elischer  * accept 0 (caller may continue), or 1 (caller must abort) as a result.
116144990b8cSJulian Elischer  *
116244990b8cSJulian Elischer  * The 'return_instead' argument tells the function if it may do a
116344990b8cSJulian Elischer  * thread_exit() or suspend, or whether the caller must abort and back
116444990b8cSJulian Elischer  * out instead.
116544990b8cSJulian Elischer  *
116644990b8cSJulian Elischer  * If the thread that set the single_threading request has set the
116744990b8cSJulian Elischer  * P_SINGLE_EXIT bit in the process flags then this call will never return
116844990b8cSJulian Elischer  * if 'return_instead' is false, but will exit.
116944990b8cSJulian Elischer  *
117044990b8cSJulian Elischer  * P_SINGLE_EXIT | return_instead == 0| return_instead != 0
117144990b8cSJulian Elischer  *---------------+--------------------+---------------------
117244990b8cSJulian Elischer  *       0       | returns 0          |   returns 0 or 1
1173353374b5SJohn Baldwin  *               | when ST ends       |   immediately
117444990b8cSJulian Elischer  *---------------+--------------------+---------------------
117544990b8cSJulian Elischer  *       1       | thread exits       |   returns 1
1176353374b5SJohn Baldwin  *               |                    |  immediately
117744990b8cSJulian Elischer  * 0 = thread_exit() or suspension ok,
117844990b8cSJulian Elischer  * other = return error instead of stopping the thread.
117944990b8cSJulian Elischer  *
118044990b8cSJulian Elischer  * While a full suspension is under effect, even a single threading
118144990b8cSJulian Elischer  * thread would be suspended if it made this call (but it shouldn't).
118244990b8cSJulian Elischer  * This call should only be made from places where
118344990b8cSJulian Elischer  * thread_exit() would be safe as that may be the outcome unless
118444990b8cSJulian Elischer  * return_instead is set.
118544990b8cSJulian Elischer  */
118644990b8cSJulian Elischer int
118744990b8cSJulian Elischer thread_suspend_check(int return_instead)
118844990b8cSJulian Elischer {
1189ecafb24bSJuli Mallett 	struct thread *td;
1190ecafb24bSJuli Mallett 	struct proc *p;
119146e47c4fSKonstantin Belousov 	int wakeup_swapper;
119244990b8cSJulian Elischer 
119344990b8cSJulian Elischer 	td = curthread;
119444990b8cSJulian Elischer 	p = td->td_proc;
119537814395SPeter Wemm 	mtx_assert(&Giant, MA_NOTOWNED);
119644990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
11978638fe7bSKonstantin Belousov 	while (thread_suspend_check_needed()) {
11981279572aSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
119944990b8cSJulian Elischer 			KASSERT(p->p_singlethread != NULL,
120044990b8cSJulian Elischer 			    ("singlethread not set"));
120144990b8cSJulian Elischer 			/*
1202e3b9bf71SJulian Elischer 			 * The only suspension in action is a
1203e3b9bf71SJulian Elischer 			 * single-threading. Single threader need not stop.
1204bd07998eSKonstantin Belousov 			 * It is safe to access p->p_singlethread unlocked
1205bd07998eSKonstantin Belousov 			 * because it can only be set to our address by us.
120644990b8cSJulian Elischer 			 */
1207e3b9bf71SJulian Elischer 			if (p->p_singlethread == td)
120844990b8cSJulian Elischer 				return (0);	/* Exempt from stopping. */
120944990b8cSJulian Elischer 		}
121045a4bfa1SDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && return_instead)
121194f0972bSDavid Xu 			return (EINTR);
121244990b8cSJulian Elischer 
1213906ac69dSDavid Xu 		/* Should we goto user boundary if we didn't come from there? */
1214906ac69dSDavid Xu 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
1215906ac69dSDavid Xu 		    (p->p_flag & P_SINGLE_BOUNDARY) && return_instead)
121694f0972bSDavid Xu 			return (ERESTART);
1217906ac69dSDavid Xu 
121844990b8cSJulian Elischer 		/*
12193077f938SKonstantin Belousov 		 * Ignore suspend requests if they are deferred.
1220d071a6faSJohn Baldwin 		 */
12213077f938SKonstantin Belousov 		if ((td->td_flags & TDF_SBDRY) != 0) {
1222d071a6faSJohn Baldwin 			KASSERT(return_instead,
1223d071a6faSJohn Baldwin 			    ("TDF_SBDRY set for unsafe thread_suspend_check"));
122446e47c4fSKonstantin Belousov 			KASSERT((td->td_flags & (TDF_SEINTR | TDF_SERESTART)) !=
122546e47c4fSKonstantin Belousov 			    (TDF_SEINTR | TDF_SERESTART),
122646e47c4fSKonstantin Belousov 			    ("both TDF_SEINTR and TDF_SERESTART"));
122746e47c4fSKonstantin Belousov 			return (TD_SBDRY_INTR(td) ? TD_SBDRY_ERRNO(td) : 0);
1228d071a6faSJohn Baldwin 		}
1229d071a6faSJohn Baldwin 
1230d071a6faSJohn Baldwin 		/*
123144990b8cSJulian Elischer 		 * If the process is waiting for us to exit,
123244990b8cSJulian Elischer 		 * this thread should just suicide.
12331279572aSDavid Xu 		 * Assumes that P_SINGLE_EXIT implies P_STOPPED_SINGLE.
123444990b8cSJulian Elischer 		 */
1235cf7d9a8cSDavid Xu 		if ((p->p_flag & P_SINGLE_EXIT) && (p->p_singlethread != td)) {
1236cf7d9a8cSDavid Xu 			PROC_UNLOCK(p);
123791d1786fSDmitry Chagin 
123891d1786fSDmitry Chagin 			/*
123991d1786fSDmitry Chagin 			 * Allow Linux emulation layer to do some work
124091d1786fSDmitry Chagin 			 * before thread suicide.
124191d1786fSDmitry Chagin 			 */
124291d1786fSDmitry Chagin 			if (__predict_false(p->p_sysent->sv_thread_detach != NULL))
124391d1786fSDmitry Chagin 				(p->p_sysent->sv_thread_detach)(td);
12442a339d9eSKonstantin Belousov 			umtx_thread_exit(td);
1245d1e7a4a5SJohn Baldwin 			kern_thr_exit(td);
1246d1e7a4a5SJohn Baldwin 			panic("stopped thread did not exit");
1247cf7d9a8cSDavid Xu 		}
124821ecd1e9SDavid Xu 
124921ecd1e9SDavid Xu 		PROC_SLOCK(p);
125021ecd1e9SDavid Xu 		thread_stopped(p);
1251a54e85fdSJeff Roberson 		if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE) {
1252a54e85fdSJeff Roberson 			if (p->p_numthreads == p->p_suspcount + 1) {
1253a54e85fdSJeff Roberson 				thread_lock(p->p_singlethread);
125484cdea97SKonstantin Belousov 				wakeup_swapper = thread_unsuspend_one(
125584cdea97SKonstantin Belousov 				    p->p_singlethread, p, false);
12567847a9daSJohn Baldwin 				if (wakeup_swapper)
12577847a9daSJohn Baldwin 					kick_proc0();
1258a54e85fdSJeff Roberson 			}
1259a54e85fdSJeff Roberson 		}
12603f9be10eSDavid Xu 		PROC_UNLOCK(p);
12617b4a950aSDavid Xu 		thread_lock(td);
126244990b8cSJulian Elischer 		/*
126344990b8cSJulian Elischer 		 * When a thread suspends, it just
1264ad1e7d28SJulian Elischer 		 * gets taken off all queues.
126544990b8cSJulian Elischer 		 */
126671fad9fdSJulian Elischer 		thread_suspend_one(td);
1267906ac69dSDavid Xu 		if (return_instead == 0) {
1268906ac69dSDavid Xu 			p->p_boundary_count++;
1269906ac69dSDavid Xu 			td->td_flags |= TDF_BOUNDARY;
1270cf19bf91SJulian Elischer 		}
12717b4a950aSDavid Xu 		PROC_SUNLOCK(p);
1272686bcb5cSJeff Roberson 		mi_switch(SW_INVOL | SWT_SUSPEND);
127344990b8cSJulian Elischer 		PROC_LOCK(p);
127444990b8cSJulian Elischer 	}
127544990b8cSJulian Elischer 	return (0);
127644990b8cSJulian Elischer }
127744990b8cSJulian Elischer 
1278478ca4b0SKonstantin Belousov /*
1279478ca4b0SKonstantin Belousov  * Check for possible stops and suspensions while executing a
1280478ca4b0SKonstantin Belousov  * casueword or similar transiently failing operation.
1281478ca4b0SKonstantin Belousov  *
1282478ca4b0SKonstantin Belousov  * The sleep argument controls whether the function can handle a stop
1283478ca4b0SKonstantin Belousov  * request itself or it should return ERESTART and the request is
1284478ca4b0SKonstantin Belousov  * proceed at the kernel/user boundary in ast.
1285478ca4b0SKonstantin Belousov  *
1286478ca4b0SKonstantin Belousov  * Typically, when retrying due to casueword(9) failure (rv == 1), we
1287478ca4b0SKonstantin Belousov  * should handle the stop requests there, with exception of cases when
1288478ca4b0SKonstantin Belousov  * the thread owns a kernel resource, for instance busied the umtx
1289300b525dSKonstantin Belousov  * key, or when functions return immediately if thread_check_susp()
1290478ca4b0SKonstantin Belousov  * returned non-zero.  On the other hand, retrying the whole lock
1291478ca4b0SKonstantin Belousov  * operation, we better not stop there but delegate the handling to
1292478ca4b0SKonstantin Belousov  * ast.
1293478ca4b0SKonstantin Belousov  *
1294478ca4b0SKonstantin Belousov  * If the request is for thread termination P_SINGLE_EXIT, we cannot
1295478ca4b0SKonstantin Belousov  * handle it at all, and simply return EINTR.
1296478ca4b0SKonstantin Belousov  */
1297478ca4b0SKonstantin Belousov int
1298478ca4b0SKonstantin Belousov thread_check_susp(struct thread *td, bool sleep)
1299478ca4b0SKonstantin Belousov {
1300478ca4b0SKonstantin Belousov 	struct proc *p;
1301478ca4b0SKonstantin Belousov 	int error;
1302478ca4b0SKonstantin Belousov 
1303478ca4b0SKonstantin Belousov 	/*
1304478ca4b0SKonstantin Belousov 	 * The check for TDF_NEEDSUSPCHK is racy, but it is enough to
1305478ca4b0SKonstantin Belousov 	 * eventually break the lockstep loop.
1306478ca4b0SKonstantin Belousov 	 */
1307478ca4b0SKonstantin Belousov 	if ((td->td_flags & TDF_NEEDSUSPCHK) == 0)
1308478ca4b0SKonstantin Belousov 		return (0);
1309478ca4b0SKonstantin Belousov 	error = 0;
1310478ca4b0SKonstantin Belousov 	p = td->td_proc;
1311478ca4b0SKonstantin Belousov 	PROC_LOCK(p);
1312478ca4b0SKonstantin Belousov 	if (p->p_flag & P_SINGLE_EXIT)
1313478ca4b0SKonstantin Belousov 		error = EINTR;
1314478ca4b0SKonstantin Belousov 	else if (P_SHOULDSTOP(p) ||
1315478ca4b0SKonstantin Belousov 	    ((p->p_flag & P_TRACED) && (td->td_dbgflags & TDB_SUSPEND)))
1316478ca4b0SKonstantin Belousov 		error = sleep ? thread_suspend_check(0) : ERESTART;
1317478ca4b0SKonstantin Belousov 	PROC_UNLOCK(p);
1318478ca4b0SKonstantin Belousov 	return (error);
1319478ca4b0SKonstantin Belousov }
1320478ca4b0SKonstantin Belousov 
132135c32a76SDavid Xu void
13226ddcc233SKonstantin Belousov thread_suspend_switch(struct thread *td, struct proc *p)
1323a54e85fdSJeff Roberson {
1324a54e85fdSJeff Roberson 
1325a54e85fdSJeff Roberson 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
1326a54e85fdSJeff Roberson 	PROC_LOCK_ASSERT(p, MA_OWNED);
13277b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
1328a54e85fdSJeff Roberson 	/*
1329a54e85fdSJeff Roberson 	 * We implement thread_suspend_one in stages here to avoid
1330a54e85fdSJeff Roberson 	 * dropping the proc lock while the thread lock is owned.
1331a54e85fdSJeff Roberson 	 */
13326ddcc233SKonstantin Belousov 	if (p == td->td_proc) {
1333a54e85fdSJeff Roberson 		thread_stopped(p);
1334a54e85fdSJeff Roberson 		p->p_suspcount++;
13356ddcc233SKonstantin Belousov 	}
13363f9be10eSDavid Xu 	PROC_UNLOCK(p);
13377b4a950aSDavid Xu 	thread_lock(td);
1338b7edba77SJeff Roberson 	td->td_flags &= ~TDF_NEEDSUSPCHK;
1339a54e85fdSJeff Roberson 	TD_SET_SUSPENDED(td);
1340c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
13417b4a950aSDavid Xu 	PROC_SUNLOCK(p);
1342a54e85fdSJeff Roberson 	DROP_GIANT();
1343686bcb5cSJeff Roberson 	mi_switch(SW_VOL | SWT_SUSPEND);
1344a54e85fdSJeff Roberson 	PICKUP_GIANT();
1345a54e85fdSJeff Roberson 	PROC_LOCK(p);
13467b4a950aSDavid Xu 	PROC_SLOCK(p);
1347a54e85fdSJeff Roberson }
1348a54e85fdSJeff Roberson 
1349a54e85fdSJeff Roberson void
135035c32a76SDavid Xu thread_suspend_one(struct thread *td)
135135c32a76SDavid Xu {
13526ddcc233SKonstantin Belousov 	struct proc *p;
135335c32a76SDavid Xu 
13546ddcc233SKonstantin Belousov 	p = td->td_proc;
13557b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
1356a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
1357e574e444SDavid Xu 	KASSERT(!TD_IS_SUSPENDED(td), ("already suspended"));
135835c32a76SDavid Xu 	p->p_suspcount++;
1359b7edba77SJeff Roberson 	td->td_flags &= ~TDF_NEEDSUSPCHK;
136071fad9fdSJulian Elischer 	TD_SET_SUSPENDED(td);
1361c5aa6b58SJeff Roberson 	sched_sleep(td, 0);
136235c32a76SDavid Xu }
136335c32a76SDavid Xu 
136484cdea97SKonstantin Belousov static int
136584cdea97SKonstantin Belousov thread_unsuspend_one(struct thread *td, struct proc *p, bool boundary)
136635c32a76SDavid Xu {
136735c32a76SDavid Xu 
1368a54e85fdSJeff Roberson 	THREAD_LOCK_ASSERT(td, MA_OWNED);
1369ad1e7d28SJulian Elischer 	KASSERT(TD_IS_SUSPENDED(td), ("Thread not suspended"));
137071fad9fdSJulian Elischer 	TD_CLR_SUSPENDED(td);
13716ddcc233SKonstantin Belousov 	td->td_flags &= ~TDF_ALLPROCSUSP;
13726ddcc233SKonstantin Belousov 	if (td->td_proc == p) {
13736ddcc233SKonstantin Belousov 		PROC_SLOCK_ASSERT(p, MA_OWNED);
137435c32a76SDavid Xu 		p->p_suspcount--;
137584cdea97SKonstantin Belousov 		if (boundary && (td->td_flags & TDF_BOUNDARY) != 0) {
137684cdea97SKonstantin Belousov 			td->td_flags &= ~TDF_BOUNDARY;
137784cdea97SKonstantin Belousov 			p->p_boundary_count--;
137884cdea97SKonstantin Belousov 		}
13796ddcc233SKonstantin Belousov 	}
138061a74c5cSJeff Roberson 	return (setrunnable(td, 0));
138135c32a76SDavid Xu }
138235c32a76SDavid Xu 
138344990b8cSJulian Elischer /*
138444990b8cSJulian Elischer  * Allow all threads blocked by single threading to continue running.
138544990b8cSJulian Elischer  */
138644990b8cSJulian Elischer void
138744990b8cSJulian Elischer thread_unsuspend(struct proc *p)
138844990b8cSJulian Elischer {
138944990b8cSJulian Elischer 	struct thread *td;
13907847a9daSJohn Baldwin 	int wakeup_swapper;
139144990b8cSJulian Elischer 
139244990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
13937b4a950aSDavid Xu 	PROC_SLOCK_ASSERT(p, MA_OWNED);
13947847a9daSJohn Baldwin 	wakeup_swapper = 0;
139544990b8cSJulian Elischer 	if (!P_SHOULDSTOP(p)) {
1396ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
1397a54e85fdSJeff Roberson 			thread_lock(td);
1398ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
139984cdea97SKonstantin Belousov 				wakeup_swapper |= thread_unsuspend_one(td, p,
140084cdea97SKonstantin Belousov 				    true);
140161a74c5cSJeff Roberson 			} else
1402a54e85fdSJeff Roberson 				thread_unlock(td);
1403ad1e7d28SJulian Elischer 		}
140484cdea97SKonstantin Belousov 	} else if (P_SHOULDSTOP(p) == P_STOPPED_SINGLE &&
140584cdea97SKonstantin Belousov 	    p->p_numthreads == p->p_suspcount) {
140644990b8cSJulian Elischer 		/*
140744990b8cSJulian Elischer 		 * Stopping everything also did the job for the single
140844990b8cSJulian Elischer 		 * threading request. Now we've downgraded to single-threaded,
140944990b8cSJulian Elischer 		 * let it continue.
141044990b8cSJulian Elischer 		 */
14116ddcc233SKonstantin Belousov 		if (p->p_singlethread->td_proc == p) {
1412a54e85fdSJeff Roberson 			thread_lock(p->p_singlethread);
14136ddcc233SKonstantin Belousov 			wakeup_swapper = thread_unsuspend_one(
141484cdea97SKonstantin Belousov 			    p->p_singlethread, p, false);
141544990b8cSJulian Elischer 		}
14166ddcc233SKonstantin Belousov 	}
14177847a9daSJohn Baldwin 	if (wakeup_swapper)
14187847a9daSJohn Baldwin 		kick_proc0();
141944990b8cSJulian Elischer }
142044990b8cSJulian Elischer 
1421ed062c8dSJulian Elischer /*
1422ed062c8dSJulian Elischer  * End the single threading mode..
1423ed062c8dSJulian Elischer  */
142444990b8cSJulian Elischer void
14256ddcc233SKonstantin Belousov thread_single_end(struct proc *p, int mode)
142644990b8cSJulian Elischer {
142744990b8cSJulian Elischer 	struct thread *td;
14287847a9daSJohn Baldwin 	int wakeup_swapper;
142944990b8cSJulian Elischer 
14306ddcc233SKonstantin Belousov 	KASSERT(mode == SINGLE_EXIT || mode == SINGLE_BOUNDARY ||
14316ddcc233SKonstantin Belousov 	    mode == SINGLE_ALLPROC || mode == SINGLE_NO_EXIT,
14326ddcc233SKonstantin Belousov 	    ("invalid mode %d", mode));
143344990b8cSJulian Elischer 	PROC_LOCK_ASSERT(p, MA_OWNED);
14346ddcc233SKonstantin Belousov 	KASSERT((mode == SINGLE_ALLPROC && (p->p_flag & P_TOTAL_STOP) != 0) ||
14356ddcc233SKonstantin Belousov 	    (mode != SINGLE_ALLPROC && (p->p_flag & P_TOTAL_STOP) == 0),
14366ddcc233SKonstantin Belousov 	    ("mode %d does not match P_TOTAL_STOP", mode));
143784cdea97SKonstantin Belousov 	KASSERT(mode == SINGLE_ALLPROC || p->p_singlethread == curthread,
143884cdea97SKonstantin Belousov 	    ("thread_single_end from other thread %p %p",
143984cdea97SKonstantin Belousov 	    curthread, p->p_singlethread));
144084cdea97SKonstantin Belousov 	KASSERT(mode != SINGLE_BOUNDARY ||
144184cdea97SKonstantin Belousov 	    (p->p_flag & P_SINGLE_BOUNDARY) != 0,
144284cdea97SKonstantin Belousov 	    ("mis-matched SINGLE_BOUNDARY flags %x", p->p_flag));
14436ddcc233SKonstantin Belousov 	p->p_flag &= ~(P_STOPPED_SINGLE | P_SINGLE_EXIT | P_SINGLE_BOUNDARY |
14446ddcc233SKonstantin Belousov 	    P_TOTAL_STOP);
14457b4a950aSDavid Xu 	PROC_SLOCK(p);
144644990b8cSJulian Elischer 	p->p_singlethread = NULL;
14477847a9daSJohn Baldwin 	wakeup_swapper = 0;
144849539972SJulian Elischer 	/*
14497847a9daSJohn Baldwin 	 * If there are other threads they may now run,
145049539972SJulian Elischer 	 * unless of course there is a blanket 'stop order'
145149539972SJulian Elischer 	 * on the process. The single threader must be allowed
145249539972SJulian Elischer 	 * to continue however as this is a bad place to stop.
145349539972SJulian Elischer 	 */
14546ddcc233SKonstantin Belousov 	if (p->p_numthreads != remain_for_mode(mode) && !P_SHOULDSTOP(p)) {
1455ad1e7d28SJulian Elischer                 FOREACH_THREAD_IN_PROC(p, td) {
1456a54e85fdSJeff Roberson 			thread_lock(td);
1457ad1e7d28SJulian Elischer 			if (TD_IS_SUSPENDED(td)) {
145884cdea97SKonstantin Belousov 				wakeup_swapper |= thread_unsuspend_one(td, p,
145984cdea97SKonstantin Belousov 				    mode == SINGLE_BOUNDARY);
146061a74c5cSJeff Roberson 			} else
1461a54e85fdSJeff Roberson 				thread_unlock(td);
146249539972SJulian Elischer 		}
1463ad1e7d28SJulian Elischer 	}
146484cdea97SKonstantin Belousov 	KASSERT(mode != SINGLE_BOUNDARY || p->p_boundary_count == 0,
146584cdea97SKonstantin Belousov 	    ("inconsistent boundary count %d", p->p_boundary_count));
14667b4a950aSDavid Xu 	PROC_SUNLOCK(p);
14677847a9daSJohn Baldwin 	if (wakeup_swapper)
14687847a9daSJohn Baldwin 		kick_proc0();
146949539972SJulian Elischer }
14704fc21c09SDaniel Eischen 
1471aae3547bSMateusz Guzik /*
1472aae3547bSMateusz Guzik  * Locate a thread by number and return with proc lock held.
1473aae3547bSMateusz Guzik  *
1474aae3547bSMateusz Guzik  * thread exit establishes proc -> tidhash lock ordering, but lookup
1475aae3547bSMateusz Guzik  * takes tidhash first and needs to return locked proc.
1476aae3547bSMateusz Guzik  *
1477aae3547bSMateusz Guzik  * The problem is worked around by relying on type-safety of both
1478aae3547bSMateusz Guzik  * structures and doing the work in 2 steps:
1479aae3547bSMateusz Guzik  * - tidhash-locked lookup which saves both thread and proc pointers
1480aae3547bSMateusz Guzik  * - proc-locked verification that the found thread still matches
1481aae3547bSMateusz Guzik  */
1482aae3547bSMateusz Guzik static bool
1483aae3547bSMateusz Guzik tdfind_hash(lwpid_t tid, pid_t pid, struct proc **pp, struct thread **tdp)
1484cf7d9a8cSDavid Xu {
1485cf7d9a8cSDavid Xu #define RUN_THRESH	16
1486aae3547bSMateusz Guzik 	struct proc *p;
1487cf7d9a8cSDavid Xu 	struct thread *td;
1488aae3547bSMateusz Guzik 	int run;
1489aae3547bSMateusz Guzik 	bool locked;
1490cf7d9a8cSDavid Xu 
1491aae3547bSMateusz Guzik 	run = 0;
149226007fe3SMateusz Guzik 	rw_rlock(TIDHASHLOCK(tid));
1493aae3547bSMateusz Guzik 	locked = true;
1494cf7d9a8cSDavid Xu 	LIST_FOREACH(td, TIDHASH(tid), td_hash) {
1495aae3547bSMateusz Guzik 		if (td->td_tid != tid) {
1496aae3547bSMateusz Guzik 			run++;
1497aae3547bSMateusz Guzik 			continue;
1498cf7d9a8cSDavid Xu 		}
1499aae3547bSMateusz Guzik 		p = td->td_proc;
1500aae3547bSMateusz Guzik 		if (pid != -1 && p->p_pid != pid) {
1501cf7d9a8cSDavid Xu 			td = NULL;
1502cf7d9a8cSDavid Xu 			break;
1503cf7d9a8cSDavid Xu 		}
1504cf7d9a8cSDavid Xu 		if (run > RUN_THRESH) {
150526007fe3SMateusz Guzik 			if (rw_try_upgrade(TIDHASHLOCK(tid))) {
1506cf7d9a8cSDavid Xu 				LIST_REMOVE(td, td_hash);
1507cf7d9a8cSDavid Xu 				LIST_INSERT_HEAD(TIDHASH(td->td_tid),
1508cf7d9a8cSDavid Xu 					td, td_hash);
150926007fe3SMateusz Guzik 				rw_wunlock(TIDHASHLOCK(tid));
1510aae3547bSMateusz Guzik 				locked = false;
1511aae3547bSMateusz Guzik 				break;
1512cf7d9a8cSDavid Xu 			}
1513cf7d9a8cSDavid Xu 		}
1514cf7d9a8cSDavid Xu 		break;
1515cf7d9a8cSDavid Xu 	}
1516aae3547bSMateusz Guzik 	if (locked)
151726007fe3SMateusz Guzik 		rw_runlock(TIDHASHLOCK(tid));
1518aae3547bSMateusz Guzik 	if (td == NULL)
1519aae3547bSMateusz Guzik 		return (false);
1520aae3547bSMateusz Guzik 	*pp = p;
1521aae3547bSMateusz Guzik 	*tdp = td;
1522aae3547bSMateusz Guzik 	return (true);
1523aae3547bSMateusz Guzik }
1524aae3547bSMateusz Guzik 
1525aae3547bSMateusz Guzik struct thread *
1526aae3547bSMateusz Guzik tdfind(lwpid_t tid, pid_t pid)
1527aae3547bSMateusz Guzik {
1528aae3547bSMateusz Guzik 	struct proc *p;
1529aae3547bSMateusz Guzik 	struct thread *td;
1530aae3547bSMateusz Guzik 
1531aae3547bSMateusz Guzik 	td = curthread;
1532aae3547bSMateusz Guzik 	if (td->td_tid == tid) {
1533aae3547bSMateusz Guzik 		if (pid != -1 && td->td_proc->p_pid != pid)
1534aae3547bSMateusz Guzik 			return (NULL);
1535aae3547bSMateusz Guzik 		PROC_LOCK(td->td_proc);
1536cf7d9a8cSDavid Xu 		return (td);
1537cf7d9a8cSDavid Xu 	}
1538cf7d9a8cSDavid Xu 
1539aae3547bSMateusz Guzik 	for (;;) {
1540aae3547bSMateusz Guzik 		if (!tdfind_hash(tid, pid, &p, &td))
1541aae3547bSMateusz Guzik 			return (NULL);
1542aae3547bSMateusz Guzik 		PROC_LOCK(p);
1543aae3547bSMateusz Guzik 		if (td->td_tid != tid) {
1544aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1545aae3547bSMateusz Guzik 			continue;
1546aae3547bSMateusz Guzik 		}
1547aae3547bSMateusz Guzik 		if (td->td_proc != p) {
1548aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1549aae3547bSMateusz Guzik 			continue;
1550aae3547bSMateusz Guzik 		}
1551aae3547bSMateusz Guzik 		if (p->p_state == PRS_NEW) {
1552aae3547bSMateusz Guzik 			PROC_UNLOCK(p);
1553aae3547bSMateusz Guzik 			return (NULL);
1554aae3547bSMateusz Guzik 		}
1555aae3547bSMateusz Guzik 		return (td);
1556aae3547bSMateusz Guzik 	}
1557aae3547bSMateusz Guzik }
1558aae3547bSMateusz Guzik 
1559cf7d9a8cSDavid Xu void
1560cf7d9a8cSDavid Xu tidhash_add(struct thread *td)
1561cf7d9a8cSDavid Xu {
156226007fe3SMateusz Guzik 	rw_wlock(TIDHASHLOCK(td->td_tid));
1563cf7d9a8cSDavid Xu 	LIST_INSERT_HEAD(TIDHASH(td->td_tid), td, td_hash);
156426007fe3SMateusz Guzik 	rw_wunlock(TIDHASHLOCK(td->td_tid));
1565cf7d9a8cSDavid Xu }
1566cf7d9a8cSDavid Xu 
1567cf7d9a8cSDavid Xu void
1568cf7d9a8cSDavid Xu tidhash_remove(struct thread *td)
1569cf7d9a8cSDavid Xu {
157026007fe3SMateusz Guzik 
157126007fe3SMateusz Guzik 	rw_wlock(TIDHASHLOCK(td->td_tid));
1572cf7d9a8cSDavid Xu 	LIST_REMOVE(td, td_hash);
157326007fe3SMateusz Guzik 	rw_wunlock(TIDHASHLOCK(td->td_tid));
1574cf7d9a8cSDavid Xu }
1575