plugins/mod_dialback.lua

   1
   2 local format = string.format;
   3 local send_s2s = require "core.s2smanager".send_to_host;
   4 local s2s_make_authenticated = require "core.s2smanager".make_authenticated;
   5 local s2s_verify_dialback = require "core.s2smanager".verify_dialback;
   6
   7 local log = require "util.logger".init("mod_dialback");
   8
   9 local xmlns_dialback = "jabber:server:dialback";
  10
  11 add_handler({"s2sin_unauthed", "s2sin"}, "verify", xmlns_dialback,
  12         function (origin, stanza)
  13                 -- We are being asked to verify the key, to ensure it was generated by us
  14                 log("debug", "verifying dialback key...");
  15                 local attr = stanza.attr;
  16                 -- FIXME: Grr, ejabberd breaks this one too?? it is black and white in XEP-220 example 34
  17                 --if attr.from ~= origin.to_host then error("invalid-from"); end
  18                 local type;
  19                 if s2s_verify_dialback(attr.id, attr.from, attr.to, stanza[1]) then
  20                         type = "valid"
  21                 else
  22                         type = "invalid"
  23                         log("warn", "Asked to verify a dialback key that was incorrect. An imposter is claiming to be %s?", attr.to);
  24                 end
  25                 log("debug", "verifyied dialback key... it is %s", type);
  26                 origin.sends2s(format("<db:verify from='%s' to='%s' id='%s' type='%s'>%s</db:verify>", attr.to, attr.from, attr.id, type, stanza[1]));
  27         end);
  28
  29 add_handler("s2sin_unauthed", "result", xmlns_dialback,
  30         function (origin, stanza)
  31                 -- he wants to be identified through dialback
  32                 -- We need to check the key with the Authoritative server
  33                 local attr = stanza.attr;
  34                 local attr = stanza.attr;
  35                 origin.from_host = attr.from;
  36                 origin.to_host = attr.to;
  37                 origin.dialback_key = stanza[1];
  38                 log("debug", "asking %s if key %s belongs to them", origin.from_host, origin.dialback_key);
  39                 send_s2s(origin.to_host, origin.from_host,
  40                         format("<db:verify from='%s' to='%s' id='%s'>%s</db:verify>", origin.to_host, origin.from_host,
  41                                 origin.streamid, origin.dialback_key));
  42                 hosts[origin.to_host].s2sout[origin.from_host].dialback_verifying = origin;
  43         end);
  44
  45 add_handler({ "s2sout_unauthed", "s2sout" }, "verify", xmlns_dialback,
  46         function (origin, stanza)
  47                 if origin.dialback_verifying then
  48                         local valid;
  49                         local attr = stanza.attr;
  50                         if attr.type == "valid" then
  51                                 s2s_make_authenticated(origin.dialback_verifying);
  52                                 valid = "valid";
  53                         else
  54                                 -- Warn the original connection that is was not verified successfully
  55                                 log("warn", "dialback for "..(origin.dialback_verifying.from_host or "(unknown)").." failed");
  56                                 valid = "invalid";
  57                         end
  58                         if not origin.dialback_verifying.sends2s then
  59                                 log("warn", "Incoming s2s session %s was closed in the meantime, so we can't notify it of the db result", tostring(origin.dialback_verifying):match("%w+$"));
  60                         else
  61                                 origin.dialback_verifying.sends2s(format("<db:result from='%s' to='%s' id='%s' type='%s'>%s</db:result>",
  62                                         attr.to, attr.from, attr.id, valid, origin.dialback_verifying.dialback_key));
  63                         end
  64                 end
  65         end);
  66
  67 add_handler({ "s2sout_unauthed", "s2sout" }, "result", xmlns_dialback,
  68         function (origin, stanza)
  69                 if stanza.attr.type == "valid" then
  70                         s2s_make_authenticated(origin);
  71                 else
  72                         -- FIXME
  73                         error("dialback failed!");
  74                 end
  75         end);