Victor Stinner | 4a3fe08 | 2020-04-14 14:26:24 +0200 | [diff] [blame] | 1 | #include "pycore_interp.h" // _PyInterpreterState.pythread_stacksize |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 2 | |
Guido van Rossum | 6602099 | 1996-06-11 18:32:18 +0000 | [diff] [blame] | 3 | /* Posix threads interface */ |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 4 | |
Guido van Rossum | 6602099 | 1996-06-11 18:32:18 +0000 | [diff] [blame] | 5 | #include <stdlib.h> |
Guido van Rossum | 9e46e56 | 1998-10-07 16:39:47 +0000 | [diff] [blame] | 6 | #include <string.h> |
Martin v. Löwis | a7a76d3 | 2002-10-04 07:21:24 +0000 | [diff] [blame] | 7 | #if defined(__APPLE__) || defined(HAVE_PTHREAD_DESTRUCTOR) |
Jack Jansen | 7668957 | 2002-01-15 20:36:14 +0000 | [diff] [blame] | 8 | #define destructor xxdestructor |
| 9 | #endif |
Guido van Rossum | 6602099 | 1996-06-11 18:32:18 +0000 | [diff] [blame] | 10 | #include <pthread.h> |
Martin v. Löwis | a7a76d3 | 2002-10-04 07:21:24 +0000 | [diff] [blame] | 11 | #if defined(__APPLE__) || defined(HAVE_PTHREAD_DESTRUCTOR) |
Jack Jansen | 7668957 | 2002-01-15 20:36:14 +0000 | [diff] [blame] | 12 | #undef destructor |
| 13 | #endif |
Guido van Rossum | 8023099 | 2001-10-12 21:49:17 +0000 | [diff] [blame] | 14 | #include <signal.h> |
Martin v. Löwis | 42ab61e | 2002-03-17 17:19:00 +0000 | [diff] [blame] | 15 | |
Jake Tesler | b121f63 | 2019-05-22 08:43:17 -0700 | [diff] [blame] | 16 | #if defined(__linux__) |
| 17 | # include <sys/syscall.h> /* syscall(SYS_gettid) */ |
| 18 | #elif defined(__FreeBSD__) |
| 19 | # include <pthread_np.h> /* pthread_getthreadid_np() */ |
David Carlier | 0b9956e | 2019-06-03 16:43:33 +0100 | [diff] [blame] | 20 | #elif defined(__OpenBSD__) |
| 21 | # include <unistd.h> /* getthrid() */ |
Michael Felt | d0eeb93 | 2019-06-14 00:34:46 +0200 | [diff] [blame] | 22 | #elif defined(_AIX) |
| 23 | # include <sys/thread.h> /* thread_self() */ |
| 24 | #elif defined(__NetBSD__) |
| 25 | # include <lwp.h> /* _lwp_self() */ |
Jake Tesler | b121f63 | 2019-05-22 08:43:17 -0700 | [diff] [blame] | 26 | #endif |
| 27 | |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 28 | /* The POSIX spec requires that use of pthread_attr_setstacksize |
| 29 | be conditional on _POSIX_THREAD_ATTR_STACKSIZE being defined. */ |
| 30 | #ifdef _POSIX_THREAD_ATTR_STACKSIZE |
| 31 | #ifndef THREAD_STACK_SIZE |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 32 | #define THREAD_STACK_SIZE 0 /* use default stack size */ |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 33 | #endif |
Ned Deily | 9a7c524 | 2011-05-28 00:19:56 -0700 | [diff] [blame] | 34 | |
Ned Deily | 7ca97d5 | 2012-03-13 11:18:18 -0700 | [diff] [blame] | 35 | /* The default stack size for new threads on OSX and BSD is small enough that |
| 36 | * we'll get hard crashes instead of 'maximum recursion depth exceeded' |
| 37 | * exceptions. |
| 38 | * |
| 39 | * The default stack sizes below are the empirically determined minimal stack |
| 40 | * sizes where a simple recursive function doesn't cause a hard crash. |
| 41 | */ |
| 42 | #if defined(__APPLE__) && defined(THREAD_STACK_SIZE) && THREAD_STACK_SIZE == 0 |
| 43 | #undef THREAD_STACK_SIZE |
Ronald Oussoren | 1a057ba | 2019-08-01 07:43:07 +0200 | [diff] [blame] | 44 | /* Note: This matches the value of -Wl,-stack_size in configure.ac */ |
| 45 | #define THREAD_STACK_SIZE 0x1000000 |
Ned Deily | 7ca97d5 | 2012-03-13 11:18:18 -0700 | [diff] [blame] | 46 | #endif |
| 47 | #if defined(__FreeBSD__) && defined(THREAD_STACK_SIZE) && THREAD_STACK_SIZE == 0 |
Ned Deily | 9a7c524 | 2011-05-28 00:19:56 -0700 | [diff] [blame] | 48 | #undef THREAD_STACK_SIZE |
| 49 | #define THREAD_STACK_SIZE 0x400000 |
| 50 | #endif |
Michael Felt | 9670ce7 | 2019-08-03 08:12:26 +0200 | [diff] [blame] | 51 | #if defined(_AIX) && defined(THREAD_STACK_SIZE) && THREAD_STACK_SIZE == 0 |
| 52 | #undef THREAD_STACK_SIZE |
| 53 | #define THREAD_STACK_SIZE 0x200000 |
| 54 | #endif |
xdegaye | 00ada2c | 2019-12-08 08:40:14 +0100 | [diff] [blame] | 55 | /* bpo-38852: test_threading.test_recursion_limit() checks that 1000 recursive |
| 56 | Python calls (default recursion limit) doesn't crash, but raise a regular |
| 57 | RecursionError exception. In debug mode, Python function calls allocates |
| 58 | more memory on the stack, so use a stack of 8 MiB. */ |
| 59 | #if defined(__ANDROID__) && defined(THREAD_STACK_SIZE) && THREAD_STACK_SIZE == 0 |
| 60 | # ifdef Py_DEBUG |
| 61 | # undef THREAD_STACK_SIZE |
| 62 | # define THREAD_STACK_SIZE 0x800000 |
| 63 | # endif |
| 64 | #endif |
pxinwr | d5dcb65 | 2020-12-10 05:47:28 +0800 | [diff] [blame] | 65 | #if defined(__VXWORKS__) && defined(THREAD_STACK_SIZE) && THREAD_STACK_SIZE == 0 |
| 66 | #undef THREAD_STACK_SIZE |
| 67 | #define THREAD_STACK_SIZE 0x100000 |
| 68 | #endif |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 69 | /* for safety, ensure a viable minimum stacksize */ |
Victor Stinner | 8c663fd | 2017-11-08 14:44:44 -0800 | [diff] [blame] | 70 | #define THREAD_STACK_MIN 0x8000 /* 32 KiB */ |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 71 | #else /* !_POSIX_THREAD_ATTR_STACKSIZE */ |
| 72 | #ifdef THREAD_STACK_SIZE |
| 73 | #error "THREAD_STACK_SIZE defined but _POSIX_THREAD_ATTR_STACKSIZE undefined" |
| 74 | #endif |
| 75 | #endif |
| 76 | |
Martin v. Löwis | 42ab61e | 2002-03-17 17:19:00 +0000 | [diff] [blame] | 77 | /* The POSIX spec says that implementations supporting the sem_* |
| 78 | family of functions must indicate this by defining |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 79 | _POSIX_SEMAPHORES. */ |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 80 | #ifdef _POSIX_SEMAPHORES |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 81 | /* On FreeBSD 4.x, _POSIX_SEMAPHORES is defined empty, so |
Martin v. Löwis | 8b8fb3d | 2005-03-28 12:34:20 +0000 | [diff] [blame] | 82 | we need to add 0 to make it work there as well. */ |
| 83 | #if (_POSIX_SEMAPHORES+0) == -1 |
Anthony Baxter | 19b2369 | 2005-03-16 04:15:07 +0000 | [diff] [blame] | 84 | #define HAVE_BROKEN_POSIX_SEMAPHORES |
| 85 | #else |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 86 | #include <semaphore.h> |
| 87 | #include <errno.h> |
| 88 | #endif |
Anthony Baxter | 19b2369 | 2005-03-16 04:15:07 +0000 | [diff] [blame] | 89 | #endif |
Guido van Rossum | 6602099 | 1996-06-11 18:32:18 +0000 | [diff] [blame] | 90 | |
Guido van Rossum | d6353e2 | 1997-05-13 17:51:13 +0000 | [diff] [blame] | 91 | |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 92 | /* Whether or not to use semaphores directly rather than emulating them with |
| 93 | * mutexes and condition variables: |
| 94 | */ |
Antoine Pitrou | 19f8edc | 2010-10-10 08:37:22 +0000 | [diff] [blame] | 95 | #if (defined(_POSIX_SEMAPHORES) && !defined(HAVE_BROKEN_POSIX_SEMAPHORES) && \ |
| 96 | defined(HAVE_SEM_TIMEDWAIT)) |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 97 | # define USE_SEMAPHORES |
| 98 | #else |
| 99 | # undef USE_SEMAPHORES |
| 100 | #endif |
| 101 | |
| 102 | |
Guido van Rossum | 8023099 | 2001-10-12 21:49:17 +0000 | [diff] [blame] | 103 | /* On platforms that don't use standard POSIX threads pthread_sigmask() |
| 104 | * isn't present. DEC threads uses sigprocmask() instead as do most |
| 105 | * other UNIX International compliant systems that don't have the full |
| 106 | * pthread implementation. |
| 107 | */ |
Jason Tishler | fac083d | 2003-07-22 15:20:49 +0000 | [diff] [blame] | 108 | #if defined(HAVE_PTHREAD_SIGMASK) && !defined(HAVE_BROKEN_PTHREAD_SIGMASK) |
Guido van Rossum | 8023099 | 2001-10-12 21:49:17 +0000 | [diff] [blame] | 109 | # define SET_THREAD_SIGMASK pthread_sigmask |
| 110 | #else |
| 111 | # define SET_THREAD_SIGMASK sigprocmask |
| 112 | #endif |
| 113 | |
| 114 | |
Antoine Pitrou | 7c3e577 | 2010-04-14 15:44:10 +0000 | [diff] [blame] | 115 | #define MICROSECONDS_TO_TIMESPEC(microseconds, ts) \ |
| 116 | do { \ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 117 | struct timeval tv; \ |
Benjamin Peterson | f1c1903 | 2019-09-10 11:37:59 +0100 | [diff] [blame] | 118 | gettimeofday(&tv, NULL); \ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 119 | tv.tv_usec += microseconds % 1000000; \ |
| 120 | tv.tv_sec += microseconds / 1000000; \ |
| 121 | tv.tv_sec += tv.tv_usec / 1000000; \ |
| 122 | tv.tv_usec %= 1000000; \ |
| 123 | ts.tv_sec = tv.tv_sec; \ |
| 124 | ts.tv_nsec = tv.tv_usec * 1000; \ |
Antoine Pitrou | 7c3e577 | 2010-04-14 15:44:10 +0000 | [diff] [blame] | 125 | } while(0) |
| 126 | |
| 127 | |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 128 | /* |
| 129 | * pthread_cond support |
| 130 | */ |
| 131 | |
| 132 | #if defined(HAVE_PTHREAD_CONDATTR_SETCLOCK) && defined(HAVE_CLOCK_GETTIME) && defined(CLOCK_MONOTONIC) |
| 133 | // monotonic is supported statically. It doesn't mean it works on runtime. |
| 134 | #define CONDATTR_MONOTONIC |
| 135 | #endif |
| 136 | |
| 137 | // NULL when pthread_condattr_setclock(CLOCK_MONOTONIC) is not supported. |
| 138 | static pthread_condattr_t *condattr_monotonic = NULL; |
| 139 | |
| 140 | static void |
Benjamin Peterson | ea62a4b | 2020-07-15 06:12:05 -0700 | [diff] [blame] | 141 | init_condattr(void) |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 142 | { |
| 143 | #ifdef CONDATTR_MONOTONIC |
| 144 | static pthread_condattr_t ca; |
| 145 | pthread_condattr_init(&ca); |
| 146 | if (pthread_condattr_setclock(&ca, CLOCK_MONOTONIC) == 0) { |
| 147 | condattr_monotonic = &ca; // Use monotonic clock |
| 148 | } |
| 149 | #endif |
| 150 | } |
| 151 | |
| 152 | int |
| 153 | _PyThread_cond_init(PyCOND_T *cond) |
| 154 | { |
| 155 | return pthread_cond_init(cond, condattr_monotonic); |
| 156 | } |
| 157 | |
| 158 | void |
| 159 | _PyThread_cond_after(long long us, struct timespec *abs) |
| 160 | { |
| 161 | #ifdef CONDATTR_MONOTONIC |
| 162 | if (condattr_monotonic) { |
| 163 | clock_gettime(CLOCK_MONOTONIC, abs); |
| 164 | abs->tv_sec += us / 1000000; |
| 165 | abs->tv_nsec += (us % 1000000) * 1000; |
| 166 | abs->tv_sec += abs->tv_nsec / 1000000000; |
| 167 | abs->tv_nsec %= 1000000000; |
| 168 | return; |
| 169 | } |
| 170 | #endif |
| 171 | |
| 172 | struct timespec ts; |
| 173 | MICROSECONDS_TO_TIMESPEC(us, ts); |
| 174 | *abs = ts; |
| 175 | } |
| 176 | |
| 177 | |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 178 | /* A pthread mutex isn't sufficient to model the Python lock type |
| 179 | * because, according to Draft 5 of the docs (P1003.4a/D5), both of the |
| 180 | * following are undefined: |
| 181 | * -> a thread tries to lock a mutex it already has locked |
| 182 | * -> a thread tries to unlock a mutex locked by a different thread |
| 183 | * pthread mutexes are designed for serializing threads over short pieces |
| 184 | * of code anyway, so wouldn't be an appropriate implementation of |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 185 | * Python's locks regardless. |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 186 | * |
| 187 | * The pthread_lock struct implements a Python lock as a "locked?" bit |
| 188 | * and a <condition, mutex> pair. In general, if the bit can be acquired |
| 189 | * instantly, it is, else the pair is used to block the thread until the |
| 190 | * bit is cleared. 9 May 1994 tim@ksr.com |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 191 | */ |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 192 | |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 193 | typedef struct { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 194 | char locked; /* 0=unlocked, 1=locked */ |
| 195 | /* a <cond, mutex> pair to handle an acquire of a locked lock */ |
| 196 | pthread_cond_t lock_released; |
| 197 | pthread_mutex_t mut; |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 198 | } pthread_lock; |
| 199 | |
Guido van Rossum | 9e46e56 | 1998-10-07 16:39:47 +0000 | [diff] [blame] | 200 | #define CHECK_STATUS(name) if (status != 0) { perror(name); error = 1; } |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 201 | #define CHECK_STATUS_PTHREAD(name) if (status != 0) { fprintf(stderr, \ |
| 202 | "%s: %s\n", name, strerror(status)); error = 1; } |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 203 | |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 204 | /* |
| 205 | * Initialization. |
| 206 | */ |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 207 | static void |
| 208 | PyThread__init_thread(void) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 209 | { |
Guido van Rossum | d21744a | 1998-09-10 03:04:40 +0000 | [diff] [blame] | 210 | #if defined(_AIX) && defined(__GNUC__) |
Antoine Pitrou | 9a00e0a | 2013-06-18 22:17:48 +0200 | [diff] [blame] | 211 | extern void pthread_init(void); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 212 | pthread_init(); |
Guido van Rossum | d21744a | 1998-09-10 03:04:40 +0000 | [diff] [blame] | 213 | #endif |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 214 | init_condattr(); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 215 | } |
| 216 | |
| 217 | /* |
| 218 | * Thread support. |
| 219 | */ |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 220 | |
Siddhesh Poyarekar | 9eea6ea | 2018-11-30 20:44:25 +0530 | [diff] [blame] | 221 | /* bpo-33015: pythread_callback struct and pythread_wrapper() cast |
| 222 | "void func(void *)" to "void* func(void *)": always return NULL. |
| 223 | |
| 224 | PyThread_start_new_thread() uses "void func(void *)" type, whereas |
| 225 | pthread_create() requires a void* return value. */ |
| 226 | typedef struct { |
| 227 | void (*func) (void *); |
| 228 | void *arg; |
| 229 | } pythread_callback; |
| 230 | |
| 231 | static void * |
| 232 | pythread_wrapper(void *arg) |
| 233 | { |
| 234 | /* copy func and func_arg and free the temporary structure */ |
| 235 | pythread_callback *callback = arg; |
| 236 | void (*func)(void *) = callback->func; |
| 237 | void *func_arg = callback->arg; |
| 238 | PyMem_RawFree(arg); |
| 239 | |
| 240 | func(func_arg); |
| 241 | return NULL; |
| 242 | } |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 243 | |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 244 | unsigned long |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 245 | PyThread_start_new_thread(void (*func)(void *), void *arg) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 246 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 247 | pthread_t th; |
| 248 | int status; |
Guido van Rossum | d0b69ec | 2001-09-10 14:10:54 +0000 | [diff] [blame] | 249 | #if defined(THREAD_STACK_SIZE) || defined(PTHREAD_SYSTEM_SCHED_SUPPORTED) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 250 | pthread_attr_t attrs; |
Jack Jansen | c51395d | 2001-08-29 15:24:53 +0000 | [diff] [blame] | 251 | #endif |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 252 | #if defined(THREAD_STACK_SIZE) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 253 | size_t tss; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 254 | #endif |
| 255 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 256 | dprintf(("PyThread_start_new_thread called\n")); |
| 257 | if (!initialized) |
| 258 | PyThread_init_thread(); |
Guido van Rossum | d6353e2 | 1997-05-13 17:51:13 +0000 | [diff] [blame] | 259 | |
Guido van Rossum | d0b69ec | 2001-09-10 14:10:54 +0000 | [diff] [blame] | 260 | #if defined(THREAD_STACK_SIZE) || defined(PTHREAD_SYSTEM_SCHED_SUPPORTED) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 261 | if (pthread_attr_init(&attrs) != 0) |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 262 | return PYTHREAD_INVALID_THREAD_ID; |
Guido van Rossum | d0b69ec | 2001-09-10 14:10:54 +0000 | [diff] [blame] | 263 | #endif |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 264 | #if defined(THREAD_STACK_SIZE) |
Victor Stinner | 50b4857 | 2018-11-01 01:51:40 +0100 | [diff] [blame] | 265 | PyThreadState *tstate = _PyThreadState_GET(); |
Eric Snow | 2ebc5ce | 2017-09-07 23:51:28 -0600 | [diff] [blame] | 266 | size_t stacksize = tstate ? tstate->interp->pythread_stacksize : 0; |
| 267 | tss = (stacksize != 0) ? stacksize : THREAD_STACK_SIZE; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 268 | if (tss != 0) { |
| 269 | if (pthread_attr_setstacksize(&attrs, tss) != 0) { |
| 270 | pthread_attr_destroy(&attrs); |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 271 | return PYTHREAD_INVALID_THREAD_ID; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 272 | } |
| 273 | } |
Jack Jansen | c51395d | 2001-08-29 15:24:53 +0000 | [diff] [blame] | 274 | #endif |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 275 | #if defined(PTHREAD_SYSTEM_SCHED_SUPPORTED) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 276 | pthread_attr_setscope(&attrs, PTHREAD_SCOPE_SYSTEM); |
Guido van Rossum | d0b69ec | 2001-09-10 14:10:54 +0000 | [diff] [blame] | 277 | #endif |
Guido van Rossum | 8023099 | 2001-10-12 21:49:17 +0000 | [diff] [blame] | 278 | |
Siddhesh Poyarekar | 9eea6ea | 2018-11-30 20:44:25 +0530 | [diff] [blame] | 279 | pythread_callback *callback = PyMem_RawMalloc(sizeof(pythread_callback)); |
| 280 | |
| 281 | if (callback == NULL) { |
| 282 | return PYTHREAD_INVALID_THREAD_ID; |
| 283 | } |
| 284 | |
| 285 | callback->func = func; |
| 286 | callback->arg = arg; |
| 287 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 288 | status = pthread_create(&th, |
Guido van Rossum | d0b69ec | 2001-09-10 14:10:54 +0000 | [diff] [blame] | 289 | #if defined(THREAD_STACK_SIZE) || defined(PTHREAD_SYSTEM_SCHED_SUPPORTED) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 290 | &attrs, |
Jack Jansen | c51395d | 2001-08-29 15:24:53 +0000 | [diff] [blame] | 291 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 292 | (pthread_attr_t*)NULL, |
Jack Jansen | c51395d | 2001-08-29 15:24:53 +0000 | [diff] [blame] | 293 | #endif |
Siddhesh Poyarekar | 9eea6ea | 2018-11-30 20:44:25 +0530 | [diff] [blame] | 294 | pythread_wrapper, callback); |
Guido van Rossum | 8023099 | 2001-10-12 21:49:17 +0000 | [diff] [blame] | 295 | |
Fred Drake | 03459a5 | 2001-11-09 16:00:41 +0000 | [diff] [blame] | 296 | #if defined(THREAD_STACK_SIZE) || defined(PTHREAD_SYSTEM_SCHED_SUPPORTED) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 297 | pthread_attr_destroy(&attrs); |
Jack Jansen | c51395d | 2001-08-29 15:24:53 +0000 | [diff] [blame] | 298 | #endif |
Siddhesh Poyarekar | 9eea6ea | 2018-11-30 20:44:25 +0530 | [diff] [blame] | 299 | |
| 300 | if (status != 0) { |
| 301 | PyMem_RawFree(callback); |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 302 | return PYTHREAD_INVALID_THREAD_ID; |
Siddhesh Poyarekar | 9eea6ea | 2018-11-30 20:44:25 +0530 | [diff] [blame] | 303 | } |
Martin v. Löwis | 910ae62 | 2003-04-19 07:44:52 +0000 | [diff] [blame] | 304 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 305 | pthread_detach(th); |
Martin v. Löwis | 910ae62 | 2003-04-19 07:44:52 +0000 | [diff] [blame] | 306 | |
Guido van Rossum | 3c28863 | 2001-10-16 21:13:49 +0000 | [diff] [blame] | 307 | #if SIZEOF_PTHREAD_T <= SIZEOF_LONG |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 308 | return (unsigned long) th; |
Guido van Rossum | 3c28863 | 2001-10-16 21:13:49 +0000 | [diff] [blame] | 309 | #else |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 310 | return (unsigned long) *(unsigned long *) &th; |
Guido van Rossum | 3c28863 | 2001-10-16 21:13:49 +0000 | [diff] [blame] | 311 | #endif |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 312 | } |
| 313 | |
Trent Mick | 635f6fb | 2000-08-23 21:33:05 +0000 | [diff] [blame] | 314 | /* XXX This implementation is considered (to quote Tim Peters) "inherently |
| 315 | hosed" because: |
Skip Montanaro | 6babcc2 | 2004-03-03 08:42:23 +0000 | [diff] [blame] | 316 | - It does not guarantee the promise that a non-zero integer is returned. |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 317 | - The cast to unsigned long is inherently unsafe. |
Jesus Cea | 736e7fc | 2011-03-14 17:36:54 +0100 | [diff] [blame] | 318 | - It is not clear that the 'volatile' (for AIX?) are any longer necessary. |
Trent Mick | 635f6fb | 2000-08-23 21:33:05 +0000 | [diff] [blame] | 319 | */ |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 320 | unsigned long |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 321 | PyThread_get_thread_ident(void) |
Guido van Rossum | e944da8 | 1994-05-23 12:43:41 +0000 | [diff] [blame] | 322 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 323 | volatile pthread_t threadid; |
| 324 | if (!initialized) |
| 325 | PyThread_init_thread(); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 326 | threadid = pthread_self(); |
Serhiy Storchaka | aefa7eb | 2017-03-23 15:48:39 +0200 | [diff] [blame] | 327 | return (unsigned long) threadid; |
Guido van Rossum | e944da8 | 1994-05-23 12:43:41 +0000 | [diff] [blame] | 328 | } |
| 329 | |
Jake Tesler | b121f63 | 2019-05-22 08:43:17 -0700 | [diff] [blame] | 330 | #ifdef PY_HAVE_THREAD_NATIVE_ID |
| 331 | unsigned long |
| 332 | PyThread_get_thread_native_id(void) |
| 333 | { |
| 334 | if (!initialized) |
| 335 | PyThread_init_thread(); |
| 336 | #ifdef __APPLE__ |
| 337 | uint64_t native_id; |
| 338 | (void) pthread_threadid_np(NULL, &native_id); |
| 339 | #elif defined(__linux__) |
| 340 | pid_t native_id; |
| 341 | native_id = syscall(SYS_gettid); |
| 342 | #elif defined(__FreeBSD__) |
| 343 | int native_id; |
| 344 | native_id = pthread_getthreadid_np(); |
David Carlier | 0b9956e | 2019-06-03 16:43:33 +0100 | [diff] [blame] | 345 | #elif defined(__OpenBSD__) |
| 346 | pid_t native_id; |
| 347 | native_id = getthrid(); |
Michael Felt | d0eeb93 | 2019-06-14 00:34:46 +0200 | [diff] [blame] | 348 | #elif defined(_AIX) |
| 349 | tid_t native_id; |
| 350 | native_id = thread_self(); |
David Carlier | 5287022 | 2019-06-12 15:37:56 +0000 | [diff] [blame] | 351 | #elif defined(__NetBSD__) |
| 352 | lwpid_t native_id; |
| 353 | native_id = _lwp_self(); |
Jake Tesler | b121f63 | 2019-05-22 08:43:17 -0700 | [diff] [blame] | 354 | #endif |
| 355 | return (unsigned long) native_id; |
| 356 | } |
| 357 | #endif |
| 358 | |
Victor Stinner | c664b34 | 2019-05-04 11:48:05 -0400 | [diff] [blame] | 359 | void _Py_NO_RETURN |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 360 | PyThread_exit_thread(void) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 361 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 362 | dprintf(("PyThread_exit_thread called\n")); |
Antoine Pitrou | 0d5e52d | 2011-05-04 20:02:30 +0200 | [diff] [blame] | 363 | if (!initialized) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 364 | exit(0); |
Antoine Pitrou | 0d5e52d | 2011-05-04 20:02:30 +0200 | [diff] [blame] | 365 | pthread_exit(0); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 366 | } |
| 367 | |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 368 | #ifdef USE_SEMAPHORES |
| 369 | |
| 370 | /* |
| 371 | * Lock support. |
| 372 | */ |
| 373 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 374 | PyThread_type_lock |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 375 | PyThread_allocate_lock(void) |
| 376 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 377 | sem_t *lock; |
| 378 | int status, error = 0; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 379 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 380 | dprintf(("PyThread_allocate_lock called\n")); |
| 381 | if (!initialized) |
| 382 | PyThread_init_thread(); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 383 | |
Victor Stinner | 80aa565 | 2013-07-07 17:17:59 +0200 | [diff] [blame] | 384 | lock = (sem_t *)PyMem_RawMalloc(sizeof(sem_t)); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 385 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 386 | if (lock) { |
| 387 | status = sem_init(lock,0,1); |
| 388 | CHECK_STATUS("sem_init"); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 389 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 390 | if (error) { |
Victor Stinner | 80aa565 | 2013-07-07 17:17:59 +0200 | [diff] [blame] | 391 | PyMem_RawFree((void *)lock); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 392 | lock = NULL; |
| 393 | } |
| 394 | } |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 395 | |
Zackery Spytz | 1a2252e | 2019-05-06 10:56:51 -0600 | [diff] [blame] | 396 | dprintf(("PyThread_allocate_lock() -> %p\n", (void *)lock)); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 397 | return (PyThread_type_lock)lock; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 398 | } |
| 399 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 400 | void |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 401 | PyThread_free_lock(PyThread_type_lock lock) |
| 402 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 403 | sem_t *thelock = (sem_t *)lock; |
| 404 | int status, error = 0; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 405 | |
Christian Heimes | 56379c0 | 2012-12-02 08:37:00 +0100 | [diff] [blame] | 406 | (void) error; /* silence unused-but-set-variable warning */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 407 | dprintf(("PyThread_free_lock(%p) called\n", lock)); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 408 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 409 | if (!thelock) |
| 410 | return; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 411 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 412 | status = sem_destroy(thelock); |
| 413 | CHECK_STATUS("sem_destroy"); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 414 | |
Victor Stinner | 80aa565 | 2013-07-07 17:17:59 +0200 | [diff] [blame] | 415 | PyMem_RawFree((void *)thelock); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 416 | } |
| 417 | |
| 418 | /* |
| 419 | * As of February 2002, Cygwin thread implementations mistakenly report error |
| 420 | * codes in the return value of the sem_ calls (like the pthread_ functions). |
| 421 | * Correct implementations return -1 and put the code in errno. This supports |
| 422 | * either. |
| 423 | */ |
| 424 | static int |
| 425 | fix_status(int status) |
| 426 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 427 | return (status == -1) ? errno : status; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 428 | } |
| 429 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 430 | PyLockStatus |
| 431 | PyThread_acquire_lock_timed(PyThread_type_lock lock, PY_TIMEOUT_T microseconds, |
| 432 | int intr_flag) |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 433 | { |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 434 | PyLockStatus success; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 435 | sem_t *thelock = (sem_t *)lock; |
| 436 | int status, error = 0; |
| 437 | struct timespec ts; |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 438 | _PyTime_t deadline = 0; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 439 | |
Christian Heimes | 56379c0 | 2012-12-02 08:37:00 +0100 | [diff] [blame] | 440 | (void) error; /* silence unused-but-set-variable warning */ |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 441 | dprintf(("PyThread_acquire_lock_timed(%p, %lld, %d) called\n", |
| 442 | lock, microseconds, intr_flag)); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 443 | |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 444 | if (microseconds > PY_TIMEOUT_MAX) { |
| 445 | Py_FatalError("Timeout larger than PY_TIMEOUT_MAX"); |
| 446 | } |
| 447 | |
| 448 | if (microseconds > 0) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 449 | MICROSECONDS_TO_TIMESPEC(microseconds, ts); |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 450 | |
| 451 | if (!intr_flag) { |
| 452 | /* cannot overflow thanks to (microseconds > PY_TIMEOUT_MAX) |
| 453 | check done above */ |
| 454 | _PyTime_t timeout = _PyTime_FromNanoseconds(microseconds * 1000); |
| 455 | deadline = _PyTime_GetMonotonicClock() + timeout; |
| 456 | } |
| 457 | } |
| 458 | |
| 459 | while (1) { |
| 460 | if (microseconds > 0) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 461 | status = fix_status(sem_timedwait(thelock, &ts)); |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 462 | } |
| 463 | else if (microseconds == 0) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 464 | status = fix_status(sem_trywait(thelock)); |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 465 | } |
| 466 | else { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 467 | status = fix_status(sem_wait(thelock)); |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 468 | } |
| 469 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 470 | /* Retry if interrupted by a signal, unless the caller wants to be |
| 471 | notified. */ |
Victor Stinner | 850a18e | 2017-10-24 16:53:32 -0700 | [diff] [blame] | 472 | if (intr_flag || status != EINTR) { |
| 473 | break; |
| 474 | } |
| 475 | |
| 476 | if (microseconds > 0) { |
| 477 | /* wait interrupted by a signal (EINTR): recompute the timeout */ |
| 478 | _PyTime_t dt = deadline - _PyTime_GetMonotonicClock(); |
| 479 | if (dt < 0) { |
| 480 | status = ETIMEDOUT; |
| 481 | break; |
| 482 | } |
| 483 | else if (dt > 0) { |
| 484 | _PyTime_t realtime_deadline = _PyTime_GetSystemClock() + dt; |
| 485 | if (_PyTime_AsTimespec(realtime_deadline, &ts) < 0) { |
| 486 | /* Cannot occur thanks to (microseconds > PY_TIMEOUT_MAX) |
| 487 | check done above */ |
| 488 | Py_UNREACHABLE(); |
| 489 | } |
| 490 | /* no need to update microseconds value, the code only care |
| 491 | if (microseconds > 0 or (microseconds == 0). */ |
| 492 | } |
| 493 | else { |
| 494 | microseconds = 0; |
| 495 | } |
| 496 | } |
| 497 | } |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 498 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 499 | /* Don't check the status if we're stopping because of an interrupt. */ |
| 500 | if (!(intr_flag && status == EINTR)) { |
| 501 | if (microseconds > 0) { |
| 502 | if (status != ETIMEDOUT) |
| 503 | CHECK_STATUS("sem_timedwait"); |
| 504 | } |
| 505 | else if (microseconds == 0) { |
| 506 | if (status != EAGAIN) |
| 507 | CHECK_STATUS("sem_trywait"); |
| 508 | } |
| 509 | else { |
| 510 | CHECK_STATUS("sem_wait"); |
| 511 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 512 | } |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 513 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 514 | if (status == 0) { |
| 515 | success = PY_LOCK_ACQUIRED; |
| 516 | } else if (intr_flag && status == EINTR) { |
| 517 | success = PY_LOCK_INTR; |
| 518 | } else { |
| 519 | success = PY_LOCK_FAILURE; |
| 520 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 521 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 522 | dprintf(("PyThread_acquire_lock_timed(%p, %lld, %d) -> %d\n", |
| 523 | lock, microseconds, intr_flag, success)); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 524 | return success; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 525 | } |
| 526 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 527 | void |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 528 | PyThread_release_lock(PyThread_type_lock lock) |
| 529 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 530 | sem_t *thelock = (sem_t *)lock; |
| 531 | int status, error = 0; |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 532 | |
Christian Heimes | 56379c0 | 2012-12-02 08:37:00 +0100 | [diff] [blame] | 533 | (void) error; /* silence unused-but-set-variable warning */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 534 | dprintf(("PyThread_release_lock(%p) called\n", lock)); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 535 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 536 | status = sem_post(thelock); |
| 537 | CHECK_STATUS("sem_post"); |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 538 | } |
| 539 | |
| 540 | #else /* USE_SEMAPHORES */ |
| 541 | |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 542 | /* |
| 543 | * Lock support. |
| 544 | */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 545 | PyThread_type_lock |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 546 | PyThread_allocate_lock(void) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 547 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 548 | pthread_lock *lock; |
| 549 | int status, error = 0; |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 550 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 551 | dprintf(("PyThread_allocate_lock called\n")); |
| 552 | if (!initialized) |
| 553 | PyThread_init_thread(); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 554 | |
Andy Lester | 7668a8b | 2020-03-24 23:26:44 -0500 | [diff] [blame] | 555 | lock = (pthread_lock *) PyMem_RawCalloc(1, sizeof(pthread_lock)); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 556 | if (lock) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 557 | lock->locked = 0; |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 558 | |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 559 | status = pthread_mutex_init(&lock->mut, NULL); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 560 | CHECK_STATUS_PTHREAD("pthread_mutex_init"); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 561 | /* Mark the pthread mutex underlying a Python mutex as |
| 562 | pure happens-before. We can't simply mark the |
| 563 | Python-level mutex as a mutex because it can be |
| 564 | acquired and released in different threads, which |
| 565 | will cause errors. */ |
| 566 | _Py_ANNOTATE_PURE_HAPPENS_BEFORE_MUTEX(&lock->mut); |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 567 | |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 568 | status = _PyThread_cond_init(&lock->lock_released); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 569 | CHECK_STATUS_PTHREAD("pthread_cond_init"); |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 570 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 571 | if (error) { |
Victor Stinner | 80aa565 | 2013-07-07 17:17:59 +0200 | [diff] [blame] | 572 | PyMem_RawFree((void *)lock); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 573 | lock = 0; |
| 574 | } |
| 575 | } |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 576 | |
Zackery Spytz | 1a2252e | 2019-05-06 10:56:51 -0600 | [diff] [blame] | 577 | dprintf(("PyThread_allocate_lock() -> %p\n", (void *)lock)); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 578 | return (PyThread_type_lock) lock; |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 579 | } |
| 580 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 581 | void |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 582 | PyThread_free_lock(PyThread_type_lock lock) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 583 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 584 | pthread_lock *thelock = (pthread_lock *)lock; |
| 585 | int status, error = 0; |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 586 | |
Antoine Pitrou | 9a00e0a | 2013-06-18 22:17:48 +0200 | [diff] [blame] | 587 | (void) error; /* silence unused-but-set-variable warning */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 588 | dprintf(("PyThread_free_lock(%p) called\n", lock)); |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 589 | |
Kristján Valur Jónsson | 187aa54 | 2012-06-05 22:17:42 +0000 | [diff] [blame] | 590 | /* some pthread-like implementations tie the mutex to the cond |
| 591 | * and must have the cond destroyed first. |
| 592 | */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 593 | status = pthread_cond_destroy( &thelock->lock_released ); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 594 | CHECK_STATUS_PTHREAD("pthread_cond_destroy"); |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 595 | |
Kristján Valur Jónsson | 187aa54 | 2012-06-05 22:17:42 +0000 | [diff] [blame] | 596 | status = pthread_mutex_destroy( &thelock->mut ); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 597 | CHECK_STATUS_PTHREAD("pthread_mutex_destroy"); |
Kristján Valur Jónsson | 187aa54 | 2012-06-05 22:17:42 +0000 | [diff] [blame] | 598 | |
Victor Stinner | 80aa565 | 2013-07-07 17:17:59 +0200 | [diff] [blame] | 599 | PyMem_RawFree((void *)thelock); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 600 | } |
| 601 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 602 | PyLockStatus |
| 603 | PyThread_acquire_lock_timed(PyThread_type_lock lock, PY_TIMEOUT_T microseconds, |
| 604 | int intr_flag) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 605 | { |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 606 | PyLockStatus success = PY_LOCK_FAILURE; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 607 | pthread_lock *thelock = (pthread_lock *)lock; |
| 608 | int status, error = 0; |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 609 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 610 | dprintf(("PyThread_acquire_lock_timed(%p, %lld, %d) called\n", |
| 611 | lock, microseconds, intr_flag)); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 612 | |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 613 | if (microseconds == 0) { |
| 614 | status = pthread_mutex_trylock( &thelock->mut ); |
| 615 | if (status != EBUSY) |
| 616 | CHECK_STATUS_PTHREAD("pthread_mutex_trylock[1]"); |
| 617 | } |
| 618 | else { |
| 619 | status = pthread_mutex_lock( &thelock->mut ); |
| 620 | CHECK_STATUS_PTHREAD("pthread_mutex_lock[1]"); |
| 621 | } |
| 622 | if (status == 0) { |
| 623 | if (thelock->locked == 0) { |
| 624 | success = PY_LOCK_ACQUIRED; |
| 625 | } |
| 626 | else if (microseconds != 0) { |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 627 | struct timespec abs; |
| 628 | if (microseconds > 0) { |
| 629 | _PyThread_cond_after(microseconds, &abs); |
| 630 | } |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 631 | /* continue trying until we get the lock */ |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 632 | |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 633 | /* mut must be locked by me -- part of the condition |
| 634 | * protocol */ |
| 635 | while (success == PY_LOCK_FAILURE) { |
| 636 | if (microseconds > 0) { |
| 637 | status = pthread_cond_timedwait( |
| 638 | &thelock->lock_released, |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 639 | &thelock->mut, &abs); |
| 640 | if (status == 1) { |
| 641 | break; |
| 642 | } |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 643 | if (status == ETIMEDOUT) |
| 644 | break; |
Inada Naoki | 001fee1 | 2019-02-20 10:00:09 +0900 | [diff] [blame] | 645 | CHECK_STATUS_PTHREAD("pthread_cond_timedwait"); |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 646 | } |
| 647 | else { |
| 648 | status = pthread_cond_wait( |
| 649 | &thelock->lock_released, |
| 650 | &thelock->mut); |
| 651 | CHECK_STATUS_PTHREAD("pthread_cond_wait"); |
| 652 | } |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 653 | |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 654 | if (intr_flag && status == 0 && thelock->locked) { |
| 655 | /* We were woken up, but didn't get the lock. We probably received |
| 656 | * a signal. Return PY_LOCK_INTR to allow the caller to handle |
| 657 | * it and retry. */ |
| 658 | success = PY_LOCK_INTR; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 659 | break; |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 660 | } |
| 661 | else if (status == 0 && !thelock->locked) { |
| 662 | success = PY_LOCK_ACQUIRED; |
| 663 | } |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 664 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 665 | } |
Antoine Pitrou | f84ac42 | 2017-06-26 20:41:07 +0200 | [diff] [blame] | 666 | if (success == PY_LOCK_ACQUIRED) thelock->locked = 1; |
| 667 | status = pthread_mutex_unlock( &thelock->mut ); |
| 668 | CHECK_STATUS_PTHREAD("pthread_mutex_unlock[1]"); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 669 | } |
Martin v. Löwis | 1509a15 | 2003-04-18 11:11:09 +0000 | [diff] [blame] | 670 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 671 | if (error) success = PY_LOCK_FAILURE; |
| 672 | dprintf(("PyThread_acquire_lock_timed(%p, %lld, %d) -> %d\n", |
| 673 | lock, microseconds, intr_flag, success)); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 674 | return success; |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 675 | } |
| 676 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 677 | void |
Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 678 | PyThread_release_lock(PyThread_type_lock lock) |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 679 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 680 | pthread_lock *thelock = (pthread_lock *)lock; |
| 681 | int status, error = 0; |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 682 | |
Antoine Pitrou | 9a00e0a | 2013-06-18 22:17:48 +0200 | [diff] [blame] | 683 | (void) error; /* silence unused-but-set-variable warning */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 684 | dprintf(("PyThread_release_lock(%p) called\n", lock)); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 685 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 686 | status = pthread_mutex_lock( &thelock->mut ); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 687 | CHECK_STATUS_PTHREAD("pthread_mutex_lock[3]"); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 688 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 689 | thelock->locked = 0; |
Guido van Rossum | b98b1b3 | 1994-05-11 08:42:04 +0000 | [diff] [blame] | 690 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 691 | /* wake up someone (anyone, if any) waiting on the lock */ |
| 692 | status = pthread_cond_signal( &thelock->lock_released ); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 693 | CHECK_STATUS_PTHREAD("pthread_cond_signal"); |
Kristján Valur Jónsson | 187aa54 | 2012-06-05 22:17:42 +0000 | [diff] [blame] | 694 | |
| 695 | status = pthread_mutex_unlock( &thelock->mut ); |
Daniel Birnstiel | d7fa6b2 | 2017-03-21 14:06:06 +0100 | [diff] [blame] | 696 | CHECK_STATUS_PTHREAD("pthread_mutex_unlock[3]"); |
Guido van Rossum | 2c8cb9f | 1994-05-09 15:12:46 +0000 | [diff] [blame] | 697 | } |
Martin v. Löwis | cc89866 | 2002-03-17 09:53:51 +0000 | [diff] [blame] | 698 | |
| 699 | #endif /* USE_SEMAPHORES */ |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 700 | |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 701 | int |
Victor Stinner | 87255be | 2020-04-07 23:11:49 +0200 | [diff] [blame] | 702 | _PyThread_at_fork_reinit(PyThread_type_lock *lock) |
| 703 | { |
| 704 | PyThread_type_lock new_lock = PyThread_allocate_lock(); |
| 705 | if (new_lock == NULL) { |
| 706 | return -1; |
| 707 | } |
| 708 | |
| 709 | /* bpo-6721, bpo-40089: The old lock can be in an inconsistent state. |
| 710 | fork() can be called in the middle of an operation on the lock done by |
| 711 | another thread. So don't call PyThread_free_lock(*lock). |
| 712 | |
| 713 | Leak memory on purpose. Don't release the memory either since the |
| 714 | address of a mutex is relevant. Putting two mutexes at the same address |
| 715 | can lead to problems. */ |
| 716 | |
| 717 | *lock = new_lock; |
| 718 | return 0; |
| 719 | } |
| 720 | |
| 721 | int |
Antoine Pitrou | 810023d | 2010-12-15 22:59:16 +0000 | [diff] [blame] | 722 | PyThread_acquire_lock(PyThread_type_lock lock, int waitflag) |
| 723 | { |
| 724 | return PyThread_acquire_lock_timed(lock, waitflag ? -1 : 0, /*intr_flag=*/0); |
| 725 | } |
| 726 | |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 727 | /* set the thread stack size. |
| 728 | * Return 0 if size is valid, -1 if size is invalid, |
| 729 | * -2 if setting stack size is not supported. |
| 730 | */ |
| 731 | static int |
| 732 | _pythread_pthread_set_stacksize(size_t size) |
| 733 | { |
| 734 | #if defined(THREAD_STACK_SIZE) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 735 | pthread_attr_t attrs; |
| 736 | size_t tss_min; |
| 737 | int rc = 0; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 738 | #endif |
| 739 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 740 | /* set to default */ |
| 741 | if (size == 0) { |
Victor Stinner | 81a7be3 | 2020-04-14 15:14:01 +0200 | [diff] [blame] | 742 | _PyInterpreterState_GET()->pythread_stacksize = 0; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 743 | return 0; |
| 744 | } |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 745 | |
| 746 | #if defined(THREAD_STACK_SIZE) |
| 747 | #if defined(PTHREAD_STACK_MIN) |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 748 | tss_min = PTHREAD_STACK_MIN > THREAD_STACK_MIN ? PTHREAD_STACK_MIN |
| 749 | : THREAD_STACK_MIN; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 750 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 751 | tss_min = THREAD_STACK_MIN; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 752 | #endif |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 753 | if (size >= tss_min) { |
| 754 | /* validate stack size by setting thread attribute */ |
| 755 | if (pthread_attr_init(&attrs) == 0) { |
| 756 | rc = pthread_attr_setstacksize(&attrs, size); |
| 757 | pthread_attr_destroy(&attrs); |
| 758 | if (rc == 0) { |
Victor Stinner | 81a7be3 | 2020-04-14 15:14:01 +0200 | [diff] [blame] | 759 | _PyInterpreterState_GET()->pythread_stacksize = size; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 760 | return 0; |
| 761 | } |
| 762 | } |
| 763 | } |
| 764 | return -1; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 765 | #else |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 766 | return -2; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 767 | #endif |
| 768 | } |
| 769 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 770 | #define THREAD_SET_STACKSIZE(x) _pythread_pthread_set_stacksize(x) |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 771 | |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 772 | |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 773 | /* Thread Local Storage (TLS) API |
| 774 | |
| 775 | This API is DEPRECATED since Python 3.7. See PEP 539 for details. |
| 776 | */ |
| 777 | |
| 778 | /* Issue #25658: On platforms where native TLS key is defined in a way that |
| 779 | cannot be safely cast to int, PyThread_create_key returns immediately a |
| 780 | failure status and other TLS functions all are no-ops. This indicates |
| 781 | clearly that the old API is not supported on platforms where it cannot be |
| 782 | used reliably, and that no effort will be made to add such support. |
| 783 | |
| 784 | Note: PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT will be unnecessary after |
| 785 | removing this API. |
| 786 | */ |
| 787 | |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 788 | int |
| 789 | PyThread_create_key(void) |
| 790 | { |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 791 | #ifdef PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 792 | pthread_key_t key; |
| 793 | int fail = pthread_key_create(&key, NULL); |
Victor Stinner | daca3d7 | 2014-08-17 22:11:06 +0200 | [diff] [blame] | 794 | if (fail) |
| 795 | return -1; |
| 796 | if (key > INT_MAX) { |
| 797 | /* Issue #22206: handle integer overflow */ |
| 798 | pthread_key_delete(key); |
| 799 | errno = ENOMEM; |
| 800 | return -1; |
| 801 | } |
| 802 | return (int)key; |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 803 | #else |
| 804 | return -1; /* never return valid key value. */ |
| 805 | #endif |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 806 | } |
| 807 | |
| 808 | void |
| 809 | PyThread_delete_key(int key) |
| 810 | { |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 811 | #ifdef PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 812 | pthread_key_delete(key); |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 813 | #endif |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 814 | } |
| 815 | |
| 816 | void |
| 817 | PyThread_delete_key_value(int key) |
| 818 | { |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 819 | #ifdef PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 820 | pthread_setspecific(key, NULL); |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 821 | #endif |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 822 | } |
| 823 | |
| 824 | int |
| 825 | PyThread_set_key_value(int key, void *value) |
| 826 | { |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 827 | #ifdef PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT |
| 828 | int fail = pthread_setspecific(key, value); |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 829 | return fail ? -1 : 0; |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 830 | #else |
| 831 | return -1; |
| 832 | #endif |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 833 | } |
| 834 | |
| 835 | void * |
| 836 | PyThread_get_key_value(int key) |
| 837 | { |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 838 | #ifdef PTHREAD_KEY_T_IS_COMPATIBLE_WITH_INT |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 839 | return pthread_getspecific(key); |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 840 | #else |
| 841 | return NULL; |
| 842 | #endif |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 843 | } |
| 844 | |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 845 | |
Kristján Valur Jónsson | 2fea9b9 | 2010-09-20 02:11:49 +0000 | [diff] [blame] | 846 | void |
| 847 | PyThread_ReInitTLS(void) |
Masayuki Yamamoto | 731e189 | 2017-10-06 19:41:34 +0900 | [diff] [blame] | 848 | { |
| 849 | } |
| 850 | |
| 851 | |
| 852 | /* Thread Specific Storage (TSS) API |
| 853 | |
| 854 | Platform-specific components of TSS API implementation. |
| 855 | */ |
| 856 | |
| 857 | int |
| 858 | PyThread_tss_create(Py_tss_t *key) |
| 859 | { |
| 860 | assert(key != NULL); |
| 861 | /* If the key has been created, function is silently skipped. */ |
| 862 | if (key->_is_initialized) { |
| 863 | return 0; |
| 864 | } |
| 865 | |
| 866 | int fail = pthread_key_create(&(key->_key), NULL); |
| 867 | if (fail) { |
| 868 | return -1; |
| 869 | } |
| 870 | key->_is_initialized = 1; |
| 871 | return 0; |
| 872 | } |
| 873 | |
| 874 | void |
| 875 | PyThread_tss_delete(Py_tss_t *key) |
| 876 | { |
| 877 | assert(key != NULL); |
| 878 | /* If the key has not been created, function is silently skipped. */ |
| 879 | if (!key->_is_initialized) { |
| 880 | return; |
| 881 | } |
| 882 | |
| 883 | pthread_key_delete(key->_key); |
| 884 | /* pthread has not provided the defined invalid value for the key. */ |
| 885 | key->_is_initialized = 0; |
| 886 | } |
| 887 | |
| 888 | int |
| 889 | PyThread_tss_set(Py_tss_t *key, void *value) |
| 890 | { |
| 891 | assert(key != NULL); |
| 892 | int fail = pthread_setspecific(key->_key, value); |
| 893 | return fail ? -1 : 0; |
| 894 | } |
| 895 | |
| 896 | void * |
| 897 | PyThread_tss_get(Py_tss_t *key) |
| 898 | { |
| 899 | assert(key != NULL); |
| 900 | return pthread_getspecific(key->_key); |
| 901 | } |