File

+
−-- luacheck: ignore 122/prosody 113/prosody
+
−
+
−local timer = require "util.timer";
+
−local portman = require "core.portmanager";
+
−local server = require "net.server";
+
−
+
−module:set_global();
+
−local orig_shutdown = prosody.shutdown;
+
−
+
−local pause = module:get_option_number("shutdown_pause", 1);
+
−
+
−function module.unload()
+
−	prosody.shutdown = orig_shutdown;
+
−end
+
−
+
−prosody.shutdown = coroutine.wrap(function (reason, code)
+
−	prosody.shutdown_reason = reason;
+
−	prosody.shutdown_code = code;
+
−	timer.add_task(pause, prosody.shutdown);
+
−	coroutine.yield(true, "shutdown initiated");
+
−	-- Close c2s ports, stop accepting new connections
+
−	portman.deactivate("c2s");
+
−	portman.deactivate("legacy_ssl");
+
−	portman.deactivate("c2s_direct_tls");
+
−
+
−	-- Close multiplexing ports to ensure c2s is not reachable via those either
+
−	portman.deactivate("multiplex");
+
−	portman.deactivate("multiplex_ssl");
+
−	portman.deactivate("proxy"); -- mod_net_proxy
+
−
+
−	-- Close all c2s sessions
+
−	for _, sess in pairs(prosody.full_sessions) do
+
−		sess:close{ condition = "system-shutdown", text = reason }
+
−	end
+
−	-- Wait for notifications to be sent
+
−	coroutine.yield(pause);
+
−	-- Event for everything else to shut down
+
−	prosody.events.fire_event("server-stopping", {
+
−		reason = reason;
+
−		code = code;
+
−	});
+
−	-- And wait
+
−	coroutine.yield(pause);
+
−	-- And stop main event loop
+
−	server.setquitting(true);
+
−	-- And wait for death
+
−	coroutine.yield(pause * 3);
+
−	-- you came back? die zombie!
+
−	os.exit(1);
+
−end);