mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Issue #16416: Fix error handling in _Py_wchar2char() _Py_char2wchar() functions

Browse source
This commit is contained in:
Victor Stinner 2012-11-12 23:32:21 +01:00
parent fc93ec5966
commit 0d92c4f667
2 changed files with 21 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

9
Objects/unicodeobject.c
View file

@ -4691,7 +4691,10 @@ onError:
#ifdef __APPLE__ #ifdef __APPLE__
/* Simplified UTF-8 decoder using surrogateescape error handler, /* Simplified UTF-8 decoder using surrogateescape error handler,
used to decode the command line arguments on Mac OS X. */ used to decode the command line arguments on Mac OS X.
Return a pointer to a newly allocated wide character string (use
PyMem_Free() to free the memory), or NULL on memory allocation error. */
wchar_t* wchar_t*
_Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size) _Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
@ -4702,10 +4705,8 @@ _Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
/* Note: size will always be longer than the resulting Unicode /* Note: size will always be longer than the resulting Unicode
character count */ character count */
if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1)) { if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1))
PyErr_NoMemory();
return NULL; return NULL;
}
unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t)); unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t));
if (!unicode) if (!unicode)
return NULL; return NULL;

25
Python/fileutils.c
View file

@ -67,10 +67,12 @@ _Py_char2wchar(const char* arg, size_t *size)
#ifdef __APPLE__ #ifdef __APPLE__
wchar_t *wstr; wchar_t *wstr;
wstr = _Py_DecodeUTF8_surrogateescape(arg, strlen(arg)); wstr = _Py_DecodeUTF8_surrogateescape(arg, strlen(arg));
if (wstr == NULL) if (size != NULL) {
return NULL; if (wstr != NULL)
if (size != NULL)
*size = wcslen(wstr); *size = wcslen(wstr);
else
*size = (size_t)-1;
}
return wstr; return wstr;
#else #else
wchar_t *res; wchar_t *res;
@ -204,22 +206,25 @@ _Py_wchar2char(const wchar_t *text, size_t *error_pos)
char *cpath; char *cpath;
unicode = PyUnicode_FromWideChar(text, wcslen(text)); unicode = PyUnicode_FromWideChar(text, wcslen(text));
if (unicode == NULL) { if (unicode == NULL)
Py_DECREF(unicode);
return NULL; return NULL;
}
bytes = _PyUnicode_AsUTF8String(unicode, "surrogateescape"); bytes = _PyUnicode_AsUTF8String(unicode, "surrogateescape");
Py_DECREF(unicode); Py_DECREF(unicode);
if (bytes == NULL) { if (bytes == NULL) {
PyErr_Clear(); PyErr_Clear();
if (error_pos != NULL)
*error_pos = (size_t)-1;
return NULL; return NULL;
} }
len = PyBytes_GET_SIZE(bytes); len = PyBytes_GET_SIZE(bytes);
cpath = PyMem_Malloc(len+1); cpath = PyMem_Malloc(len+1);
if (cpath == NULL) { if (cpath == NULL) {
PyErr_Clear();
Py_DECREF(bytes); Py_DECREF(bytes);
if (error_pos != NULL)
*error_pos = (size_t)-1;
return NULL; return NULL;
} }
memcpy(cpath, PyBytes_AsString(bytes), len + 1); memcpy(cpath, PyBytes_AsString(bytes), len + 1);
@ -231,9 +236,6 @@ _Py_wchar2char(const wchar_t *text, size_t *error_pos)
size_t i, size, converted; size_t i, size, converted;
wchar_t c, buf[2]; wchar_t c, buf[2];
if (error_pos != NULL)
*error_pos = (size_t)-1;
/* The function works in two steps: /* The function works in two steps:
1. compute the length of the output buffer in bytes (size) 1. compute the length of the output buffer in bytes (size)
2. outputs the bytes */ 2. outputs the bytes */
@ -280,8 +282,11 @@ _Py_wchar2char(const wchar_t *text, size_t *error_pos)
size += 1; /* nul byte at the end */ size += 1; /* nul byte at the end */
result = PyMem_Malloc(size); result = PyMem_Malloc(size);
if (result == NULL) if (result == NULL) {
if (error_pos != NULL)
*error_pos = (size_t)-1;
return NULL; return NULL;
}
bytes = result; bytes = result;
} }
return result; return result;