Issue #18408: _PyUnicodeWriter_Finish() now clears its buffer attribute in all

cases, so _PyUnicodeWriter_Dealloc() can be called after finish.
This commit is contained in:
Victor Stinner 2013-07-09 00:37:24 +02:00
parent 7979926616
commit 9e6b4d715c

View file

@ -13159,18 +13159,21 @@ _PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
{
PyObject *str;
if (writer->pos == 0) {
Py_XDECREF(writer->buffer);
Py_CLEAR(writer->buffer);
_Py_RETURN_UNICODE_EMPTY();
}
if (writer->readonly) {
assert(PyUnicode_GET_LENGTH(writer->buffer) == writer->pos);
return writer->buffer;
str = writer->buffer;
writer->buffer = NULL;
assert(PyUnicode_GET_LENGTH(str) == writer->pos);
return str;
}
if (PyUnicode_GET_LENGTH(writer->buffer) != writer->pos) {
PyObject *newbuffer;
newbuffer = resize_compact(writer->buffer, writer->pos);
if (newbuffer == NULL) {
Py_DECREF(writer->buffer);
writer->buffer = NULL;
return NULL;
}
writer->buffer = newbuffer;