]> git.proxmox.com Git - mirror_frr.git/blobdiff - lib/seqlock.c
Merge pull request #5280 from qlyoung/doc-clean-topotest-json
[mirror_frr.git] / lib / seqlock.c
index 223d14952ceb5b4a164ae428377a3ebe8bee6e73..c05ec19db4dcb9e36534ab90b93458b641f2f7ba 100644 (file)
@@ -25,6 +25,7 @@
 #include "config.h"
 #endif
 
+#include <string.h>
 #include <unistd.h>
 #include <limits.h>
 #include <errno.h>
 
 #include "seqlock.h"
 
+/****************************************
+ * OS specific synchronization wrappers *
+ ****************************************/
+
+/*
+ * Linux: sys_futex()
+ */
 #ifdef HAVE_SYNC_LINUX_FUTEX
-/* Linux-specific - sys_futex() */
 #include <sys/syscall.h>
 #include <linux/futex.h>
 
-static long sys_futex(void *addr1, int op, int val1, struct timespec *timeout,
-               void *addr2, int val3)
+static long sys_futex(void *addr1, int op, int val1,
+                     const struct timespec *timeout, void *addr2, int val3)
 {
        return syscall(SYS_futex, addr1, op, val1, timeout, addr2, val3);
 }
 
 #define wait_once(sqlo, val)   \
        sys_futex((int *)&sqlo->pos, FUTEX_WAIT, (int)val, NULL, NULL, 0)
+#define wait_time(sqlo, val, time, reltime)    \
+       sys_futex((int *)&sqlo->pos, FUTEX_WAIT_BITSET, (int)val, time, \
+                 NULL, ~0U)
 #define wait_poke(sqlo)                \
        sys_futex((int *)&sqlo->pos, FUTEX_WAKE, INT_MAX, NULL, NULL, 0)
 
+/*
+ * OpenBSD: sys_futex(), almost the same as on Linux
+ */
 #elif defined(HAVE_SYNC_OPENBSD_FUTEX)
-/* OpenBSD variant of the above.  untested, not upstream in OpenBSD. */
 #include <sys/syscall.h>
 #include <sys/futex.h>
 
+#define TIME_RELATIVE          1
+
 #define wait_once(sqlo, val)   \
        futex((int *)&sqlo->pos, FUTEX_WAIT, (int)val, NULL, NULL, 0)
+#define wait_time(sqlo, val, time, reltime)    \
+       futex((int *)&sqlo->pos, FUTEX_WAIT, (int)val, reltime, NULL, 0)
 #define wait_poke(sqlo)                \
        futex((int *)&sqlo->pos, FUTEX_WAKE, INT_MAX, NULL, NULL, 0)
 
+/*
+ * FreeBSD: _umtx_op()
+ */
 #elif defined(HAVE_SYNC_UMTX_OP)
-/* FreeBSD-specific: umtx_op() */
 #include <sys/umtx.h>
 
 #define wait_once(sqlo, val)   \
        _umtx_op((void *)&sqlo->pos, UMTX_OP_WAIT_UINT, val, NULL, NULL)
+static int wait_time(struct seqlock *sqlo, uint32_t val,
+                     const struct timespec *abstime,
+                     const struct timespec *reltime)
+{
+       struct _umtx_time t;
+       t._flags = UMTX_ABSTIME;
+       t._clockid = CLOCK_MONOTONIC;
+       memcpy(&t._timeout, abstime, sizeof(t._timeout));
+       return _umtx_op((void *)&sqlo->pos, UMTX_OP_WAIT_UINT, val,
+                (void *)(uintptr_t) sizeof(t), &t);
+}
 #define wait_poke(sqlo)                \
        _umtx_op((void *)&sqlo->pos, UMTX_OP_WAKE, INT_MAX, NULL, NULL)
 
-#else
-/* generic version.  used on *BSD, Solaris and OSX.
+/*
+ * generic version.  used on NetBSD, Solaris and OSX.  really shitty.
  */
+#else
+
+#define TIME_ABS_REALTIME      1
 
 #define wait_init(sqlo)                do { \
                pthread_mutex_init(&sqlo->lock, NULL); \
