Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-06-23 15:30:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

py3: Fix conversion of 1-byte python strings to ICU strings

Browse Source
This commit is contained in:
Kovid Goyal 2019-04-02 15:56:06 +05:30
parent bea653d594
commit 8bb15338f5
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/calibre/utils/icu_calibre_utils.h
View File

@ -134,19 +134,18 @@ static UChar* python_to_icu(PyObject *obj, int32_t *osz) {
switch(PyUnicode_KIND(obj)) {
case PyUnicode_1BYTE_KIND:
ans = (UChar*) malloc((sz+1) * sizeof(UChar));
if (ans == NULL) {
PyErr_NoMemory();
return NULL;
}
u_strFromUTF8(
ans, sz + 1,
(int32_t*) osz,
(char*) PyUnicode_1BYTE_DATA(obj),
(int32_t) sz,
&status);
case PyUnicode_1BYTE_KIND: {
Py_ssize_t data_sz;
const char *utf8_data = PyUnicode_AsUTF8AndSize(obj, &data_sz);
if (!utf8_data) return NULL;
size_t buf_sz = (sz > data_sz ? sz : data_sz) + 1;
ans = (UChar*) malloc(buf_sz * sizeof(UChar));
if (ans == NULL) { PyErr_NoMemory(); return NULL; }
u_strFromUTF8Lenient(ans, buf_sz, (int32_t*) osz, utf8_data, (int32_t)data_sz, &status);
// add null terminator
ans[buf_sz-1] = 0;
break;
}
case PyUnicode_2BYTE_KIND:
ans = (UChar*) malloc((sz+1) * sizeof(UChar));
data = PyUnicode_2BYTE_DATA(obj);