bpo-32436: Fix potential NULL dereference (#5286)

8 years ago · 0bad4d63c6
1 changed files with 4 additions and 0 deletions
--- a/Python/hamt.c
+++ b/Python/hamt.c
@ -1055,6 +1055,10 @@ hamt_node_bitmap_without(PyHamtNode_Bitmap *self,
 #endif

                PyHamtNode_Bitmap *clone = hamt_node_bitmap_clone(self);
+                if (clone == NULL) {
+                    return W_ERROR;
+                }
+
                Py_SETREF(clone->b_array[val_idx],
                          (PyObject *)sub_node);  /* borrow */