@@ -80,6 +112,9 @@ static long sys_futex(void *addr1, int op, int val1, struct timespec *timeout,
        } while (0)
 #define wait_prep(sqlo)                pthread_mutex_lock(&sqlo->lock)
 #define wait_once(sqlo, val)   pthread_cond_wait(&sqlo->wake, &sqlo->lock)
+#define wait_time(sqlo, val, time, reltime) \
+                               pthread_cond_timedwait(&sqlo->wake, \
+                                                      &sqlo->lock, time);
 #define wait_done(sqlo)                pthread_mutex_unlock(&sqlo->lock)
 #define wait_poke(sqlo)                do { \
                pthread_mutex_lock(&sqlo->lock); \
@@ -103,18 +138,112 @@ void seqlock_wait(struct seqlock *sqlo, seqlock_val_t val)
        seqlock_assert_valid(val);
 
        wait_prep(sqlo);
-       while (1) {
-               cur = atomic_load_explicit(&sqlo->pos, memory_order_acquire);
-               if (!(cur & 1))
+       cur = atomic_load_explicit(&sqlo->pos, memory_order_relaxed);
+
+       while (cur & SEQLOCK_HELD) {
+               cal = SEQLOCK_VAL(cur) - val - 1;
+               assert(cal < 0x40000000 || cal > 0xc0000000);
+               if (cal < 0x80000000)
                        break;
-               cal = cur - val - 1;
+
+               if ((cur & SEQLOCK_WAITERS)
+                   || atomic_compare_exchange_weak_explicit(
+                               &sqlo->pos, &cur, cur | SEQLOCK_WAITERS,
+                               memory_order_relaxed, memory_order_relaxed)) {
+                       wait_once(sqlo, cur | SEQLOCK_WAITERS);
+                       cur = atomic_load_explicit(&sqlo->pos,
+                               memory_order_relaxed);
+               }
+               /* else: we failed to swap in cur because it just changed */
+       }
+       wait_done(sqlo);
+}
+
+bool seqlock_timedwait(struct seqlock *sqlo, seqlock_val_t val,
+                      const struct timespec *abs_monotime_limit)
+{
+/*
+ * ABS_REALTIME - used on NetBSD, Solaris and OSX
+ */
+#if TIME_ABS_REALTIME
+#define time_arg1 &abs_rt
+#define time_arg2 NULL
+#define time_prep
+       struct timespec curmono, abs_rt;
+
+       clock_gettime(CLOCK_MONOTONIC, &curmono);
+       clock_gettime(CLOCK_REALTIME, &abs_rt);
+
+       abs_rt.tv_nsec += abs_monotime_limit->tv_nsec - curmono.tv_nsec;
+       if (abs_rt.tv_nsec < 0) {
+               abs_rt.tv_sec--;
+               abs_rt.tv_nsec += 1000000000;
+       } else if (abs_rt.tv_nsec >= 1000000000) {
+               abs_rt.tv_sec++;
+               abs_rt.tv_nsec -= 1000000000;
+       }
+       abs_rt.tv_sec += abs_monotime_limit->tv_sec - curmono.tv_sec;
+
+/*
+ * RELATIVE - used on OpenBSD (might get a patch to get absolute monotime)
+ */
+#elif TIME_RELATIVE
+       struct timespec reltime;
+
+#define time_arg1 abs_monotime_limit
+#define time_arg2 &reltime
+#define time_prep \
+       clock_gettime(CLOCK_MONOTONIC, &reltime);                              \
+       reltime.tv_sec = abs_monotime_limit.tv_sec - reltime.tv_sec;           \
+       reltime.tv_nsec = abs_monotime_limit.tv_nsec - reltime.tv_nsec;        \
+       if (reltime.tv_nsec < 0) {                                             \
+               reltime.tv_sec--;                                              \
+               reltime.tv_nsec += 1000000000;                                 \
+       }
+/*
+ * FreeBSD & Linux: absolute time re. CLOCK_MONOTONIC
+ */
+#else
+#define time_arg1 abs_monotime_limit
+#define time_arg2 NULL
+#define time_prep
+#endif
+
+       bool ret = true;
+       seqlock_val_t cur, cal;
+
+       seqlock_assert_valid(val);
+
+       wait_prep(sqlo);
+       cur = atomic_load_explicit(&sqlo->pos, memory_order_relaxed);
+
+       while (cur & SEQLOCK_HELD) {
+               cal = SEQLOCK_VAL(cur) - val - 1;
                assert(cal < 0x40000000 || cal > 0xc0000000);
                if (cal < 0x80000000)
                        break;
 
-               wait_once(sqlo, cur);
+               if ((cur & SEQLOCK_WAITERS)
+                   || atomic_compare_exchange_weak_explicit(
+                               &sqlo->pos, &cur, cur | SEQLOCK_WAITERS,
+                               memory_order_relaxed, memory_order_relaxed)) {
+                       int rv;
+
+                       time_prep
+
+                       rv = wait_time(sqlo, cur | SEQLOCK_WAITERS, time_arg1,
+                                      time_arg2);
+                       if (rv) {
+                               ret = false;
+                               break;
+                       }
+                       cur = atomic_load_explicit(&sqlo->pos,
+                               memory_order_relaxed);
+               }
        }
        wait_done(sqlo);
+
+       return ret;
 }
 
 bool seqlock_check(struct seqlock *sqlo, seqlock_val_t val)
