Issue #874900: fix behaviour of threading module after a fork.

Reviewed by Gregory P. Smith.

2 files changed, 10 insertions, 5 deletions

diff --git a/Lib/test/test_threading.py b/Lib/test/test_threading.py
index 775e312..a7f2322 100644
--- a/Lib/test/test_threading.py
+++ b/Lib/test/test_threading.py
@@ -347,6 +347,9 @@ class ThreadJoinOnShutdown(unittest.TestCase):
             def joiningfunc(mainthread):
                 mainthread.join()
                 print('end of thread')
+                # stdout is fully buffered because not a tty, we have to flush
+                # before exit.
+                sys.stdout.flush()
         \n""" + script
 
         import subprocess
@@ -388,8 +391,7 @@ class ThreadJoinOnShutdown(unittest.TestCase):
             """
         self._run_and_join(script)
 
-    # XXX This test hangs!
-    def Xtest_3_join_in_forked_from_thread(self):
+    def test_3_join_in_forked_from_thread(self):
         # Like the test above, but fork() was called from a worker thread
         # In the forked process, the main Thread object must be marked as stopped.
         import os
diff --git a/Lib/threading.py b/Lib/threading.py
index 1d59cb0..b541d77 100644
--- a/Lib/threading.py
+++ b/Lib/threading.py
@@ -835,16 +835,19 @@ def _after_fork():
     new_active = {}
     current = current_thread()
     with _active_limbo_lock:
-        for ident, thread in _active.items():
+        for thread in _active.values():
             if thread is current:
-                # There is only one active thread.
+                # There is only one active thread. We reset the ident to
+                # its new value since it can have changed.
+                ident = _get_ident()
+                thread._ident = ident
                 new_active[ident] = thread
             else:
                 # All the others are already stopped.
                 # We don't call _Thread__stop() because it tries to acquire
                 # thread._Thread__block which could also have been held while
                 # we forked.
-                thread._Thread__stopped = True
+                thread._stopped = True
 
         _limbo.clear()
         _active.clear()