Merge 0.7->trunk
[prosody.git] / core / hostmanager.lua
index 15484574eb8e890d1d576f097868cd99fb1823b0..eb88044995d63b5cf036e9d1d3d258096ff93e3c 100644 (file)
@@ -1,6 +1,6 @@
 -- Prosody IM
--- Copyright (C) 2008-2009 Matthew Wild
--- Copyright (C) 2008-2009 Waqas Hussain
+-- Copyright (C) 2008-2010 Matthew Wild
+-- Copyright (C) 2008-2010 Waqas Hussain
 -- 
 -- This project is MIT/X11 licensed. Please see the
 -- COPYING file in the source package for more information.
@@ -9,12 +9,18 @@
 local ssl = ssl
 
 local hosts = hosts;
+local certmanager = require "core.certmanager";
 local configmanager = require "core.configmanager";
 local eventmanager = require "core.eventmanager";
+local modulemanager = require "core.modulemanager";
 local events_new = require "util.events".new;
 
--- These are the defaults if not overridden in the config
-local default_ssl_ctx = { mode = "client", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
+local uuid_gen = require "util.uuid".generate;
+
+if not _G.prosody.incoming_s2s then
+       require "core.s2smanager";
+end
+local incoming_s2s = _G.prosody.incoming_s2s;
 
 local log = require "util.logger".init("hostmanager");
 
@@ -26,12 +32,19 @@ local hosts_loaded_once;
 
 local function load_enabled_hosts(config)
        local defined_hosts = config or configmanager.getconfig();
+       local activated_any_host;
        
        for host, host_config in pairs(defined_hosts) do
-               if host ~= "*" and (host_config.core.enabled == nil or host_config.core.enabled) and not host_config.core.component_module then
+               if host ~= "*" and host_config.core.enabled ~= false and not host_config.core.component_module then
+                       activated_any_host = true;
                        activate(host, host_config);
                end
        end
+       
+       if not activated_any_host then
+               log("error", "No hosts defined in the config file. This may cause unexpected behaviour as no modules will be loaded.");
+       end
+       
        eventmanager.fire_event("hosts-activated", defined_hosts);
        hosts_loaded_once = true;
 end
@@ -39,11 +52,12 @@ end
 eventmanager.add_event_hook("server-starting", load_enabled_hosts);
 
 function activate(host, host_config)
-       hosts[host] = {type = "local", connected = true, sessions = {}, 
-                      host = host, s2sout = {}, events = events_new(), 
-                      disallow_s2s = configmanager.get(host, "core", "disallow_s2s") 
-                        or (configmanager.get(host, "core", "anonymous_login") 
-                            and (configmanager.get(host, "core", "disallow_s2s") ~= false))
+       hosts[host] = {type = "local", connected = true, sessions = {},
+                       host = host, s2sout = {}, events = events_new(),
+                       disallow_s2s = configmanager.get(host, "core", "disallow_s2s")
+                               or (configmanager.get(host, "core", "anonymous_login")
+                               and (configmanager.get(host, "core", "disallow_s2s") ~= false));
+                       dialback_secret = configmanager.get(host, "core", "dialback_secret") or uuid_gen();
                      };
        for option_name in pairs(host_config.core) do
                if option_name:match("_ports$") then
@@ -51,30 +65,51 @@ function activate(host, host_config)
                end
        end
        
-       if ssl then
-               local ssl_config = host_config.core.ssl or configmanager.get("*", "core", "ssl");
-               if ssl_config then
-                       hosts[host].ssl_ctx = ssl.newcontext(setmetatable(ssl_config, { __index = default_ssl_ctx }));
-               end
-        end
-
+       hosts[host].ssl_ctx = certmanager.create_context(host, "client", host_config); -- for outgoing connections
+       hosts[host].ssl_ctx_in = certmanager.create_context(host, "server", host_config); -- for incoming connections
+       
        log((hosts_loaded_once and "info") or "debug", "Activated host: %s", host);
        eventmanager.fire_event("host-activated", host, host_config);
 end
 
-function deactivate(host)
+function deactivate(host, reason)
        local host_session = hosts[host];
        log("info", "Deactivating host: %s", host);
        eventmanager.fire_event("host-deactivating", host, host_session);
        
+       reason = reason or { condition = "host-gone", text = "This server has stopped serving "..host };
+       
        -- Disconnect local users, s2s connections
-       for user, session_list in pairs(host_session.sessions) do
-               for resource, session in pairs(session_list) do
-                       session:close("host-gone");
+       if host_session.sessions then
+               for username, user in pairs(host_session.sessions) do
+                       for resource, session in pairs(user.sessions) do
+                               log("debug", "Closing connection for %s@%s/%s", username, host, resource);
+                               session:close(reason);
+                       end
                end
        end
-       -- Components?
-       
+       if host_session.s2sout then
+               for remotehost, session in pairs(host_session.s2sout) do
+                       if session.close then
+                               log("debug", "Closing outgoing connection to %s", remotehost);
+                               if session.srv_hosts then session.srv_hosts = nil; end
+                               session:close(reason);
+                       end
+               end
+       end
+       for remote_session in pairs(incoming_s2s) do
+               if remote_session.to_host == host then
+                       log("debug", "Closing incoming connection from %s", remote_session.from_host or "<unknown>");
+                       remote_session:close(reason);
+               end
+       end
+
+       if host_session.modules then
+               for module in pairs(host_session.modules) do
+                       modulemanager.unload(host, module);
+               end
+       end
+
        hosts[host] = nil;
        eventmanager.fire_event("host-deactivated", host);
        log("info", "Deactivated host: %s", host);