pythongh-115136: Fix possible NULL deref in getpath_joinpath() (pythonGH-115137)

dTenebrae · web-flow · commit 9e90313320a2 · 2024-02-08T08:40:38.000Z
Signed-off-by: Artem Chernyshev &lt;artem.chernyshev@red-soft.ru&gt;
diff --git a/Modules/getpath.c b/Modules/getpath.c
@@ -262,6 +262,10 @@ getpath_joinpath(PyObject *Py_UNUSED(self), PyObject *args)
     }
     /* Convert all parts to wchar and accumulate max final length */
     wchar_t **parts = (wchar_t **)PyMem_Malloc(n * sizeof(wchar_t *));
+    if (parts == NULL) {
+        PyErr_NoMemory();
+        return NULL;
+    }
     memset(parts, 0, n * sizeof(wchar_t *));
     Py_ssize_t cchFinal = 0;
     Py_ssize_t first = 0;
