Merge 0.10->trunk
[prosody.git] / core / certmanager.lua
index 837fe23192ef6f4882c8d882c430025e8e18bff9..29a5a6c8403906428bddec299950a79efaf5f506 100644 (file)
@@ -6,13 +6,26 @@
 -- COPYING file in the source package for more information.
 --
 
+local softreq = require"util.dependencies".softreq;
+local ssl = softreq"ssl";
+if not ssl then
+       return {
+               create_context = function ()
+                       return nil, "LuaSec (required for encryption) was not found";
+               end;
+               reload_ssl_config = function () end;
+       }
+end
+
 local configmanager = require "core.configmanager";
 local log = require "util.logger".init("certmanager");
-local ssl = ssl;
-local ssl_newcontext = ssl and ssl.newcontext;
+local ssl_context = ssl.context or softreq"ssl.context";
+local ssl_x509 = ssl.x509 or softreq"ssl.x509";
+local ssl_newcontext = ssl.newcontext;
 local new_config = require"util.sslconfig".new;
+local stat = require "lfs".attributes;
 
-local tostring = tostring;
+local tonumber, tostring = tonumber, tostring;
 local pairs = pairs;
 local type = type;
 local io_open = io.open;
@@ -22,31 +35,71 @@ local prosody = prosody;
 local resolve_path = require"util.paths".resolve_relative_path;
 local config_path = prosody.paths.config;
 
-local luasec_has_noticket, luasec_has_verifyext, luasec_has_no_compression;
-if ssl then
-       local luasec_major, luasec_minor = ssl._VERSION:match("^(%d+)%.(%d+)");
-       luasec_has_noticket = tonumber(luasec_major)>0 or tonumber(luasec_minor)>=4;
-       luasec_has_verifyext = tonumber(luasec_major)>0 or tonumber(luasec_minor)>=5;
-       luasec_has_no_compression = tonumber(luasec_major)>0 or tonumber(luasec_minor)>=5;
-end
+local luasec_major, luasec_minor = ssl._VERSION:match("^(%d+)%.(%d+)");
+local luasec_version = tonumber(luasec_major) * 100 + tonumber(luasec_minor);
+local luasec_has = {
+       -- TODO If LuaSec ever starts exposing these things itself, use that instead
+       cipher_server_preference = luasec_version >= 2;
+       no_ticket = luasec_version >= 4;
+       no_compression = luasec_version >= 5;
+       single_dh_use = luasec_version >= 2;
+       single_ecdh_use = luasec_version >= 2;
+};
 
-module "certmanager"
+local _ENV = nil;
 
 -- Global SSL options if not overridden per-host
 local global_ssl_config = configmanager.get("*", "ssl");
 
