core/sessionmanager.lua

   1
   2 local tonumber, tostring = tonumber, tostring;
   3 local ipairs = ipairs;
   4
   5 local m_random = math.random;
   6 local format = string.format;
   7
   8 local print = print;
   9
  10 local hosts = hosts;
  11
  12 local modulemanager = require "core.modulemanager";
  13 local log = require "util.logger".init("sessionmanager");
  14 local error = error;
  15 local uuid_generate = require "util.uuid".uuid_generate;
  16 module "sessionmanager"
  17
  18 function new_session(conn)
  19         local session = { conn = conn, notopen = true, priority = 0, type = "c2s_unauthed" };
  20         local w = conn.write;
  21         session.send = function (t) w(tostring(t)); end
  22         return session;
  23 end
  24
  25 function destroy_session(session)
  26 end
  27
  28 function send_to_session(session, data)
  29         log("debug", "Sending: %s", tostring(data));
  30         session.conn.write(tostring(data));
  31 end
  32
  33 function make_authenticated(session, username)
  34         session.username = username;
  35         session.resource = resource;
  36         if session.type == "c2s_unauthed" then
  37                 session.type = "c2s";
  38         end
  39 end
  40
  41 function bind_resource(session, resource)
  42         if not session.username then return false, "auth"; end
  43         if session.resource then return false, "constraint"; end -- We don't support binding multiple resources
  44         resource = resource or uuid_generate();
  45         --FIXME: Randomly-generated resources must be unique per-user, and never conflict with existing
  46
  47         if not hosts[session.host].sessions[session.username] then
  48                 hosts[session.host].sessions[session.username] = { sessions = {} };
  49         else
  50                 if hosts[session.host].sessions[session.username].sessions[resource] then
  51                         -- Resource conflict
  52                         return false, "conflict";
  53                 end
  54         end
  55
  56         session.resource = resource;
  57         session.full_jid = session.username .. '@' .. session.host .. '/' .. resource;
  58         hosts[session.host].sessions[session.username].sessions[resource] = session;
  59
  60         return true;
  61 end
  62
  63 function streamopened(session, attr)
  64                                                 local send = session.send;
  65                                                 session.host = attr.to or error("Client failed to specify destination hostname");
  66                                                 session.version = tonumber(attr.version) or 0;
  67                                                 session.streamid = m_random(1000000, 99999999);
  68                                                 print(session, session.host, "Client opened stream");
  69                                                 send("<?xml version='1.0'?>");
  70                                                 send(format("<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' id='%s' from='%s' version='1.0'>", session.streamid, session.host));
  71
  72                                                 local features = {};
  73                                                 modulemanager.fire_event("stream-features", session, features);
  74
  75                                                 send("<stream:features>");
  76
  77                                                 for _, feature in ipairs(features) do
  78                                                         send_to_session(session, tostring(feature));
  79                                                 end
  80
  81                                                 send("</stream:features>");
  82                                                 log("info", "core", "Stream opened successfully");
  83                                                 session.notopen = nil;
  84 end
  85
  86 return _M;