From 63680781b37c1f57998cc423660a98591f908527 Mon Sep 17 00:00:00 2001
From: zefie <zefie@zefie.net>
Date: Thu, 12 Aug 2021 01:40:12 -0400
Subject: [PATCH] just in case

---
 zefie_wtvp_minisrv/app.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/zefie_wtvp_minisrv/app.js b/zefie_wtvp_minisrv/app.js
index 91262753..6fa4354a 100644
--- a/zefie_wtvp_minisrv/app.js
+++ b/zefie_wtvp_minisrv/app.js
@@ -739,6 +739,7 @@ async function sendToClient(socket, headers_obj, data) {
 async function sendToSocket(socket, data) {
     var chunk_size = 16384;
     var can_write = true;
+    var close_socket = false;
     var expected_data_out = 0;
     while ((socket.bytesWritten == 0 || socket.bytesWritten != expected_data_out) && can_write) {
         if (expected_data_out === 0) expected_data_out = data.byteLength + (socket_sessions[socket.id].socket_total_written || 0);
@@ -747,6 +748,11 @@ async function sendToSocket(socket, data) {
         var data_left = (expected_data_out - socket.bytesWritten);
         // buffer size = lesser of chunk_size or size remaining
         var buffer_size = (data_left >= chunk_size) ? chunk_size : data_left;
+        if (buffer_size < 0) {
+            socket.destroy();
+            close_socket = true;
+            break;
+        }
         var offset = (data.byteLength - data_left);
         var chunk = new Buffer.alloc(buffer_size);
         data.copy(chunk, 0, offset, (offset + buffer_size));
@@ -758,7 +764,7 @@ async function sendToSocket(socket, data) {
             break;
         }
     }
-    if (socket.bytesWritten == expected_data_out) {
+    if (socket.bytesWritten == expected_data_out || close_socket) {
         socket_sessions[socket.id].socket_total_written = socket.bytesWritten;
         if (socket_sessions[socket.id].expecting_post_data) delete socket_sessions[socket.id].expecting_post_data;
         if (socket_sessions[socket.id].header_buffer) delete socket_sessions[socket.id].header_buffer;