Add lock_globals() and unlock_globals() functions (for when you really need to use...
[prosody.git] / prosody
diff --git a/prosody b/prosody
index f63fc50dd1a875d3368c5d02957a24ecb39e2732..42e4eb502d909fe33133eef6b2f6686daf1c79be 100755 (executable)
--- a/prosody
+++ b/prosody
@@ -1,5 +1,5 @@
 #!/usr/bin/env lua
--- Prosody IM v0.3
+-- Prosody IM v0.4
 -- Copyright (C) 2008-2009 Matthew Wild
 -- Copyright (C) 2008-2009 Waqas Hussain
 -- 
@@ -7,14 +7,14 @@
 -- COPYING file in the source package for more information.
 --
 
--- Config here --
+-- Will be modified by configure script if run --
 
 CFG_SOURCEDIR=nil;
 CFG_CONFIGDIR=os.getenv("PROSODY_CFGDIR");
 CFG_PLUGINDIR=nil;
 CFG_DATADIR=os.getenv("PROSODY_DATADIR");
 
--- -- -- -- -- --
+-- -- -- -- -- -- -- ---- -- -- -- -- -- -- -- --
 
 if CFG_SOURCEDIR then
        package.path = CFG_SOURCEDIR.."/?.lua;"..package.path
@@ -32,29 +32,38 @@ pcall(require, "luarocks.require")
 
 
 config = require "core.configmanager"
-log = require "util.logger".init("general");
-
--- Disable log output, needs to read from config
--- require "util.logger".setwriter(function () end);
 
 do
        -- TODO: Check for other formats when we add support for them
        -- Use lfs? Make a new conf/ dir?
-       local ok, err = config.load((CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
+       local ok, level, err = config.load((CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
        if not ok then
                print("");
                print("**************************");
-               print("Prosody was unable to find the configuration file.");
-               print("We looked for: "..(CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
-               print("A sample config file is included in the Prosody download called prosody.cfg.lua.dist");
-               print("Copy or rename it to prosody.cfg.lua and edit as necessary.");
+               if level == "parser" then
+                       print("A problem occured while reading the config file "..(CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
+                       local err_line, err_message = tostring(err):match("%[string .-%]:(%d*): (.*)");
+                       print("Error"..(err_line and (" on line "..err_line) or "")..": "..(err_message or tostring(err)));
+                       print("");
+               elseif level == "file" then
+                       print("Prosody was unable to find the configuration file.");
+                       print("We looked for: "..(CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
+                       print("A sample config file is included in the Prosody download called prosody.cfg.lua.dist");
+                       print("Copy or rename it to prosody.cfg.lua and edit as necessary.");
+               end
                print("More help on configuring Prosody can be found at http://prosody.im/doc/configure");
                print("Good luck!");
                print("**************************");
+               print("");
                os.exit(1);
        end
 end
 
+log = require "util.logger".init("general");
+
+-- Disable log output, needs to read from config
+-- require "util.logger".setwriter(function () end);
+
 require "util.dependencies"
 
 local server = require "net.server"
@@ -76,7 +85,12 @@ require "core.usermanager"
 require "core.sessionmanager"
 require "core.stanza_router"
 
---[[
+require "util.array"
+require "util.iterators"
+
+-- Commented to protect us from 
+-- the second kind of people
+--[[ 
 pcall(require, "remdebug.engine");
 if remdebug then remdebug.engine.start() end
 ]]
@@ -93,15 +107,14 @@ require "util.jid"
 
 local data_path = config.get("*", "core", "data_path") or CFG_DATADIR or "data";
 require "util.datamanager".set_data_path(data_path);
+require "util.datamanager".set_callback(function(username, host, datastore)
+       return config.get(host, "core", "anonymous_login");
+end);
 
 ----------- End of out-of-place code --------------
 
 eventmanager.fire_event("server-starting");
 
-
--- setup error handling
-setmetatable(_G, { __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 });
-
 local global_ssl_ctx = config.get("*", "core", "ssl");
 if global_ssl_ctx then
        local default_ssl_ctx = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
@@ -109,9 +122,9 @@ if global_ssl_ctx then
 end
 
 -- start listening on sockets
-local function do_ports(option, listener, default, conntype)
+function net_activate_ports(option, listener, default, conntype)
        local ports = config.get("*", "core", option) or default;
-       --if type(ports) == "number" then ports = {ports} end;
+       if type(ports) == "number" then ports = {ports} end;
        
        if type(ports) ~= "table" then
                log("error", "core."..option.." is not a table");
@@ -126,9 +139,9 @@ local function do_ports(option, listener, default, conntype)
        end
 end
 
-do_ports("c2s_ports", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
-do_ports("s2s_ports", "xmppserver", {5269}, "tcp");
-do_ports("legacy_ssl_ports", "xmppclient", {}, "ssl");
+net_activate_ports("c2s_ports", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
+net_activate_ports("s2s_ports", "xmppserver", {5269}, "tcp");
+net_activate_ports("legacy_ssl_ports", "xmppclient", {}, "ssl");
 
 if config.get("*", "core", "console_enabled") then
        if cl.get("console") then
@@ -138,6 +151,36 @@ if config.get("*", "core", "console_enabled") then
        end
 end
 
+-- setup error handling
+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 unlock_globals()
+       setmetatable(_G, nil);
+end
+
+function lock_globals()
+       setmetatable(_G, locked_globals_mt);
+end
+
+lock_globals();
+
 eventmanager.fire_event("server-started");
 
-server.loop();
+local quitting;
+while not quitting do
+       xpcall(server.loop, function (err)
+                                       if err:match("%d*: interrupted!$") then
+                                               quitting = true;
+                                               return;
+                                       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
+                                       
+                                       eventmanager.fire_event("very-bad-error", "*", err, traceback);
+                               end);
+end