Verify heappop argument is a list.

f5d615af · Raymond Hettinger · 45d9ebb5 · f5d615af
Commit f5d615af authored Nov 15, 2003 by Raymond Hettinger
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

Modules/heapqmodule.c Modules/heapqmodule.c +5 -0

No files found.
--- a/Modules/heapqmodule.c
+++ b/Modules/heapqmodule.c
@@ -119,6 +119,11 @@ heappop(PyObject *self, PyObject *heap)
 	PyObject *lastelt, *returnitem;
 	int n;

+	if (!PyList_Check(heap)) {
+		PyErr_SetString(PyExc_ValueError, "heap argument must be a list");
+		return NULL;
+	}
+
 	/* # raises appropriate IndexError if heap is empty */
 	n = PyList_GET_SIZE(heap);
 	if (n == 0) {