util.sasl.scram: Fixed global access.
[prosody.git] / util / sasl / scram.lua
index be82c60e0f8adcd4e4e2af456f2a2b5914cf2c90..ed7d7bc385a0de3fbfd33c8b4223c0deff1db26d 100644 (file)
@@ -1,5 +1,5 @@
 -- sasl.lua v0.4
--- Copyright (C) 2008-2009 Tobias Markmann
+-- Copyright (C) 2008-2010 Tobias Markmann
 --
 --    All rights reserved.
 --
@@ -15,17 +15,29 @@ local s_match = string.match;
 local type = type
 local string = string
 local base64 = require "util.encodings".base64;
-local xor = require "bit".bxor
 local hmac_sha1 = require "util.hmac".sha1;
 local sha1 = require "util.hashes".sha1;
 local generate_uuid = require "util.uuid".generate;
 local saslprep = require "util.encodings".stringprep.saslprep;
 local log = require "util.logger".init("sasl");
+local t_concat = table.concat;
+local char = string.char;
+local byte = string.byte;
 
 module "scram"
 
 --=========================
 --SASL SCRAM-SHA-1 according to draft-ietf-sasl-scram-10
+
+--[[
+Supported Authentication Backends
+
+scram-{MECH}:
+       function(username, realm)
+               return salted_password, iteration_count, salt, state;
+       end
+]]
+
 local default_i = 4096
 
 local function bp( b )
@@ -36,17 +48,19 @@ local function bp( b )
        return result
 end
 
+local xor_map = {0;1;2;3;4;5;6;7;8;9;10;11;12;13;14;15;1;0;3;2;5;4;7;6;9;8;11;10;13;12;15;14;2;3;0;1;6;7;4;5;10;11;8;9;14;15;12;13;3;2;1;0;7;6;5;4;11;10;9;8;15;14;13;12;4;5;6;7;0;1;2;3;12;13;14;15;8;9;10;11;5;4;7;6;1;0;3;2;13;12;15;14;9;8;11;10;6;7;4;5;2;3;0;1;14;15;12;13;10;11;8;9;7;6;5;4;3;2;1;0;15;14;13;12;11;10;9;8;8;9;10;11;12;13;14;15;0;1;2;3;4;5;6;7;9;8;11;10;13;12;15;14;1;0;3;2;5;4;7;6;10;11;8;9;14;15;12;13;2;3;0;1;6;7;4;5;11;10;9;8;15;14;13;12;3;2;1;0;7;6;5;4;12;13;14;15;8;9;10;11;4;5;6;7;0;1;2;3;13;12;15;14;9;8;11;10;5;4;7;6;1;0;3;2;14;15;12;13;10;11;8;9;6;7;4;5;2;3;0;1;15;14;13;12;11;10;9;8;7;6;5;4;3;2;1;0;};
+
+local result = {};
 local function binaryXOR( a, b )
-       if a:len() > b:len() then
-               b = string.rep("\0", a:len() - b:len())..b
-       elseif string.len(a) < string.len(b) then
-               a = string.rep("\0", b:len() - a:len())..a
-       end
-       local result = ""
-       for i=1, a:len() do
-               result = result..string.char(xor(a:byte(i), b:byte(i)))
+       for i=1, #a do
+               local x, y = byte(a, i), byte(b, i);
+               local lowx, lowy = x % 16, y % 16;
+               local hix, hiy = (x - lowx) / 16, (y - lowy) / 16;
+               local lowr, hir = xor_map[lowx * 16 + lowy + 1], xor_map[hix * 16 + hiy + 1];
+               local r = hir * 16 + lowr;
+               result[i] = char(r)
        end
-       return result
+       return t_concat(result);
 end
 
 -- hash algorithm independent Hi(PBKDF2) implementation
@@ -78,77 +92,95 @@ local function validate_username(username)
        return username;
 end
 
-local function scram_sha_1(self, message)
-       if not self.state then self["state"] = {} end
+local function scram_gen(hash_name, H_f, HMAC_f)
+       local function scram_hash(self, message)
+               if not self.state then self["state"] = {} end
        
-       if not self.state.name then
-               -- we are processing client_first_message
-               local client_first_message = message;
-               self.state["client_first_message"] = client_first_message;
-               self.state["name"] = client_first_message:match("n=(.+),r=")
-               self.state["clientnonce"] = client_first_message:match("r=([^,]+)")
-               
-               if not self.state.name or not self.state.clientnonce then
-                       return "failure", "malformed-request";
-               end
-               
-               self.state.name = validate_username(self.state.name);
                if not self.state.name then
-                       log("debug", "Username violates either SASLprep or contains forbidden character sequences.")
-                       return "failure", "malformed-request", "Invalid username.";
-               end
+                       -- we are processing client_first_message
+                       local client_first_message = message;
+                       self.state["client_first_message"] = client_first_message;
+                       self.state["name"] = client_first_message:match("n=(.+),r=")
+                       self.state["clientnonce"] = client_first_message:match("r=([^,]+)")
                
-               self.state["servernonce"] = generate_uuid();
-               self.state["salt"] = generate_uuid();
+                       if not self.state.name or not self.state.clientnonce then
+                               return "failure", "malformed-request";
+                       end
                
-               local server_first_message = "r="..self.state.clientnonce..self.state.servernonce..",s="..base64.encode(self.state.salt)..",i="..default_i;
-               self.state["server_first_message"] = server_first_message;
-               return "challenge", server_first_message
-       else
-               if type(message) ~= "string" then return "failure", "malformed-request" end
-               -- we are processing client_final_message
-               local client_final_message = message;
+                       self.state.name = validate_username(self.state.name);
+                       if not self.state.name then
+                               log("debug", "Username violates either SASLprep or contains forbidden character sequences.")
+                               return "failure", "malformed-request", "Invalid username.";
+                       end
                
