GH-141312: Allow only integers to longrangeiter_setstate state (GH-141317)

This fixes an assertion error when the new computed start is not an integer.
This commit is contained in:
Sergey Miryanov 2025-11-14 19:52:01 +05:00 committed by GitHub
parent 49e74210cb
commit 10bec7c1eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 17 additions and 0 deletions

View file

@ -1042,6 +1042,11 @@ longrangeiter_reduce(PyObject *op, PyObject *Py_UNUSED(ignored))
static PyObject *
longrangeiter_setstate(PyObject *op, PyObject *state)
{
if (!PyLong_CheckExact(state)) {
PyErr_Format(PyExc_TypeError, "state must be an int, not %T", state);
return NULL;
}
longrangeiterobject *r = (longrangeiterobject*)op;
PyObject *zero = _PyLong_GetZero(); // borrowed reference
int cmp;