util/sasl.lua

   1 require "base64"
   2 sasl = {}
   3
   4 function sasl:new_plain(onAuth, onSuccess, onFail, onWrite)
   5         local object = { mechanism = "PLAIN", onAuth = onAuth, onSuccess = onSuccess, onFail = onFail,
   6                                         onWrite = onWrite}
   7         local challenge = base64.encode("");
   8         onWrite(stanza.stanza("challenge", {xmlns = "urn:ietf:params:xml:ns:xmpp-sasl"}):text(challenge))
   9         object.feed =   function(self, stanza)
  10                                                 if (stanza.name ~= "response") then self.onFail() end
  11                                                 if (stanza.attr.xmlns ~= "urn:ietf:params:xml:ns:xmpp-sasl") then self.onFail() end
  12                                                 local response = base64.decode(stanza.tag[1])
  13                                                 local authorization = string.match(response, [[([^&\0]+)]])
  14                                                 local authentication = string.match(response, [[\0([^&\0]+)\0]])
  15                                                 local password = string.match(response, [[\0[^&\0]+\0([^&\0]+)]])
  16                                                 if self.onAuth(authorization, password) == true then
  17                                                         self.onWrite(stanza.stanza("success", {xmlns = "urn:ietf:params:xml:ns:xmpp-sasl"}))
  18                                                         self.onSuccess()
  19                                                 else
  20                                                         self.onWrite(stanza.stanza("failure", {xmlns = "urn:ietf:params:xml:ns:xmpp-sasl"}):tag("temporary-auth-failure"));
  21                                                 end
  22                                         end
  23         return object
  24 end
  25
  26 function sasl:new(mechanism, onAuth, onSuccess, onFail, onWrite)
  27         local object
  28         if mechanism == "PLAIN" then object = new_plain(onAuth, onSuccess, onFail, onWrite)
  29         else onFail()
  30         end
  31         return object
  32 end
  33
  34 module "sasl"