util.iterators: Add skip() to skip the first n items of an iterator

[prosody.git] / util / sasl.lua

diff --git a/util/sasl.lua b/util/sasl.lua
index a7e151a233fb3b5fd948313b204d0fe9bb999360..5d1b9f17f3d5e2e8e76f517b522bf45ebfee20c7 100644 (file)
--- a/util/sasl.lua
+++ b/util/sasl.lua
@@ -88,17 +88,20 @@ end
 
 -- get a list of possible SASL mechanims to use
 function method:mechanisms()
-       local mechanisms = {}
-       for backend, f in pairs(self.profile) do
-               if backend_mechanism[backend] then
-                       for _, mechanism in ipairs(backend_mechanism[backend]) do
-                               if not self.restrict:contains(mechanism) then
-                                       mechanisms[mechanism] = true;
+       local mechanisms = self.mechs;
+       if not mechanisms then
+               mechanisms = {}
+               for backend, f in pairs(self.profile) do
+                       if backend_mechanism[backend] then
+                               for _, mechanism in ipairs(backend_mechanism[backend]) do
+                                       if not self.restrict:contains(mechanism) then
+                                               mechanisms[mechanism] = true;
+                                       end
                                end
                        end
                end
+               self.mechs = mechanisms;
        end
-       self.mechs = mechanisms;
        return mechanisms;
 end
 
@@ -108,7 +111,7 @@ function method:select(mechanism)
                return false;
        end
        
-       self.mech_i = self.mechs[mechanism];
+       self.mech_i = mechanisms[self:mechanisms()[mechanism] and mechanism];
        return (self.mech_i ~= nil);
 end