prosody-modules: mod_s2s_keepalive/mod_s2s

annotate mod_s2s_keepalive/mod_s2s_keepalive.lua @ 4260:c539334dd01a

mod_http_oauth2: Rescope oauth client config into users' storage This produces client_id of the form owner@host/random and prevents clients from being deleted by registering an account with the same name and then deleting the account, as well as having the client automatically be deleted when the owner account is removed. On one hand, this leaks the bare JID of the creator to users. On the other hand, it makes it obvious who made the oauth application. This module is experimental and only for developers, so this can be changed if a better method comes up.

author	Kim Alvefur <zash@zash.se>
date	Sat, 21 Nov 2020 23:55:10 +0100
parents	93a980ac1816
children	edde5905744a

rev	line source
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	1 local st = require "util.stanza";
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	2 local watchdog = require "util.watchdog";
4203 c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	3 local dt = require "util.datetime";
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	4
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	5 local keepalive_servers = module:get_option_set("keepalive_servers");
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	6 local keepalive_interval = module:get_option_number("keepalive_interval", 60);
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	7 local keepalive_timeout = module:get_option_number("keepalive_timeout", 593);
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	8
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	9 local host = module.host;
3764 07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	10 local s2sout = prosody.hosts[host].s2sout;
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	11
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	12 local function send_pings()
3764 07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	13 local ping_hosts = {};
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	14
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	15 for remote_domain, session in pairs(s2sout) do
3771 98e1e3ce307d mod_s2s_keepalive: Invert check to work with bidi connections Kim Alvefur <zash@zash.se> parents: 3770 diff changeset	16 if session.type ~= "s2sout_unauthed"
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	17 and (not(keepalive_servers) or keepalive_servers:contains(remote_domain)) then
4204 a5930a185806 mod_s2s_keepalive: Fix name of timestamp function Kim Alvefur <zash@zash.se> parents: 4203 diff changeset	18 session.sends2s(st.iq({ to = remote_domain, type = "get", from = host, id = "keepalive:"..dt.datetime()})
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	19 :tag("ping", { xmlns = "urn:xmpp:ping" })
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	20 );
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	21 end
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	22 end
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	23
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	24 for session in pairs(prosody.incoming_s2s) do
3771 98e1e3ce307d mod_s2s_keepalive: Invert check to work with bidi connections Kim Alvefur <zash@zash.se> parents: 3770 diff changeset	25 if session.type ~= "s2sin_unauthed"
3772 22f02716819f mod_s2s_keepalive: Isolate source host of pings Kim Alvefur <zash@zash.se> parents: 3771 diff changeset	26 and session.to_host == host
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	27 and (not(keepalive_servers) or keepalive_servers:contains(session.from_host)) then
3764 07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	28 if not s2sout[session.from_host] then ping_hosts[session.from_host] = true; end
1264 2db2c03dfb95 mod_s2s_keepalive: Don't send directly on the connection, use sends2s Kim Alvefur <zash@zash.se> parents: 1110 diff changeset	29 session.sends2s " ";
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	30 -- If the connection is dead, this should make it time out.
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	31 end
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	32 end
3764 07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	33
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	34 -- ping remotes we only have s2sin from
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	35 for remote_domain in pairs(ping_hosts) do
4204 a5930a185806 mod_s2s_keepalive: Fix name of timestamp function Kim Alvefur <zash@zash.se> parents: 4203 diff changeset	36 module:send(st.iq({ to = remote_domain, type = "get", from = host, id = "keepalive:"..dt.datetime() })
3764 07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	37 :tag("ping", { xmlns = "urn:xmpp:ping" })
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	38 );
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	39 end
07a1faa24261 mod_s2s_keepalive: Ping remotes we only have s2sin established from Kim Alvefur <zash@zash.se> parents: 3723 diff changeset	40
1110 97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	41 return keepalive_interval;
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	42 end
97e238ce37ce mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace Kim Alvefur <zash@zash.se> parents: diff changeset	43
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	44 module:hook("s2sin-established", function (event)
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	45 local session = event.session;
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	46 if session.watchdog_keepalive then return end -- in case mod_bidi fires this twice
3833 580862decd77 mod_s2s_keepalive: Respect keepalive_servers when creating watchdogs Kim Alvefur <zash@zash.se> parents: 3772 diff changeset	47 if keepalive_servers and not keepalive_servers:contains(session.from_host) then return end
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	48 session.watchdog_keepalive = watchdog.new(keepalive_timeout, function ()
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	49 session.log("info", "Keepalive ping timed out, closing connection");
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	50 session:close("connection-timeout");
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	51 end);
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	52 end);
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	53
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	54 module:hook("s2sout-established", function (event)
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	55 local session = event.session;
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	56 if session.watchdog_keepalive then return end -- in case mod_bidi fires this twice
3833 580862decd77 mod_s2s_keepalive: Respect keepalive_servers when creating watchdogs Kim Alvefur <zash@zash.se> parents: 3772 diff changeset	57 if keepalive_servers and not keepalive_servers:contains(session.from_host) then return end
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	58 session.watchdog_keepalive = watchdog.new(keepalive_timeout, function ()
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	59 session.log("info", "Keepalive ping timed out, closing connection");
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	60 session:close("connection-timeout");
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	61 end);
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	62 end);
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	63
4212 593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	64 module:hook("iq/host", function (event)
4203 c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	65 local stanza = event.stanza;
4212 593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	66 if stanza.attr.type ~= "result" and stanza.attr.type == "error" then
593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	67 return -- not a reply iq stanza
593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	68 end
4203 c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	69 if not (stanza.attr.id and stanza.attr.id:sub(1, #"keepalive:") == "keepalive:") then
c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	70 return -- not a reply to this module
c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	71 end
c4002aae4ad3 mod_s2s_keepalive: Use timestamp as iq @id Kim Alvefur <zash@zash.se> parents: 3833 diff changeset	72
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	73 local origin = event.origin;
4212 593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	74 if origin.dummy then return end -- Probably a sendq bounce
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	75 if origin.watchdog_keepalive then
4212 593fd9e0a435 mod_s2s_keepalive: Fix response handler (thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4204 diff changeset	76 origin.log("debug", "Resetting keepalive watchdog")
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	77 origin.watchdog_keepalive:reset();
11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	78 end
3766 f547eafb5a6d mod_s2s_keepalive: Fix s2sout watchdog reset Kim Alvefur <zash@zash.se> parents: 3765 diff changeset	79 if s2sout[origin.from_host] and s2sout[origin.from_host].watchdog_keepalive then
f547eafb5a6d mod_s2s_keepalive: Fix s2sout watchdog reset Kim Alvefur <zash@zash.se> parents: 3765 diff changeset	80 s2sout[origin.from_host].watchdog_keepalive:reset();
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	81 end
3770 ae34ee0867f0 mod_s2s_keepalive: Mark ping response event as handled Kim Alvefur <zash@zash.se> parents: 3769 diff changeset	82 return true;
3765 11878130f266 mod_s2s_keepalive: Use a watchdog to close unresponsive sessions (fixes #1457) Kim Alvefur <zash@zash.se> parents: 3764 diff changeset	83 end);
4213 93a980ac1816 mod_s2s_keepalive: Restore timer start (Thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4212 diff changeset	84
93a980ac1816 mod_s2s_keepalive: Restore timer start (Thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 4212 diff changeset	85 module:add_timer(keepalive_interval, send_pings);

Mercurial > prosody-modules

annotate mod_s2s_keepalive/mod_s2s_keepalive.lua @ 4260:c539334dd01a