bpo-36436: Fix _testcapi.pymem_buffer_overflow() (GH-12560)

Handle memory allocation failure.

bpo-36436: Fix _testcapi.pymem_buffer_overflow() (GH-12560)
Handle memory allocation failure.
414b1cde · Victor Stinner · GitHub · 871309c7 · 414b1cde · 414b1cde
Commit 414b1cde authored Mar 26, 2019 by Victor Stinner Committed by GitHub Mar 26, 2019
Showing with 5 additions and 0 deletions

Misc/NEWS.d/next/Tests/2019-03-26-13-49-21.bpo-36436.yAtN0V.rst ...EWS.d/next/Tests/2019-03-26-13-49-21.bpo-36436.yAtN0V.rst +1 -0

Modules/_testcapimodule.c Modules/_testcapimodule.c +4 -0

No files found.
--- a/Misc/NEWS.d/next/Tests/2019-03-26-13-49-21.bpo-36436.yAtN0V.rst
+++ b/Misc/NEWS.d/next/Tests/2019-03-26-13-49-21.bpo-36436.yAtN0V.rst
+Fix ``_testcapi.pymem_buffer_overflow()``: handle memory allocation failure.
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -4184,6 +4184,10 @@ pymem_buffer_overflow(PyObject *self, PyObject *args)
    /* Deliberate buffer overflow to check that PyMem_Free() detects
       the overflow when debug hooks are installed. */
    buffer = PyMem_Malloc(16);
+    if (buffer == NULL) {
+        PyErr_NoMemory();
+        return NULL;
+    }
    buffer[16] = 'x';
    PyMem_Free(buffer);