xref: /freebsd/lib/libthr/thread/thr_umtx.h (revision 8042f26d)
1 /*-
2  * Copyright (c) 2005 David Xu <davidxu@freebsd.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #ifndef _THR_FBSD_UMTX_H_
30 #define _THR_FBSD_UMTX_H_
31 
32 #include <strings.h>
33 #include <sys/umtx.h>
34 
35 #define DEFAULT_UMUTEX	{0, 0, {0, 0}, {0, 0, 0, 0}}
36 
37 typedef long umtx_t;
38 
39 int __thr_umutex_lock(struct umutex *mtx) __hidden;
40 int __thr_umutex_timedlock(struct umutex *mtx,
41 	const struct timespec *timeout) __hidden;
42 int __thr_umutex_unlock(struct umutex *mtx) __hidden;
43 int __thr_umutex_trylock(struct umutex *mtx) __hidden;
44 int __thr_umutex_set_ceiling(struct umutex *mtx, uint32_t ceiling,
45 	uint32_t *oldceiling) __hidden;
46 
47 void _thr_umutex_init(struct umutex *mtx) __hidden;
48 int _thr_umtx_wait(volatile umtx_t *mtx, umtx_t exp,
49 	const struct timespec *timeout) __hidden;
50 int _thr_umtx_wake(volatile umtx_t *mtx, int count) __hidden;
51 
52 static inline int
53 _thr_umutex_trylock(struct umutex *mtx, uint32_t id)
54 {
55     if (atomic_cmpset_acq_32(&mtx->m_owner, UMUTEX_UNOWNED, id))
56 	return (0);
57     if ((mtx->m_flags & UMUTEX_PRIO_PROTECT) == 0)
58     	return (EBUSY);
59     return (__thr_umutex_trylock(mtx));
60 }
61 
62 static inline int
63 _thr_umutex_lock(struct umutex *mtx, uint32_t id)
64 {
65     if (atomic_cmpset_acq_32(&mtx->m_owner, UMUTEX_UNOWNED, id))
66 	return (0);
67     return (__thr_umutex_lock(mtx));
68 }
69 
70 static inline int
71 _thr_umutex_timedlock(struct umutex *mtx, uint32_t id,
72 	const struct timespec *timeout)
73 {
74     if (atomic_cmpset_acq_32(&mtx->m_owner, UMUTEX_UNOWNED, id))
75 	return (0);
76     return (__thr_umutex_timedlock(mtx, timeout));
77 }
78 
79 static inline int
80 _thr_umutex_unlock(struct umutex *mtx, uint32_t id)
81 {
82     if (atomic_cmpset_rel_32(&mtx->m_owner, id, UMUTEX_UNOWNED))
83 	return (0);
84     return (__thr_umutex_unlock(mtx));
85 }
86 
87 #endif
88