plugins/mod_posix.lua

   1
   2 local want_pposix_version = "0.3.0";
   3
   4 local pposix = assert(require "util.pposix");
   5 if pposix._VERSION ~= want_pposix_version then module:log("warn", "Unknown version (%s) of binary pposix module, expected %s", tostring(pposix._VERSION), want_pposix_version); end
   6
   7 local config_get = require "core.configmanager".get;
   8 local logger_set = require "util.logger".setwriter;
   9
  10 module.host = "*"; -- we're a global module
  11
  12 if not config_get("*", "core", "no_daemonize") then
  13         local function daemonize_server()
  14                 local logwriter;
  15
  16                 local logfilename = config_get("*", "core", "log");
  17                 if logfilename == "syslog" then
  18                         pposix.syslog_open("prosody");
  19                         pposix.syslog_setminlevel(config.get("*", "core", "minimum_log_level") or "info");
  20                                 local syslog, format = pposix.syslog_log, string.format;
  21                                 logwriter = function (name, level, message, ...)
  22                                                         if ... then
  23                                                                 syslog(level, format(message, ...));
  24                                                         else
  25                                                                 syslog(level, message);
  26                                                         end
  27                                                 end;
  28                 elseif logfilename then
  29                         local logfile = io.open(logfilename, "a+");
  30                         if logfile then
  31                                 local write, format, flush = logfile.write, string.format, logfile.flush;
  32                                 logwriter = function (name, level, message, ...)
  33                                                         if ... then
  34                                                                 write(logfile, name, "\t", level, "\t", format(message, ...), "\n");
  35                                                         else
  36                                                                 write(logfile, name, "\t" , level, "\t", message, "\n");
  37                                                         end
  38                                                         flush(logfile);
  39                                                 end;
  40                         end
  41                 else
  42                         log("debug", "No logging specified, will continue with default");
  43                 end
  44
  45                 local ok, ret = pposix.daemonize();
  46                 if not ok then
  47                         log("error", "Failed to daemonize: %s", ret);
  48                 elseif ret and ret > 0 then
  49                         os.exit(0);
  50                 else
  51                         if logwriter then
  52                                 local ok, ret = logger_set(logwriter);
  53                                 if not ok then
  54                                         log("error", "Couldn't set new log output: %s", ret);
  55                                 end
  56                         end
  57                         log("info", "Successfully daemonized to PID %d", pposix.getpid());
  58
  59                         local pidfile = config.get("*", "core", "pidfile");
  60                         if pidfile then
  61                                 local pf, err = io.open(pidfile, "w+");
  62                                 if not pf then
  63                                         log("error", "Couldn't write pidfile; %s", err);
  64                                 else
  65                                         pf:write(tostring(pposix.getpid()));
  66                                         pf:close();
  67                                 end
  68                         end
  69                 end
  70         end
  71         module:add_event_hook("server-starting", daemonize_server);
  72 end