# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1639753208 -3600
# Node ID d2380fd5e421864221df5cf038f0d5308b753978
# Parent  d308f690139733f3ca5e7b2c24a32646c2f7cb54
mod_smacks: Preserve counter values on shutdown

diff -r d308f6901397 -r d2380fd5e421 plugins/mod_smacks.lua
--- a/plugins/mod_smacks.lua	Fri Dec 17 15:59:06 2021 +0100
+++ b/plugins/mod_smacks.lua	Fri Dec 17 16:00:08 2021 +0100
@@ -633,3 +633,25 @@
 
 module:hook("s2s-read-timeout", handle_read_timeout);
 module:hook("c2s-read-timeout", handle_read_timeout);
+
+module:hook_global("server-stopping", function()
+	-- Close smacks-enaled sessions ourselves instead of letting mod_c2s close
+	-- it, which invalidates the smacks session. This allows preserving the
+	-- counter value, so it can be communicated to the client when it tries to
+	-- resume the lost session after a restart.
+	for _, user in pairs(local_sessions) do
+		for _, session in pairs(user.sessions) do
+			if session.resumption_token then
+				if old_session_registry:set(session.username, session.resumption_token, { h = session.handled_stanza_count }) then
+					session.resumption_token = nil;
+					if session.conn then
+						session.conn:close()
+						session.conn = nil;
+						-- Now when mod_c2s gets here, it will immediately destroy the
+						-- session since it is unconnected.
+					end
+				end
+			end
+		end
+	end
+end, -90);