@@ -123,26 +252,32 @@ bool seqlock_check(struct seqlock *sqlo, seqlock_val_t val)
 
        seqlock_assert_valid(val);
 
-       cur = atomic_load_explicit(&sqlo->pos, memory_order_acquire);
-       if (!(cur & 1))
+       cur = atomic_load_explicit(&sqlo->pos, memory_order_relaxed);
+       if (!(cur & SEQLOCK_HELD))
                return 1;
-       cur -= val;
+       cur = SEQLOCK_VAL(cur) - val - 1;
        assert(cur < 0x40000000 || cur > 0xc0000000);
        return cur < 0x80000000;
 }
 
 void seqlock_acquire_val(struct seqlock *sqlo, seqlock_val_t val)
 {
+       seqlock_val_t prev;
+
        seqlock_assert_valid(val);
 
-       atomic_store_explicit(&sqlo->pos, val, memory_order_release);
-       wait_poke(sqlo);
+       prev = atomic_exchange_explicit(&sqlo->pos, val, memory_order_relaxed);
+       if (prev & SEQLOCK_WAITERS)
+               wait_poke(sqlo);
 }
 
 void seqlock_release(struct seqlock *sqlo)
 {
-       atomic_store_explicit(&sqlo->pos, 0, memory_order_release);
-       wait_poke(sqlo);
+       seqlock_val_t prev;
+
+       prev = atomic_exchange_explicit(&sqlo->pos, 0, memory_order_relaxed);
+       if (prev & SEQLOCK_WAITERS)
+               wait_poke(sqlo);
 }
 
 void seqlock_init(struct seqlock *sqlo)
@@ -154,14 +289,23 @@ void seqlock_init(struct seqlock *sqlo)
 
 seqlock_val_t seqlock_cur(struct seqlock *sqlo)
 {
-       return atomic_load_explicit(&sqlo->pos, memory_order_acquire);
+       return SEQLOCK_VAL(atomic_load_explicit(&sqlo->pos,
+                                               memory_order_relaxed));
 }
 
 seqlock_val_t seqlock_bump(struct seqlock *sqlo)
 {
-       seqlock_val_t val;
+       seqlock_val_t val, cur;
+
+       cur = atomic_load_explicit(&sqlo->pos, memory_order_relaxed);
+       seqlock_assert_valid(cur);
+
+       do {
+               val = SEQLOCK_VAL(cur) + SEQLOCK_INCR;
+       } while (!atomic_compare_exchange_weak_explicit(&sqlo->pos, &cur, val,
+                       memory_order_relaxed, memory_order_relaxed));
 
-       val = atomic_fetch_add_explicit(&sqlo->pos, 2, memory_order_release);
-       wait_poke(sqlo);
+       if (cur & SEQLOCK_WAITERS)
+               wait_poke(sqlo);
        return val;
 }