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