synch.h revision 211876
1/*- 2 * Copyright (c) 2009-2010 The FreeBSD Foundation 3 * All rights reserved. 4 * 5 * This software was developed by Pawel Jakub Dawidek under sponsorship from 6 * the FreeBSD Foundation. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 * 29 * $FreeBSD: head/sbin/hastd/synch.h 211876 2010-08-27 13:58:38Z pjd $ 30 */ 31 32#ifndef _SYNCH_H_ 33#define _SYNCH_H_ 34 35#include <assert.h> 36#include <errno.h> 37#include <pthread.h> 38#include <pthread_np.h> 39#include <stdbool.h> 40#include <time.h> 41 42static __inline void 43mtx_init(pthread_mutex_t *lock) 44{ 45 int error; 46 47 error = pthread_mutex_init(lock, NULL); 48 assert(error == 0); 49} 50static __inline void 51mtx_lock(pthread_mutex_t *lock) 52{ 53 int error; 54 55 error = pthread_mutex_lock(lock); 56 assert(error == 0); 57} 58static __inline bool 59mtx_trylock(pthread_mutex_t *lock) 60{ 61 int error; 62 63 error = pthread_mutex_trylock(lock); 64 assert(error == 0 || error == EBUSY); 65 return (error == 0); 66} 67static __inline void 68mtx_unlock(pthread_mutex_t *lock) 69{ 70 int error; 71 72 error = pthread_mutex_unlock(lock); 73 assert(error == 0); 74} 75static __inline bool 76mtx_owned(pthread_mutex_t *lock) 77{ 78 79 return (pthread_mutex_isowned_np(lock) != 0); 80} 81 82static __inline void 83rw_init(pthread_rwlock_t *lock) 84{ 85 int error; 86 87 error = pthread_rwlock_init(lock, NULL); 88 assert(error == 0); 89} 90static __inline void 91rw_rlock(pthread_rwlock_t *lock) 92{ 93 int error; 94 95 error = pthread_rwlock_rdlock(lock); 96 assert(error == 0); 97} 98static __inline void 99rw_wlock(pthread_rwlock_t *lock) 100{ 101 int error; 102 103 error = pthread_rwlock_wrlock(lock); 104 assert(error == 0); 105} 106static __inline void 107rw_unlock(pthread_rwlock_t *lock) 108{ 109 int error; 110 111 error = pthread_rwlock_unlock(lock); 112 assert(error == 0); 113} 114 115static __inline void 116cv_init(pthread_cond_t *cv) 117{ 118 pthread_condattr_t attr; 119 int error; 120 121 error = pthread_condattr_init(&attr); 122 assert(error == 0); 123 error = pthread_condattr_setclock(&attr, CLOCK_MONOTONIC); 124 assert(error == 0); 125 error = pthread_cond_init(cv, &attr); 126 assert(error == 0); 127} 128static __inline void 129cv_wait(pthread_cond_t *cv, pthread_mutex_t *lock) 130{ 131 int error; 132 133 error = pthread_cond_wait(cv, lock); 134 assert(error == 0); 135} 136static __inline bool 137cv_timedwait(pthread_cond_t *cv, pthread_mutex_t *lock, int timeout) 138{ 139 struct timespec ts; 140 int error; 141 142 if (timeout == 0) { 143 cv_wait(cv, lock); 144 return (false); 145 } 146 147 error = clock_gettime(CLOCK_MONOTONIC, &ts); 148 assert(error == 0); 149 ts.tv_sec += timeout; 150 error = pthread_cond_timedwait(cv, lock, &ts); 151 assert(error == 0 || error == ETIMEDOUT); 152 return (error == ETIMEDOUT); 153} 154static __inline void 155cv_signal(pthread_cond_t *cv) 156{ 157 int error; 158 159 error = pthread_cond_signal(cv); 160 assert(error == 0); 161} 162static __inline void 163cv_broadcast(pthread_cond_t *cv) 164{ 165 int error; 166 167 error = pthread_cond_broadcast(cv); 168 assert(error == 0); 169} 170#endif /* !_SYNCH_H_ */ 171