[3.13] gh-146093: Fix csv _set_str(): check if PyUnicode_DecodeASCII() failed (GH...

gh-146093: Fix csv _set_str(): check if PyUnicode_DecodeASCII() failed (GH-146113)

The function can fail on a memory allocation failure.

Bug reported by devdanzin.
(cherry picked from commit 724c7c8146f44a7c737ec4588a1ee4b9db994f6f)

Co-authored-by: Victor Stinner <vstinner@python.org>

diff --git a/Modules/_csv.c b/Modules/_csv.c
index aea071cc1433289484f13efdc76cfc2c74e35b86..0b7ff2575fcfec621b92bbec7e686dba39bc81b3 100644 (file)
--- a/Modules/_csv.c
+++ b/Modules/_csv.c
@@ -302,8 +302,12 @@ _set_char(const char *name, Py_UCS4 *target, PyObject *src, Py_UCS4 dflt)
 static int
 _set_str(const char *name, PyObject **target, PyObject *src, const char *dflt)
 {
-    if (src == NULL)
+    if (src == NULL) {
         *target = PyUnicode_DecodeASCII(dflt, strlen(dflt), NULL);
+        if (*target == NULL) {
+            return -1;
+        }
+    }
     else {
         if (src == Py_None)
             *target = NULL;