Diff

mod_smacks/mod_smacks.lua @ 2624:c110b6bfe5d1

mod_smacks: Prevent ack loop on misbehaving clients
author tmolitor <thilo@eightysoft.de>
date Wed, 15 Mar 2017 16:24:03 +0100
parent 2623:a65260300708
child 2670:6e01878103c0
line wrap: on
line diff
--- a/mod_smacks/mod_smacks.lua	Wed Mar 15 16:21:55 2017 +0100
+++ b/mod_smacks/mod_smacks.lua	Wed Mar 15 16:24:03 2017 +0100
@@ -138,33 +138,36 @@
 			end
 		end);
 
-local function request_ack_if_needed(session)
+local function request_ack_if_needed(session, force)
 	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 = stoppable_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;
-				if not session.delayed_ack_timer then
-					session.delayed_ack_timer = stoppable_timer(delayed_ack_timeout, function()
-						delayed_ack_function(session);
-					end);
+	if session.awaiting_ack == nil then
+		if (#queue > max_unacked_stanzas and session.last_queue_count ~= #queue) or force then
+			session.log("debug", "Queuing <r> (in a moment)");
+			session.awaiting_ack = false;
+			session.awaiting_ack_timer = stoppable_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;
+					if not session.delayed_ack_timer then
+						session.delayed_ack_timer = stoppable_timer(delayed_ack_timeout, function()
+							delayed_ack_function(session);
+						end);
+					end
 				end
-			end
-		end);
+			end);
+		end
+		-- Trigger "smacks-ack-delayed"-event if we added new (ackable) stanzas to the outgoing queue
+		-- and there isn't already a timer for this event running.
+		-- If we wouldn't do this, stanzas added to the queue after the first "smacks-ack-delayed"-event
+		-- would not trigger this event (again).
+		if #queue > max_unacked_stanzas and session.awaiting_ack and session.delayed_ack_timer == nil then
+			session.log("debug", "Calling delayed_ack_function directly (still waiting for ack)");
+			delayed_ack_function(session);
+		end
 	end
-	-- Trigger "smacks-ack-delayed"-event if we added new (ackable) stanzas to the outgoing queue
-	-- and there isn't already a timer for this event running.
-	-- If we wouldn't do this, stanzas added to the queue after the first "smacks-ack-delayed"-event
-	-- would not trigger this event (again).
-	if #queue > max_unacked_stanzas and session.awaiting_ack and session.delayed_ack_timer == nil then
-		session.log("debug", "Calling delayed_ack_function directly (still waiting for ack)");
-		delayed_ack_function(session);
-	end
+	session.last_queue_count = #queue;
 end
 
 local function outgoing_stanza_filter(stanza, session)
@@ -184,7 +187,7 @@
 			session.log("debug", "hibernating, stanza queued");
 			return nil;
 		end
-		request_ack_if_needed(session);
+		request_ack_if_needed(session, false);
 	end
 	return stanza;
 end
@@ -337,7 +340,7 @@
 	end
 	origin.log("debug", "#queue = %d", #queue);
 	origin.last_acknowledged_stanza = origin.last_acknowledged_stanza + handled_stanza_count;
-	request_ack_if_needed(origin)
+	request_ack_if_needed(origin, false)
 	return true;
 end
 module:hook_stanza(xmlns_sm2, "a", handle_a);
@@ -502,6 +505,7 @@
 			session.log("warn", "Tried to send stanza on old session migrated by smacks resume (maybe there is a bug?): %s", tostring(stanza));
 			return false;
 		end
+		request_ack_if_needed(original_session, true);
 	else
 		module:log("warn", "Client %s@%s[%s] tried to resume stream for %s@%s[%s]",
 			session.username or "?", session.host or "?", session.type,