util-src/pposix.c
changeset 2438 819ba949c7bc
parent 2437 b1ba2473fd91
child 2441 d72078946a16
--- a/util-src/pposix.c	Sun Jan 10 20:21:48 2010 +0000
+++ b/util-src/pposix.c	Sun Jan 10 21:48:25 2010 +0000
@@ -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 @@
 	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 @@
 	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");