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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
-- Prosody IM
-- Copyright (C) 2008-2010 Matthew Wild
-- Copyright (C) 2008-2010 Waqas Hussain
--
-- This project is MIT/X11 licensed. Please see the
-- COPYING file in the source package for more information.
--
if not (prosody and prosody.config_loaded) then
-- This module only supports loading inside Prosody, outside Prosody
-- you should directly require net.server_select or server_event, etc.
error(debug.traceback("Loading outside Prosody or Prosody not yet initialized"), 0);
end
local log = require "util.logger".init("net.server");
local have_util_poll = pcall(require, "util.poll");
local default_backend = have_util_poll and "epoll" or "select";
local server_type = require "core.configmanager".get("*", "network_backend") or default_backend;
if require "core.configmanager".get("*", "use_libevent") then
server_type = "event";
end
if server_type == "event" then
if not pcall(require, "luaevent.core") then
log("error", "libevent not found, falling back to select()");
server_type = "select"
end
end
local server;
local set_config;
if server_type == "event" then
server = require "net.server_event";
local defaults = {};
for k,v in pairs(server.cfg) do
defaults[k] = v;
end
function set_config(settings)
local event_settings = {
ACCEPT_DELAY = settings.accept_retry_interval;
ACCEPT_QUEUE = settings.tcp_backlog;
CLEAR_DELAY = settings.event_clear_interval;
CONNECT_TIMEOUT = settings.connect_timeout;
DEBUG = settings.debug;
HANDSHAKE_TIMEOUT = settings.ssl_handshake_timeout;
MAX_CONNECTIONS = settings.max_connections;
MAX_HANDSHAKE_ATTEMPTS = settings.max_ssl_handshake_roundtrips;
MAX_READ_LENGTH = settings.max_receive_buffer_size;
MAX_SEND_LENGTH = settings.max_send_buffer_size;
READ_TIMEOUT = settings.read_timeout;
WRITE_TIMEOUT = settings.send_timeout;
};
for k,default in pairs(defaults) do
server.cfg[k] = event_settings[k] or default;
end
end
elseif server_type == "select" then
server = require "net.server_select";
local defaults = {};
for k,v in pairs(server.getsettings()) do
defaults[k] = v;
end
function set_config(settings)
local select_settings = {};
for k,default in pairs(defaults) do
select_settings[k] = settings[k] or default;
end
server.changesettings(select_settings);
end
else
server = require("net.server_"..server_type);
set_config = server.set_config;
if not server.get_backend then
function server.get_backend()
return server_type;
end
end
end
-- If server.hook_signal exists, replace signal.signal()
local has_signal, signal = pcall(require, "util.signal");
if has_signal then
if server.hook_signal then
function signal.signal(signal_id, handler)
if type(signal_id) == "string" then
signal_id = signal[signal_id:upper()];
end
if type(signal_id) ~= "number" then
return false, "invalid-signal";
end
return server.hook_signal(signal_id, handler);
end
else
server.hook_signal = signal.signal;
end
else
if not server.hook_signal then
server.hook_signal = function()
return false, "signal hooking not supported"
end
end
end
if prosody and set_config then
local config_get = require "core.configmanager".get;
local function load_config()
local settings = config_get("*", "network_settings") or {};
return set_config(settings);
end
load_config();
prosody.events.add_handler("config-reloaded", load_config);
end
-- require "net.server" shall now forever return this,
-- ie. server_select or server_event as chosen above.
return server;
|
