include/internal/refcount.h

   1 /*
   2  * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL license (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9 #ifndef HEADER_INTERNAL_REFCOUNT_H
  10 # define HEADER_INTERNAL_REFCOUNT_H
  11
  12 # if defined(__STDC_VERSION__) && __STDC_VERSION__ >= 201112L \
  13      && !defined(__STDC_NO_ATOMICS__)
  14 # include <stdatomic.h>
  15 # define HAVE_C11_ATOMICS
  16 # endif
  17
  18 # if defined(HAVE_C11_ATOMICS) && defined(ATOMIC_INT_LOCK_FREE) \
  19      && ATOMIC_INT_LOCK_FREE > 0
  20
  21 # define HAVE_ATOMICS 1
  22
  23 typedef _Atomic int CRYPTO_REF_COUNT;
  24
  25 static ossl_inline int CRYPTO_UP_REF(_Atomic int *val, int *ret, void *lock)
  26 {
  27     *ret = atomic_fetch_add_explicit(val, 1, memory_order_relaxed) + 1;
  28     return 1;
  29 }
  30
  31 static ossl_inline int CRYPTO_DOWN_REF(_Atomic int *val, int *ret, void *lock)
  32 {
  33     *ret = atomic_fetch_sub_explicit(val, 1, memory_order_release) - 1;
  34     if (*ret == 0)
  35         atomic_thread_fence(memory_order_acquire);
  36     return 1;
  37 }
  38
  39 # elif defined(__GNUC__) && defined(__ATOMIC_RELAXED) && __GCC_ATOMIC_INT_LOCK_FREE > 0
  40
  41 # define HAVE_ATOMICS 1
  42
  43 typedef int CRYPTO_REF_COUNT;
  44
  45 static ossl_inline int CRYPTO_UP_REF(int *val, int *ret, void *lock)
  46 {
  47     *ret = __atomic_fetch_add(val, 1, __ATOMIC_RELAXED) + 1;
  48     return 1;
  49 }
  50
  51 static ossl_inline int CRYPTO_DOWN_REF(int *val, int *ret, void *lock)
  52 {
  53     *ret = __atomic_fetch_sub(val, 1, __ATOMIC_RELEASE) - 1;
  54     if (*ret == 0)
  55         __atomic_thread_fence(__ATOMIC_ACQUIRE);
  56     return 1;
  57 }
  58
  59 # else
  60
  61 typedef int CRYPTO_REF_COUNT;
  62
  63 # define CRYPTO_UP_REF(val, ret, lock) CRYPTO_atomic_add(val, 1, ret, lock)
  64 # define CRYPTO_DOWN_REF(val, ret, lock) CRYPTO_atomic_add(val, -1, ret, lock)
  65
  66 # endif
  67 #endif