+local global_certificates = configmanager.get("*", "certificates") or "certs";
+
+local crt_try = { "", "/%s.crt", "/%s/fullchain.pem", "/%s.pem", };
+local key_try = { "", "/%s.key", "/%s/privkey.pem",   "/%s.pem", };
+
+local function find_cert(user_certs, name)
+       local certs = resolve_path(config_path, user_certs or global_certificates);
+       for i = 1, #crt_try do
+               local crt_path = certs .. crt_try[i]:format(name);
+               local key_path = certs .. key_try[i]:format(name);
+
+               if stat(crt_path, "mode") == "file" then
+                       if key_path:sub(-4) == ".crt" then
+                               key_path = key_path:sub(1, -4) .. "key";
+                               if stat(key_path, "mode") == "file" then
+                                       return { certificate = crt_path, key = key_path };
+                               end
+                       elseif stat(key_path, "mode") == "file" then
+                               return { certificate = crt_path, key = key_path };
+                       end
+               end
+       end
+end
+
+local function find_host_cert(host)
+       if not host then return nil; end
+       return find_cert(configmanager.get(host, "certificate"), host) or find_host_cert(host:match("%.(.+)$"));
+end
+
+local function find_service_cert(service, port)
+       local cert_config = configmanager.get("*", service.."_certificate");
+       if type(cert_config) == "table" then
+               cert_config = cert_config[port] or cert_config.default;
+       end
+       return find_cert(cert_config, service);
+end
+
 -- Built-in defaults
 local core_defaults = {
        capath = "/etc/ssl/certs";
+       depth = 9;
        protocol = "tlsv1+";
-       verify = (ssl and ssl.x509 and { "peer", "client_once", }) or "none";
+       verify = (ssl_x509 and { "peer", "client_once", }) or "none";
        options = {
-               cipher_server_preference = true;
-               no_ticket = luasec_has_noticket;
-               no_compression = luasec_has_no_compression and configmanager.get("*", "ssl_compression") ~= true;
-               -- Has no_compression? Then it has these too...
-               single_dh_use = luasec_has_no_compression;
-               single_ecdh_use = luasec_has_no_compression;
+               cipher_server_preference = luasec_has.cipher_server_preference;
+               no_ticket = luasec_has.no_ticket;
+               no_compression = luasec_has.no_compression and configmanager.get("*", "ssl_compression") ~= true;
+               single_dh_use = luasec_has.single_dh_use;
+               single_ecdh_use = luasec_has.single_ecdh_use;
        };
        verifyext = { "lsec_continue", "lsec_ignore_purpose" };
        curve = "secp384r1";
@@ -56,24 +109,28 @@ local path_options = { -- These we pass through resolve_path()
        key = true, certificate = true, cafile = true, capath = true, dhparam = true
 }
 
-if ssl and not luasec_has_verifyext and ssl.x509 then
+if luasec_version < 5 and ssl_x509 then
        -- COMPAT mw/luasec-hg
        for i=1,#core_defaults.verifyext do -- Remove lsec_ prefix
                core_defaults.verify[#core_defaults.verify+1] = core_defaults.verifyext[i]:sub(6);
        end
 end
 
-function create_context(host, mode, ...)
-       if not ssl then return nil, "LuaSec (required for encryption) was not found"; end
-
+local function create_context(host, mode, ...)
        local cfg = new_config();
        cfg:apply(core_defaults);
-       cfg:apply(global_ssl_config);
+       local service_name, port = host:match("^(%w+) port (%d+)$");
+       if service_name then
+               cfg:apply(find_service_cert(service_name, tonumber(port)));
+       else
+               cfg:apply(find_host_cert(host));
+       end
        cfg:apply({
                mode = mode,
                -- We can't read the password interactively when daemonized
                password = function() log("error", "Encrypted certificate for %s requires 'ssl' 'password' to be set in config", host); end;
        });
+       cfg:apply(global_ssl_config);
 
        for i = select('#', ...), 1, -1 do
                cfg:apply(select(i, ...));
@@ -88,6 +145,8 @@ function create_context(host, mode, ...)
        for option in pairs(path_options) do
                if type(user_ssl_config[option]) == "string" then
                        user_ssl_config[option] = resolve_path(config_path, user_ssl_config[option]);
+               else
+                       user_ssl_config[option] = nil;
                end
        end
 
@@ -108,7 +167,7 @@ function create_context(host, mode, ...)
        -- of it ourselves (W/A for #x)
        if ctx and user_ssl_config.ciphers then
                local success;
-               success, err = ssl.context.setcipher(ctx, user_ssl_config.ciphers);
+               success, err = ssl_context.setcipher(ctx, user_ssl_config.ciphers);
                if not success then ctx = nil; end
        end
 
@@ -138,16 +197,19 @@ function create_context(host, mode, ...)
                        log("error", "SSL/TLS: Error initialising for %s: %s", host, err);
                end
        end
-       return ctx, err;
+       return ctx, err, user_ssl_config;
 end
 
-function reload_ssl_config()
+local function reload_ssl_config()
        global_ssl_config = configmanager.get("*", "ssl");
-       if luasec_has_no_compression then
+       if luasec_has.no_compression then
                core_defaults.options.no_compression = configmanager.get("*", "ssl_compression") ~= true;
        end
 end
 
 prosody.events.add_handler("config-reloaded", reload_ssl_config);
 
-return _M;
+return {
+       create_context = create_context;
+       reload_ssl_config = reload_ssl_config;
+};