# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1661532780 -7200
# Node ID 7d4a95ba9b6c6ce4491b989c147c71c11e96cf0b
# Parent  2200f0c6b3f1d3209a3339009c960a95151674c0
mod_s2s: Fix firing buffer drain events

Fixes the same kind of issue as in 65563530375b but once and for all,
while improving similarity between incoming and outgoing connections.

diff -r 2200f0c6b3f1 -r 7d4a95ba9b6c plugins/mod_s2s.lua
--- a/plugins/mod_s2s.lua	Mon Aug 15 18:56:22 2022 +0200
+++ b/plugins/mod_s2s.lua	Fri Aug 26 18:53:00 2022 +0200
@@ -464,6 +464,7 @@
 		end
 		if session.type == "s2sin_unauthed" and to and not session.to_host then
 			session.to_host = to;
+			session.host = to;
 		elseif to ~= session.to_host then
 			session:close({ condition = "improper-addressing", text = "New stream 'to' attribute does not match original" });
 			return;
@@ -888,8 +889,7 @@
 function listener.onreadtimeout(conn)
 	local session = sessions[conn];
 	if session then
-		local host = session.host or session.to_host;
-		return (hosts[host] or prosody).events.fire_event("s2s-read-timeout", { session = session });
+		return (hosts[session.host] or prosody).events.fire_event("s2s-read-timeout", { session = session });
 	end
 end