Mercurial > prosody-hg

comparison plugins/mod_net_multiplex.lua @ 4797:e239668aa6d2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge 0.9->trunk
author Matthew Wild <mwild1@gmail.com>
date Sun, 29 Apr 2012 02:10:55 +0100
parents d5739b8b7161
children bcabea740c00
comparison
equal deleted inserted replaced
4796:04a34287dc12 4797:e239668aa6d2
1 module:set_global();
2
3 local max_buffer_len = module:get_option_number("multiplex_buffer_size", 1024);
4
5 local portmanager = require "core.portmanager";
6
7 local available_services = {};
8
9 local function add_service(service)
10 local multiplex_pattern = service.multiplex and service.multiplex.pattern;
11 if multiplex_pattern then
12 module:log("debug", "Adding multiplex service %q with pattern %q", service.name, multiplex_pattern);
13 available_services[service] = multiplex_pattern;
14 else
15 module:log("debug", "Service %q is not multiplex-capable", service.name);
16 end
17 end
18 module:hook("service-added", function (event) add_service(event.service); end);
19 module:hook("service-removed", function (event) available_services[event.service] = nil; end);
20
21 for service_name, services in pairs(portmanager.get_registered_services()) do
22 for i, service in ipairs(services) do
23 add_service(service);
24 end
25 end
26
27 local buffers = {};
28
29 local listener = { default_mode = "*a" };
30
31 function listener.onconnect()
32 end
33
34 function listener.onincoming(conn, data)
35 if not data then return; end
36 local buf = buffers[conn];
37 buffers[conn] = nil;
38 buf = buf and buf..data or data;
39 for service, multiplex_pattern in pairs(available_services) do
40 if buf:match(multiplex_pattern) then
41 module:log("debug", "Routing incoming connection to %s", service.name);
42 local listener = service.listener;
43 conn:setlistener(listener);
44 local onconnect = listener.onconnect;
45 if onconnect then onconnect(conn) end
46 return listener.onincoming(conn, buf);
47 end
48 end
49 if #buf > max_buffer_len then -- Give up
50 conn:close();
51 else
52 buffers[conn] = buf;
53 end
54 end
55
56 function listener.ondisconnect(conn, err)
57 buffers[conn] = nil; -- warn if no buffer?
58 end
59
60 module:add_item("net-provider", {
61 name = "multiplex";
62 config_prefix = "";
63 listener = listener;
64 });
65
66 module:provides("net", {
67 name = "multiplex_ssl";
68 config_prefix = "ssl";
69 listener = listener;
70 });