diff Python/thread.c @ 47664:db99f8752710 legacy-trunk

#1683 prevent forking from interfering in threading storage This should prevent some test_multiprocessing failures
author Benjamin Peterson <benjamin@python.org>
date Fri, 13 Jun 2008 00:09:47 +0000
parents 132d908471c8
children a8a8232cb541
line wrap: on
line diff
--- a/Python/thread.c	Thu Jun 12 22:33:06 2008 +0000
+++ b/Python/thread.c	Fri Jun 13 00:09:47 2008 +0000
@@ -381,4 +381,35 @@
 	PyThread_release_lock(keymutex);
 }
 
+/* Forget everything not associated with the current thread id.
+ * This function is called from PyOS_AfterFork().  It is necessary
+ * because other thread ids which were in use at the time of the fork
+ * may be reused for new threads created in the forked process.
+ */
+void
+PyThread_ReInitTLS(void)
+{
+	long id = PyThread_get_thread_ident();
+	struct key *p, **q;
+
+	if (!keymutex)
+		return;
+	
+	/* As with interpreter_lock in PyEval_ReInitThreads()
+	   we just create a new lock without freeing the old one */
+	keymutex = PyThread_allocate_lock();
+
+	/* Delete all keys which do not match the current thread id */
+	q = &keyhead;
+	while ((p = *q) != NULL) {
+		if (p->id != id) {
+			*q = p->next;
+			free((void *)p);
+			/* NB This does *not* free p->value! */
+		}
+		else
+			q = &p->next;
+	}
+}
+
 #endif /* Py_HAVE_NATIVE_TLS */