gh-101892: Fix `SystemError` when a callable iterator call exhausts the iterator (GH-101896)

(cherry picked from commit 705487c655)

Co-authored-by: Raj <51259329+workingpayload@users.noreply.github.com>
Co-authored-by: Oleg Iarygin <oleg@arhadthedev.net>
This commit is contained in:
Miss Islington (bot) 2023-03-04 06:46:17 -08:00 committed by GitHub
parent 00791f23b7
commit 06a3bb8c94
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 30 additions and 2 deletions

View File

@ -348,6 +348,31 @@ def spam(state=[0]):
return i
self.check_iterator(iter(spam, 20), list(range(10)), pickle=False)
def test_iter_function_concealing_reentrant_exhaustion(self):
# gh-101892: Test two-argument iter() with a function that
# exhausts its associated iterator but forgets to either return
# a sentinel value or raise StopIteration.
HAS_MORE = 1
NO_MORE = 2
def exhaust(iterator):
"""Exhaust an iterator without raising StopIteration."""
list(iterator)
def spam():
# Touching the iterator with exhaust() below will call
# spam() once again so protect against recursion.
if spam.is_recursive_call:
return NO_MORE
spam.is_recursive_call = True
exhaust(spam.iterator)
return HAS_MORE
spam.is_recursive_call = False
spam.iterator = iter(spam, NO_MORE)
with self.assertRaises(StopIteration):
next(spam.iterator)
# Test exception propagation through function iterator
def test_exception_function(self):
def spam(state=[0]):

View File

@ -0,0 +1,3 @@
Callable iterators no longer raise :class:`SystemError` when the
callable object exhausts the iterator but forgets to either return a
sentinel value or raise :class:`StopIteration`.

View File

@ -222,7 +222,7 @@ calliter_iternext(calliterobject *it)
}
result = _PyObject_CallNoArgs(it->it_callable);
if (result != NULL) {
if (result != NULL && it->it_sentinel != NULL){
int ok;
ok = PyObject_RichCompareBool(it->it_sentinel, result, Py_EQ);
@ -230,7 +230,6 @@ calliter_iternext(calliterobject *it)
return result; /* Common case, fast path */
}
Py_DECREF(result);
if (ok > 0) {
Py_CLEAR(it->it_callable);
Py_CLEAR(it->it_sentinel);
@ -241,6 +240,7 @@ calliter_iternext(calliterobject *it)
Py_CLEAR(it->it_callable);
Py_CLEAR(it->it_sentinel);
}
Py_XDECREF(result);
return NULL;
}