From ed3b838988ba656cdbb184fcb6589654c81ab276 Mon Sep 17 00:00:00 2001
From: Georg Brandl <georg@python.org>
Date: Mon, 12 Mar 2007 13:15:14 +0000
Subject: [PATCH] Check the keys of the locals dict -- they need not be a list.

---
 Objects/object.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/Objects/object.c b/Objects/object.c
index e2d1b134fb0..f4ae4f3dfb9 100644
--- a/Objects/object.c
+++ b/Objects/object.c
@@ -1349,6 +1349,7 @@ merge_class_dict(PyObject* dict, PyObject* aclass)
 static PyObject *
 _dir_locals()
 {
+	PyObject *names;
 	PyObject *locals = PyEval_GetLocals();
 
 	if (locals == NULL) {
@@ -1356,8 +1357,18 @@ _dir_locals()
 		return NULL;
 	}
 
+	names = PyMapping_Keys(locals);
+	if (!names)
+		return NULL;
+	if (!PyList_Check(names)) {
+		PyErr_Format(PyExc_TypeError,
+			"dir(): expected keys() of locals to be a list, "
+			"not '%.200s'", names->ob_type->tp_name);
+		Py_DECREF(names);
+		return NULL;
+	}
 	/* the locals don't need to be DECREF'd */
-	return PyMapping_Keys(locals);
+	return names;
 }
 
 /* Helper for PyObject_Dir of type objects: returns __dict__ and __bases__.