From: Tim Peters <tim.peters@gmail.com>
Date: Wed, 12 Sep 2001 07:54:51 +0000 (+0000)
Subject: If interning an instance of a string subclass, intern a real string object
X-Git-Tag: v2.2.1c1~1827
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=111f60964ee438e64d82cb884f9dfdec10321b7c;p=python

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.
---

diff --git a/Lib/test/test_descr.py b/Lib/test/test_descr.py
index 8d9e81e12e..b0299790ac 100644
--- a/Lib/test/test_descr.py
+++ b/Lib/test/test_descr.py
@@ -1529,6 +1529,20 @@ def inherits():
     verify(s.lower().__class__ is str)
     verify(s.lower() == base)
 
+    s = madstring("x y")
+    verify(intern(s).__class__ is str)
+    verify(intern(s) is intern("x y"))
+    verify(intern(s) == "x y")
+
+    i = intern("y x")
+    s = madstring("y x")
+    verify(intern(s).__class__ is str)
+    verify(intern(s) is i)
+
+    s = madstring(i)
+    verify(intern(s).__class__ is str)
+    verify(intern(s) is i)
+
     class madunicode(unicode):
         _rev = None
         def rev(self):
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();
 }