1143 lines
25 KiB
C
1143 lines
25 KiB
C
/*
|
|
* lposix.c
|
|
* POSIX library for Lua 5.1.
|
|
* Luiz Henrique de Figueiredo <lhf@tecgraf.puc-rio.br>
|
|
* 07 Apr 2006 23:17:49
|
|
* Clean up and bug fixes by Leo Razoumov <slonik.az@gmail.com> 2006-10-11 <!LR>
|
|
* Based on original by Claudio Terra for Lua 3.x.
|
|
* With contributions by Roberto Ierusalimschy.
|
|
*/
|
|
|
|
#include <sys/stat.h>
|
|
#include <sys/times.h>
|
|
#include <sys/types.h>
|
|
#include <sys/utsname.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <glob.h>
|
|
#include <grp.h>
|
|
#include <libgen.h>
|
|
#include <limits.h>
|
|
#include <poll.h>
|
|
#include <pwd.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <syslog.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include <utime.h>
|
|
|
|
#define MYNAME "posix"
|
|
#define MYVERSION MYNAME " library for " LUA_VERSION " / Jan 2008"
|
|
|
|
#ifndef ENABLE_SYSLOG
|
|
#define ENABLE_SYSLOG 1
|
|
#endif
|
|
|
|
#include "lua.h"
|
|
#include "lualib.h"
|
|
#include "lauxlib.h"
|
|
|
|
#include "modemuncher.h"
|
|
|
|
/* compatibility with Lua 5.0 */
|
|
#ifndef LUA_VERSION_NUM
|
|
static int luaL_checkoption (lua_State *L, int narg, const char *def,
|
|
const char *const lst[]) {
|
|
const char *name = (def) ? luaL_optstring(L, narg, def) :
|
|
luaL_checkstring(L, narg);
|
|
int i = luaL_findstring(name, lst);
|
|
if (i == -1)
|
|
luaL_argerror(L, narg, lua_pushfstring(L, "invalid option '%s'", name));
|
|
return i;
|
|
}
|
|
#define lua_pushinteger lua_pushnumber
|
|
#define lua_createtable(L,a,r) lua_newtable(L)
|
|
#define LUA_FILEHANDLE "FILE*"
|
|
|
|
#define lua_setfield(l,i,k)
|
|
#define lua_getfield(l,i,k)
|
|
|
|
#endif
|
|
|
|
static const struct { char c; mode_t b; } M[] =
|
|
{
|
|
{'r', S_IRUSR}, {'w', S_IWUSR}, {'x', S_IXUSR},
|
|
{'r', S_IRGRP}, {'w', S_IWGRP}, {'x', S_IXGRP},
|
|
{'r', S_IROTH}, {'w', S_IWOTH}, {'x', S_IXOTH},
|
|
};
|
|
|
|
|
|
static void pushmode(lua_State *L, mode_t mode)
|
|
{
|
|
char m[9];
|
|
int i;
|
|
for (i=0; i<9; i++) m[i]= (mode & M[i].b) ? M[i].c : '-';
|
|
if (mode & S_ISUID) m[2]= (mode & S_IXUSR) ? 's' : 'S';
|
|
if (mode & S_ISGID) m[5]= (mode & S_IXGRP) ? 's' : 'S';
|
|
lua_pushlstring(L, m, 9);
|
|
}
|
|
|
|
typedef void (*Selector)(lua_State *L, int i, const void *data);
|
|
|
|
static int doselection(lua_State *L, int i, int n,
|
|
const char *const S[],
|
|
Selector F,
|
|
const void *data)
|
|
{
|
|
if (lua_isnone(L, i) || lua_istable(L, i))
|
|
{
|
|
int j;
|
|
if (lua_isnone(L, i)) lua_createtable(L,0,n); else lua_settop(L, i);
|
|
for (j=0; S[j]!=NULL; j++)
|
|
{
|
|
lua_pushstring(L, S[j]);
|
|
F(L, j, data);
|
|
lua_settable(L, -3);
|
|
}
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
int k,n=lua_gettop(L);
|
|
for (k=i; k<=n; k++)
|
|
{
|
|
int j=luaL_checkoption(L, k, NULL, S);
|
|
F(L, j, data);
|
|
lua_replace(L, k);
|
|
}
|
|
return n-i+1;
|
|
}
|
|
}
|
|
#define doselection(L,i,S,F,d) (doselection)(L,i,sizeof(S)/sizeof(*S)-1,S,F,d)
|
|
|
|
static int pusherror(lua_State *L, const char *info)
|
|
{
|
|
lua_pushnil(L);
|
|
if (info==NULL)
|
|
lua_pushstring(L, strerror(errno));
|
|
else
|
|
lua_pushfstring(L, "%s: %s", info, strerror(errno));
|
|
lua_pushinteger(L, errno);
|
|
return 3;
|
|
}
|
|
|
|
static int pushresult(lua_State *L, int i, const char *info)
|
|
{
|
|
if (i==-1) return pusherror(L, info);
|
|
lua_pushinteger(L, i);
|
|
return 1;
|
|
}
|
|
|
|
static void badoption(lua_State *L, int i, const char *what, int option)
|
|
{
|
|
luaL_argerror(L, 2,
|
|
lua_pushfstring(L, "unknown %s option '%c'", what, option));
|
|
}
|
|
|
|
static uid_t mygetuid(lua_State *L, int i)
|
|
{
|
|
if (lua_isnone(L, i))
|
|
return -1;
|
|
else if (lua_isnumber(L, i))
|
|
return (uid_t) lua_tonumber(L, i);
|
|
else if (lua_isstring(L, i))
|
|
{
|
|
struct passwd *p=getpwnam(lua_tostring(L, i));
|
|
return (p==NULL) ? -1 : p->pw_uid;
|
|
}
|
|
else
|
|
return luaL_typerror(L, i, "string or number");
|
|
}
|
|
|
|
static gid_t mygetgid(lua_State *L, int i)
|
|
{
|
|
if (lua_isnone(L, i))
|
|
return -1;
|
|
else if (lua_isnumber(L, i))
|
|
return (gid_t) lua_tonumber(L, i);
|
|
else if (lua_isstring(L, i))
|
|
{
|
|
struct group *g=getgrnam(lua_tostring(L, i));
|
|
return (g==NULL) ? -1 : g->gr_gid;
|
|
}
|
|
else
|
|
return luaL_typerror(L, i, "string or number");
|
|
}
|
|
|
|
|
|
static int Perrno(lua_State *L) /** errno([n]) */
|
|
{
|
|
int n = luaL_optint(L, 1, errno);
|
|
lua_pushstring(L, strerror(n));
|
|
lua_pushinteger(L, n);
|
|
return 2;
|
|
}
|
|
|
|
|
|
static int Pbasename(lua_State *L) /** basename(path) */
|
|
{
|
|
char b[PATH_MAX];
|
|
size_t len;
|
|
const char *path = luaL_checklstring(L, 1, &len);
|
|
if (len>=sizeof(b)) luaL_argerror(L, 1, "too long");
|
|
lua_pushstring(L, basename(strcpy(b,path)));
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pdirname(lua_State *L) /** dirname(path) */
|
|
{
|
|
char b[PATH_MAX];
|
|
size_t len;
|
|
const char *path = luaL_checklstring(L, 1, &len);
|
|
if (len>=sizeof(b)) luaL_argerror(L, 1, "too long");
|
|
lua_pushstring(L, dirname(strcpy(b,path)));
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pdir(lua_State *L) /** dir([path]) */
|
|
{
|
|
const char *path = luaL_optstring(L, 1, ".");
|
|
DIR *d = opendir(path);
|
|
if (d == NULL)
|
|
return pusherror(L, path);
|
|
else
|
|
{
|
|
int i;
|
|
struct dirent *entry;
|
|
lua_newtable(L);
|
|
for (i=1; (entry = readdir(d)) != NULL; i++)
|
|
{
|
|
lua_pushstring(L, entry->d_name);
|
|
lua_rawseti(L, -2, i);
|
|
}
|
|
closedir(d);
|
|
lua_pushinteger(L, i-1);
|
|
return 2;
|
|
}
|
|
}
|
|
|
|
static int Pglob(lua_State *L) /** glob(pattern) */
|
|
{
|
|
const char *pattern = luaL_optstring(L, 1, "*");
|
|
glob_t globres;
|
|
|
|
if (glob(pattern, 0, NULL, &globres))
|
|
return pusherror(L, pattern);
|
|
else
|
|
{
|
|
int i;
|
|
lua_newtable(L);
|
|
for (i=1; i<=globres.gl_pathc; i++) {
|
|
lua_pushstring(L, globres.gl_pathv[i-1]);
|
|
lua_rawseti(L, -2, i);
|
|
}
|
|
globfree(&globres);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
static int aux_files(lua_State *L)
|
|
{
|
|
DIR **p = (DIR **)lua_touserdata(L, lua_upvalueindex(1));
|
|
DIR *d = *p;
|
|
struct dirent *entry;
|
|
if (d == NULL) return 0;
|
|
entry = readdir(d);
|
|
if (entry == NULL)
|
|
{
|
|
closedir(d);
|
|
*p=NULL;
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
lua_pushstring(L, entry->d_name);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
static int dir_gc (lua_State *L)
|
|
{
|
|
DIR *d = *(DIR **)lua_touserdata(L, 1);
|
|
if (d!=NULL) closedir(d);
|
|
return 0;
|
|
}
|
|
|
|
static int Pfiles(lua_State *L) /** files([path]) */
|
|
{
|
|
const char *path = luaL_optstring(L, 1, ".");
|
|
DIR **d = (DIR **)lua_newuserdata(L, sizeof(DIR *));
|
|
if (luaL_newmetatable(L, MYNAME " dir handle"))
|
|
{
|
|
lua_pushliteral(L, "__gc");
|
|
lua_pushcfunction(L, dir_gc);
|
|
lua_settable(L, -3);
|
|
}
|
|
lua_setmetatable(L, -2);
|
|
*d = opendir(path);
|
|
if (*d == NULL) return pusherror(L, path);
|
|
lua_pushcclosure(L, aux_files, 1);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pgetcwd(lua_State *L) /** getcwd() */
|
|
{
|
|
char b[PATH_MAX];
|
|
if (getcwd(b, sizeof(b)) == NULL) return pusherror(L, ".");
|
|
lua_pushstring(L, b);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pmkdir(lua_State *L) /** mkdir(path) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
return pushresult(L, mkdir(path, 0777), path);
|
|
}
|
|
|
|
|
|
static int Pchdir(lua_State *L) /** chdir(path) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
return pushresult(L, chdir(path), path);
|
|
}
|
|
|
|
static int Prmdir(lua_State *L) /** rmdir(path) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
return pushresult(L, rmdir(path), path);
|
|
}
|
|
|
|
|
|
static int Punlink(lua_State *L) /** unlink(path) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
return pushresult(L, unlink(path), path);
|
|
}
|
|
|
|
static int Plink(lua_State *L) /** link(old,new,[symbolic]) */
|
|
{
|
|
const char *oldpath = luaL_checkstring(L, 1);
|
|
const char *newpath = luaL_checkstring(L, 2);
|
|
return pushresult(L,
|
|
(lua_toboolean(L,3) ? symlink : link)(oldpath, newpath), NULL);
|
|
}
|
|
|
|
|
|
static int Preadlink(lua_State *L) /** readlink(path) */
|
|
{
|
|
char b[PATH_MAX];
|
|
const char *path = luaL_checkstring(L, 1);
|
|
int n = readlink(path, b, sizeof(b));
|
|
if (n==-1) return pusherror(L, path);
|
|
lua_pushlstring(L, b, n);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Paccess(lua_State *L) /** access(path,[mode]) */
|
|
{
|
|
int mode=F_OK;
|
|
const char *path=luaL_checkstring(L, 1);
|
|
const char *s;
|
|
for (s=luaL_optstring(L, 2, "f"); *s!=0 ; s++)
|
|
switch (*s)
|
|
{
|
|
case ' ': break;
|
|
case 'r': mode |= R_OK; break;
|
|
case 'w': mode |= W_OK; break;
|
|
case 'x': mode |= X_OK; break;
|
|
case 'f': mode |= F_OK; break;
|
|
default: badoption(L, 2, "mode", *s); break;
|
|
}
|
|
return pushresult(L, access(path, mode), path);
|
|
}
|
|
|
|
|
|
static int myfclose (lua_State *L) {
|
|
FILE **p = (FILE **)lua_touserdata(L, 1);
|
|
int rc = fclose(*p);
|
|
if (rc == 0) *p = NULL;
|
|
return pushresult(L, rc, NULL);
|
|
}
|
|
|
|
static int pushfile (lua_State *L, int id, const char *mode) {
|
|
FILE **f = (FILE **)lua_newuserdata(L, sizeof(FILE *));
|
|
*f = NULL;
|
|
luaL_getmetatable(L, LUA_FILEHANDLE);
|
|
lua_setmetatable(L, -2);
|
|
lua_getfield(L, LUA_REGISTRYINDEX, "POSIX_PIPEFILE");
|
|
if (lua_isnil(L, -1)) {
|
|
lua_pop(L, 1);
|
|
lua_newtable(L);
|
|
lua_pushvalue(L, -1);
|
|
lua_pushcfunction(L, myfclose);
|
|
lua_setfield(L, -2, "__close");
|
|
lua_setfield(L, LUA_REGISTRYINDEX, "POSIX_PIPEFILE");
|
|
}
|
|
lua_setfenv(L, -2);
|
|
*f = fdopen(id, mode);
|
|
return (*f != NULL);
|
|
}
|
|
|
|
static int Ppipe(lua_State *L) /** pipe() */
|
|
{
|
|
int fd[2];
|
|
if (pipe(fd)==-1) return pusherror(L, NULL);
|
|
if (!pushfile(L, fd[0], "r") || !pushfile(L, fd[1], "w"))
|
|
return pusherror(L, "pipe");
|
|
return 2;
|
|
}
|
|
|
|
|
|
static int Pfileno(lua_State *L) /** fileno(filehandle) */
|
|
{
|
|
FILE *f = *(FILE**) luaL_checkudata(L, 1, LUA_FILEHANDLE);
|
|
return pushresult(L, fileno(f), NULL);
|
|
}
|
|
|
|
|
|
static int Pfdopen(lua_State *L) /** fdopen(fd, mode) */
|
|
{
|
|
int fd = luaL_checkint(L, 1);
|
|
const char *mode = luaL_checkstring(L, 2);
|
|
if (!pushfile(L, fd, mode))
|
|
return pusherror(L, "fdpoen");
|
|
return 1;
|
|
}
|
|
|
|
|
|
/* helper func for Pdup */
|
|
static const char *filemode(int fd)
|
|
{
|
|
const char *m;
|
|
int mode = fcntl(fd, F_GETFL);
|
|
if (mode < 0)
|
|
return NULL;
|
|
switch (mode & O_ACCMODE) {
|
|
case O_RDONLY: m = "r"; break;
|
|
case O_WRONLY: m = "w"; break;
|
|
default: m = "rw"; break;
|
|
}
|
|
return m;
|
|
}
|
|
|
|
static int Pdup(lua_State *L) /** dup(old,[new]) */
|
|
{
|
|
FILE **oldf = (FILE**)luaL_checkudata(L, 1, LUA_FILEHANDLE);
|
|
FILE **newf = (FILE **)lua_touserdata(L, 2);
|
|
int fd;
|
|
const char *msg = "dup2";
|
|
fflush(*newf);
|
|
if (newf == NULL) {
|
|
fd = dup(fileno(*oldf));
|
|
msg = "dup";
|
|
} else {
|
|
fflush(*newf);
|
|
fd = dup2(fileno(*oldf), fileno(*newf));
|
|
}
|
|
|
|
if ((fd < 0) || !pushfile(L, fd, filemode(fd)))
|
|
return pusherror(L, msg);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pmkfifo(lua_State *L) /** mkfifo(path) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
return pushresult(L, mkfifo(path, 0777), path);
|
|
}
|
|
|
|
|
|
static int runexec(lua_State *L, int use_shell)
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
int i,n=lua_gettop(L);
|
|
char **argv = lua_newuserdata(L,(n+1)*sizeof(char*));
|
|
argv[0] = (char*)path;
|
|
for (i=1; i<n; i++) argv[i] = (char*)luaL_checkstring(L, i+1);
|
|
argv[n] = NULL;
|
|
if (use_shell) {
|
|
execvp(path, argv);
|
|
} else {
|
|
execv(path, argv);
|
|
}
|
|
return pusherror(L, path);
|
|
}
|
|
|
|
|
|
static int Pexec(lua_State *L) /** exec(path,[args]) */
|
|
{
|
|
return runexec(L, 0);
|
|
}
|
|
|
|
|
|
static int Pexecp(lua_State *L) /** execp(path,[args]) */
|
|
{
|
|
return runexec(L, 1);
|
|
}
|
|
|
|
|
|
static int Pfork(lua_State *L) /** fork() */
|
|
{
|
|
return pushresult(L, fork(), NULL);
|
|
}
|
|
|
|
/* from http://lua-users.org/lists/lua-l/2007-11/msg00346.html */
|
|
static int Ppoll(lua_State *L) /** poll(filehandle, timeout) */
|
|
{
|
|
struct pollfd fds;
|
|
FILE* file = *(FILE**)luaL_checkudata(L,1,LUA_FILEHANDLE);
|
|
int timeout = luaL_checkint(L,2);
|
|
fds.fd = fileno(file);
|
|
fds.events = POLLIN;
|
|
return pushresult(L, poll(&fds,1,timeout), NULL);
|
|
}
|
|
|
|
static int Pwait(lua_State *L) /** wait([pid]) */
|
|
{
|
|
int status;
|
|
pid_t pid = luaL_optint(L, 1, -1);
|
|
pid = waitpid(pid, &status, 0);
|
|
if (pid == -1) return pusherror(L, NULL);
|
|
lua_pushinteger(L, pid);
|
|
if (WIFEXITED(status))
|
|
{
|
|
lua_pushliteral(L,"exited");
|
|
lua_pushinteger(L, WEXITSTATUS(status));
|
|
return 3;
|
|
}
|
|
else if (WIFSIGNALED(status))
|
|
{
|
|
lua_pushliteral(L,"killed");
|
|
lua_pushinteger(L, WTERMSIG(status));
|
|
return 3;
|
|
}
|
|
else if (WIFSTOPPED(status))
|
|
{
|
|
lua_pushliteral(L,"stopped");
|
|
lua_pushinteger(L, WSTOPSIG(status));
|
|
return 3;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pkill(lua_State *L) /** kill(pid,[sig]) */
|
|
{
|
|
pid_t pid = luaL_checkint(L, 1);
|
|
int sig = luaL_optint(L, 2, SIGTERM);
|
|
return pushresult(L, kill(pid, sig), NULL);
|
|
}
|
|
|
|
static int Psetpid(lua_State *L) /** setpid(option,...) */
|
|
{
|
|
const char *what=luaL_checkstring(L, 1);
|
|
switch (*what)
|
|
{
|
|
case 'U':
|
|
return pushresult(L, seteuid(mygetuid(L, 2)), NULL);
|
|
case 'u':
|
|
return pushresult(L, setuid(mygetuid(L, 2)), NULL);
|
|
case 'G':
|
|
return pushresult(L, setegid(mygetgid(L, 2)), NULL);
|
|
case 'g':
|
|
return pushresult(L, setgid(mygetgid(L, 2)), NULL);
|
|
case 's':
|
|
return pushresult(L, setsid(), NULL);
|
|
case 'p':
|
|
{
|
|
pid_t pid = luaL_checkint(L, 2);
|
|
pid_t pgid = luaL_checkint(L, 3);
|
|
return pushresult(L, setpgid(pid,pgid), NULL);
|
|
}
|
|
default:
|
|
badoption(L, 2, "id", *what);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
static int Psleep(lua_State *L) /** sleep(seconds) */
|
|
{
|
|
unsigned int seconds = luaL_checkint(L, 1);
|
|
lua_pushinteger(L, sleep(seconds));
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Psetenv(lua_State *L) /** setenv(name,value,[over]) */
|
|
{
|
|
const char *name=luaL_checkstring(L, 1);
|
|
const char *value=luaL_optstring(L, 2, NULL);
|
|
if (value==NULL)
|
|
{
|
|
unsetenv(name);
|
|
return pushresult(L, 0, NULL);
|
|
}
|
|
else
|
|
{
|
|
int overwrite=lua_isnoneornil(L, 3) || lua_toboolean(L, 3);
|
|
return pushresult(L, setenv(name,value,overwrite), NULL);
|
|
}
|
|
}
|
|
|
|
|
|
static int Pgetenv(lua_State *L) /** getenv([name]) */
|
|
{
|
|
if (lua_isnone(L, 1))
|
|
{
|
|
extern char **environ;
|
|
char **e;
|
|
lua_newtable(L);
|
|
for (e=environ; *e!=NULL; e++)
|
|
{
|
|
char *s=*e;
|
|
char *eq=strchr(s, '=');
|
|
if (eq==NULL) /* will this ever happen? */
|
|
{
|
|
lua_pushstring(L,s);
|
|
lua_pushboolean(L,1);
|
|
}
|
|
else
|
|
{
|
|
lua_pushlstring(L,s,eq-s);
|
|
lua_pushstring(L,eq+1);
|
|
}
|
|
lua_settable(L,-3);
|
|
}
|
|
}
|
|
else
|
|
lua_pushstring(L, getenv(luaL_checkstring(L, 1)));
|
|
return 1;
|
|
}
|
|
|
|
static int Pumask(lua_State *L) /** umask([mode]) */
|
|
{/* <!LR> from old lposix-5.0 version */
|
|
char m[10];
|
|
mode_t mode;
|
|
umask(mode=umask(0));
|
|
mode=(~mode)&0777;
|
|
if (!lua_isnone(L, 1))
|
|
{
|
|
if (mode_munch(&mode, luaL_checkstring(L, 1)))
|
|
{
|
|
lua_pushnil(L);
|
|
return 1;
|
|
}
|
|
mode&=0777;
|
|
umask(~mode);
|
|
}
|
|
modechopper(mode, m);
|
|
lua_pushstring(L, m);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pchmod(lua_State *L) /** chmod(path,mode) */
|
|
{
|
|
mode_t mode;
|
|
struct stat s;
|
|
const char *path = luaL_checkstring(L, 1);
|
|
const char *modestr = luaL_checkstring(L, 2);
|
|
if (stat(path, &s)) return pusherror(L, path);
|
|
mode = s.st_mode;
|
|
if (mode_munch(&mode, modestr)) luaL_argerror(L, 2, "bad mode");
|
|
return pushresult(L, chmod(path, mode), path);
|
|
}
|
|
|
|
|
|
static int Pchown(lua_State *L) /** chown(path,uid,gid) */
|
|
{
|
|
const char *path = luaL_checkstring(L, 1);
|
|
uid_t uid = mygetuid(L, 2);
|
|
gid_t gid = mygetgid(L, 3);
|
|
return pushresult(L, chown(path, uid, gid), path);
|
|
}
|
|
|
|
|
|
static int Putime(lua_State *L) /** utime(path,[mtime,atime]) */
|
|
{
|
|
struct utimbuf times;
|
|
time_t currtime = time(NULL);
|
|
const char *path = luaL_checkstring(L, 1);
|
|
times.modtime = luaL_optnumber(L, 2, currtime);
|
|
times.actime = luaL_optnumber(L, 3, currtime);
|
|
return pushresult(L, utime(path, ×), path);
|
|
}
|
|
|
|
|
|
static void FgetID(lua_State *L, int i, const void *data)
|
|
{
|
|
switch (i)
|
|
{
|
|
case 0: lua_pushinteger(L, getegid()); break;
|
|
case 1: lua_pushinteger(L, geteuid()); break;
|
|
case 2: lua_pushinteger(L, getgid()); break;
|
|
case 3: lua_pushinteger(L, getuid()); break;
|
|
case 4: lua_pushinteger(L, getpgrp()); break;
|
|
case 5: lua_pushinteger(L, getpid()); break;
|
|
case 6: lua_pushinteger(L, getppid()); break;
|
|
}
|
|
}
|
|
|
|
static const char *const SgetID[] =
|
|
{
|
|
"egid", "euid", "gid", "uid", "pgrp", "pid", "ppid", NULL
|
|
};
|
|
|
|
static int Pgetpid(lua_State *L) /** getpid([options]) */
|
|
{
|
|
return doselection(L, 1, SgetID, FgetID, NULL);
|
|
}
|
|
|
|
|
|
static int Phostid(lua_State *L) /** hostid() */
|
|
{
|
|
char b[32];
|
|
sprintf(b,"%ld",gethostid());
|
|
lua_pushstring(L, b);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pttyname(lua_State *L) /** ttyname([fd]) */
|
|
{
|
|
int fd=luaL_optint(L, 1, 0);
|
|
lua_pushstring(L, ttyname(fd));
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pctermid(lua_State *L) /** ctermid() */
|
|
{
|
|
char b[L_ctermid];
|
|
lua_pushstring(L, ctermid(b));
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pgetlogin(lua_State *L) /** getlogin() */
|
|
{
|
|
lua_pushstring(L, getlogin());
|
|
return 1;
|
|
}
|
|
|
|
|
|
static void Fgetpasswd(lua_State *L, int i, const void *data)
|
|
{
|
|
const struct passwd *p=data;
|
|
switch (i)
|
|
{
|
|
case 0: lua_pushstring(L, p->pw_name); break;
|
|
case 1: lua_pushinteger(L, p->pw_uid); break;
|
|
case 2: lua_pushinteger(L, p->pw_gid); break;
|
|
case 3: lua_pushstring(L, p->pw_dir); break;
|
|
case 4: lua_pushstring(L, p->pw_shell); break;
|
|
/* not strictly POSIX */
|
|
case 5: lua_pushstring(L, p->pw_gecos); break;
|
|
case 6: lua_pushstring(L, p->pw_passwd); break;
|
|
}
|
|
}
|
|
|
|
static const char *const Sgetpasswd[] =
|
|
{
|
|
"name", "uid", "gid", "dir", "shell", "gecos", "passwd", NULL
|
|
};
|
|
|
|
|
|
static int Pgetpasswd(lua_State *L) /** getpasswd(name|id,[sel]) */
|
|
{
|
|
struct passwd *p=NULL;
|
|
if (lua_isnoneornil(L, 1))
|
|
p = getpwuid(geteuid());
|
|
else if (lua_isnumber(L, 1))
|
|
p = getpwuid((uid_t)lua_tonumber(L, 1));
|
|
else if (lua_isstring(L, 1))
|
|
p = getpwnam(lua_tostring(L, 1));
|
|
else
|
|
luaL_typerror(L, 1, "string or number");
|
|
if (p==NULL)
|
|
lua_pushnil(L);
|
|
else
|
|
return doselection(L, 2, Sgetpasswd, Fgetpasswd, p);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int Pgetgroup(lua_State *L) /** getgroup(name|id) */
|
|
{
|
|
struct group *g=NULL;
|
|
if (lua_isnumber(L, 1))
|
|
g = getgrgid((gid_t)lua_tonumber(L, 1));
|
|
else if (lua_isstring(L, 1))
|
|
g = getgrnam(lua_tostring(L, 1));
|
|
else
|
|
luaL_typerror(L, 1, "string or number");
|
|
if (g==NULL)
|
|
lua_pushnil(L);
|
|
else
|
|
{
|
|
int i;
|
|
lua_newtable(L);
|
|
lua_pushliteral(L, "name");
|
|
lua_pushstring(L, g->gr_name);
|
|
lua_settable(L, -3);
|
|
lua_pushliteral(L, "gid");
|
|
lua_pushinteger(L, g->gr_gid);
|
|
lua_settable(L, -3);
|
|
for (i=0; g->gr_mem[i]!=NULL; i++)
|
|
{
|
|
lua_pushstring(L, g->gr_mem[i]);
|
|
lua_rawseti(L, -2, i);
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
|
|
struct mytimes
|
|
{
|
|
struct tms t;
|
|
clock_t elapsed;
|
|
};
|
|
|
|
/* #define pushtime(L,x) lua_pushnumber(L,((lua_Number)x)/CLOCKS_PER_SEC) */
|
|
#define pushtime(L,x) lua_pushnumber(L, ((lua_Number)x)/clk_tck)
|
|
|
|
static void Ftimes(lua_State *L, int i, const void *data)
|
|
{
|
|
static long clk_tck = 0;
|
|
const struct mytimes *t=data;
|
|
|
|
if( !clk_tck){ clk_tck= sysconf(_SC_CLK_TCK);}
|
|
switch (i)
|
|
{
|
|
case 0: pushtime(L, t->t.tms_utime); break;
|
|
case 1: pushtime(L, t->t.tms_stime); break;
|
|
case 2: pushtime(L, t->t.tms_cutime); break;
|
|
case 3: pushtime(L, t->t.tms_cstime); break;
|
|
case 4: pushtime(L, t->elapsed); break;
|
|
}
|
|
}
|
|
|
|
static const char *const Stimes[] =
|
|
{
|
|
"utime", "stime", "cutime", "cstime", "elapsed", NULL
|
|
};
|
|
|
|
static int Ptimes(lua_State *L) /** times([options]) */
|
|
{
|
|
struct mytimes t;
|
|
t.elapsed = times(&t.t);
|
|
return doselection(L, 1, Stimes, Ftimes, &t);
|
|
}
|
|
|
|
|
|
static const char *filetype(mode_t m)
|
|
{
|
|
if (S_ISREG(m)) return "regular";
|
|
else if (S_ISLNK(m)) return "link";
|
|
else if (S_ISDIR(m)) return "directory";
|
|
else if (S_ISCHR(m)) return "character device";
|
|
else if (S_ISBLK(m)) return "block device";
|
|
else if (S_ISFIFO(m)) return "fifo";
|
|
else if (S_ISSOCK(m)) return "socket";
|
|
else return "?";
|
|
}
|
|
|
|
static void Fstat(lua_State *L, int i, const void *data)
|
|
{
|
|
const struct stat *s=data;
|
|
switch (i)
|
|
{
|
|
case 0: pushmode(L, s->st_mode); break;
|
|
case 1: lua_pushinteger(L, s->st_ino); break;
|
|
case 2: lua_pushinteger(L, s->st_dev); break;
|
|
case 3: lua_pushinteger(L, s->st_nlink); break;
|
|
case 4: lua_pushinteger(L, s->st_uid); break;
|
|
case 5: lua_pushinteger(L, s->st_gid); break;
|
|
case 6: lua_pushinteger(L, s->st_size); break;
|
|
case 7: lua_pushinteger(L, s->st_atime); break;
|
|
case 8: lua_pushinteger(L, s->st_mtime); break;
|
|
case 9: lua_pushinteger(L, s->st_ctime); break;
|
|
case 10:lua_pushstring(L, filetype(s->st_mode)); break;
|
|
}
|
|
}
|
|
|
|
static const char *const Sstat[] =
|
|
{
|
|
"mode", "ino", "dev", "nlink", "uid", "gid",
|
|
"size", "atime", "mtime", "ctime", "type",
|
|
NULL
|
|
};
|
|
|
|
static int Pstat(lua_State *L) /** stat(path,[options]) */
|
|
{
|
|
struct stat s;
|
|
const char *path=luaL_checkstring(L, 1);
|
|
if (lstat(path,&s)==-1) return pusherror(L, path);
|
|
return doselection(L, 2, Sstat, Fstat, &s);
|
|
}
|
|
|
|
|
|
static int Puname(lua_State *L) /** uname([string]) */
|
|
{
|
|
struct utsname u;
|
|
luaL_Buffer b;
|
|
const char *s;
|
|
if (uname(&u)==-1) return pusherror(L, NULL);
|
|
luaL_buffinit(L, &b);
|
|
for (s=luaL_optstring(L, 1, "%s %n %r %v %m"); *s; s++)
|
|
if (*s!='%')
|
|
luaL_putchar(&b, *s);
|
|
else switch (*++s)
|
|
{
|
|
case '%': luaL_putchar(&b, *s); break;
|
|
case 'm': luaL_addstring(&b,u.machine); break;
|
|
case 'n': luaL_addstring(&b,u.nodename); break;
|
|
case 'r': luaL_addstring(&b,u.release); break;
|
|
case 's': luaL_addstring(&b,u.sysname); break;
|
|
case 'v': luaL_addstring(&b,u.version); break;
|
|
default: badoption(L, 2, "format", *s); break;
|
|
}
|
|
luaL_pushresult(&b);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static const int Kpathconf[] =
|
|
{
|
|
_PC_LINK_MAX, _PC_MAX_CANON, _PC_MAX_INPUT, _PC_NAME_MAX, _PC_PATH_MAX,
|
|
_PC_PIPE_BUF, _PC_CHOWN_RESTRICTED, _PC_NO_TRUNC, _PC_VDISABLE,
|
|
-1
|
|
};
|
|
|
|
static void Fpathconf(lua_State *L, int i, const void *data)
|
|
{
|
|
const char *path=data;
|
|
lua_pushinteger(L, pathconf(path, Kpathconf[i]));
|
|
}
|
|
|
|
static const char *const Spathconf[] =
|
|
{
|
|
"link_max", "max_canon", "max_input", "name_max", "path_max",
|
|
"pipe_buf", "chown_restricted", "no_trunc", "vdisable",
|
|
NULL
|
|
};
|
|
|
|
static int Ppathconf(lua_State *L) /** pathconf([path,options]) */
|
|
{
|
|
const char *path = luaL_optstring(L, 1, ".");
|
|
return doselection(L, 2, Spathconf, Fpathconf, path);
|
|
}
|
|
|
|
|
|
static const int Ksysconf[] =
|
|
{
|
|
_SC_ARG_MAX, _SC_CHILD_MAX, _SC_CLK_TCK, _SC_NGROUPS_MAX, _SC_STREAM_MAX,
|
|
_SC_TZNAME_MAX, _SC_OPEN_MAX, _SC_JOB_CONTROL, _SC_SAVED_IDS, _SC_VERSION,
|
|
-1
|
|
};
|
|
|
|
static void Fsysconf(lua_State *L, int i, const void *data)
|
|
{
|
|
lua_pushinteger(L, sysconf(Ksysconf[i]));
|
|
}
|
|
|
|
static const char *const Ssysconf[] =
|
|
{
|
|
"arg_max", "child_max", "clk_tck", "ngroups_max", "stream_max",
|
|
"tzname_max", "open_max", "job_control", "saved_ids", "version",
|
|
NULL
|
|
};
|
|
|
|
static int Psysconf(lua_State *L) /** sysconf([options]) */
|
|
{
|
|
return doselection(L, 1, Ssysconf, Fsysconf, NULL);
|
|
}
|
|
|
|
#if ENABLE_SYSLOG
|
|
/* syslog funcs */
|
|
static int Popenlog(lua_State *L) /** openlog(ident, [option], [facility]) */
|
|
{
|
|
const char *ident = luaL_checkstring(L, 1);
|
|
int option = 0;
|
|
int facility = luaL_optint(L, 3, LOG_USER);
|
|
const char *s = luaL_optstring(L, 2, "");
|
|
while (*s) {
|
|
switch (*s) {
|
|
case ' ': break;
|
|
case 'c': option |= LOG_CONS; break;
|
|
case 'n': option |= LOG_NDELAY; break;
|
|
case 'e': option |= LOG_PERROR; break;
|
|
case 'p': option |= LOG_PID; break;
|
|
default: badoption(L, 2, "option", *s); break;
|
|
}
|
|
s++;
|
|
}
|
|
openlog(ident, option, facility);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int Psyslog(lua_State *L) /** syslog(priority, message) */
|
|
{
|
|
int priority = luaL_checkint(L, 1);
|
|
const char *msg = luaL_checkstring(L, 2);
|
|
syslog(priority, "%s", msg);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int Pcloselog(lua_State *L) /** closelog() */
|
|
{
|
|
closelog();
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
#if 0 /* dtrg: disabled */
|
|
/*
|
|
* XXX: GNU and BSD handle the forward declaration of crypt() in different
|
|
* and annoying ways (especially GNU). Declare it here just to make sure
|
|
* that it's there
|
|
*/
|
|
char *crypt(const char *, const char *);
|
|
|
|
static int Pcrypt(lua_State *L)
|
|
{
|
|
const char *str, *salt;
|
|
char *res;
|
|
|
|
str = luaL_checkstring(L, 1);
|
|
salt = luaL_checkstring(L, 2);
|
|
if (strlen(salt) < 2)
|
|
luaL_error(L, "not enough salt");
|
|
|
|
res = crypt(str, salt);
|
|
lua_pushstring(L, res);
|
|
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
static const luaL_reg R[] =
|
|
{
|
|
{"access", Paccess},
|
|
{"basename", Pbasename},
|
|
{"chdir", Pchdir},
|
|
{"chmod", Pchmod},
|
|
{"chown", Pchown},
|
|
#if 0
|
|
{"crypt", Pcrypt},
|
|
#endif
|
|
{"ctermid", Pctermid},
|
|
{"dirname", Pdirname},
|
|
{"dir", Pdir},
|
|
{"dup", Pdup},
|
|
{"errno", Perrno},
|
|
{"exec", Pexec},
|
|
{"execp", Pexecp},
|
|
{"fdopen", Pfdopen},
|
|
{"fileno", Pfileno},
|
|
{"files", Pfiles},
|
|
{"fork", Pfork},
|
|
{"getcwd", Pgetcwd},
|
|
{"getenv", Pgetenv},
|
|
{"getgroup", Pgetgroup},
|
|
{"getlogin", Pgetlogin},
|
|
{"getpasswd", Pgetpasswd},
|
|
{"getpid", Pgetpid},
|
|
{"glob", Pglob},
|
|
{"hostid", Phostid},
|
|
{"kill", Pkill},
|
|
{"link", Plink},
|
|
{"mkdir", Pmkdir},
|
|
{"mkfifo", Pmkfifo},
|
|
{"pathconf", Ppathconf},
|
|
{"pipe", Ppipe},
|
|
{"readlink", Preadlink},
|
|
{"rmdir", Prmdir},
|
|
{"rpoll", Ppoll},
|
|
{"setenv", Psetenv},
|
|
{"setpid", Psetpid},
|
|
{"sleep", Psleep},
|
|
{"stat", Pstat},
|
|
{"sysconf", Psysconf},
|
|
{"times", Ptimes},
|
|
{"ttyname", Pttyname},
|
|
{"unlink", Punlink},
|
|
{"umask", Pumask},
|
|
{"uname", Puname},
|
|
{"utime", Putime},
|
|
{"wait", Pwait},
|
|
|
|
#if ENABLE_SYSLOG
|
|
{"openlog", Popenlog},
|
|
{"syslog", Psyslog},
|
|
{"closelog", Pcloselog},
|
|
#endif
|
|
|
|
{NULL, NULL}
|
|
};
|
|
|
|
#define set_const(key, value) \
|
|
lua_pushliteral(L, key); \
|
|
lua_pushnumber(L, value); \
|
|
lua_settable(L, -3)
|
|
|
|
LUALIB_API int luaopen_posix (lua_State *L)
|
|
{
|
|
luaL_register(L,MYNAME,R);
|
|
lua_pushliteral(L,"version"); /** version */
|
|
lua_pushliteral(L,MYVERSION);
|
|
lua_settable(L,-3);
|
|
|
|
#if ENABLE_SYSLOG
|
|
set_const("LOG_AUTH", LOG_AUTH);
|
|
set_const("LOG_AUTHPRIV", LOG_AUTHPRIV);
|
|
set_const("LOG_CRON", LOG_CRON);
|
|
set_const("LOG_DAEMON", LOG_DAEMON);
|
|
set_const("LOG_FTP", LOG_FTP);
|
|
set_const("LOG_KERN", LOG_KERN);
|
|
set_const("LOG_LOCAL0", LOG_LOCAL0);
|
|
set_const("LOG_LOCAL1", LOG_LOCAL1);
|
|
set_const("LOG_LOCAL2", LOG_LOCAL2);
|
|
set_const("LOG_LOCAL3", LOG_LOCAL3);
|
|
set_const("LOG_LOCAL4", LOG_LOCAL4);
|
|
set_const("LOG_LOCAL5", LOG_LOCAL5);
|
|
set_const("LOG_LOCAL6", LOG_LOCAL6);
|
|
set_const("LOG_LOCAL7", LOG_LOCAL7);
|
|
set_const("LOG_LPR", LOG_LPR);
|
|
set_const("LOG_MAIL", LOG_MAIL);
|
|
set_const("LOG_NEWS", LOG_NEWS);
|
|
set_const("LOG_SYSLOG", LOG_SYSLOG);
|
|
set_const("LOG_USER", LOG_USER);
|
|
set_const("LOG_UUCP", LOG_UUCP);
|
|
|
|
set_const("LOG_EMERG", LOG_EMERG);
|
|
set_const("LOG_ALERT", LOG_ALERT);
|
|
set_const("LOG_CRIT", LOG_CRIT);
|
|
set_const("LOG_ERR", LOG_ERR);
|
|
set_const("LOG_WARNING", LOG_WARNING);
|
|
set_const("LOG_NOTICE", LOG_NOTICE);
|
|
set_const("LOG_INFO", LOG_INFO);
|
|
set_const("LOG_DEBUG", LOG_DEBUG);
|
|
#endif
|
|
|
|
|
|
return 1;
|
|
}
|
|
|
|
/*EOF*/
|