Software `/` code `/` prosody-modules

Annotate

mod_smacks/mod_smacks.lua @ 1870:6a9f91ad4f95

mod_log_mark/README: Also mention logging

author	Kim Alvefur <zash@zash.se>
date	Thu, 24 Sep 2015 00:02:15 +0200
parent	1733:9abd3dce619a
child	1881:3683eb95bc1a

rev	line source
1670 5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	1 -- XEP-0198: Stream Management for Prosody IM
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	2 --
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	3 -- Copyright (C) 2010-2015 Matthew Wild
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	4 -- Copyright (C) 2010 Waqas Hussain
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	5 -- Copyright (C) 2012-2015 Kim Alvefur
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	6 -- Copyright (C) 2012 Thijs Alkemade
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	7 -- Copyright (C) 2014 Florian Zeitz
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	8 --
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	9 -- This project is MIT/X11 licensed. Please see the
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	10 -- COPYING file in the source package for more information.
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	11 --
5f5ff061b316 mod_smacks: Add license header Kim Alvefur <zash@zash.se> parents: 1637 diff changeset	12
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	13 local st = require "util.stanza";
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	14 local uuid_generate = require "util.uuid".generate;
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	15
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	16 local t_insert, t_remove = table.insert, table.remove;
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	17 local math_min = math.min;
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	18 local os_time = os.time;
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	19 local tonumber, tostring = tonumber, tostring;
201 bc24f58a0d39 mod_smacks: Use filters for catching incoming stanzas (more reliable and efficient), also add some logic to make compatible with the stream resumption module (coming soon) Matthew Wild <mwild1@gmail.com> parents: 200 diff changeset	20 local add_filter = require "util.filters".add_filter;
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	21 local timer = require "util.timer";
593 db2a40cbd6ef Add a <delay> to stanzas that are queued (and don't have one already), so clients can show them with the original timestamp. Thijs Alkemade <thijsalkemade@gmail.com> parents: 592 diff changeset	22 local datetime = require "util.datetime";
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	23
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	24 local xmlns_sm2 = "urn:xmpp:sm:2";
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	25 local xmlns_sm3 = "urn:xmpp:sm:3";
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	26 local xmlns_errors = "urn:ietf:params:xml:ns:xmpp-stanzas";
593 db2a40cbd6ef Add a <delay> to stanzas that are queued (and don't have one already), so clients can show them with the original timestamp. Thijs Alkemade <thijsalkemade@gmail.com> parents: 592 diff changeset	27 local xmlns_delay = "urn:xmpp:delay";
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	28
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	29 local sm2_attr = { xmlns = xmlns_sm2 };
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	30 local sm3_attr = { xmlns = xmlns_sm3 };
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	31
641 06a78cba6499 mod_smacks: get_option+_number Matthew Wild <mwild1@gmail.com> parents: 640 diff changeset	32 local resume_timeout = module:get_option_number("smacks_hibernation_time", 300);
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	33 local s2s_smacks = module:get_option_boolean("smacks_enabled_s2s", false);
642 842a8a3b0d81 mod_smacks: Make smacks_max_unacked_stanzas configurable Matthew Wild <mwild1@gmail.com> parents: 641 diff changeset	34 local max_unacked_stanzas = module:get_option_number("smacks_max_unacked_stanzas", 0);
754 713c6791fbcc mod_smacks: Import prosody.core_process_stanza() Kim Alvefur <zash@zash.se> parents: 642 diff changeset	35 local core_process_stanza = prosody.core_process_stanza;
757 92c6f84ec446 mod_smacks: Use require to import (thanks a lot, autocomplete) Kim Alvefur <zash@zash.se> parents: 756 diff changeset	36 local sessionmanager = require"core.sessionmanager";
200 64a573203c20 mod_smacks: Better logic for deciding what is a stanza and what is not, and deciding when to send ack requests Matthew Wild <mwild1@gmail.com> parents: 160 diff changeset	37
640 d87131b29bbd mod_smacks: Remove dependency on connlisteners (use sessions table shared by mod_c2s directly) Matthew Wild <mwild1@gmail.com> parents: 625 diff changeset	38 local c2s_sessions = module:shared("/*/c2s/sessions");
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	39 local session_registry = {};
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	40
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	41 local function can_do_smacks(session, advertise_only)
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	42 if session.smacks then return false, "unexpected-request", "Stream management is already enabled"; end
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	43
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	44 local session_type = session.type;
595 7693724881b3 Fix a typo in mod_smacks (type -> session_type). Thijs Alkemade <thijsalkemade@gmail.com> parents: 594 diff changeset	45 if session_type == "c2s" then
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	46 if not(advertise_only) and not(session.resource) then -- Fail unless we're only advertising sm
592 f9c73c1249cd Update smacks to urn:xmpp:sm:3. Fix typo in can_do_smacks. Thijs Alkemade <thijsalkemade@gmail.com> parents: 591 diff changeset	47 return false, "unexpected-request", "Client must bind a resource before enabling stream management";
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	48 end
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	49 return true;
595 7693724881b3 Fix a typo in mod_smacks (type -> session_type). Thijs Alkemade <thijsalkemade@gmail.com> parents: 594 diff changeset	50 elseif s2s_smacks and (session_type == "s2sin" or session_type == "s2sout") then
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	51 return true;
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	52 end
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	53 return false, "service-unavailable", "Stream management is not available for this stream";
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	54 end
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	55
263 41f1cac40560 mod_smacks: Fixed to use the correct events API. Waqas Hussain <waqas20@gmail.com> parents: 258 diff changeset	56 module:hook("stream-features",
41f1cac40560 mod_smacks: Fixed to use the correct events API. Waqas Hussain <waqas20@gmail.com> parents: 258 diff changeset	57 function (event)
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	58 if can_do_smacks(event.origin, true) then
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	59 event.features:tag("sm", sm2_attr):tag("optional"):up():up();
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	60 event.features:tag("sm", sm3_attr):tag("optional"):up():up();
589 57ac609444c4 mod_smacks: Only advertise stream features when a stream is authenticated, and doesn't already have smacks enabled Matthew Wild <mwild1@gmail.com> parents: 588 diff changeset	61 end
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	62 end);
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	63
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	64 module:hook("s2s-stream-features",
263 41f1cac40560 mod_smacks: Fixed to use the correct events API. Waqas Hussain <waqas20@gmail.com> parents: 258 diff changeset	65 function (event)
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	66 if can_do_smacks(event.origin, true) then
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	67 event.features:tag("sm", sm2_attr):tag("optional"):up():up();
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	68 event.features:tag("sm", sm3_attr):tag("optional"):up():up();
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	69 end
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	70 end);
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	71
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	72 local function outgoing_stanza_filter(stanza, session)
1597 dc0cf2ba0e1a mod_smacks: Stanzas don't have a ':' in their name Kim Alvefur <zash@zash.se> parents: 1539 diff changeset	73 local is_stanza = stanza.attr and not stanza.attr.xmlns and not stanza.name:find":";
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	74 if is_stanza and not stanza._cached then -- Stanza in default stream namespace
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	75 local queue = session.outgoing_stanza_queue;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	76 local cached_stanza = st.clone(stanza);
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	77 cached_stanza._cached = true;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	78
1599 8e006226e4c5 mod_smacks: Don't attach timestamps to 'iq' stanzas Kim Alvefur <zash@zash.se> parents: 1598 diff changeset	79 if cached_stanza and cached_stanza.name ~= "iq" and cached_stanza:get_child("delay", xmlns_delay) == nil then
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	80 cached_stanza = cached_stanza:tag("delay", { xmlns = xmlns_delay, from = session.host, stamp = datetime.datetime()});
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	81 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	82
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	83 queue[#queue+1] = cached_stanza;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	84 session.log("debug", "#queue = %d", #queue);
1522 d4a4ed31567e mod_smacks: Trick session.send() into not returning nil or mod_message will act as if messages were not delivered Kim Alvefur <zash@zash.se> parents: 1520 diff changeset	85 if session.hibernating then
d4a4ed31567e mod_smacks: Trick session.send() into not returning nil or mod_message will act as if messages were not delivered Kim Alvefur <zash@zash.se> parents: 1520 diff changeset	86 session.log("debug", "hibernating, stanza queued");
d4a4ed31567e mod_smacks: Trick session.send() into not returning nil or mod_message will act as if messages were not delivered Kim Alvefur <zash@zash.se> parents: 1520 diff changeset	87 return ""; -- Hack to make session.send() not return nil
d4a4ed31567e mod_smacks: Trick session.send() into not returning nil or mod_message will act as if messages were not delivered Kim Alvefur <zash@zash.se> parents: 1520 diff changeset	88 end
1539 05fa54404012 mod_smacks: Send ack requests by serializing and concatenating them to outgoing stanzas, using a timer might cause high CPU usage Kim Alvefur <zash@zash.se> parents: 1531 diff changeset	89 if #queue > max_unacked_stanzas and not session.awaiting_ack then
05fa54404012 mod_smacks: Send ack requests by serializing and concatenating them to outgoing stanzas, using a timer might cause high CPU usage Kim Alvefur <zash@zash.se> parents: 1531 diff changeset	90 session.awaiting_ack = true;
05fa54404012 mod_smacks: Send ack requests by serializing and concatenating them to outgoing stanzas, using a timer might cause high CPU usage Kim Alvefur <zash@zash.se> parents: 1531 diff changeset	91 return tostring(stanza)..tostring(st.stanza("r", { xmlns = session.smacks }));
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	92 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	93 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	94 return stanza;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	95 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	96
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	97 local function count_incoming_stanzas(stanza, session)
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	98 if not stanza.attr.xmlns then
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	99 session.handled_stanza_count = session.handled_stanza_count + 1;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	100 session.log("debug", "Handled %d incoming stanzas", session.handled_stanza_count);
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	101 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	102 return stanza;
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	103 end
2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	104
1529 16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	105 local function wrap_session_out(session, resume)
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	106 if not resume then
1528 5ccb3ee2cf72 mod_smacks: Remove variable not used after 2881d532f385 Kim Alvefur <zash@zash.se> parents: 1527 diff changeset	107 session.outgoing_stanza_queue = {};
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	108 session.last_acknowledged_stanza = 0;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	109 end
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	110
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	111 add_filter(session, "stanzas/out", outgoing_stanza_filter, -1000);
988 c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	112
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	113 local session_close = session.close;
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	114 function session.close(...)
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	115 if session.resumption_token then
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	116 session_registry[session.resumption_token] = nil;
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	117 session.resumption_token = nil;
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	118 end
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	119 return session_close(...);
c15cea87036f mod_smacks: Wrap session:close() and make the session not resumable Kim Alvefur <zash@zash.se> parents: 987 diff changeset	120 end
1529 16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	121 return session;
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	122 end
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	123
1529 16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	124 local function wrap_session_in(session, resume)
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	125 if not resume then
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	126 session.handled_stanza_count = 0;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	127 end
1703 a53385a78a64 mod_smacks: Add stanza counting filter when resuming session Kim Alvefur <zash@zash.se> parents: 1670 diff changeset	128 add_filter(session, "stanzas/in", count_incoming_stanzas, 1000);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	129
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	130 return session;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	131 end
201 bc24f58a0d39 mod_smacks: Use filters for catching incoming stanzas (more reliable and efficient), also add some logic to make compatible with the stream resumption module (coming soon) Matthew Wild <mwild1@gmail.com> parents: 200 diff changeset	132
1529 16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	133 local function wrap_session(session, resume)
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	134 wrap_session_out(session, resume);
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	135 wrap_session_in(session, resume);
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	136 return session;
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	137 end
16893646a458 mod_smacks: Move wrap_session() contents into separate direction-specific routines Kim Alvefur <zash@zash.se> parents: 1528 diff changeset	138
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	139 function handle_enable(session, stanza, xmlns_sm)
591 36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	140 local ok, err, err_text = can_do_smacks(session);
36003cae2370 mod_smacks: Consolidate logic for deciding whether to advertise or allow smacks for a given session, and fix an issue with not allowing s2s connections to enable smacks because of not binding a resource... (thanks xnyphs) Matthew Wild <mwild1@gmail.com> parents: 590 diff changeset	141 if not ok then
590 40b707d7a809 mod_smacks: Handle a client trying to <enable> twice, or trying to <enable> before resource binding (thanks Thijs Alkemade) Matthew Wild <mwild1@gmail.com> parents: 589 diff changeset	142 session.log("warn", "Failed to enable smacks: %s", err_text); -- TODO: XEP doesn't say we can send error text, should it?
1527 06ecc5b3ca46 mod_smacks: Send failure correctly on s2s Kim Alvefur <zash@zash.se> parents: 1526 diff changeset	143 (session.sends2s or session.send)(st.stanza("failed", { xmlns = xmlns_sm }):tag(err, { xmlns = xmlns_errors}));
590 40b707d7a809 mod_smacks: Handle a client trying to <enable> twice, or trying to <enable> before resource binding (thanks Thijs Alkemade) Matthew Wild <mwild1@gmail.com> parents: 589 diff changeset	144 return true;
40b707d7a809 mod_smacks: Handle a client trying to <enable> twice, or trying to <enable> before resource binding (thanks Thijs Alkemade) Matthew Wild <mwild1@gmail.com> parents: 589 diff changeset	145 end
40b707d7a809 mod_smacks: Handle a client trying to <enable> twice, or trying to <enable> before resource binding (thanks Thijs Alkemade) Matthew Wild <mwild1@gmail.com> parents: 589 diff changeset	146
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	147 module:log("debug", "Enabling stream management");
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	148 session.smacks = xmlns_sm;
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	149
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	150 wrap_session(session, false);
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	151
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	152 local resume_token;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	153 local resume = stanza.attr.resume;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	154 if resume == "true" or resume == "1" then
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	155 resume_token = uuid_generate();
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	156 session_registry[resume_token] = session;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	157 session.resumption_token = resume_token;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	158 end
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	159 (session.sends2s or session.send)(st.stanza("enabled", { xmlns = xmlns_sm, id = resume_token, resume = resume }));
44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	160 return true;
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	161 end
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	162 module:hook_stanza(xmlns_sm2, "enable", function (session, stanza) return handle_enable(session, stanza, xmlns_sm2); end, 100);
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	163 module:hook_stanza(xmlns_sm3, "enable", function (session, stanza) return handle_enable(session, stanza, xmlns_sm3); end, 100);
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	164
1530 fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	165 module:hook_stanza("http://etherx.jabber.org/streams", "features",
fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	166 function (session, stanza)
1531 7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	167 module:add_timer(0, function ()
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	168 if can_do_smacks(session) then
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	169 if stanza:get_child("sm", xmlns_sm3) then
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	170 session.sends2s(st.stanza("enable", sm3_attr));
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	171 session.smacks = xmlns_sm3;
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	172 elseif stanza:get_child("sm", xmlns_sm2) then
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	173 session.sends2s(st.stanza("enable", sm2_attr));
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	174 session.smacks = xmlns_sm2;
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	175 else
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	176 return;
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	177 end
7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	178 wrap_session_out(session, false);
1530 fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	179 end
1531 7d86fc477993 mod_smacks: Delay initiation of outgoing smacks on s2s until after queued stanzas are sent Kim Alvefur <zash@zash.se> parents: 1530 diff changeset	180 end);
1530 fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	181 end);
fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	182
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	183 function handle_enabled(session, stanza, xmlns_sm)
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	184 module:log("debug", "Enabling stream management");
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	185 session.smacks = xmlns_sm;
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	186
1530 fb7cd669f41b mod_smacks: Initiate outgoing smacks on s2s when sending request and incoming when the remote says enabled Kim Alvefur <zash@zash.se> parents: 1529 diff changeset	187 wrap_session_in(session, false);
576 44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	188
44b69c3d5351 mod_smacks: Fix smacks on s2s connections, but disable it by default. Kim Alvefur <zash@zash.se> parents: 478 diff changeset	189 -- FIXME Resume?
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	190
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	191 return true;
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	192 end
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	193 module:hook_stanza(xmlns_sm2, "enabled", function (session, stanza) return handle_enabled(session, stanza, xmlns_sm2); end, 100);
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	194 module:hook_stanza(xmlns_sm3, "enabled", function (session, stanza) return handle_enabled(session, stanza, xmlns_sm3); end, 100);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	195
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	196 function handle_r(origin, stanza, xmlns_sm)
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	197 if not origin.smacks then
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	198 module:log("debug", "Received ack request from non-smack-enabled session");
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	199 return;
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	200 end
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	201 module:log("debug", "Received ack request, acking for %d", origin.handled_stanza_count);
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	202 -- Reply with <a>
255 9b9b089407b1 mod_smacks: Fix to reply to stream for s2s sessions Matthew Wild <mwild1@gmail.com> parents: 220 diff changeset	203 (origin.sends2s or origin.send)(st.stanza("a", { xmlns = xmlns_sm, h = tostring(origin.handled_stanza_count) }));
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	204 return true;
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	205 end
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	206 module:hook_stanza(xmlns_sm2, "r", function (origin, stanza) return handle_r(origin, stanza, xmlns_sm2); end);
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	207 module:hook_stanza(xmlns_sm3, "r", function (origin, stanza) return handle_r(origin, stanza, xmlns_sm3); end);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	208
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	209 function handle_a(origin, stanza)
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	210 if not origin.smacks then return; end
200 64a573203c20 mod_smacks: Better logic for deciding what is a stanza and what is not, and deciding when to send ack requests Matthew Wild <mwild1@gmail.com> parents: 160 diff changeset	211 origin.awaiting_ack = nil;
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	212 -- Remove handled stanzas from outgoing_stanza_queue
258 36648205b10a mod_smacks: Add commented log statement for future debugging Matthew Wild <mwild1@gmail.com> parents: 257 diff changeset	213 --log("debug", "ACK: h=%s, last=%s", stanza.attr.h or "", origin.last_acknowledged_stanza or "");
1407 b631c8a8b9e7 mod_smacks: Assert that the 'h' attribute is present and a number Kim Alvefur <zash@zash.se> parents: 1406 diff changeset	214 local h = tonumber(stanza.attr.h);
b631c8a8b9e7 mod_smacks: Assert that the 'h' attribute is present and a number Kim Alvefur <zash@zash.se> parents: 1406 diff changeset	215 if not h then
b631c8a8b9e7 mod_smacks: Assert that the 'h' attribute is present and a number Kim Alvefur <zash@zash.se> parents: 1406 diff changeset	216 origin:close{ condition = "invalid-xml"; text = "Missing or invalid 'h' attribute"; };
b631c8a8b9e7 mod_smacks: Assert that the 'h' attribute is present and a number Kim Alvefur <zash@zash.se> parents: 1406 diff changeset	217 end
b631c8a8b9e7 mod_smacks: Assert that the 'h' attribute is present and a number Kim Alvefur <zash@zash.se> parents: 1406 diff changeset	218 local handled_stanza_count = h-origin.last_acknowledged_stanza;
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	219 local queue = origin.outgoing_stanza_queue;
263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	220 if handled_stanza_count > #queue then
1416 7ddb522d9b28 mod_smacks: Fix logging (Thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 1408 diff changeset	221 origin.log("warn", "The client says it handled %d new stanzas, but we only sent %d :)",
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	222 handled_stanza_count, #queue);
1416 7ddb522d9b28 mod_smacks: Fix logging (Thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 1408 diff changeset	223 origin.log("debug", "Client h: %d, our h: %d", tonumber(stanza.attr.h), origin.last_acknowledged_stanza);
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	224 for i=1,#queue do
1416 7ddb522d9b28 mod_smacks: Fix logging (Thanks Ge0rG) Kim Alvefur <zash@zash.se> parents: 1408 diff changeset	225 origin.log("debug", "Q item %d: %s", i, tostring(queue[i]));
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	226 end
263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	227 end
263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	228 for i=1,math_min(handled_stanza_count,#queue) do
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	229 t_remove(origin.outgoing_stanza_queue, 1);
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	230 end
1406 7d76dd2310ef mod_smacks: Add more debug logging Kim Alvefur <zash@zash.se> parents: 1405 diff changeset	231 origin.log("debug", "#queue = %d", #queue);
220 263858d40ceb mod_smacks: Fix the logic for handling outgoing stanzas and ack requests Matthew Wild <mwild1@gmail.com> parents: 202 diff changeset	232 origin.last_acknowledged_stanza = origin.last_acknowledged_stanza + handled_stanza_count;
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	233 return true;
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	234 end
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	235 module:hook_stanza(xmlns_sm2, "a", handle_a);
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	236 module:hook_stanza(xmlns_sm3, "a", handle_a);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	237
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	238 --TODO: Optimise... incoming stanzas should be handled by a per-session
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	239 -- function that has a counter as an upvalue (no table indexing for increments,
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	240 -- and won't slow non-198 sessions). We can also then remove the .handled flag
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	241 -- on stanzas
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	242
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	243 function handle_unacked_stanzas(session)
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	244 local queue = session.outgoing_stanza_queue;
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	245 local error_attr = { type = "cancel" };
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	246 if #queue > 0 then
202 d11478ae374e mod_smacks: Clean outgoing stanza queue correctly on session close Matthew Wild <mwild1@gmail.com> parents: 201 diff changeset	247 session.outgoing_stanza_queue = {};
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	248 for i=1,#queue do
256 57de4a7840ef mod_smacks: Fixes for storing the unacked stanzas so that they can be properly replayed to clients on stream resume Matthew Wild <mwild1@gmail.com> parents: 255 diff changeset	249 local reply = st.reply(queue[i]);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	250 if reply.attr.to ~= session.full_jid then
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	251 reply.attr.type = "error";
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	252 reply:tag("error", error_attr)
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	253 :tag("recipient-unavailable", {xmlns = "urn:ietf:params:xml:ns:xmpp-stanzas"});
256 57de4a7840ef mod_smacks: Fixes for storing the unacked stanzas so that they can be properly replayed to clients on stream resume Matthew Wild <mwild1@gmail.com> parents: 255 diff changeset	254 core_process_stanza(session, reply);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	255 end
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	256 end
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	257 end
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	258 end
9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	259
622 ce39df945de1 mod_smacks: Use pre-resource-unbind (0.9 feature) to fix reloadability and multiple host support Matthew Wild <mwild1@gmail.com> parents: 595 diff changeset	260 module:hook("pre-resource-unbind", function (event)
ce39df945de1 mod_smacks: Use pre-resource-unbind (0.9 feature) to fix reloadability and multiple host support Matthew Wild <mwild1@gmail.com> parents: 595 diff changeset	261 local session, err = event.session, event.error;
1024 d7655e634c30 mod_smacks: Allow resumption if the TCP connection is closed from our end, c15cea87036f ensures distinction from cleanly closed streams (thanks Lance) Kim Alvefur <zash@zash.se> parents: 994 diff changeset	262 if session.smacks then
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	263 if not session.resumption_token then
08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	264 local queue = session.outgoing_stanza_queue;
08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	265 if #queue > 0 then
1405 26a2092e289f mod_smacks: Log things attached to sessions instead of the module Kim Alvefur <zash@zash.se> parents: 1343 diff changeset	266 session.log("warn", "Destroying session with %d unacked stanzas", #queue);
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	267 handle_unacked_stanzas(session);
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	268 end
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	269 else
586 f733e7599ed6 mod_smacks: Add logging to hibernation and session destruction (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 576 diff changeset	270 session.log("debug", "mod_smacks hibernating session for up to %d seconds", resume_timeout);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	271 local hibernate_time = os_time(); -- Track the time we went into hibernation
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	272 session.hibernating = hibernate_time;
478 db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	273 local resumption_token = session.resumption_token;
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	274 timer.add_task(resume_timeout, function ()
586 f733e7599ed6 mod_smacks: Add logging to hibernation and session destruction (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 576 diff changeset	275 session.log("debug", "mod_smacks hibernation timeout reached...");
478 db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	276 -- We need to check the current resumption token for this resource
db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	277 -- matches the smacks session this timer is for in case it changed
db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	278 -- (for example, the client may have bound a new resource and
db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	279 -- started a new smacks session, or not be using smacks)
810 464ed063a8f2 mod_smacks: Less table indexing! Kim Alvefur <zash@zash.se> parents: 757 diff changeset	280 local curr_session = full_sessions[session.full_jid];
1598 043d08448d87 mod_smacks: Remove negation of condition, most likely a leftover from debugging Kim Alvefur <zash@zash.se> parents: 1597 diff changeset	281 if session.destroyed then
987 fabff75bfc3f mod_smacks: If a hibernating session was destroyed before the timeout, don't destroy it again or say that it was resumed Kim Alvefur <zash@zash.se> parents: 925 diff changeset	282 session.log("debug", "The session has already been destroyed");
fabff75bfc3f mod_smacks: If a hibernating session was destroyed before the timeout, don't destroy it again or say that it was resumed Kim Alvefur <zash@zash.se> parents: 925 diff changeset	283 elseif curr_session and curr_session.resumption_token == resumption_token
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	284 -- Check the hibernate time still matches what we think it is,
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	285 -- otherwise the session resumed and re-hibernated.
478 db0f065c4e09 mod_smacks: Don't destroy a session that binds the same resource as a hibernating smacks session (thanks xnyhps for tracking down the problem, though I've used a different fix) Matthew Wild <mwild1@gmail.com> parents: 345 diff changeset	286 and session.hibernating == hibernate_time then
586 f733e7599ed6 mod_smacks: Add logging to hibernation and session destruction (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 576 diff changeset	287 session.log("debug", "Destroying session for hibernating too long");
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	288 session_registry[session.resumption_token] = nil;
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	289 session.resumption_token = nil;
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	290 sessionmanager.destroy_session(session);
586 f733e7599ed6 mod_smacks: Add logging to hibernation and session destruction (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 576 diff changeset	291 else
f733e7599ed6 mod_smacks: Add logging to hibernation and session destruction (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 576 diff changeset	292 session.log("debug", "Session resumed before hibernation timeout, all is well")
257 08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	293 end
08fa42e1ab06 mod_smacks: Fixes for monkey-patched sessionmanager.destroy to handle stream resumption, and to fall back to stock destroy() if the session is not smacks-enabled. Matthew Wild <mwild1@gmail.com> parents: 256 diff changeset	294 end);
622 ce39df945de1 mod_smacks: Use pre-resource-unbind (0.9 feature) to fix reloadability and multiple host support Matthew Wild <mwild1@gmail.com> parents: 595 diff changeset	295 return true; -- Postpone destruction for now
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	296 end
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	297
160 9a7671720dec mod_smacks: XEP-0198 Stream Management acks. Initial commit - very rough, useful mainly for testing at the moment, most certainly contains bugs :) Matthew Wild <mwild1@gmail.com> parents: diff changeset	298 end
622 ce39df945de1 mod_smacks: Use pre-resource-unbind (0.9 feature) to fix reloadability and multiple host support Matthew Wild <mwild1@gmail.com> parents: 595 diff changeset	299 end);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	300
1732 2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	301 local function handle_s2s_destroyed(event)
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	302 local session = event.session;
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	303 local queue = session.outgoing_stanza_queue;
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	304 if queue and #queue > 0 then
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	305 session.log("warn", "Destroying session with %d unacked stanzas", #queue);
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	306 handle_unacked_stanzas(session);
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	307 end
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	308 end
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	309
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	310 module:hook("s2sout-destroyed", handle_s2s_destroyed);
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	311 module:hook("s2sin-destroyed", handle_s2s_destroyed);
2f9ee9ed6267 mod_smacks: Handle unacked messages in s2s queues Kim Alvefur <zash@zash.se> parents: 1706 diff changeset	312
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	313 function handle_resume(session, stanza, xmlns_sm)
925 720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	314 if session.full_jid then
994 487cd02aada0 mod_smacks: Complain a little louder about clients trying to resume after resource binding Kim Alvefur <zash@zash.se> parents: 988 diff changeset	315 session.log("warn", "Tried to resume after resource binding");
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	316 session.send(st.stanza("failed", { xmlns = xmlns_sm })
925 720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	317 :tag("unexpected-request", { xmlns = xmlns_errors })
720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	318 );
720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	319 return true;
720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	320 end
720b8268778e mod_smacks: Don't allow steam resumption onto a stream after resource binding Kim Alvefur <zash@zash.se> parents: 908 diff changeset	321
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	322 local id = stanza.attr.previd;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	323 local original_session = session_registry[id];
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	324 if not original_session then
588 8042558336b6 mod_smacks: Log message when client tries to resume unknown session Matthew Wild <mwild1@gmail.com> parents: 587 diff changeset	325 session.log("debug", "Tried to resume non-existent session with id %s", id);
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	326 session.send(st.stanza("failed", { xmlns = xmlns_sm })
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	327 :tag("item-not-found", { xmlns = xmlns_errors })
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	328 );
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	329 elseif session.username == original_session.username
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	330 and session.host == original_session.host then
587 322a14acd974 mod_smacks: Add log message on resume Matthew Wild <mwild1@gmail.com> parents: 586 diff changeset	331 session.log("debug", "mod_smacks resuming existing session...");
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	332 -- TODO: All this should move to sessionmanager (e.g. session:replace(new_session))
623 c1f3958695ea mod_smacks: If a resumed session still has a connection open, close that connection before resuming Matthew Wild <mwild1@gmail.com> parents: 622 diff changeset	333 if original_session.conn then
c1f3958695ea mod_smacks: If a resumed session still has a connection open, close that connection before resuming Matthew Wild <mwild1@gmail.com> parents: 622 diff changeset	334 session.log("debug", "mod_smacks closing an old connection for this session");
c1f3958695ea mod_smacks: If a resumed session still has a connection open, close that connection before resuming Matthew Wild <mwild1@gmail.com> parents: 622 diff changeset	335 local conn = original_session.conn;
640 d87131b29bbd mod_smacks: Remove dependency on connlisteners (use sessions table shared by mod_c2s directly) Matthew Wild <mwild1@gmail.com> parents: 625 diff changeset	336 c2s_sessions[conn] = nil;
623 c1f3958695ea mod_smacks: If a resumed session still has a connection open, close that connection before resuming Matthew Wild <mwild1@gmail.com> parents: 622 diff changeset	337 conn:close();
c1f3958695ea mod_smacks: If a resumed session still has a connection open, close that connection before resuming Matthew Wild <mwild1@gmail.com> parents: 622 diff changeset	338 end
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	339 original_session.ip = session.ip;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	340 original_session.conn = session.conn;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	341 original_session.send = session.send;
1637 ad6694f7b13c mod_smacks: Don't restore filters from original session after resumption, use new filters Matthew Wild <mwild1@gmail.com> parents: 1599 diff changeset	342 original_session.filter = session.filter;
1706 e4867211cddb mod_smacks: Set session upvalue of filter() to correct session Kim Alvefur <zash@zash.se> parents: 1705 diff changeset	343 original_session.filter.session = original_session;
1705 6fa220a9f36d mod_smacks: Move set of filters from new session to session being resumed Kim Alvefur <zash@zash.se> parents: 1704 diff changeset	344 original_session.filters = session.filters;
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	345 original_session.stream = session.stream;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	346 original_session.secure = session.secure;
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	347 original_session.hibernating = nil;
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	348 wrap_session(original_session, true);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	349 -- Inform xmppstream of the new session (passed to its callbacks)
1520 2881d532f385 mod_smacks: Use filters for queuing outgoing stanzas instead of wrapping session.send() Kim Alvefur <zash@zash.se> parents: 1518 diff changeset	350 original_session.stream:set_session(original_session);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	351 -- Similar for connlisteners
640 d87131b29bbd mod_smacks: Remove dependency on connlisteners (use sessions table shared by mod_c2s directly) Matthew Wild <mwild1@gmail.com> parents: 625 diff changeset	352 c2s_sessions[session.conn] = original_session;
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	353
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	354 session.send(st.stanza("resumed", { xmlns = xmlns_sm,
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	355 h = original_session.handled_stanza_count, previd = id }));
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	356
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	357 -- Fake an <a> with the h of the <resume/> from the client
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	358 original_session:dispatch_stanza(st.stanza("a", { xmlns = xmlns_sm,
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	359 h = stanza.attr.h }));
1343 7dbde05b48a9 all the things: Remove trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 1325 diff changeset	360
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	361 -- Ok, we need to re-send any stanzas that the client didn't see
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	362 -- ...they are what is now left in the outgoing stanza queue
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	363 local queue = original_session.outgoing_stanza_queue;
1406 7d76dd2310ef mod_smacks: Add more debug logging Kim Alvefur <zash@zash.se> parents: 1405 diff changeset	364 session.log("debug", "#queue = %d", #queue);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	365 for i=1,#queue do
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	366 session.send(queue[i]);
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	367 end
1406 7d76dd2310ef mod_smacks: Add more debug logging Kim Alvefur <zash@zash.se> parents: 1405 diff changeset	368 session.log("debug", "#queue = %d -- after send", #queue);
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	369 else
755 bab7c4ace803 mod_smacks: Fix log statement Kim Alvefur <zash@zash.se> parents: 754 diff changeset	370 module:log("warn", "Client %s@%s[%s] tried to resume stream for %s@%s[%s]",
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	371 session.username or "?", session.host or "?", session.type,
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	372 original_session.username or "?", original_session.host or "?", original_session.type);
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	373 session.send(st.stanza("failed", { xmlns = xmlns_sm })
345 445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	374 :tag("not-authorized", { xmlns = xmlns_errors }));
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	375 end
445178d15b51 mod_smacks: Merge mod_fastreconnect (resumption support), fix a number of bugs, refactor the code and add some more comments to explain process Matthew Wild <mwild1@gmail.com> parents: 263 diff changeset	376 return true;
1298 659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	377 end
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	378 module:hook_stanza(xmlns_sm2, "resume", function (session, stanza) return handle_resume(session, stanza, xmlns_sm2); end);
659da45a2b4b mod_smacks: Handle both version 2 and version 3 namespace Florian Zeitz <florob@babelmonkeys.de> parents: 1293 diff changeset	379 module:hook_stanza(xmlns_sm3, "resume", function (session, stanza) return handle_resume(session, stanza, xmlns_sm3); end);
1733 9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	380
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	381 local function handle_read_timeout(event)
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	382 local session = event.session;
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	383 if session.smacks then
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	384 if session.awaiting_ack then
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	385 return false; -- Kick the session
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	386 end
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	387 (session.sends2s or session.send)(st.stanza("r", { xmlns = session.smacks }));
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	388 session.awaiting_ack = true;
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	389 return true;
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	390 end
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	391 end
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	392
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	393 module:hook("s2s-read-timeout", handle_read_timeout);
9abd3dce619a mod_smacks: Handle the read timeout event (0.10+ only) and send an ack request, close the connection the second time if unanswerd Kim Alvefur <zash@zash.se> parents: 1732 diff changeset	394 module:hook("c2s-read-timeout", handle_read_timeout);

prosody-modules

6a9f91ad4f95

mod_smacks/mod_smacks.lua

Software / code / prosody-modules

Annotate

mod_smacks/mod_smacks.lua @ 1870:6a9f91ad4f95

Software `/` code `/` prosody-modules