Comparison

mod_component: Add read timeout handler (same behaviour as c2s and s2s)

comparison

equal deleted inserted replaced

-:60d3b53a36f7
+:679746cdf3cc
 local log = module._log;
 local opt_keepalives = module:get_option_boolean("component_tcp_keepalives", module:get_option_boolean("tcp_keepalives", true));
 local sessions = module:shared("sessions");
+local function keepalive(event)
+	local session = event.session;
+	if not session.notopen then
+		return event.session.send(' ');
+	end
+end
 function module.add_host(module)
 	if module:get_host_type() ~= "component" then
 		error("Don't load mod_component manually, it should be for a component, please see http://prosody.im/doc/components", 0);
 	end
 	module:hook("message/full", handle_stanza, -1);
 	module:hook("presence/full", handle_stanza, -1);
 	module:hook("iq/host", handle_stanza, -1);
 	module:hook("message/host", handle_stanza, -1);
 	module:hook("presence/host", handle_stanza, -1);
-end
+	module:hook("component-read-timeout", keepalive, -1);
+end
+module:hook("component-read-timeout", keepalive, -1);
 --- Network and stream part ---
 local xmlns_component = 'jabber:component:accept';
 function listener.ondetach(conn)
 	sessions[conn] = nil;
 end
+function listener.onreadtimeout(conn)
+	local session = sessions[conn];
+	if session then
+		return (hosts[session.host] or prosody).events.fire_event("component-read-timeout", { session = session });
+	end
+end
 module:provides("net", {
 	name = "component";
 	private = true;
 	listener = listener;
 	default_port = 5347;