configmanager: nameprep VirtualHost and Component names

[prosody.git] / net / server.lua

diff --git a/net/server.lua b/net/server.lua
index a23257358c5a3af3f09b8180956e2e725776b04e..375e7081f670bb760f46fe8df757215a34e9617d 100644 (file)
--- a/net/server.lua
+++ b/net/server.lua
@@ -1,26 +1,84 @@
-local have_luaevent = pcall(require, "luaevent.core");
-local use_luaevent = require "core.configmanager".get("*", "core", "use_libevent");
+-- 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.
+--
+
+local use_luaevent = prosody and require "core.configmanager".get("*", "use_libevent");
+
+if use_luaevent then
+       use_luaevent = pcall(require, "luaevent.core");
+       if not use_luaevent then
+               log("error", "libevent not found, falling back to select()");
+       end
+end
 
 local server;
 
-if have_luaevent and use_luaevent == true then
+if use_luaevent then
        server = require "net.server_event";
-       -- util.timer requires "net.server", so instead of having
-       -- Lua look for, and load us again (causing a loop) - set this here
-       -- (usually it isn't set until we return, look down there...)
-       package.loaded["net.server"] = server;
-       
-       -- Backwards compatibility for timers, addtimer
-       -- called a function roughly every second
-       local add_task = require "util.timer".add_task;
-       function server.addtimer(f)
-               return add_task(1, function (...) f(...); return 1; end);
+
+       -- Overwrite signal.signal() because we need to ask libevent to
+       -- handle them instead
+       local ok, signal = pcall(require, "util.signal");
+       if ok and signal then
+               local _signal_signal = signal.signal;
+               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
        end
 else
+       use_luaevent = false;
        server = require "net.server_select";
-       package.loaded["net.server"] = server;
+end
+
+if prosody then
+       local config_get = require "core.configmanager".get;
+       local defaults = {};
+       for k,v in pairs(server.cfg or server.getsettings()) do
+               defaults[k] = v;
+       end
+       local function load_config()
+               local settings = config_get("*", "network_settings") or {};
+               if use_luaevent then
+                       local event_settings = {
+                               ACCEPT_DELAY = settings.event_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
+               else
+                       local select_settings = {};
+                       for k,default in pairs(defaults) do
+                               select_settings[k] = settings[k] or default;
+                       end
+                       server.changesettings(select_settings);
+               end
+       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; 
+return server;