From 0a1b8cba90d467e85be87cd8b10b4d31caa8765a Mon Sep 17 00:00:00 2001 From: Victor Stinner Date: Sat, 16 Oct 2010 22:55:47 +0000 Subject: [PATCH] _Py_wrealpath() uses _Py_char2wchar() to decode the result, to support surrogate characters. --- Python/fileutils.c | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/Python/fileutils.c b/Python/fileutils.c index 147636f2c78..b8910b7be64 100644 --- a/Python/fileutils.c +++ b/Python/fileutils.c @@ -353,6 +353,7 @@ _Py_wrealpath(const wchar_t *path, { char *cpath; char cresolved_path[PATH_MAX]; + wchar_t *wresolved_path; char *res; size_t r; cpath = _Py_wchar2char(path); @@ -364,11 +365,20 @@ _Py_wrealpath(const wchar_t *path, PyMem_Free(cpath); if (res == NULL) return NULL; - r = mbstowcs(resolved_path, cresolved_path, resolved_path_size); - if (r == (size_t)-1 || r >= PATH_MAX) { + + wresolved_path = _Py_char2wchar(cresolved_path); + if (wresolved_path == NULL) { errno = EINVAL; return NULL; } + r = wcslen(wresolved_path); + if (resolved_path_size <= r) { + PyMem_Free(wresolved_path); + errno = EINVAL; + return NULL; + } + wcsncpy(resolved_path, wresolved_path, resolved_path_size); + PyMem_Free(wresolved_path); return resolved_path; } #endif