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