From: Victor Stinner <victor.stinner@gmail.com>
Date: Mon, 8 Jul 2013 20:29:55 +0000 (+0200)
Subject: Issue #18408: Fix _PyUnicodeWriter_Finish(): clear writer->buffer,
X-Git-Tag: v3.4.0a1~281
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=15a0bd39658461165fd3b56ba92be5f53f8cc5a0;p=python

Issue #18408: Fix _PyUnicodeWriter_Finish(): clear writer->buffer,
so _PyUnicodeWriter_Dealloc() can be called on the writer after finish.
---

diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
index ac5f66aeaa..96c7c5825e 100644
--- a/Objects/unicodeobject.c
+++ b/Objects/unicodeobject.c
@@ -13157,6 +13157,7 @@ _PyUnicodeWriter_WriteCstr(_PyUnicodeWriter *writer, const char *str, Py_ssize_t
 PyObject *
 _PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
 {
+    PyObject *str;
     if (writer->pos == 0) {
         Py_XDECREF(writer->buffer);
         _Py_RETURN_UNICODE_EMPTY();
@@ -13174,8 +13175,10 @@ _PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
         }
         writer->buffer = newbuffer;
     }
-    assert(_PyUnicode_CheckConsistency(writer->buffer, 1));
-    return unicode_result_ready(writer->buffer);
+    str = writer->buffer;
+    writer->buffer = NULL;
+    assert(_PyUnicode_CheckConsistency(str, 1));
+    return unicode_result_ready(str);
 }
 
 void