aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorKim Alvefur <zash@zash.se>2017-02-10 21:25:12 +0100
committerKim Alvefur <zash@zash.se>2017-02-10 21:25:12 +0100
commita586901308157ae912d86292adf3ca6704217521 (patch)
tree51c4d53e2363c57f4540a50f6288a19f3e4f45f6
parent0e99534951484e5c1ab98897272c92ea6b13f8ff (diff)
downloadprosody-a586901308157ae912d86292adf3ca6704217521.tar.gz
prosody-a586901308157ae912d86292adf3ca6704217521.zip
mod_component: Add read timeout handler (same behaviour as c2s and s2s)
-rw-r--r--plugins/mod_component.lua18
1 files changed, 18 insertions, 0 deletions
diff --git a/plugins/mod_component.lua b/plugins/mod_component.lua
index c46a6266..4da1baf7 100644
--- a/plugins/mod_component.lua
+++ b/plugins/mod_component.lua
@@ -29,6 +29,13 @@ local opt_keepalives = module:get_option_boolean("component_tcp_keepalives", mod
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);
@@ -135,8 +142,12 @@ function module.add_host(module)
module:hook("iq/host", handle_stanza, -1);
module:hook("message/host", handle_stanza, -1);
module:hook("presence/host", handle_stanza, -1);
+
+ 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';
@@ -332,6 +343,13 @@ 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;