From f32d91cc5772ed893879e996e52b6c3bcebf8016 Mon Sep 17 00:00:00 2001
From: James R <justsomejames2@gmail.com>
Date: Thu, 14 May 2020 17:19:25 -0700
Subject: [PATCH] Resize response body buffer as needed

(cherry picked from commit 85fef2655cc78c9453aed423f5a1df6011ce3262)
---
 src/http-mserv.c | 27 +++++++++++++++++++--------
 1 file changed, 19 insertions(+), 8 deletions(-)

diff --git a/src/http-mserv.c b/src/http-mserv.c
index 2864d806f..2c52773c8 100644
--- a/src/http-mserv.c
+++ b/src/http-mserv.c
@@ -25,6 +25,9 @@ Documentation available here.
 #include "i_tcp.h"/* for current_port */
 #include "i_threads.h"
 
+/* reasonable default I guess?? */
+#define DEFAULT_BUFFER_SIZE (4096)
+
 /* I just stop myself from making macros anymore. */
 #define Blame( ... ) \
 	CONS_Printf("\x85" __VA_ARGS__)
@@ -65,16 +68,25 @@ static size_t
 HMS_on_read (char *s, size_t _1, size_t n, void *userdata)
 {
 	struct HMS_buffer *buffer;
+	size_t blocks;
+
 	(void)_1;
+
 	buffer = userdata;
-	if (n < (size_t)( buffer->end - buffer->needle ))
+
+	if (n >= (size_t)( buffer->end - buffer->needle ))
 	{
-		memcpy(&buffer->buffer[buffer->needle], s, n);
-		buffer->needle += n;
-		return n;
+		/* resize to next multiple of buffer size */
+		blocks = ( n / DEFAULT_BUFFER_SIZE + 1 );
+		buffer->end = ( blocks * DEFAULT_BUFFER_SIZE );
+
+		buffer->buffer = realloc(buffer->buffer, buffer->end);
 	}
-	else
-		return 0;
+
+	memcpy(&buffer->buffer[buffer->needle], s, n);
+	buffer->needle += n;
+
+	return n;
 }
 
 static struct HMS_buffer *
@@ -134,8 +146,7 @@ HMS_connect (const char *format, ...)
 
 	buffer = malloc(sizeof *buffer);
 	buffer->curl = curl;
-	/* I just allocated 4k and fuck it! */
-	buffer->end = 4096;
+	buffer->end = DEFAULT_BUFFER_SIZE;
 	buffer->buffer = malloc(buffer->end);
 	buffer->needle = 0;