_Py_wreadlink(): catch _Py_char2wchar() failure

350147b5 · Victor Stinner · 3f711f4a · 350147b5
Commit 350147b5 authored Oct 16, 2010 by Victor Stinner
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

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

No files found.
--- a/Python/fileutils.c
+++ b/Python/fileutils.c
@@ -326,6 +326,10 @@ _Py_wreadlink(const wchar_t *path, wchar_t *buf, size_t bufsiz)
    }
    cbuf[res] = '\0'; /* buf will be null terminated */
    wbuf = _Py_char2wchar(cbuf);
+    if (wbuf == NULL) {
+        errno = EINVAL;
+        return -1;
+    }
    r1 = wcslen(wbuf);
    if (bufsiz <= r1) {
        PyMem_Free(wbuf);