]> granicus.if.org Git - python/commitdiff
Issue 19158: a rare race in BoundedSemaphore could allow .release() too often.
authorTim Peters <tim@python.org>
Wed, 9 Oct 2013 01:55:51 +0000 (20:55 -0500)
committerTim Peters <tim@python.org>
Wed, 9 Oct 2013 01:55:51 +0000 (20:55 -0500)
Lib/test/test_threading.py
Lib/threading.py

index fef3314c16fed677cd98ded0bd2a72d73aa0ac09..0ebeb39cbdc660516ddf4235187c2f0980d9513d 100644 (file)
@@ -468,6 +468,24 @@ class ThreadTests(BaseTestCase):
                 self.assertEqual(0, status)
 
 
+    def test_BoundedSemaphore_limit(self):
+        # BoundedSemaphore should raise ValueError if released too often.
+        for limit in range(1, 10):
+            bs = threading.BoundedSemaphore(limit)
+            threads = [threading.Thread(target=bs.acquire)
+                       for _ in range(limit)]
+            for t in threads:
+                t.start()
+            for t in threads:
+                t.join()
+            threads = [threading.Thread(target=bs.release)
+                       for _ in range(limit)]
+            for t in threads:
+                t.start()
+            for t in threads:
+                t.join()
+            self.assertRaises(ValueError, bs.release)
+
 class ThreadJoinOnShutdown(BaseTestCase):
 
     # Between fork() and exec(), only async-safe functions are allowed (issues
index c98a0062314ad137976d7ac122a14f8301651208..3d4952b586308568b02939f95891eca38ba7b6ac 100644 (file)
@@ -283,9 +283,11 @@ class BoundedSemaphore(Semaphore):
         self._initial_value = value
 
     def release(self):
-        if self._value >= self._initial_value:
-            raise ValueError("Semaphore released too many times")
-        return Semaphore.release(self)
+        with self._cond:
+            if self._value >= self._initial_value:
+                raise ValueError("Semaphore released too many times")
+            self._value += 1
+            self._cond.notify()
 
 
 class Event: