]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-126240: handle `NULL` returned by `_Py_asdl_expr_seq_new` (#126241)
authorBénédikt Tran <10796600+picnixz@users.noreply.github.com>
Thu, 31 Oct 2024 14:24:07 +0000 (15:24 +0100)
committerGitHub <noreply@github.com>
Thu, 31 Oct 2024 14:24:07 +0000 (17:24 +0300)
check return value of `_Py_asdl_expr_seq_new`

Parser/action_helpers.c

index cb21777f56618926ac8748cebb6599ffbf7d94b5..5ac1dd7813689cd71aa4ce2beb00b0230786de9a 100644 (file)
@@ -1128,6 +1128,9 @@ expr_ty _PyPegen_collect_call_seqs(Parser *p, asdl_expr_seq *a, asdl_seq *b,
     }
 
     asdl_expr_seq *args = _Py_asdl_expr_seq_new(total_len, arena);
+    if (args == NULL) {
+        return NULL;
+    }
 
     Py_ssize_t i = 0;
     for (i = 0; i < args_len; i++) {
@@ -1298,6 +1301,9 @@ unpack_top_level_joined_strs(Parser *p, asdl_expr_seq *raw_expressions)
     }
 
     asdl_expr_seq *expressions = _Py_asdl_expr_seq_new(req_size, p->arena);
+    if (expressions == NULL) {
+        return NULL;
+    }
 
     Py_ssize_t raw_index, req_index = 0;
     for (raw_index = 0; raw_index < raw_size; raw_index++) {
@@ -1490,6 +1496,9 @@ expr_ty _PyPegen_formatted_value(Parser *p, expr_ty expression, Token *debug, Re
         }
 
         asdl_expr_seq *values = _Py_asdl_expr_seq_new(2, arena);
+        if (values == NULL) {
+            return NULL;
+        }
         asdl_seq_SET(values, 0, debug_text);
         asdl_seq_SET(values, 1, formatted_value);
         return _PyAST_JoinedStr(values, lineno, col_offset, debug_end_line, debug_end_offset, p->arena);