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