# HG changeset patch # User Matthew Wild # Date 1453835700 0 # Node ID 256a5e3591db120363d7e5c1b4c121aec701f4b8 # Parent acf86edeb1ccaaa522f4c53aa5c08f3411c5a4a9 mod_pinger: Added from /files/ diff -r acf86edeb1cc -r 256a5e3591db mod_pinger/mod_pinger.lua --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_pinger/mod_pinger.lua Tue Jan 26 19:15:00 2016 +0000 @@ -0,0 +1,50 @@ +local new_watchdog = require "util.watchdog".new; +local filters = require "util.filters"; +local st = require "util.stanza"; + +local idle_timeout = module:get_option_number("c2s_idle_timeout", 300); +local ping_timeout = module:get_option_number("c2s_ping_timeout", 30); + +function update_watchdog(data, session) + session.idle_watchdog:reset(); + session.idle_pinged = nil; + return data; +end + +function check_session(watchdog) + local session = watchdog.session; + if not session.idle_pinged then + session.idle_pinged = true; + if session.smacks and not session.awaiting_ack then + session.send(st.stanza("r", { xmlns = "urn:xmpp:sm:2" })) -- TODO: hardcoded sm:2 + else + session.send(st.iq({ type = "get", from = module.host, id = "idle-check" }) + :tag("ping", { xmlns = "urn:xmpp:ping" })); + end + return ping_timeout; -- Call us again after ping_timeout + else + module:log("info", "Client %q silent for too long, closing...", session.full_jid); + session:close("connection-timeout"); + end +end + + +function watch_session(session) + if not session.idle_watchdog + and not session.requests then -- Don't watch BOSH connections (BOSH already has timeouts) + session.idle_watchdog = new_watchdog(idle_timeout, check_session); + session.idle_watchdog.session = session; + filters.add_filter(session, "bytes/in", update_watchdog); + end +end + +function unwatch_session(session) + if session.idle_watchdog then + session.idle_watchdog:cancel(); + session.idle_watchdog = nil; + filters.remove_filter(session, "bytes/in", update_watchdog); + end +end + +module:hook("resource-bind", function (event) watch_session(event.session); end); +module:hook("resource-unbind", function (event) unwatch_session(event.session); end);