[3.13] gh-126461: Fix _Unpickler_ReadFromFile() error handling (GH-12…

…6485) (#126495) gh-126461: Fix _Unpickler_ReadFromFile() error handling (GH-126485) Handle _Unpickler_SetStringInput() failure. (cherry picked from commit a1c57bc) Co-authored-by: Victor Stinner <[email protected]>
python · Nov 6, 2024 · 83827ad · 83827ad
1 parent 6e9da38
commit 83827ad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Modules/_pickle.c b/Modules/_pickle.c
@@ -1287,6 +1287,10 @@ _Unpickler_ReadFromFile(UnpicklerObject *self, Py_ssize_t n)
             else {
                 read_size = _Unpickler_SetStringInput(self, data);
                 Py_DECREF(data);
+                if (read_size < 0) {
+                    return -1;
+                }
+
                 self->prefetched_idx = 0;
                 if (n <= read_size)
                     return n;