commit be487a65f18e1be5fde03e2977fff4be53cc2fbf
author Serhiy Storchaka <storchaka@gmail.com> Mon Mar 06 21:21:41 2017 +0200
committer GitHub <noreply@github.com> Mon Mar 06 21:21:41 2017 +0200
tree 680cef96de35559eb9b476fb9ac0d80b63f4ee1b
parent d908fd9ee1c307f7066023eb2031c0f509036cbc

bpo-15954: Check return code of wcsxfrm(). (#508)

Modules/_localemodule.c

diff --git a/Modules/_localemodule.c b/Modules/_localemodule.c
index feb3802..f5c126a 100644
--- a/Modules/_localemodule.c
+++ b/Modules/_localemodule.c
@@ -260,7 +260,12 @@
         PyErr_NoMemory();
         goto exit;
     }
+    errno = 0;
     n2 = wcsxfrm(buf, s, n1);
+    if (errno) {
+        PyErr_SetFromErrno(PyExc_OSError);
+        goto exit;
+    }
     if (n2 >= (size_t)n1) {
         /* more space needed */
         wchar_t * new_buf = PyMem_Realloc(buf, (n2+1)*sizeof(wchar_t));
@@ -269,14 +274,17 @@
             goto exit;
         }
         buf = new_buf;
+        errno = 0;
         n2 = wcsxfrm(buf, s, n2+1);
+        if (errno) {
+            PyErr_SetFromErrno(PyExc_OSError);
+            goto exit;
+        }
     }
     result = PyUnicode_FromWideChar(buf, n2);
 exit:
-    if (buf)
-        PyMem_Free(buf);
-    if (s)
-        PyMem_Free(s);
+    PyMem_Free(buf);
+    PyMem_Free(s);
     return result;
 }
 #endif