lib/libthr/thread/thr_umtx.c

   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 unmodified, this list of conditions, and the following
  10  *    disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  * $FreeBSD$
  27  *
  28  */
  29
  30 #include "thr_private.h"
  31 #include "thr_umtx.h"
  32
  33 void
  34 _thr_umutex_init(struct umutex *mtx)
  35 {
  36         static struct umutex default_mtx = DEFAULT_UMUTEX;
  37
  38         *mtx = default_mtx;
  39 }
  40
  41 int
  42 __thr_umutex_lock(struct umutex *mtx)
  43 {
  44         if (_umtx_op(mtx, UMTX_OP_MUTEX_LOCK, 0, 0, 0) != -1)
  45                 return 0;
  46         return (errno);
  47 }
  48
  49 int
  50 __thr_umutex_timedlock(struct umutex *mtx,
  51         const struct timespec *timeout)
  52 {
  53         if (timeout && (timeout->tv_sec < 0 || (timeout->tv_sec == 0 &&
  54                 timeout->tv_nsec <= 0)))
  55                 return (ETIMEDOUT);
  56         if (_umtx_op(mtx, UMTX_OP_MUTEX_LOCK, 0, 0,
  57                 __DECONST(void *, timeout)) != -1)
  58                 return (0);
  59         return (errno);
  60 }
  61
  62 int
  63 __thr_umutex_unlock(struct umutex *mtx)
  64 {
  65         if (_umtx_op(mtx, UMTX_OP_MUTEX_UNLOCK, 0, 0, 0) != -1)
  66                 return (0);
  67         return (errno);
  68 }
  69
  70 int
  71 __thr_umutex_trylock(struct umutex *mtx)
  72 {
  73         if (_umtx_op(mtx, UMTX_OP_MUTEX_TRYLOCK, 0, 0, 0) != -1)
  74                 return (0);
  75         return (errno);
  76 }
  77
  78 int
  79 __thr_umutex_set_ceiling(struct umutex *mtx, uint32_t ceiling,
  80         uint32_t *oldceiling)
  81 {
  82         if (_umtx_op(mtx, UMTX_OP_SET_CEILING, ceiling, oldceiling, 0) != -1)
  83                 return (0);
  84         return (errno);
  85 }
  86
  87 int
  88 _thr_umtx_wait(volatile long *mtx, long id, const struct timespec *timeout)
  89 {
  90         if (timeout && (timeout->tv_sec < 0 || (timeout->tv_sec == 0 &&
  91                 timeout->tv_nsec <= 0)))
  92                 return (ETIMEDOUT);
  93         if (_umtx_op(__DEVOLATILE(void *, mtx), UMTX_OP_WAIT, id, 0,
  94                 __DECONST(void*, timeout)) != -1)
  95                 return (0);
  96         return (errno);
  97 }
  98
  99 int
 100 _thr_umtx_wait_uint(volatile u_int *mtx, u_int id, const struct timespec *timeout)
 101 {
 102         if (timeout && (timeout->tv_sec < 0 || (timeout->tv_sec == 0 &&
 103                 timeout->tv_nsec <= 0)))
 104                 return (ETIMEDOUT);
 105         if (_umtx_op(__DEVOLATILE(void *, mtx), UMTX_OP_WAIT_UINT, id, 0,
 106                 __DECONST(void*, timeout)) != -1)
 107                 return (0);
 108         return (errno);
 109 }
 110
 111 int
 112 _thr_umtx_wake(volatile void *mtx, int nr_wakeup)
 113 {
 114         if (_umtx_op(__DEVOLATILE(void *, mtx), UMTX_OP_WAKE,
 115                 nr_wakeup, 0, 0) != -1)
 116                 return (0);
 117         return (errno);
 118 }
 119
 120 void
 121 _thr_ucond_init(struct ucond *cv)
 122 {
 123         bzero(cv, sizeof(struct ucond));
 124 }
 125
 126 int
 127 _thr_ucond_wait(struct ucond *cv, struct umutex *m,
 128         const struct timespec *timeout, int check_unparking)
 129 {
 130         if (timeout && (timeout->tv_sec < 0 || (timeout->tv_sec == 0 &&
 131             timeout->tv_nsec <= 0))) {
 132                 __thr_umutex_unlock(m);
 133                 return (ETIMEDOUT);
 134         }
 135         if (_umtx_op(cv, UMTX_OP_CV_WAIT,
 136                      check_unparking ? UMTX_CHECK_UNPARKING : 0,
 137                      m, __DECONST(void*, timeout)) != -1) {
 138                 return (0);
 139         }
 140         return (errno);
 141 }
 142
 143 int
 144 _thr_ucond_signal(struct ucond *cv)
 145 {
 146         if (!cv->c_has_waiters)
 147                 return (0);
 148         if (_umtx_op(cv, UMTX_OP_CV_SIGNAL, 0, NULL, NULL) != -1)
 149                 return (0);
 150         return (errno);
 151 }
 152
 153 int
 154 _thr_ucond_broadcast(struct ucond *cv)
 155 {
 156         if (!cv->c_has_waiters)
 157                 return (0);
 158         if (_umtx_op(cv, UMTX_OP_CV_BROADCAST, 0, NULL, NULL) != -1)
 159                 return (0);
 160         return (errno);
 161 }