1#include "pthread_impl.h"
2#include "fork_impl.h"
3
4volatile size_t __pthread_tsd_size = sizeof(void *) * PTHREAD_KEYS_MAX;
5void *__pthread_tsd_main[PTHREAD_KEYS_MAX] = { 0 };
6
7static void (*keys[PTHREAD_KEYS_MAX])(void *);
8
9static pthread_rwlock_t key_lock = PTHREAD_RWLOCK_INITIALIZER;
10
11static pthread_key_t next_key;
12
13static void nodtor(void *dummy)
14{
15}
16
17static void dummy_0(void)
18{
19}
20
21weak_alias(dummy_0, __tl_lock);
22weak_alias(dummy_0, __tl_unlock);
23
24#ifdef __wasilibc_unmodified_upstream // WASI lacks fork
25void __pthread_key_atfork(int who)
26{
27 if (who<0) __pthread_rwlock_rdlock(&key_lock);
28 else if (!who) __pthread_rwlock_unlock(&key_lock);
29 else key_lock = (pthread_rwlock_t)PTHREAD_RWLOCK_INITIALIZER;
30}
31#endif
32
33int __pthread_key_create(pthread_key_t *k, void (*dtor)(void *))
34{
35 pthread_t self = __pthread_self();
36
37 /* This can only happen in the main thread before
38 * pthread_create has been called. */
39 if (!self->tsd) self->tsd = __pthread_tsd_main;
40
41 /* Purely a sentinel value since null means slot is free. */
42 if (!dtor) dtor = nodtor;
43
44 __pthread_rwlock_wrlock(&key_lock);
45 pthread_key_t j = next_key;
46 do {
47 if (!keys[j]) {
48 keys[next_key = *k = j] = dtor;
49 __pthread_rwlock_unlock(&key_lock);
50 return 0;
51 }
52 } while ((j=(j+1)%PTHREAD_KEYS_MAX) != next_key);
53
54 __pthread_rwlock_unlock(&key_lock);
55 return EAGAIN;
56}
57
58int __pthread_key_delete(pthread_key_t k)
59{
60 sigset_t set;
61 pthread_t self = __pthread_self(), td=self;
62
63#ifdef __wasilibc_unmodified_upstream
64 __block_app_sigs(&set);
65#endif
66 __pthread_rwlock_wrlock(&key_lock);
67
68 __tl_lock();
69 do td->tsd[k] = 0;
70 while ((td=td->next)!=self);
71 __tl_unlock();
72
73 keys[k] = 0;
74
75 __pthread_rwlock_unlock(&key_lock);
76#ifdef __wasilibc_unmodified_upstream
77 __restore_sigs(&set);
78#endif
79
80 return 0;
81}
82
83void __pthread_tsd_run_dtors()
84{
85 pthread_t self = __pthread_self();
86 int i, j;
87 for (j=0; self->tsd_used && j<PTHREAD_DESTRUCTOR_ITERATIONS; j++) {
88 __pthread_rwlock_rdlock(&key_lock);
89 self->tsd_used = 0;
90 for (i=0; i<PTHREAD_KEYS_MAX; i++) {
91 void *val = self->tsd[i];
92 void (*dtor)(void *) = keys[i];
93 self->tsd[i] = 0;
94 if (val && dtor && dtor != nodtor) {
95 __pthread_rwlock_unlock(&key_lock);
96 dtor(val);
97 __pthread_rwlock_rdlock(&key_lock);
98 }
99 }
100 __pthread_rwlock_unlock(&key_lock);
101 }
102}
103
104weak_alias(__pthread_key_create, pthread_key_create);
105weak_alias(__pthread_key_delete, pthread_key_delete);