crypto/threads_pthread.c

   1 /*
   2  * Copyright 2016-2018 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (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
  10 #include <openssl/crypto.h>
  11 #include "internal/cryptlib.h"
  12
  13 #if defined(__sun)
  14 # include <atomic.h>
  15 #endif
  16
  17 #if defined(OPENSSL_THREADS) && !defined(CRYPTO_TDEBUG) && !defined(OPENSSL_SYS_WINDOWS)
  18
  19 # if defined(OPENSSL_SYS_UNIX)
  20 #  include <sys/types.h>
  21 #  include <unistd.h>
  22 #endif
  23
  24 # ifdef PTHREAD_RWLOCK_INITIALIZER
  25 #  define USE_RWLOCK
  26 # endif
  27
  28 CRYPTO_RWLOCK *CRYPTO_THREAD_lock_new(void)
  29 {
  30 # ifdef USE_RWLOCK
  31     CRYPTO_RWLOCK *lock;
  32
  33     if ((lock = OPENSSL_zalloc(sizeof(pthread_rwlock_t))) == NULL) {
  34         /* Don't set error, to avoid recursion blowup. */
  35         return NULL;
  36     }
  37
  38     if (pthread_rwlock_init(lock, NULL) != 0) {
  39         OPENSSL_free(lock);
  40         return NULL;
  41     }
  42 # else
  43     pthread_mutexattr_t attr;
  44     CRYPTO_RWLOCK *lock;
  45
  46     if ((lock = OPENSSL_zalloc(sizeof(pthread_mutex_t))) == NULL) {
  47         /* Don't set error, to avoid recursion blowup. */
  48         return NULL;
  49     }
  50
  51     pthread_mutexattr_init(&attr);
  52     pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
  53
  54     if (pthread_mutex_init(lock, &attr) != 0) {
  55         pthread_mutexattr_destroy(&attr);
  56         OPENSSL_free(lock);
  57         return NULL;
  58     }
  59
  60     pthread_mutexattr_destroy(&attr);
  61 # endif
  62
  63     return lock;
  64 }
  65
  66 int CRYPTO_THREAD_read_lock(CRYPTO_RWLOCK *lock)
  67 {
  68 # ifdef USE_RWLOCK
  69     if (pthread_rwlock_rdlock(lock) != 0)
  70         return 0;
  71 # else
  72     if (pthread_mutex_lock(lock) != 0)
  73         return 0;
  74 # endif
  75
  76     return 1;
  77 }
  78
  79 int CRYPTO_THREAD_write_lock(CRYPTO_RWLOCK *lock)
  80 {
  81 # ifdef USE_RWLOCK
  82     if (pthread_rwlock_wrlock(lock) != 0)
  83         return 0;
  84 # else
  85     if (pthread_mutex_lock(lock) != 0)
  86         return 0;
  87 # endif
  88
  89     return 1;
  90 }
  91
  92 int CRYPTO_THREAD_unlock(CRYPTO_RWLOCK *lock)
  93 {
  94 # ifdef USE_RWLOCK
  95     if (pthread_rwlock_unlock(lock) != 0)
  96         return 0;
  97 # else
  98     if (pthread_mutex_unlock(lock) != 0)
  99         return 0;
 100 # endif
 101
 102     return 1;
 103 }
 104
 105 void CRYPTO_THREAD_lock_free(CRYPTO_RWLOCK *lock)
 106 {
 107     if (lock == NULL)
 108         return;
 109
 110 # ifdef USE_RWLOCK
 111     pthread_rwlock_destroy(lock);
 112 # else
 113     pthread_mutex_destroy(lock);
 114 # endif
 115     OPENSSL_free(lock);
 116
 117     return;
 118 }
 119
 120 int CRYPTO_THREAD_run_once(CRYPTO_ONCE *once, void (*init)(void))
 121 {
 122     if (pthread_once(once, init) != 0)
 123         return 0;
 124
 125     return 1;
 126 }
 127
 128 int CRYPTO_THREAD_init_local(CRYPTO_THREAD_LOCAL *key, void (*cleanup)(void *))
 129 {
 130     if (pthread_key_create(key, cleanup) != 0)
 131         return 0;
 132
 133     return 1;
 134 }
 135
 136 void *CRYPTO_THREAD_get_local(CRYPTO_THREAD_LOCAL *key)
 137 {
 138     return pthread_getspecific(*key);
 139 }
 140
 141 int CRYPTO_THREAD_set_local(CRYPTO_THREAD_LOCAL *key, void *val)
 142 {
 143     if (pthread_setspecific(*key, val) != 0)
 144         return 0;
 145
 146     return 1;
 147 }
 148
 149 int CRYPTO_THREAD_cleanup_local(CRYPTO_THREAD_LOCAL *key)
 150 {
 151     if (pthread_key_delete(*key) != 0)
 152         return 0;
 153
 154     return 1;
 155 }
 156
 157 CRYPTO_THREAD_ID CRYPTO_THREAD_get_current_id(void)
 158 {
 159     return pthread_self();
 160 }
 161
 162 int CRYPTO_THREAD_compare_id(CRYPTO_THREAD_ID a, CRYPTO_THREAD_ID b)
 163 {
 164     return pthread_equal(a, b);
 165 }
 166
 167 int CRYPTO_atomic_add(int *val, int amount, int *ret, CRYPTO_RWLOCK *lock)
 168 {
 169 # if defined(__GNUC__) && defined(__ATOMIC_ACQ_REL)
 170     if (__atomic_is_lock_free(sizeof(*val), val)) {
 171         *ret = __atomic_add_fetch(val, amount, __ATOMIC_ACQ_REL);
 172         return 1;
 173     }
 174 # elif defined(__sun) && (defined(__SunOS_5_10) || defined(__SunOS_5_11))
 175     /* This will work for all future Solaris versions. */
 176     if (ret != NULL) {
 177         *ret = atomic_add_int_nv((volatile unsigned int *)val, amount);
 178         return 1;
 179     }
 180 # endif
 181     if (!CRYPTO_THREAD_write_lock(lock))
 182         return 0;
 183
 184     *val += amount;
 185     *ret  = *val;
 186
 187     if (!CRYPTO_THREAD_unlock(lock))
 188         return 0;
 189
 190     return 1;
 191 }
 192
 193 # ifndef FIPS_MODE
 194 /* TODO(3.0): No fork protection in FIPS module yet! */
 195
 196 #  ifdef OPENSSL_SYS_UNIX
 197 static pthread_once_t fork_once_control = PTHREAD_ONCE_INIT;
 198
 199 static void fork_once_func(void)
 200 {
 201     pthread_atfork(OPENSSL_fork_prepare,
 202                    OPENSSL_fork_parent, OPENSSL_fork_child);
 203 }
 204 #  endif
 205
 206 int openssl_init_fork_handlers(void)
 207 {
 208 #  ifdef OPENSSL_SYS_UNIX
 209     if (pthread_once(&fork_once_control, fork_once_func) == 0)
 210         return 1;
 211 #  endif
 212     return 0;
 213 }
 214 # endif /* FIPS_MODE */
 215
 216 int openssl_get_fork_id(void)
 217 {
 218     return getpid();
 219 }
 220 #endif