core/hostmanager.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 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 ssl = ssl
  10
  11 local hosts = hosts;
  12 local configmanager = require "core.configmanager";
  13 local eventmanager = require "core.eventmanager";
  14 local modulemanager = require "core.modulemanager";
  15 local events_new = require "util.events".new;
  16
  17 local uuid_gen = require "util.uuid".generate;
  18
  19 if not _G.prosody.incoming_s2s then
  20         require "core.s2smanager";
  21 end
  22 local incoming_s2s = _G.prosody.incoming_s2s;
  23
  24 -- These are the defaults if not overridden in the config
  25 local default_ssl_ctx = { mode = "client", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none", options = "no_sslv2"; };
  26 local default_ssl_ctx_in = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none", options = "no_sslv2"; };
  27
  28 local log = require "util.logger".init("hostmanager");
  29
  30 local pairs, setmetatable = pairs, setmetatable;
  31
  32 module "hostmanager"
  33
  34 local hosts_loaded_once;
  35
  36 local function load_enabled_hosts(config)
  37         local defined_hosts = config or configmanager.getconfig();
  38
  39         for host, host_config in pairs(defined_hosts) do
  40                 if host ~= "*" and (host_config.core.enabled == nil or host_config.core.enabled) and not host_config.core.component_module then
  41                         activate(host, host_config);
  42                 end
  43         end
  44         eventmanager.fire_event("hosts-activated", defined_hosts);
  45         hosts_loaded_once = true;
  46 end
  47
  48 eventmanager.add_event_hook("server-starting", load_enabled_hosts);
  49
  50 function activate(host, host_config)
  51         hosts[host] = {type = "local", connected = true, sessions = {},
  52                         host = host, s2sout = {}, events = events_new(),
  53                         disallow_s2s = configmanager.get(host, "core", "disallow_s2s")
  54                                 or (configmanager.get(host, "core", "anonymous_login")
  55                                 and (configmanager.get(host, "core", "disallow_s2s") ~= false));
  56                         dialback_secret = configmanager.get(host, "core", "dialback_secret") or uuid_gen();
  57                       };
  58         for option_name in pairs(host_config.core) do
  59                 if option_name:match("_ports$") then
  60                         log("warn", "%s: Option '%s' has no effect for virtual hosts - put it in global Host \"*\" instead", host, option_name);
  61                 end
  62         end
  63
  64         if ssl then
  65                 local ssl_config = host_config.core.ssl or configmanager.get("*", "core", "ssl");
  66                 if ssl_config then
  67                         hosts[host].ssl_ctx = ssl.newcontext(setmetatable(ssl_config, { __index = default_ssl_ctx }));
  68                         hosts[host].ssl_ctx_in = ssl.newcontext(setmetatable(ssl_config, { __index = default_ssl_ctx_in }));
  69                 end
  70         end
  71
  72         log((hosts_loaded_once and "info") or "debug", "Activated host: %s", host);
  73         eventmanager.fire_event("host-activated", host, host_config);
  74 end
  75
  76 function deactivate(host, reason)
  77         local host_session = hosts[host];
  78         log("info", "Deactivating host: %s", host);
  79         eventmanager.fire_event("host-deactivating", host, host_session);
  80
  81         reason = reason or { condition = "host-gone", text = "This server has stopped serving "..host };
  82
  83         -- Disconnect local users, s2s connections
  84         if host_session.sessions then
  85                 for username, user in pairs(host_session.sessions) do
  86                         for resource, session in pairs(user.sessions) do
  87                                 log("debug", "Closing connection for %s@%s/%s", username, host, resource);
  88                                 session:close(reason);
  89                         end
  90                 end
  91         end
  92         if host_session.s2sout then
  93                 for remotehost, session in pairs(host_session.s2sout) do
  94                         if session.close then
  95                                 log("debug", "Closing outgoing connection to %s", remotehost);
  96                                 if session.srv_hosts then session.srv_hosts = nil; end
  97                                 session:close(reason);
  98                         end
  99                 end
 100         end
 101         for remote_session in pairs(incoming_s2s) do
 102                 if remote_session.to_host == host then
 103                         log("debug", "Closing incoming connection from %s", remote_session.from_host or "<unknown>");
 104                         remote_session:close(reason);
 105                 end
 106         end
 107
 108         if host_session.modules then
 109                 for module in pairs(host_session.modules) do
 110                         modulemanager.unload(host, module);
 111                 end
 112         end
 113
 114         hosts[host] = nil;
 115         eventmanager.fire_event("host-deactivated", host);
 116         log("info", "Deactivated host: %s", host);
 117 end
 118
 119 function getconfig(name)
 120 end
 121
 122 return _M;