diff Modules/_threadmodule.c @ 56440:29908f2bc93f

Merged revisions 74524 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r74524 | gregory.p.smith | 2009-08-20 04:39:38 -0500 (Thu, 20 Aug 2009) | 2 lines Add weakref support to the thread.lock type. ........
author Benjamin Peterson <benjamin@python.org>
date Sat, 10 Oct 2009 01:16:07 +0000
parents 5458c20aa0cf
children 6cf84e5a6dd7
line wrap: on
line diff
--- a/Modules/_threadmodule.c	Sat Oct 10 00:13:32 2009 +0000
+++ b/Modules/_threadmodule.c	Sat Oct 10 01:16:07 2009 +0000
@@ -3,6 +3,7 @@
 /* Interface to Sjoerd's portable C thread library */
 
 #include "Python.h"
+#include "structmember.h" /* offsetof */
 
 #ifndef WITH_THREAD
 #error "Error!  The rest of Python is not compiled with thread support."
@@ -20,12 +21,15 @@
 typedef struct {
 	PyObject_HEAD
 	PyThread_type_lock lock_lock;
+	PyObject *in_weakreflist;
 } lockobject;
 
 static void
 lock_dealloc(lockobject *self)
 {
 	assert(self->lock_lock);
+	if (self->in_weakreflist != NULL)
+		PyObject_ClearWeakRefs((PyObject *) self);
 	/* Unlock the lock so it's safe to free it */
 	PyThread_acquire_lock(self->lock_lock, 0);
 	PyThread_release_lock(self->lock_lock);
@@ -145,7 +149,7 @@
 	0,				/*tp_traverse*/
 	0,				/*tp_clear*/
 	0,				/*tp_richcompare*/
-	0,				/*tp_weaklistoffset*/
+	offsetof(lockobject, in_weakreflist), /*tp_weaklistoffset*/
 	0,				/*tp_iter*/
 	0,				/*tp_iternext*/
 	lock_methods,			/*tp_methods*/
@@ -159,6 +163,7 @@
 	if (self == NULL)
 		return NULL;
 	self->lock_lock = PyThread_allocate_lock();
+	self->in_weakreflist = NULL;
 	if (self->lock_lock == NULL) {
 		PyObject_Del(self);
 		self = NULL;