Mercurial > cpython-lru_cache-weakref
diff Modules/threadmodule.c @ 64300:b2e8dcd15040 2.7
Merged revisions 84344 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/branches/py3k
........
r84344 | antoine.pitrou | 2010-08-28 20:17:03 +0200 (sam., 28 août 2010) | 4 lines
Issue #1868: Eliminate subtle timing issues in thread-local objects by
getting rid of the cached copy of thread-local attribute dictionary.
........
| author | Antoine Pitrou <solipsis@pitrou.net> |
|---|---|
| date | Sat, 28 Aug 2010 18:29:13 +0000 |
| parents | 756f8b398230 |
| children | b36cb4602e21 |
line wrap: on
line diff
--- a/Modules/threadmodule.c Thu Aug 26 06:20:13 2010 +0000 +++ b/Modules/threadmodule.c Sat Aug 28 18:29:13 2010 +0000 @@ -14,6 +14,7 @@ #include "pythread.h" static PyObject *ThreadError; +static PyObject *str_dict; static long nb_threads = 0; /* Lock objects */ @@ -264,8 +265,6 @@ PyObject *key; PyObject *args; PyObject *kw; - /* The current thread's local dict (necessary for tp_dictoffset) */ - PyObject *dict; PyObject *weakreflist; /* List of weak references to self */ /* A {localdummy weakref -> localdict} dict */ PyObject *dummies; @@ -277,9 +276,9 @@ static PyObject *_ldict(localobject *self); static PyObject *_localdummy_destroyed(PyObject *meth_self, PyObject *dummyweakref); -/* Create and register the dummy for the current thread, as well as the - corresponding local dict */ -static int +/* Create and register the dummy for the current thread. + Returns a borrowed reference of the corresponding local dict */ +static PyObject * _local_create_dummy(localobject *self) { PyObject *tdict, *ldict = NULL, *wr = NULL; @@ -315,15 +314,14 @@ goto err; Py_CLEAR(dummy); - Py_CLEAR(self->dict); - self->dict = ldict; - return 0; + Py_DECREF(ldict); + return ldict; err: Py_XDECREF(ldict); Py_XDECREF(wr); Py_XDECREF(dummy); - return -1; + return NULL; } static PyObject * @@ -369,7 +367,7 @@ if (self->wr_callback == NULL) goto err; - if (_local_create_dummy(self) < 0) + if (_local_create_dummy(self) == NULL) goto err; return (PyObject *)self; @@ -385,7 +383,6 @@ Py_VISIT(self->args); Py_VISIT(self->kw); Py_VISIT(self->dummies); - Py_VISIT(self->dict); return 0; } @@ -396,7 +393,6 @@ Py_CLEAR(self->args); Py_CLEAR(self->kw); Py_CLEAR(self->dummies); - Py_CLEAR(self->dict); Py_CLEAR(self->wr_callback); /* Remove all strong references to dummies from the thread states */ if (self->key @@ -442,9 +438,9 @@ dummy = PyDict_GetItem(tdict, self->key); if (dummy == NULL) { - if (_local_create_dummy(self) < 0) + ldict = _local_create_dummy(self); + if (ldict == NULL) return NULL; - ldict = self->dict; if (Py_TYPE(self)->tp_init != PyBaseObject_Type.tp_init && Py_TYPE(self)->tp_init((PyObject*)self, @@ -461,14 +457,6 @@ ldict = ((localdummyobject *) dummy)->localdict; } - /* The call to tp_init above may have caused another thread to run. - Install our ldict again. */ - if (self->dict != ldict) { - Py_INCREF(ldict); - Py_CLEAR(self->dict); - self->dict = ldict; - } - return ldict; } @@ -476,29 +464,25 @@ local_setattro(localobject *self, PyObject *name, PyObject *v) { PyObject *ldict; + int r; ldict = _ldict(self); if (ldict == NULL) return -1; - return PyObject_GenericSetAttr((PyObject *)self, name, v); -} + r = PyObject_RichCompareBool(name, str_dict, Py_EQ); + if (r == 1) { + PyErr_Format(PyExc_AttributeError, + "'%.50s' object attribute '__dict__' is read-only", + Py_TYPE(self)->tp_name); + return -1; + } + if (r == -1) + return -1; -static PyObject * -local_getdict(localobject *self, void *closure) -{ - PyObject *ldict; - ldict = _ldict(self); - Py_XINCREF(ldict); - return ldict; + return _PyObject_GenericSetAttrWithDict((PyObject *)self, name, v, ldict); } -static PyGetSetDef local_getset[] = { - {"__dict__", (getter)local_getdict, (setter)NULL, - "Local-data dictionary", NULL}, - {NULL} /* Sentinel */ -}; - static PyObject *local_getattro(localobject *, PyObject *); static PyTypeObject localtype = { @@ -532,12 +516,12 @@ /* tp_iternext */ 0, /* tp_methods */ 0, /* tp_members */ 0, - /* tp_getset */ local_getset, + /* tp_getset */ 0, /* tp_base */ 0, /* tp_dict */ 0, /* internal use */ /* tp_descr_get */ 0, /* tp_descr_set */ 0, - /* tp_dictoffset */ offsetof(localobject, dict), + /* tp_dictoffset */ 0, /* tp_init */ 0, /* tp_alloc */ 0, /* tp_new */ local_new, @@ -549,20 +533,29 @@ local_getattro(localobject *self, PyObject *name) { PyObject *ldict, *value; + int r; ldict = _ldict(self); if (ldict == NULL) return NULL; + r = PyObject_RichCompareBool(name, str_dict, Py_EQ); + if (r == 1) { + Py_INCREF(ldict); + return ldict; + } + if (r == -1) + return NULL; + if (Py_TYPE(self) != &localtype) /* use generic lookup for subtypes */ - return PyObject_GenericGetAttr((PyObject *)self, name); + return _PyObject_GenericGetAttrWithDict((PyObject *)self, name, ldict); /* Optimization: just look in dict ourselves */ value = PyDict_GetItem(ldict, name); if (value == NULL) /* Fall back on generic to get __class__ and __dict__ */ - return PyObject_GenericGetAttr((PyObject *)self, name); + return _PyObject_GenericGetAttrWithDict((PyObject *)self, name, ldict); Py_INCREF(value); return value; @@ -587,8 +580,6 @@ PyObject *ldict; ldict = PyDict_GetItem(self->dummies, dummyweakref); if (ldict != NULL) { - if (ldict == self->dict) - Py_CLEAR(self->dict); PyDict_DelItem(self->dummies, dummyweakref); } if (PyErr_Occurred()) @@ -931,6 +922,10 @@ nb_threads = 0; + str_dict = PyString_InternFromString("__dict__"); + if (str_dict == NULL) + return; + /* Initialize the C thread library */ PyThread_init_thread(); }
