nixio: Export more constants, introduce {g,s}et{g,u}id, getpid, getppid

This commit is contained in:
Steven Barth 2009-03-05 19:20:40 +00:00
parent d6e251adbe
commit f151bb804f
2 changed files with 63 additions and 1 deletions

View file

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

View file

@ -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}
};