Merge with 0.5
authorMatthew Wild <mwild1@gmail.com>
Wed, 29 Jul 2009 17:11:37 +0000 (18:11 +0100)
committerMatthew Wild <mwild1@gmail.com>
Wed, 29 Jul 2009 17:11:37 +0000 (18:11 +0100)
1  2 
prosody

diff --combined prosody
index e9f00598875278829e720cc1b79f6e23b5fccc0e,4984ed15f906fa766d4c3781f7852da1f45b68f1..0618a1f3d3b52aa15750ba134edf08c4178b1fee
+++ b/prosody
@@@ -9,9 -9,9 +9,9 @@@
  
  -- Will be modified by configure script if run --
  
- CFG_SOURCEDIR=nil;
+ CFG_SOURCEDIR=os.getenv("PROSODY_SRCDIR");
  CFG_CONFIGDIR=os.getenv("PROSODY_CFGDIR");
- CFG_PLUGINDIR=nil;
+ CFG_PLUGINDIR=os.getenv("PROSODY_PLUGINDIR");
  CFG_DATADIR=os.getenv("PROSODY_DATADIR");
  
  -- -- -- -- -- -- -- ---- -- -- -- -- -- -- -- --
@@@ -36,7 -36,7 +36,7 @@@ pcall(require, "luarocks.require"
  
  config = require "core.configmanager"
  
 -do
 +function read_config()
        -- TODO: Check for other formats when we add support for them
        -- Use lfs? Make a new conf/ dir?
        local ok, level, err = config.load((CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
        end
  end
  
 ---- Initialize logging
 -require "core.loggingmanager"
 -
 ---- Check runtime dependencies
 -require "util.dependencies"
 -
 ---- Load socket framework
 -local server = require "net.server"
 -
 -bare_sessions = {};
 -full_sessions = {};
 -hosts = {};
 -
 --- Global 'prosody' object
 -prosody = {};
 -local prosody = prosody;
 -
 -prosody.bare_sessions = bare_sessions;
 -prosody.full_sessions = full_sessions;
 -prosody.hosts = hosts;
 -
 -prosody.paths = { source = CFG_SOURCEDIR, config = CFG_CONFIGDIR, 
 -                  plugins = CFG_PLUGINDIR, data = CFG_DATADIR };
 -
 -prosody.arg = arg;
 -
 -prosody.events = require "util.events".new();
 +function load_libraries()
 +      --- Initialize logging
 +      require "core.loggingmanager"
 +      
 +      --- Check runtime dependencies
 +      require "util.dependencies"
 +      
 +      --- Load socket framework
 +      server = require "net.server"
 +end   
 +
 +function init_global_state()
 +      bare_sessions = {};
 +      full_sessions = {};
 +      hosts = {};
 +
 +      -- Global 'prosody' object
 +      prosody = {};
 +      local prosody = prosody;
 +      
 +      prosody.bare_sessions = bare_sessions;
 +      prosody.full_sessions = full_sessions;
 +      prosody.hosts = hosts;
 +      
 +      prosody.paths = { source = CFG_SOURCEDIR, config = CFG_CONFIGDIR, 
 +                        plugins = CFG_PLUGINDIR, data = CFG_DATADIR };
 +      
 +      prosody.arg = _G.arg;
  
 --- Try to determine version
 -local version_file = io.open((CFG_SOURCEDIR or ".").."/prosody.version");
 -if version_file then
 -      prosody.version = version_file:read("*a"):gsub("%s*$", "");
 -      version_file:close();
 -      if #prosody.version == 12 and prosody.version:match("^[a-f0-9]+$") then
 -              prosody.version = "hg:"..prosody.version;
 +      prosody.events = require "util.events".new();
 +      
 +      
 +      -- Function to reload the config file
 +      function prosody.reload_config()
 +              log("info", "Reloading configuration file");
 +              prosody.events.fire_event("reloading-config");
 +              local ok, level, err = config.load((rawget(_G, "CFG_CONFIGDIR") or ".").."/prosody.cfg.lua");
 +              if not ok then
 +                      if level == "parser" then
 +                              log("error", "There was an error parsing the configuration file: %s", tostring(err));
 +                      elseif level == "file" then
 +                              log("error", "Couldn't read the config file when trying to reload: %s", tostring(err));
 +                      end
 +              end
 +              return ok, (err and tostring(level)..": "..tostring(err)) or nil;
        end
 -else
 -      prosody.version = "unknown";
 -end
  
 -log("info", "Hello and welcome to Prosody version %s", prosody.version);
 -
 ---- Load and initialise core modules
 -require "util.import"
 -require "core.xmlhandlers"
 -require "core.rostermanager"
 -require "core.eventmanager"
 -require "core.hostmanager"
 -require "core.modulemanager"
 -require "core.usermanager"
 -require "core.sessionmanager"
 -require "core.stanza_router"
 -
 -require "util.array"
 -require "util.iterators"
 -require "util.timer"
 -
 --- Commented to protect us from 
 --- the second kind of people
 ---[[ 
 -pcall(require, "remdebug.engine");
 -if remdebug then remdebug.engine.start() end
 -]]
 -
 -local cl = require "net.connlisteners";
 -
 -require "util.stanza"
 -require "util.jid"
 -
 -------------------------------------------------------------------------
 -
 -
 -------------- Begin code without a home ---------------------
 -
 -local data_path = config.get("*", "core", "data_path") or CFG_DATADIR or "data";
 -require "util.datamanager".set_data_path(data_path);
 -require "util.datamanager".add_callback(function(username, host, datastore, data)
 -      if config.get(host, "core", "anonymous_login") then
 -              return false;
 +      -- Function to reopen logfiles
 +      function prosody.reopen_logfiles()
 +              log("info", "Re-opening log files");
 +              eventmanager.fire_event("reopen-log-files"); -- Handled by appropriate log sinks
 +              prosody.events.fire_event("reopen-log-files");
        end
 -      return username, host, datastore, data;
 -end);
  
 ------------ End of out-of-place code --------------
 +      -- Function to initiate prosody shutdown
 +      function prosody.shutdown(reason)
 +              log("info", "Shutting down: %s", reason or "unknown reason");
 +              prosody.shutdown_reason = reason;
 +              prosody.events.fire_event("server-stopping", {reason = reason});
 +              server.setquitting(true);
 +      end
 +end
  
 --- Function to reload the config file
 -function prosody.reload_config()
 -      log("info", "Reloading configuration file");
 -      prosody.events.fire_event("reloading-config");
 -      local ok, level, err = config.load((rawget(_G, "CFG_CONFIGDIR") or ".").."/prosody.cfg.lua");
 -      if not ok then
 -              if level == "parser" then
 -                      log("error", "There was an error parsing the configuration file: %s", tostring(err));
 -              elseif level == "file" then
 -                      log("error", "Couldn't read the config file when trying to reload: %s", tostring(err));
 +function read_version()
 +      -- Try to determine version
 +      local version_file = io.open((CFG_SOURCEDIR or ".").."/prosody.version");
 +      if version_file then
 +              prosody.version = version_file:read("*a"):gsub("%s*$", "");
 +              version_file:close();
 +              if #prosody.version == 12 and prosody.version:match("^[a-f0-9]+$") then
 +                      prosody.version = "hg:"..prosody.version;
                end
 +      else
 +              prosody.version = "unknown";
        end
  end
  
 --- Function to reopen logfiles
 -function prosody.reopen_logfiles()
 -      log("info", "Re-opening log files");
 -      eventmanager.fire_event("reopen-log-files"); -- Handled by appropriate log sinks
 -      prosody.events.fire_event("reopen-log-files");
 +function load_secondary_libraries()
 +      --- Load and initialise core modules
 +      require "util.import"
 +      require "core.xmlhandlers"
 +      require "core.rostermanager"
 +      require "core.eventmanager"
 +      require "core.hostmanager"
 +      require "core.modulemanager"
 +      require "core.usermanager"
 +      require "core.sessionmanager"
 +      require "core.stanza_router"
 +
 +      require "util.array"
 +      require "util.iterators"
 +      require "util.timer"
 +      require "util.helpers"
 +      
 +      -- Commented to protect us from 
 +      -- the second kind of people
 +      --[[ 
 +      pcall(require, "remdebug.engine");
 +      if remdebug then remdebug.engine.start() end
 +      ]]
 +
 +      require "net.connlisteners";
 +      
 +      require "util.stanza"
 +      require "util.jid"
  end
  
 --- Function to initiate prosody shutdown
 -function prosody.shutdown(reason)
 -      log("info", "Shutting down: %s", reason or "unknown reason");
 -      prosody.events.fire_event("server-stopping", {reason = reason});
 -      server.setquitting(true);
 +function init_data_store()
 +      local data_path = config.get("*", "core", "data_path") or CFG_DATADIR or "data";
 +      require "util.datamanager".set_data_path(data_path);
 +      require "util.datamanager".add_callback(function(username, host, datastore, data)
 +              if config.get(host, "core", "anonymous_login") then
 +                      return false;
 +              end
 +              return username, host, datastore, data;
 +      end);
  end
  
 --- Signal to modules that we are ready to start
 -eventmanager.fire_event("server-starting");
 -prosody.events.fire_event("server-starting");
 +function prepare_to_start()
 +      -- Signal to modules that we are ready to start
 +      eventmanager.fire_event("server-starting");
 +      prosody.events.fire_event("server-starting");
  
 --- Load SSL settings from config, and create a ctx table
 -local global_ssl_ctx = ssl and config.get("*", "core", "ssl");
 -if global_ssl_ctx then
 -      local default_ssl_ctx = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
 -      setmetatable(global_ssl_ctx, { __index = default_ssl_ctx });
 -end
 +      -- Load SSL settings from config, and create a ctx table
 +      local global_ssl_ctx = ssl and config.get("*", "core", "ssl");
 +      if global_ssl_ctx then
 +              local default_ssl_ctx = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
 +              setmetatable(global_ssl_ctx, { __index = default_ssl_ctx });
 +      end
  
 --- start listening on sockets
 -function net_activate_ports(option, listener, default, conntype)
 -      if not cl.get(listener) then return; end
 -      local ports = config.get("*", "core", option.."_ports") or default;
 -      if type(ports) == "number" then ports = {ports} end;
 -      
 -      if type(ports) ~= "table" then
 -              log("error", "core."..option.." is not a table");
 -      else
 -              for _, port in ipairs(ports) do
 -                      if type(port) ~= "number" then
 -                              log("error", "Non-numeric "..option.."_ports: "..tostring(port));
 -                      else
 -                              cl.start(listener, { 
 -                                      ssl = conntype ~= "tcp" and global_ssl_ctx,
 -                                      port = port,
 -                                      interface = config.get("*", "core", option.."_interface") 
 -                                              or cl.get(listener).default_interface 
 -                                              or config.get("*", "core", "interface"),
 -                                      type = conntype
 -                              });
 +      local cl = require "net.connlisteners";
 +      -- start listening on sockets
 +      function net_activate_ports(option, listener, default, conntype)
 +              if not cl.get(listener) then return; end
 +              local ports = config.get("*", "core", option.."_ports") or default;
 +              if type(ports) == "number" then ports = {ports} end;
 +              
 +              if type(ports) ~= "table" then
 +                      log("error", "core."..option.." is not a table");
 +              else
 +                      for _, port in ipairs(ports) do
 +                              if type(port) ~= "number" then
 +                                      log("error", "Non-numeric "..option.."_ports: "..tostring(port));
 +                              else
 +                                      cl.start(listener, { 
 +                                              ssl = conntype ~= "tcp" and global_ssl_ctx,
 +                                              port = port,
 +                                              interface = config.get("*", "core", option.."_interface") 
 +                                                      or cl.get(listener).default_interface 
 +                                                      or config.get("*", "core", "interface"),
 +                                              type = conntype
 +                                      });
 +                              end
                        end
                end
        end
 -end
 -
 -net_activate_ports("c2s", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
 -net_activate_ports("s2s", "xmppserver", {5269}, "tcp");
 -net_activate_ports("component", "xmppcomponent", {}, "tcp");
 -net_activate_ports("legacy_ssl", "xmppclient", {}, "ssl");
 -net_activate_ports("console", "console", {5582}, "tcp");
 -
 --- Catch global accesses --
 -local locked_globals_mt = { __index = function (t, k) error("Attempt to read a non-existent global '"..k.."'", 2); end, __newindex = function (t, k, v) error("Attempt to set a global: "..tostring(k).." = "..tostring(v), 2); end }
  
 -function prosody.unlock_globals()
 -      setmetatable(_G, nil);
 -end
 +      net_activate_ports("c2s", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
 +      net_activate_ports("s2s", "xmppserver", {5269}, "tcp");
 +      net_activate_ports("component", "xmppcomponent", {}, "tcp");
 +      net_activate_ports("legacy_ssl", "xmppclient", {}, "ssl");
 +      net_activate_ports("console", "console", {5582}, "tcp");
 +
 +      prosody.start_time = os.time();
 +end   
 +
 +function init_global_protection()
 +      -- Catch global accesses --
 +      local locked_globals_mt = { __index = function (t, k) error("Attempt to read a non-existent global '"..k.."'", 2); end, __newindex = function (t, k, v) error("Attempt to set a global: "..tostring(k).." = "..tostring(v), 2); end }
 +              
 +      function prosody.unlock_globals()
 +              setmetatable(_G, nil);
 +      end
 +      
 +      function prosody.lock_globals()
 +              setmetatable(_G, locked_globals_mt);
 +      end
  
 -function prosody.lock_globals()
 -      setmetatable(_G, locked_globals_mt);
 +      -- And lock now...
 +      prosody.lock_globals();
  end
  
 --- And lock now...
 -prosody.lock_globals();
 -
 -prosody.start_time = os.time();
 -
 -eventmanager.fire_event("server-started");
 -prosody.events.fire_event("server-started");
 -
 --- Error handler for errors that make it this far
 -local function catch_uncaught_error(err)
 -      if err:match("%d*: interrupted!$") then
 -              return "quitting";
 +function loop()
 +      -- Error handler for errors that make it this far
 +      local function catch_uncaught_error(err)
 +              if err:match("%d*: interrupted!$") then
 +                      return "quitting";
 +              end
 +              
 +              log("error", "Top-level error, please report:\n%s", tostring(err));
 +              local traceback = debug.traceback("", 2);
 +              if traceback then
 +                      log("error", "%s", traceback);
 +              end
 +              
 +              prosody.events.fire_event("very-bad-error", {error = err, traceback = traceback});
        end
        
 -      log("error", "Top-level error, please report:\n%s", tostring(err));
 -      local traceback = debug.traceback("", 2);
 -      if traceback then
 -              log("error", "%s", traceback);
 +      while select(2, xpcall(server.loop, catch_uncaught_error)) ~= "quitting" do
 +              socket.sleep(0.2);
        end
 -      
 -      prosody.events.fire_event("very-bad-error", {error = err, traceback = traceback});
 -end
 -
 -while select(2, xpcall(server.loop, catch_uncaught_error)) ~= "quitting" do
 -      socket.sleep(0.2);
  end
  
 -log("info", "Shutdown status: Cleaning up");
 -prosody.events.fire_event("server-cleanup");
 -
 --- Ok, we're quitting I know, but we
 --- need to do some tidying before we go :)
 -server.setquitting(false);
 -
 -log("info", "Shutdown status: Closing all active sessions");
 -for hostname, host in pairs(hosts) do
 -      log("debug", "Shutdown status: Closing client connections for %s", hostname)
 -      if host.sessions then
 -              for username, user in pairs(host.sessions) do
 -                      for resource, session in pairs(user.sessions) do
 -                              log("debug", "Closing connection for %s@%s/%s", username, hostname, resource);
 -                              session:close("system-shutdown");
 +function cleanup()
 +      log("info", "Shutdown status: Cleaning up");
 +      prosody.events.fire_event("server-cleanup");
 +      
 +      -- Ok, we're quitting I know, but we
 +      -- need to do some tidying before we go :)
 +      server.setquitting(false);
 +      
 +      log("info", "Shutdown status: Closing all active sessions");
 +      for hostname, host in pairs(hosts) do
 +              log("debug", "Shutdown status: Closing client connections for %s", hostname)
 +              if host.sessions then
 +                      local reason = { condition = "system-shutdown", text = "Server is shutting down" };
 +                      if prosody.shutdown_reason then
 +                              reason.text = reason.text..": "..prosody.shutdown_reason;
 +                      end
 +                      for username, user in pairs(host.sessions) do
 +                              for resource, session in pairs(user.sessions) do
 +                                      log("debug", "Closing connection for %s@%s/%s", username, hostname, resource);
 +                                      session:close(reason);
 +                              end
                        end
                end
 -      end
        
 -      log("debug", "Shutdown status: Closing outgoing s2s connections from %s", hostname);
 -      if host.s2sout then
 -              for remotehost, session in pairs(host.s2sout) do
 -                      if session.close then
 -                              session:close("system-shutdown");
 -                      else
 -                              log("warn", "Unable to close outgoing s2s session to %s, no session:close()?!", remotehost);
 +              log("debug", "Shutdown status: Closing outgoing s2s connections from %s", hostname);
 +              if host.s2sout then
 +                      for remotehost, session in pairs(host.s2sout) do
 +                              if session.close then
 +                                      session:close("system-shutdown");
 +                              else
 +                                      log("warn", "Unable to close outgoing s2s session to %s, no session:close()?!", remotehost);
 +                              end
                        end
                end
        end
 +
 +      log("info", "Shutdown status: Closing all server connections");
 +      server.closeall();
 +      
 +      server.setquitting(true);
  end
  
 -log("info", "Shutdown status: Closing all server connections");
 -server.closeall();
 +read_config();
 +load_libraries();
 +init_global_state();
 +read_version();
 +log("info", "Hello and welcome to Prosody version %s", prosody.version);
 +load_secondary_libraries();
 +init_data_store();
 +prepare_to_start();
 +init_global_protection();
 +
 +eventmanager.fire_event("server-started");
 +prosody.events.fire_event("server-started");
  
 -server.setquitting(true);
 +loop();
  
 +log("info", "Shutting down...");
 +cleanup();
  eventmanager.fire_event("server-stopped");
  prosody.events.fire_event("server-stopped");
 -log("info", "Shutdown status: Complete!");
 +log("info", "Shutdown complete");
 +