summaryrefslogtreecommitdiffstats
path: root/Python/fileutils.c
diff options
context:
space:
mode:
Diffstat (limited to 'Python/fileutils.c')
-rw-r--r--Python/fileutils.c9
1 files changed, 7 insertions, 2 deletions
diff --git a/Python/fileutils.c b/Python/fileutils.c
index 076f510..cfafd86 100644
--- a/Python/fileutils.c
+++ b/Python/fileutils.c
@@ -307,6 +307,7 @@ _Py_wreadlink(const wchar_t *path, wchar_t *buf, size_t bufsiz)
{
char *cpath;
char cbuf[PATH_MAX];
+ wchar_t *wbuf;
int res;
size_t r1;
@@ -324,11 +325,15 @@ _Py_wreadlink(const wchar_t *path, wchar_t *buf, size_t bufsiz)
return -1;
}
cbuf[res] = '\0'; /* buf will be null terminated */
- r1 = mbstowcs(buf, cbuf, bufsiz);
- if (r1 == -1) {
+ wbuf = _Py_char2wchar(cbuf);
+ r1 = wcslen(wbuf);
+ if (bufsiz <= r1) {
+ PyMem_Free(wbuf);
errno = EINVAL;
return -1;
}
+ wcsncpy(buf, wbuf, bufsiz);
+ PyMem_Free(wbuf);
return (int)r1;
}
#endif