Merge 0.10->trunk
[prosody.git] / plugins / mod_saslauth.lua
index df60aefa36d09a6278555a9bb65ba9d596da1651..bb36600b2e066641e59fc975c4a01281b90f23dd 100644 (file)
@@ -13,13 +13,13 @@ local sm_bind_resource = require "core.sessionmanager".bind_resource;
 local sm_make_authenticated = require "core.sessionmanager".make_authenticated;
 local base64 = require "util.encodings".base64;
 
-local cert_verify_identity = require "util.x509".verify_identity;
-
 local usermanager_get_sasl_handler = require "core.usermanager".get_sasl_handler;
 local tostring = tostring;
 
-local secure_auth_only = module:get_option("c2s_require_encryption") or module:get_option("require_encryption");
-local allow_unencrypted_plain_auth = module:get_option("allow_unencrypted_plain_auth")
+local secure_auth_only = module:get_option_boolean("c2s_require_encryption", module:get_option_boolean("require_encryption", false));
+local allow_unencrypted_plain_auth = module:get_option_boolean("allow_unencrypted_plain_auth", false)
+local insecure_mechanisms = module:get_option_set("insecure_sasl_mechanisms", allow_unencrypted_plain_auth and {} or {"PLAIN", "LOGIN"});
+local disabled_mechanisms = module:get_option_set("disable_sasl_mechanisms", { "DIGEST-MD5" });
 
 local log = module._log;
 
@@ -28,15 +28,15 @@ local xmlns_bind ='urn:ietf:params:xml:ns:xmpp-bind';
 
 local function build_reply(status, ret, err_msg)
        local reply = st.stanza(status, {xmlns = xmlns_sasl});
-       if status == "challenge" then
-               --log("debug", "CHALLENGE: %s", ret or "");
-               reply:text(base64.encode(ret or ""));
-       elseif status == "failure" then
+       if status == "failure" then
                reply:tag(ret):up();
                if err_msg then reply:tag("text"):text(err_msg); end
-       elseif status == "success" then
-               --log("debug", "SUCCESS: %s", ret or "");
-               reply:text(base64.encode(ret or ""));
+       elseif status == "challenge" or status == "success" then
+               if ret == "" then
+                       reply:text("=")
+               elseif ret then
+                       reply:text(base64.encode(ret));
+               end
        else
                module:log("error", "Unknown sasl status: %s", status);
        end
@@ -99,12 +99,10 @@ module:hook_stanza(xmlns_sasl, "failure", function (session, stanza)
        module:log("info", "SASL EXTERNAL with %s failed", session.to_host)
        -- TODO: Log the failure reason
        session.external_auth = "failed"
+       session:close();
+       return true;
 end, 500)
 