-               self.state["proof"] = client_final_message:match("p=(.+)");
-               self.state["nonce"] = client_final_message:match("r=(.+),p=");
-               self.state["channelbinding"] = client_final_message:match("c=(.+),r=");
-               if not self.state.proof or not self.state.nonce or not self.state.channelbinding then
-                       return "failure", "malformed-request", "Missing an attribute(p, r or c) in SASL message.";
-               end
+                       self.state["servernonce"] = generate_uuid();
+                       
+                       -- retreive credentials
+                       if self.profile.plain then
+                               local password, state = self.profile.plain(self.state.name, self.realm)
+                               if state == nil then return "failure", "not-authorized"
+                               elseif state == false then return "failure", "account-disabled" end
+                               
+                               password = saslprep(password);
+                               if not password then
+                                       log("debug", "Password violates SASLprep.");
+                                       return "failure", "not-authorized", "Invalid password."
+                               end
+                               self.state.salt = generate_uuid();
+                               self.state.iteration_count = default_i;
+                               self.state.salted_password = Hi(HMAC_f, password, self.state.salt, default_i);
+                       elseif self.profile["scram_"..hash_name] then
+                               local salted_password, iteration_count, salt, state = self.profile["scram-"..hash_name](self.state.name, self.realm);
+                               if state == nil then return "failure", "not-authorized"
+                               elseif state == false then return "failure", "account-disabled" end
+                               
+                               self.state.salted_password = salted_password;
+                               self.state.iteration_count = iteration_count;
+                               self.state.salt = salt
+                       end
+               
+                       local server_first_message = "r="..self.state.clientnonce..self.state.servernonce..",s="..base64.encode(self.state.salt)..",i="..self.state.iteration_count;
+                       self.state["server_first_message"] = server_first_message;
+                       return "challenge", server_first_message
+               else
+                       if type(message) ~= "string" then return "failure", "malformed-request" end
+                       -- we are processing client_final_message
+                       local client_final_message = message;
                
-               local password;
-               if self.profile.plain then
-                       local password, state = self.profile.plain(self.state.name, self.realm)
-                       if state == nil then return "failure", "not-authorized"
-                       elseif state == false then return "failure", "account-disabled" end
-                       password = saslprep(password);
-                       if not password then
-                               log("debug", "Password violates SASLprep.");
-                               return "failure", "not-authorized", "Invalid password."
+                       self.state["proof"] = client_final_message:match("p=(.+)");
+                       self.state["nonce"] = client_final_message:match("r=(.+),p=");
+                       self.state["channelbinding"] = client_final_message:match("c=(.+),r=");
+                       if not self.state.proof or not self.state.nonce or not self.state.channelbinding then
+                               return "failure", "malformed-request", "Missing an attribute(p, r or c) in SASL message.";
                        end
-               end
                
-               local SaltedPassword = Hi(hmac_sha1, password, self.state.salt, default_i)
-               local ClientKey = hmac_sha1(SaltedPassword, "Client Key")
-               local ServerKey = hmac_sha1(SaltedPassword, "Server Key")
-               local StoredKey = sha1(ClientKey)
-               local AuthMessage = "n=" .. s_match(self.state.client_first_message,"n=(.+)") .. "," .. self.state.server_first_message .. "," .. s_match(client_final_message, "(.+),p=.+")
-               local ClientSignature = hmac_sha1(StoredKey, AuthMessage)
-               local ClientProof     = binaryXOR(ClientKey, ClientSignature)
-               local ServerSignature = hmac_sha1(ServerKey, AuthMessage)
+                       local SaltedPassword = self.state.salted_password;
+                       local ClientKey = HMAC_f(SaltedPassword, "Client Key")
+                       local ServerKey = HMAC_f(SaltedPassword, "Server Key")
+                       local StoredKey = H_f(ClientKey)
+                       local AuthMessage = "n=" .. s_match(self.state.client_first_message,"n=(.+)") .. "," .. self.state.server_first_message .. "," .. s_match(client_final_message, "(.+),p=.+")
+                       local ClientSignature = HMAC_f(StoredKey, AuthMessage)
+                       local ClientProof     = binaryXOR(ClientKey, ClientSignature)
+                       local ServerSignature = HMAC_f(ServerKey, AuthMessage)
                
-               if base64.encode(ClientProof) == self.state.proof then
-                       local server_final_message = "v="..base64.encode(ServerSignature);
-                       self["username"] = self.state.name;
-                       return "success", server_final_message;
-               else
-                       return "failure", "not-authorized", "The response provided by the client doesn't match the one we calculated.";
+                       if base64.encode(ClientProof) == self.state.proof then
+                               local server_final_message = "v="..base64.encode(ServerSignature);
+                               self["username"] = self.state.name;
+                               return "success", server_final_message;
+                       else
+                               return "failure", "not-authorized", "The response provided by the client doesn't match the one we calculated.";
+                       end
                end
        end
+       return scram_hash;
 end
 
 function init(registerMechanism)
-       registerMechanism("SCRAM-SHA-1", {"plain"}, scram_sha_1);
+       local function registerSCRAMMechanism(hash_name, hash, hmac_hash)
+               registerMechanism("SCRAM-"..hash_name, {"plain", "scram_"..(hash_name:lower())}, scram_gen(hash_name:lower(), hash, hmac_hash));
+       end
+       
+       registerSCRAMMechanism("SHA-1", sha1, hmac_sha1);
 end
 
 return _M;
\ No newline at end of file