view mod_proxy65_whitelist/mod_proxy65_whitelist.lua @ 2349:48686b399a88

mod_proxy65_whitelist: Add proxy hosts enabled after module is loaded (fixes #494)
author Kim Alvefur <zash@zash.se>
date Mon, 07 Nov 2016 17:35:11 +0100
parents 4ef0a1a499fa
children 67990e045d4f
line wrap: on
line source

local allowed_streamhosts = module:get_option_set("allowed_streamhosts", {}); -- eg proxy.eu.jabber.org

if module:get_option_boolean("allow_local_streamhosts", true) then
	for hostname, host in pairs(hosts) do
		if host.modules.proxy65 then
			allowed_streamhosts:add(hostname);
		end
	end
end

module:hook_global("host-activated", function (host)
	if hosts[host].modules.proxy65 then
		allowed_streamhosts:add(host);
	end
end);

local function filter_streamhosts(tag)
	if tag.name == "streamhost" and not allowed_streamhosts:contains(tag.attr.jid) then
		return nil;
	end
	return tag;
end

module:hook("iq/full", function (event)
	local stanza, origin = event.stanza, event.origin;
	if stanza.attr.type == "set" then
		local payload = stanza:get_child("query", "http://jabber.org/protocol/bytestreams");
		if payload then
			payload:maptags(filter_streamhosts);
		end
	end
end, 1);