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

added check_file function to check for closed FILE*

Browse Source 
use registry for associating pathname with diriter
master
mark 21 years ago
parent 8e3b4ac49d
commit 3c9cbc3761
3 changed files with 65 additions and 62 deletions
Show Stats Download Patch File Download Diff File

2
posix/Makefile
Unescape Escape View File

@ -7,6 +7,8 @@ LUA = /home/mark/src/lang/lua/lua51
ex-OBJS = ex.o spawn.o ex-OBJS = ex.o spawn.o
default: ex.dll
ex.so: $(ex-OBJS); $(CC) -shared -o $@ $(ex-OBJS) ex.so: $(ex-OBJS); $(CC) -shared -o $@ $(ex-OBJS)
#LIBS = -L$(LUA)/lib -llua51 spawn.a #LIBS = -L$(LUA)/lib -llua51 spawn.a

57
posix/ex.c
Unescape Escape View File

@ -13,6 +13,7 @@
#include <fcntl.h> #include <fcntl.h>
#include <dirent.h> #include <dirent.h>
#include <sys/stat.h> #include <sys/stat.h>
#include "spawn.h" #include "spawn.h"
@ -122,6 +123,13 @@ static int ex_currentdir(lua_State *L)
} }
static FILE *check_file(lua_State *L, int idx)
{
FILE **pf = checkuserdata(L, idx, LUA_FILEHANDLE);
if (!*pf) return luaL_error(L, "attempt to use a closed file"), NULL;
return *pf;
}
/* pathname/file -- entry */ /* pathname/file -- entry */
static int ex_dirent(lua_State *L) static int ex_dirent(lua_State *L)
{ {
@ -134,8 +142,8 @@ static int ex_dirent(lua_State *L)
return push_error(L); return push_error(L);
} break; } break;
case LUA_TUSERDATA: { case LUA_TUSERDATA: {
FILE **pf = checkuserdata(L, 1, LUA_FILEHANDLE); FILE *f = check_file(L, 1);
if (-1 == fstat(fileno(*pf), &st)) if (-1 == fstat(fileno(f), &st))
return push_error(L); return push_error(L);
} break; } break;
} }
@ -159,49 +167,46 @@ static int ex_dirent(lua_State *L)
#define DIR_HANDLE "DIR*" #define DIR_HANDLE "DIR*"
struct diriter { struct diriter {
DIR *dir; DIR *dir;
size_t pathlen;
char pathname[PATH_MAX + 1];
}; };
/* ...diriter... -- ...diriter... pathname */
static int diriter_getpathname(lua_State *L, int index) static int diriter_getpathname(lua_State *L, int index)
{ {
struct diriter *pi = lua_touserdata(L, index); lua_pushvalue(L, index);
lua_pushlstring(L, pi->pathname, pi->pathlen); lua_gettable(L, LUA_REGISTRYINDEX);
return 1; return 1;
} }
/* ...diriter... pathname -- ...diriter... */
static int diriter_setpathname(lua_State *L, int index) static int diriter_setpathname(lua_State *L, int index)
{ {
struct diriter *pi = lua_touserdata(L, index);
size_t len; size_t len;
const char *path = lua_tolstring(L, -1, &len); const char *path = lua_tolstring(L, -1, &len);
if (len >= sizeof pi->pathname - 1) if (path && path[len - 1] != *LUA_DIRSEP) {
return luaL_argerror(L, 1, "pathname too long");
if (path[len - 1] != *LUA_DIRSEP) {
lua_pushliteral(L, LUA_DIRSEP); lua_pushliteral(L, LUA_DIRSEP);
lua_concat(L, 2); lua_concat(L, 2);
path = lua_tostring(L, -1);
len++;
} }
memcpy(pi->pathname, path, len + 1); lua_pushvalue(L, index); /* ... pathname diriter */
pi->pathlen = len; lua_insert(L, -2); /* ... diriter pathname */
lua_pop(L, 1); lua_settable(L, LUA_REGISTRYINDEX); /* ... */
return 0; return 0;
} }
/* dir -- */ /* diriter -- diriter */
static int diriter_close(lua_State *L) static int diriter_close(lua_State *L)
{ {
struct diriter *pi = lua_touserdata(L, 1); struct diriter *pi = lua_touserdata(L, 1);
if (pi->dir) { if (pi->dir) {
closedir(pi->dir); closedir(pi->dir);
pi->dir = 0; pi->dir = 0;
lua_pushnil(L);
diriter_setpathname(L, 1);
} }
return 0; return 0;
} }
/* pathname -- iter state nil */ /* pathname -- iter state nil */
/* dir ... -- entry */ /* diriter ... -- entry */
static int ex_dir(lua_State *L) static int ex_dir(lua_State *L)
{ {
const char *pathname; const char *pathname;
@ -228,13 +233,13 @@ static int ex_dir(lua_State *L)
pi->dir = 0; pi->dir = 0;
return push_error(L); return push_error(L);
} }
lua_newtable(L); /* dir ... entry */ lua_newtable(L); /* diriter ... entry */
diriter_getpathname(L, 1); /* dir ... entry dirpath */ diriter_getpathname(L, 1); /* diriter ... entry dirpath */
lua_pushstring(L, d->d_name); /* dir ... entry dirpath name */ lua_pushstring(L, d->d_name); /* diriter ... entry dirpath name */
lua_pushliteral(L, "name"); /* dir ... entry dirpath name "name" */ lua_pushliteral(L, "name"); /* diriter ... entry dirpath name "name" */
lua_pushvalue(L, -2); /* dir ... entry dirpath name "name" name */ lua_pushvalue(L, -2); /* diriter ... entry dirpath name "name" name */
lua_settable(L, -5); /* dir ... entry dirpath name */ lua_settable(L, -5); /* diriter ... entry dirpath name */
lua_concat(L, 2); /* dir ... entry fullpath */ lua_concat(L, 2); /* diriter ... entry fullpath */
lua_replace(L, 1); /* fullpath ... entry */ lua_replace(L, 1); /* fullpath ... entry */
lua_replace(L, 2); /* fullpath entry ... */ lua_replace(L, 2); /* fullpath entry ... */
return ex_dirent(L); return ex_dirent(L);
@ -264,11 +269,11 @@ static int file_lock(lua_State *L, FILE *f, const char *mode, long offset, long
/* file mode [offset [length]] -- true/nil error */ /* file mode [offset [length]] -- true/nil error */
static int ex_lock(lua_State *L) static int ex_lock(lua_State *L)
{ {
FILE **pf = checkuserdata(L, 1, LUA_FILEHANDLE); FILE *f = check_file(L, 1);
const char *mode = luaL_checkstring(L, 2); const char *mode = luaL_checkstring(L, 2);
long offset = luaL_optnumber(L, 3, 0); long offset = luaL_optnumber(L, 3, 0);
long length = luaL_optnumber(L, 4, 0); long length = luaL_optnumber(L, 4, 0);
return file_lock(L, *pf, mode, offset, length); return file_lock(L, f, mode, offset, length);
} }
/* file [offset [length]] -- true/nil error */ /* file [offset [length]] -- true/nil error */

68
w32api/ex.c
Unescape Escape View File

@ -14,8 +14,6 @@
#include "spawn.h" #include "spawn.h"
#define debug(...) fprintf(stderr,__VA_ARGS__)
/* Generally useful function -- what luaL_checkudata() should do */ /* Generally useful function -- what luaL_checkudata() should do */
extern void *checkuserdata(lua_State *L, int idx, const char *tname) extern void *checkuserdata(lua_State *L, int idx, const char *tname)
@ -146,6 +144,13 @@ static int ex_currentdir(lua_State *L)
} }
static FILE *check_file(lua_State *L, int idx)
{
FILE **pf = checkuserdata(L, idx, LUA_FILEHANDLE);
if (!*pf) return luaL_error(L, "attempt to use a closed file"), NULL;
return *pf;
}
static BOOL GetFileInformationByPath(LPCSTR name, BY_HANDLE_FILE_INFORMATION *pinfo) static BOOL GetFileInformationByPath(LPCSTR name, BY_HANDLE_FILE_INFORMATION *pinfo)
{ {
HANDLE h = CreateFile(name, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0); HANDLE h = CreateFile(name, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);
@ -193,10 +198,9 @@ static int ex_dirent(lua_State *L)
size = get_size(name); size = get_size(name);
} break; } break;
case LUA_TUSERDATA: { case LUA_TUSERDATA: {
FILE **pf = checkuserdata(L, 1, LUA_FILEHANDLE); FILE *f = check_file(L, 1);
BY_HANDLE_FILE_INFORMATION info; BY_HANDLE_FILE_INFORMATION info;
BOOL ret = GetFileInformationByHandle(get_handle(*pf), &info); if (!GetFileInformationByHandle(get_handle(f), &info))
if (!ret)
return push_error(L); return push_error(L);
attr = info.dwFileAttributes; attr = info.dwFileAttributes;
size = info.nFileSizeHigh; size <<= 32; size += info.nFileSizeLow; size = info.nFileSizeHigh; size <<= 32; size += info.nFileSizeLow;
@ -223,49 +227,46 @@ static int ex_dirent(lua_State *L)
struct diriter { struct diriter {
HANDLE hf; HANDLE hf;
WIN32_FIND_DATA fd; WIN32_FIND_DATA fd;
size_t pathlen;
char pathname[MAX_PATH + 1];
}; };
/* ...diriter... -- ...diriter... pathname */
static int diriter_getpathname(lua_State *L, int index) static int diriter_getpathname(lua_State *L, int index)
{ {
struct diriter *pi = lua_touserdata(L, index); lua_pushvalue(L, index);
lua_pushlstring(L, pi->pathname, pi->pathlen); lua_gettable(L, LUA_REGISTRYINDEX);
return 1; return 1;
} }
/* ...diriter... pathname -- ...diriter... */
static int diriter_setpathname(lua_State *L, int index) static int diriter_setpathname(lua_State *L, int index)
{ {
struct diriter *pi = lua_touserdata(L, index);
size_t len; size_t len;
const char *path = lua_tolstring(L, -1, &len); const char *path = lua_tolstring(L, -1, &len);
if (len >= sizeof pi->pathname - 1)
return luaL_argerror(L, 1, "pathname too long");
if (path[len - 1] != *LUA_DIRSEP) { if (path[len - 1] != *LUA_DIRSEP) {
lua_pushliteral(L, LUA_DIRSEP); lua_pushliteral(L, LUA_DIRSEP);
lua_concat(L, 2); lua_concat(L, 2);
path = lua_tostring(L, -1);
len++;
} }
memcpy(pi->pathname, path, len + 1); lua_pushvalue(L, index); /* ... pathname diriter */
pi->pathlen = len; lua_insert(L, -2); /* ... diriter pathname */
lua_pop(L, 1); lua_settable(L, LUA_REGISTRYINDEX); /* ... */
return 0; return 0;
} }
/* dir -- */ /* diriter -- diriter */
static int diriter_close(lua_State *L) static int diriter_close(lua_State *L)
{ {
struct diriter *pi = lua_touserdata(L, 1); struct diriter *pi = lua_touserdata(L, 1);
if (pi->hf != INVALID_HANDLE_VALUE) { if (pi->hf != INVALID_HANDLE_VALUE) {
FindClose(pi->hf); FindClose(pi->hf);
pi->hf = INVALID_HANDLE_VALUE; pi->hf = INVALID_HANDLE_VALUE;
lua_pushnil(L);
diriter_setpathname(L, 1);
} }
return 0; return 0;
} }
/* pathname -- iter state nil */ /* pathname -- iter state nil */
/* dir ... -- entry */ /* diriter ... -- entry */
static int ex_dir(lua_State *L) static int ex_dir(lua_State *L)
{ {
const char *pathname; const char *pathname;
@ -279,36 +280,31 @@ static int ex_dir(lua_State *L)
pathname = lua_tostring(L, -1); pathname = lua_tostring(L, -1);
lua_pushcfunction(L, ex_dir); /* pathname ... pat iter */ lua_pushcfunction(L, ex_dir); /* pathname ... pat iter */
pi = lua_newuserdata(L, sizeof *pi); /* pathname ... pat iter state */ pi = lua_newuserdata(L, sizeof *pi); /* pathname ... pat iter state */
debug("FindFirstFile(\"%s\")\n", pathname);
pi->hf = FindFirstFile(pathname, &pi->fd); pi->hf = FindFirstFile(pathname, &pi->fd);
if (pi->hf == INVALID_HANDLE_VALUE) if (pi->hf == INVALID_HANDLE_VALUE)
return push_error(L); return push_error(L);
luaL_getmetatable(L, DIR_HANDLE); /* pathname ... pat iter state M */ luaL_getmetatable(L, DIR_HANDLE); /* pathname ... pat iter state M */
lua_setmetatable(L, -2); /* pathname ... pat iter state */ lua_setmetatable(L, -2); /* pathname ... pat iter state */
lua_pushvalue(L, 1); /* pathname ... pat iter state pathname */ lua_pushvalue(L, 1); /* pathname ... pat iter state pathname */
diriter_setpathname(L, -2); /* pathname ... pat iter state */ diriter_setpathname(L, -2); /* pathname ... pat iter state */
debug("returned DIR:%p\n", lua_topointer(L, -1));
return 2; return 2;
case LUA_TUSERDATA: case LUA_TUSERDATA:
debug("received DIR:%p\n", lua_topointer(L, 1));
pi = checkuserdata(L, 1, DIR_HANDLE); pi = checkuserdata(L, 1, DIR_HANDLE);
if (pi->hf == INVALID_HANDLE_VALUE) if (pi->hf == INVALID_HANDLE_VALUE)
return 0; return 0;
debug("Found: %s\n", pi->fd.cFileName); lua_newtable(L); /* diriter ... entry */
lua_newtable(L); /* dir ... entry */ diriter_getpathname(L, 1); /* diriter ... entry dirpath */
diriter_getpathname(L, 1); /* dir ... entry dirpath */ lua_pushstring(L, pi->fd.cFileName); /* diriter ... entry dirpath name */
lua_pushstring(L, pi->fd.cFileName); /* dir ... entry dirpath name */ lua_pushliteral(L, "name"); /* diriter ... entry dirpath name "name" */
lua_pushliteral(L, "name"); /* dir ... entry dirpath name "name" */ lua_pushvalue(L, -2); /* diriter ... entry dirpath name "name" name */
lua_pushvalue(L, -2); /* dir ... entry dirpath name "name" name */ lua_settable(L, -5); /* diriter ... entry dirpath name */
lua_settable(L, -5); /* dir ... entry dirpath name */ lua_concat(L, 2); /* diriter ... entry fullpath */
lua_concat(L, 2); /* dir ... entry fullpath */
if (!FindNextFile(pi->hf, &pi->fd)) { if (!FindNextFile(pi->hf, &pi->fd)) {
FindClose(pi->hf); FindClose(pi->hf);
pi->hf = INVALID_HANDLE_VALUE; pi->hf = INVALID_HANDLE_VALUE;
} }
lua_replace(L, 1); /* fullpath ... entry */ lua_replace(L, 1); /* fullpath ... entry */
lua_replace(L, 2); /* fullpath entry ... */ lua_replace(L, 2); /* fullpath entry ... */
debug("passing off to dirent\n");
return ex_dirent(L); return ex_dirent(L);
} }
/*NOTREACHED*/ /*NOTREACHED*/
@ -342,11 +338,11 @@ static int file_lock(lua_State *L, FILE *f, const char *mode, long offset, long
/* file mode [offset [length]] -- true/nil error */ /* file mode [offset [length]] -- true/nil error */
static int ex_lock(lua_State *L) static int ex_lock(lua_State *L)
{ {
FILE **pf = checkuserdata(L, 1, LUA_FILEHANDLE); FILE *f = check_file(L, 1);
const char *mode = luaL_checkstring(L, 2); const char *mode = luaL_checkstring(L, 2);
long offset = luaL_optnumber(L, 3, 0); long offset = luaL_optnumber(L, 3, 0);
long length = luaL_optnumber(L, 4, 0); long length = luaL_optnumber(L, 4, 0);
return file_lock(L, *pf, mode, offset, length); return file_lock(L, f, mode, offset, length);
} }
/* file [offset [length]] -- true/nil error */ /* file [offset [length]] -- true/nil error */
@ -386,8 +382,8 @@ static int ex_pipe(lua_State *L)
} }
/* 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)
{ {
struct spawn_params params = {L}; struct spawn_params params = {L};

Write Preview
Loading…
Cancel
Save