diff options
| -rw-r--r-- | libs/nixio/src/nixio.c | 7 | ||||
| -rw-r--r-- | libs/nixio/src/process.c | 57 |
2 files changed, 63 insertions, 1 deletions
diff --git a/libs/nixio/src/nixio.c b/libs/nixio/src/nixio.c index e44f5fd470..d433ba4fd2 100644 --- a/libs/nixio/src/nixio.c +++ b/libs/nixio/src/nixio.c @@ -130,9 +130,10 @@ LUALIB_API int luaopen_nixio(lua_State *L) { lua_setfield(L, -2, "version"); /* some constants */ - lua_createtable(L, 0, 11); + lua_createtable(L, 0, 16); NIXIO_PUSH_CONSTANT(EACCES); + NIXIO_PUSH_CONSTANT(EINTR); NIXIO_PUSH_CONSTANT(ENOSYS); NIXIO_PUSH_CONSTANT(EINVAL); NIXIO_PUSH_CONSTANT(EWOULDBLOCK); @@ -143,6 +144,10 @@ LUALIB_API int luaopen_nixio(lua_State *L) { NIXIO_PUSH_CONSTANT(SIGINT); NIXIO_PUSH_CONSTANT(SIGTERM); NIXIO_PUSH_CONSTANT(SIGKILL); + NIXIO_PUSH_CONSTANT(SIGHUP); + NIXIO_PUSH_CONSTANT(SIGSTOP); + NIXIO_PUSH_CONSTANT(SIGCONT); + NIXIO_PUSH_CONSTANT(SIGSEGV); lua_setfield(L, -2, "const"); diff --git a/libs/nixio/src/process.c b/libs/nixio/src/process.c index 1f2ae153fb..19fa10ef2c 100644 --- a/libs/nixio/src/process.c +++ b/libs/nixio/src/process.c @@ -17,10 +17,13 @@ */ #include "nixio.h" +#include <pwd.h> +#include <grp.h> #include <unistd.h> #include <errno.h> #include <string.h> #include <sys/wait.h> +#include <sys/types.h> static int nixio_fork(lua_State *L) { pid_t pid = fork(); @@ -81,12 +84,66 @@ static int nixio_kill(lua_State *L) { return nixio__pstatus(L, !kill(luaL_checkint(L, 1), luaL_checkint(L, 2))); } +static int nixio_getpid(lua_State *L) { + lua_pushinteger(L, getpid()); + return 1; +} + +static int nixio_getppid(lua_State *L) { + lua_pushinteger(L, getppid()); + return 1; +} + +static int nixio_getuid(lua_State *L) { + lua_pushinteger(L, getuid()); + return 1; +} + +static int nixio_getgid(lua_State *L) { + lua_pushinteger(L, getgid()); + return 1; +} + +static int nixio_setgid(lua_State *L) { + gid_t gid; + if (lua_isstring(L, 1)) { + struct group *g = getgrnam(lua_tostring(L, 1)); + gid = (!g) ? -1 : g->gr_gid; + } else if (lua_isnumber(L, 1)) { + gid = lua_tointeger(L, 1); + } else { + return luaL_argerror(L, 1, "supported values: <groupname>, <gid>"); + } + + return nixio__pstatus(L, !setgid(gid)); +} + +static int nixio_setuid(lua_State *L) { + uid_t uid; + if (lua_isstring(L, 1)) { + struct passwd *p = getpwnam(lua_tostring(L, 1)); + uid = (!p) ? -1 : p->pw_uid; + } else if (lua_isnumber(L, 1)) { + uid = lua_tointeger(L, 1); + } else { + return luaL_argerror(L, 1, "supported values: <username>, <uid>"); + } + + return nixio__pstatus(L, !setuid(uid)); +} + /* module table */ static const luaL_reg R[] = { {"fork", nixio_fork}, {"wait", nixio_wait}, {"kill", nixio_kill}, + {"getpid", nixio_getpid}, + {"getppid", nixio_getppid}, + {"getuid", nixio_getuid}, + {"getgid", nixio_getgid}, + {"setuid", nixio_setuid}, + {"setgid", nixio_setgid}, {NULL, NULL} }; |