Software / code / prosody-modules
Diff
mod_smacks/mod_smacks.lua @ 2491:5fbca7de2088
mod_smacks: Send out more ack requests where needed
Under some circumstances it was possible that more than "max_unacked_stanzas"
where left in the outgoing stanza queue without forcing an ack.
This could happen, when more stanzas entered the queue while the last ack request
was still unanswered.
Now the test "#queue > max_unacked_stanzas" is done upon receiving
an ack as well as when sending out stanzas, which fixes this bug.
| author | tmolitor <thilo@eightysoft.de> |
|---|---|
| date | Sun, 12 Feb 2017 19:27:50 +0100 |
| parent | 2417:5e7badecf7fe |
| child | 2494:d300ae5dba87 |
line wrap: on
line diff
--- a/mod_smacks/mod_smacks.lua Fri Feb 10 22:50:45 2017 +0100 +++ b/mod_smacks/mod_smacks.lua Sun Feb 12 19:27:50 2017 +0100 @@ -5,7 +5,7 @@ -- Copyright (C) 2012-2015 Kim Alvefur -- Copyright (C) 2012 Thijs Alkemade -- Copyright (C) 2014 Florian Zeitz --- Copyright (C) 2016 Thilo Molitor +-- Copyright (C) 2016-2017 Thilo Molitor -- -- This project is MIT/X11 licensed. Please see the -- COPYING file in the source package for more information. @@ -80,6 +80,25 @@ end end); +local function request_ack_if_needed(session) + local queue = session.outgoing_stanza_queue; + if #queue > max_unacked_stanzas and session.awaiting_ack == nil then + session.log("debug", "Queuing <r> (in a moment)"); + session.awaiting_ack = false; + session.awaiting_ack_timer = module:add_timer(1e-06, function () + if not session.awaiting_ack then + session.log("debug", "Sending <r> (inside timer, before send)"); + (session.sends2s or session.send)(st.stanza("r", { xmlns = session.smacks })) + session.log("debug", "Sending <r> (inside timer, after send)"); + session.awaiting_ack = true; + session.delayed_ack_timer = module:add_timer(delayed_ack_timeout, function() + delayed_ack_function(session); + end); + end + end); + end +end + local function outgoing_stanza_filter(stanza, session) local is_stanza = stanza.attr and not stanza.attr.xmlns and not stanza.name:find":"; if is_stanza and not stanza._cached then -- Stanza in default stream namespace @@ -97,21 +116,7 @@ session.log("debug", "hibernating, stanza queued"); return nil; end - if #queue > max_unacked_stanzas and session.awaiting_ack == nil then - session.log("debug", "Queuing <r> (in a moment)"); - session.awaiting_ack = false; - session.awaiting_ack_timer = module:add_timer(1e-06, function () - if not session.awaiting_ack then - session.log("debug", "Sending <r> (before send)"); - (session.sends2s or session.send)(st.stanza("r", { xmlns = session.smacks })) - session.log("debug", "Sending <r> (after send)"); - session.awaiting_ack = true; - session.delayed_ack_timer = module:add_timer(delayed_ack_timeout, function() - delayed_ack_function(session); - end); - end - end); - end + request_ack_if_needed(session); end return stanza; end @@ -238,7 +243,7 @@ origin.delayed_ack_timer:stop(); end -- Remove handled stanzas from outgoing_stanza_queue - --log("debug", "ACK: h=%s, last=%s", stanza.attr.h or "", origin.last_acknowledged_stanza or ""); + log("debug", "ACK: h=%s, last=%s", stanza.attr.h or "", origin.last_acknowledged_stanza or ""); local h = tonumber(stanza.attr.h); if not h then origin:close{ condition = "invalid-xml"; text = "Missing or invalid 'h' attribute"; }; @@ -258,6 +263,7 @@ end origin.log("debug", "#queue = %d", #queue); origin.last_acknowledged_stanza = origin.last_acknowledged_stanza + handled_stanza_count; + request_ack_if_needed(origin) return true; end module:hook_stanza(xmlns_sm2, "a", handle_a);
