core.modulemanager: Some refactoring to make upcoming changes a little easier

[prosody.git] / util-src / pposix.c

diff --git a/util-src/pposix.c b/util-src/pposix.c
index c98414cca2d6cdc5e78ed93b756f68b1088fd925..9d10ec7ef0d16762290599209bb0862b7427e75f 100644 (file)
--- a/util-src/pposix.c
+++ b/util-src/pposix.c
@@ -21,6 +21,8 @@
    POSIX support functions for Lua
 */
 
+#define MODULE_VERSION "0.3.0"
+
 #include <stdlib.h>
 #include <unistd.h>
 #include <libgen.h>
@@ -28,9 +30,16 @@
 #include <sys/stat.h>
 #include <fcntl.h>
 
+#include <syslog.h>
+
+#include <string.h>
+
 #include "lua.h"
+#include "lauxlib.h"
 
-static int daemonize(lua_State *L)
+/* Daemonization support */
+
+static int lc_daemonize(lua_State *L)
 {
 
        pid_t pid;
@@ -53,13 +62,11 @@ static int daemonize(lua_State *L)
        else if(pid != 0)
        {
                /* We are the parent process */
-               printf("We are the parent, pid of child is %d\n", (int)pid);
                lua_pushboolean(L, 1);
                lua_pushnumber(L, pid);
                return 2;
        }
        
-       printf("We are the child, pid reports %d\n", (int)pid);
        /* and we are the child process */
        if(setsid() == -1)
        {
@@ -85,10 +92,167 @@ static int daemonize(lua_State *L)
        return 2;
 }
 
+/* Syslog support */
+
+char *facility_strings[] =     {       "auth",
+                                       "authpriv",
+                                       "cron",
+                                       "daemon",
+                                       "ftp",
+                                       "kern",
+                                       "local0",
+                                       "local1",
+                                       "local2",
+                                       "local3",
+                                       "local4",
+                                       "local5",
+                                       "local6",
+                                       "local7",
+                                       "lpr",
+                                       "mail",
+                                       "syslog",
+                                       "user",
+                                       "uucp",
+                                       NULL
+                               };
+int facility_constants[] =     {
+                                       LOG_AUTH,
+                                       LOG_AUTHPRIV,
+                                       LOG_CRON,
+                                       LOG_DAEMON,
+                                       LOG_FTP,
+                                       LOG_KERN,
+                                       LOG_LOCAL0,
+                                       LOG_LOCAL1,
+                                       LOG_LOCAL2,
+                                       LOG_LOCAL3,
+                                       LOG_LOCAL4,
+                                       LOG_LOCAL5,
+                                       LOG_LOCAL6,
+                                       LOG_LOCAL7,
+                                       LOG_LPR,
+                                       LOG_MAIL,
+                                       LOG_NEWS,
+                                       LOG_SYSLOG,
+                                       LOG_USER,
+                                       LOG_UUCP,
+                                       -1
+                               };
+
+/* "
+       The parameter ident in the call of openlog() is probably stored  as-is.
+       Thus,  if  the  string  it  points  to  is  changed, syslog() may start
+       prepending the changed string, and if the string it points to ceases to
+       exist,  the  results  are  undefined.  Most portable is to use a string
+       constant.
+   " -- syslog manpage
+*/ 
+char* syslog_ident = NULL;
+
+int lc_syslog_open(lua_State* L)
+{
+       int facility = luaL_checkoption(L, 2, "daemon", &facility_strings);
+       facility = facility_constants[facility];
+
+       luaL_checkstring(L, 1);
+       
+       if(syslog_ident)
+               free(syslog_ident);
+       
+       syslog_ident = strdup(lua_tostring(L, 1));
+       
+       openlog(syslog_ident, LOG_PID, facility);
+       return 0;
+}
+
+char *level_strings[] =        {
+                               "debug",
+                               "info",
+                               "notice",
+                               "warn",
+                               "error",
+                               NULL
+                       };
+int level_constants[] =        {
+                               LOG_DEBUG,
+                               LOG_INFO,
+                               LOG_NOTICE,
+                               LOG_WARNING,
+                               LOG_EMERG,
+                               -1
+                       };
+int lc_syslog_log(lua_State* L)
+{
+       int level = luaL_checkoption(L, 1, "notice", &level_strings);
+       level = level_constants[level];
+
+       luaL_checkstring(L, 2);
+
+       syslog(level, "%s", lua_tostring(L, 2));
+       return 0;
+}
+
+int lc_syslog_close(lua_State* L)
+{
+       closelog();
+       if(syslog_ident)
+       {
+               free(syslog_ident);
+               syslog_ident = NULL;
+       }
+       return 0;
+}
+
+int lc_syslog_setmask(lua_State* L)
+{
+       int level_idx = luaL_checkoption(L, 1, "notice", &level_strings);
+       int mask = 0;
+       do
+       {
+               mask |= LOG_MASK(level_constants[level_idx]);
+       } while (++level_idx<=4);
+
+       setlogmask(mask);
+       return 0;
+}
+
+/* getpid */
+
+int lc_getpid(lua_State* L)
+{
+       lua_pushinteger(L, getpid());
+       return 1;
+}
+
+/* Register functions */
+
 int luaopen_util_pposix(lua_State *L)
 {
        lua_newtable(L);
-       lua_pushcfunction(L, daemonize);
+
+       lua_pushcfunction(L, lc_daemonize);
        lua_setfield(L, -2, "daemonize");
+
+       lua_pushcfunction(L, lc_syslog_open);
+       lua_setfield(L, -2, "syslog_open");
+
+       lua_pushcfunction(L, lc_syslog_close);
+       lua_setfield(L, -2, "syslog_close");
+
+       lua_pushcfunction(L, lc_syslog_log);
+       lua_setfield(L, -2, "syslog_log");
+
+       lua_pushcfunction(L, lc_syslog_setmask);
+       lua_setfield(L, -2, "syslog_setminlevel");
+
+       lua_pushcfunction(L, lc_getpid);
+       lua_setfield(L, -2, "getpid");
+
+       lua_pushliteral(L, "pposix");
+       lua_setfield(L, -2, "_NAME");
+
+       lua_pushliteral(L, MODULE_VERSION);
+       lua_setfield(L, -2, "_VERSION");
+       
        return 1;
 };