brejela
/
lua-ex
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Attempt to merge implementations

Browse Source
master
mark 21 years ago
parent 05bc6c5bf9
commit 2ace11ed4c
4 changed files with 207 additions and 143 deletions
Show Stats Download Patch File Download Diff File

3
posix/Makefile
Unescape Escape View File

@ -2,7 +2,8 @@ CFLAGS = -std=c99 $(WARNINGS) $(DEFINES) $(INCLUDES)
DEFINES = -D_POSIX_SOURCE -D_XOPEN_SOURCE=600 -D_POSIX_C_SOURCE=200112L -U__STRICT_ANSI__ DEFINES = -D_POSIX_SOURCE -D_XOPEN_SOURCE=600 -D_POSIX_C_SOURCE=200112L -U__STRICT_ANSI__
INCLUDES = -I$(LUA)/include -I. INCLUDES = -I$(LUA)/include -I.
WARNINGS = -W -Wall WARNINGS = -W -Wall
LUA = /home/mark/src/lang/lua/lua-5.1-rc2 #LUA = /home/mark/src/lang/lua/lua-5.1-rc2
LUA = /home/mark/src/lang/lua/lua51
ex.so: ex.o; $(CC) -shared -o $@ ex.o ex.so: ex.o; $(CC) -shared -o $@ ex.o

177
posix/ex.c
Unescape Escape View File

