bpo-32827: Fix usage of _PyUnicodeWriter_Prepare() in decoding errors handler. (GH-5636)

This commit is contained in:
Serhiy Storchaka 2018-02-13 08:27:33 +02:00 committed by GitHub
parent 688722cedd
commit b7e2d67f7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4256,7 +4256,7 @@ unicode_decode_call_errorhandler_writer(
}
if (need_to_grow) {
writer->overallocate = 1;
if (_PyUnicodeWriter_Prepare(writer, writer->min_length,
if (_PyUnicodeWriter_Prepare(writer, writer->min_length - writer->pos,
PyUnicode_MAX_CHAR_VALUE(repunicode)) == -1)
goto onError;
}
@ -6085,9 +6085,7 @@ _PyUnicode_DecodeUnicodeEscape(const char *s,
&writer)) {
goto onError;
}
if (_PyUnicodeWriter_Prepare(&writer, writer.min_length, 127) < 0) {
goto onError;
}
assert(end - s <= writer.size - writer.pos);
#undef WRITE_ASCII_CHAR
#undef WRITE_CHAR
@ -6364,9 +6362,7 @@ PyUnicode_DecodeRawUnicodeEscape(const char *s,
&writer)) {
goto onError;
}
if (_PyUnicodeWriter_Prepare(&writer, writer.min_length, 127) < 0) {
goto onError;
}
assert(end - s <= writer.size - writer.pos);
#undef WRITE_CHAR
}