Fix some more seg-faults on encoding.

ultrajson · Feb 12, 2022 · 86e5fa7 · 86e5fa7
1 parent 7fd7dbf
commit 86e5fa7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/ultrajsonenc.c b/lib/ultrajsonenc.c
@@ -175,6 +175,7 @@ static FASTCALL_ATTR INLINE_PREFIX void FASTCALL_MSVC Buffer_AppendShortHexUnche
 
 static int Buffer_EscapeStringUnvalidated (JSONObjectEncoder *enc, const char *io, const char *end)
 {
+  Buffer_Reserve(enc, RESERVE_STRING(end - io));
   char *of = (char *) enc->offset;
 
   for (;;)
@@ -722,7 +723,7 @@ static void encode(JSOBJ obj, JSONObjectEncoder *enc, const char *name, size_t c
       if (count > 0) {
         // Reserve space for the indentation plus the newline and the closing
         // bracket.
-        Buffer_Reserve (enc, enc->indent * enc->level + 2);
+        Buffer_Reserve (enc, enc->indent * enc->level + 4);
         Buffer_AppendIndentNewlineUnchecked (enc);
         Buffer_AppendIndentUnchecked (enc, enc->level);
       }