commit 1a5856bf9295fa73995898d576e0bedf016aee1f
author Serhiy Storchaka <storchaka@gmail.com> Sat Apr 22 02:48:11 2017 +0300
committer Victor Stinner <victor.stinner@gmail.com> Sat Apr 22 01:48:11 2017 +0200
tree 32c1f6c2ad33c8374bc5753f520e609339962fd2
parent a36e939aeb3b5a2c56561eb24f0e339eee9f3f9d

bpo-29867: Add asserts in PyTuple_GET_SIZE, PyList_GET_SIZE and PySet_GET_SIZE. (#751)

Include/listobject.h

diff --git a/Include/listobject.h b/Include/listobject.h
index 31843b5..6057279 100644
--- a/Include/listobject.h
+++ b/Include/listobject.h
@@ -71,7 +71,7 @@
 #ifndef Py_LIMITED_API
 #define PyList_GET_ITEM(op, i) (((PyListObject *)(op))->ob_item[i])
 #define PyList_SET_ITEM(op, i, v) (((PyListObject *)(op))->ob_item[i] = (v))
-#define PyList_GET_SIZE(op)    Py_SIZE(op)
+#define PyList_GET_SIZE(op)    (assert(PyList_Check(op)),Py_SIZE(op))
 #define _PyList_ITEMS(op)      (((PyListObject *)(op))->ob_item)
 #endif
 

Include/setobject.h

diff --git a/Include/setobject.h b/Include/setobject.h
index 87ec1c8..fc0ea83 100644
--- a/Include/setobject.h
+++ b/Include/setobject.h
@@ -64,7 +64,7 @@
     PyObject *weakreflist;      /* List of weak references */
 } PySetObject;
 
-#define PySet_GET_SIZE(so) (((PySetObject *)(so))->used)
+#define PySet_GET_SIZE(so) (assert(PyAnySet_Check(so)),(((PySetObject *)(so))->used))
 
 PyAPI_DATA(PyObject *) _PySet_Dummy;
 

Include/tupleobject.h

diff --git a/Include/tupleobject.h b/Include/tupleobject.h
index c273ce7..98c2622 100644
--- a/Include/tupleobject.h
+++ b/Include/tupleobject.h
@@ -56,7 +56,7 @@
 /* Macro, trading safety for speed */
 #ifndef Py_LIMITED_API
 #define PyTuple_GET_ITEM(op, i) (((PyTupleObject *)(op))->ob_item[i])
-#define PyTuple_GET_SIZE(op)    Py_SIZE(op)
+#define PyTuple_GET_SIZE(op)    (assert(PyTuple_Check(op)),Py_SIZE(op))
 
 /* Macro, *only* to be used to fill in brand new tuples */
 #define PyTuple_SET_ITEM(op, i, v) (((PyTupleObject *)(op))->ob_item[i] = v)

Objects/odictobject.c

diff --git a/Objects/odictobject.c b/Objects/odictobject.c
index c2cef21..771dcc3 100644
--- a/Objects/odictobject.c
+++ b/Objects/odictobject.c
@@ -1519,7 +1519,7 @@
             count++;
         }
         if (count < PyList_GET_SIZE(pieces))
-            PyList_GET_SIZE(pieces) = count;
+            Py_SIZE(pieces) = count;
     }
     else {
         PyObject *items = _PyObject_CallMethodIdObjArgs((PyObject *)self,