Add lock_globals() and unlock_globals() functions (for when you really need to use...
[prosody.git] / prosody
diff --git a/prosody b/prosody
index b4a18c678e095da49a6e2583ddb5a8d282ae2f71..42e4eb502d909fe33133eef6b2f6686daf1c79be 100755 (executable)
--- a/prosody
+++ b/prosody
@@ -1,31 +1,20 @@
 #!/usr/bin/env lua
--- Prosody IM v0.2
--- Copyright (C) 2008 Matthew Wild
--- Copyright (C) 2008 Waqas Hussain
+-- Prosody IM v0.4
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
 -- 
--- This program is free software; you can redistribute it and/or
--- modify it under the terms of the GNU General Public License
--- as published by the Free Software Foundation; either version 2
--- of the License, or (at your option) any later version.
--- 
--- This program is distributed in the hope that it will be useful,
--- but WITHOUT ANY WARRANTY; without even the implied warranty of
--- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--- GNU General Public License for more details.
--- 
--- You should have received a copy of the GNU General Public License
--- along with this program; if not, write to the Free Software
--- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+-- This project is MIT/X11 licensed. Please see the
+-- 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
@@ -43,36 +32,43 @@ 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
-               log("error", "Couldn't load config file: %s", err);
-               log("info", "Falling back to old config file format...")
-               ok, err = pcall(dofile, "lxmppd.cfg");
-               if not ok then
-                       log("error", "Old config format loading failed too: %s", err);
-               else
-                       for _, host in ipairs(_G.config.hosts) do
-                               config.set(host, "core", "defined", true);
-                       end
-                       
-                       config.set("*", "core", "modules_enabled", _G.config.modules);
-                       config.set("*", "core", "ssl", _G.config.ssl_ctx);
+               print("");
+               print("**************************");
+               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
 
-local server = require "net.server"
+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"
+
+
 -- Maps connections to sessions --
 sessions = {};
 hosts = {};
@@ -82,7 +78,6 @@ hosts = {};
 require "util.import"
 require "core.xmlhandlers"
 require "core.rostermanager"
-require "core.offlinemessage"
 require "core.eventmanager"
 require "core.hostmanager"
 require "core.modulemanager"
@@ -90,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
 ]]
@@ -107,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"; };
@@ -123,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");
@@ -134,24 +133,54 @@ local function do_ports(option, listener, default, conntype)
                        if type(port) ~= "number" then
                                log("error", "Non-numeric "..option..": "..tostring(port));
                        else
-                               cl.start(listener, { ssl = global_ssl_ctx, port = port, type = conntype });
+                               cl.start(listener, { ssl = conntype ~= "tcp" and global_ssl_ctx, port = port, type = conntype });
                        end
                end
        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
-               cl.start("console")
+               cl.start("console", { interface = config.get("*", "core", "console_interface") or "127.0.0.1" })
        else
                log("error", "Console is enabled, but the console module appears not to be loaded");
        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