diff mod_host_blacklist/mod_host_blacklist.lua @ 1180:513aa2e0c045

mod_host_blacklist: Similar to mod_s2s_blacklist, but stanza-based. Works between hosts on the same server, at a slight performance impact.
author Matthew Wild <mwild1@gmail.com>
date Tue, 03 Sep 2013 11:07:00 +0100
parents
children 005b0429cf46
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_host_blacklist/mod_host_blacklist.lua	Tue Sep 03 11:07:00 2013 +0100
@@ -0,0 +1,25 @@
+local jid_split = require "util.jid".split;
+local st = require "util.stanza";
+local set = require "util.set";
+local select = select;
+
+local blacklist = module:get_option_inherited_set("host_blacklist", {});
+
+local function stanza_checker(attr)
+	return function (event)
+		local host = select(2, jid_split(stanza.attr[attr]));
+		if blacklist:contains(host) then
+			module:send(st.error_reply(event.stanza, "cancel", "not-allowed", "Communication with this domain is restricted"));
+		end
+	end
+end
+
+check_incoming_stanza = stanza_checker("from");
+check_outgoing_stanza = stanza_checker("to");
+
+for stanza_type in set.new{"presence","message","iq"}:items() do
+	for jid_type in set.new{"bare", "full", "host"}:items() do
+		module:hook("pre-"..stanza_type.."/"..jid_type, check_outgoing_stanza, 500);
+		module:hook(stanza_type.."/"..jid_type, check_incoming_stanza, 500);
+	end
+end