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

0bad4d63 · Yury Selivanov · GitHub · 9d411c11 · 0bad4d63
Commit 0bad4d63 authored Jan 23, 2018 by Yury Selivanov Committed by GitHub Jan 23, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

Python/hamt.c Python/hamt.c +4 -0

No files found.
--- 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 */