core/hostmanager.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9 local configmanager = require "core.configmanager";
  10 local modulemanager = require "core.modulemanager";
  11 local events_new = require "util.events".new;
  12 local disco_items = require "util.multitable".new();
  13 local NULL = {};
  14
  15 local jid_split = require "util.jid".split;
  16 local uuid_gen = require "util.uuid".generate;
  17
  18 local log = require "util.logger".init("hostmanager");
  19
  20 local hosts = prosody.hosts;
  21 local prosody_events = prosody.events;
  22 if not _G.prosody.incoming_s2s then
  23         require "core.s2smanager";
  24 end
  25 local incoming_s2s = _G.prosody.incoming_s2s;
  26 local core_route_stanza = _G.prosody.core_route_stanza;
  27
  28 local pairs, select, rawget = pairs, select, rawget;
  29 local tostring, type = tostring, type;
  30
  31 module "hostmanager"
  32
  33 local hosts_loaded_once;
  34
  35 local function load_enabled_hosts(config)
  36         local defined_hosts = config or configmanager.getconfig();
  37         local activated_any_host;
  38
  39         for host, host_config in pairs(defined_hosts) do
  40                 if host ~= "*" and host_config.enabled ~= false then
  41                         if not host_config.component_module then
  42                                 activated_any_host = true;
  43                         end
  44                         activate(host, host_config);
  45                 end
  46         end
  47
  48         if not activated_any_host then
  49                 log("error", "No active VirtualHost entries in the config file. This may cause unexpected behaviour as no modules will be loaded.");
  50         end
  51
  52         prosody_events.fire_event("hosts-activated", defined_hosts);
  53         hosts_loaded_once = true;
  54 end
  55
  56 prosody_events.add_handler("server-starting", load_enabled_hosts);
  57
  58 local function host_send(stanza)
  59         core_route_stanza(nil, stanza);
  60 end
  61
  62 function activate(host, host_config)
  63         if rawget(hosts, host) then return nil, "The host "..host.." is already activated"; end
  64         host_config = host_config or configmanager.getconfig()[host];
  65         if not host_config then return nil, "Couldn't find the host "..tostring(host).." defined in the current config"; end
  66         local host_session = {
  67                 host = host;
  68                 s2sout = {};
  69                 events = events_new();
  70                 send = host_send;
  71                 modules = {};
  72         };
  73         if not host_config.component_module then -- host
  74                 host_session.type = "local";
  75                 host_session.sessions = {};
  76         else -- component
  77                 host_session.type = "component";
  78         end
  79         hosts[host] = host_session;
  80         if not host:match("[@/]") then
  81                 disco_items:set(host:match("%.(.*)") or "*", host, host_config.name or true);
  82         end
  83         for option_name in pairs(host_config) do
  84                 if option_name:match("_ports$") or option_name:match("_interface$") then
  85                         log("warn", "%s: Option '%s' has no effect for virtual hosts - put it in the server-wide section instead", host, option_name);
  86                 end
  87         end
  88
  89         log((hosts_loaded_once and "info") or "debug", "Activated host: %s", host);
  90         prosody_events.fire_event("host-activated", host);
  91         return true;
  92 end
  93
  94 function deactivate(host, reason)
  95         local host_session = hosts[host];
  96         if not host_session then return nil, "The host "..tostring(host).." is not activated"; end
  97         log("info", "Deactivating host: %s", host);
  98         prosody_events.fire_event("host-deactivating", { host = host, host_session = host_session, reason = reason });
  99
 100         if type(reason) ~= "table" then
 101                 reason = { condition = "host-gone", text = tostring(reason or "This server has stopped serving "..host) };
 102         end
 103
 104         -- Disconnect local users, s2s connections
 105         -- TODO: These should move to mod_c2s and mod_s2s (how do they know they're being unloaded and not reloaded?)
 106         if host_session.sessions then
 107                 for username, user in pairs(host_session.sessions) do
 108                         for resource, session in pairs(user.sessions) do
 109                                 log("debug", "Closing connection for %s@%s/%s", username, host, resource);
 110                                 session:close(reason);
 111                         end
 112                 end
 113         end
 114         if host_session.s2sout then
 115                 for remotehost, session in pairs(host_session.s2sout) do
 116                         if session.close then
 117                                 log("debug", "Closing outgoing connection to %s", remotehost);
 118                                 if session.srv_hosts then session.srv_hosts = nil; end
 119                                 session:close(reason);
 120                         end
 121                 end
 122         end
 123         for remote_session in pairs(incoming_s2s) do
 124                 if remote_session.to_host == host then
 125                         log("debug", "Closing incoming connection from %s", remote_session.from_host or "<unknown>");
 126                         remote_session:close(reason);
 127                 end
 128         end
 129
 130         -- TODO: This should be done in modulemanager
 131         if host_session.modules then
 132                 for module in pairs(host_session.modules) do
 133                         modulemanager.unload(host, module);
 134                 end
 135         end
 136
 137         hosts[host] = nil;
 138         if not host:match("[@/]") then
 139                 disco_items:remove(host:match("%.(.*)") or "*", host);
 140         end
 141         prosody_events.fire_event("host-deactivated", host);
 142         log("info", "Deactivated host: %s", host);
 143         return true;
 144 end
 145
 146 function get_children(host)
 147         return disco_items:get(host) or NULL;
 148 end
 149
 150 return _M;