-module:hook_stanza(xmlns_sasl, "failure", function (session, stanza)
-       -- TODO: Dialback wasn't loaded.  Do something useful.
-end, 90)
-
 module:hook_stanza("http://etherx.jabber.org/streams", "features", function (session, stanza)
        if session.type ~= "s2sout_unauthed" or not session.secure then return; end
 
@@ -124,71 +122,52 @@ module:hook_stanza("http://etherx.jabber.org/streams", "features", function (ses
 end, 150);
 
 local function s2s_external_auth(session, stanza)
+       if session.external_auth ~= "offered" then return end -- Unexpected request
+
        local mechanism = stanza.attr.mechanism;
 
-       if not session.secure then
-               if mechanism == "EXTERNAL" then
-                       session.sends2s(build_reply("failure", "encryption-required"))
-               else
-                       session.sends2s(build_reply("failure", "invalid-mechanism"))
-               end
+       if mechanism ~= "EXTERNAL" then
+               session.sends2s(build_reply("failure", "invalid-mechanism"));
                return true;
        end
 
-       if mechanism ~= "EXTERNAL" or session.cert_chain_status ~= "valid" then
-               session.sends2s(build_reply("failure", "invalid-mechanism"))
+       if not session.secure then
+               session.sends2s(build_reply("failure", "encryption-required"));
                return true;
        end
 
-       local text = stanza[1]
+       local text = stanza[1];
        if not text then
-               session.sends2s(build_reply("failure", "malformed-request"))
-               return true
+               session.sends2s(build_reply("failure", "malformed-request"));
+               return true;
        end
 
-       -- Either the value is "=" and we've already verified the external
-       -- cert identity, or the value is a string and either matches the
-       -- from_host (
-
-       text = base64.decode(text)
+       text = base64.decode(text);
        if not text then
-               session.sends2s(build_reply("failure", "incorrect-encoding"))
+               session.sends2s(build_reply("failure", "incorrect-encoding"));
                return true;
        end
 
-       if session.cert_identity_status == "valid" then
-               if text ~= "" and text ~= session.from_host then
-                       session.sends2s(build_reply("failure", "invalid-authzid"))
-                       return true
-               end
-       else
-               if text == "" then
-                       session.sends2s(build_reply("failure", "invalid-authzid"))
-                       return true
-               end
-
-               local cert = session.conn:socket():getpeercertificate()
-               if (cert_verify_identity(text, "xmpp-server", cert)) then
-                       session.cert_identity_status = "valid"
-               else
-                       session.cert_identity_status = "invalid"
-                       session.sends2s(build_reply("failure", "invalid-authzid"))
-                       return true
-               end
+       -- The text value is either "" or equals session.from_host
+       if not ( text == "" or text == session.from_host ) then
+               session.sends2s(build_reply("failure", "invalid-authzid"));
+               return true;
        end
 
-       session.external_auth = "succeeded"
-
-       if not session.from_host then
-               session.from_host = text;
+       -- We've already verified the external cert identity before offering EXTERNAL
+       if session.cert_chain_status ~= "valid" or session.cert_identity_status ~= "valid" then
+               session.sends2s(build_reply("failure", "not-authorized"));
+               session:close();
+               return true;
        end
-       session.sends2s(build_reply("success"))
 
-       local domain = text ~= "" and text or session.from_host;
-       module:log("info", "Accepting SASL EXTERNAL identity from %s", domain);
-       module:fire_event("s2s-authenticated", { session = session, host = domain });
+       -- Success!
+       session.external_auth = "succeeded";
+       session.sends2s(build_reply("success"));
+       module:log("info", "Accepting SASL EXTERNAL identity from %s", session.from_host);
+       module:fire_event("s2s-authenticated", { session = session, host = session.from_host });
        session:reset_stream();
-       return true
+       return true;
 end
 
 module:hook("stanza/urn:ietf:params:xml:ns:xmpp-sasl:auth", function(event)
@@ -206,9 +185,12 @@ module:hook("stanza/urn:ietf:params:xml:ns:xmpp-sasl:auth", function(event)
                session.sasl_handler = usermanager_get_sasl_handler(module.host, session);
        end
        local mechanism = stanza.attr.mechanism;
-       if not session.secure and (secure_auth_only or (mechanism == "PLAIN" and not allow_unencrypted_plain_auth)) then
+       if not session.secure and (secure_auth_only or insecure_mechanisms:contains(mechanism)) then
                session.send(build_reply("failure", "encryption-required"));
                return true;
+       elseif disabled_mechanisms:contains(mechanism) then
+               session.send(build_reply("failure", "invalid-mechanism"));
+               return true;
        end
        local valid_mechanism = session.sasl_handler:select(mechanism);
        if not valid_mechanism then
@@ -232,6 +214,10 @@ module:hook("stanza/urn:ietf:params:xml:ns:xmpp-sasl:abort", function(event)
        return true;
 end);
 
+local function tls_unique(self)
+       return self.userdata["tls-unique"]:getpeerfinished();
+end
+
 local mechanisms_attr = { xmlns='urn:ietf:params:xml:ns:xmpp-sasl' };
 local bind_attr = { xmlns='urn:ietf:params:xml:ns:xmpp-bind' };
 local xmpp_session_attr = { xmlns='urn:ietf:params:xml:ns:xmpp-session' };
@@ -241,24 +227,32 @@ module:hook("stream-features", function(event)
                if secure_auth_only and not origin.secure then
                        return;
                end
-               origin.sasl_handler = usermanager_get_sasl_handler(module.host, origin);
+               local sasl_handler = usermanager_get_sasl_handler(module.host, origin)
+               origin.sasl_handler = sasl_handler;
                if origin.encrypted then
                        -- check wether LuaSec has the nifty binding to the function needed for tls-unique
                        -- FIXME: would be nice to have this check only once and not for every socket
-                       if origin.conn:socket().getpeerfinished and origin.sasl_handler.add_cb_handler then
-                               origin.sasl_handler:add_cb_handler("tls-unique", function(self)
-                                       return self.userdata:getpeerfinished();
-                               end);
-                               origin.sasl_handler["userdata"] = origin.conn:socket();
+                       if sasl_handler.add_cb_handler then
+                               local socket = origin.conn:socket();
+                               if socket.getpeerfinished then
+                                       sasl_handler:add_cb_handler("tls-unique", tls_unique);
+                               end
+                               sasl_handler["userdata"] = {
+                                       ["tls-unique"] = socket;
+                               };
                        end
                end
                local mechanisms = st.stanza("mechanisms", mechanisms_attr);
-               for mechanism in pairs(origin.sasl_handler:mechanisms()) do
-                       if mechanism ~= "PLAIN" or origin.secure or allow_unencrypted_plain_auth then
+               for mechanism in pairs(sasl_handler:mechanisms()) do
+                       if (not disabled_mechanisms:contains(mechanism)) and (origin.secure or not insecure_mechanisms:contains(mechanism)) then
                                mechanisms:tag("mechanism"):text(mechanism):up();
                        end
                end
-               if mechanisms[1] then features:add_child(mechanisms); end
+               if mechanisms[1] then
+                       features:add_child(mechanisms);
+               else
+                       (origin.log or log)("warn", "No SASL mechanisms to offer");
+               end
        else
                features:tag("bind", bind_attr):tag("required"):up():up();
                features:tag("session", xmpp_session_attr):tag("optional"):up():up();
@@ -268,10 +262,10 @@ end);
 module:hook("s2s-stream-features", function(event)
        local origin, features = event.origin, event.features;
        if origin.secure and origin.type == "s2sin_unauthed" then
-               -- Offer EXTERNAL if chain is valid and either we didn't validate
-               -- the identity or it passed.
-               if origin.cert_chain_status == "valid" and origin.cert_identity_status ~= "invalid" then --TODO: Configurable
-                       module:log("debug", "Offering SASL EXTERNAL")
+               -- Offer EXTERNAL only if both chain and identity is valid.
+               if origin.cert_chain_status == "valid" and origin.cert_identity_status == "valid" then
+                       module:log("debug", "Offering SASL EXTERNAL");
+                       origin.external_auth = "offered"
                        features:tag("mechanisms", { xmlns = xmlns_sasl })
                                :tag("mechanism"):text("EXTERNAL")
                        :up():up();
@@ -284,7 +278,7 @@ module:hook("iq/self/urn:ietf:params:xml:ns:xmpp-bind:bind", function(event)
        local resource;
        if stanza.attr.type == "set" then
                local bind = stanza.tags[1];
-               resource = bind:child_with_name("resource");
+               resource = bind:get_child("resource");
                resource = resource and #resource.tags == 0 and resource[1] or nil;
        end
        local success, err_type, err, err_msg = sm_bind_resource(origin, resource);