1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
// Copyright 2015 The BoringSSL Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Ensure we can't call OPENSSL_malloc circularly.
#define _BORINGSSL_PROHIBIT_OPENSSL_MALLOC
#include "internal.h"
#if defined(OPENSSL_PTHREADS)
#include <assert.h>
#include <pthread.h>
#include <stdlib.h>
#include <string.h>
BSSL_NAMESPACE_BEGIN
void StaticMutex::LockRead() { BSSL_CHECK(pthread_rwlock_rdlock(&lock_) == 0); }
void StaticMutex::UnlockRead() {
BSSL_CHECK(pthread_rwlock_unlock(&lock_) == 0);
}
void StaticMutex::LockWrite() {
BSSL_CHECK(pthread_rwlock_wrlock(&lock_) == 0);
}
void StaticMutex::UnlockWrite() {
BSSL_CHECK(pthread_rwlock_unlock(&lock_) == 0);
}
Mutex::~Mutex() { pthread_rwlock_destroy(&lock_); }
void CRYPTO_once(CRYPTO_once_t *once, void (*init)()) {
BSSL_CHECK(pthread_once(once, init) == 0);
}
static pthread_mutex_t g_destructors_lock = PTHREAD_MUTEX_INITIALIZER;
static thread_local_destructor_t g_destructors[NUM_OPENSSL_THREAD_LOCALS];
// thread_local_destructor is called when a thread exits. It releases thread
// local data for that thread only.
static void thread_local_destructor(void *arg) {
if (arg == nullptr) {
return;
}
thread_local_destructor_t destructors[NUM_OPENSSL_THREAD_LOCALS];
if (pthread_mutex_lock(&g_destructors_lock) != 0) {
return;
}
OPENSSL_memcpy(destructors, g_destructors, sizeof(destructors));
pthread_mutex_unlock(&g_destructors_lock);
unsigned i;
void **pointers = reinterpret_cast<void **>(arg);
for (i = 0; i < NUM_OPENSSL_THREAD_LOCALS; i++) {
if (destructors[i] != nullptr) {
destructors[i](pointers[i]);
}
}
free(pointers);
}
static pthread_once_t g_thread_local_init_once = PTHREAD_ONCE_INIT;
static pthread_key_t g_thread_local_key;
static int g_thread_local_key_created = 0;
static void thread_local_init() {
g_thread_local_key_created =
pthread_key_create(&g_thread_local_key, thread_local_destructor) == 0;
}
void *CRYPTO_get_thread_local(thread_local_data_t index) {
CRYPTO_once(&g_thread_local_init_once, thread_local_init);
if (!g_thread_local_key_created) {
return nullptr;
}
void **pointers =
reinterpret_cast<void **>(pthread_getspecific(g_thread_local_key));
if (pointers == nullptr) {
return nullptr;
}
return pointers[index];
}
int CRYPTO_set_thread_local(thread_local_data_t index, void *value,
thread_local_destructor_t destructor) {
CRYPTO_once(&g_thread_local_init_once, thread_local_init);
if (!g_thread_local_key_created) {
destructor(value);
return 0;
}
void **pointers =
reinterpret_cast<void **>(pthread_getspecific(g_thread_local_key));
if (pointers == nullptr) {
pointers = reinterpret_cast<void **>(
malloc(sizeof(void *) * NUM_OPENSSL_THREAD_LOCALS));
if (pointers == nullptr) {
destructor(value);
return 0;
}
OPENSSL_memset(pointers, 0, sizeof(void *) * NUM_OPENSSL_THREAD_LOCALS);
if (pthread_setspecific(g_thread_local_key, pointers) != 0) {
free(pointers);
destructor(value);
return 0;
}
}
if (pthread_mutex_lock(&g_destructors_lock) != 0) {
destructor(value);
return 0;
}
g_destructors[index] = destructor;
pthread_mutex_unlock(&g_destructors_lock);
pointers[index] = value;
return 1;
}
BSSL_NAMESPACE_END
#endif // OPENSSL_PTHREADS
|
