mod_s2s: Buffer stream error + stream closing tag
authorKim Alvefur <zash@zash.se>
Wed, 03 Mar 2021 13:34:00 +0100
changeset 11423 2c26dfc1977f
parent 11422 f4b76e500768
child 11424 f768db80aee0
mod_s2s: Buffer stream error + stream closing tag In "opportunistic writes" mode, prevents ondisconnect from happening while writing the stream closing tag.
plugins/mod_s2s.lua
--- a/plugins/mod_s2s.lua	Wed Mar 03 13:30:19 2021 +0100
+++ b/plugins/mod_s2s.lua	Wed Mar 03 13:34:00 2021 +0100
@@ -503,6 +503,8 @@
 local function session_close(session, reason, remote_reason, bounce_reason)
 	local log = session.log or log;
 	if session.conn then
+		local conn = session.conn;
+		conn:pause_writes(); -- until :close
 		if session.notopen then
 			if session.direction == "incoming" then
 				session:open_stream(session.to_host, session.from_host);
@@ -540,8 +542,9 @@
 		session.log("info", "%s s2s stream %s->%s closed: %s", session.direction:gsub("^.", string.upper),
 			session.from_host or "(unknown host)", session.to_host or "(unknown host)", reason or "stream closed");
 
+		conn:resume_writes();
+
 		-- Authenticated incoming stream may still be sending us stanzas, so wait for </stream:stream> from remote
-		local conn = session.conn;
 		if reason == nil and not session.notopen and session.direction == "incoming" then
 			add_task(stream_close_timeout, function ()
 				if not session.destroyed then