From 111f60964ee438e64d82cb884f9dfdec10321b7c Mon Sep 17 00:00:00 2001 From: Tim Peters Date: Wed, 12 Sep 2001 07:54:51 +0000 Subject: If interning an instance of a string subclass, intern a real string object with the same value instead. This ensures that a string (or string subclass) object's ob_sinterned pointer is always a str (or NULL), and that the dict of interned strings only has strs as keys. --- Objects/stringobject.c | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) (limited to 'Objects/stringobject.c') diff --git a/Objects/stringobject.c b/Objects/stringobject.c index 3c03b9ee2a..99a16ed174 100644 --- a/Objects/stringobject.c +++ b/Objects/stringobject.c @@ -3553,10 +3553,26 @@ PyString_InternInPlace(PyObject **p) Py_DECREF(s); return; } - t = (PyObject *)s; - if (PyDict_SetItem(interned, t, t) == 0) { - s->ob_sinterned = t; - return; + /* Ensure that only true string objects appear in the intern dict, + and as the value of ob_sinterned. */ + if (PyString_CheckExact(s)) { + t = (PyObject *)s; + if (PyDict_SetItem(interned, t, t) == 0) { + s->ob_sinterned = t; + return; + } + } + else { + t = PyString_FromStringAndSize(PyString_AS_STRING(s), + PyString_GET_SIZE(s)); + if (t != NULL) { + if (PyDict_SetItem(interned, t, t) == 0) { + *p = s->ob_sinterned = t; + Py_DECREF(s); + return; + } + Py_DECREF(t); + } } PyErr_Clear(); } -- cgit v1.2.1