GET_ITER may leave the iterable on the stack

python · markshannon · May 27, 2025 · Apr 10, 2025 · Apr 11, 2025 · Apr 15, 2025
commit 6d1b93e760b1312c2b938f89c87066be6195b303
diff --git a/Lib/test/test_list.py b/Lib/test/test_list.py
@@ -365,5 +365,19 @@ def lappend(l, x, y):
         rc, _, _ = assert_python_ok("-c", code)
         self.assertEqual(rc, 0)
 
+    def test_list_overwrite_local(self):
+        """Test that overwriting the last reference to the
+           iterable doesn't prematurely free the iterable"""
+
+        def foo(x):
+            r = 0
+            for i in x:
+                r += i
+                x = None
+            return r
+
+        self.assertEqual(foo(list(range(10))), 45)
+
+
 if __name__ == "__main__":
     unittest.main()
@@ -2871,6 +2871,7 @@ optimize_load_fast(cfg_builder *g)
 
                 // Opcodes that consume no inputs
                 case GET_ANEXT:
+                case GET_ITER:
                 case GET_LEN:
                 case IMPORT_FROM:
                 case MATCH_KEYS: