File

plugins/mod_iq.lua @ 1268:dc1f95b37024

mod_iq: Correctly handle the lack of 'to' on IQs
author Waqas Hussain <waqas20@gmail.com>
date Mon, 01 Jun 2009 11:16:48 +0500
parent 1267:1bf897de6c24
child 1288:d2dc0954ebfd
line wrap: on
line source


local st = require "util.stanza";
local jid_split = require "util.jid".split;
local user_exists = require "core.usermanager".user_exists;

local full_sessions = full_sessions;
local bare_sessions = bare_sessions;

module:hook("iq/full", function(data)
	-- IQ to full JID recieved
	local origin, stanza = data.origin, data.stanza;

	local session = full_sessions[stanza.attr.to];
	if session then
		-- TODO fire post processing event
		session.send(stanza);
	else -- resource not online
		if stanza.attr.type == "get" or stanza.attr.type == "set" then
			origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
		end
	end
	return true;
end);

module:hook("iq/bare", function(data)
	-- IQ to bare JID recieved
	local origin, stanza = data.origin, data.stanza;

	local to = stanza.attr.to;
	if to and not bare_sessions[to] then -- quick check for account existance
		local node, host = jid_split(to);
		if not user_exists(node, host) then -- full check for account existance
			if stanza.attr.type == "get" or stanza.attr.type == "set" then
				origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
			end
			return true;
		end
	end
	-- TODO fire post processing events
	if #stanza.tags == 1 then
		return module:fire_event("iq/bare/"..stanza.tags[1].attr.xmlns..":"..stanza.tags[1].name, data);
	else
		return true; -- TODO do something with results and errors
	end
end);

module:hook("iq/host", function(data)
	-- IQ to a local host recieved
	local origin, stanza = data.origin, data.stanza;

	if #stanza.tags == 1 then
		return module:fire_event("iq/host/"..stanza.tags[1].attr.xmlns..":"..stanza.tags[1].name, data);
	else
		return true; -- TODO do something with results and errors
	end
end);