diff options
| -rw-r--r-- | plugins/mod_posix.lua | 2 | ||||
| -rw-r--r-- | util-src/pposix.c | 18 |
2 files changed, 18 insertions, 2 deletions
diff --git a/plugins/mod_posix.lua b/plugins/mod_posix.lua index 697930ca..0f28041c 100644 --- a/plugins/mod_posix.lua +++ b/plugins/mod_posix.lua @@ -7,7 +7,7 @@ -- -local want_pposix_version = "0.3.1"; +local want_pposix_version = "0.3.2"; local pposix = assert(require "util.pposix"); 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 diff --git a/util-src/pposix.c b/util-src/pposix.c index ae3e1bb8..b47faaaf 100644 --- a/util-src/pposix.c +++ b/util-src/pposix.c @@ -13,9 +13,10 @@ * POSIX support functions for Lua */ -#define MODULE_VERSION "0.3.1" +#define MODULE_VERSION "0.3.2" #include <stdlib.h> +#include <math.h> #include <unistd.h> #include <libgen.h> #include <sys/resource.h> @@ -358,6 +359,18 @@ int lc_setgid(lua_State* L) return 2; } +int lc_umask(lua_State* L) +{ + char old_mode_string[7]; + mode_t old_mode = umask(strtoul(luaL_checkstring(L, 1), NULL, 8)); + + snprintf(old_mode_string, sizeof(old_mode_string), "%03o", old_mode); + old_mode_string[sizeof(old_mode_string)-1] = 0; + lua_pushstring(L, old_mode_string); + + return 1; +} + /* Like POSIX's setrlimit()/getrlimit() API functions. * * Syntax: @@ -506,6 +519,9 @@ int luaopen_util_pposix(lua_State *L) lua_pushcfunction(L, lc_setgid); lua_setfield(L, -2, "setgid"); + lua_pushcfunction(L, lc_umask); + lua_setfield(L, -2, "umask"); + lua_pushcfunction(L, lc_setrlimit); lua_setfield(L, -2, "setrlimit"); |