diff options
| author | Neal Norwitz <nnorwitz@gmail.com> | 2006-09-05 02:30:10 (GMT) |
|---|---|---|
| committer | Neal Norwitz <nnorwitz@gmail.com> | 2006-09-05 02:30:10 (GMT) |
| commit | 7ae5f294658caaf4de4afabbc779317cca0217da (patch) | |
| tree | cdfd534ecbc67e4f357d2ee10024301fac1f963c /Modules/itertoolsmodule.c | |
| parent | 29a5fdb7caf372144a085247d63440d1abb51b5c (diff) | |
| download | cpython-7ae5f294658caaf4de4afabbc779317cca0217da.zip cpython-7ae5f294658caaf4de4afabbc779317cca0217da.tar.gz cpython-7ae5f294658caaf4de4afabbc779317cca0217da.tar.bz2 | |
Backport fix for SF bug #1550714, itertools.tee raises SystemError
Diffstat (limited to 'Modules/itertoolsmodule.c')
| -rw-r--r-- | Modules/itertoolsmodule.c | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index d913890..a41f55b 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -618,11 +618,15 @@ static PyTypeObject tee_type = { static PyObject * tee(PyObject *self, PyObject *args) { - int i, n=2; + Py_ssize_t i, n=2; PyObject *it, *iterable, *copyable, *result; - if (!PyArg_ParseTuple(args, "O|i", &iterable, &n)) + if (!PyArg_ParseTuple(args, "O|n", &iterable, &n)) return NULL; + if (n < 0) { + PyErr_SetString(PyExc_ValueError, "n must be >= 0"); + return NULL; + } result = PyTuple_New(n); if (result == NULL) return NULL; |