gh-74953: _PyThread_cond_after() uses _PyTime_t

vstinner · vstinner · commit bcc9696314d9 · 2022-06-21T14:41:08.000+02:00
pthread _PyThread_cond_after() implementation now uses the _PyTime_t
type to handle properly overflow: clamp to the maximum value.
diff --git a/Python/condvar.h b/Python/condvar.h
@@ -68,9 +68,9 @@ void _PyThread_cond_after(long long us, struct timespec *abs);
 Py_LOCAL_INLINE(int)
 PyCOND_TIMEDWAIT(PyCOND_T *cond, PyMUTEX_T *mut, long long us)
 {
-    struct timespec abs;
-    _PyThread_cond_after(us, &abs);
-    int ret = pthread_cond_timedwait(cond, mut, &abs);
+    struct timespec abs_timeout;
+    _PyThread_cond_after(us, &abs_timeout);
+    int ret = pthread_cond_timedwait(cond, mut, &abs_timeout);
     if (ret == ETIMEDOUT) {
         return 1;
     }
diff --git a/Python/thread_pthread.h b/Python/thread_pthread.h
@@ -156,23 +156,23 @@ _PyThread_cond_init(PyCOND_T *cond)
     return pthread_cond_init(cond, condattr_monotonic);
 }
 
+
 void
 _PyThread_cond_after(long long us, struct timespec *abs)
 {
+    _PyTime_t timeout = _PyTime_FromMicrosecondsClamp(us);
+    _PyTime_t t;
 #ifdef CONDATTR_MONOTONIC
     if (condattr_monotonic) {
-        clock_gettime(CLOCK_MONOTONIC, abs);
-        abs->tv_sec  += us / 1000000;
-        abs->tv_nsec += (us % 1000000) * 1000;
-        abs->tv_sec  += abs->tv_nsec / 1000000000;
-        abs->tv_nsec %= 1000000000;
-        return;
+        t = _PyTime_GetMonotonicClock();
     }
+    else
 #endif
-
-    struct timespec ts;
-    MICROSECONDS_TO_TIMESPEC(us, ts);
-    *abs = ts;
+    {
+        t = _PyTime_GetSystemClock();
+    }
+    t = _PyTime_Add(t, timeout);
+    _PyTime_AsTimespec_clamp(t, abs);
 }
 
 
@@ -639,17 +639,17 @@ PyThread_acquire_lock_timed(PyThread_type_lock lock, PY_TIMEOUT_T microseconds,
         goto unlock;
     }
 
-    struct timespec abs;
+    struct timespec abs_timeout;
     if (microseconds > 0) {
-        _PyThread_cond_after(microseconds, &abs);
+        _PyThread_cond_after(microseconds, &abs_timeout);
     }
     // Continue trying until we get the lock
 
     // mut must be locked by me -- part of the condition protocol
     while (1) {
         if (microseconds > 0) {
             status = pthread_cond_timedwait(&thelock->lock_released,
-                                            &thelock->mut, &abs);
+                                            &thelock->mut, &abs_timeout);
             if (status == 1) {
                 break;
             }

Original file line number	Diff line number	Diff line change
`@@ -68,9 +68,9 @@ void _PyThread_cond_after(long long us, struct timespec *abs);`
`68`	`68`	`Py_LOCAL_INLINE(int)`
`69`	`69`	`PyCOND_TIMEDWAIT(PyCOND_T cond, PyMUTEX_T mut, long long us)`
`70`	`70`	`{`
`71`		`- struct timespec abs;`
`72`		`- _PyThread_cond_after(us, &abs);`
`73`		`- int ret = pthread_cond_timedwait(cond, mut, &abs);`
	`71`	`+ struct timespec abs_timeout;`
	`72`	`+ _PyThread_cond_after(us, &abs_timeout);`
	`73`	`+ int ret = pthread_cond_timedwait(cond, mut, &abs_timeout);`
`74`	`74`	`if (ret == ETIMEDOUT) {`
`75`	`75`	`return 1;`
`76`	`76`	`}`