# HG changeset patch
# User Matthew Wild <mwild1@gmail.com>
# Date 1600357308 -3600
# Node ID fa1821b56f75a56a5050ba06e18c9ea77cb7a9f8
# Parent  ddd0007e0f1b85e43631232b954cfe8380122e3e
mod_websocket: handle full frame buffer and raise stream error

diff -r ddd0007e0f1b -r fa1821b56f75 plugins/mod_websocket.lua
--- a/plugins/mod_websocket.lua	Thu Sep 17 13:04:46 2020 +0100
+++ b/plugins/mod_websocket.lua	Thu Sep 17 16:41:48 2020 +0100
@@ -275,7 +275,11 @@
 	-- max frame header is 22 bytes
 	local frameBuffer = dbuffer.new(stanza_size_limit + 22, frame_fragment_limit);
 	add_filter(session, "bytes/in", function(data)
-		frameBuffer:write(data);
+		if not frameBuffer:write(data) then
+			session.log("warn", "websocket frame buffer full - terminating session");
+			session:close({ condition = "resource-constraint", text = "frame buffer exceeded" });
+			return;
+		end
 
 		local cache = {};
 		local frame, length = parse_frame(frameBuffer);