net/server.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9 local use_luaevent = prosody and require "core.configmanager".get("*", "use_libevent");
  10
  11 if use_luaevent then
  12         use_luaevent = pcall(require, "luaevent.core");
  13         if not use_luaevent then
  14                 log("error", "libevent not found, falling back to select()");
  15         end
  16 end
  17
  18 local server;
  19
  20 if use_luaevent then
  21         server = require "net.server_event";
  22
  23         -- Overwrite signal.signal() because we need to ask libevent to
  24         -- handle them instead
  25         local ok, signal = pcall(require, "util.signal");
  26         if ok and signal then
  27                 local _signal_signal = signal.signal;
  28                 function signal.signal(signal_id, handler)
  29                         if type(signal_id) == "string" then
  30                                 signal_id = signal[signal_id:upper()];
  31                         end
  32                         if type(signal_id) ~= "number" then
  33                                 return false, "invalid-signal";
  34                         end
  35                         return server.hook_signal(signal_id, handler);
  36                 end
  37         end
  38 else
  39         use_luaevent = false;
  40         server = require "net.server_select";
  41 end
  42
  43 if prosody then
  44         local config_get = require "core.configmanager".get;
  45         local defaults = {};
  46         for k,v in pairs(server.cfg or server.getsettings()) do
  47                 defaults[k] = v;
  48         end
  49         local function load_config()
  50                 local settings = config_get("*", "network_settings") or {};
  51                 if use_luaevent then
  52                         local event_settings = {
  53                                 ACCEPT_DELAY = settings.accept_retry_interval;
  54                                 ACCEPT_QUEUE = settings.tcp_backlog;
  55                                 CLEAR_DELAY = settings.event_clear_interval;
  56                                 CONNECT_TIMEOUT = settings.connect_timeout;
  57                                 DEBUG = settings.debug;
  58                                 HANDSHAKE_TIMEOUT = settings.ssl_handshake_timeout;
  59                                 MAX_CONNECTIONS = settings.max_connections;
  60                                 MAX_HANDSHAKE_ATTEMPTS = settings.max_ssl_handshake_roundtrips;
  61                                 MAX_READ_LENGTH = settings.max_receive_buffer_size;
  62                                 MAX_SEND_LENGTH = settings.max_send_buffer_size;
  63                                 READ_TIMEOUT = settings.read_timeout;
  64                                 WRITE_TIMEOUT = settings.send_timeout;
  65                         };
  66
  67                         for k,default in pairs(defaults) do
  68                                 server.cfg[k] = event_settings[k] or default;
  69                         end
  70                 else
  71                         local select_settings = {};
  72                         for k,default in pairs(defaults) do
  73                                 select_settings[k] = settings[k] or default;
  74                         end
  75                         server.changesettings(select_settings);
  76                 end
  77         end
  78         load_config();
  79         prosody.events.add_handler("config-reloaded", load_config);
  80 end
  81
  82 -- require "net.server" shall now forever return this,
  83 -- ie. server_select or server_event as chosen above.
  84 return server;