author | Matthew Wild <mwild1@gmail.com> |
Tue, 16 Apr 2024 12:05:47 +0100 | |
changeset 5887 | 6c11e1b4066d |
parent 3117 | 8298b06e6603 |
permissions | -rw-r--r-- |
2038 | 1 |
local new_watchdog = require "util.watchdog".new; |
2 |
local filters = require "util.filters"; |
|
3 |
local st = require "util.stanza"; |
|
4 |
||
5 |
local idle_timeout = module:get_option_number("c2s_idle_timeout", 300); |
|
6 |
local ping_timeout = module:get_option_number("c2s_ping_timeout", 30); |
|
7 |
||
8 |
function update_watchdog(data, session) |
|
3117
8298b06e6603
mod_pinger: work around updates on stale sessions
Georg Lukas <georg@op-co.de>
parents:
3107
diff
changeset
|
9 |
if session.idle_watchdog then |
8298b06e6603
mod_pinger: work around updates on stale sessions
Georg Lukas <georg@op-co.de>
parents:
3107
diff
changeset
|
10 |
session.idle_watchdog:reset(); |
8298b06e6603
mod_pinger: work around updates on stale sessions
Georg Lukas <georg@op-co.de>
parents:
3107
diff
changeset
|
11 |
session.idle_pinged = nil; |
8298b06e6603
mod_pinger: work around updates on stale sessions
Georg Lukas <georg@op-co.de>
parents:
3107
diff
changeset
|
12 |
end |
2038 | 13 |
return data; |
14 |
end |
|
15 |
||
16 |
function check_session(watchdog) |
|
17 |
local session = watchdog.session; |
|
3107
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
18 |
if session.smacks then |
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
19 |
unwatch_session(session); |
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
20 |
return; |
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
21 |
end |
2038 | 22 |
if not session.idle_pinged then |
23 |
session.idle_pinged = true; |
|
3107
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
24 |
session.send(st.iq({ type = "get", from = module.host, id = "idle-check" }) |
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
25 |
:tag("ping", { xmlns = "urn:xmpp:ping" })); |
2038 | 26 |
return ping_timeout; -- Call us again after ping_timeout |
27 |
else |
|
28 |
module:log("info", "Client %q silent for too long, closing...", session.full_jid); |
|
29 |
session:close("connection-timeout"); |
|
30 |
end |
|
31 |
end |
|
32 |
||
33 |
||
34 |
function watch_session(session) |
|
35 |
if not session.idle_watchdog |
|
36 |
and not session.requests then -- Don't watch BOSH connections (BOSH already has timeouts) |
|
37 |
session.idle_watchdog = new_watchdog(idle_timeout, check_session); |
|
38 |
session.idle_watchdog.session = session; |
|
39 |
filters.add_filter(session, "bytes/in", update_watchdog); |
|
40 |
end |
|
41 |
end |
|
42 |
||
43 |
function unwatch_session(session) |
|
44 |
if session.idle_watchdog then |
|
3107
5bf79bb3cf7e
Neuter 0198 from mod_pinger, fix #712
Georg Lukas <georg@op-co.de>
parents:
2678
diff
changeset
|
45 |
filters.remove_filter(session, "bytes/in", update_watchdog); |
2038 | 46 |
session.idle_watchdog:cancel(); |
47 |
session.idle_watchdog = nil; |
|
48 |
end |
|
49 |
end |
|
50 |
||
51 |
module:hook("resource-bind", function (event) watch_session(event.session); end); |
|
52 |
module:hook("resource-unbind", function (event) unwatch_session(event.session); end); |