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) == 0)
  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)) == 0)
  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) == 0)
  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) == 0)
  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) == 0)
  83                 return (0);
  84         return (errno);
  85 }
  86
  87 int
  88 _thr_umtx_wait(volatile umtx_t *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)) == 0)
  95                 return (0);
  96         return (errno);
  97 }
  98
  99 int
 100 _thr_umtx_wake(volatile umtx_t *mtx, int nr_wakeup)
 101 {
 102         if (_umtx_op(__DEVOLATILE(void *, mtx), UMTX_OP_WAKE,
 103                 nr_wakeup, 0, 0) == 0)
 104                 return (0);
 105         return (errno);
 106 }
 107
 108 void
 109 _thr_ucond_init(struct ucond *cv)
 110 {
 111         bzero(cv, sizeof(struct ucond));
 112 }
 113
 114 int
 115 _thr_ucond_wait(struct ucond *cv, struct umutex *m,
 116         const struct timespec *timeout, int check_unparking)
 117 {
 118         if (timeout && (timeout->tv_sec < 0 || (timeout->tv_sec == 0 &&
 119             timeout->tv_nsec <= 0))) {
 120                 __thr_umutex_unlock(m);
 121                 return (ETIMEDOUT);
 122         }
 123         if (_umtx_op(cv, UMTX_OP_CV_WAIT,
 124                      check_unparking ? UMTX_CHECK_UNPARKING : 0,
 125                      m, __DECONST(void*, timeout)) == 0) {
 126                 return (0);
 127         }
 128         return (errno);
 129 }
 130
 131 int
 132 _thr_ucond_signal(struct ucond *cv)
 133 {
 134         if (!cv->c_has_waiters)
 135                 return (0);
 136         if (_umtx_op(cv, UMTX_OP_CV_SIGNAL, 0, NULL, NULL) == 0)
 137                 return (0);
 138         return (errno);
 139 }
 140
 141 int
 142 _thr_ucond_broadcast(struct ucond *cv)
 143 {
 144         if (!cv->c_has_waiters)
 145                 return (0);
 146         if (_umtx_op(cv, UMTX_OP_CV_BROADCAST, 0, NULL, NULL) == 0)
 147                 return (0);
 148         return (errno);
 149 }