util.pposix: Add pposix.umask(), bump version to 0.3.2 (and do the same in mod_posix)

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");