Mercurial > lxmppd-hg-xmllex / file comparison
comparison: core/componentmanager.lua
core/componentmanager.lua
- changeset 3603
- fdc671381411
- parent 3602
- a977d8be6277
equal
deleted
inserted
replaced
| 3602:a977d8be6277 | 3603:fdc671381411 |
|---|---|
| 4 -- | 4 -- |
| 5 -- This project is MIT/X11 licensed. Please see the | 5 -- This project is MIT/X11 licensed. Please see the |
| 6 -- COPYING file in the source package for more information. | 6 -- COPYING file in the source package for more information. |
| 7 -- | 7 -- |
| 8 | 8 |
| 9 local prosody = _G.prosody; | |
| 10 local log = require "util.logger".init("componentmanager"); | 9 local log = require "util.logger".init("componentmanager"); |
| 11 local certmanager = require "core.certmanager"; | |
| 12 local configmanager = require "core.configmanager"; | |
| 13 local modulemanager = require "core.modulemanager"; | |
| 14 local jid_split = require "util.jid".split; | |
| 15 local fire_event = prosody.events.fire_event; | |
| 16 local events_new = require "util.events".new; | |
| 17 local st = require "util.stanza"; | |
| 18 local prosody, hosts = prosody, prosody.hosts; | 10 local prosody, hosts = prosody, prosody.hosts; |
| 19 local uuid_gen = require "util.uuid".generate; | |
| 20 | |
| 21 local pairs, setmetatable, type, tostring = pairs, setmetatable, type, tostring; | |
| 22 | 11 |
| 23 local components = {}; | 12 local components = {}; |
| 24 | 13 |
| 25 module "componentmanager" | 14 module "componentmanager" |
| 26 | 15 |
| 27 local function default_component_handler(origin, stanza) | |
| 28 log("warn", "Stanza being handled by default component; bouncing error for: %s", stanza:top_tag()); | |
| 29 if stanza.attr.type ~= "error" and stanza.attr.type ~= "result" then | |
| 30 origin.send(st.error_reply(stanza, "wait", "service-unavailable", "Component unavailable")); | |
| 31 end | |
| 32 end | |
| 33 | |
| 34 function create_component(host, component, events) | |
| 35 -- TODO check for host well-formedness | |
| 36 return { type = "component", host = host, s2sout = {}, | |
| 37 events = events or events_new(), | |
| 38 dialback_secret = configmanager.get(host, "core", "dialback_secret") or uuid_gen(), | |
| 39 disallow_s2s = configmanager.get(host, "core", "disallow_s2s"); }; | |
| 40 end | |
| 41 | |
| 42 function register_component(host, component) | 16 function register_component(host, component) |
| 43 if not hosts[host] or hosts[host].type == 'component' then | 17 if hosts[host] and hosts[host].type == 'component' then |
| 44 local old_events = hosts[host] and hosts[host].events; | |
| 45 | |
| 46 components[host] = component; | 18 components[host] = component; |
| 47 hosts[host] = create_component(host, component, old_events); | |
| 48 | |
| 49 -- Add events object if not already one | |
| 50 if not hosts[host].events then | |
| 51 hosts[host].events = old_events or events_new(); | |
| 52 end | |
| 53 | |
| 54 if not hosts[host].dialback_secret then | |
| 55 hosts[host].dialback_secret = configmanager.get(host, "core", "dialback_secret") or uuid_gen(); | |
| 56 end | |
| 57 | |
| 58 modulemanager.load(host, "dialback"); | |
| 59 modulemanager.load(host, "tls"); | |
| 60 log("debug", "component added: "..host); | 19 log("debug", "component added: "..host); |
| 61 return hosts[host]; | 20 return hosts[host]; |
| 62 else | 21 else |
| 63 log("error", "Attempt to set component for existing host: "..host); | 22 log("error", "Attempt to set component for existing host: "..host); |
| 64 end | 23 end |
| 65 end | 24 end |
| 66 | 25 |
| 67 function deregister_component(host) | 26 function deregister_component(host) |
| 68 if components[host] then | 27 if components[host] then |
| 69 modulemanager.unload(host, "tls"); | 28 components[host] = nil; |
| 70 modulemanager.unload(host, "dialback"); | |
| 71 local host_config = configmanager.getconfig()[host]; | |
| 72 if host_config and ((host_config.core.enabled == nil or host_config.core.enabled) and type(host_config.core.component_module) == "string") then | |
| 73 -- Set default handler | |
| 74 components[host] = default_component_handler; | |
| 75 else | |
| 76 -- Component not in config, or disabled, remove | |
| 77 hosts[host] = nil; -- FIXME do proper unload of all modules and other cleanup before removing | |
| 78 components[host] = nil; | |
| 79 end | |
| 80 log("debug", "component removed: "..host); | 29 log("debug", "component removed: "..host); |
| 81 return true; | 30 return true; |
| 82 else | 31 else |
| 83 log("error", "Attempt to remove component for non-existing host: "..host); | 32 log("error", "Attempt to remove component for non-existing host: "..host); |
| 84 end | 33 end |
