From a1c57bcfd2bcbc55ff858407e09c1d8d8cee44e6 Mon Sep 17 00:00:00 2001
From: Victor Stinner <vstinner@python.org>
Date: Wed, 6 Nov 2024 14:24:46 +0100
Subject: [PATCH] gh-126461: Fix _Unpickler_ReadFromFile() error handling
 (#126485)

Handle _Unpickler_SetStringInput() failure.
---
 Modules/_pickle.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Modules/_pickle.c b/Modules/_pickle.c
index b2bd9545c1b130..863da6878409f3 100644
--- a/Modules/_pickle.c
+++ b/Modules/_pickle.c
@@ -1288,6 +1288,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;