kern_umtx.c revision 177849
1139804Simp/*-
2139013Sdavidxu * Copyright (c) 2004, David Xu <davidxu@freebsd.org>
3112904Sjeff * Copyright (c) 2002, Jeffrey Roberson <jeff@freebsd.org>
4112904Sjeff * All rights reserved.
5112904Sjeff *
6112904Sjeff * Redistribution and use in source and binary forms, with or without
7112904Sjeff * modification, are permitted provided that the following conditions
8112904Sjeff * are met:
9112904Sjeff * 1. Redistributions of source code must retain the above copyright
10112904Sjeff *    notice unmodified, this list of conditions, and the following
11112904Sjeff *    disclaimer.
12112904Sjeff * 2. Redistributions in binary form must reproduce the above copyright
13112904Sjeff *    notice, this list of conditions and the following disclaimer in the
14112904Sjeff *    documentation and/or other materials provided with the distribution.
15112904Sjeff *
16112904Sjeff * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17112904Sjeff * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18112904Sjeff * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19112904Sjeff * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20112904Sjeff * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21112904Sjeff * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22112904Sjeff * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23112904Sjeff * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24112904Sjeff * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25112904Sjeff * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26112904Sjeff */
27112904Sjeff
28116182Sobrien#include <sys/cdefs.h>
29116182Sobrien__FBSDID("$FreeBSD: head/sys/kern/kern_umtx.c 177849 2008-04-02 04:26:59Z davidxu $");
30116182Sobrien
31162536Sdavidxu#include "opt_compat.h"
32112904Sjeff#include <sys/param.h>
33112904Sjeff#include <sys/kernel.h>
34131431Smarcel#include <sys/limits.h>
35112904Sjeff#include <sys/lock.h>
36115765Sjeff#include <sys/malloc.h>
37112904Sjeff#include <sys/mutex.h>
38164033Srwatson#include <sys/priv.h>
39112904Sjeff#include <sys/proc.h>
40161678Sdavidxu#include <sys/sched.h>
41165369Sdavidxu#include <sys/smp.h>
42161678Sdavidxu#include <sys/sysctl.h>
43112904Sjeff#include <sys/sysent.h>
44112904Sjeff#include <sys/systm.h>
45112904Sjeff#include <sys/sysproto.h>
46139013Sdavidxu#include <sys/eventhandler.h>
47112904Sjeff#include <sys/umtx.h>
48112904Sjeff
49139013Sdavidxu#include <vm/vm.h>
50139013Sdavidxu#include <vm/vm_param.h>
51139013Sdavidxu#include <vm/pmap.h>
52139013Sdavidxu#include <vm/vm_map.h>
53139013Sdavidxu#include <vm/vm_object.h>
54139013Sdavidxu
55165369Sdavidxu#include <machine/cpu.h>
56165369Sdavidxu
57162536Sdavidxu#ifdef COMPAT_IA32
58162536Sdavidxu#include <compat/freebsd32/freebsd32_proto.h>
59162536Sdavidxu#endif
60162536Sdavidxu
61161678Sdavidxu#define TYPE_SIMPLE_LOCK	0
62161678Sdavidxu#define TYPE_SIMPLE_WAIT	1
63161678Sdavidxu#define TYPE_NORMAL_UMUTEX	2
64161678Sdavidxu#define TYPE_PI_UMUTEX		3
65161678Sdavidxu#define TYPE_PP_UMUTEX		4
66161678Sdavidxu#define TYPE_CV			5
67177848Sdavidxu#define TYPE_RWLOCK		6
68139013Sdavidxu
69161678Sdavidxu/* Key to represent a unique userland synchronous object */
70139013Sdavidxustruct umtx_key {
71161678Sdavidxu	int	hash;
72139013Sdavidxu	int	type;
73161678Sdavidxu	int	shared;
74139013Sdavidxu	union {
75139013Sdavidxu		struct {
76139013Sdavidxu			vm_object_t	object;
77161678Sdavidxu			uintptr_t	offset;
78139013Sdavidxu		} shared;
79139013Sdavidxu		struct {
80161678Sdavidxu			struct vmspace	*vs;
81161678Sdavidxu			uintptr_t	addr;
82139013Sdavidxu		} private;
83139013Sdavidxu		struct {
84161678Sdavidxu			void		*a;
85161678Sdavidxu			uintptr_t	b;
86139013Sdavidxu		} both;
87139013Sdavidxu	} info;
88139013Sdavidxu};
89139013Sdavidxu
90161678Sdavidxu/* Priority inheritance mutex info. */
91161678Sdavidxustruct umtx_pi {
92161678Sdavidxu	/* Owner thread */
93161678Sdavidxu	struct thread		*pi_owner;
94161678Sdavidxu
95161678Sdavidxu	/* Reference count */
96161678Sdavidxu	int			pi_refcount;
97161678Sdavidxu
98161678Sdavidxu 	/* List entry to link umtx holding by thread */
99161678Sdavidxu	TAILQ_ENTRY(umtx_pi)	pi_link;
100161678Sdavidxu
101161678Sdavidxu	/* List entry in hash */
102161678Sdavidxu	TAILQ_ENTRY(umtx_pi)	pi_hashlink;
103161678Sdavidxu
104161678Sdavidxu	/* List for waiters */
105161678Sdavidxu	TAILQ_HEAD(,umtx_q)	pi_blocked;
106161678Sdavidxu
107161678Sdavidxu	/* Identify a userland lock object */
108161678Sdavidxu	struct umtx_key		pi_key;
109161678Sdavidxu};
110161678Sdavidxu
111161678Sdavidxu/* A userland synchronous object user. */
112115765Sjeffstruct umtx_q {
113161678Sdavidxu	/* Linked list for the hash. */
114161678Sdavidxu	TAILQ_ENTRY(umtx_q)	uq_link;
115161678Sdavidxu
116161678Sdavidxu	/* Umtx key. */
117161678Sdavidxu	struct umtx_key		uq_key;
118161678Sdavidxu
119161678Sdavidxu	/* Umtx flags. */
120161678Sdavidxu	int			uq_flags;
121161678Sdavidxu#define UQF_UMTXQ	0x0001
122161678Sdavidxu
123161678Sdavidxu	/* The thread waits on. */
124161678Sdavidxu	struct thread		*uq_thread;
125161678Sdavidxu
126161678Sdavidxu	/*
127161678Sdavidxu	 * Blocked on PI mutex. read can use chain lock
128170300Sjeff	 * or umtx_lock, write must have both chain lock and
129170300Sjeff	 * umtx_lock being hold.
130161678Sdavidxu	 */
131161678Sdavidxu	struct umtx_pi		*uq_pi_blocked;
132161678Sdavidxu
133161678Sdavidxu	/* On blocked list */
134161678Sdavidxu	TAILQ_ENTRY(umtx_q)	uq_lockq;
135161678Sdavidxu
136161678Sdavidxu	/* Thread contending with us */
137161678Sdavidxu	TAILQ_HEAD(,umtx_pi)	uq_pi_contested;
138161678Sdavidxu
139161742Sdavidxu	/* Inherited priority from PP mutex */
140161678Sdavidxu	u_char			uq_inherited_pri;
141115765Sjeff};
142115765Sjeff
143161678SdavidxuTAILQ_HEAD(umtxq_head, umtx_q);
144161678Sdavidxu
145161678Sdavidxu/* Userland lock object's wait-queue chain */
146138224Sdavidxustruct umtxq_chain {
147161678Sdavidxu	/* Lock for this chain. */
148161678Sdavidxu	struct mtx		uc_lock;
149161678Sdavidxu
150161678Sdavidxu	/* List of sleep queues. */
151177848Sdavidxu	struct umtxq_head	uc_queue[2];
152177848Sdavidxu#define UMTX_SHARED_QUEUE	0
153177848Sdavidxu#define UMTX_EXCLUSIVE_QUEUE	1
154161678Sdavidxu
155161678Sdavidxu	/* Busy flag */
156161678Sdavidxu	char			uc_busy;
157161678Sdavidxu
158161678Sdavidxu	/* Chain lock waiters */
159158377Sdavidxu	int			uc_waiters;
160161678Sdavidxu
161161678Sdavidxu	/* All PI in the list */
162161678Sdavidxu	TAILQ_HEAD(,umtx_pi)	uc_pi_list;
163138224Sdavidxu};
164115765Sjeff
165161678Sdavidxu#define	UMTXQ_LOCKED_ASSERT(uc)		mtx_assert(&(uc)->uc_lock, MA_OWNED)
166161678Sdavidxu
167161678Sdavidxu/*
168161678Sdavidxu * Don't propagate time-sharing priority, there is a security reason,
169161678Sdavidxu * a user can simply introduce PI-mutex, let thread A lock the mutex,
170161678Sdavidxu * and let another thread B block on the mutex, because B is
171161678Sdavidxu * sleeping, its priority will be boosted, this causes A's priority to
172161678Sdavidxu * be boosted via priority propagating too and will never be lowered even
173161678Sdavidxu * if it is using 100%CPU, this is unfair to other processes.
174161678Sdavidxu */
175161678Sdavidxu
176163709Sjb#define UPRI(td)	(((td)->td_user_pri >= PRI_MIN_TIMESHARE &&\
177163709Sjb			  (td)->td_user_pri <= PRI_MAX_TIMESHARE) ?\
178163709Sjb			 PRI_MAX_TIMESHARE : (td)->td_user_pri)
179161678Sdavidxu
180138224Sdavidxu#define	GOLDEN_RATIO_PRIME	2654404609U
181138224Sdavidxu#define	UMTX_CHAINS		128
182138224Sdavidxu#define	UMTX_SHIFTS		(__WORD_BIT - 7)
183115765Sjeff
184161678Sdavidxu#define THREAD_SHARE		0
185161678Sdavidxu#define PROCESS_SHARE		1
186161678Sdavidxu#define AUTO_SHARE		2
187161678Sdavidxu
188161678Sdavidxu#define	GET_SHARE(flags)	\
189161678Sdavidxu    (((flags) & USYNC_PROCESS_SHARED) == 0 ? THREAD_SHARE : PROCESS_SHARE)
190161678Sdavidxu
191177848Sdavidxu#define BUSY_SPINS		200
192177848Sdavidxu
193161678Sdavidxustatic uma_zone_t		umtx_pi_zone;
194161678Sdavidxustatic struct umtxq_chain	umtxq_chains[UMTX_CHAINS];
195138224Sdavidxustatic MALLOC_DEFINE(M_UMTX, "umtx", "UMTX queue memory");
196161678Sdavidxustatic int			umtx_pi_allocated;
197115310Sjeff
198161678SdavidxuSYSCTL_NODE(_debug, OID_AUTO, umtx, CTLFLAG_RW, 0, "umtx debug");
199161678SdavidxuSYSCTL_INT(_debug_umtx, OID_AUTO, umtx_pi_allocated, CTLFLAG_RD,
200161678Sdavidxu    &umtx_pi_allocated, 0, "Allocated umtx_pi");
201161678Sdavidxu
202161678Sdavidxustatic void umtxq_sysinit(void *);
203161678Sdavidxustatic void umtxq_hash(struct umtx_key *key);
204161678Sdavidxustatic struct umtxq_chain *umtxq_getchain(struct umtx_key *key);
205139013Sdavidxustatic void umtxq_lock(struct umtx_key *key);
206139013Sdavidxustatic void umtxq_unlock(struct umtx_key *key);
207139257Sdavidxustatic void umtxq_busy(struct umtx_key *key);
208139257Sdavidxustatic void umtxq_unbusy(struct umtx_key *key);
209177848Sdavidxustatic void umtxq_insert_queue(struct umtx_q *uq, int q);
210177848Sdavidxustatic void umtxq_remove_queue(struct umtx_q *uq, int q);
211161678Sdavidxustatic int umtxq_sleep(struct umtx_q *uq, const char *wmesg, int timo);
212139257Sdavidxustatic int umtxq_count(struct umtx_key *key);
213139013Sdavidxustatic int umtx_key_match(const struct umtx_key *k1, const struct umtx_key *k2);
214161678Sdavidxustatic int umtx_key_get(void *addr, int type, int share,
215139013Sdavidxu	struct umtx_key *key);
216139013Sdavidxustatic void umtx_key_release(struct umtx_key *key);
217163697Sdavidxustatic struct umtx_pi *umtx_pi_alloc(int);
218161678Sdavidxustatic void umtx_pi_free(struct umtx_pi *pi);
219174701Sdavidxustatic void umtx_pi_adjust_locked(struct thread *td, u_char oldpri);
220161678Sdavidxustatic int do_unlock_pp(struct thread *td, struct umutex *m, uint32_t flags);
221161678Sdavidxustatic void umtx_thread_cleanup(struct thread *td);
222161678Sdavidxustatic void umtx_exec_hook(void *arg __unused, struct proc *p __unused,
223161678Sdavidxu	struct image_params *imgp __unused);
224161678SdavidxuSYSINIT(umtx, SI_SUB_EVENTHANDLER+1, SI_ORDER_MIDDLE, umtxq_sysinit, NULL);
225115310Sjeff
226177848Sdavidxu#define umtxq_signal(key, nwake)	umtxq_signal_queue((key), (nwake), UMTX_SHARED_QUEUE)
227177848Sdavidxu#define umtxq_insert(uq)	umtxq_insert_queue((uq), UMTX_SHARED_QUEUE)
228177848Sdavidxu#define umtxq_remove(uq)	umtxq_remove_queue((uq), UMTX_SHARED_QUEUE)
229177848Sdavidxu
230170300Sjeffstatic struct mtx umtx_lock;
231170300Sjeff
232161678Sdavidxustatic void
233161678Sdavidxuumtxq_sysinit(void *arg __unused)
234161678Sdavidxu{
235161678Sdavidxu	int i;
236138224Sdavidxu
237161678Sdavidxu	umtx_pi_zone = uma_zcreate("umtx pi", sizeof(struct umtx_pi),
238161678Sdavidxu		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
239161678Sdavidxu	for (i = 0; i < UMTX_CHAINS; ++i) {
240161678Sdavidxu		mtx_init(&umtxq_chains[i].uc_lock, "umtxql", NULL,
241161678Sdavidxu			 MTX_DEF | MTX_DUPOK);
242177848Sdavidxu		TAILQ_INIT(&umtxq_chains[i].uc_queue[0]);
243177848Sdavidxu		TAILQ_INIT(&umtxq_chains[i].uc_queue[1]);
244161678Sdavidxu		TAILQ_INIT(&umtxq_chains[i].uc_pi_list);
245161678Sdavidxu		umtxq_chains[i].uc_busy = 0;
246161678Sdavidxu		umtxq_chains[i].uc_waiters = 0;
247161678Sdavidxu	}
248170300Sjeff	mtx_init(&umtx_lock, "umtx lock", NULL, MTX_SPIN);
249161678Sdavidxu	EVENTHANDLER_REGISTER(process_exec, umtx_exec_hook, NULL,
250161678Sdavidxu	    EVENTHANDLER_PRI_ANY);
251161678Sdavidxu}
252161678Sdavidxu
253143149Sdavidxustruct umtx_q *
254143149Sdavidxuumtxq_alloc(void)
255143149Sdavidxu{
256161678Sdavidxu	struct umtx_q *uq;
257161678Sdavidxu
258161678Sdavidxu	uq = malloc(sizeof(struct umtx_q), M_UMTX, M_WAITOK | M_ZERO);
259161678Sdavidxu	TAILQ_INIT(&uq->uq_pi_contested);
260161678Sdavidxu	uq->uq_inherited_pri = PRI_MAX;
261161678Sdavidxu	return (uq);
262143149Sdavidxu}
263143149Sdavidxu
264143149Sdavidxuvoid
265143149Sdavidxuumtxq_free(struct umtx_q *uq)
266143149Sdavidxu{
267143149Sdavidxu	free(uq, M_UMTX);
268143149Sdavidxu}
269143149Sdavidxu
270161678Sdavidxustatic inline void
271139013Sdavidxuumtxq_hash(struct umtx_key *key)
272138224Sdavidxu{
273161678Sdavidxu	unsigned n = (uintptr_t)key->info.both.a + key->info.both.b;
274161678Sdavidxu	key->hash = ((n * GOLDEN_RATIO_PRIME) >> UMTX_SHIFTS) % UMTX_CHAINS;
275138224Sdavidxu}
276138224Sdavidxu
277139013Sdavidxustatic inline int
278139013Sdavidxuumtx_key_match(const struct umtx_key *k1, const struct umtx_key *k2)
279139013Sdavidxu{
280139013Sdavidxu	return (k1->type == k2->type &&
281161678Sdavidxu		k1->info.both.a == k2->info.both.a &&
282161678Sdavidxu	        k1->info.both.b == k2->info.both.b);
283139013Sdavidxu}
284139013Sdavidxu
285161678Sdavidxustatic inline struct umtxq_chain *
286161678Sdavidxuumtxq_getchain(struct umtx_key *key)
287139013Sdavidxu{
288161678Sdavidxu	return (&umtxq_chains[key->hash]);
289139013Sdavidxu}
290139013Sdavidxu
291161678Sdavidxu/*
292177848Sdavidxu * Lock a chain.
293161678Sdavidxu */
294138224Sdavidxustatic inline void
295177848Sdavidxuumtxq_lock(struct umtx_key *key)
296139257Sdavidxu{
297161678Sdavidxu	struct umtxq_chain *uc;
298139257Sdavidxu
299161678Sdavidxu	uc = umtxq_getchain(key);
300177848Sdavidxu	mtx_lock(&uc->uc_lock);
301139257Sdavidxu}
302139257Sdavidxu
303161678Sdavidxu/*
304177848Sdavidxu * Unlock a chain.
305161678Sdavidxu */
306139257Sdavidxustatic inline void
307177848Sdavidxuumtxq_unlock(struct umtx_key *key)
308139257Sdavidxu{
309161678Sdavidxu	struct umtxq_chain *uc;
310139257Sdavidxu
311161678Sdavidxu	uc = umtxq_getchain(key);
312177848Sdavidxu	mtx_unlock(&uc->uc_lock);
313139257Sdavidxu}
314139257Sdavidxu
315161678Sdavidxu/*
316177848Sdavidxu * Set chain to busy state when following operation
317177848Sdavidxu * may be blocked (kernel mutex can not be used).
318161678Sdavidxu */
319139257Sdavidxustatic inline void
320177848Sdavidxuumtxq_busy(struct umtx_key *key)
321138224Sdavidxu{
322161678Sdavidxu	struct umtxq_chain *uc;
323161678Sdavidxu
324161678Sdavidxu	uc = umtxq_getchain(key);
325177848Sdavidxu	mtx_assert(&uc->uc_lock, MA_OWNED);
326177848Sdavidxu	if (uc->uc_busy) {
327177848Sdavidxu		int count = BUSY_SPINS;
328177848Sdavidxu		if (count > 0) {
329177848Sdavidxu			umtxq_unlock(key);
330177848Sdavidxu			while (uc->uc_busy && --count > 0)
331177848Sdavidxu				cpu_spinwait();
332177848Sdavidxu			umtxq_lock(key);
333177848Sdavidxu		}
334177848Sdavidxu		while (uc->uc_busy != 0) {
335177848Sdavidxu			uc->uc_waiters++;
336177848Sdavidxu			msleep(uc, &uc->uc_lock, 0, "umtxqb", 0);
337177848Sdavidxu			uc->uc_waiters--;
338177848Sdavidxu		}
339177848Sdavidxu	}
340177848Sdavidxu	uc->uc_busy = 1;
341138224Sdavidxu}
342138224Sdavidxu
343161678Sdavidxu/*
344177848Sdavidxu * Unbusy a chain.
345161678Sdavidxu */
346138225Sdavidxustatic inline void
347177848Sdavidxuumtxq_unbusy(struct umtx_key *key)
348138224Sdavidxu{
349161678Sdavidxu	struct umtxq_chain *uc;
350161678Sdavidxu
351161678Sdavidxu	uc = umtxq_getchain(key);
352177848Sdavidxu	mtx_assert(&uc->uc_lock, MA_OWNED);
353177848Sdavidxu	KASSERT(uc->uc_busy != 0, ("not busy"));
354177848Sdavidxu	uc->uc_busy = 0;
355177848Sdavidxu	if (uc->uc_waiters)
356177848Sdavidxu		wakeup_one(uc);
357138224Sdavidxu}
358138224Sdavidxu
359139013Sdavidxustatic inline void
360177848Sdavidxuumtxq_insert_queue(struct umtx_q *uq, int q)
361115765Sjeff{
362161678Sdavidxu	struct umtxq_chain *uc;
363139013Sdavidxu
364161678Sdavidxu	uc = umtxq_getchain(&uq->uq_key);
365161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
366177848Sdavidxu	TAILQ_INSERT_TAIL(&uc->uc_queue[q], uq, uq_link);
367158718Sdavidxu	uq->uq_flags |= UQF_UMTXQ;
368139013Sdavidxu}
369139013Sdavidxu
370139013Sdavidxustatic inline void
371177848Sdavidxuumtxq_remove_queue(struct umtx_q *uq, int q)
372139013Sdavidxu{
373161678Sdavidxu	struct umtxq_chain *uc;
374161678Sdavidxu
375161678Sdavidxu	uc = umtxq_getchain(&uq->uq_key);
376161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
377158718Sdavidxu	if (uq->uq_flags & UQF_UMTXQ) {
378177848Sdavidxu		TAILQ_REMOVE(&uc->uc_queue[q], uq, uq_link);
379158718Sdavidxu		uq->uq_flags &= ~UQF_UMTXQ;
380139013Sdavidxu	}
381139013Sdavidxu}
382139013Sdavidxu
383161678Sdavidxu/*
384161678Sdavidxu * Check if there are multiple waiters
385161678Sdavidxu */
386139013Sdavidxustatic int
387139013Sdavidxuumtxq_count(struct umtx_key *key)
388139013Sdavidxu{
389161678Sdavidxu	struct umtxq_chain *uc;
390115765Sjeff	struct umtx_q *uq;
391161678Sdavidxu	int count = 0;
392115765Sjeff
393161678Sdavidxu	uc = umtxq_getchain(key);
394161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
395177848Sdavidxu	TAILQ_FOREACH(uq, &uc->uc_queue[UMTX_SHARED_QUEUE], uq_link) {
396139013Sdavidxu		if (umtx_key_match(&uq->uq_key, key)) {
397139013Sdavidxu			if (++count > 1)
398139013Sdavidxu				break;
399139013Sdavidxu		}
400115765Sjeff	}
401139013Sdavidxu	return (count);
402115765Sjeff}
403115765Sjeff
404161678Sdavidxu/*
405161678Sdavidxu * Check if there are multiple PI waiters and returns first
406161678Sdavidxu * waiter.
407161678Sdavidxu */
408139257Sdavidxustatic int
409161678Sdavidxuumtxq_count_pi(struct umtx_key *key, struct umtx_q **first)
410161678Sdavidxu{
411161678Sdavidxu	struct umtxq_chain *uc;
412161678Sdavidxu	struct umtx_q *uq;
413161678Sdavidxu	int count = 0;
414161678Sdavidxu
415161678Sdavidxu	*first = NULL;
416161678Sdavidxu	uc = umtxq_getchain(key);
417161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
418177848Sdavidxu	TAILQ_FOREACH(uq, &uc->uc_queue[UMTX_SHARED_QUEUE], uq_link) {
419161678Sdavidxu		if (umtx_key_match(&uq->uq_key, key)) {
420161678Sdavidxu			if (++count > 1)
421161678Sdavidxu				break;
422161678Sdavidxu			*first = uq;
423161678Sdavidxu		}
424161678Sdavidxu	}
425161678Sdavidxu	return (count);
426161678Sdavidxu}
427161678Sdavidxu
428161678Sdavidxu/*
429161678Sdavidxu * Wake up threads waiting on an userland object.
430161678Sdavidxu */
431177848Sdavidxu
432161678Sdavidxustatic int
433177848Sdavidxuumtxq_signal_queue(struct umtx_key *key, int n_wake, int q)
434115765Sjeff{
435161678Sdavidxu	struct umtxq_chain *uc;
436139257Sdavidxu	struct umtx_q *uq, *next;
437161678Sdavidxu	int ret;
438115765Sjeff
439139257Sdavidxu	ret = 0;
440161678Sdavidxu	uc = umtxq_getchain(key);
441161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
442177848Sdavidxu	TAILQ_FOREACH_SAFE(uq, &uc->uc_queue[q], uq_link, next) {
443139013Sdavidxu		if (umtx_key_match(&uq->uq_key, key)) {
444177848Sdavidxu			umtxq_remove_queue(uq, q);
445161678Sdavidxu			wakeup(uq);
446139257Sdavidxu			if (++ret >= n_wake)
447139257Sdavidxu				break;
448139013Sdavidxu		}
449139013Sdavidxu	}
450139257Sdavidxu	return (ret);
451138224Sdavidxu}
452138224Sdavidxu
453177848Sdavidxu
454161678Sdavidxu/*
455161678Sdavidxu * Wake up specified thread.
456161678Sdavidxu */
457161678Sdavidxustatic inline void
458161678Sdavidxuumtxq_signal_thread(struct umtx_q *uq)
459161678Sdavidxu{
460161678Sdavidxu	struct umtxq_chain *uc;
461161678Sdavidxu
462161678Sdavidxu	uc = umtxq_getchain(&uq->uq_key);
463161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
464161678Sdavidxu	umtxq_remove(uq);
465161678Sdavidxu	wakeup(uq);
466161678Sdavidxu}
467161678Sdavidxu
468161678Sdavidxu/*
469161678Sdavidxu * Put thread into sleep state, before sleeping, check if
470161678Sdavidxu * thread was removed from umtx queue.
471161678Sdavidxu */
472138224Sdavidxustatic inline int
473161678Sdavidxuumtxq_sleep(struct umtx_q *uq, const char *wmesg, int timo)
474138224Sdavidxu{
475161678Sdavidxu	struct umtxq_chain *uc;
476161678Sdavidxu	int error;
477161678Sdavidxu
478161678Sdavidxu	uc = umtxq_getchain(&uq->uq_key);
479161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
480161678Sdavidxu	if (!(uq->uq_flags & UQF_UMTXQ))
481161678Sdavidxu		return (0);
482161678Sdavidxu	error = msleep(uq, &uc->uc_lock, PCATCH, wmesg, timo);
483139751Sdavidxu	if (error == EWOULDBLOCK)
484139751Sdavidxu		error = ETIMEDOUT;
485139751Sdavidxu	return (error);
486138224Sdavidxu}
487138224Sdavidxu
488161678Sdavidxu/*
489161678Sdavidxu * Convert userspace address into unique logical address.
490161678Sdavidxu */
491139013Sdavidxustatic int
492161678Sdavidxuumtx_key_get(void *addr, int type, int share, struct umtx_key *key)
493139013Sdavidxu{
494161678Sdavidxu	struct thread *td = curthread;
495139013Sdavidxu	vm_map_t map;
496139013Sdavidxu	vm_map_entry_t entry;
497139013Sdavidxu	vm_pindex_t pindex;
498139013Sdavidxu	vm_prot_t prot;
499139013Sdavidxu	boolean_t wired;
500139013Sdavidxu
501161678Sdavidxu	key->type = type;
502161678Sdavidxu	if (share == THREAD_SHARE) {
503161678Sdavidxu		key->shared = 0;
504161678Sdavidxu		key->info.private.vs = td->td_proc->p_vmspace;
505161678Sdavidxu		key->info.private.addr = (uintptr_t)addr;
506163677Sdavidxu	} else {
507163677Sdavidxu		MPASS(share == PROCESS_SHARE || share == AUTO_SHARE);
508161678Sdavidxu		map = &td->td_proc->p_vmspace->vm_map;
509161678Sdavidxu		if (vm_map_lookup(&map, (vm_offset_t)addr, VM_PROT_WRITE,
510161678Sdavidxu		    &entry, &key->info.shared.object, &pindex, &prot,
511161678Sdavidxu		    &wired) != KERN_SUCCESS) {
512161678Sdavidxu			return EFAULT;
513161678Sdavidxu		}
514161678Sdavidxu
515161678Sdavidxu		if ((share == PROCESS_SHARE) ||
516161678Sdavidxu		    (share == AUTO_SHARE &&
517161678Sdavidxu		     VM_INHERIT_SHARE == entry->inheritance)) {
518161678Sdavidxu			key->shared = 1;
519161678Sdavidxu			key->info.shared.offset = entry->offset + entry->start -
520161678Sdavidxu				(vm_offset_t)addr;
521161678Sdavidxu			vm_object_reference(key->info.shared.object);
522161678Sdavidxu		} else {
523161678Sdavidxu			key->shared = 0;
524161678Sdavidxu			key->info.private.vs = td->td_proc->p_vmspace;
525161678Sdavidxu			key->info.private.addr = (uintptr_t)addr;
526161678Sdavidxu		}
527161678Sdavidxu		vm_map_lookup_done(map, entry);
528139013Sdavidxu	}
529139013Sdavidxu
530161678Sdavidxu	umtxq_hash(key);
531139013Sdavidxu	return (0);
532139013Sdavidxu}
533139013Sdavidxu
534161678Sdavidxu/*
535161678Sdavidxu * Release key.
536161678Sdavidxu */
537139013Sdavidxustatic inline void
538139013Sdavidxuumtx_key_release(struct umtx_key *key)
539139013Sdavidxu{
540161678Sdavidxu	if (key->shared)
541139013Sdavidxu		vm_object_deallocate(key->info.shared.object);
542139013Sdavidxu}
543139013Sdavidxu
544161678Sdavidxu/*
545161678Sdavidxu * Lock a umtx object.
546161678Sdavidxu */
547139013Sdavidxustatic int
548163449Sdavidxu_do_lock_umtx(struct thread *td, struct umtx *umtx, u_long id, int timo)
549112904Sjeff{
550143149Sdavidxu	struct umtx_q *uq;
551163449Sdavidxu	u_long owner;
552163449Sdavidxu	u_long old;
553138224Sdavidxu	int error = 0;
554112904Sjeff
555143149Sdavidxu	uq = td->td_umtxq;
556161678Sdavidxu
557112904Sjeff	/*
558161678Sdavidxu	 * Care must be exercised when dealing with umtx structure. It
559112904Sjeff	 * can fault on any access.
560112904Sjeff	 */
561112904Sjeff	for (;;) {
562112904Sjeff		/*
563112904Sjeff		 * Try the uncontested case.  This should be done in userland.
564112904Sjeff		 */
565163449Sdavidxu		owner = casuword(&umtx->u_owner, UMTX_UNOWNED, id);
566112904Sjeff
567138224Sdavidxu		/* The acquire succeeded. */
568138224Sdavidxu		if (owner == UMTX_UNOWNED)
569138224Sdavidxu			return (0);
570138224Sdavidxu
571115765Sjeff		/* The address was invalid. */
572115765Sjeff		if (owner == -1)
573115765Sjeff			return (EFAULT);
574115765Sjeff
575115765Sjeff		/* If no one owns it but it is contested try to acquire it. */
576115765Sjeff		if (owner == UMTX_CONTESTED) {
577163449Sdavidxu			owner = casuword(&umtx->u_owner,
578139013Sdavidxu			    UMTX_CONTESTED, id | UMTX_CONTESTED);
579115765Sjeff
580138224Sdavidxu			if (owner == UMTX_CONTESTED)
581138224Sdavidxu				return (0);
582138224Sdavidxu
583115765Sjeff			/* The address was invalid. */
584115765Sjeff			if (owner == -1)
585115765Sjeff				return (EFAULT);
586115765Sjeff
587115765Sjeff			/* If this failed the lock has changed, restart. */
588115765Sjeff			continue;
589112904Sjeff		}
590112904Sjeff
591138224Sdavidxu		/*
592138224Sdavidxu		 * If we caught a signal, we have retried and now
593138224Sdavidxu		 * exit immediately.
594138224Sdavidxu		 */
595161678Sdavidxu		if (error != 0)
596138224Sdavidxu			return (error);
597112904Sjeff
598161678Sdavidxu		if ((error = umtx_key_get(umtx, TYPE_SIMPLE_LOCK,
599161678Sdavidxu			AUTO_SHARE, &uq->uq_key)) != 0)
600161678Sdavidxu			return (error);
601161678Sdavidxu
602161678Sdavidxu		umtxq_lock(&uq->uq_key);
603161678Sdavidxu		umtxq_busy(&uq->uq_key);
604161678Sdavidxu		umtxq_insert(uq);
605161678Sdavidxu		umtxq_unbusy(&uq->uq_key);
606161678Sdavidxu		umtxq_unlock(&uq->uq_key);
607161678Sdavidxu
608112904Sjeff		/*
609112904Sjeff		 * Set the contested bit so that a release in user space
610112904Sjeff		 * knows to use the system call for unlock.  If this fails
611112904Sjeff		 * either some one else has acquired the lock or it has been
612112904Sjeff		 * released.
613112904Sjeff		 */
614163449Sdavidxu		old = casuword(&umtx->u_owner, owner, owner | UMTX_CONTESTED);
615112904Sjeff
616112904Sjeff		/* The address was invalid. */
617112967Sjake		if (old == -1) {
618143149Sdavidxu			umtxq_lock(&uq->uq_key);
619143149Sdavidxu			umtxq_remove(uq);
620143149Sdavidxu			umtxq_unlock(&uq->uq_key);
621143149Sdavidxu			umtx_key_release(&uq->uq_key);
622115765Sjeff			return (EFAULT);
623112904Sjeff		}
624112904Sjeff
625112904Sjeff		/*
626115765Sjeff		 * We set the contested bit, sleep. Otherwise the lock changed
627117685Smtm		 * and we need to retry or we lost a race to the thread
628117685Smtm		 * unlocking the umtx.
629112904Sjeff		 */
630143149Sdavidxu		umtxq_lock(&uq->uq_key);
631161678Sdavidxu		if (old == owner)
632161678Sdavidxu			error = umtxq_sleep(uq, "umtx", timo);
633143149Sdavidxu		umtxq_remove(uq);
634143149Sdavidxu		umtxq_unlock(&uq->uq_key);
635143149Sdavidxu		umtx_key_release(&uq->uq_key);
636112904Sjeff	}
637117743Smtm
638117743Smtm	return (0);
639112904Sjeff}
640112904Sjeff
641161678Sdavidxu/*
642161678Sdavidxu * Lock a umtx object.
643161678Sdavidxu */
644139013Sdavidxustatic int
645163449Sdavidxudo_lock_umtx(struct thread *td, struct umtx *umtx, u_long id,
646140245Sdavidxu	struct timespec *timeout)
647112904Sjeff{
648140245Sdavidxu	struct timespec ts, ts2, ts3;
649139013Sdavidxu	struct timeval tv;
650140245Sdavidxu	int error;
651139013Sdavidxu
652140245Sdavidxu	if (timeout == NULL) {
653162536Sdavidxu		error = _do_lock_umtx(td, umtx, id, 0);
654162030Sdavidxu		/* Mutex locking is restarted if it is interrupted. */
655162030Sdavidxu		if (error == EINTR)
656162030Sdavidxu			error = ERESTART;
657139013Sdavidxu	} else {
658140245Sdavidxu		getnanouptime(&ts);
659140245Sdavidxu		timespecadd(&ts, timeout);
660140245Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, timeout);
661139013Sdavidxu		for (;;) {
662162536Sdavidxu			error = _do_lock_umtx(td, umtx, id, tvtohz(&tv));
663140245Sdavidxu			if (error != ETIMEDOUT)
664140245Sdavidxu				break;
665140245Sdavidxu			getnanouptime(&ts2);
666140245Sdavidxu			if (timespeccmp(&ts2, &ts, >=)) {
667139751Sdavidxu				error = ETIMEDOUT;
668139013Sdavidxu				break;
669139013Sdavidxu			}
670140245Sdavidxu			ts3 = ts;
671140245Sdavidxu			timespecsub(&ts3, &ts2);
672140245Sdavidxu			TIMESPEC_TO_TIMEVAL(&tv, &ts3);
673139013Sdavidxu		}
674162030Sdavidxu		/* Timed-locking is not restarted. */
675162030Sdavidxu		if (error == ERESTART)
676162030Sdavidxu			error = EINTR;
677139013Sdavidxu	}
678139013Sdavidxu	return (error);
679139013Sdavidxu}
680139013Sdavidxu
681161678Sdavidxu/*
682161678Sdavidxu * Unlock a umtx object.
683161678Sdavidxu */
684139013Sdavidxustatic int
685163449Sdavidxudo_unlock_umtx(struct thread *td, struct umtx *umtx, u_long id)
686139013Sdavidxu{
687139013Sdavidxu	struct umtx_key key;
688163449Sdavidxu	u_long owner;
689163449Sdavidxu	u_long old;
690139257Sdavidxu	int error;
691139257Sdavidxu	int count;
692112904Sjeff
693112904Sjeff	/*
694112904Sjeff	 * Make sure we own this mtx.
695112904Sjeff	 */
696163449Sdavidxu	owner = fuword(__DEVOLATILE(u_long *, &umtx->u_owner));
697161678Sdavidxu	if (owner == -1)
698115765Sjeff		return (EFAULT);
699115765Sjeff
700139013Sdavidxu	if ((owner & ~UMTX_CONTESTED) != id)
701115765Sjeff		return (EPERM);
702112904Sjeff
703161678Sdavidxu	/* This should be done in userland */
704161678Sdavidxu	if ((owner & UMTX_CONTESTED) == 0) {
705163449Sdavidxu		old = casuword(&umtx->u_owner, owner, UMTX_UNOWNED);
706161678Sdavidxu		if (old == -1)
707161678Sdavidxu			return (EFAULT);
708161678Sdavidxu		if (old == owner)
709161678Sdavidxu			return (0);
710161855Sdavidxu		owner = old;
711161678Sdavidxu	}
712161678Sdavidxu
713117685Smtm	/* We should only ever be in here for contested locks */
714161678Sdavidxu	if ((error = umtx_key_get(umtx, TYPE_SIMPLE_LOCK, AUTO_SHARE,
715161678Sdavidxu		&key)) != 0)
716139257Sdavidxu		return (error);
717139257Sdavidxu
718139257Sdavidxu	umtxq_lock(&key);
719139257Sdavidxu	umtxq_busy(&key);
720139257Sdavidxu	count = umtxq_count(&key);
721139257Sdavidxu	umtxq_unlock(&key);
722139257Sdavidxu
723117743Smtm	/*
724117743Smtm	 * When unlocking the umtx, it must be marked as unowned if
725117743Smtm	 * there is zero or one thread only waiting for it.
726117743Smtm	 * Otherwise, it must be marked as contested.
727117743Smtm	 */
728163449Sdavidxu	old = casuword(&umtx->u_owner, owner,
729163449Sdavidxu		count <= 1 ? UMTX_UNOWNED : UMTX_CONTESTED);
730139257Sdavidxu	umtxq_lock(&key);
731161678Sdavidxu	umtxq_signal(&key,1);
732139257Sdavidxu	umtxq_unbusy(&key);
733139257Sdavidxu	umtxq_unlock(&key);
734139257Sdavidxu	umtx_key_release(&key);
735115765Sjeff	if (old == -1)
736115765Sjeff		return (EFAULT);
737138224Sdavidxu	if (old != owner)
738138224Sdavidxu		return (EINVAL);
739115765Sjeff	return (0);
740112904Sjeff}
741139013Sdavidxu
742162536Sdavidxu#ifdef COMPAT_IA32
743162536Sdavidxu
744161678Sdavidxu/*
745162536Sdavidxu * Lock a umtx object.
746162536Sdavidxu */
747162536Sdavidxustatic int
748162536Sdavidxu_do_lock_umtx32(struct thread *td, uint32_t *m, uint32_t id, int timo)
749162536Sdavidxu{
750162536Sdavidxu	struct umtx_q *uq;
751162536Sdavidxu	uint32_t owner;
752162536Sdavidxu	uint32_t old;
753162536Sdavidxu	int error = 0;
754162536Sdavidxu
755162536Sdavidxu	uq = td->td_umtxq;
756162536Sdavidxu
757162536Sdavidxu	/*
758162536Sdavidxu	 * Care must be exercised when dealing with umtx structure. It
759162536Sdavidxu	 * can fault on any access.
760162536Sdavidxu	 */
761162536Sdavidxu	for (;;) {
762162536Sdavidxu		/*
763162536Sdavidxu		 * Try the uncontested case.  This should be done in userland.
764162536Sdavidxu		 */
765162536Sdavidxu		owner = casuword32(m, UMUTEX_UNOWNED, id);
766162536Sdavidxu
767162536Sdavidxu		/* The acquire succeeded. */
768162536Sdavidxu		if (owner == UMUTEX_UNOWNED)
769162536Sdavidxu			return (0);
770162536Sdavidxu
771162536Sdavidxu		/* The address was invalid. */
772162536Sdavidxu		if (owner == -1)
773162536Sdavidxu			return (EFAULT);
774162536Sdavidxu
775162536Sdavidxu		/* If no one owns it but it is contested try to acquire it. */
776162536Sdavidxu		if (owner == UMUTEX_CONTESTED) {
777162536Sdavidxu			owner = casuword32(m,
778162536Sdavidxu			    UMUTEX_CONTESTED, id | UMUTEX_CONTESTED);
779162536Sdavidxu			if (owner == UMUTEX_CONTESTED)
780162536Sdavidxu				return (0);
781162536Sdavidxu
782162536Sdavidxu			/* The address was invalid. */
783162536Sdavidxu			if (owner == -1)
784162536Sdavidxu				return (EFAULT);
785162536Sdavidxu
786162536Sdavidxu			/* If this failed the lock has changed, restart. */
787162536Sdavidxu			continue;
788162536Sdavidxu		}
789162536Sdavidxu
790162536Sdavidxu		/*
791162536Sdavidxu		 * If we caught a signal, we have retried and now
792162536Sdavidxu		 * exit immediately.
793162536Sdavidxu		 */
794162536Sdavidxu		if (error != 0)
795162536Sdavidxu			return (error);
796162536Sdavidxu
797162536Sdavidxu		if ((error = umtx_key_get(m, TYPE_SIMPLE_LOCK,
798162536Sdavidxu			AUTO_SHARE, &uq->uq_key)) != 0)
799162536Sdavidxu			return (error);
800162536Sdavidxu
801162536Sdavidxu		umtxq_lock(&uq->uq_key);
802162536Sdavidxu		umtxq_busy(&uq->uq_key);
803162536Sdavidxu		umtxq_insert(uq);
804162536Sdavidxu		umtxq_unbusy(&uq->uq_key);
805162536Sdavidxu		umtxq_unlock(&uq->uq_key);
806162536Sdavidxu
807162536Sdavidxu		/*
808162536Sdavidxu		 * Set the contested bit so that a release in user space
809162536Sdavidxu		 * knows to use the system call for unlock.  If this fails
810162536Sdavidxu		 * either some one else has acquired the lock or it has been
811162536Sdavidxu		 * released.
812162536Sdavidxu		 */
813162536Sdavidxu		old = casuword32(m, owner, owner | UMUTEX_CONTESTED);
814162536Sdavidxu
815162536Sdavidxu		/* The address was invalid. */
816162536Sdavidxu		if (old == -1) {
817162536Sdavidxu			umtxq_lock(&uq->uq_key);
818162536Sdavidxu			umtxq_remove(uq);
819162536Sdavidxu			umtxq_unlock(&uq->uq_key);
820162536Sdavidxu			umtx_key_release(&uq->uq_key);
821162536Sdavidxu			return (EFAULT);
822162536Sdavidxu		}
823162536Sdavidxu
824162536Sdavidxu		/*
825162536Sdavidxu		 * We set the contested bit, sleep. Otherwise the lock changed
826162536Sdavidxu		 * and we need to retry or we lost a race to the thread
827162536Sdavidxu		 * unlocking the umtx.
828162536Sdavidxu		 */
829162536Sdavidxu		umtxq_lock(&uq->uq_key);
830162536Sdavidxu		if (old == owner)
831162536Sdavidxu			error = umtxq_sleep(uq, "umtx", timo);
832162536Sdavidxu		umtxq_remove(uq);
833162536Sdavidxu		umtxq_unlock(&uq->uq_key);
834162536Sdavidxu		umtx_key_release(&uq->uq_key);
835162536Sdavidxu	}
836162536Sdavidxu
837162536Sdavidxu	return (0);
838162536Sdavidxu}
839162536Sdavidxu
840162536Sdavidxu/*
841162536Sdavidxu * Lock a umtx object.
842162536Sdavidxu */
843162536Sdavidxustatic int
844162536Sdavidxudo_lock_umtx32(struct thread *td, void *m, uint32_t id,
845162536Sdavidxu	struct timespec *timeout)
846162536Sdavidxu{
847162536Sdavidxu	struct timespec ts, ts2, ts3;
848162536Sdavidxu	struct timeval tv;
849162536Sdavidxu	int error;
850162536Sdavidxu
851162536Sdavidxu	if (timeout == NULL) {
852162536Sdavidxu		error = _do_lock_umtx32(td, m, id, 0);
853162536Sdavidxu		/* Mutex locking is restarted if it is interrupted. */
854162536Sdavidxu		if (error == EINTR)
855162536Sdavidxu			error = ERESTART;
856162536Sdavidxu	} else {
857162536Sdavidxu		getnanouptime(&ts);
858162536Sdavidxu		timespecadd(&ts, timeout);
859162536Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, timeout);
860162536Sdavidxu		for (;;) {
861162536Sdavidxu			error = _do_lock_umtx32(td, m, id, tvtohz(&tv));
862162536Sdavidxu			if (error != ETIMEDOUT)
863162536Sdavidxu				break;
864162536Sdavidxu			getnanouptime(&ts2);
865162536Sdavidxu			if (timespeccmp(&ts2, &ts, >=)) {
866162536Sdavidxu				error = ETIMEDOUT;
867162536Sdavidxu				break;
868162536Sdavidxu			}
869162536Sdavidxu			ts3 = ts;
870162536Sdavidxu			timespecsub(&ts3, &ts2);
871162536Sdavidxu			TIMESPEC_TO_TIMEVAL(&tv, &ts3);
872162536Sdavidxu		}
873162536Sdavidxu		/* Timed-locking is not restarted. */
874162536Sdavidxu		if (error == ERESTART)
875162536Sdavidxu			error = EINTR;
876162536Sdavidxu	}
877162536Sdavidxu	return (error);
878162536Sdavidxu}
879162536Sdavidxu
880162536Sdavidxu/*
881162536Sdavidxu * Unlock a umtx object.
882162536Sdavidxu */
883162536Sdavidxustatic int
884162536Sdavidxudo_unlock_umtx32(struct thread *td, uint32_t *m, uint32_t id)
885162536Sdavidxu{
886162536Sdavidxu	struct umtx_key key;
887162536Sdavidxu	uint32_t owner;
888162536Sdavidxu	uint32_t old;
889162536Sdavidxu	int error;
890162536Sdavidxu	int count;
891162536Sdavidxu
892162536Sdavidxu	/*
893162536Sdavidxu	 * Make sure we own this mtx.
894162536Sdavidxu	 */
895162536Sdavidxu	owner = fuword32(m);
896162536Sdavidxu	if (owner == -1)
897162536Sdavidxu		return (EFAULT);
898162536Sdavidxu
899162536Sdavidxu	if ((owner & ~UMUTEX_CONTESTED) != id)
900162536Sdavidxu		return (EPERM);
901162536Sdavidxu
902162536Sdavidxu	/* This should be done in userland */
903162536Sdavidxu	if ((owner & UMUTEX_CONTESTED) == 0) {
904162536Sdavidxu		old = casuword32(m, owner, UMUTEX_UNOWNED);
905162536Sdavidxu		if (old == -1)
906162536Sdavidxu			return (EFAULT);
907162536Sdavidxu		if (old == owner)
908162536Sdavidxu			return (0);
909162536Sdavidxu		owner = old;
910162536Sdavidxu	}
911162536Sdavidxu
912162536Sdavidxu	/* We should only ever be in here for contested locks */
913162536Sdavidxu	if ((error = umtx_key_get(m, TYPE_SIMPLE_LOCK, AUTO_SHARE,
914162536Sdavidxu		&key)) != 0)
915162536Sdavidxu		return (error);
916162536Sdavidxu
917162536Sdavidxu	umtxq_lock(&key);
918162536Sdavidxu	umtxq_busy(&key);
919162536Sdavidxu	count = umtxq_count(&key);
920162536Sdavidxu	umtxq_unlock(&key);
921162536Sdavidxu
922162536Sdavidxu	/*
923162536Sdavidxu	 * When unlocking the umtx, it must be marked as unowned if
924162536Sdavidxu	 * there is zero or one thread only waiting for it.
925162536Sdavidxu	 * Otherwise, it must be marked as contested.
926162536Sdavidxu	 */
927162536Sdavidxu	old = casuword32(m, owner,
928162536Sdavidxu		count <= 1 ? UMUTEX_UNOWNED : UMUTEX_CONTESTED);
929162536Sdavidxu	umtxq_lock(&key);
930162536Sdavidxu	umtxq_signal(&key,1);
931162536Sdavidxu	umtxq_unbusy(&key);
932162536Sdavidxu	umtxq_unlock(&key);
933162536Sdavidxu	umtx_key_release(&key);
934162536Sdavidxu	if (old == -1)
935162536Sdavidxu		return (EFAULT);
936162536Sdavidxu	if (old != owner)
937162536Sdavidxu		return (EINVAL);
938162536Sdavidxu	return (0);
939162536Sdavidxu}
940162536Sdavidxu#endif
941162536Sdavidxu
942162536Sdavidxu/*
943161678Sdavidxu * Fetch and compare value, sleep on the address if value is not changed.
944161678Sdavidxu */
945139013Sdavidxustatic int
946163449Sdavidxudo_wait(struct thread *td, void *addr, u_long id,
947162536Sdavidxu	struct timespec *timeout, int compat32)
948139013Sdavidxu{
949143149Sdavidxu	struct umtx_q *uq;
950140245Sdavidxu	struct timespec ts, ts2, ts3;
951139013Sdavidxu	struct timeval tv;
952163449Sdavidxu	u_long tmp;
953140245Sdavidxu	int error = 0;
954139013Sdavidxu
955143149Sdavidxu	uq = td->td_umtxq;
956162536Sdavidxu	if ((error = umtx_key_get(addr, TYPE_SIMPLE_WAIT, AUTO_SHARE,
957161678Sdavidxu	    &uq->uq_key)) != 0)
958139013Sdavidxu		return (error);
959161678Sdavidxu
960161678Sdavidxu	umtxq_lock(&uq->uq_key);
961161678Sdavidxu	umtxq_insert(uq);
962161678Sdavidxu	umtxq_unlock(&uq->uq_key);
963162536Sdavidxu	if (compat32 == 0)
964162536Sdavidxu		tmp = fuword(addr);
965162536Sdavidxu        else
966162536Sdavidxu		tmp = fuword32(addr);
967139427Sdavidxu	if (tmp != id) {
968143149Sdavidxu		umtxq_lock(&uq->uq_key);
969143149Sdavidxu		umtxq_remove(uq);
970143149Sdavidxu		umtxq_unlock(&uq->uq_key);
971140245Sdavidxu	} else if (timeout == NULL) {
972143149Sdavidxu		umtxq_lock(&uq->uq_key);
973164839Sdavidxu		error = umtxq_sleep(uq, "uwait", 0);
974161678Sdavidxu		umtxq_remove(uq);
975143149Sdavidxu		umtxq_unlock(&uq->uq_key);
976139013Sdavidxu	} else {
977140245Sdavidxu		getnanouptime(&ts);
978140245Sdavidxu		timespecadd(&ts, timeout);
979140245Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, timeout);
980161678Sdavidxu		umtxq_lock(&uq->uq_key);
981139013Sdavidxu		for (;;) {
982164839Sdavidxu			error = umtxq_sleep(uq, "uwait", tvtohz(&tv));
983161678Sdavidxu			if (!(uq->uq_flags & UQF_UMTXQ))
984161678Sdavidxu				break;
985140245Sdavidxu			if (error != ETIMEDOUT)
986140245Sdavidxu				break;
987161678Sdavidxu			umtxq_unlock(&uq->uq_key);
988140245Sdavidxu			getnanouptime(&ts2);
989140245Sdavidxu			if (timespeccmp(&ts2, &ts, >=)) {
990139751Sdavidxu				error = ETIMEDOUT;
991161678Sdavidxu				umtxq_lock(&uq->uq_key);
992139013Sdavidxu				break;
993139013Sdavidxu			}
994140245Sdavidxu			ts3 = ts;
995140245Sdavidxu			timespecsub(&ts3, &ts2);
996140245Sdavidxu			TIMESPEC_TO_TIMEVAL(&tv, &ts3);
997161678Sdavidxu			umtxq_lock(&uq->uq_key);
998139013Sdavidxu		}
999143149Sdavidxu		umtxq_remove(uq);
1000143149Sdavidxu		umtxq_unlock(&uq->uq_key);
1001139013Sdavidxu	}
1002143149Sdavidxu	umtx_key_release(&uq->uq_key);
1003139257Sdavidxu	if (error == ERESTART)
1004139257Sdavidxu		error = EINTR;
1005139013Sdavidxu	return (error);
1006139013Sdavidxu}
1007139013Sdavidxu
1008161678Sdavidxu/*
1009161678Sdavidxu * Wake up threads sleeping on the specified address.
1010161678Sdavidxu */
1011151692Sdavidxuint
1012151692Sdavidxukern_umtx_wake(struct thread *td, void *uaddr, int n_wake)
1013139013Sdavidxu{
1014139013Sdavidxu	struct umtx_key key;
1015139257Sdavidxu	int ret;
1016139013Sdavidxu
1017161678Sdavidxu	if ((ret = umtx_key_get(uaddr, TYPE_SIMPLE_WAIT, AUTO_SHARE,
1018161678Sdavidxu	   &key)) != 0)
1019139257Sdavidxu		return (ret);
1020139258Sdavidxu	umtxq_lock(&key);
1021139257Sdavidxu	ret = umtxq_signal(&key, n_wake);
1022139258Sdavidxu	umtxq_unlock(&key);
1023139257Sdavidxu	umtx_key_release(&key);
1024139013Sdavidxu	return (0);
1025139013Sdavidxu}
1026139013Sdavidxu
1027161678Sdavidxu/*
1028161678Sdavidxu * Lock PTHREAD_PRIO_NONE protocol POSIX mutex.
1029161678Sdavidxu */
1030161678Sdavidxustatic int
1031161678Sdavidxu_do_lock_normal(struct thread *td, struct umutex *m, uint32_t flags, int timo,
1032161678Sdavidxu	int try)
1033161678Sdavidxu{
1034161678Sdavidxu	struct umtx_q *uq;
1035161678Sdavidxu	uint32_t owner, old, id;
1036161678Sdavidxu	int error = 0;
1037161678Sdavidxu
1038161678Sdavidxu	id = td->td_tid;
1039161678Sdavidxu	uq = td->td_umtxq;
1040161678Sdavidxu
1041161678Sdavidxu	/*
1042161678Sdavidxu	 * Care must be exercised when dealing with umtx structure. It
1043161678Sdavidxu	 * can fault on any access.
1044161678Sdavidxu	 */
1045161678Sdavidxu	for (;;) {
1046161678Sdavidxu		/*
1047161678Sdavidxu		 * Try the uncontested case.  This should be done in userland.
1048161678Sdavidxu		 */
1049161678Sdavidxu		owner = casuword32(&m->m_owner, UMUTEX_UNOWNED, id);
1050161678Sdavidxu
1051161678Sdavidxu		/* The acquire succeeded. */
1052161678Sdavidxu		if (owner == UMUTEX_UNOWNED)
1053161678Sdavidxu			return (0);
1054161678Sdavidxu
1055161678Sdavidxu		/* The address was invalid. */
1056161678Sdavidxu		if (owner == -1)
1057161678Sdavidxu			return (EFAULT);
1058161678Sdavidxu
1059161678Sdavidxu		/* If no one owns it but it is contested try to acquire it. */
1060161678Sdavidxu		if (owner == UMUTEX_CONTESTED) {
1061161678Sdavidxu			owner = casuword32(&m->m_owner,
1062161678Sdavidxu			    UMUTEX_CONTESTED, id | UMUTEX_CONTESTED);
1063161678Sdavidxu
1064161678Sdavidxu			if (owner == UMUTEX_CONTESTED)
1065161678Sdavidxu				return (0);
1066161678Sdavidxu
1067161678Sdavidxu			/* The address was invalid. */
1068161678Sdavidxu			if (owner == -1)
1069161678Sdavidxu				return (EFAULT);
1070161678Sdavidxu
1071161678Sdavidxu			/* If this failed the lock has changed, restart. */
1072161678Sdavidxu			continue;
1073161678Sdavidxu		}
1074161678Sdavidxu
1075161678Sdavidxu		if ((flags & UMUTEX_ERROR_CHECK) != 0 &&
1076161678Sdavidxu		    (owner & ~UMUTEX_CONTESTED) == id)
1077161678Sdavidxu			return (EDEADLK);
1078161678Sdavidxu
1079161678Sdavidxu		if (try != 0)
1080161678Sdavidxu			return (EBUSY);
1081161678Sdavidxu
1082161678Sdavidxu		/*
1083161678Sdavidxu		 * If we caught a signal, we have retried and now
1084161678Sdavidxu		 * exit immediately.
1085161678Sdavidxu		 */
1086161678Sdavidxu		if (error != 0)
1087161678Sdavidxu			return (error);
1088161678Sdavidxu
1089161678Sdavidxu		if ((error = umtx_key_get(m, TYPE_NORMAL_UMUTEX,
1090161678Sdavidxu		    GET_SHARE(flags), &uq->uq_key)) != 0)
1091161678Sdavidxu			return (error);
1092161678Sdavidxu
1093161678Sdavidxu		umtxq_lock(&uq->uq_key);
1094161678Sdavidxu		umtxq_busy(&uq->uq_key);
1095161678Sdavidxu		umtxq_insert(uq);
1096161678Sdavidxu		umtxq_unbusy(&uq->uq_key);
1097161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1098161678Sdavidxu
1099161678Sdavidxu		/*
1100161678Sdavidxu		 * Set the contested bit so that a release in user space
1101161678Sdavidxu		 * knows to use the system call for unlock.  If this fails
1102161678Sdavidxu		 * either some one else has acquired the lock or it has been
1103161678Sdavidxu		 * released.
1104161678Sdavidxu		 */
1105161678Sdavidxu		old = casuword32(&m->m_owner, owner, owner | UMUTEX_CONTESTED);
1106161678Sdavidxu
1107161678Sdavidxu		/* The address was invalid. */
1108161678Sdavidxu		if (old == -1) {
1109161678Sdavidxu			umtxq_lock(&uq->uq_key);
1110161678Sdavidxu			umtxq_remove(uq);
1111161678Sdavidxu			umtxq_unlock(&uq->uq_key);
1112161678Sdavidxu			umtx_key_release(&uq->uq_key);
1113161678Sdavidxu			return (EFAULT);
1114161678Sdavidxu		}
1115161678Sdavidxu
1116161678Sdavidxu		/*
1117161678Sdavidxu		 * We set the contested bit, sleep. Otherwise the lock changed
1118161678Sdavidxu		 * and we need to retry or we lost a race to the thread
1119161678Sdavidxu		 * unlocking the umtx.
1120161678Sdavidxu		 */
1121161678Sdavidxu		umtxq_lock(&uq->uq_key);
1122161678Sdavidxu		if (old == owner)
1123161678Sdavidxu			error = umtxq_sleep(uq, "umtxn", timo);
1124161678Sdavidxu		umtxq_remove(uq);
1125161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1126161678Sdavidxu		umtx_key_release(&uq->uq_key);
1127161678Sdavidxu	}
1128161678Sdavidxu
1129161678Sdavidxu	return (0);
1130161678Sdavidxu}
1131161678Sdavidxu
1132161678Sdavidxu/*
1133161678Sdavidxu * Lock PTHREAD_PRIO_NONE protocol POSIX mutex.
1134161678Sdavidxu */
1135161678Sdavidxu/*
1136161678Sdavidxu * Unlock PTHREAD_PRIO_NONE protocol POSIX mutex.
1137161678Sdavidxu */
1138161678Sdavidxustatic int
1139161678Sdavidxudo_unlock_normal(struct thread *td, struct umutex *m, uint32_t flags)
1140161678Sdavidxu{
1141161678Sdavidxu	struct umtx_key key;
1142161678Sdavidxu	uint32_t owner, old, id;
1143161678Sdavidxu	int error;
1144161678Sdavidxu	int count;
1145161678Sdavidxu
1146161678Sdavidxu	id = td->td_tid;
1147161678Sdavidxu	/*
1148161678Sdavidxu	 * Make sure we own this mtx.
1149161678Sdavidxu	 */
1150163449Sdavidxu	owner = fuword32(__DEVOLATILE(uint32_t *, &m->m_owner));
1151161678Sdavidxu	if (owner == -1)
1152161678Sdavidxu		return (EFAULT);
1153161678Sdavidxu
1154161678Sdavidxu	if ((owner & ~UMUTEX_CONTESTED) != id)
1155161678Sdavidxu		return (EPERM);
1156161678Sdavidxu
1157161678Sdavidxu	/* This should be done in userland */
1158161678Sdavidxu	if ((owner & UMUTEX_CONTESTED) == 0) {
1159161678Sdavidxu		old = casuword32(&m->m_owner, owner, UMUTEX_UNOWNED);
1160161678Sdavidxu		if (old == -1)
1161161678Sdavidxu			return (EFAULT);
1162161678Sdavidxu		if (old == owner)
1163161678Sdavidxu			return (0);
1164161855Sdavidxu		owner = old;
1165161678Sdavidxu	}
1166161678Sdavidxu
1167161678Sdavidxu	/* We should only ever be in here for contested locks */
1168161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_NORMAL_UMUTEX, GET_SHARE(flags),
1169161678Sdavidxu	    &key)) != 0)
1170161678Sdavidxu		return (error);
1171161678Sdavidxu
1172161678Sdavidxu	umtxq_lock(&key);
1173161678Sdavidxu	umtxq_busy(&key);
1174161678Sdavidxu	count = umtxq_count(&key);
1175161678Sdavidxu	umtxq_unlock(&key);
1176161678Sdavidxu
1177161678Sdavidxu	/*
1178161678Sdavidxu	 * When unlocking the umtx, it must be marked as unowned if
1179161678Sdavidxu	 * there is zero or one thread only waiting for it.
1180161678Sdavidxu	 * Otherwise, it must be marked as contested.
1181161678Sdavidxu	 */
1182161678Sdavidxu	old = casuword32(&m->m_owner, owner,
1183161678Sdavidxu		count <= 1 ? UMUTEX_UNOWNED : UMUTEX_CONTESTED);
1184161678Sdavidxu	umtxq_lock(&key);
1185161678Sdavidxu	umtxq_signal(&key,1);
1186161678Sdavidxu	umtxq_unbusy(&key);
1187161678Sdavidxu	umtxq_unlock(&key);
1188161678Sdavidxu	umtx_key_release(&key);
1189161678Sdavidxu	if (old == -1)
1190161678Sdavidxu		return (EFAULT);
1191161678Sdavidxu	if (old != owner)
1192161678Sdavidxu		return (EINVAL);
1193161678Sdavidxu	return (0);
1194161678Sdavidxu}
1195161678Sdavidxu
1196161678Sdavidxustatic inline struct umtx_pi *
1197163697Sdavidxuumtx_pi_alloc(int flags)
1198161678Sdavidxu{
1199161678Sdavidxu	struct umtx_pi *pi;
1200161678Sdavidxu
1201163697Sdavidxu	pi = uma_zalloc(umtx_pi_zone, M_ZERO | flags);
1202161678Sdavidxu	TAILQ_INIT(&pi->pi_blocked);
1203161678Sdavidxu	atomic_add_int(&umtx_pi_allocated, 1);
1204161678Sdavidxu	return (pi);
1205161678Sdavidxu}
1206161678Sdavidxu
1207161678Sdavidxustatic inline void
1208161678Sdavidxuumtx_pi_free(struct umtx_pi *pi)
1209161678Sdavidxu{
1210161678Sdavidxu	uma_zfree(umtx_pi_zone, pi);
1211161678Sdavidxu	atomic_add_int(&umtx_pi_allocated, -1);
1212161678Sdavidxu}
1213161678Sdavidxu
1214161678Sdavidxu/*
1215161678Sdavidxu * Adjust the thread's position on a pi_state after its priority has been
1216161678Sdavidxu * changed.
1217161678Sdavidxu */
1218161678Sdavidxustatic int
1219161678Sdavidxuumtx_pi_adjust_thread(struct umtx_pi *pi, struct thread *td)
1220161678Sdavidxu{
1221161678Sdavidxu	struct umtx_q *uq, *uq1, *uq2;
1222161678Sdavidxu	struct thread *td1;
1223161678Sdavidxu
1224170300Sjeff	mtx_assert(&umtx_lock, MA_OWNED);
1225161678Sdavidxu	if (pi == NULL)
1226161678Sdavidxu		return (0);
1227161678Sdavidxu
1228161678Sdavidxu	uq = td->td_umtxq;
1229161678Sdavidxu
1230161678Sdavidxu	/*
1231161678Sdavidxu	 * Check if the thread needs to be moved on the blocked chain.
1232161678Sdavidxu	 * It needs to be moved if either its priority is lower than
1233161678Sdavidxu	 * the previous thread or higher than the next thread.
1234161678Sdavidxu	 */
1235161678Sdavidxu	uq1 = TAILQ_PREV(uq, umtxq_head, uq_lockq);
1236161678Sdavidxu	uq2 = TAILQ_NEXT(uq, uq_lockq);
1237161678Sdavidxu	if ((uq1 != NULL && UPRI(td) < UPRI(uq1->uq_thread)) ||
1238161678Sdavidxu	    (uq2 != NULL && UPRI(td) > UPRI(uq2->uq_thread))) {
1239161678Sdavidxu		/*
1240161678Sdavidxu		 * Remove thread from blocked chain and determine where
1241161678Sdavidxu		 * it should be moved to.
1242161678Sdavidxu		 */
1243161678Sdavidxu		TAILQ_REMOVE(&pi->pi_blocked, uq, uq_lockq);
1244161678Sdavidxu		TAILQ_FOREACH(uq1, &pi->pi_blocked, uq_lockq) {
1245161678Sdavidxu			td1 = uq1->uq_thread;
1246161678Sdavidxu			MPASS(td1->td_proc->p_magic == P_MAGIC);
1247161678Sdavidxu			if (UPRI(td1) > UPRI(td))
1248161678Sdavidxu				break;
1249161678Sdavidxu		}
1250161678Sdavidxu
1251161678Sdavidxu		if (uq1 == NULL)
1252161678Sdavidxu			TAILQ_INSERT_TAIL(&pi->pi_blocked, uq, uq_lockq);
1253161678Sdavidxu		else
1254161678Sdavidxu			TAILQ_INSERT_BEFORE(uq1, uq, uq_lockq);
1255161678Sdavidxu	}
1256161678Sdavidxu	return (1);
1257161678Sdavidxu}
1258161678Sdavidxu
1259161678Sdavidxu/*
1260161678Sdavidxu * Propagate priority when a thread is blocked on POSIX
1261161678Sdavidxu * PI mutex.
1262161678Sdavidxu */
1263161678Sdavidxustatic void
1264161678Sdavidxuumtx_propagate_priority(struct thread *td)
1265161678Sdavidxu{
1266161678Sdavidxu	struct umtx_q *uq;
1267161678Sdavidxu	struct umtx_pi *pi;
1268161678Sdavidxu	int pri;
1269161678Sdavidxu
1270170300Sjeff	mtx_assert(&umtx_lock, MA_OWNED);
1271161678Sdavidxu	pri = UPRI(td);
1272161678Sdavidxu	uq = td->td_umtxq;
1273161678Sdavidxu	pi = uq->uq_pi_blocked;
1274161678Sdavidxu	if (pi == NULL)
1275161678Sdavidxu		return;
1276161678Sdavidxu
1277161678Sdavidxu	for (;;) {
1278161678Sdavidxu		td = pi->pi_owner;
1279161678Sdavidxu		if (td == NULL)
1280161678Sdavidxu			return;
1281161678Sdavidxu
1282161678Sdavidxu		MPASS(td->td_proc != NULL);
1283161678Sdavidxu		MPASS(td->td_proc->p_magic == P_MAGIC);
1284161678Sdavidxu
1285161678Sdavidxu		if (UPRI(td) <= pri)
1286161678Sdavidxu			return;
1287161678Sdavidxu
1288170300Sjeff		thread_lock(td);
1289161678Sdavidxu		sched_lend_user_prio(td, pri);
1290170300Sjeff		thread_unlock(td);
1291161678Sdavidxu
1292161678Sdavidxu		/*
1293161678Sdavidxu		 * Pick up the lock that td is blocked on.
1294161678Sdavidxu		 */
1295161678Sdavidxu		uq = td->td_umtxq;
1296161678Sdavidxu		pi = uq->uq_pi_blocked;
1297161678Sdavidxu		/* Resort td on the list if needed. */
1298161678Sdavidxu		if (!umtx_pi_adjust_thread(pi, td))
1299161678Sdavidxu			break;
1300161678Sdavidxu	}
1301161678Sdavidxu}
1302161678Sdavidxu
1303161678Sdavidxu/*
1304161678Sdavidxu * Unpropagate priority for a PI mutex when a thread blocked on
1305161678Sdavidxu * it is interrupted by signal or resumed by others.
1306161678Sdavidxu */
1307161678Sdavidxustatic void
1308161678Sdavidxuumtx_unpropagate_priority(struct umtx_pi *pi)
1309161678Sdavidxu{
1310161678Sdavidxu	struct umtx_q *uq, *uq_owner;
1311161678Sdavidxu	struct umtx_pi *pi2;
1312174701Sdavidxu	int pri, oldpri;
1313161678Sdavidxu
1314170300Sjeff	mtx_assert(&umtx_lock, MA_OWNED);
1315161678Sdavidxu
1316161678Sdavidxu	while (pi != NULL && pi->pi_owner != NULL) {
1317161678Sdavidxu		pri = PRI_MAX;
1318161678Sdavidxu		uq_owner = pi->pi_owner->td_umtxq;
1319161678Sdavidxu
1320161678Sdavidxu		TAILQ_FOREACH(pi2, &uq_owner->uq_pi_contested, pi_link) {
1321161678Sdavidxu			uq = TAILQ_FIRST(&pi2->pi_blocked);
1322161678Sdavidxu			if (uq != NULL) {
1323161678Sdavidxu				if (pri > UPRI(uq->uq_thread))
1324161678Sdavidxu					pri = UPRI(uq->uq_thread);
1325161678Sdavidxu			}
1326161678Sdavidxu		}
1327161678Sdavidxu
1328161678Sdavidxu		if (pri > uq_owner->uq_inherited_pri)
1329161678Sdavidxu			pri = uq_owner->uq_inherited_pri;
1330170300Sjeff		thread_lock(pi->pi_owner);
1331174701Sdavidxu		oldpri = pi->pi_owner->td_user_pri;
1332161678Sdavidxu		sched_unlend_user_prio(pi->pi_owner, pri);
1333170300Sjeff		thread_unlock(pi->pi_owner);
1334174701Sdavidxu		umtx_pi_adjust_locked(pi->pi_owner, oldpri);
1335161678Sdavidxu		pi = uq_owner->uq_pi_blocked;
1336161678Sdavidxu	}
1337161678Sdavidxu}
1338161678Sdavidxu
1339161678Sdavidxu/*
1340161678Sdavidxu * Insert a PI mutex into owned list.
1341161678Sdavidxu */
1342161678Sdavidxustatic void
1343161678Sdavidxuumtx_pi_setowner(struct umtx_pi *pi, struct thread *owner)
1344161678Sdavidxu{
1345161678Sdavidxu	struct umtx_q *uq_owner;
1346161678Sdavidxu
1347161678Sdavidxu	uq_owner = owner->td_umtxq;
1348170300Sjeff	mtx_assert(&umtx_lock, MA_OWNED);
1349161678Sdavidxu	if (pi->pi_owner != NULL)
1350161678Sdavidxu		panic("pi_ower != NULL");
1351161678Sdavidxu	pi->pi_owner = owner;
1352161678Sdavidxu	TAILQ_INSERT_TAIL(&uq_owner->uq_pi_contested, pi, pi_link);
1353161678Sdavidxu}
1354161678Sdavidxu
1355161678Sdavidxu/*
1356161678Sdavidxu * Claim ownership of a PI mutex.
1357161678Sdavidxu */
1358161678Sdavidxustatic int
1359161678Sdavidxuumtx_pi_claim(struct umtx_pi *pi, struct thread *owner)
1360161678Sdavidxu{
1361161678Sdavidxu	struct umtx_q *uq, *uq_owner;
1362161678Sdavidxu
1363161678Sdavidxu	uq_owner = owner->td_umtxq;
1364170300Sjeff	mtx_lock_spin(&umtx_lock);
1365161678Sdavidxu	if (pi->pi_owner == owner) {
1366170300Sjeff		mtx_unlock_spin(&umtx_lock);
1367161678Sdavidxu		return (0);
1368161678Sdavidxu	}
1369161678Sdavidxu
1370161678Sdavidxu	if (pi->pi_owner != NULL) {
1371161678Sdavidxu		/*
1372161678Sdavidxu		 * userland may have already messed the mutex, sigh.
1373161678Sdavidxu		 */
1374170300Sjeff		mtx_unlock_spin(&umtx_lock);
1375161678Sdavidxu		return (EPERM);
1376161678Sdavidxu	}
1377161678Sdavidxu	umtx_pi_setowner(pi, owner);
1378161678Sdavidxu	uq = TAILQ_FIRST(&pi->pi_blocked);
1379161678Sdavidxu	if (uq != NULL) {
1380161678Sdavidxu		int pri;
1381161678Sdavidxu
1382161678Sdavidxu		pri = UPRI(uq->uq_thread);
1383170300Sjeff		thread_lock(owner);
1384161678Sdavidxu		if (pri < UPRI(owner))
1385161678Sdavidxu			sched_lend_user_prio(owner, pri);
1386170300Sjeff		thread_unlock(owner);
1387161678Sdavidxu	}
1388170300Sjeff	mtx_unlock_spin(&umtx_lock);
1389161678Sdavidxu	return (0);
1390161678Sdavidxu}
1391161678Sdavidxu
1392174701Sdavidxustatic void
1393174701Sdavidxuumtx_pi_adjust_locked(struct thread *td, u_char oldpri)
1394161599Sdavidxu{
1395161678Sdavidxu	struct umtx_q *uq;
1396161678Sdavidxu	struct umtx_pi *pi;
1397161678Sdavidxu
1398161678Sdavidxu	uq = td->td_umtxq;
1399161678Sdavidxu	/*
1400161678Sdavidxu	 * Pick up the lock that td is blocked on.
1401161678Sdavidxu	 */
1402161678Sdavidxu	pi = uq->uq_pi_blocked;
1403161678Sdavidxu	MPASS(pi != NULL);
1404161678Sdavidxu
1405161678Sdavidxu	/* Resort the turnstile on the list. */
1406161678Sdavidxu	if (!umtx_pi_adjust_thread(pi, td))
1407161678Sdavidxu		return;
1408161678Sdavidxu
1409161678Sdavidxu	/*
1410161678Sdavidxu	 * If our priority was lowered and we are at the head of the
1411161678Sdavidxu	 * turnstile, then propagate our new priority up the chain.
1412161678Sdavidxu	 */
1413161678Sdavidxu	if (uq == TAILQ_FIRST(&pi->pi_blocked) && UPRI(td) < oldpri)
1414161678Sdavidxu		umtx_propagate_priority(td);
1415161599Sdavidxu}
1416161599Sdavidxu
1417161678Sdavidxu/*
1418174701Sdavidxu * Adjust a thread's order position in its blocked PI mutex,
1419174701Sdavidxu * this may result new priority propagating process.
1420174701Sdavidxu */
1421174701Sdavidxuvoid
1422174701Sdavidxuumtx_pi_adjust(struct thread *td, u_char oldpri)
1423174701Sdavidxu{
1424174707Sdavidxu	struct umtx_q *uq;
1425174707Sdavidxu	struct umtx_pi *pi;
1426174707Sdavidxu
1427174707Sdavidxu	uq = td->td_umtxq;
1428174701Sdavidxu	mtx_lock_spin(&umtx_lock);
1429174707Sdavidxu	/*
1430174707Sdavidxu	 * Pick up the lock that td is blocked on.
1431174707Sdavidxu	 */
1432174707Sdavidxu	pi = uq->uq_pi_blocked;
1433174707Sdavidxu	if (pi != NULL)
1434174707Sdavidxu		umtx_pi_adjust_locked(td, oldpri);
1435174701Sdavidxu	mtx_unlock_spin(&umtx_lock);
1436174701Sdavidxu}
1437174701Sdavidxu
1438174701Sdavidxu/*
1439161678Sdavidxu * Sleep on a PI mutex.
1440161678Sdavidxu */
1441161678Sdavidxustatic int
1442161678Sdavidxuumtxq_sleep_pi(struct umtx_q *uq, struct umtx_pi *pi,
1443161678Sdavidxu	uint32_t owner, const char *wmesg, int timo)
1444161678Sdavidxu{
1445161678Sdavidxu	struct umtxq_chain *uc;
1446161678Sdavidxu	struct thread *td, *td1;
1447161678Sdavidxu	struct umtx_q *uq1;
1448161678Sdavidxu	int pri;
1449161678Sdavidxu	int error = 0;
1450161678Sdavidxu
1451161678Sdavidxu	td = uq->uq_thread;
1452161678Sdavidxu	KASSERT(td == curthread, ("inconsistent uq_thread"));
1453161678Sdavidxu	uc = umtxq_getchain(&uq->uq_key);
1454161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
1455161678Sdavidxu	umtxq_insert(uq);
1456161678Sdavidxu	if (pi->pi_owner == NULL) {
1457161678Sdavidxu		/* XXX
1458161678Sdavidxu		 * Current, We only support process private PI-mutex,
1459161678Sdavidxu		 * non-contended PI-mutexes are locked in userland.
1460161678Sdavidxu		 * Process shared PI-mutex should always be initialized
1461161678Sdavidxu		 * by kernel and be registered in kernel, locking should
1462161678Sdavidxu		 * always be done by kernel to avoid security problems.
1463161678Sdavidxu		 * For process private PI-mutex, we can find owner
1464161678Sdavidxu		 * thread and boost its priority safely.
1465161678Sdavidxu		 */
1466161678Sdavidxu		PROC_LOCK(curproc);
1467161678Sdavidxu		td1 = thread_find(curproc, owner);
1468170300Sjeff		mtx_lock_spin(&umtx_lock);
1469161678Sdavidxu		if (td1 != NULL && pi->pi_owner == NULL) {
1470161678Sdavidxu			uq1 = td1->td_umtxq;
1471161678Sdavidxu			umtx_pi_setowner(pi, td1);
1472161678Sdavidxu		}
1473161678Sdavidxu		PROC_UNLOCK(curproc);
1474161678Sdavidxu	} else {
1475170300Sjeff		mtx_lock_spin(&umtx_lock);
1476161678Sdavidxu	}
1477161678Sdavidxu
1478161678Sdavidxu	TAILQ_FOREACH(uq1, &pi->pi_blocked, uq_lockq) {
1479161678Sdavidxu		pri = UPRI(uq1->uq_thread);
1480161678Sdavidxu		if (pri > UPRI(td))
1481161678Sdavidxu			break;
1482161678Sdavidxu	}
1483161678Sdavidxu
1484161678Sdavidxu	if (uq1 != NULL)
1485161678Sdavidxu		TAILQ_INSERT_BEFORE(uq1, uq, uq_lockq);
1486161678Sdavidxu	else
1487161678Sdavidxu		TAILQ_INSERT_TAIL(&pi->pi_blocked, uq, uq_lockq);
1488161678Sdavidxu
1489161678Sdavidxu	uq->uq_pi_blocked = pi;
1490174701Sdavidxu	thread_lock(td);
1491161678Sdavidxu	td->td_flags |= TDF_UPIBLOCKED;
1492174701Sdavidxu	thread_unlock(td);
1493170300Sjeff	mtx_unlock_spin(&umtx_lock);
1494161678Sdavidxu	umtxq_unlock(&uq->uq_key);
1495161678Sdavidxu
1496170300Sjeff	mtx_lock_spin(&umtx_lock);
1497161678Sdavidxu	umtx_propagate_priority(td);
1498170300Sjeff	mtx_unlock_spin(&umtx_lock);
1499161678Sdavidxu
1500161678Sdavidxu	umtxq_lock(&uq->uq_key);
1501161678Sdavidxu	if (uq->uq_flags & UQF_UMTXQ) {
1502161678Sdavidxu		error = msleep(uq, &uc->uc_lock, PCATCH, wmesg, timo);
1503161678Sdavidxu		if (error == EWOULDBLOCK)
1504161678Sdavidxu			error = ETIMEDOUT;
1505161678Sdavidxu		if (uq->uq_flags & UQF_UMTXQ) {
1506161678Sdavidxu			umtxq_busy(&uq->uq_key);
1507161678Sdavidxu			umtxq_remove(uq);
1508161678Sdavidxu			umtxq_unbusy(&uq->uq_key);
1509161678Sdavidxu		}
1510161678Sdavidxu	}
1511161678Sdavidxu	umtxq_unlock(&uq->uq_key);
1512161678Sdavidxu
1513170300Sjeff	mtx_lock_spin(&umtx_lock);
1514161678Sdavidxu	uq->uq_pi_blocked = NULL;
1515174701Sdavidxu	thread_lock(td);
1516161678Sdavidxu	td->td_flags &= ~TDF_UPIBLOCKED;
1517174701Sdavidxu	thread_unlock(td);
1518161678Sdavidxu	TAILQ_REMOVE(&pi->pi_blocked, uq, uq_lockq);
1519161678Sdavidxu	umtx_unpropagate_priority(pi);
1520170300Sjeff	mtx_unlock_spin(&umtx_lock);
1521161678Sdavidxu
1522161678Sdavidxu	umtxq_lock(&uq->uq_key);
1523161678Sdavidxu
1524161678Sdavidxu	return (error);
1525161678Sdavidxu}
1526161678Sdavidxu
1527161678Sdavidxu/*
1528161678Sdavidxu * Add reference count for a PI mutex.
1529161678Sdavidxu */
1530161678Sdavidxustatic void
1531161678Sdavidxuumtx_pi_ref(struct umtx_pi *pi)
1532161678Sdavidxu{
1533161678Sdavidxu	struct umtxq_chain *uc;
1534161678Sdavidxu
1535161678Sdavidxu	uc = umtxq_getchain(&pi->pi_key);
1536161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
1537161678Sdavidxu	pi->pi_refcount++;
1538161678Sdavidxu}
1539161678Sdavidxu
1540161678Sdavidxu/*
1541161678Sdavidxu * Decrease reference count for a PI mutex, if the counter
1542161678Sdavidxu * is decreased to zero, its memory space is freed.
1543161678Sdavidxu */
1544161678Sdavidxustatic void
1545161678Sdavidxuumtx_pi_unref(struct umtx_pi *pi)
1546161678Sdavidxu{
1547161678Sdavidxu	struct umtxq_chain *uc;
1548161678Sdavidxu	int free = 0;
1549161678Sdavidxu
1550161678Sdavidxu	uc = umtxq_getchain(&pi->pi_key);
1551161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
1552161678Sdavidxu	KASSERT(pi->pi_refcount > 0, ("invalid reference count"));
1553161678Sdavidxu	if (--pi->pi_refcount == 0) {
1554170300Sjeff		mtx_lock_spin(&umtx_lock);
1555161678Sdavidxu		if (pi->pi_owner != NULL) {
1556161678Sdavidxu			TAILQ_REMOVE(&pi->pi_owner->td_umtxq->uq_pi_contested,
1557161678Sdavidxu				pi, pi_link);
1558161678Sdavidxu			pi->pi_owner = NULL;
1559161678Sdavidxu		}
1560161678Sdavidxu		KASSERT(TAILQ_EMPTY(&pi->pi_blocked),
1561161678Sdavidxu			("blocked queue not empty"));
1562170300Sjeff		mtx_unlock_spin(&umtx_lock);
1563161678Sdavidxu		TAILQ_REMOVE(&uc->uc_pi_list, pi, pi_hashlink);
1564161678Sdavidxu		free = 1;
1565161678Sdavidxu	}
1566161678Sdavidxu	if (free)
1567161678Sdavidxu		umtx_pi_free(pi);
1568161678Sdavidxu}
1569161678Sdavidxu
1570161678Sdavidxu/*
1571161678Sdavidxu * Find a PI mutex in hash table.
1572161678Sdavidxu */
1573161678Sdavidxustatic struct umtx_pi *
1574161678Sdavidxuumtx_pi_lookup(struct umtx_key *key)
1575161678Sdavidxu{
1576161678Sdavidxu	struct umtxq_chain *uc;
1577161678Sdavidxu	struct umtx_pi *pi;
1578161678Sdavidxu
1579161678Sdavidxu	uc = umtxq_getchain(key);
1580161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
1581161678Sdavidxu
1582161678Sdavidxu	TAILQ_FOREACH(pi, &uc->uc_pi_list, pi_hashlink) {
1583161678Sdavidxu		if (umtx_key_match(&pi->pi_key, key)) {
1584161678Sdavidxu			return (pi);
1585161678Sdavidxu		}
1586161678Sdavidxu	}
1587161678Sdavidxu	return (NULL);
1588161678Sdavidxu}
1589161678Sdavidxu
1590161678Sdavidxu/*
1591161678Sdavidxu * Insert a PI mutex into hash table.
1592161678Sdavidxu */
1593161678Sdavidxustatic inline void
1594161678Sdavidxuumtx_pi_insert(struct umtx_pi *pi)
1595161678Sdavidxu{
1596161678Sdavidxu	struct umtxq_chain *uc;
1597161678Sdavidxu
1598161678Sdavidxu	uc = umtxq_getchain(&pi->pi_key);
1599161678Sdavidxu	UMTXQ_LOCKED_ASSERT(uc);
1600161678Sdavidxu	TAILQ_INSERT_TAIL(&uc->uc_pi_list, pi, pi_hashlink);
1601161678Sdavidxu}
1602161678Sdavidxu
1603161678Sdavidxu/*
1604161678Sdavidxu * Lock a PI mutex.
1605161678Sdavidxu */
1606161678Sdavidxustatic int
1607161678Sdavidxu_do_lock_pi(struct thread *td, struct umutex *m, uint32_t flags, int timo,
1608161678Sdavidxu	int try)
1609161678Sdavidxu{
1610161678Sdavidxu	struct umtx_q *uq;
1611161678Sdavidxu	struct umtx_pi *pi, *new_pi;
1612161678Sdavidxu	uint32_t id, owner, old;
1613161678Sdavidxu	int error;
1614161678Sdavidxu
1615161678Sdavidxu	id = td->td_tid;
1616161678Sdavidxu	uq = td->td_umtxq;
1617161678Sdavidxu
1618161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_PI_UMUTEX, GET_SHARE(flags),
1619161678Sdavidxu	    &uq->uq_key)) != 0)
1620161678Sdavidxu		return (error);
1621163697Sdavidxu	umtxq_lock(&uq->uq_key);
1622163697Sdavidxu	pi = umtx_pi_lookup(&uq->uq_key);
1623163697Sdavidxu	if (pi == NULL) {
1624163697Sdavidxu		new_pi = umtx_pi_alloc(M_NOWAIT);
1625163697Sdavidxu		if (new_pi == NULL) {
1626161678Sdavidxu			umtxq_unlock(&uq->uq_key);
1627163697Sdavidxu			new_pi = umtx_pi_alloc(M_WAITOK);
1628161678Sdavidxu			new_pi->pi_key = uq->uq_key;
1629161678Sdavidxu			umtxq_lock(&uq->uq_key);
1630161678Sdavidxu			pi = umtx_pi_lookup(&uq->uq_key);
1631163697Sdavidxu			if (pi != NULL) {
1632161678Sdavidxu				umtx_pi_free(new_pi);
1633163697Sdavidxu				new_pi = NULL;
1634161678Sdavidxu			}
1635161678Sdavidxu		}
1636163697Sdavidxu		if (new_pi != NULL) {
1637163697Sdavidxu			new_pi->pi_key = uq->uq_key;
1638163697Sdavidxu			umtx_pi_insert(new_pi);
1639163697Sdavidxu			pi = new_pi;
1640163697Sdavidxu		}
1641163697Sdavidxu	}
1642163697Sdavidxu	umtx_pi_ref(pi);
1643163697Sdavidxu	umtxq_unlock(&uq->uq_key);
1644161678Sdavidxu
1645163697Sdavidxu	/*
1646163697Sdavidxu	 * Care must be exercised when dealing with umtx structure.  It
1647163697Sdavidxu	 * can fault on any access.
1648163697Sdavidxu	 */
1649163697Sdavidxu	for (;;) {
1650161678Sdavidxu		/*
1651161678Sdavidxu		 * Try the uncontested case.  This should be done in userland.
1652161678Sdavidxu		 */
1653161678Sdavidxu		owner = casuword32(&m->m_owner, UMUTEX_UNOWNED, id);
1654161678Sdavidxu
1655161678Sdavidxu		/* The acquire succeeded. */
1656161678Sdavidxu		if (owner == UMUTEX_UNOWNED) {
1657161678Sdavidxu			error = 0;
1658161678Sdavidxu			break;
1659161678Sdavidxu		}
1660161678Sdavidxu
1661161678Sdavidxu		/* The address was invalid. */
1662161678Sdavidxu		if (owner == -1) {
1663161678Sdavidxu			error = EFAULT;
1664161678Sdavidxu			break;
1665161678Sdavidxu		}
1666161678Sdavidxu
1667161678Sdavidxu		/* If no one owns it but it is contested try to acquire it. */
1668161678Sdavidxu		if (owner == UMUTEX_CONTESTED) {
1669161678Sdavidxu			owner = casuword32(&m->m_owner,
1670161678Sdavidxu			    UMUTEX_CONTESTED, id | UMUTEX_CONTESTED);
1671161678Sdavidxu
1672161678Sdavidxu			if (owner == UMUTEX_CONTESTED) {
1673161678Sdavidxu				umtxq_lock(&uq->uq_key);
1674161678Sdavidxu				error = umtx_pi_claim(pi, td);
1675161678Sdavidxu				umtxq_unlock(&uq->uq_key);
1676161678Sdavidxu				break;
1677161678Sdavidxu			}
1678161678Sdavidxu
1679161678Sdavidxu			/* The address was invalid. */
1680161678Sdavidxu			if (owner == -1) {
1681161678Sdavidxu				error = EFAULT;
1682161678Sdavidxu				break;
1683161678Sdavidxu			}
1684161678Sdavidxu
1685161678Sdavidxu			/* If this failed the lock has changed, restart. */
1686161678Sdavidxu			continue;
1687161678Sdavidxu		}
1688161678Sdavidxu
1689161678Sdavidxu		if ((flags & UMUTEX_ERROR_CHECK) != 0 &&
1690161678Sdavidxu		    (owner & ~UMUTEX_CONTESTED) == id) {
1691161678Sdavidxu			error = EDEADLK;
1692161678Sdavidxu			break;
1693161678Sdavidxu		}
1694161678Sdavidxu
1695161678Sdavidxu		if (try != 0) {
1696161678Sdavidxu			error = EBUSY;
1697161678Sdavidxu			break;
1698161678Sdavidxu		}
1699161678Sdavidxu
1700161678Sdavidxu		/*
1701161678Sdavidxu		 * If we caught a signal, we have retried and now
1702161678Sdavidxu		 * exit immediately.
1703161678Sdavidxu		 */
1704161678Sdavidxu		if (error != 0)
1705161678Sdavidxu			break;
1706161678Sdavidxu
1707161678Sdavidxu		umtxq_lock(&uq->uq_key);
1708161678Sdavidxu		umtxq_busy(&uq->uq_key);
1709161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1710161678Sdavidxu
1711161678Sdavidxu		/*
1712161678Sdavidxu		 * Set the contested bit so that a release in user space
1713161678Sdavidxu		 * knows to use the system call for unlock.  If this fails
1714161678Sdavidxu		 * either some one else has acquired the lock or it has been
1715161678Sdavidxu		 * released.
1716161678Sdavidxu		 */
1717161678Sdavidxu		old = casuword32(&m->m_owner, owner, owner | UMUTEX_CONTESTED);
1718161678Sdavidxu
1719161678Sdavidxu		/* The address was invalid. */
1720161678Sdavidxu		if (old == -1) {
1721161678Sdavidxu			umtxq_lock(&uq->uq_key);
1722161678Sdavidxu			umtxq_unbusy(&uq->uq_key);
1723161678Sdavidxu			umtxq_unlock(&uq->uq_key);
1724161678Sdavidxu			error = EFAULT;
1725161678Sdavidxu			break;
1726161678Sdavidxu		}
1727161678Sdavidxu
1728161678Sdavidxu		umtxq_lock(&uq->uq_key);
1729161678Sdavidxu		umtxq_unbusy(&uq->uq_key);
1730161678Sdavidxu		/*
1731161678Sdavidxu		 * We set the contested bit, sleep. Otherwise the lock changed
1732161678Sdavidxu		 * and we need to retry or we lost a race to the thread
1733161678Sdavidxu		 * unlocking the umtx.
1734161678Sdavidxu		 */
1735161678Sdavidxu		if (old == owner)
1736161678Sdavidxu			error = umtxq_sleep_pi(uq, pi, owner & ~UMUTEX_CONTESTED,
1737161678Sdavidxu				 "umtxpi", timo);
1738161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1739161678Sdavidxu	}
1740161678Sdavidxu
1741163697Sdavidxu	umtxq_lock(&uq->uq_key);
1742163697Sdavidxu	umtx_pi_unref(pi);
1743163697Sdavidxu	umtxq_unlock(&uq->uq_key);
1744161678Sdavidxu
1745161678Sdavidxu	umtx_key_release(&uq->uq_key);
1746161678Sdavidxu	return (error);
1747161678Sdavidxu}
1748161678Sdavidxu
1749161678Sdavidxu/*
1750161678Sdavidxu * Unlock a PI mutex.
1751161678Sdavidxu */
1752161678Sdavidxustatic int
1753161678Sdavidxudo_unlock_pi(struct thread *td, struct umutex *m, uint32_t flags)
1754161678Sdavidxu{
1755161678Sdavidxu	struct umtx_key key;
1756161678Sdavidxu	struct umtx_q *uq_first, *uq_first2, *uq_me;
1757161678Sdavidxu	struct umtx_pi *pi, *pi2;
1758161678Sdavidxu	uint32_t owner, old, id;
1759161678Sdavidxu	int error;
1760161678Sdavidxu	int count;
1761161678Sdavidxu	int pri;
1762161678Sdavidxu
1763161678Sdavidxu	id = td->td_tid;
1764161678Sdavidxu	/*
1765161678Sdavidxu	 * Make sure we own this mtx.
1766161678Sdavidxu	 */
1767163449Sdavidxu	owner = fuword32(__DEVOLATILE(uint32_t *, &m->m_owner));
1768161678Sdavidxu	if (owner == -1)
1769161678Sdavidxu		return (EFAULT);
1770161678Sdavidxu
1771161678Sdavidxu	if ((owner & ~UMUTEX_CONTESTED) != id)
1772161678Sdavidxu		return (EPERM);
1773161678Sdavidxu
1774161678Sdavidxu	/* This should be done in userland */
1775161678Sdavidxu	if ((owner & UMUTEX_CONTESTED) == 0) {
1776161678Sdavidxu		old = casuword32(&m->m_owner, owner, UMUTEX_UNOWNED);
1777161678Sdavidxu		if (old == -1)
1778161678Sdavidxu			return (EFAULT);
1779161678Sdavidxu		if (old == owner)
1780161678Sdavidxu			return (0);
1781161855Sdavidxu		owner = old;
1782161678Sdavidxu	}
1783161678Sdavidxu
1784161678Sdavidxu	/* We should only ever be in here for contested locks */
1785161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_PI_UMUTEX, GET_SHARE(flags),
1786161678Sdavidxu	    &key)) != 0)
1787161678Sdavidxu		return (error);
1788161678Sdavidxu
1789161678Sdavidxu	umtxq_lock(&key);
1790161678Sdavidxu	umtxq_busy(&key);
1791161678Sdavidxu	count = umtxq_count_pi(&key, &uq_first);
1792161678Sdavidxu	if (uq_first != NULL) {
1793161678Sdavidxu		pi = uq_first->uq_pi_blocked;
1794161678Sdavidxu		if (pi->pi_owner != curthread) {
1795161678Sdavidxu			umtxq_unbusy(&key);
1796161678Sdavidxu			umtxq_unlock(&key);
1797161678Sdavidxu			/* userland messed the mutex */
1798161678Sdavidxu			return (EPERM);
1799161678Sdavidxu		}
1800161678Sdavidxu		uq_me = curthread->td_umtxq;
1801170300Sjeff		mtx_lock_spin(&umtx_lock);
1802161678Sdavidxu		pi->pi_owner = NULL;
1803161678Sdavidxu		TAILQ_REMOVE(&uq_me->uq_pi_contested, pi, pi_link);
1804161678Sdavidxu		uq_first = TAILQ_FIRST(&pi->pi_blocked);
1805161678Sdavidxu		pri = PRI_MAX;
1806161678Sdavidxu		TAILQ_FOREACH(pi2, &uq_me->uq_pi_contested, pi_link) {
1807161678Sdavidxu			uq_first2 = TAILQ_FIRST(&pi2->pi_blocked);
1808161678Sdavidxu			if (uq_first2 != NULL) {
1809161678Sdavidxu				if (pri > UPRI(uq_first2->uq_thread))
1810161678Sdavidxu					pri = UPRI(uq_first2->uq_thread);
1811161678Sdavidxu			}
1812161678Sdavidxu		}
1813170300Sjeff		thread_lock(curthread);
1814161678Sdavidxu		sched_unlend_user_prio(curthread, pri);
1815170300Sjeff		thread_unlock(curthread);
1816170300Sjeff		mtx_unlock_spin(&umtx_lock);
1817161678Sdavidxu	}
1818161678Sdavidxu	umtxq_unlock(&key);
1819161678Sdavidxu
1820161678Sdavidxu	/*
1821161678Sdavidxu	 * When unlocking the umtx, it must be marked as unowned if
1822161678Sdavidxu	 * there is zero or one thread only waiting for it.
1823161678Sdavidxu	 * Otherwise, it must be marked as contested.
1824161678Sdavidxu	 */
1825161678Sdavidxu	old = casuword32(&m->m_owner, owner,
1826161678Sdavidxu		count <= 1 ? UMUTEX_UNOWNED : UMUTEX_CONTESTED);
1827161678Sdavidxu
1828161678Sdavidxu	umtxq_lock(&key);
1829161678Sdavidxu	if (uq_first != NULL)
1830161678Sdavidxu		umtxq_signal_thread(uq_first);
1831161678Sdavidxu	umtxq_unbusy(&key);
1832161678Sdavidxu	umtxq_unlock(&key);
1833161678Sdavidxu	umtx_key_release(&key);
1834161678Sdavidxu	if (old == -1)
1835161678Sdavidxu		return (EFAULT);
1836161678Sdavidxu	if (old != owner)
1837161678Sdavidxu		return (EINVAL);
1838161678Sdavidxu	return (0);
1839161678Sdavidxu}
1840161678Sdavidxu
1841161678Sdavidxu/*
1842161678Sdavidxu * Lock a PP mutex.
1843161678Sdavidxu */
1844161678Sdavidxustatic int
1845161678Sdavidxu_do_lock_pp(struct thread *td, struct umutex *m, uint32_t flags, int timo,
1846161678Sdavidxu	int try)
1847161678Sdavidxu{
1848161678Sdavidxu	struct umtx_q *uq, *uq2;
1849161678Sdavidxu	struct umtx_pi *pi;
1850161678Sdavidxu	uint32_t ceiling;
1851161678Sdavidxu	uint32_t owner, id;
1852161678Sdavidxu	int error, pri, old_inherited_pri, su;
1853161678Sdavidxu
1854161678Sdavidxu	id = td->td_tid;
1855161678Sdavidxu	uq = td->td_umtxq;
1856161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_PP_UMUTEX, GET_SHARE(flags),
1857161678Sdavidxu	    &uq->uq_key)) != 0)
1858161678Sdavidxu		return (error);
1859164033Srwatson	su = (priv_check(td, PRIV_SCHED_RTPRIO) == 0);
1860161678Sdavidxu	for (;;) {
1861161678Sdavidxu		old_inherited_pri = uq->uq_inherited_pri;
1862161678Sdavidxu		umtxq_lock(&uq->uq_key);
1863161678Sdavidxu		umtxq_busy(&uq->uq_key);
1864161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1865161678Sdavidxu
1866161678Sdavidxu		ceiling = RTP_PRIO_MAX - fuword32(&m->m_ceilings[0]);
1867161678Sdavidxu		if (ceiling > RTP_PRIO_MAX) {
1868161678Sdavidxu			error = EINVAL;
1869161678Sdavidxu			goto out;
1870161678Sdavidxu		}
1871161678Sdavidxu
1872170300Sjeff		mtx_lock_spin(&umtx_lock);
1873161678Sdavidxu		if (UPRI(td) < PRI_MIN_REALTIME + ceiling) {
1874170300Sjeff			mtx_unlock_spin(&umtx_lock);
1875161678Sdavidxu			error = EINVAL;
1876161678Sdavidxu			goto out;
1877161678Sdavidxu		}
1878161678Sdavidxu		if (su && PRI_MIN_REALTIME + ceiling < uq->uq_inherited_pri) {
1879161678Sdavidxu			uq->uq_inherited_pri = PRI_MIN_REALTIME + ceiling;
1880170300Sjeff			thread_lock(td);
1881161678Sdavidxu			if (uq->uq_inherited_pri < UPRI(td))
1882161678Sdavidxu				sched_lend_user_prio(td, uq->uq_inherited_pri);
1883170300Sjeff			thread_unlock(td);
1884161678Sdavidxu		}
1885170300Sjeff		mtx_unlock_spin(&umtx_lock);
1886161678Sdavidxu
1887161678Sdavidxu		owner = casuword32(&m->m_owner,
1888161678Sdavidxu		    UMUTEX_CONTESTED, id | UMUTEX_CONTESTED);
1889161678Sdavidxu
1890161678Sdavidxu		if (owner == UMUTEX_CONTESTED) {
1891161678Sdavidxu			error = 0;
1892161678Sdavidxu			break;
1893161678Sdavidxu		}
1894161678Sdavidxu
1895161678Sdavidxu		/* The address was invalid. */
1896161678Sdavidxu		if (owner == -1) {
1897161678Sdavidxu			error = EFAULT;
1898161678Sdavidxu			break;
1899161678Sdavidxu		}
1900161678Sdavidxu
1901161678Sdavidxu		if ((flags & UMUTEX_ERROR_CHECK) != 0 &&
1902161678Sdavidxu		    (owner & ~UMUTEX_CONTESTED) == id) {
1903161678Sdavidxu			error = EDEADLK;
1904161678Sdavidxu			break;
1905161678Sdavidxu		}
1906161678Sdavidxu
1907161678Sdavidxu		if (try != 0) {
1908161678Sdavidxu			error = EBUSY;
1909161678Sdavidxu			break;
1910161678Sdavidxu		}
1911161678Sdavidxu
1912161678Sdavidxu		/*
1913161678Sdavidxu		 * If we caught a signal, we have retried and now
1914161678Sdavidxu		 * exit immediately.
1915161678Sdavidxu		 */
1916161678Sdavidxu		if (error != 0)
1917161678Sdavidxu			break;
1918161678Sdavidxu
1919161678Sdavidxu		umtxq_lock(&uq->uq_key);
1920161678Sdavidxu		umtxq_insert(uq);
1921161678Sdavidxu		umtxq_unbusy(&uq->uq_key);
1922161678Sdavidxu		error = umtxq_sleep(uq, "umtxpp", timo);
1923161678Sdavidxu		umtxq_remove(uq);
1924161678Sdavidxu		umtxq_unlock(&uq->uq_key);
1925161678Sdavidxu
1926170300Sjeff		mtx_lock_spin(&umtx_lock);
1927161678Sdavidxu		uq->uq_inherited_pri = old_inherited_pri;
1928161678Sdavidxu		pri = PRI_MAX;
1929161678Sdavidxu		TAILQ_FOREACH(pi, &uq->uq_pi_contested, pi_link) {
1930161678Sdavidxu			uq2 = TAILQ_FIRST(&pi->pi_blocked);
1931161678Sdavidxu			if (uq2 != NULL) {
1932161678Sdavidxu				if (pri > UPRI(uq2->uq_thread))
1933161678Sdavidxu					pri = UPRI(uq2->uq_thread);
1934161678Sdavidxu			}
1935161678Sdavidxu		}
1936161678Sdavidxu		if (pri > uq->uq_inherited_pri)
1937161678Sdavidxu			pri = uq->uq_inherited_pri;
1938170300Sjeff		thread_lock(td);
1939161678Sdavidxu		sched_unlend_user_prio(td, pri);
1940170300Sjeff		thread_unlock(td);
1941170300Sjeff		mtx_unlock_spin(&umtx_lock);
1942161678Sdavidxu	}
1943161678Sdavidxu
1944161678Sdavidxu	if (error != 0) {
1945170300Sjeff		mtx_lock_spin(&umtx_lock);
1946161678Sdavidxu		uq->uq_inherited_pri = old_inherited_pri;
1947161678Sdavidxu		pri = PRI_MAX;
1948161678Sdavidxu		TAILQ_FOREACH(pi, &uq->uq_pi_contested, pi_link) {
1949161678Sdavidxu			uq2 = TAILQ_FIRST(&pi->pi_blocked);
1950161678Sdavidxu			if (uq2 != NULL) {
1951161678Sdavidxu				if (pri > UPRI(uq2->uq_thread))
1952161678Sdavidxu					pri = UPRI(uq2->uq_thread);
1953161678Sdavidxu			}
1954161678Sdavidxu		}
1955161678Sdavidxu		if (pri > uq->uq_inherited_pri)
1956161678Sdavidxu			pri = uq->uq_inherited_pri;
1957170300Sjeff		thread_lock(td);
1958161678Sdavidxu		sched_unlend_user_prio(td, pri);
1959170300Sjeff		thread_unlock(td);
1960170300Sjeff		mtx_unlock_spin(&umtx_lock);
1961161678Sdavidxu	}
1962161678Sdavidxu
1963161678Sdavidxuout:
1964161678Sdavidxu	umtxq_lock(&uq->uq_key);
1965161678Sdavidxu	umtxq_unbusy(&uq->uq_key);
1966161678Sdavidxu	umtxq_unlock(&uq->uq_key);
1967161678Sdavidxu	umtx_key_release(&uq->uq_key);
1968161678Sdavidxu	return (error);
1969161678Sdavidxu}
1970161678Sdavidxu
1971161678Sdavidxu/*
1972161678Sdavidxu * Unlock a PP mutex.
1973161678Sdavidxu */
1974161678Sdavidxustatic int
1975161678Sdavidxudo_unlock_pp(struct thread *td, struct umutex *m, uint32_t flags)
1976161678Sdavidxu{
1977161678Sdavidxu	struct umtx_key key;
1978161678Sdavidxu	struct umtx_q *uq, *uq2;
1979161678Sdavidxu	struct umtx_pi *pi;
1980161678Sdavidxu	uint32_t owner, id;
1981161678Sdavidxu	uint32_t rceiling;
1982161926Sdavidxu	int error, pri, new_inherited_pri, su;
1983161678Sdavidxu
1984161678Sdavidxu	id = td->td_tid;
1985161678Sdavidxu	uq = td->td_umtxq;
1986164033Srwatson	su = (priv_check(td, PRIV_SCHED_RTPRIO) == 0);
1987161678Sdavidxu
1988161678Sdavidxu	/*
1989161678Sdavidxu	 * Make sure we own this mtx.
1990161678Sdavidxu	 */
1991163449Sdavidxu	owner = fuword32(__DEVOLATILE(uint32_t *, &m->m_owner));
1992161678Sdavidxu	if (owner == -1)
1993161678Sdavidxu		return (EFAULT);
1994161678Sdavidxu
1995161678Sdavidxu	if ((owner & ~UMUTEX_CONTESTED) != id)
1996161678Sdavidxu		return (EPERM);
1997161678Sdavidxu
1998161678Sdavidxu	error = copyin(&m->m_ceilings[1], &rceiling, sizeof(uint32_t));
1999161678Sdavidxu	if (error != 0)
2000161678Sdavidxu		return (error);
2001161678Sdavidxu
2002161678Sdavidxu	if (rceiling == -1)
2003161678Sdavidxu		new_inherited_pri = PRI_MAX;
2004161678Sdavidxu	else {
2005161678Sdavidxu		rceiling = RTP_PRIO_MAX - rceiling;
2006161678Sdavidxu		if (rceiling > RTP_PRIO_MAX)
2007161678Sdavidxu			return (EINVAL);
2008161678Sdavidxu		new_inherited_pri = PRI_MIN_REALTIME + rceiling;
2009161678Sdavidxu	}
2010161678Sdavidxu
2011161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_PP_UMUTEX, GET_SHARE(flags),
2012161678Sdavidxu	    &key)) != 0)
2013161678Sdavidxu		return (error);
2014161678Sdavidxu	umtxq_lock(&key);
2015161678Sdavidxu	umtxq_busy(&key);
2016161678Sdavidxu	umtxq_unlock(&key);
2017161678Sdavidxu	/*
2018161678Sdavidxu	 * For priority protected mutex, always set unlocked state
2019161678Sdavidxu	 * to UMUTEX_CONTESTED, so that userland always enters kernel
2020161678Sdavidxu	 * to lock the mutex, it is necessary because thread priority
2021161678Sdavidxu	 * has to be adjusted for such mutex.
2022161678Sdavidxu	 */
2023163449Sdavidxu	error = suword32(__DEVOLATILE(uint32_t *, &m->m_owner),
2024163449Sdavidxu		UMUTEX_CONTESTED);
2025161678Sdavidxu
2026161678Sdavidxu	umtxq_lock(&key);
2027161678Sdavidxu	if (error == 0)
2028161678Sdavidxu		umtxq_signal(&key, 1);
2029161678Sdavidxu	umtxq_unbusy(&key);
2030161678Sdavidxu	umtxq_unlock(&key);
2031161678Sdavidxu
2032161678Sdavidxu	if (error == -1)
2033161678Sdavidxu		error = EFAULT;
2034161678Sdavidxu	else {
2035170300Sjeff		mtx_lock_spin(&umtx_lock);
2036161926Sdavidxu		if (su != 0)
2037161926Sdavidxu			uq->uq_inherited_pri = new_inherited_pri;
2038161678Sdavidxu		pri = PRI_MAX;
2039161678Sdavidxu		TAILQ_FOREACH(pi, &uq->uq_pi_contested, pi_link) {
2040161678Sdavidxu			uq2 = TAILQ_FIRST(&pi->pi_blocked);
2041161678Sdavidxu			if (uq2 != NULL) {
2042161678Sdavidxu				if (pri > UPRI(uq2->uq_thread))
2043161678Sdavidxu					pri = UPRI(uq2->uq_thread);
2044161678Sdavidxu			}
2045161678Sdavidxu		}
2046161678Sdavidxu		if (pri > uq->uq_inherited_pri)
2047161678Sdavidxu			pri = uq->uq_inherited_pri;
2048170300Sjeff		thread_lock(td);
2049161678Sdavidxu		sched_unlend_user_prio(td, pri);
2050170300Sjeff		thread_unlock(td);
2051170300Sjeff		mtx_unlock_spin(&umtx_lock);
2052161678Sdavidxu	}
2053161678Sdavidxu	umtx_key_release(&key);
2054161678Sdavidxu	return (error);
2055161678Sdavidxu}
2056161678Sdavidxu
2057161678Sdavidxustatic int
2058161678Sdavidxudo_set_ceiling(struct thread *td, struct umutex *m, uint32_t ceiling,
2059161678Sdavidxu	uint32_t *old_ceiling)
2060161678Sdavidxu{
2061161678Sdavidxu	struct umtx_q *uq;
2062161678Sdavidxu	uint32_t save_ceiling;
2063161678Sdavidxu	uint32_t owner, id;
2064161678Sdavidxu	uint32_t flags;
2065161678Sdavidxu	int error;
2066161678Sdavidxu
2067161678Sdavidxu	flags = fuword32(&m->m_flags);
2068161678Sdavidxu	if ((flags & UMUTEX_PRIO_PROTECT) == 0)
2069161678Sdavidxu		return (EINVAL);
2070161678Sdavidxu	if (ceiling > RTP_PRIO_MAX)
2071161678Sdavidxu		return (EINVAL);
2072161678Sdavidxu	id = td->td_tid;
2073161678Sdavidxu	uq = td->td_umtxq;
2074161678Sdavidxu	if ((error = umtx_key_get(m, TYPE_PP_UMUTEX, GET_SHARE(flags),
2075161678Sdavidxu	   &uq->uq_key)) != 0)
2076161678Sdavidxu		return (error);
2077161678Sdavidxu	for (;;) {
2078161678Sdavidxu		umtxq_lock(&uq->uq_key);
2079161678Sdavidxu		umtxq_busy(&uq->uq_key);
2080161678Sdavidxu		umtxq_unlock(&uq->uq_key);
2081161678Sdavidxu
2082161678Sdavidxu		save_ceiling = fuword32(&m->m_ceilings[0]);
2083161678Sdavidxu
2084161678Sdavidxu		owner = casuword32(&m->m_owner,
2085161678Sdavidxu		    UMUTEX_CONTESTED, id | UMUTEX_CONTESTED);
2086161678Sdavidxu
2087161678Sdavidxu		if (owner == UMUTEX_CONTESTED) {
2088161678Sdavidxu			suword32(&m->m_ceilings[0], ceiling);
2089163449Sdavidxu			suword32(__DEVOLATILE(uint32_t *, &m->m_owner),
2090163449Sdavidxu				UMUTEX_CONTESTED);
2091161678Sdavidxu			error = 0;
2092161678Sdavidxu			break;
2093161678Sdavidxu		}
2094161678Sdavidxu
2095161678Sdavidxu		/* The address was invalid. */
2096161678Sdavidxu		if (owner == -1) {
2097161678Sdavidxu			error = EFAULT;
2098161678Sdavidxu			break;
2099161678Sdavidxu		}
2100161678Sdavidxu
2101161678Sdavidxu		if ((owner & ~UMUTEX_CONTESTED) == id) {
2102161678Sdavidxu			suword32(&m->m_ceilings[0], ceiling);
2103161678Sdavidxu			error = 0;
2104161678Sdavidxu			break;
2105161678Sdavidxu		}
2106161678Sdavidxu
2107161678Sdavidxu		/*
2108161678Sdavidxu		 * If we caught a signal, we have retried and now
2109161678Sdavidxu		 * exit immediately.
2110161678Sdavidxu		 */
2111161678Sdavidxu		if (error != 0)
2112161678Sdavidxu			break;
2113161678Sdavidxu
2114161678Sdavidxu		/*
2115161678Sdavidxu		 * We set the contested bit, sleep. Otherwise the lock changed
2116161678Sdavidxu		 * and we need to retry or we lost a race to the thread
2117161678Sdavidxu		 * unlocking the umtx.
2118161678Sdavidxu		 */
2119161678Sdavidxu		umtxq_lock(&uq->uq_key);
2120161678Sdavidxu		umtxq_insert(uq);
2121161678Sdavidxu		umtxq_unbusy(&uq->uq_key);
2122161678Sdavidxu		error = umtxq_sleep(uq, "umtxpp", 0);
2123161678Sdavidxu		umtxq_remove(uq);
2124161678Sdavidxu		umtxq_unlock(&uq->uq_key);
2125161678Sdavidxu	}
2126161678Sdavidxu	umtxq_lock(&uq->uq_key);
2127161678Sdavidxu	if (error == 0)
2128161678Sdavidxu		umtxq_signal(&uq->uq_key, INT_MAX);
2129161678Sdavidxu	umtxq_unbusy(&uq->uq_key);
2130161678Sdavidxu	umtxq_unlock(&uq->uq_key);
2131161678Sdavidxu	umtx_key_release(&uq->uq_key);
2132161678Sdavidxu	if (error == 0 && old_ceiling != NULL)
2133161678Sdavidxu		suword32(old_ceiling, save_ceiling);
2134161678Sdavidxu	return (error);
2135161678Sdavidxu}
2136161678Sdavidxu
2137162030Sdavidxustatic int
2138162030Sdavidxu_do_lock_umutex(struct thread *td, struct umutex *m, int flags, int timo,
2139162030Sdavidxu	int try)
2140162030Sdavidxu{
2141162030Sdavidxu	switch(flags & (UMUTEX_PRIO_INHERIT | UMUTEX_PRIO_PROTECT)) {
2142162030Sdavidxu	case 0:
2143162030Sdavidxu		return (_do_lock_normal(td, m, flags, timo, try));
2144162030Sdavidxu	case UMUTEX_PRIO_INHERIT:
2145162030Sdavidxu		return (_do_lock_pi(td, m, flags, timo, try));
2146162030Sdavidxu	case UMUTEX_PRIO_PROTECT:
2147162030Sdavidxu		return (_do_lock_pp(td, m, flags, timo, try));
2148162030Sdavidxu	}
2149162030Sdavidxu	return (EINVAL);
2150162030Sdavidxu}
2151162030Sdavidxu
2152161678Sdavidxu/*
2153161678Sdavidxu * Lock a userland POSIX mutex.
2154161678Sdavidxu */
2155161678Sdavidxustatic int
2156162030Sdavidxudo_lock_umutex(struct thread *td, struct umutex *m,
2157162030Sdavidxu	struct timespec *timeout, int try)
2158161678Sdavidxu{
2159162030Sdavidxu	struct timespec ts, ts2, ts3;
2160162030Sdavidxu	struct timeval tv;
2161161678Sdavidxu	uint32_t flags;
2162162030Sdavidxu	int error;
2163161678Sdavidxu
2164161678Sdavidxu	flags = fuword32(&m->m_flags);
2165161678Sdavidxu	if (flags == -1)
2166161678Sdavidxu		return (EFAULT);
2167161678Sdavidxu
2168162030Sdavidxu	if (timeout == NULL) {
2169162030Sdavidxu		error = _do_lock_umutex(td, m, flags, 0, try);
2170162030Sdavidxu		/* Mutex locking is restarted if it is interrupted. */
2171162030Sdavidxu		if (error == EINTR)
2172162030Sdavidxu			error = ERESTART;
2173162030Sdavidxu	} else {
2174162030Sdavidxu		getnanouptime(&ts);
2175162030Sdavidxu		timespecadd(&ts, timeout);
2176162030Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, timeout);
2177162030Sdavidxu		for (;;) {
2178162030Sdavidxu			error = _do_lock_umutex(td, m, flags, tvtohz(&tv), try);
2179162030Sdavidxu			if (error != ETIMEDOUT)
2180162030Sdavidxu				break;
2181162030Sdavidxu			getnanouptime(&ts2);
2182162030Sdavidxu			if (timespeccmp(&ts2, &ts, >=)) {
2183162030Sdavidxu				error = ETIMEDOUT;
2184162030Sdavidxu				break;
2185162030Sdavidxu			}
2186162030Sdavidxu			ts3 = ts;
2187162030Sdavidxu			timespecsub(&ts3, &ts2);
2188162030Sdavidxu			TIMESPEC_TO_TIMEVAL(&tv, &ts3);
2189162030Sdavidxu		}
2190162030Sdavidxu		/* Timed-locking is not restarted. */
2191162030Sdavidxu		if (error == ERESTART)
2192162030Sdavidxu			error = EINTR;
2193161742Sdavidxu	}
2194162030Sdavidxu	return (error);
2195161678Sdavidxu}
2196161678Sdavidxu
2197161678Sdavidxu/*
2198161678Sdavidxu * Unlock a userland POSIX mutex.
2199161678Sdavidxu */
2200161678Sdavidxustatic int
2201161678Sdavidxudo_unlock_umutex(struct thread *td, struct umutex *m)
2202161678Sdavidxu{
2203161678Sdavidxu	uint32_t flags;
2204161678Sdavidxu
2205161678Sdavidxu	flags = fuword32(&m->m_flags);
2206161678Sdavidxu	if (flags == -1)
2207161678Sdavidxu		return (EFAULT);
2208161678Sdavidxu
2209161855Sdavidxu	switch(flags & (UMUTEX_PRIO_INHERIT | UMUTEX_PRIO_PROTECT)) {
2210161855Sdavidxu	case 0:
2211161855Sdavidxu		return (do_unlock_normal(td, m, flags));
2212161855Sdavidxu	case UMUTEX_PRIO_INHERIT:
2213161855Sdavidxu		return (do_unlock_pi(td, m, flags));
2214161855Sdavidxu	case UMUTEX_PRIO_PROTECT:
2215161855Sdavidxu		return (do_unlock_pp(td, m, flags));
2216161855Sdavidxu	}
2217161678Sdavidxu
2218161855Sdavidxu	return (EINVAL);
2219161678Sdavidxu}
2220161678Sdavidxu
2221164839Sdavidxustatic int
2222164839Sdavidxudo_cv_wait(struct thread *td, struct ucond *cv, struct umutex *m,
2223164876Sdavidxu	struct timespec *timeout, u_long wflags)
2224164839Sdavidxu{
2225164839Sdavidxu	struct umtx_q *uq;
2226164839Sdavidxu	struct timeval tv;
2227164839Sdavidxu	struct timespec cts, ets, tts;
2228164839Sdavidxu	uint32_t flags;
2229164839Sdavidxu	int error;
2230164839Sdavidxu
2231164839Sdavidxu	uq = td->td_umtxq;
2232164839Sdavidxu	flags = fuword32(&cv->c_flags);
2233164839Sdavidxu	error = umtx_key_get(cv, TYPE_CV, GET_SHARE(flags), &uq->uq_key);
2234164839Sdavidxu	if (error != 0)
2235164839Sdavidxu		return (error);
2236164839Sdavidxu	umtxq_lock(&uq->uq_key);
2237164839Sdavidxu	umtxq_busy(&uq->uq_key);
2238164839Sdavidxu	umtxq_insert(uq);
2239164839Sdavidxu	umtxq_unlock(&uq->uq_key);
2240164839Sdavidxu
2241164839Sdavidxu	/*
2242164839Sdavidxu	 * The magic thing is we should set c_has_waiters to 1 before
2243164839Sdavidxu	 * releasing user mutex.
2244164839Sdavidxu	 */
2245164839Sdavidxu	suword32(__DEVOLATILE(uint32_t *, &cv->c_has_waiters), 1);
2246164839Sdavidxu
2247164839Sdavidxu	umtxq_lock(&uq->uq_key);
2248164839Sdavidxu	umtxq_unbusy(&uq->uq_key);
2249164839Sdavidxu	umtxq_unlock(&uq->uq_key);
2250164839Sdavidxu
2251164839Sdavidxu	error = do_unlock_umutex(td, m);
2252164839Sdavidxu
2253164839Sdavidxu	umtxq_lock(&uq->uq_key);
2254164839Sdavidxu	if (error == 0) {
2255164876Sdavidxu		if ((wflags & UMTX_CHECK_UNPARKING) &&
2256164876Sdavidxu		    (td->td_pflags & TDP_WAKEUP)) {
2257164876Sdavidxu			td->td_pflags &= ~TDP_WAKEUP;
2258164876Sdavidxu			error = EINTR;
2259164876Sdavidxu		} else if (timeout == NULL) {
2260164839Sdavidxu			error = umtxq_sleep(uq, "ucond", 0);
2261164839Sdavidxu		} else {
2262164839Sdavidxu			getnanouptime(&ets);
2263164839Sdavidxu			timespecadd(&ets, timeout);
2264164839Sdavidxu			TIMESPEC_TO_TIMEVAL(&tv, timeout);
2265164839Sdavidxu			for (;;) {
2266164839Sdavidxu				error = umtxq_sleep(uq, "ucond", tvtohz(&tv));
2267164839Sdavidxu				if (error != ETIMEDOUT)
2268164839Sdavidxu					break;
2269164839Sdavidxu				getnanouptime(&cts);
2270164839Sdavidxu				if (timespeccmp(&cts, &ets, >=)) {
2271164839Sdavidxu					error = ETIMEDOUT;
2272164839Sdavidxu					break;
2273164839Sdavidxu				}
2274164839Sdavidxu				tts = ets;
2275164839Sdavidxu				timespecsub(&tts, &cts);
2276164839Sdavidxu				TIMESPEC_TO_TIMEVAL(&tv, &tts);
2277164839Sdavidxu			}
2278164839Sdavidxu		}
2279164839Sdavidxu	}
2280164839Sdavidxu
2281164839Sdavidxu	if (error != 0) {
2282164839Sdavidxu		if ((uq->uq_flags & UQF_UMTXQ) == 0) {
2283164839Sdavidxu			/*
2284164839Sdavidxu			 * If we concurrently got do_cv_signal()d
2285164839Sdavidxu			 * and we got an error or UNIX signals or a timeout,
2286164839Sdavidxu			 * then, perform another umtxq_signal to avoid
2287164839Sdavidxu			 * consuming the wakeup. This may cause supurious
2288164839Sdavidxu			 * wakeup for another thread which was just queued,
2289164839Sdavidxu			 * but SUSV3 explicitly allows supurious wakeup to
2290164839Sdavidxu			 * occur, and indeed a kernel based implementation
2291164839Sdavidxu			 * can not avoid it.
2292164839Sdavidxu			 */
2293164876Sdavidxu			if (!umtxq_signal(&uq->uq_key, 1))
2294164876Sdavidxu				error = 0;
2295164839Sdavidxu		}
2296164839Sdavidxu		if (error == ERESTART)
2297164839Sdavidxu			error = EINTR;
2298164839Sdavidxu	}
2299164839Sdavidxu	umtxq_remove(uq);
2300164839Sdavidxu	umtxq_unlock(&uq->uq_key);
2301164839Sdavidxu	umtx_key_release(&uq->uq_key);
2302164839Sdavidxu	return (error);
2303164839Sdavidxu}
2304164839Sdavidxu
2305164839Sdavidxu/*
2306164839Sdavidxu * Signal a userland condition variable.
2307164839Sdavidxu */
2308164839Sdavidxustatic int
2309164839Sdavidxudo_cv_signal(struct thread *td, struct ucond *cv)
2310164839Sdavidxu{
2311164839Sdavidxu	struct umtx_key key;
2312164839Sdavidxu	int error, cnt, nwake;
2313164839Sdavidxu	uint32_t flags;
2314164839Sdavidxu
2315164839Sdavidxu	flags = fuword32(&cv->c_flags);
2316164839Sdavidxu	if ((error = umtx_key_get(cv, TYPE_CV, GET_SHARE(flags), &key)) != 0)
2317164839Sdavidxu		return (error);
2318164839Sdavidxu	umtxq_lock(&key);
2319164839Sdavidxu	umtxq_busy(&key);
2320164839Sdavidxu	cnt = umtxq_count(&key);
2321164839Sdavidxu	nwake = umtxq_signal(&key, 1);
2322164839Sdavidxu	if (cnt <= nwake) {
2323164839Sdavidxu		umtxq_unlock(&key);
2324164839Sdavidxu		error = suword32(
2325164839Sdavidxu		    __DEVOLATILE(uint32_t *, &cv->c_has_waiters), 0);
2326164839Sdavidxu		umtxq_lock(&key);
2327164839Sdavidxu	}
2328164839Sdavidxu	umtxq_unbusy(&key);
2329164839Sdavidxu	umtxq_unlock(&key);
2330164839Sdavidxu	umtx_key_release(&key);
2331164839Sdavidxu	return (error);
2332164839Sdavidxu}
2333164839Sdavidxu
2334164839Sdavidxustatic int
2335164839Sdavidxudo_cv_broadcast(struct thread *td, struct ucond *cv)
2336164839Sdavidxu{
2337164839Sdavidxu	struct umtx_key key;
2338164839Sdavidxu	int error;
2339164839Sdavidxu	uint32_t flags;
2340164839Sdavidxu
2341164839Sdavidxu	flags = fuword32(&cv->c_flags);
2342164839Sdavidxu	if ((error = umtx_key_get(cv, TYPE_CV, GET_SHARE(flags), &key)) != 0)
2343164839Sdavidxu		return (error);
2344164839Sdavidxu
2345164839Sdavidxu	umtxq_lock(&key);
2346164839Sdavidxu	umtxq_busy(&key);
2347164839Sdavidxu	umtxq_signal(&key, INT_MAX);
2348164839Sdavidxu	umtxq_unlock(&key);
2349164839Sdavidxu
2350164839Sdavidxu	error = suword32(__DEVOLATILE(uint32_t *, &cv->c_has_waiters), 0);
2351164839Sdavidxu
2352164839Sdavidxu	umtxq_lock(&key);
2353164839Sdavidxu	umtxq_unbusy(&key);
2354164839Sdavidxu	umtxq_unlock(&key);
2355164839Sdavidxu
2356164839Sdavidxu	umtx_key_release(&key);
2357164839Sdavidxu	return (error);
2358164839Sdavidxu}
2359164839Sdavidxu
2360177848Sdavidxustatic int
2361177848Sdavidxudo_rw_rdlock(struct thread *td, struct urwlock *rwlock, long fflag, int timo)
2362177848Sdavidxu{
2363177848Sdavidxu	struct umtx_q *uq;
2364177848Sdavidxu	uint32_t flags, wrflags;
2365177848Sdavidxu	int32_t state, oldstate;
2366177848Sdavidxu	int32_t blocked_readers;
2367177848Sdavidxu	int error;
2368177848Sdavidxu
2369177848Sdavidxu	uq = td->td_umtxq;
2370177848Sdavidxu	flags = fuword32(&rwlock->rw_flags);
2371177848Sdavidxu	error = umtx_key_get(rwlock, TYPE_RWLOCK, GET_SHARE(flags), &uq->uq_key);
2372177848Sdavidxu	if (error != 0)
2373177848Sdavidxu		return (error);
2374177848Sdavidxu
2375177848Sdavidxu	wrflags = URWLOCK_WRITE_OWNER;
2376177848Sdavidxu	if (!(fflag & URWLOCK_PREFER_READER) && !(flags & URWLOCK_PREFER_READER))
2377177848Sdavidxu		wrflags |= URWLOCK_WRITE_WAITERS;
2378177848Sdavidxu
2379177848Sdavidxu	for (;;) {
2380177848Sdavidxu		state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2381177848Sdavidxu		/* try to lock it */
2382177848Sdavidxu		while (!(state & wrflags)) {
2383177848Sdavidxu			if (__predict_false(URWLOCK_READER_COUNT(state) == URWLOCK_MAX_READERS)) {
2384177848Sdavidxu				umtx_key_release(&uq->uq_key);
2385177848Sdavidxu				return (EAGAIN);
2386177848Sdavidxu			}
2387177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state, state + 1);
2388177848Sdavidxu			if (oldstate == state) {
2389177848Sdavidxu				umtx_key_release(&uq->uq_key);
2390177848Sdavidxu				return (0);
2391177848Sdavidxu			}
2392177848Sdavidxu			state = oldstate;
2393177848Sdavidxu		}
2394177848Sdavidxu
2395177848Sdavidxu		if (error)
2396177848Sdavidxu			break;
2397177848Sdavidxu
2398177848Sdavidxu		/* grab monitor lock */
2399177848Sdavidxu		umtxq_lock(&uq->uq_key);
2400177848Sdavidxu		umtxq_busy(&uq->uq_key);
2401177848Sdavidxu		umtxq_unlock(&uq->uq_key);
2402177848Sdavidxu
2403177848Sdavidxu		/* set read contention bit */
2404177848Sdavidxu		while ((state & wrflags) && !(state & URWLOCK_READ_WAITERS)) {
2405177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state, state | URWLOCK_READ_WAITERS);
2406177848Sdavidxu			if (oldstate == state)
2407177848Sdavidxu				goto sleep;
2408177848Sdavidxu			state = oldstate;
2409177848Sdavidxu		}
2410177848Sdavidxu
2411177848Sdavidxu		/* state is changed while setting flags, restart */
2412177848Sdavidxu		if (!(state & wrflags)) {
2413177848Sdavidxu			umtxq_lock(&uq->uq_key);
2414177848Sdavidxu			umtxq_unbusy(&uq->uq_key);
2415177848Sdavidxu			umtxq_unlock(&uq->uq_key);
2416177848Sdavidxu			continue;
2417177848Sdavidxu		}
2418177848Sdavidxu
2419177848Sdavidxusleep:
2420177848Sdavidxu		/* contention bit is set, before sleeping, increase read waiter count */
2421177848Sdavidxu		blocked_readers = fuword32(&rwlock->rw_blocked_readers);
2422177848Sdavidxu		suword32(&rwlock->rw_blocked_readers, blocked_readers+1);
2423177848Sdavidxu
2424177848Sdavidxu		while (state & wrflags) {
2425177848Sdavidxu			umtxq_lock(&uq->uq_key);
2426177848Sdavidxu			umtxq_insert(uq);
2427177848Sdavidxu			umtxq_unbusy(&uq->uq_key);
2428177848Sdavidxu
2429177848Sdavidxu			error = umtxq_sleep(uq, "urdlck", timo);
2430177848Sdavidxu
2431177848Sdavidxu			umtxq_busy(&uq->uq_key);
2432177848Sdavidxu			umtxq_remove(uq);
2433177848Sdavidxu			umtxq_unlock(&uq->uq_key);
2434177848Sdavidxu			if (error)
2435177848Sdavidxu				break;
2436177848Sdavidxu			state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2437177848Sdavidxu		}
2438177848Sdavidxu
2439177848Sdavidxu		/* decrease read waiter count, and may clear read contention bit */
2440177848Sdavidxu		blocked_readers = fuword32(&rwlock->rw_blocked_readers);
2441177848Sdavidxu		suword32(&rwlock->rw_blocked_readers, blocked_readers-1);
2442177848Sdavidxu		if (blocked_readers == 1) {
2443177848Sdavidxu			state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2444177848Sdavidxu			for (;;) {
2445177848Sdavidxu				oldstate = casuword32(&rwlock->rw_state, state,
2446177848Sdavidxu					 state & ~URWLOCK_READ_WAITERS);
2447177848Sdavidxu				if (oldstate == state)
2448177848Sdavidxu					break;
2449177848Sdavidxu				state = oldstate;
2450177848Sdavidxu			}
2451177848Sdavidxu		}
2452177848Sdavidxu
2453177848Sdavidxu		umtxq_lock(&uq->uq_key);
2454177848Sdavidxu		umtxq_unbusy(&uq->uq_key);
2455177848Sdavidxu		umtxq_unlock(&uq->uq_key);
2456177848Sdavidxu	}
2457177848Sdavidxu	umtx_key_release(&uq->uq_key);
2458177848Sdavidxu	return (error);
2459177848Sdavidxu}
2460177848Sdavidxu
2461177848Sdavidxustatic int
2462177848Sdavidxudo_rw_rdlock2(struct thread *td, void *obj, long val, struct timespec *timeout)
2463177848Sdavidxu{
2464177848Sdavidxu	struct timespec ts, ts2, ts3;
2465177848Sdavidxu	struct timeval tv;
2466177848Sdavidxu	int error;
2467177848Sdavidxu
2468177848Sdavidxu	getnanouptime(&ts);
2469177848Sdavidxu	timespecadd(&ts, timeout);
2470177848Sdavidxu	TIMESPEC_TO_TIMEVAL(&tv, timeout);
2471177848Sdavidxu	for (;;) {
2472177848Sdavidxu		error = do_rw_rdlock(td, obj, val, tvtohz(&tv));
2473177848Sdavidxu		if (error != ETIMEDOUT)
2474177848Sdavidxu			break;
2475177848Sdavidxu		getnanouptime(&ts2);
2476177848Sdavidxu		if (timespeccmp(&ts2, &ts, >=)) {
2477177848Sdavidxu			error = ETIMEDOUT;
2478177848Sdavidxu			break;
2479177848Sdavidxu		}
2480177848Sdavidxu		ts3 = ts;
2481177848Sdavidxu		timespecsub(&ts3, &ts2);
2482177848Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, &ts3);
2483177848Sdavidxu	}
2484177849Sdavidxu	if (error == ERESTART)
2485177849Sdavidxu		error = EINTR;
2486177848Sdavidxu	return (error);
2487177848Sdavidxu}
2488177848Sdavidxu
2489177848Sdavidxustatic int
2490177848Sdavidxudo_rw_wrlock(struct thread *td, struct urwlock *rwlock, int timo)
2491177848Sdavidxu{
2492177848Sdavidxu	struct umtx_q *uq;
2493177848Sdavidxu	uint32_t flags;
2494177848Sdavidxu	int32_t state, oldstate;
2495177848Sdavidxu	int32_t blocked_writers;
2496177848Sdavidxu	int error;
2497177848Sdavidxu
2498177848Sdavidxu	uq = td->td_umtxq;
2499177848Sdavidxu	flags = fuword32(&rwlock->rw_flags);
2500177848Sdavidxu	error = umtx_key_get(rwlock, TYPE_RWLOCK, GET_SHARE(flags), &uq->uq_key);
2501177848Sdavidxu	if (error != 0)
2502177848Sdavidxu		return (error);
2503177848Sdavidxu
2504177848Sdavidxu	for (;;) {
2505177848Sdavidxu		state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2506177848Sdavidxu		while (!(state & URWLOCK_WRITE_OWNER) && URWLOCK_READER_COUNT(state) == 0) {
2507177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state, state | URWLOCK_WRITE_OWNER);
2508177848Sdavidxu			if (oldstate == state) {
2509177848Sdavidxu				umtx_key_release(&uq->uq_key);
2510177848Sdavidxu				return (0);
2511177848Sdavidxu			}
2512177848Sdavidxu			state = oldstate;
2513177848Sdavidxu		}
2514177848Sdavidxu
2515177848Sdavidxu		if (error)
2516177848Sdavidxu			break;
2517177848Sdavidxu
2518177848Sdavidxu		/* grab monitor lock */
2519177848Sdavidxu		umtxq_lock(&uq->uq_key);
2520177848Sdavidxu		umtxq_busy(&uq->uq_key);
2521177848Sdavidxu		umtxq_unlock(&uq->uq_key);
2522177848Sdavidxu
2523177848Sdavidxu		while (((state & URWLOCK_WRITE_OWNER) || URWLOCK_READER_COUNT(state) != 0) &&
2524177848Sdavidxu		       (state & URWLOCK_WRITE_WAITERS) == 0) {
2525177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state, state | URWLOCK_WRITE_WAITERS);
2526177848Sdavidxu			if (oldstate == state)
2527177848Sdavidxu				goto sleep;
2528177848Sdavidxu			state = oldstate;
2529177848Sdavidxu		}
2530177848Sdavidxu
2531177848Sdavidxu		if (!(state & URWLOCK_WRITE_OWNER) && URWLOCK_READER_COUNT(state) == 0) {
2532177848Sdavidxu			umtxq_lock(&uq->uq_key);
2533177848Sdavidxu			umtxq_unbusy(&uq->uq_key);
2534177848Sdavidxu			umtxq_unlock(&uq->uq_key);
2535177848Sdavidxu			continue;
2536177848Sdavidxu		}
2537177848Sdavidxusleep:
2538177848Sdavidxu		blocked_writers = fuword32(&rwlock->rw_blocked_writers);
2539177848Sdavidxu		suword32(&rwlock->rw_blocked_writers, blocked_writers+1);
2540177848Sdavidxu
2541177848Sdavidxu		while ((state & URWLOCK_WRITE_OWNER) || URWLOCK_READER_COUNT(state) != 0) {
2542177848Sdavidxu			umtxq_lock(&uq->uq_key);
2543177848Sdavidxu			umtxq_insert_queue(uq, UMTX_EXCLUSIVE_QUEUE);
2544177848Sdavidxu			umtxq_unbusy(&uq->uq_key);
2545177848Sdavidxu
2546177848Sdavidxu			error = umtxq_sleep(uq, "uwrlck", timo);
2547177848Sdavidxu
2548177848Sdavidxu			umtxq_busy(&uq->uq_key);
2549177848Sdavidxu			umtxq_remove_queue(uq, UMTX_EXCLUSIVE_QUEUE);
2550177848Sdavidxu			umtxq_unlock(&uq->uq_key);
2551177848Sdavidxu			if (error)
2552177848Sdavidxu				break;
2553177848Sdavidxu			state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2554177848Sdavidxu		}
2555177848Sdavidxu
2556177848Sdavidxu		blocked_writers = fuword32(&rwlock->rw_blocked_writers);
2557177848Sdavidxu		suword32(&rwlock->rw_blocked_writers, blocked_writers-1);
2558177848Sdavidxu		if (blocked_writers == 1) {
2559177848Sdavidxu			state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2560177848Sdavidxu			for (;;) {
2561177848Sdavidxu				oldstate = casuword32(&rwlock->rw_state, state,
2562177848Sdavidxu					 state & ~URWLOCK_WRITE_WAITERS);
2563177848Sdavidxu				if (oldstate == state)
2564177848Sdavidxu					break;
2565177848Sdavidxu				state = oldstate;
2566177848Sdavidxu			}
2567177848Sdavidxu		}
2568177848Sdavidxu
2569177848Sdavidxu		umtxq_lock(&uq->uq_key);
2570177848Sdavidxu		umtxq_unbusy(&uq->uq_key);
2571177848Sdavidxu		umtxq_unlock(&uq->uq_key);
2572177848Sdavidxu	}
2573177848Sdavidxu
2574177848Sdavidxu	umtx_key_release(&uq->uq_key);
2575177848Sdavidxu	return (error);
2576177848Sdavidxu}
2577177848Sdavidxu
2578177848Sdavidxustatic int
2579177848Sdavidxudo_rw_wrlock2(struct thread *td, void *obj, struct timespec *timeout)
2580177848Sdavidxu{
2581177848Sdavidxu	struct timespec ts, ts2, ts3;
2582177848Sdavidxu	struct timeval tv;
2583177848Sdavidxu	int error;
2584177848Sdavidxu
2585177848Sdavidxu	getnanouptime(&ts);
2586177848Sdavidxu	timespecadd(&ts, timeout);
2587177848Sdavidxu	TIMESPEC_TO_TIMEVAL(&tv, timeout);
2588177848Sdavidxu	for (;;) {
2589177848Sdavidxu		error = do_rw_wrlock(td, obj, tvtohz(&tv));
2590177848Sdavidxu		if (error != ETIMEDOUT)
2591177848Sdavidxu			break;
2592177848Sdavidxu		getnanouptime(&ts2);
2593177848Sdavidxu		if (timespeccmp(&ts2, &ts, >=)) {
2594177848Sdavidxu			error = ETIMEDOUT;
2595177848Sdavidxu			break;
2596177848Sdavidxu		}
2597177848Sdavidxu		ts3 = ts;
2598177848Sdavidxu		timespecsub(&ts3, &ts2);
2599177848Sdavidxu		TIMESPEC_TO_TIMEVAL(&tv, &ts3);
2600177848Sdavidxu	}
2601177849Sdavidxu	if (error == ERESTART)
2602177849Sdavidxu		error = EINTR;
2603177848Sdavidxu	return (error);
2604177848Sdavidxu}
2605177848Sdavidxu
2606177848Sdavidxustatic int
2607177848Sdavidxudo_rwlock_unlock(struct thread *td, struct urwlock *rwlock)
2608177848Sdavidxu{
2609177848Sdavidxu	struct umtx_q *uq;
2610177848Sdavidxu	uint32_t flags;
2611177848Sdavidxu	int32_t state, oldstate;
2612177848Sdavidxu	int error, q, count;
2613177848Sdavidxu
2614177848Sdavidxu	uq = td->td_umtxq;
2615177848Sdavidxu	flags = fuword32(&rwlock->rw_flags);
2616177848Sdavidxu	error = umtx_key_get(rwlock, TYPE_RWLOCK, GET_SHARE(flags), &uq->uq_key);
2617177848Sdavidxu	if (error != 0)
2618177848Sdavidxu		return (error);
2619177848Sdavidxu
2620177848Sdavidxu	state = fuword32(__DEVOLATILE(int32_t *, &rwlock->rw_state));
2621177848Sdavidxu	if (state & URWLOCK_WRITE_OWNER) {
2622177848Sdavidxu		for (;;) {
2623177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state,
2624177848Sdavidxu				state & ~URWLOCK_WRITE_OWNER);
2625177848Sdavidxu			if (oldstate != state) {
2626177848Sdavidxu				state = oldstate;
2627177848Sdavidxu				if (!(oldstate & URWLOCK_WRITE_OWNER)) {
2628177848Sdavidxu					error = EPERM;
2629177848Sdavidxu					goto out;
2630177848Sdavidxu				}
2631177848Sdavidxu			} else
2632177848Sdavidxu				break;
2633177848Sdavidxu		}
2634177848Sdavidxu	} else if (URWLOCK_READER_COUNT(state) != 0) {
2635177848Sdavidxu		for (;;) {
2636177848Sdavidxu			oldstate = casuword32(&rwlock->rw_state, state,
2637177848Sdavidxu				state - 1);
2638177848Sdavidxu			if (oldstate != state) {
2639177848Sdavidxu				state = oldstate;
2640177848Sdavidxu				if (URWLOCK_READER_COUNT(oldstate) == 0) {
2641177848Sdavidxu					error = EPERM;
2642177848Sdavidxu					goto out;
2643177848Sdavidxu				}
2644177848Sdavidxu			}
2645177848Sdavidxu			else
2646177848Sdavidxu				break;
2647177848Sdavidxu		}
2648177848Sdavidxu	} else {
2649177848Sdavidxu		error = EPERM;
2650177848Sdavidxu		goto out;
2651177848Sdavidxu	}
2652177848Sdavidxu
2653177848Sdavidxu	count = 0;
2654177848Sdavidxu
2655177848Sdavidxu	if (!(flags & URWLOCK_PREFER_READER)) {
2656177848Sdavidxu		if (state & URWLOCK_WRITE_WAITERS) {
2657177848Sdavidxu			count = 1;
2658177848Sdavidxu			q = UMTX_EXCLUSIVE_QUEUE;
2659177848Sdavidxu		} else if (state & URWLOCK_READ_WAITERS) {
2660177848Sdavidxu			count = INT_MAX;
2661177848Sdavidxu			q = UMTX_SHARED_QUEUE;
2662177848Sdavidxu		}
2663177848Sdavidxu	} else {
2664177848Sdavidxu		if (state & URWLOCK_READ_WAITERS) {
2665177848Sdavidxu			count = INT_MAX;
2666177848Sdavidxu			q = UMTX_SHARED_QUEUE;
2667177848Sdavidxu		} else if (state & URWLOCK_WRITE_WAITERS) {
2668177848Sdavidxu			count = 1;
2669177848Sdavidxu			q = UMTX_EXCLUSIVE_QUEUE;
2670177848Sdavidxu		}
2671177848Sdavidxu	}
2672177848Sdavidxu
2673177848Sdavidxu	if (count) {
2674177848Sdavidxu		umtxq_lock(&uq->uq_key);
2675177848Sdavidxu		umtxq_busy(&uq->uq_key);
2676177848Sdavidxu		umtxq_signal_queue(&uq->uq_key, count, q);
2677177848Sdavidxu		umtxq_unbusy(&uq->uq_key);
2678177848Sdavidxu		umtxq_unlock(&uq->uq_key);
2679177848Sdavidxu	}
2680177848Sdavidxuout:
2681177848Sdavidxu	umtx_key_release(&uq->uq_key);
2682177848Sdavidxu	return (error);
2683177848Sdavidxu}
2684177848Sdavidxu
2685139013Sdavidxuint
2686139013Sdavidxu_umtx_lock(struct thread *td, struct _umtx_lock_args *uap)
2687139013Sdavidxu    /* struct umtx *umtx */
2688139013Sdavidxu{
2689162536Sdavidxu	return _do_lock_umtx(td, uap->umtx, td->td_tid, 0);
2690139013Sdavidxu}
2691139013Sdavidxu
2692139013Sdavidxuint
2693139013Sdavidxu_umtx_unlock(struct thread *td, struct _umtx_unlock_args *uap)
2694139013Sdavidxu    /* struct umtx *umtx */
2695139013Sdavidxu{
2696162536Sdavidxu	return do_unlock_umtx(td, uap->umtx, td->td_tid);
2697139013Sdavidxu}
2698139013Sdavidxu
2699162536Sdavidxustatic int
2700162536Sdavidxu__umtx_op_lock_umtx(struct thread *td, struct _umtx_op_args *uap)
2701139013Sdavidxu{
2702162536Sdavidxu	struct timespec *ts, timeout;
2703139013Sdavidxu	int error;
2704139013Sdavidxu
2705162536Sdavidxu	/* Allow a null timespec (wait forever). */
2706162536Sdavidxu	if (uap->uaddr2 == NULL)
2707162536Sdavidxu		ts = NULL;
2708162536Sdavidxu	else {
2709162536Sdavidxu		error = copyin(uap->uaddr2, &timeout, sizeof(timeout));
2710162536Sdavidxu		if (error != 0)
2711162536Sdavidxu			return (error);
2712162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2713162536Sdavidxu		    timeout.tv_nsec < 0) {
2714162536Sdavidxu			return (EINVAL);
2715161678Sdavidxu		}
2716162536Sdavidxu		ts = &timeout;
2717162536Sdavidxu	}
2718162536Sdavidxu	return (do_lock_umtx(td, uap->obj, uap->val, ts));
2719162536Sdavidxu}
2720162536Sdavidxu
2721162536Sdavidxustatic int
2722162536Sdavidxu__umtx_op_unlock_umtx(struct thread *td, struct _umtx_op_args *uap)
2723162536Sdavidxu{
2724162536Sdavidxu	return (do_unlock_umtx(td, uap->obj, uap->val));
2725162536Sdavidxu}
2726162536Sdavidxu
2727162536Sdavidxustatic int
2728162536Sdavidxu__umtx_op_wait(struct thread *td, struct _umtx_op_args *uap)
2729162536Sdavidxu{
2730162536Sdavidxu	struct timespec *ts, timeout;
2731162536Sdavidxu	int error;
2732162536Sdavidxu
2733162536Sdavidxu	if (uap->uaddr2 == NULL)
2734162536Sdavidxu		ts = NULL;
2735162536Sdavidxu	else {
2736162536Sdavidxu		error = copyin(uap->uaddr2, &timeout, sizeof(timeout));
2737162536Sdavidxu		if (error != 0)
2738162536Sdavidxu			return (error);
2739162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2740162536Sdavidxu		    timeout.tv_nsec < 0)
2741162536Sdavidxu			return (EINVAL);
2742162536Sdavidxu		ts = &timeout;
2743162536Sdavidxu	}
2744162536Sdavidxu	return do_wait(td, uap->obj, uap->val, ts, 0);
2745162536Sdavidxu}
2746162536Sdavidxu
2747162536Sdavidxustatic int
2748173800Sdavidxu__umtx_op_wait_uint(struct thread *td, struct _umtx_op_args *uap)
2749173800Sdavidxu{
2750173800Sdavidxu	struct timespec *ts, timeout;
2751173800Sdavidxu	int error;
2752173800Sdavidxu
2753173800Sdavidxu	if (uap->uaddr2 == NULL)
2754173800Sdavidxu		ts = NULL;
2755173800Sdavidxu	else {
2756173800Sdavidxu		error = copyin(uap->uaddr2, &timeout, sizeof(timeout));
2757173800Sdavidxu		if (error != 0)
2758173800Sdavidxu			return (error);
2759173800Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2760173800Sdavidxu		    timeout.tv_nsec < 0)
2761173800Sdavidxu			return (EINVAL);
2762173800Sdavidxu		ts = &timeout;
2763173800Sdavidxu	}
2764173800Sdavidxu	return do_wait(td, uap->obj, uap->val, ts, 1);
2765173800Sdavidxu}
2766173800Sdavidxu
2767173800Sdavidxustatic int
2768162536Sdavidxu__umtx_op_wake(struct thread *td, struct _umtx_op_args *uap)
2769162536Sdavidxu{
2770162536Sdavidxu	return (kern_umtx_wake(td, uap->obj, uap->val));
2771162536Sdavidxu}
2772162536Sdavidxu
2773162536Sdavidxustatic int
2774162536Sdavidxu__umtx_op_lock_umutex(struct thread *td, struct _umtx_op_args *uap)
2775162536Sdavidxu{
2776162536Sdavidxu	struct timespec *ts, timeout;
2777162536Sdavidxu	int error;
2778162536Sdavidxu
2779162536Sdavidxu	/* Allow a null timespec (wait forever). */
2780162536Sdavidxu	if (uap->uaddr2 == NULL)
2781162536Sdavidxu		ts = NULL;
2782162536Sdavidxu	else {
2783162536Sdavidxu		error = copyin(uap->uaddr2, &timeout,
2784162536Sdavidxu		    sizeof(timeout));
2785162536Sdavidxu		if (error != 0)
2786162536Sdavidxu			return (error);
2787162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2788162536Sdavidxu		    timeout.tv_nsec < 0) {
2789162536Sdavidxu			return (EINVAL);
2790139013Sdavidxu		}
2791162536Sdavidxu		ts = &timeout;
2792139013Sdavidxu	}
2793162536Sdavidxu	return do_lock_umutex(td, uap->obj, ts, 0);
2794162536Sdavidxu}
2795162536Sdavidxu
2796162536Sdavidxustatic int
2797162536Sdavidxu__umtx_op_trylock_umutex(struct thread *td, struct _umtx_op_args *uap)
2798162536Sdavidxu{
2799162536Sdavidxu	return do_lock_umutex(td, uap->obj, NULL, 1);
2800162536Sdavidxu}
2801162536Sdavidxu
2802162536Sdavidxustatic int
2803162536Sdavidxu__umtx_op_unlock_umutex(struct thread *td, struct _umtx_op_args *uap)
2804162536Sdavidxu{
2805162536Sdavidxu	return do_unlock_umutex(td, uap->obj);
2806162536Sdavidxu}
2807162536Sdavidxu
2808162536Sdavidxustatic int
2809162536Sdavidxu__umtx_op_set_ceiling(struct thread *td, struct _umtx_op_args *uap)
2810162536Sdavidxu{
2811162536Sdavidxu	return do_set_ceiling(td, uap->obj, uap->val, uap->uaddr1);
2812162536Sdavidxu}
2813162536Sdavidxu
2814164839Sdavidxustatic int
2815164839Sdavidxu__umtx_op_cv_wait(struct thread *td, struct _umtx_op_args *uap)
2816164839Sdavidxu{
2817164839Sdavidxu	struct timespec *ts, timeout;
2818164839Sdavidxu	int error;
2819164839Sdavidxu
2820164839Sdavidxu	/* Allow a null timespec (wait forever). */
2821164839Sdavidxu	if (uap->uaddr2 == NULL)
2822164839Sdavidxu		ts = NULL;
2823164839Sdavidxu	else {
2824164839Sdavidxu		error = copyin(uap->uaddr2, &timeout,
2825164839Sdavidxu		    sizeof(timeout));
2826164839Sdavidxu		if (error != 0)
2827164839Sdavidxu			return (error);
2828164839Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2829164839Sdavidxu		    timeout.tv_nsec < 0) {
2830164839Sdavidxu			return (EINVAL);
2831164839Sdavidxu		}
2832164839Sdavidxu		ts = &timeout;
2833164839Sdavidxu	}
2834164876Sdavidxu	return (do_cv_wait(td, uap->obj, uap->uaddr1, ts, uap->val));
2835164839Sdavidxu}
2836164839Sdavidxu
2837164839Sdavidxustatic int
2838164839Sdavidxu__umtx_op_cv_signal(struct thread *td, struct _umtx_op_args *uap)
2839164839Sdavidxu{
2840164839Sdavidxu	return do_cv_signal(td, uap->obj);
2841164839Sdavidxu}
2842164839Sdavidxu
2843164839Sdavidxustatic int
2844164839Sdavidxu__umtx_op_cv_broadcast(struct thread *td, struct _umtx_op_args *uap)
2845164839Sdavidxu{
2846164839Sdavidxu	return do_cv_broadcast(td, uap->obj);
2847164839Sdavidxu}
2848164839Sdavidxu
2849177848Sdavidxustatic int
2850177848Sdavidxu__umtx_op_rw_rdlock(struct thread *td, struct _umtx_op_args *uap)
2851177848Sdavidxu{
2852177848Sdavidxu	struct timespec timeout;
2853177848Sdavidxu	int error;
2854177848Sdavidxu
2855177848Sdavidxu	/* Allow a null timespec (wait forever). */
2856177848Sdavidxu	if (uap->uaddr2 == NULL) {
2857177848Sdavidxu		error = do_rw_rdlock(td, uap->obj, uap->val, 0);
2858177848Sdavidxu	} else {
2859177848Sdavidxu		error = copyin(uap->uaddr2, &timeout,
2860177848Sdavidxu		    sizeof(timeout));
2861177848Sdavidxu		if (error != 0)
2862177848Sdavidxu			return (error);
2863177848Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2864177848Sdavidxu		    timeout.tv_nsec < 0) {
2865177848Sdavidxu			return (EINVAL);
2866177848Sdavidxu		}
2867177848Sdavidxu		error = do_rw_rdlock2(td, uap->obj, uap->val, &timeout);
2868177848Sdavidxu	}
2869177848Sdavidxu	return (error);
2870177848Sdavidxu}
2871177848Sdavidxu
2872177848Sdavidxustatic int
2873177848Sdavidxu__umtx_op_rw_wrlock(struct thread *td, struct _umtx_op_args *uap)
2874177848Sdavidxu{
2875177848Sdavidxu	struct timespec timeout;
2876177848Sdavidxu	int error;
2877177848Sdavidxu
2878177848Sdavidxu	/* Allow a null timespec (wait forever). */
2879177848Sdavidxu	if (uap->uaddr2 == NULL) {
2880177848Sdavidxu		error = do_rw_wrlock(td, uap->obj, 0);
2881177848Sdavidxu	} else {
2882177848Sdavidxu		error = copyin(uap->uaddr2, &timeout,
2883177848Sdavidxu		    sizeof(timeout));
2884177848Sdavidxu		if (error != 0)
2885177848Sdavidxu			return (error);
2886177848Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2887177848Sdavidxu		    timeout.tv_nsec < 0) {
2888177848Sdavidxu			return (EINVAL);
2889177848Sdavidxu		}
2890177848Sdavidxu
2891177848Sdavidxu		error = do_rw_wrlock2(td, uap->obj, &timeout);
2892177848Sdavidxu	}
2893177848Sdavidxu	return (error);
2894177848Sdavidxu}
2895177848Sdavidxu
2896177848Sdavidxustatic int
2897177848Sdavidxu__umtx_op_rw_unlock(struct thread *td, struct _umtx_op_args *uap)
2898177848Sdavidxu{
2899177848Sdavidxu	return do_rwlock_unlock(td, uap->obj);
2900177848Sdavidxu}
2901177848Sdavidxu
2902162536Sdavidxutypedef int (*_umtx_op_func)(struct thread *td, struct _umtx_op_args *uap);
2903162536Sdavidxu
2904162536Sdavidxustatic _umtx_op_func op_table[] = {
2905162536Sdavidxu	__umtx_op_lock_umtx,		/* UMTX_OP_LOCK */
2906162536Sdavidxu	__umtx_op_unlock_umtx,		/* UMTX_OP_UNLOCK */
2907162536Sdavidxu	__umtx_op_wait,			/* UMTX_OP_WAIT */
2908162536Sdavidxu	__umtx_op_wake,			/* UMTX_OP_WAKE */
2909162536Sdavidxu	__umtx_op_trylock_umutex,	/* UMTX_OP_MUTEX_TRYLOCK */
2910162536Sdavidxu	__umtx_op_lock_umutex,		/* UMTX_OP_MUTEX_LOCK */
2911162536Sdavidxu	__umtx_op_unlock_umutex,	/* UMTX_OP_MUTEX_UNLOCK */
2912164839Sdavidxu	__umtx_op_set_ceiling,		/* UMTX_OP_SET_CEILING */
2913164839Sdavidxu	__umtx_op_cv_wait,		/* UMTX_OP_CV_WAIT*/
2914164839Sdavidxu	__umtx_op_cv_signal,		/* UMTX_OP_CV_SIGNAL */
2915173800Sdavidxu	__umtx_op_cv_broadcast,		/* UMTX_OP_CV_BROADCAST */
2916177848Sdavidxu	__umtx_op_wait_uint,		/* UMTX_OP_WAIT_UINT */
2917177848Sdavidxu	__umtx_op_rw_rdlock,		/* UMTX_OP_RW_RDLOCK */
2918177848Sdavidxu	__umtx_op_rw_wrlock,		/* UMTX_OP_RW_WRLOCK */
2919177848Sdavidxu	__umtx_op_rw_unlock		/* UMTX_OP_RW_UNLOCK */
2920162536Sdavidxu};
2921162536Sdavidxu
2922162536Sdavidxuint
2923162536Sdavidxu_umtx_op(struct thread *td, struct _umtx_op_args *uap)
2924162536Sdavidxu{
2925163678Sdavidxu	if ((unsigned)uap->op < UMTX_OP_MAX)
2926162536Sdavidxu		return (*op_table[uap->op])(td, uap);
2927162536Sdavidxu	return (EINVAL);
2928162536Sdavidxu}
2929162536Sdavidxu
2930162536Sdavidxu#ifdef COMPAT_IA32
2931163046Sdavidxuint
2932163046Sdavidxufreebsd32_umtx_lock(struct thread *td, struct freebsd32_umtx_lock_args *uap)
2933163046Sdavidxu    /* struct umtx *umtx */
2934163046Sdavidxu{
2935163046Sdavidxu	return (do_lock_umtx32(td, (uint32_t *)uap->umtx, td->td_tid, NULL));
2936163046Sdavidxu}
2937163046Sdavidxu
2938163046Sdavidxuint
2939163046Sdavidxufreebsd32_umtx_unlock(struct thread *td, struct freebsd32_umtx_unlock_args *uap)
2940163046Sdavidxu    /* struct umtx *umtx */
2941163046Sdavidxu{
2942163046Sdavidxu	return (do_unlock_umtx32(td, (uint32_t *)uap->umtx, td->td_tid));
2943163046Sdavidxu}
2944163046Sdavidxu
2945162536Sdavidxustruct timespec32 {
2946162536Sdavidxu	u_int32_t tv_sec;
2947162536Sdavidxu	u_int32_t tv_nsec;
2948162536Sdavidxu};
2949162536Sdavidxu
2950162536Sdavidxustatic inline int
2951162536Sdavidxucopyin_timeout32(void *addr, struct timespec *tsp)
2952162536Sdavidxu{
2953162536Sdavidxu	struct timespec32 ts32;
2954162536Sdavidxu	int error;
2955162536Sdavidxu
2956162536Sdavidxu	error = copyin(addr, &ts32, sizeof(struct timespec32));
2957162536Sdavidxu	if (error == 0) {
2958162536Sdavidxu		tsp->tv_sec = ts32.tv_sec;
2959162536Sdavidxu		tsp->tv_nsec = ts32.tv_nsec;
2960162536Sdavidxu	}
2961140421Sdavidxu	return (error);
2962139013Sdavidxu}
2963161678Sdavidxu
2964162536Sdavidxustatic int
2965162536Sdavidxu__umtx_op_lock_umtx_compat32(struct thread *td, struct _umtx_op_args *uap)
2966162536Sdavidxu{
2967162536Sdavidxu	struct timespec *ts, timeout;
2968162536Sdavidxu	int error;
2969162536Sdavidxu
2970162536Sdavidxu	/* Allow a null timespec (wait forever). */
2971162536Sdavidxu	if (uap->uaddr2 == NULL)
2972162536Sdavidxu		ts = NULL;
2973162536Sdavidxu	else {
2974162536Sdavidxu		error = copyin_timeout32(uap->uaddr2, &timeout);
2975162536Sdavidxu		if (error != 0)
2976162536Sdavidxu			return (error);
2977162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
2978162536Sdavidxu		    timeout.tv_nsec < 0) {
2979162536Sdavidxu			return (EINVAL);
2980162536Sdavidxu		}
2981162536Sdavidxu		ts = &timeout;
2982162536Sdavidxu	}
2983162536Sdavidxu	return (do_lock_umtx32(td, uap->obj, uap->val, ts));
2984162536Sdavidxu}
2985162536Sdavidxu
2986162536Sdavidxustatic int
2987162536Sdavidxu__umtx_op_unlock_umtx_compat32(struct thread *td, struct _umtx_op_args *uap)
2988162536Sdavidxu{
2989162536Sdavidxu	return (do_unlock_umtx32(td, uap->obj, (uint32_t)uap->val));
2990162536Sdavidxu}
2991162536Sdavidxu
2992162536Sdavidxustatic int
2993162536Sdavidxu__umtx_op_wait_compat32(struct thread *td, struct _umtx_op_args *uap)
2994162536Sdavidxu{
2995162536Sdavidxu	struct timespec *ts, timeout;
2996162536Sdavidxu	int error;
2997162536Sdavidxu
2998162536Sdavidxu	if (uap->uaddr2 == NULL)
2999162536Sdavidxu		ts = NULL;
3000162536Sdavidxu	else {
3001162536Sdavidxu		error = copyin_timeout32(uap->uaddr2, &timeout);
3002162536Sdavidxu		if (error != 0)
3003162536Sdavidxu			return (error);
3004162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
3005162536Sdavidxu		    timeout.tv_nsec < 0)
3006162536Sdavidxu			return (EINVAL);
3007162536Sdavidxu		ts = &timeout;
3008162536Sdavidxu	}
3009162536Sdavidxu	return do_wait(td, uap->obj, uap->val, ts, 1);
3010162536Sdavidxu}
3011162536Sdavidxu
3012162536Sdavidxustatic int
3013162536Sdavidxu__umtx_op_lock_umutex_compat32(struct thread *td, struct _umtx_op_args *uap)
3014162536Sdavidxu{
3015162536Sdavidxu	struct timespec *ts, timeout;
3016162536Sdavidxu	int error;
3017162536Sdavidxu
3018162536Sdavidxu	/* Allow a null timespec (wait forever). */
3019162536Sdavidxu	if (uap->uaddr2 == NULL)
3020162536Sdavidxu		ts = NULL;
3021162536Sdavidxu	else {
3022162536Sdavidxu		error = copyin_timeout32(uap->uaddr2, &timeout);
3023162536Sdavidxu		if (error != 0)
3024162536Sdavidxu			return (error);
3025162536Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
3026162536Sdavidxu		    timeout.tv_nsec < 0)
3027162536Sdavidxu			return (EINVAL);
3028162536Sdavidxu		ts = &timeout;
3029162536Sdavidxu	}
3030162536Sdavidxu	return do_lock_umutex(td, uap->obj, ts, 0);
3031162536Sdavidxu}
3032162536Sdavidxu
3033164839Sdavidxustatic int
3034164839Sdavidxu__umtx_op_cv_wait_compat32(struct thread *td, struct _umtx_op_args *uap)
3035164839Sdavidxu{
3036164839Sdavidxu	struct timespec *ts, timeout;
3037164839Sdavidxu	int error;
3038164839Sdavidxu
3039164839Sdavidxu	/* Allow a null timespec (wait forever). */
3040164839Sdavidxu	if (uap->uaddr2 == NULL)
3041164839Sdavidxu		ts = NULL;
3042164839Sdavidxu	else {
3043164839Sdavidxu		error = copyin_timeout32(uap->uaddr2, &timeout);
3044164839Sdavidxu		if (error != 0)
3045164839Sdavidxu			return (error);
3046164839Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
3047164839Sdavidxu		    timeout.tv_nsec < 0)
3048164839Sdavidxu			return (EINVAL);
3049164839Sdavidxu		ts = &timeout;
3050164839Sdavidxu	}
3051164876Sdavidxu	return (do_cv_wait(td, uap->obj, uap->uaddr1, ts, uap->val));
3052164839Sdavidxu}
3053164839Sdavidxu
3054177848Sdavidxustatic int
3055177848Sdavidxu__umtx_op_rw_rdlock_compat32(struct thread *td, struct _umtx_op_args *uap)
3056177848Sdavidxu{
3057177848Sdavidxu	struct timespec timeout;
3058177848Sdavidxu	int error;
3059177848Sdavidxu
3060177848Sdavidxu	/* Allow a null timespec (wait forever). */
3061177848Sdavidxu	if (uap->uaddr2 == NULL) {
3062177848Sdavidxu		error = do_rw_rdlock(td, uap->obj, uap->val, 0);
3063177848Sdavidxu	} else {
3064177848Sdavidxu		error = copyin(uap->uaddr2, &timeout,
3065177848Sdavidxu		    sizeof(timeout));
3066177848Sdavidxu		if (error != 0)
3067177848Sdavidxu			return (error);
3068177848Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
3069177848Sdavidxu		    timeout.tv_nsec < 0) {
3070177848Sdavidxu			return (EINVAL);
3071177848Sdavidxu		}
3072177848Sdavidxu		error = do_rw_rdlock2(td, uap->obj, uap->val, &timeout);
3073177848Sdavidxu	}
3074177848Sdavidxu	return (error);
3075177848Sdavidxu}
3076177848Sdavidxu
3077177848Sdavidxustatic int
3078177848Sdavidxu__umtx_op_rw_wrlock_compat32(struct thread *td, struct _umtx_op_args *uap)
3079177848Sdavidxu{
3080177848Sdavidxu	struct timespec timeout;
3081177848Sdavidxu	int error;
3082177848Sdavidxu
3083177848Sdavidxu	/* Allow a null timespec (wait forever). */
3084177848Sdavidxu	if (uap->uaddr2 == NULL) {
3085177848Sdavidxu		error = do_rwlock_wrlock(td, uap->obj, 0);
3086177848Sdavidxu	} else {
3087177848Sdavidxu		error = copyin_timeout32(uap->uaddr2, &timeout);
3088177848Sdavidxu		if (error != 0)
3089177848Sdavidxu			return (error);
3090177848Sdavidxu		if (timeout.tv_nsec >= 1000000000 ||
3091177848Sdavidxu		    timeout.tv_nsec < 0) {
3092177848Sdavidxu			return (EINVAL);
3093177848Sdavidxu		}
3094177848Sdavidxu
3095177848Sdavidxu		error = do_rw_wrlock2(td, uap->obj, uap->val, &timeout);
3096177848Sdavidxu	}
3097177848Sdavidxu	return (error);
3098177848Sdavidxu}
3099177848Sdavidxu
3100162536Sdavidxustatic _umtx_op_func op_table_compat32[] = {
3101162536Sdavidxu	__umtx_op_lock_umtx_compat32,	/* UMTX_OP_LOCK */
3102162536Sdavidxu	__umtx_op_unlock_umtx_compat32,	/* UMTX_OP_UNLOCK */
3103162536Sdavidxu	__umtx_op_wait_compat32,	/* UMTX_OP_WAIT */
3104162536Sdavidxu	__umtx_op_wake,			/* UMTX_OP_WAKE */
3105162550Sdavidxu	__umtx_op_trylock_umutex,	/* UMTX_OP_MUTEX_LOCK */
3106162536Sdavidxu	__umtx_op_lock_umutex_compat32,	/* UMTX_OP_MUTEX_TRYLOCK */
3107162536Sdavidxu	__umtx_op_unlock_umutex,	/* UMTX_OP_MUTEX_UNLOCK	*/
3108164839Sdavidxu	__umtx_op_set_ceiling,		/* UMTX_OP_SET_CEILING */
3109164839Sdavidxu	__umtx_op_cv_wait_compat32,	/* UMTX_OP_CV_WAIT*/
3110164839Sdavidxu	__umtx_op_cv_signal,		/* UMTX_OP_CV_SIGNAL */
3111173800Sdavidxu	__umtx_op_cv_broadcast,		/* UMTX_OP_CV_BROADCAST */
3112177848Sdavidxu	__umtx_op_wait_compat32,	/* UMTX_OP_WAIT_UINT */
3113177848Sdavidxu	__umtx_op_rw_rdlock_compat32,	/* UMTX_OP_RW_RDLOCK */
3114177848Sdavidxu	__umtx_op_rw_wrlock_compat32,	/* UMTX_OP_RW_WRLOCK */
3115177848Sdavidxu	__umtx_op_rw_unlock		/* UMTX_OP_RW_UNLOCK */
3116162536Sdavidxu};
3117162536Sdavidxu
3118162536Sdavidxuint
3119162536Sdavidxufreebsd32_umtx_op(struct thread *td, struct freebsd32_umtx_op_args *uap)
3120162536Sdavidxu{
3121163678Sdavidxu	if ((unsigned)uap->op < UMTX_OP_MAX)
3122162536Sdavidxu		return (*op_table_compat32[uap->op])(td,
3123162536Sdavidxu			(struct _umtx_op_args *)uap);
3124162536Sdavidxu	return (EINVAL);
3125162536Sdavidxu}
3126162536Sdavidxu#endif
3127162536Sdavidxu
3128161678Sdavidxuvoid
3129161678Sdavidxuumtx_thread_init(struct thread *td)
3130161678Sdavidxu{
3131161678Sdavidxu	td->td_umtxq = umtxq_alloc();
3132161678Sdavidxu	td->td_umtxq->uq_thread = td;
3133161678Sdavidxu}
3134161678Sdavidxu
3135161678Sdavidxuvoid
3136161678Sdavidxuumtx_thread_fini(struct thread *td)
3137161678Sdavidxu{
3138161678Sdavidxu	umtxq_free(td->td_umtxq);
3139161678Sdavidxu}
3140161678Sdavidxu
3141161678Sdavidxu/*
3142161678Sdavidxu * It will be called when new thread is created, e.g fork().
3143161678Sdavidxu */
3144161678Sdavidxuvoid
3145161678Sdavidxuumtx_thread_alloc(struct thread *td)
3146161678Sdavidxu{
3147161678Sdavidxu	struct umtx_q *uq;
3148161678Sdavidxu
3149161678Sdavidxu	uq = td->td_umtxq;
3150161678Sdavidxu	uq->uq_inherited_pri = PRI_MAX;
3151161678Sdavidxu
3152161678Sdavidxu	KASSERT(uq->uq_flags == 0, ("uq_flags != 0"));
3153161678Sdavidxu	KASSERT(uq->uq_thread == td, ("uq_thread != td"));
3154161678Sdavidxu	KASSERT(uq->uq_pi_blocked == NULL, ("uq_pi_blocked != NULL"));
3155161678Sdavidxu	KASSERT(TAILQ_EMPTY(&uq->uq_pi_contested), ("uq_pi_contested is not empty"));
3156161678Sdavidxu}
3157161678Sdavidxu
3158161678Sdavidxu/*
3159161678Sdavidxu * exec() hook.
3160161678Sdavidxu */
3161161678Sdavidxustatic void
3162161678Sdavidxuumtx_exec_hook(void *arg __unused, struct proc *p __unused,
3163161678Sdavidxu	struct image_params *imgp __unused)
3164161678Sdavidxu{
3165161678Sdavidxu	umtx_thread_cleanup(curthread);
3166161678Sdavidxu}
3167161678Sdavidxu
3168161678Sdavidxu/*
3169161678Sdavidxu * thread_exit() hook.
3170161678Sdavidxu */
3171161678Sdavidxuvoid
3172161678Sdavidxuumtx_thread_exit(struct thread *td)
3173161678Sdavidxu{
3174161678Sdavidxu	umtx_thread_cleanup(td);
3175161678Sdavidxu}
3176161678Sdavidxu
3177161678Sdavidxu/*
3178161678Sdavidxu * clean up umtx data.
3179161678Sdavidxu */
3180161678Sdavidxustatic void
3181161678Sdavidxuumtx_thread_cleanup(struct thread *td)
3182161678Sdavidxu{
3183161678Sdavidxu	struct umtx_q *uq;
3184161678Sdavidxu	struct umtx_pi *pi;
3185161678Sdavidxu
3186161678Sdavidxu	if ((uq = td->td_umtxq) == NULL)
3187161678Sdavidxu		return;
3188161678Sdavidxu
3189170300Sjeff	mtx_lock_spin(&umtx_lock);
3190161678Sdavidxu	uq->uq_inherited_pri = PRI_MAX;
3191161678Sdavidxu	while ((pi = TAILQ_FIRST(&uq->uq_pi_contested)) != NULL) {
3192161678Sdavidxu		pi->pi_owner = NULL;
3193161678Sdavidxu		TAILQ_REMOVE(&uq->uq_pi_contested, pi, pi_link);
3194161678Sdavidxu	}
3195174701Sdavidxu	thread_lock(td);
3196161678Sdavidxu	td->td_flags &= ~TDF_UBORROWING;
3197174701Sdavidxu	thread_unlock(td);
3198170300Sjeff	mtx_unlock_spin(&umtx_lock);
3199161678Sdavidxu}
3200