mod_admin_telnet: Add server:memory() command to view details of Prosody's memory...
[prosody.git] / plugins / mod_saslauth.lua
index 4906d01f6fa9f6d52ad7f3448268c13615b879cf..201cc477de9eed5980a6ec31eaeb6a298afd5b21 100644 (file)
@@ -13,7 +13,8 @@ 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 nodeprep = require "util.encodings".stringprep.nodeprep;
+local cert_verify_identity = require "util.x509".verify_identity;
+
 local usermanager_get_sasl_handler = require "core.usermanager".get_sasl_handler;
 local tostring = tostring;
 
@@ -24,7 +25,6 @@ local log = module._log;
 
 local xmlns_sasl ='urn:ietf:params:xml:ns:xmpp-sasl';
 local xmlns_bind ='urn:ietf:params:xml:ns:xmpp-bind';
-local xmlns_stanzas ='urn:ietf:params:xml:ns:xmpp-stanzas';
 
 local function build_reply(status, ret, err_msg)
        local reply = st.stanza(status, {xmlns = xmlns_sasl});
@@ -45,16 +45,17 @@ end
 
 local function handle_status(session, status, ret, err_msg)
        if status == "failure" then
+               module:fire_event("authentication-failure", { session = session, condition = ret, text = err_msg });
                session.sasl_handler = session.sasl_handler:clean_clone();
        elseif status == "success" then
-               local username = nodeprep(session.sasl_handler.username);
-
                local ok, err = sm_make_authenticated(session, session.sasl_handler.username);
                if ok then
+                       module:fire_event("authentication-success", { session = session });
                        session.sasl_handler = nil;
                        session:reset_stream();
                else
                        module:log("warn", "SASL succeeded but username was invalid");
+                       module:fire_event("authentication-failure", { session = session, condition = "not-authorized", text = err });
                        session.sasl_handler = session.sasl_handler:clean_clone();
                        return "failure", "not-authorized", "User authenticated successfully, but username was invalid";
                end
@@ -81,15 +82,128 @@ local function sasl_process_cdata(session, stanza)
        return true;
 end
 
+module:hook_stanza(xmlns_sasl, "success", function (session, stanza)
+       if session.type ~= "s2sout_unauthed" or session.external_auth ~= "attempting" then return; end
+       module:log("debug", "SASL EXTERNAL with %s succeeded", session.to_host);
+       session.external_auth = "succeeded"
+       session:reset_stream();
+       session:open_stream(session.from_host, session.to_host);
+
+       module:fire_event("s2s-authenticated", { session = session, host = session.to_host });
+       return true;
+end)
+
+module:hook_stanza(xmlns_sasl, "failure", function (session, stanza)
+       if session.type ~= "s2sout_unauthed" or session.external_auth ~= "attempting" then return; end
+
+       module:log("info", "SASL EXTERNAL with %s failed", session.to_host)
+       -- TODO: Log the failure reason
+       session.external_auth = "failed"
+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
+
+       local mechanisms = stanza:get_child("mechanisms", xmlns_sasl)
+       if mechanisms then
+               for mech in mechanisms:childtags() do
+                       if mech[1] == "EXTERNAL" then
+                               module:log("debug", "Initiating SASL EXTERNAL with %s", session.to_host);
+                               local reply = st.stanza("auth", {xmlns = xmlns_sasl, mechanism = "EXTERNAL"});
+                               reply:text(base64.encode(session.from_host))
+                               session.sends2s(reply)
+                               session.external_auth = "attempting"
+                               return true
+                       end
+               end
+       end
+end, 150);
+
+local function s2s_external_auth(session, stanza)
+       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
+               return true;
+       end
+
+       if mechanism ~= "EXTERNAL" or session.cert_chain_status ~= "valid" then
+               session.sends2s(build_reply("failure", "invalid-mechanism"))
+               return true;
+       end
+
+       local text = stanza[1]
+       if not text then
+               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)
+       if not text then
+               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
+       end
+
+       session.external_auth = "succeeded"
+
+       if not session.from_host then
+               session.from_host = text;
+       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 });
+       session:reset_stream();
+       return true
+end
+
 module:hook("stanza/urn:ietf:params:xml:ns:xmpp-sasl:auth", function(event)
        local session, stanza = event.origin, event.stanza;
+       if session.type == "s2sin_unauthed" then
+               return s2s_external_auth(session, stanza)
+       end
+
        if session.type ~= "c2s_unauthed" then return; end
 
        if session.sasl_handler and session.sasl_handler.selected then
                session.sasl_handler = nil; -- allow starting a new SASL negotiation before completing an old one
        end
        if not session.sasl_handler then
-               session.sasl_handler = usermanager_get_sasl_handler(module.host);
+               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
@@ -127,20 +241,34 @@ 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);
-               features:tag("mechanisms", mechanisms_attr);
+               origin.sasl_handler = usermanager_get_sasl_handler(module.host, origin);
+               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
-                               features:tag("mechanism"):text(mechanism):up();
+                               mechanisms:tag("mechanism"):text(mechanism):up();
                        end
                end
-               features:up();
+               if mechanisms[1] then features:add_child(mechanisms); end
        else
                features:tag("bind", bind_attr):tag("required"):up():up();
                features:tag("session", xmpp_session_attr):tag("optional"):up():up();
        end
 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")
+                       features:tag("mechanisms", { xmlns = xmlns_sasl })
+                               :tag("mechanism"):text("EXTERNAL")
+                       :up():up();
+               end
+       end
+end);
+
 module:hook("iq/self/urn:ietf:params:xml:ns:xmpp-bind:bind", function(event)
        local origin, stanza = event.origin, event.stanza;
        local resource;