diff options
Diffstat (limited to 'Objects/longobject.c')
| -rw-r--r-- | Objects/longobject.c | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/Objects/longobject.c b/Objects/longobject.c index 35d121120d..7374fce9dd 100644 --- a/Objects/longobject.c +++ b/Objects/longobject.c @@ -1123,17 +1123,19 @@ PyLong_FromString(char *str, char **pend, int base) PyObject * PyLong_FromUnicode(Py_UNICODE *u, int length, int base) { - char buffer[256]; + PyObject *result; + char *buffer = PyMem_MALLOC(length+1); - if (length >= sizeof(buffer)) { - PyErr_SetString(PyExc_ValueError, - "long() literal too large to convert"); - return NULL; - } - if (PyUnicode_EncodeDecimal(u, length, buffer, NULL)) + if (buffer == NULL) return NULL; - return PyLong_FromString(buffer, NULL, base); + if (PyUnicode_EncodeDecimal(u, length, buffer, NULL)) { + PyMem_FREE(buffer); + return NULL; + } + result = PyLong_FromString(buffer, NULL, base); + PyMem_FREE(buffer); + return result; } #endif |