crypto/threads_win.c

   1 /* ====================================================================
   2  * Copyright (c) 2016 The OpenSSL Project.  All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  *
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in
  13  *    the documentation and/or other materials provided with the
  14  *    distribution.
  15  *
  16  * 3. All advertising materials mentioning features or use of this
  17  *    software must display the following acknowledgment:
  18  *    "This product includes software developed by the OpenSSL Project
  19  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
  20  *
  21  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  22  *    endorse or promote products derived from this software without
  23  *    prior written permission. For written permission, please contact
  24  *    openssl-core@openssl.org.
  25  *
  26  * 5. Products derived from this software may not be called "OpenSSL"
  27  *    nor may "OpenSSL" appear in their names without prior written
  28  *    permission of the OpenSSL Project.
  29  *
  30  * 6. Redistributions of any form whatsoever must retain the following
  31  *    acknowledgment:
  32  *    "This product includes software developed by the OpenSSL Project
  33  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
  34  *
  35  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  36  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  37  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  38  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  39  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  40  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  41  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  42  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  43  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  44  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  45  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  46  * OF THE POSSIBILITY OF SUCH DAMAGE.
  47  * ====================================================================
  48  */
  49
  50 #include <openssl/crypto.h>
  51 #include "internal/threads.h"
  52
  53 #if defined(OPENSSL_THREADS) && !defined(CRYPTO_TDEBUG) && defined(OPENSSL_SYS_WINDOWS)
  54
  55 CRYPTO_RWLOCK *CRYPTO_THREAD_lock_new(void)
  56 {
  57     CRYPTO_RWLOCK *lock = OPENSSL_zalloc(sizeof(CRITICAL_SECTION));
  58     if (lock == NULL)
  59         return NULL;
  60
  61     /* 0x400 is the spin count value suggested in the documentation */
  62     if (!InitializeCriticalSectionAndSpinCount(lock, 0x400)) {
  63         OPENSSL_free(lock);
  64         return NULL;
  65     }
  66
  67     return lock;
  68 }
  69
  70 int CRYPTO_THREAD_read_lock(CRYPTO_RWLOCK *lock)
  71 {
  72     EnterCriticalSection(lock);
  73     return 1;
  74 }
  75
  76 int CRYPTO_THREAD_write_lock(CRYPTO_RWLOCK *lock)
  77 {
  78     EnterCriticalSection(lock);
  79     return 1;
  80 }
  81
  82 int CRYPTO_THREAD_unlock(CRYPTO_RWLOCK *lock)
  83 {
  84     LeaveCriticalSection(lock);
  85     return 1;
  86 }
  87
  88 void CRYPTO_THREAD_lock_free(CRYPTO_RWLOCK *lock)
  89 {
  90     if (lock == NULL)
  91         return;
  92
  93     DeleteCriticalSection(lock);
  94     OPENSSL_free(lock);
  95
  96     return;
  97 }
  98
  99 # if _WIN32_WINNT < 0x0600
 100
 101 #  define ONCE_UNINITED     0
 102 #  define ONCE_ININIT       1
 103 #  define ONCE_DONE         2
 104
 105 int CRYPTO_THREAD_run_once(CRYPTO_ONCE *once, void (*init)(void))
 106 {
 107     LONG volatile *lock = (LONG *)once;
 108     LONG result;
 109
 110     if (*lock == ONCE_DONE)
 111         return 1;
 112
 113     do {
 114         result = InterlockedCompareExchange(lock, ONCE_ININIT, ONCE_UNINITED);
 115         if (result == ONCE_UNINITED) {
 116             init();
 117             *lock = ONCE_DONE;
 118             return 1;
 119         }
 120     } while (result == ONCE_ININIT);
 121
 122     return (*lock == ONCE_DONE);
 123 }
 124
 125 # else
 126
 127 BOOL CALLBACK once_cb(PINIT_ONCE once, PVOID p, PVOID *pp)
 128 {
 129     void (*init)(void) = p;
 130
 131     init();
 132
 133     return TRUE;
 134 }
 135
 136 int CRYPTO_THREAD_run_once(CRYPTO_ONCE *once, void (*init)(void))
 137 {
 138     if (InitOnceExecuteOnce(once, once_cb, init, NULL))
 139         return 1;
 140
 141     return 0;
 142 }
 143
 144 # endif
 145
 146 int CRYPTO_THREAD_init_local(CRYPTO_THREAD_LOCAL *key, void (*cleanup)(void *))
 147 {
 148     *key = TlsAlloc();
 149     if (*key == TLS_OUT_OF_INDEXES)
 150         return 0;
 151
 152     return 1;
 153 }
 154
 155 void *CRYPTO_THREAD_get_local(CRYPTO_THREAD_LOCAL *key)
 156 {
 157     return TlsGetValue(*key);
 158 }
 159
 160 int CRYPTO_THREAD_set_local(CRYPTO_THREAD_LOCAL *key, void *val)
 161 {
 162     if (TlsSetValue(*key, val) == 0)
 163         return 0;
 164
 165     return 1;
 166 }
 167
 168 int CRYPTO_THREAD_cleanup_local(CRYPTO_THREAD_LOCAL *key)
 169 {
 170     if (TlsFree(*key) == 0)
 171         return 0;
 172
 173     return 1;
 174 }
 175
 176 CRYPTO_THREAD_ID CRYPTO_THREAD_get_current_id(void)
 177 {
 178     return GetCurrentThreadId();
 179 }
 180
 181 int CRYPTO_THREAD_compare_id(CRYPTO_THREAD_ID a, CRYPTO_THREAD_ID b)
 182 {
 183     return (a == b);
 184 }
 185
 186 int CRYPTO_atomic_add(int *val, int amount, int *ret, CRYPTO_RWLOCK *lock)
 187 {
 188     *ret = InterlockedExchangeAdd(val, amount) + amount;
 189     return 1;
 190 }
 191
 192 #endif