summaryrefslogtreecommitdiffstats
path: root/Modules/itertoolsmodule.c
diff options
context:
space:
mode:
authorKristján Valur Jónsson <sweskman@gmail.com>2015-09-12 15:30:23 (GMT)
committerKristján Valur Jónsson <sweskman@gmail.com>2015-09-12 15:30:23 (GMT)
commit95c3e6cb2243acc0a102d9fa59efece1427a3936 (patch)
treeec383c38241ed66bbb1bc17dbd1b68bcb3306a79 /Modules/itertoolsmodule.c
parent0424eaf7533b0f386f6038dbb99f89e30fbbff76 (diff)
parent102764a1f6927955b9a907005ffd07216ebe1d96 (diff)
downloadcpython-95c3e6cb2243acc0a102d9fa59efece1427a3936.zip
cpython-95c3e6cb2243acc0a102d9fa59efece1427a3936.tar.gz
cpython-95c3e6cb2243acc0a102d9fa59efece1427a3936.tar.bz2
Issue #25021: Merge from 3.3 to 3.4
Diffstat (limited to 'Modules/itertoolsmodule.c')
-rw-r--r--Modules/itertoolsmodule.c12
1 files changed, 10 insertions, 2 deletions
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c
index f5fa3fb..4693fbc 100644
--- a/Modules/itertoolsmodule.c
+++ b/Modules/itertoolsmodule.c
@@ -2236,13 +2236,21 @@ product_setstate(productobject *lz, PyObject *state)
{
PyObject* indexObject = PyTuple_GET_ITEM(state, i);
Py_ssize_t index = PyLong_AsSsize_t(indexObject);
+ PyObject* pool;
+ Py_ssize_t poolsize;
if (index < 0 && PyErr_Occurred())
return NULL; /* not an integer */
+ pool = PyTuple_GET_ITEM(lz->pools, i);
+ poolsize = PyTuple_GET_SIZE(pool);
+ if (poolsize == 0) {
+ lz->stopped = 1;
+ Py_RETURN_NONE;
+ }
/* clamp the index */
if (index < 0)
index = 0;
- else if (index > n-1)
- index = n-1;
+ else if (index > poolsize-1)
+ index = poolsize-1;
lz->indices[i] = index;
}