mirror of
https://github.com/python/cpython.git
synced 2026-05-06 12:49:07 -04:00
GH-121012: Set index to -1 when list iterators become exhausted in tier 2 (GH-121483)
This commit is contained in:
@@ -299,6 +299,15 @@ class ListTest(list_tests.CommonTest):
|
||||
lst = [X(), X()]
|
||||
X() in lst
|
||||
|
||||
def test_tier2_invalidates_iterator(self):
|
||||
# GH-121012
|
||||
for _ in range(100):
|
||||
a = [1, 2, 3]
|
||||
it = iter(a)
|
||||
for _ in it:
|
||||
pass
|
||||
a.append(4)
|
||||
self.assertEqual(list(it), [])
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
@@ -0,0 +1,2 @@
|
||||
Tier 2 execution now ensures that list iterators remain exhausted, once they
|
||||
become exhausted.
|
||||
+4
-1
@@ -2967,7 +2967,10 @@ dummy_func(
|
||||
assert(Py_TYPE(iter_o) == &PyListIter_Type);
|
||||
PyListObject *seq = it->it_seq;
|
||||
EXIT_IF(seq == NULL);
|
||||
EXIT_IF((size_t)it->it_index >= (size_t)PyList_GET_SIZE(seq));
|
||||
if ((size_t)it->it_index >= (size_t)PyList_GET_SIZE(seq)) {
|
||||
it->it_index = -1;
|
||||
EXIT_IF(1);
|
||||
}
|
||||
}
|
||||
|
||||
op(_ITER_NEXT_LIST, (iter -- iter, next)) {
|
||||
|
||||
Generated
+5
-2
@@ -3055,8 +3055,11 @@
|
||||
JUMP_TO_JUMP_TARGET();
|
||||
}
|
||||
if ((size_t)it->it_index >= (size_t)PyList_GET_SIZE(seq)) {
|
||||
UOP_STAT_INC(uopcode, miss);
|
||||
JUMP_TO_JUMP_TARGET();
|
||||
it->it_index = -1;
|
||||
if (1) {
|
||||
UOP_STAT_INC(uopcode, miss);
|
||||
JUMP_TO_JUMP_TARGET();
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user