From abf3e7015e1f77b2d665c1e0e4e4477100d8623b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Mon, 31 Jan 2022 19:59:42 +0100
Subject: [PATCH] http: don't send response if body is too long for buffer

---
 aquatic_http/src/workers/socket.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/aquatic_http/src/workers/socket.rs b/aquatic_http/src/workers/socket.rs
index 31c764b..7dc7e6e 100644
--- a/aquatic_http/src/workers/socket.rs
+++ b/aquatic_http/src/workers/socket.rs
@@ -429,6 +429,12 @@ impl Connection {
 
         position += body_len;
 
+        if position + 2 > self.response_buffer.len() {
+            ::log::error!("Response buffer is too short for response");
+
+            return Err(anyhow::anyhow!("Response buffer is too short for response"));
+        }
+
         (&mut self.response_buffer[position..position + 2]).copy_from_slice(b"\r\n");
 
         position += 2;