mirror of https://github.com/odrling/Aegisub
568 lines
14 KiB
C
568 lines
14 KiB
C
/*
|
|
** I/O library.
|
|
** Copyright (C) 2005-2017 Mike Pall. See Copyright Notice in luajit.h
|
|
**
|
|
** Major portions taken verbatim or adapted from the Lua interpreter.
|
|
** Copyright (C) 1994-2011 Lua.org, PUC-Rio. See Copyright Notice in lua.h
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
|
|
#define lib_io_c
|
|
#define LUA_LIB
|
|
|
|
#include "lua.h"
|
|
#include "lauxlib.h"
|
|
#include "lualib.h"
|
|
|
|
#include "lj_obj.h"
|
|
#include "lj_gc.h"
|
|
#include "lj_err.h"
|
|
#include "lj_buf.h"
|
|
#include "lj_str.h"
|
|
#include "lj_state.h"
|
|
#include "lj_strfmt.h"
|
|
#include "lj_ff.h"
|
|
#include "lj_lib.h"
|
|
|
|
#if LJ_TARGET_WINDOWS
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
|
|
static int widen(const char *in, wchar_t *out)
|
|
{
|
|
return MultiByteToWideChar(CP_UTF8, 0, in, -1, out, MAX_PATH);
|
|
}
|
|
#endif
|
|
|
|
/* Userdata payload for I/O file. */
|
|
typedef struct IOFileUD {
|
|
FILE *fp; /* File handle. */
|
|
uint32_t type; /* File type. */
|
|
} IOFileUD;
|
|
|
|
#define IOFILE_TYPE_FILE 0 /* Regular file. */
|
|
#define IOFILE_TYPE_PIPE 1 /* Pipe. */
|
|
#define IOFILE_TYPE_STDF 2 /* Standard file handle. */
|
|
#define IOFILE_TYPE_MASK 3
|
|
|
|
#define IOFILE_FLAG_CLOSE 4 /* Close after io.lines() iterator. */
|
|
|
|
#define IOSTDF_UD(L, id) (&gcref(G(L)->gcroot[(id)])->ud)
|
|
#define IOSTDF_IOF(L, id) ((IOFileUD *)uddata(IOSTDF_UD(L, (id))))
|
|
|
|
/* -- Open/close helpers -------------------------------------------------- */
|
|
|
|
static IOFileUD *io_tofilep(lua_State *L)
|
|
{
|
|
if (!(L->base < L->top && tvisudata(L->base) &&
|
|
udataV(L->base)->udtype == UDTYPE_IO_FILE))
|
|
lj_err_argtype(L, 1, "FILE*");
|
|
return (IOFileUD *)uddata(udataV(L->base));
|
|
}
|
|
|
|
static IOFileUD *io_tofile(lua_State *L)
|
|
{
|
|
IOFileUD *iof = io_tofilep(L);
|
|
if (iof->fp == NULL)
|
|
lj_err_caller(L, LJ_ERR_IOCLFL);
|
|
return iof;
|
|
}
|
|
|
|
static FILE *io_stdfile(lua_State *L, ptrdiff_t id)
|
|
{
|
|
IOFileUD *iof = IOSTDF_IOF(L, id);
|
|
if (iof->fp == NULL)
|
|
lj_err_caller(L, LJ_ERR_IOSTDCL);
|
|
return iof->fp;
|
|
}
|
|
|
|
static IOFileUD *io_file_new(lua_State *L)
|
|
{
|
|
IOFileUD *iof = (IOFileUD *)lua_newuserdata(L, sizeof(IOFileUD));
|
|
GCudata *ud = udataV(L->top-1);
|
|
ud->udtype = UDTYPE_IO_FILE;
|
|
/* NOBARRIER: The GCudata is new (marked white). */
|
|
setgcrefr(ud->metatable, curr_func(L)->c.env);
|
|
iof->fp = NULL;
|
|
iof->type = IOFILE_TYPE_FILE;
|
|
return iof;
|
|
}
|
|
|
|
static IOFileUD *io_file_open(lua_State *L, const char *mode)
|
|
{
|
|
const char *fname = strdata(lj_lib_checkstr(L, 1));
|
|
IOFileUD *iof = io_file_new(L);
|
|
#if LJ_TARGET_WINDOWS
|
|
wchar_t wfname[MAX_PATH];
|
|
wchar_t wmode[MAX_PATH];
|
|
if (!widen(fname, wfname) || !widen(mode, wmode))
|
|
luaL_argerror(L, 1, lj_strfmt_pushf(L, "%s: failed to convert path to utf-16", fname));
|
|
iof->fp = _wfopen(wfname, wmode);
|
|
#else
|
|
iof->fp = fopen(fname, mode);
|
|
#endif
|
|
if (iof->fp == NULL)
|
|
luaL_argerror(L, 1, lj_strfmt_pushf(L, "%s: %s", fname, strerror(errno)));
|
|
return iof;
|
|
}
|
|
|
|
static int io_file_close(lua_State *L, IOFileUD *iof)
|
|
{
|
|
int ok;
|
|
if ((iof->type & IOFILE_TYPE_MASK) == IOFILE_TYPE_FILE) {
|
|
ok = (fclose(iof->fp) == 0);
|
|
} else if ((iof->type & IOFILE_TYPE_MASK) == IOFILE_TYPE_PIPE) {
|
|
int stat = -1;
|
|
#if LJ_TARGET_POSIX
|
|
stat = pclose(iof->fp);
|
|
#elif LJ_TARGET_WINDOWS && !LJ_TARGET_XBOXONE
|
|
stat = _pclose(iof->fp);
|
|
#else
|
|
lua_assert(0);
|
|
return 0;
|
|
#endif
|
|
#if LJ_52
|
|
iof->fp = NULL;
|
|
return luaL_execresult(L, stat);
|
|
#else
|
|
ok = (stat != -1);
|
|
#endif
|
|
} else {
|
|
lua_assert((iof->type & IOFILE_TYPE_MASK) == IOFILE_TYPE_STDF);
|
|
setnilV(L->top++);
|
|
lua_pushliteral(L, "cannot close standard file");
|
|
return 2;
|
|
}
|
|
iof->fp = NULL;
|
|
return luaL_fileresult(L, ok, NULL);
|
|
}
|
|
|
|
/* -- Read/write helpers -------------------------------------------------- */
|
|
|
|
static int io_file_readnum(lua_State *L, FILE *fp)
|
|
{
|
|
lua_Number d;
|
|
if (fscanf(fp, LUA_NUMBER_SCAN, &d) == 1) {
|
|
if (LJ_DUALNUM) {
|
|
int32_t i = lj_num2int(d);
|
|
if (d == (lua_Number)i && !tvismzero((cTValue *)&d)) {
|
|
setintV(L->top++, i);
|
|
return 1;
|
|
}
|
|
}
|
|
setnumV(L->top++, d);
|
|
return 1;
|
|
} else {
|
|
setnilV(L->top++);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
static int io_file_readline(lua_State *L, FILE *fp, MSize chop)
|
|
{
|
|
MSize m = LUAL_BUFFERSIZE, n = 0, ok = 0;
|
|
char *buf;
|
|
for (;;) {
|
|
buf = lj_buf_tmp(L, m);
|
|
if (fgets(buf+n, m-n, fp) == NULL) break;
|
|
n += (MSize)strlen(buf+n);
|
|
ok |= n;
|
|
if (n && buf[n-1] == '\n') { n -= chop; break; }
|
|
if (n >= m - 64) m += m;
|
|
}
|
|
setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n));
|
|
lj_gc_check(L);
|
|
return (int)ok;
|
|
}
|
|
|
|
static void io_file_readall(lua_State *L, FILE *fp)
|
|
{
|
|
MSize m, n;
|
|
for (m = LUAL_BUFFERSIZE, n = 0; ; m += m) {
|
|
char *buf = lj_buf_tmp(L, m);
|
|
n += (MSize)fread(buf+n, 1, m-n, fp);
|
|
if (n != m) {
|
|
setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n));
|
|
lj_gc_check(L);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int io_file_readlen(lua_State *L, FILE *fp, MSize m)
|
|
{
|
|
if (m) {
|
|
char *buf = lj_buf_tmp(L, m);
|
|
MSize n = (MSize)fread(buf, 1, m, fp);
|
|
setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n));
|
|
lj_gc_check(L);
|
|
return (n > 0 || m == 0);
|
|
} else {
|
|
int c = getc(fp);
|
|
ungetc(c, fp);
|
|
setstrV(L, L->top++, &G(L)->strempty);
|
|
return (c != EOF);
|
|
}
|
|
}
|
|
|
|
static int io_file_read(lua_State *L, FILE *fp, int start)
|
|
{
|
|
int ok, n, nargs = (int)(L->top - L->base) - start;
|
|
clearerr(fp);
|
|
if (nargs == 0) {
|
|
ok = io_file_readline(L, fp, 1);
|
|
n = start+1; /* Return 1 result. */
|
|
} else {
|
|
/* The results plus the buffers go on top of the args. */
|
|
luaL_checkstack(L, nargs+LUA_MINSTACK, "too many arguments");
|
|
ok = 1;
|
|
for (n = start; nargs-- && ok; n++) {
|
|
if (tvisstr(L->base+n)) {
|
|
const char *p = strVdata(L->base+n);
|
|
if (p[0] == '*') p++;
|
|
if (p[0] == 'n')
|
|
ok = io_file_readnum(L, fp);
|
|
else if ((p[0] & ~0x20) == 'L')
|
|
ok = io_file_readline(L, fp, (p[0] == 'l'));
|
|
else if (p[0] == 'a')
|
|
io_file_readall(L, fp);
|
|
else
|
|
lj_err_arg(L, n+1, LJ_ERR_INVFMT);
|
|
} else if (tvisnumber(L->base+n)) {
|
|
ok = io_file_readlen(L, fp, (MSize)lj_lib_checkint(L, n+1));
|
|
} else {
|
|
lj_err_arg(L, n+1, LJ_ERR_INVOPT);
|
|
}
|
|
}
|
|
}
|
|
if (ferror(fp))
|
|
return luaL_fileresult(L, 0, NULL);
|
|
if (!ok)
|
|
setnilV(L->top-1); /* Replace last result with nil. */
|
|
return n - start;
|
|
}
|
|
|
|
static int io_file_write(lua_State *L, FILE *fp, int start)
|
|
{
|
|
cTValue *tv;
|
|
int status = 1;
|
|
for (tv = L->base+start; tv < L->top; tv++) {
|
|
MSize len;
|
|
const char *p = lj_strfmt_wstrnum(L, tv, &len);
|
|
if (!p)
|
|
lj_err_argt(L, (int)(tv - L->base) + 1, LUA_TSTRING);
|
|
status = status && (fwrite(p, 1, len, fp) == len);
|
|
}
|
|
if (LJ_52 && status) {
|
|
L->top = L->base+1;
|
|
if (start == 0)
|
|
setudataV(L, L->base, IOSTDF_UD(L, GCROOT_IO_OUTPUT));
|
|
return 1;
|
|
}
|
|
return luaL_fileresult(L, status, NULL);
|
|
}
|
|
|
|
static int io_file_iter(lua_State *L)
|
|
{
|
|
GCfunc *fn = curr_func(L);
|
|
IOFileUD *iof = uddata(udataV(&fn->c.upvalue[0]));
|
|
int n = fn->c.nupvalues - 1;
|
|
if (iof->fp == NULL)
|
|
lj_err_caller(L, LJ_ERR_IOCLFL);
|
|
L->top = L->base;
|
|
if (n) { /* Copy upvalues with options to stack. */
|
|
if (n > LUAI_MAXCSTACK)
|
|
lj_err_caller(L, LJ_ERR_STKOV);
|
|
lj_state_checkstack(L, (MSize)n);
|
|
memcpy(L->top, &fn->c.upvalue[1], n*sizeof(TValue));
|
|
L->top += n;
|
|
}
|
|
n = io_file_read(L, iof->fp, 0);
|
|
if (ferror(iof->fp))
|
|
lj_err_callermsg(L, strVdata(L->top-2));
|
|
if (tvisnil(L->base) && (iof->type & IOFILE_FLAG_CLOSE)) {
|
|
io_file_close(L, iof); /* Return values are ignored. */
|
|
return 0;
|
|
}
|
|
return n;
|
|
}
|
|
|
|
static int io_file_lines(lua_State *L)
|
|
{
|
|
int n = (int)(L->top - L->base);
|
|
if (n > LJ_MAX_UPVAL)
|
|
lj_err_caller(L, LJ_ERR_UNPACK);
|
|
lua_pushcclosure(L, io_file_iter, n);
|
|
return 1;
|
|
}
|
|
|
|
/* -- I/O file methods ---------------------------------------------------- */
|
|
|
|
#define LJLIB_MODULE_io_method
|
|
|
|
LJLIB_CF(io_method_close)
|
|
{
|
|
IOFileUD *iof = L->base < L->top ? io_tofile(L) :
|
|
IOSTDF_IOF(L, GCROOT_IO_OUTPUT);
|
|
return io_file_close(L, iof);
|
|
}
|
|
|
|
LJLIB_CF(io_method_read)
|
|
{
|
|
return io_file_read(L, io_tofile(L)->fp, 1);
|
|
}
|
|
|
|
LJLIB_CF(io_method_write) LJLIB_REC(io_write 0)
|
|
{
|
|
return io_file_write(L, io_tofile(L)->fp, 1);
|
|
}
|
|
|
|
LJLIB_CF(io_method_flush) LJLIB_REC(io_flush 0)
|
|
{
|
|
return luaL_fileresult(L, fflush(io_tofile(L)->fp) == 0, NULL);
|
|
}
|
|
|
|
LJLIB_CF(io_method_seek)
|
|
{
|
|
FILE *fp = io_tofile(L)->fp;
|
|
int opt = lj_lib_checkopt(L, 2, 1, "\3set\3cur\3end");
|
|
int64_t ofs = 0;
|
|
cTValue *o;
|
|
int res;
|
|
if (opt == 0) opt = SEEK_SET;
|
|
else if (opt == 1) opt = SEEK_CUR;
|
|
else if (opt == 2) opt = SEEK_END;
|
|
o = L->base+2;
|
|
if (o < L->top) {
|
|
if (tvisint(o))
|
|
ofs = (int64_t)intV(o);
|
|
else if (tvisnum(o))
|
|
ofs = (int64_t)numV(o);
|
|
else if (!tvisnil(o))
|
|
lj_err_argt(L, 3, LUA_TNUMBER);
|
|
}
|
|
#if LJ_TARGET_POSIX
|
|
res = fseeko(fp, ofs, opt);
|
|
#elif _MSC_VER >= 1400
|
|
res = _fseeki64(fp, ofs, opt);
|
|
#elif defined(__MINGW32__)
|
|
res = fseeko64(fp, ofs, opt);
|
|
#else
|
|
res = fseek(fp, (long)ofs, opt);
|
|
#endif
|
|
if (res)
|
|
return luaL_fileresult(L, 0, NULL);
|
|
#if LJ_TARGET_POSIX
|
|
ofs = ftello(fp);
|
|
#elif _MSC_VER >= 1400
|
|
ofs = _ftelli64(fp);
|
|
#elif defined(__MINGW32__)
|
|
ofs = ftello64(fp);
|
|
#else
|
|
ofs = (int64_t)ftell(fp);
|
|
#endif
|
|
setint64V(L->top-1, ofs);
|
|
return 1;
|
|
}
|
|
|
|
LJLIB_CF(io_method_setvbuf)
|
|
{
|
|
FILE *fp = io_tofile(L)->fp;
|
|
int opt = lj_lib_checkopt(L, 2, -1, "\4full\4line\2no");
|
|
size_t sz = (size_t)lj_lib_optint(L, 3, LUAL_BUFFERSIZE);
|
|
if (opt == 0) opt = _IOFBF;
|
|
else if (opt == 1) opt = _IOLBF;
|
|
else if (opt == 2) opt = _IONBF;
|
|
return luaL_fileresult(L, setvbuf(fp, NULL, opt, sz) == 0, NULL);
|
|
}
|
|
|
|
LJLIB_CF(io_method_lines)
|
|
{
|
|
io_tofile(L);
|
|
return io_file_lines(L);
|
|
}
|
|
|
|
LJLIB_CF(io_method___gc)
|
|
{
|
|
IOFileUD *iof = io_tofilep(L);
|
|
if (iof->fp != NULL && (iof->type & IOFILE_TYPE_MASK) != IOFILE_TYPE_STDF)
|
|
io_file_close(L, iof);
|
|
return 0;
|
|
}
|
|
|
|
LJLIB_CF(io_method___tostring)
|
|
{
|
|
IOFileUD *iof = io_tofilep(L);
|
|
if (iof->fp != NULL)
|
|
lua_pushfstring(L, "file (%p)", iof->fp);
|
|
else
|
|
lua_pushliteral(L, "file (closed)");
|
|
return 1;
|
|
}
|
|
|
|
LJLIB_PUSH(top-1) LJLIB_SET(__index)
|
|
|
|
#include "lj_libdef.h"
|
|
|
|
/* -- I/O library functions ----------------------------------------------- */
|
|
|
|
#define LJLIB_MODULE_io
|
|
|
|
LJLIB_PUSH(top-2) LJLIB_SET(!) /* Set environment. */
|
|
|
|
LJLIB_CF(io_open)
|
|
{
|
|
const char *fname = strdata(lj_lib_checkstr(L, 1));
|
|
GCstr *s = lj_lib_optstr(L, 2);
|
|
const char *mode = s ? strdata(s) : "r";
|
|
IOFileUD *iof = io_file_new(L);
|
|
#if LJ_TARGET_WINDOWS
|
|
wchar_t wfname[MAX_PATH];
|
|
wchar_t wmode[MAX_PATH];
|
|
if (widen(fname, wfname) && widen(mode, wmode))
|
|
iof->fp = _wfopen(wfname, wmode);
|
|
#else
|
|
iof->fp = fopen(fname, mode);
|
|
#endif
|
|
return iof->fp != NULL ? 1 : luaL_fileresult(L, 0, fname);
|
|
}
|
|
|
|
LJLIB_CF(io_popen)
|
|
{
|
|
#if LJ_TARGET_POSIX || (LJ_TARGET_WINDOWS && !LJ_TARGET_XBOXONE)
|
|
const char *fname = strdata(lj_lib_checkstr(L, 1));
|
|
GCstr *s = lj_lib_optstr(L, 2);
|
|
const char *mode = s ? strdata(s) : "r";
|
|
IOFileUD *iof = io_file_new(L);
|
|
iof->type = IOFILE_TYPE_PIPE;
|
|
#if LJ_TARGET_POSIX
|
|
fflush(NULL);
|
|
iof->fp = popen(fname, mode);
|
|
#else
|
|
wchar_t wfname[MAX_PATH];
|
|
wchar_t wmode[MAX_PATH];
|
|
if (widen(fname, wfname) && widen(mode, wmode))
|
|
iof->fp = _wpopen(wfname, wmode);
|
|
#endif
|
|
return iof->fp != NULL ? 1 : luaL_fileresult(L, 0, fname);
|
|
#else
|
|
return luaL_error(L, LUA_QL("popen") " not supported");
|
|
#endif
|
|
}
|
|
|
|
LJLIB_CF(io_tmpfile)
|
|
{
|
|
IOFileUD *iof = io_file_new(L);
|
|
#if LJ_TARGET_PS3 || LJ_TARGET_PS4 || LJ_TARGET_PSVITA
|
|
iof->fp = NULL; errno = ENOSYS;
|
|
#else
|
|
iof->fp = tmpfile();
|
|
#endif
|
|
return iof->fp != NULL ? 1 : luaL_fileresult(L, 0, NULL);
|
|
}
|
|
|
|
LJLIB_CF(io_close)
|
|
{
|
|
return lj_cf_io_method_close(L);
|
|
}
|
|
|
|
LJLIB_CF(io_read)
|
|
{
|
|
return io_file_read(L, io_stdfile(L, GCROOT_IO_INPUT), 0);
|
|
}
|
|
|
|
LJLIB_CF(io_write) LJLIB_REC(io_write GCROOT_IO_OUTPUT)
|
|
{
|
|
return io_file_write(L, io_stdfile(L, GCROOT_IO_OUTPUT), 0);
|
|
}
|
|
|
|
LJLIB_CF(io_flush) LJLIB_REC(io_flush GCROOT_IO_OUTPUT)
|
|
{
|
|
return luaL_fileresult(L, fflush(io_stdfile(L, GCROOT_IO_OUTPUT)) == 0, NULL);
|
|
}
|
|
|
|
static int io_std_getset(lua_State *L, ptrdiff_t id, const char *mode)
|
|
{
|
|
if (L->base < L->top && !tvisnil(L->base)) {
|
|
if (tvisudata(L->base)) {
|
|
io_tofile(L);
|
|
L->top = L->base+1;
|
|
} else {
|
|
io_file_open(L, mode);
|
|
}
|
|
/* NOBARRIER: The standard I/O handles are GC roots. */
|
|
setgcref(G(L)->gcroot[id], gcV(L->top-1));
|
|
} else {
|
|
setudataV(L, L->top++, IOSTDF_UD(L, id));
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
LJLIB_CF(io_input)
|
|
{
|
|
return io_std_getset(L, GCROOT_IO_INPUT, "r");
|
|
}
|
|
|
|
LJLIB_CF(io_output)
|
|
{
|
|
return io_std_getset(L, GCROOT_IO_OUTPUT, "w");
|
|
}
|
|
|
|
LJLIB_CF(io_lines)
|
|
{
|
|
if (L->base == L->top) setnilV(L->top++);
|
|
if (!tvisnil(L->base)) { /* io.lines(fname) */
|
|
IOFileUD *iof = io_file_open(L, "r");
|
|
iof->type = IOFILE_TYPE_FILE|IOFILE_FLAG_CLOSE;
|
|
L->top--;
|
|
setudataV(L, L->base, udataV(L->top));
|
|
} else { /* io.lines() iterates over stdin. */
|
|
setudataV(L, L->base, IOSTDF_UD(L, GCROOT_IO_INPUT));
|
|
}
|
|
return io_file_lines(L);
|
|
}
|
|
|
|
LJLIB_CF(io_type)
|
|
{
|
|
cTValue *o = lj_lib_checkany(L, 1);
|
|
if (!(tvisudata(o) && udataV(o)->udtype == UDTYPE_IO_FILE))
|
|
setnilV(L->top++);
|
|
else if (((IOFileUD *)uddata(udataV(o)))->fp != NULL)
|
|
lua_pushliteral(L, "file");
|
|
else
|
|
lua_pushliteral(L, "closed file");
|
|
return 1;
|
|
}
|
|
|
|
#include "lj_libdef.h"
|
|
|
|
/* ------------------------------------------------------------------------ */
|
|
|
|
static GCobj *io_std_new(lua_State *L, FILE *fp, const char *name)
|
|
{
|
|
IOFileUD *iof = (IOFileUD *)lua_newuserdata(L, sizeof(IOFileUD));
|
|
GCudata *ud = udataV(L->top-1);
|
|
ud->udtype = UDTYPE_IO_FILE;
|
|
/* NOBARRIER: The GCudata is new (marked white). */
|
|
setgcref(ud->metatable, gcV(L->top-3));
|
|
iof->fp = fp;
|
|
iof->type = IOFILE_TYPE_STDF;
|
|
lua_setfield(L, -2, name);
|
|
return obj2gco(ud);
|
|
}
|
|
|
|
LUALIB_API int luaopen_io(lua_State *L)
|
|
{
|
|
LJ_LIB_REG(L, NULL, io_method);
|
|
copyTV(L, L->top, L->top-1); L->top++;
|
|
lua_setfield(L, LUA_REGISTRYINDEX, LUA_FILEHANDLE);
|
|
LJ_LIB_REG(L, LUA_IOLIBNAME, io);
|
|
setgcref(G(L)->gcroot[GCROOT_IO_INPUT], io_std_new(L, stdin, "stdin"));
|
|
setgcref(G(L)->gcroot[GCROOT_IO_OUTPUT], io_std_new(L, stdout, "stdout"));
|
|
io_std_new(L, stderr, "stderr");
|
|
return 1;
|
|
}
|
|
|