mod_c2s, mod_s2s: Fire an event on read timeouts

diff --git a/plugins/mod_c2s.lua b/plugins/mod_c2s.lua
index d87b34152a80e501557c472c45ae340b741e6417..91bde5742f844f6019797129f7e15a64412da32c 100644 (file)
--- a/plugins/mod_c2s.lua
+++ b/plugins/mod_c2s.lua
@@ -265,15 +265,23 @@ end
 function listener.onreadtimeout(conn)
        local session = sessions[conn];
        if session then
-               return session.send(' ');
+               return (hosts[session.host] or prosody).events.fire_event("c2s-read-timeout", { session = session });
        end
 end
 
+local function keepalive(event)
+       return event.session.send(' ');
+end
+
 function listener.associate_session(conn, session)
        sessions[conn] = session;
 end
 
-function module.add_host() end
+function module.add_host(module)
+       module:hook("c2s-read-timeout", keepalive, -1);
+end
+
+module:hook("c2s-read-timeout", keepalive, -1);
 
 module:hook("server-stopping", function(event)
        local reason = event.reason;

diff --git a/plugins/mod_s2s/mod_s2s.lua b/plugins/mod_s2s/mod_s2s.lua
index bce617ca69d6f6e42402f2ebe895082ed49543ff..5e50e88b6207ca3085067560430bdf4c1b071656 100644 (file)
--- a/plugins/mod_s2s/mod_s2s.lua
+++ b/plugins/mod_s2s/mod_s2s.lua
@@ -135,6 +135,10 @@ function route_to_new_session(event)
        return true;
 end
 
+local function keepalive(event)
+       return event.session.sends2s(' ');
+end
+
 function module.add_host(module)
        if module:get_option_boolean("disallow_s2s", false) then
                module:log("warn", "The 'disallow_s2s' config option is deprecated, please see http://prosody.im/doc/s2s#disabling");
@@ -143,6 +147,7 @@ function module.add_host(module)
        module:hook("route/remote", route_to_existing_session, -1);
        module:hook("route/remote", route_to_new_session, -10);
        module:hook("s2s-authenticated", make_authenticated, -1);
+       module:hook("s2s-read-timeout", keepalive, -1);
 end
 
 -- Stream is authorised, and ready for normal stanzas
@@ -628,7 +633,7 @@ end
 function listener.onreadtimeout(conn)
        local session = sessions[conn];
        if session then
-               return session.sends2s(' ');
+               return (hosts[session.host] or prosody).events.fire_event("s2s-read-timeout", { session = session });
        end
 end