gh-116480: Fix _PySemaphore_Wait call during thread deletion

colesbury · colesbury · commit e4aa866ff637 · 2024-03-07T23:27:23.000Z
In general, when `_PyThreadState_GET()` is non-NULL then the current
thread is "attached", but there is a small window during
`PyThreadState_DeleteCurrent()` where that's not true:
tstate_delete_common is called when the thread is detached, but before
current_fast_clear().

This updates _PySemaphore_Wait() to handle that case.
diff --git a/Python/parking_lot.c b/Python/parking_lot.c
@@ -194,14 +194,17 @@ _PySemaphore_Wait(_PySemaphore *sema, PyTime_t timeout, int detach)
     PyThreadState *tstate = NULL;
     if (detach) {
         tstate = _PyThreadState_GET();
-        if (tstate) {
-            PyEval_ReleaseThread(tstate);
+        // Only detach if we are attached
+        if (tstate && tstate->state != _Py_THREAD_ATTACHED) {
+            tstate = NULL;
         }
     }
 
+    if (tstate) {
+        PyEval_ReleaseThread(tstate);
+    }
     int res = _PySemaphore_PlatformWait(sema, timeout);
-
-    if (detach && tstate) {
+    if (tstate) {
         PyEval_AcquireThread(tstate);
     }
     return res;
