net/multiplex_listener.lua

   1
   2 local connlisteners_register = require "net.connlisteners".register;
   3 local connlisteners_get = require "net.connlisteners".get;
   4
   5 local httpserver_listener = connlisteners_get("httpserver");
   6 local xmppserver_listener = connlisteners_get("xmppserver");
   7 local xmppclient_listener = connlisteners_get("xmppclient");
   8 local xmppcomponent_listener = connlisteners_get("xmppcomponent");
   9
  10 local server = { default_mode = "*a" };
  11
  12 local buffer = {};
  13
  14 function server.onincoming(conn, data)
  15         if not data then return; end
  16         local buf = buffer[conn];
  17         buffer[conn] = nil;
  18         buf = buf and buf..data or data;
  19         if buf:match("^[a-zA-Z]") then
  20                 local listener = httpserver_listener;
  21                 conn:setlistener(listener);
  22                 listener.onincoming(conn, buf);
  23         elseif buf:match(">") then
  24                 local listener;
  25                 local xmlns = buf:match("%sxmlns%s*=%s*['\"]([^'\"]*)");
  26                 if xmlns == "jabber:server" then
  27                         listener = xmppserver_listener;
  28                 elseif xmlns == "jabber:component:accept" then
  29                         listener = xmppcomponent_listener;
  30                 else
  31                         listener = xmppclient_listener;
  32                 end
  33                 conn:setlistener(listener);
  34                 listener.onincoming(conn, buf);
  35         elseif #buf > 1024 then
  36                 conn:close();
  37         else
  38                 buffer[conn] = buf;
  39         end
  40 end
  41
  42 function server.ondisconnect(conn, err)
  43         buffer[conn] = nil; -- warn if no buffer?
  44 end
  45
  46 connlisteners_register("multiplex", server);