@ -3,7 +3,7 @@
#include <string.h> #include <string.h>
#include <ctype.h> #include <ctype.h>
#include <errno.h> #include <errno.h>
#include <assert.h> #include <limits.h>
#include <lua.h> #include <lua.h>
#include <lualib.h> #include <lualib.h>
@ -14,8 +14,6 @@
#include <spawn.h> #include <spawn.h>
#include <sys/wait.h> #include <sys/wait.h>
#define debug(...) fprintf(stderr,__VA_ARGS__)
/* Generally useful function -- what luaL_checkudata() should do */ /* Generally useful function -- what luaL_checkudata() should do */
static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname) static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname)
@ -27,7 +25,7 @@ static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname)
/* -- nil error */ /* -- nil error */
static int posix_error(lua_State *L) static int push_error(lua_State *L)
{ {
lua_pushnil(L); lua_pushnil(L);
lua_pushstring(L, strerror(errno)); lua_pushstring(L, strerror(errno));
@ -53,7 +51,7 @@ static int ex_setenv(lua_State *L)
const char *nam = luaL_checkstring(L, 1); const char *nam = luaL_checkstring(L, 1);
const char *val = luaL_checkstring(L, 2); const char *val = luaL_checkstring(L, 2);
if (-1 == setenv(nam, val, 1)) if (-1 == setenv(nam, val, 1))
return posix_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -63,7 +61,7 @@ static int ex_unsetenv(lua_State *L)
{ {
const char *nam = luaL_checkstring(L, 1); const char *nam = luaL_checkstring(L, 1);
if (-1 == unsetenv(nam)) if (-1 == unsetenv(nam))
return posix_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -74,7 +72,7 @@ static int ex_environ(lua_State *L)
const char *nam, *val, *end; const char *nam, *val, *end;
const char **env; const char **env;
lua_newtable(L); lua_newtable(L);
for (env = environ; (nam = *env); env++) { for (env = (const char **)environ; (nam = *env); env++) {
end = strchr(val = strchr(nam, '=') + 1, '\0'); end = strchr(val = strchr(nam, '=') + 1, '\0');
lua_pushlstring(L, nam, val - nam - 1); lua_pushlstring(L, nam, val - nam - 1);
lua_pushlstring(L, val, end - val); lua_pushlstring(L, val, end - val);
@ -98,7 +96,7 @@ static int ex_chdir(lua_State *L)
{ {
const char *pathname = luaL_checkstring(L, 1); const char *pathname = luaL_checkstring(L, 1);
if (-1 == chdir(pathname)) if (-1 == chdir(pathname))
return posix_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -108,7 +106,7 @@ static int ex_mkdir(lua_State *L)
{ {
const char *pathname = luaL_checkstring(L, 1); const char *pathname = luaL_checkstring(L, 1);
if (-1 == mkdir(pathname, 0777)) if (-1 == mkdir(pathname, 0777))
return posix_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -118,7 +116,7 @@ static int ex_currentdir(lua_State *L)
{ {
char pathname[PATH_MAX + 1]; char pathname[PATH_MAX + 1];
if (!getcwd(pathname, sizeof pathname)) if (!getcwd(pathname, sizeof pathname))
return posix_error(L); return push_error(L);
lua_pushstring(L, pathname); lua_pushstring(L, pathname);
return 1; return 1;
} }
@ -143,7 +141,7 @@ static int file_lock(lua_State *L, FILE *f, const char *mode, long offset, long
k.l_start = offset; k.l_start = offset;
k.l_len = length; k.l_len = length;
if (-1 == fcntl(fileno(f), F_SETLK, &k)) if (-1 == fcntl(fileno(f), F_SETLK, &k))
return posix_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -186,51 +184,87 @@ static int ex_pipe(lua_State *L)
{ {
int fd[2]; int fd[2];
if (-1 == pipe(fd)) if (-1 == pipe(fd))
return posix_error(L); return push_error(L);
return make_pipe(L, fdopen(fd[0], "r"), fdopen(fd[1], "w")); return make_pipe(L, fdopen(fd[0], "r"), fdopen(fd[1], "w"));
} }
/* ... argtab -- ... */ /* ... argtab -- ... */
static const char **build_args(lua_State *L) static const char **build_vector(lua_State *L)
{ {
size_t i, n = lua_objlen(L, -1); size_t i, n = lua_objlen(L, -1);
const char **argv = lua_newuserdata(L, (n + 2) * sizeof *argv); const char **vec = lua_newuserdata(L, (n + 2) * sizeof *vec);
/* ... argtab argv */ /* ... arr vec */
for (i = 0; i <= n; i++) { for (i = 0; i <= n; i++) {
lua_rawgeti(L, -2, i); /* ... argtab argv argi */ lua_rawgeti(L, -2, i); /* ... arr vec elem */
argv[i] = lua_tostring(L, -1); vec[i] = lua_tostring(L, -1);
lua_pop(L, 1); /* ... argtab argv */ lua_pop(L, 1); /* ... arr vec */
} }
argv[n + 1] = 0; vec[n + 1] = 0;
lua_replace(L, -2); /* ... argv */ lua_replace(L, -2); /* ... vector */
return argv; return vec;
}
struct spawn_params {
lua_State *L;
const char *command, **argv, **envp;
posix_spawn_file_actions_t file_actions;
int has_actions;
};
/* filename ... */
static void spawn_param_filename(struct spawn_params *p)
{
/* XXX confusing */
p->command = luaL_checkstring(p->L, 1);
}
static void spawn_param_defaults(struct spawn_params *p)
{
p->argv = lua_newuserdata(p->L, 2 * sizeof *p->argv);
p->argv[0] = p->command;
p->argv[1] = 0;
p->envp = (const char **)environ;
p->has_actions = 0;
}
/* ... argtab */
static void spawn_param_args(struct spawn_params *p)
{
const char **argv = build_vector(p->L);
if (!argv[0]) argv[0] = p->command;
p->argv = argv;
} }
/* ... envtab -- */ /* ... envtab/nil */
static const char **build_env(lua_State *L) static void spawn_param_env(struct spawn_params *p)
{ {
size_t i = 0; size_t i = 0;
luaL_Buffer estr; luaL_Buffer estr;
luaL_buffinit(L, &estr); if (lua_isnil(p->L, -1)) {
lua_newtable(L); /* ... envtab a */ p->envp = (const char **)environ;
lua_pushnil(L); /* ... envtab a nil */ return;
for (i = 0; lua_next(L, -3); i++) {/* ... envtab a k v */ }
luaL_buffinit(p->L, &estr);
lua_newtable(p->L); /* ... envtab arr */
lua_pushnil(p->L); /* ... envtab arr nil */
for (i = 0; lua_next(p->L, -3); i++) { /* ... envtab arr k v */
luaL_prepbuffer(&estr); luaL_prepbuffer(&estr);
lua_pushvalue(L, -2); /* ... envtab a k v k */ lua_pushvalue(p->L, -2); /* ... envtab arr k v k */
luaL_addvalue(&estr); luaL_addvalue(&estr);
luaL_putchar(&estr, '='); luaL_putchar(&estr, '=');
lua_pop(L, 1); /* ... envtab a k v */ lua_pop(p->L, 1); /* ... envtab arr k v */
luaL_addvalue(&estr); luaL_addvalue(&estr);
lua_pop(L, 1); /* ... envtab a k */ lua_pop(p->L, 1); /* ... envtab arr k */
luaL_pushresult(&estr); /* ... envtab a k estr */ luaL_pushresult(&estr); /* ... envtab arr k estr */
lua_rawseti(L, -3, i); /* ... envtab a[n]=estr k */ lua_rawseti(p->L, -3, i); /* ... envtab arr[n]=estr k */
} /* ... envtab a */ } /* ... envtab arr */
lua_replace(L, -2); /* ... a */ lua_replace(p->L, -2); /* ... arr */
return build_args(L); /* ... */ build_vector(p->L); /* ... arr */
} }
/* */ /* _ opts ... */
static int get_redirect(lua_State *L, posix_spawn_file_actions_t *file_actions, static int get_redirect(lua_State *L, posix_spawn_file_actions_t *file_actions,
const char *stdname, int descriptor) const char *stdname, int descriptor)
{ {
@ -243,22 +277,38 @@ static int get_redirect(lua_State *L, posix_spawn_file_actions_t *file_actions,
return ret; return ret;
} }
/* _ opts ... */
static void spawn_param_redirects(struct spawn_params *p)
{
posix_spawn_file_actions_init(&p->file_actions);
p->has_actions = 1;
get_redirect(p->L, &p->file_actions, "stdin", STDIN_FILENO);
get_redirect(p->L, &p->file_actions, "stdout", STDOUT_FILENO);
get_redirect(p->L, &p->file_actions, "stderr", STDERR_FILENO);
}
#define PROCESS_HANDLE "process" #define PROCESS_HANDLE "process"
struct process { struct process {
pid_t pid;
int status; int status;
pid_t pid;
}; };
static int spawn_param_execute(struct spawn_params *p, struct process *proc)
{
int ret;
proc->status = -1;
ret = posix_spawnp(&proc->pid, p->command, &p->file_actions, 0, (char *const *)p->argv, (char *const *)p->envp);
if (p->has_actions)
posix_spawn_file_actions_destroy(&p->file_actions);
return ret == 0;
}
/* filename [args-opts] -- true/nil,error */ /* filename [args-opts] -- true/nil,error */
/* args-opts -- true/nil,error */ /* args-opts -- true/nil,error */
static int ex_spawn(lua_State *L) static int ex_spawn(lua_State *L)
{ {
const char *command; struct spawn_params params = {L};
const char **argv;
const char **envp;
posix_spawn_file_actions_t file_actions, *pactions = 0;
struct process *proc; struct process *proc;
int ret;
if (lua_type(L, 1) == LUA_TTABLE) { if (lua_type(L, 1) == LUA_TTABLE) {
lua_getfield(L, 1, "command"); /* opts ... cmd */ lua_getfield(L, 1, "command"); /* opts ... cmd */
@ -282,18 +332,14 @@ static int ex_spawn(lua_State *L)
} }
} }
/* XXX confusing */ /* get filename to execute */
command = luaL_checkstring(L, 1); spawn_param_filename(&params);
/* get arguments, environment, and redirections */ /* get arguments, environment, and redirections */
switch (lua_type(L, 2)) { switch (lua_type(L, 2)) {
default: luaL_argerror(L, 2, "expected options table"); break; default: luaL_argerror(L, 2, "expected options table"); break;
case LUA_TNONE: case LUA_TNONE:
argv = lua_newuserdata(L, 2 * sizeof *argv); spawn_param_defaults(&params);
argv[0] = command;
argv[1] = 0;
envp = (const char **)environ;
pactions = 0;
break; break;
case LUA_TTABLE: case LUA_TTABLE:
lua_getfield(L, 2, "args"); /* cmd opts ... argtab */ lua_getfield(L, 2, "args"); /* cmd opts ... argtab */
@ -306,43 +352,24 @@ static int ex_spawn(lua_State *L)
case LUA_TTABLE: case LUA_TTABLE:
if (lua_objlen(L, 2) > 0) if (lua_objlen(L, 2) > 0)
luaL_error(L, "cannot specify both the args option and array values"); luaL_error(L, "cannot specify both the args option and array values");
argv = build_args(L); /* cmd opts ... */ spawn_param_args(&params); /* cmd opts ... */
if (!argv[0]) argv[0] = command;
break; break;
} }
lua_getfield(L, 2, "env"); lua_getfield(L, 2, "env"); /* cmd opts ... envtab */
switch (lua_type(L, -1)) { switch (lua_type(L, -1)) {
default: luaL_error(L, "env option must be a table"); break; default: luaL_error(L, "env option must be a table"); break;
case LUA_TNIL: case LUA_TNIL:
envp = (const char **)environ;
break;
case LUA_TTABLE: case LUA_TTABLE:
envp = build_env(L); /* cmd opts ... */ spawn_param_env(&params); /* cmd opts ... */
break; break;
} }
posix_spawn_file_actions_init(&file_actions); spawn_param_redirects(&params);
if (get_redirect(L, &file_actions, "stdin", STDIN_FILENO)
| get_redirect(L, &file_actions, "stdout", STDOUT_FILENO)
| get_redirect(L, &file_actions, "stderr", STDERR_FILENO)) {
pactions = &file_actions;
}
else {
posix_spawn_file_actions_destroy(&file_actions);
pactions = 0;
}
} }
proc = lua_newuserdata(L, sizeof *proc); /* cmd opts ... proc */ proc = lua_newuserdata(L, sizeof *proc); /* cmd opts ... proc */
luaL_getmetatable(L, PROCESS_HANDLE); /* cmd opts ... proc M */ luaL_getmetatable(L, PROCESS_HANDLE); /* cmd opts ... proc M */
lua_setmetatable(L, -2); /* cmd opts ... proc */ lua_setmetatable(L, -2); /* cmd opts ... proc */
proc->status = -1; if (!spawn_param_execute(&params, proc))
assert(argv && argv[0]); return push_error(L);
assert(envp && envp[0]);
ret = posix_spawnp(&proc->pid, command, pactions, 0, (char *const *)argv, (char *const *)envp);
debug("returns %d:%ld\n", ret, (long)proc->pid);
if (pactions)
posix_spawn_file_actions_destroy(pactions);
if (0 != ret)
return posix_error(L);
return 1; /* ... proc */ return 1; /* ... proc */
} }
@ -353,7 +380,7 @@ static int process_wait(lua_State *L)
if (p->status == -1) { if (p->status == -1) {
int status; int status;
if (-1 == waitpid(p->pid, &status, 0)) if (-1 == waitpid(p->pid, &status, 0))
return posix_error(L); return push_error(L);
p->status = WEXITSTATUS(status); p->status = WEXITSTATUS(status);
} }
lua_pushnumber(L, p->status); lua_pushnumber(L, p->status);

3
w32api/Makefile
Unescape Escape View File

@ -3,7 +3,8 @@ CFLAGS = $(WARNINGS) $(DEFINES) $(INCLUDES)
DEFINES = -DWIN32_LEAN_AND_MEAN DEFINES = -DWIN32_LEAN_AND_MEAN
INCLUDES = -I${LUA}/include INCLUDES = -I${LUA}/include
WARNINGS = -W -Wall -Wno-missing-braces WARNINGS = -W -Wall -Wno-missing-braces
LUA = /home/mark/src/lang/lua/lua-5.1-rc2 #LUA = /home/mark/src/lang/lua/lua-5.1-rc2
LUA = /home/mark/src/lang/lua/lua51
LUALIBS = -L$(LUA)/lib -llua51 LUALIBS = -L$(LUA)/lib -llua51
ex.dll: ex.o ex.dll: ex.o

167
w32api/ex.c
Unescape Escape View File

@ -12,8 +12,6 @@
#include <fcntl.h> #include <fcntl.h>
#include <sys/locking.h> #include <sys/locking.h>
#define debug(...) fprintf(stderr,__VA_ARGS__)
/* Generally useful function -- what luaL_checkudata() should do */ /* Generally useful function -- what luaL_checkudata() should do */
static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname) static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname)
@ -25,7 +23,7 @@ static void *luaL_checkuserdata(lua_State *L, int idx, const char *tname)
/* -- nil error */ /* -- nil error */
static int windows_error(lua_State *L) static int push_error(lua_State *L)
{ {
DWORD error = GetLastError(); DWORD error = GetLastError();
char buffer[1024]; char buffer[1024];
@ -54,7 +52,7 @@ static int ex_getenv(lua_State *L)
size_t len; size_t len;
len = GetEnvironmentVariable(nam, val, sizeof val); len = GetEnvironmentVariable(nam, val, sizeof val);
if (sizeof val < len) if (sizeof val < len)
return windows_error(L); return push_error(L);
else if (len == 0) else if (len == 0)
lua_pushnil(L); lua_pushnil(L);
else else
@ -68,7 +66,7 @@ static int ex_setenv(lua_State *L)
const char *nam = luaL_checkstring(L, 1); const char *nam = luaL_checkstring(L, 1);
const char *val = luaL_checkstring(L, 2); const char *val = luaL_checkstring(L, 2);
if (!SetEnvironmentVariable(nam, val)) if (!SetEnvironmentVariable(nam, val))
return windows_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -78,7 +76,7 @@ static int ex_unsetenv(lua_State *L)
{ {
const char *nam = luaL_checkstring(L, 1); const char *nam = luaL_checkstring(L, 1);
if (!SetEnvironmentVariable(nam, 0)) if (!SetEnvironmentVariable(nam, 0))
return windows_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -89,13 +87,11 @@ static int ex_environ(lua_State *L)
const char *nam, *val, *end; const char *nam, *val, *end;
const char *envs; const char *envs;
if (!(envs = GetEnvironmentStrings())) if (!(envs = GetEnvironmentStrings()))
return windows_error(L); return push_error(L);
lua_newtable(L); lua_newtable(L);
for (nam = envs; *nam; nam = end + 1) { for (nam = envs; *nam; nam = end + 1) {
val = strchr(nam, '='); end = strchr(val = strchr(nam, '=') + 1, '\0');
end = strchr(val, '\0'); lua_pushlstring(L, nam, val - nam - 1);
lua_pushlstring(L, nam, val - nam);
val++;
lua_pushlstring(L, val, end - val); lua_pushlstring(L, val, end - val);
lua_settable(L, -3); lua_settable(L, -3);
} }
@ -117,7 +113,7 @@ static int ex_chdir(lua_State *L)
{ {
const char *pathname = luaL_checkstring(L, 1); const char *pathname = luaL_checkstring(L, 1);
if (!SetCurrentDirectory(pathname)) if (!SetCurrentDirectory(pathname))
return windows_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -127,7 +123,7 @@ static int ex_mkdir(lua_State *L)
{ {
const char *pathname = luaL_checkstring(L, 1); const char *pathname = luaL_checkstring(L, 1);
if (!CreateDirectory(pathname, 0)) if (!CreateDirectory(pathname, 0))
return windows_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -138,7 +134,7 @@ static int ex_currentdir(lua_State *L)
char pathname[PATH_MAX + 1]; char pathname[PATH_MAX + 1];
size_t len; size_t len;
if (!(len = GetCurrentDirectory(sizeof pathname, pathname))) if (!(len = GetCurrentDirectory(sizeof pathname, pathname)))
return windows_error(L); return push_error(L);
lua_pushlstring(L, pathname, len); lua_pushlstring(L, pathname, len);
return 1; return 1;
} }
@ -174,7 +170,7 @@ static int file_lock(lua_State *L, FILE *f, const char *mode, long offset, long
ret = flags ? LockFileEx(h, flags, 0, len.LowPart, len.HighPart, &ov) ret = flags ? LockFileEx(h, flags, 0, len.LowPart, len.HighPart, &ov)
: UnlockFileEx(h, 0, len.LowPart, len.HighPart, &ov); : UnlockFileEx(h, 0, len.LowPart, len.HighPart, &ov);
if (!ret) if (!ret)
return windows_error(L); return push_error(L);
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
return 1; return 1;
} }
@ -217,17 +213,46 @@ static int ex_pipe(lua_State *L)
{ {
HANDLE ph[2]; HANDLE ph[2];
if (0 == CreatePipe(ph+0, ph+1, 0, 0)) if (0 == CreatePipe(ph+0, ph+1, 0, 0))
return windows_error(L); return push_error(L);
return make_pipe(L, return make_pipe(L,
_fdopen(_open_osfhandle((long)ph[0], _O_RDONLY), "r"), _fdopen(_open_osfhandle((long)ph[0], _O_RDONLY), "r"),
_fdopen(_open_osfhandle((long)ph[1], _O_WRONLY), "w")); _fdopen(_open_osfhandle((long)ph[1], _O_WRONLY), "w"));
} }
struct spawn_params {
lua_State *L;
const char *cmdline;
const char *environment;
STARTUPINFO si;
};
static int needs_quoting(const char *s) static int needs_quoting(const char *s)
{ {
return s[0] != '"' && strchr(s, ' '); return s[0] != '"' && strchr(s, ' ');
} }
/* filename ... */
static void spawn_param_filename(struct spawn_params *p)
{
/* XXX luaL_checkstring is confusing here */
p->cmdline = luaL_checkstring(p->L, 1);
if (needs_quoting(p->cmdline)) {
lua_pushliteral(p->L, "\""); /* cmd ... q */
lua_pushvalue(p->L, 1); /* cmd ... q cmd */
lua_pushvalue(p->L, -2); /* cmd ... q cmd q */
lua_concat(p->L, 3); /* cmd ... "cmd" */
lua_replace(p->L, 1); /* "cmd" ... */
}
}
/* -- */
static void spawn_param_default(struct spawn_params *p)
{
p->environment = 0;
memset(&p->si, 0, sizeof p->si);
p->si.cb = sizeof p->si;
}
/* {arg1,arg 2} => " arg1 \"arg2\"" */ /* {arg1,arg 2} => " arg1 \"arg2\"" */
static const char *concat_args(lua_State *L) static const char *concat_args(lua_State *L)
{ {
@ -250,6 +275,16 @@ static const char *concat_args(lua_State *L)
return lua_tostring(L, -1); return lua_tostring(L, -1);
} }
/* cmd opts ... argtab */
static void spawn_param_args(struct spawn_params *p)
{
concat_args(p->L); /* cmd opts ... argstr */
lua_pushvalue(p->L, 1); /* cmd opts ... argstr cmd */
lua_replace(p->L, -2); /* cmd opts ... cmd argstr */
lua_concat(p->L, 2); /* cmd opts ... cmdline */
p->cmdline = lua_tostring(p->L, -1);
}
/* {nam1=val1,nam2=val2} => "nam1=val1\0nam2=val2\0" */ /* {nam1=val1,nam2=val2} => "nam1=val1\0nam2=val2\0" */
static const char *concat_env(lua_State *L) static const char *concat_env(lua_State *L)
{ {
@ -274,7 +309,19 @@ static const char *concat_env(lua_State *L)
return lua_tostring(L, -1); return lua_tostring(L, -1);
} }
/* XXX document me */ /* ... envtab */
static void spawn_param_env(struct spawn_params *p)
{
if (lua_isnil(p->L, -1)) {
p->environment = 0;
lua_pop(p->L, 1);
}
else {
p->environment = concat_env(p->L);
}
}
/* _ opts ... */
static int get_redirect(lua_State *L, const char *stdname, HANDLE *ph) static int get_redirect(lua_State *L, const char *stdname, HANDLE *ph)
{ {
int ret; int ret;
@ -288,22 +335,44 @@ static int get_redirect(lua_State *L, const char *stdname, HANDLE *ph)
return ret; return ret;
} }
/* _ opts ... */
static void spawn_param_redirect(struct spawn_params *p)
{
p->si.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
p->si.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
p->si.hStdError = GetStdHandle(STD_ERROR_HANDLE);
if (get_redirect(p->L, "stdin", &p->si.hStdInput)
| get_redirect(p->L, "stdout", &p->si.hStdOutput)
| get_redirect(p->L, "stderr", &p->si.hStdError))
p->si.dwFlags = STARTF_USESTDHANDLES;
}
#define PROCESS_HANDLE "process" #define PROCESS_HANDLE "process"
struct process { struct process {
HANDLE hProcess;
int status; int status;
HANDLE hProcess;
}; };
static int spawn_param_execute(struct spawn_params *p, struct process *proc)
{
char *c = strdup(p->cmdline);
char *e = (char *)p->environment; // strdup(p->environment);
PROCESS_INFORMATION pi;
proc->status = -1;
/* XXX does CreateProcess modify its environment argument? */
int ret = CreateProcess(0, c, 0, 0, 0, 0, e, 0, &p->si, &pi);
if (e) free(e);
free(c);
if (ret) proc->hProcess = pi.hProcess;
return ret;
}
/* filename [args-opts] -- true/nil,error */ /* filename [args-opts] -- true/nil,error */
/* args-opts -- true/nil,error */ /* args-opts -- true/nil,error */
static int ex_spawn(lua_State *L) static int ex_spawn(lua_State *L)
{ {
const char *cmdline; struct spawn_params params = {L};
const char *environment; struct process *proc;
struct process *p;
STARTUPINFO si = {sizeof si};
PROCESS_INFORMATION pi;
BOOL ret;
if (lua_type(L, 1) == LUA_TTABLE) { if (lua_type(L, 1) == LUA_TTABLE) {
lua_getfield(L, 1, "command"); /* opts ... cmd */ lua_getfield(L, 1, "command"); /* opts ... cmd */
@ -328,21 +397,13 @@ static int ex_spawn(lua_State *L)
} }
/* get command */ /* get command */
/* XXX luaL_checkstring is confusing here */ spawn_param_filename(&params);
cmdline = luaL_checkstring(L, 1);
if (needs_quoting(cmdline)) {
lua_pushliteral(L, "\""); /* cmd ... q */
lua_pushvalue(L, 1); /* cmd ... q cmd */
lua_pushvalue(L, -2); /* cmd ... q cmd q */
lua_concat(L, 3); /* cmd ... "cmd" */
lua_replace(L, 1); /* "cmd" ... */
}
/* get arguments, environment, and redirections */ /* get arguments, environment, and redirections */
switch (lua_type(L, 2)) { switch (lua_type(L, 2)) {
default: luaL_argerror(L, 2, "expected options table"); break; default: luaL_argerror(L, 2, "expected options table"); break;
case LUA_TNONE: case LUA_TNONE:
environment = 0; spawn_param_default(&params);
break; break;
case LUA_TTABLE: case LUA_TTABLE:
lua_getfield(L, 2, "args"); /* cmd opts ... argtab */ lua_getfield(L, 2, "args"); /* cmd opts ... argtab */
@ -350,56 +411,30 @@ static int ex_spawn(lua_State *L)
default: luaL_error(L, "args option must be an array"); break; default: luaL_error(L, "args option must be an array"); break;
case LUA_TNIL: case LUA_TNIL:
lua_pop(L, 1); /* cmd opts ... */ lua_pop(L, 1); /* cmd opts ... */
if (lua_objlen(L, 2) == 0) break;
lua_pushvalue(L, 2); /* cmd opts ... opts */ lua_pushvalue(L, 2); /* cmd opts ... opts */
if (0) /*FALLTHRU*/ if (0) /*FALLTHRU*/
case LUA_TTABLE: case LUA_TTABLE:
if (lua_objlen(L, 2) > 0) if (lua_objlen(L, 2) > 0)
luaL_error(L, "cannot specify both the args option and array values"); luaL_error(L, "cannot specify both the args option and array values");
concat_args(L); /* cmd opts ... argstr */ spawn_param_args(&params);
lua_pushvalue(L, 1); /* cmd opts ... argstr cmd */
lua_replace(L, -2); /* cmd opts ... cmd argstr */
lua_concat(L, 2); /* cmd opts ... cmdline */
cmdline = lua_tostring(L, -1);
break; break;
} }
lua_getfield(L, 2, "env"); /* cmd opts ... envtab */ lua_getfield(L, 2, "env"); /* cmd opts ... envtab */
switch (lua_type(L, -1)) { switch (lua_type(L, -1)) {
default: luaL_error(L, "env option must be a table"); break; default: luaL_error(L, "env option must be a table"); break;
case LUA_TNIL: case LUA_TNIL:
environment = 0;
lua_pop(L, 1); /* cmd opts */
break;
case LUA_TTABLE: case LUA_TTABLE:
environment = concat_env(L); /* cmd opts ... envstr */ spawn_param_env(&params);
break; break;
} }
si.hStdInput = GetStdHandle(STD_INPUT_HANDLE); spawn_param_redirect(&params);
si.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
si.hStdError = GetStdHandle(STD_ERROR_HANDLE);
if (get_redirect(L, "stdin", &si.hStdInput) /* cmd opts ... in? */
| get_redirect(L, "stdout", &si.hStdOutput) /* cmd opts ... out? */
| get_redirect(L, "stderr", &si.hStdError)) /* cmd opts ... err? */
si.dwFlags = STARTF_USESTDHANDLES;
break; break;
} }
proc = lua_newuserdata(L, sizeof *proc); /* cmd opts ... proc */
p = lua_newuserdata(L, sizeof *p); /* cmd opts ... proc */
luaL_getmetatable(L, PROCESS_HANDLE); /* cmd opts ... proc M */ luaL_getmetatable(L, PROCESS_HANDLE); /* cmd opts ... proc M */
lua_setmetatable(L, -2); /* cmd opts ... proc */ lua_setmetatable(L, -2); /* cmd opts ... proc */
p->status = -1; if (!spawn_param_execute(&params, proc))
return push_error(L);
/* XXX does CreateProcess modify its environment argument? */
cmdline = strdup(cmdline);
if (environment) environment = strdup(environment);
debug("CreateProcess(%s)\n", cmdline);
ret = CreateProcess(0, (char *)cmdline, 0, 0, 0, 0, (char *)environment, 0, &si, &pi);
// if (environment) free((char *)environment);
// free((char *)cmdline);
if (!ret)
return windows_error(L);
p->hProcess = pi.hProcess;
return 1; /* ... proc */ return 1; /* ... proc */
} }

Write Preview
Loading…
Cancel
Save