src/util/futex.h

   1 /*
   2  * Copyright © 2015 Intel
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  */
  23
  24 #ifndef UTIL_FUTEX_H
  25 #define UTIL_FUTEX_H
  26
  27 #if defined(HAVE_LINUX_FUTEX_H)
  28
  29 #include <limits.h>
  30 #include <stdint.h>
  31 #include <unistd.h>
  32 #include <linux/futex.h>
  33 #include <sys/syscall.h>
  34 #include <sys/time.h>
  35
  36 static inline long sys_futex(void *addr1, int op, int val1, const struct timespec *timeout, void *addr2, int val3)
  37 {
  38    return syscall(SYS_futex, addr1, op, val1, timeout, addr2, val3);
  39 }
  40
  41 static inline int futex_wake(uint32_t *addr, int count)
  42 {
  43    return sys_futex(addr, FUTEX_WAKE, count, NULL, NULL, 0);
  44 }
  45
  46 static inline int futex_wait(uint32_t *addr, int32_t value, const struct timespec *timeout)
  47 {
  48    /* FUTEX_WAIT_BITSET with FUTEX_BITSET_MATCH_ANY is equivalent to
  49     * FUTEX_WAIT, except that it treats the timeout as absolute. */
  50    return sys_futex(addr, FUTEX_WAIT_BITSET, value, timeout, NULL,
  51                     FUTEX_BITSET_MATCH_ANY);
  52 }
  53
  54 #elif defined(__FreeBSD__)
  55
  56 #include <assert.h>
  57 #include <errno.h>
  58 #include <fcntl.h>
  59 #include <sys/types.h>
  60 #include <sys/umtx.h>
  61 #include <sys/time.h>
  62
  63 static inline int futex_wake(uint32_t *addr, int count)
  64 {
  65    assert(count == (int)(uint32_t)count); /* Check that bits weren't discarded */
  66    return _umtx_op(addr, UMTX_OP_WAKE, (uint32_t)count, NULL, NULL) == -1 ? errno : 0;
  67 }
  68
  69 static inline int futex_wait(uint32_t *addr, int32_t value, struct timespec *timeout)
  70 {
  71    void *uaddr = NULL, *uaddr2 = NULL;
  72    struct _umtx_time tmo = {
  73       ._flags = UMTX_ABSTIME,
  74       ._clockid = CLOCK_MONOTONIC
  75    };
  76
  77    assert(value == (int)(uint32_t)value); /* Check that bits weren't discarded */
  78
  79    if (timeout != NULL) {
  80       tmo._timeout = *timeout;
  81       uaddr = (void *)(uintptr_t)sizeof(tmo);
  82       uaddr2 = (void *)&tmo;
  83    }
  84
  85    return _umtx_op(addr, UMTX_OP_WAIT_UINT, (uint32_t)value, uaddr, uaddr2) == -1 ? errno : 0;
  86 }
  87
  88 #elif defined(__OpenBSD__)
  89
  90 #include <sys/time.h>
  91 #include <sys/futex.h>
  92
  93 static inline int futex_wake(uint32_t *addr, int count)
  94 {
  95    return futex(addr, FUTEX_WAKE, count, NULL, NULL);
  96 }
  97
  98 static inline int futex_wait(uint32_t *addr, int32_t value, const struct timespec *timeout)
  99 {
 100    struct timespec tsrel, tsnow;
 101    clock_gettime(CLOCK_MONOTONIC, &tsnow);
 102    timespecsub(timeout, &tsrel, &tsrel);
 103    return futex(addr, FUTEX_WAIT, value, &tsrel, NULL);
 104 }
 105
 106 #endif
 107
 108 #endif /* UTIL_FUTEX_H */