sbin/hastd/synch.h

   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$
  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
  42 static __inline void
  43 mtx_init(pthread_mutex_t *lock)
  44 {
  45         int error;
  46
  47         error = pthread_mutex_init(lock, NULL);
  48         assert(error == 0);
  49 }
  50 static __inline void
  51 mtx_destroy(pthread_mutex_t *lock)
  52 {
  53         int error;
  54
  55         error = pthread_mutex_destroy(lock);
  56         assert(error == 0);
  57 }
  58 static __inline void
  59 mtx_lock(pthread_mutex_t *lock)
  60 {
  61         int error;
  62
  63         error = pthread_mutex_lock(lock);
  64         assert(error == 0);
  65 }
  66 static __inline bool
  67 mtx_trylock(pthread_mutex_t *lock)
  68 {
  69         int error;
  70
  71         error = pthread_mutex_trylock(lock);
  72         assert(error == 0 || error == EBUSY);
  73         return (error == 0);
  74 }
  75 static __inline void
  76 mtx_unlock(pthread_mutex_t *lock)
  77 {
  78         int error;
  79
  80         error = pthread_mutex_unlock(lock);
  81         assert(error == 0);
  82 }
  83 static __inline bool
  84 mtx_owned(pthread_mutex_t *lock)
  85 {
  86
  87         return (pthread_mutex_isowned_np(lock) != 0);
  88 }
  89
  90 static __inline void
  91 rw_init(pthread_rwlock_t *lock)
  92 {
  93         int error;
  94
  95         error = pthread_rwlock_init(lock, NULL);
  96         assert(error == 0);
  97 }
  98 static __inline void
  99 rw_destroy(pthread_rwlock_t *lock)
 100 {
 101         int error;
 102
 103         error = pthread_rwlock_destroy(lock);
 104         assert(error == 0);
 105 }
 106 static __inline void
 107 rw_rlock(pthread_rwlock_t *lock)
 108 {
 109         int error;
 110
 111         error = pthread_rwlock_rdlock(lock);
 112         assert(error == 0);
 113 }
 114 static __inline void
 115 rw_wlock(pthread_rwlock_t *lock)
 116 {
 117         int error;
 118
 119         error = pthread_rwlock_wrlock(lock);
 120         assert(error == 0);
 121 }
 122 static __inline void
 123 rw_unlock(pthread_rwlock_t *lock)
 124 {
 125         int error;
 126
 127         error = pthread_rwlock_unlock(lock);
 128         assert(error == 0);
 129 }
 130
 131 static __inline void
 132 cv_init(pthread_cond_t *cv)
 133 {
 134         pthread_condattr_t attr;
 135         int error;
 136
 137         error = pthread_condattr_init(&attr);
 138         assert(error == 0);
 139         error = pthread_condattr_setclock(&attr, CLOCK_MONOTONIC);
 140         assert(error == 0);
 141         error = pthread_cond_init(cv, &attr);
 142         assert(error == 0);
 143         error = pthread_condattr_destroy(&attr);
 144         assert(error == 0);
 145 }
 146 static __inline void
 147 cv_wait(pthread_cond_t *cv, pthread_mutex_t *lock)
 148 {
 149         int error;
 150
 151         error = pthread_cond_wait(cv, lock);
 152         assert(error == 0);
 153 }
 154 static __inline bool
 155 cv_timedwait(pthread_cond_t *cv, pthread_mutex_t *lock, int timeout)
 156 {
 157         struct timespec ts;
 158         int error;
 159
 160         if (timeout == 0) {
 161                 cv_wait(cv, lock);
 162                 return (false);
 163         }
 164
 165         error = clock_gettime(CLOCK_MONOTONIC, &ts);
 166         assert(error == 0);
 167         ts.tv_sec += timeout;
 168         error = pthread_cond_timedwait(cv, lock, &ts);
 169         assert(error == 0 || error == ETIMEDOUT);
 170         return (error == ETIMEDOUT);
 171 }
 172 static __inline void
 173 cv_signal(pthread_cond_t *cv)
 174 {
 175         int error;
 176
 177         error = pthread_cond_signal(cv);
 178         assert(error == 0);
 179 }
 180 static __inline void
 181 cv_broadcast(pthread_cond_t *cv)
 182 {
 183         int error;
 184
 185         error = pthread_cond_broadcast(cv);
 186         assert(error == 0);
 187 }
 188 #endif  /* !_SYNCH_H_ */