mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-05 14:17:37 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_component: Updated to use events for hooking stanzas instead of the component stanza handler, and the on_destroy callback.

Browse source
This commit is contained in:
Waqas Hussain 2010-11-10 01:51:03 +05:00
parent 87615e95fc
commit cac4ec8aca
1 changed files with 41 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

49
plugins/mod_component.lua
View file

@ -22,11 +22,48 @@ local st = require "util.stanza";
local log = module._log; local log = module._log;
local main_session, send;
local function on_destroy(session, err)
if main_session == session then
main_session = nil;
send = nil;
session.on_destroy = nil;
hosts[session.host].connected = nil;
end
end
local function handle_stanza(event)
local stanza = event.stanza;
if send then
stanza.attr.xmlns = nil;
send(stanza);
else
log("warn", "Stanza being handled by default component; bouncing error for: %s", stanza:top_tag());
if stanza.attr.type ~= "error" and stanza.attr.type ~= "result" then
event.origin.send(st.error_reply(stanza, "wait", "service-unavailable", "Component unavailable"));
end
end
end
module:hook("iq/bare", handle_stanza);
module:hook("message/bare", handle_stanza);
module:hook("presence/bare", handle_stanza);
module:hook("iq/full", handle_stanza);
module:hook("message/full", handle_stanza);
module:hook("presence/full", handle_stanza);
module:hook("iq/host", handle_stanza);
module:hook("message/host", handle_stanza);
module:hook("presence/host", handle_stanza);
cm_register_component(module.host, function() end);
--- Handle authentication attempts by components --- Handle authentication attempts by components
function handle_component_auth(event) function handle_component_auth(event)
local session, stanza = event.origin, event.stanza; local session, stanza = event.origin, event.stanza;
if session.type ~= "component" then return; end if session.type ~= "component" then return; end
if main_session == session then return; end
log("info", "Handling component auth"); log("info", "Handling component auth");
if (not session.host) or #stanza.tags > 0 then if (not session.host) or #stanza.tags > 0 then
@ -57,14 +94,10 @@ function handle_component_auth(event)
session.component_validate_from = module:get_option_boolean("validate_from_addresses") ~= false; session.component_validate_from = module:get_option_boolean("validate_from_addresses") ~= false;
-- If component not already created for this host, create one now -- If component not already created for this host, create one now
if not hosts[session.host].connected then if not main_session then
local send = session.send; send = session.send;
session.component_session = cm_register_component(session.host, function (_, data) main_session = session;
if data.attr and data.attr.xmlns == "jabber:client" then session.on_destroy = on_destroy;
data.attr.xmlns = nil;
end
return send(data);
end);
hosts[session.host].connected = true; hosts[session.host].connected = true;
log("info", "Component successfully registered"); log("info", "Component successfully registered");
else else