3 ** Copyright (C) 2005-2017 Mike Pall. See Copyright Notice in luajit.h
5 ** Major portions taken verbatim or adapted from the Lua interpreter.
6 ** Copyright (C) 1994-2011 Lua.org, PUC-Rio. See Copyright Notice in lua.h
27 /* Userdata payload for I/O file. */
28 typedef struct IOFileUD
{
29 FILE *fp
; /* File handle. */
30 uint32_t type
; /* File type. */
33 #define IOFILE_TYPE_FILE 0 /* Regular file. */
34 #define IOFILE_TYPE_PIPE 1 /* Pipe. */
35 #define IOFILE_TYPE_STDF 2 /* Standard file handle. */
36 #define IOFILE_TYPE_MASK 3
38 #define IOFILE_FLAG_CLOSE 4 /* Close after io.lines() iterator. */
40 #define IOSTDF_UD(L, id) (&gcref(G(L)->gcroot[(id)])->ud)
41 #define IOSTDF_IOF(L, id) ((IOFileUD *)uddata(IOSTDF_UD(L, (id))))
43 /* -- Open/close helpers -------------------------------------------------- */
45 static IOFileUD
*io_tofilep(lua_State
*L
)
47 if (!(L
->base
< L
->top
&& tvisudata(L
->base
) &&
48 udataV(L
->base
)->udtype
== UDTYPE_IO_FILE
))
49 lj_err_argtype(L
, 1, "FILE*");
50 return (IOFileUD
*)uddata(udataV(L
->base
));
53 static IOFileUD
*io_tofile(lua_State
*L
)
55 IOFileUD
*iof
= io_tofilep(L
);
57 lj_err_caller(L
, LJ_ERR_IOCLFL
);
61 static FILE *io_stdfile(lua_State
*L
, ptrdiff_t id
)
63 IOFileUD
*iof
= IOSTDF_IOF(L
, id
);
65 lj_err_caller(L
, LJ_ERR_IOSTDCL
);
69 static IOFileUD
*io_file_new(lua_State
*L
)
71 IOFileUD
*iof
= (IOFileUD
*)lua_newuserdata(L
, sizeof(IOFileUD
));
72 GCudata
*ud
= udataV(L
->top
-1);
73 ud
->udtype
= UDTYPE_IO_FILE
;
74 /* NOBARRIER: The GCudata is new (marked white). */
75 setgcrefr(ud
->metatable
, curr_func(L
)->c
.env
);
77 iof
->type
= IOFILE_TYPE_FILE
;
81 static IOFileUD
*io_file_open(lua_State
*L
, const char *mode
)
83 const char *fname
= strdata(lj_lib_checkstr(L
, 1));
84 IOFileUD
*iof
= io_file_new(L
);
85 iof
->fp
= fopen(fname
, mode
);
87 luaL_argerror(L
, 1, lj_str_pushf(L
, "%s: %s", fname
, strerror(errno
)));
91 static int io_file_close(lua_State
*L
, IOFileUD
*iof
)
94 if ((iof
->type
& IOFILE_TYPE_MASK
) == IOFILE_TYPE_FILE
) {
95 ok
= (fclose(iof
->fp
) == 0);
96 } else if ((iof
->type
& IOFILE_TYPE_MASK
) == IOFILE_TYPE_PIPE
) {
99 stat
= pclose(iof
->fp
);
100 #elif LJ_TARGET_WINDOWS
101 stat
= _pclose(iof
->fp
);
108 return luaL_execresult(L
, stat
);
113 lua_assert((iof
->type
& IOFILE_TYPE_MASK
) == IOFILE_TYPE_STDF
);
115 lua_pushliteral(L
, "cannot close standard file");
119 return luaL_fileresult(L
, ok
, NULL
);
122 /* -- Read/write helpers -------------------------------------------------- */
124 static int io_file_readnum(lua_State
*L
, FILE *fp
)
127 if (fscanf(fp
, LUA_NUMBER_SCAN
, &d
) == 1) {
129 int32_t i
= lj_num2int(d
);
130 if (d
== (lua_Number
)i
&& !tvismzero((cTValue
*)&d
)) {
131 setintV(L
->top
++, i
);
135 setnumV(L
->top
++, d
);
143 static int io_file_readline(lua_State
*L
, FILE *fp
, MSize chop
)
145 MSize m
= LUAL_BUFFERSIZE
, n
= 0, ok
= 0;
148 buf
= lj_str_needbuf(L
, &G(L
)->tmpbuf
, m
);
149 if (fgets(buf
+n
, m
-n
, fp
) == NULL
) break;
150 n
+= (MSize
)strlen(buf
+n
);
152 if (n
&& buf
[n
-1] == '\n') { n
-= chop
; break; }
153 if (n
>= m
- 64) m
+= m
;
155 setstrV(L
, L
->top
++, lj_str_new(L
, buf
, (size_t)n
));
160 static void io_file_readall(lua_State
*L
, FILE *fp
)
163 for (m
= LUAL_BUFFERSIZE
, n
= 0; ; m
+= m
) {
164 char *buf
= lj_str_needbuf(L
, &G(L
)->tmpbuf
, m
);
165 n
+= (MSize
)fread(buf
+n
, 1, m
-n
, fp
);
167 setstrV(L
, L
->top
++, lj_str_new(L
, buf
, (size_t)n
));
174 static int io_file_readlen(lua_State
*L
, FILE *fp
, MSize m
)
177 char *buf
= lj_str_needbuf(L
, &G(L
)->tmpbuf
, m
);
178 MSize n
= (MSize
)fread(buf
, 1, m
, fp
);
179 setstrV(L
, L
->top
++, lj_str_new(L
, buf
, (size_t)n
));
181 return (n
> 0 || m
== 0);
185 setstrV(L
, L
->top
++, &G(L
)->strempty
);
190 static int io_file_read(lua_State
*L
, FILE *fp
, int start
)
192 int ok
, n
, nargs
= (int)(L
->top
- L
->base
) - start
;
195 ok
= io_file_readline(L
, fp
, 1);
196 n
= start
+1; /* Return 1 result. */
198 /* The results plus the buffers go on top of the args. */
199 luaL_checkstack(L
, nargs
+LUA_MINSTACK
, "too many arguments");
201 for (n
= start
; nargs
-- && ok
; n
++) {
202 if (tvisstr(L
->base
+n
)) {
203 const char *p
= strVdata(L
->base
+n
);
205 lj_err_arg(L
, n
+1, LJ_ERR_INVOPT
);
207 ok
= io_file_readnum(L
, fp
);
208 else if ((p
[1] & ~0x20) == 'L')
209 ok
= io_file_readline(L
, fp
, (p
[1] == 'l'));
210 else if (p
[1] == 'a')
211 io_file_readall(L
, fp
);
213 lj_err_arg(L
, n
+1, LJ_ERR_INVFMT
);
214 } else if (tvisnumber(L
->base
+n
)) {
215 ok
= io_file_readlen(L
, fp
, (MSize
)lj_lib_checkint(L
, n
+1));
217 lj_err_arg(L
, n
+1, LJ_ERR_INVOPT
);
222 return luaL_fileresult(L
, 0, NULL
);
224 setnilV(L
->top
-1); /* Replace last result with nil. */
228 static int io_file_write(lua_State
*L
, FILE *fp
, int start
)
232 for (tv
= L
->base
+start
; tv
< L
->top
; tv
++) {
234 MSize len
= strV(tv
)->len
;
235 status
= status
&& (fwrite(strVdata(tv
), 1, len
, fp
) == len
);
236 } else if (tvisint(tv
)) {
237 char buf
[LJ_STR_INTBUF
];
238 char *p
= lj_str_bufint(buf
, intV(tv
));
239 size_t len
= (size_t)(buf
+LJ_STR_INTBUF
-p
);
240 status
= status
&& (fwrite(p
, 1, len
, fp
) == len
);
241 } else if (tvisnum(tv
)) {
242 status
= status
&& (fprintf(fp
, LUA_NUMBER_FMT
, numV(tv
)) > 0);
244 lj_err_argt(L
, (int)(tv
- L
->base
) + 1, LUA_TSTRING
);
247 if (LJ_52
&& status
) {
250 setudataV(L
, L
->base
, IOSTDF_UD(L
, GCROOT_IO_OUTPUT
));
253 return luaL_fileresult(L
, status
, NULL
);
256 static int io_file_iter(lua_State
*L
)
258 GCfunc
*fn
= curr_func(L
);
259 IOFileUD
*iof
= uddata(udataV(&fn
->c
.upvalue
[0]));
260 int n
= fn
->c
.nupvalues
- 1;
262 lj_err_caller(L
, LJ_ERR_IOCLFL
);
264 if (n
) { /* Copy upvalues with options to stack. */
265 if (n
> LUAI_MAXCSTACK
)
266 lj_err_caller(L
, LJ_ERR_STKOV
);
267 lj_state_checkstack(L
, (MSize
)n
);
268 memcpy(L
->top
, &fn
->c
.upvalue
[1], n
*sizeof(TValue
));
271 n
= io_file_read(L
, iof
->fp
, 0);
273 lj_err_callermsg(L
, strVdata(L
->top
-2));
274 if (tvisnil(L
->base
) && (iof
->type
& IOFILE_FLAG_CLOSE
)) {
275 io_file_close(L
, iof
); /* Return values are ignored. */
281 static int io_file_lines(lua_State
*L
)
283 int n
= (int)(L
->top
- L
->base
);
284 if (n
> LJ_MAX_UPVAL
)
285 lj_err_caller(L
, LJ_ERR_UNPACK
);
286 lua_pushcclosure(L
, io_file_iter
, n
);
290 /* -- I/O file methods ---------------------------------------------------- */
292 #define LJLIB_MODULE_io_method
294 LJLIB_CF(io_method_close
)
296 IOFileUD
*iof
= L
->base
< L
->top
? io_tofile(L
) :
297 IOSTDF_IOF(L
, GCROOT_IO_OUTPUT
);
298 return io_file_close(L
, iof
);
301 LJLIB_CF(io_method_read
)
303 return io_file_read(L
, io_tofile(L
)->fp
, 1);
306 LJLIB_CF(io_method_write
) LJLIB_REC(io_write
0)
308 return io_file_write(L
, io_tofile(L
)->fp
, 1);
311 LJLIB_CF(io_method_flush
) LJLIB_REC(io_flush
0)
313 return luaL_fileresult(L
, fflush(io_tofile(L
)->fp
) == 0, NULL
);
316 LJLIB_CF(io_method_seek
)
318 FILE *fp
= io_tofile(L
)->fp
;
319 int opt
= lj_lib_checkopt(L
, 2, 1, "\3set\3cur\3end");
323 if (opt
== 0) opt
= SEEK_SET
;
324 else if (opt
== 1) opt
= SEEK_CUR
;
325 else if (opt
== 2) opt
= SEEK_END
;
329 ofs
= (int64_t)intV(o
);
331 ofs
= (int64_t)numV(o
);
332 else if (!tvisnil(o
))
333 lj_err_argt(L
, 3, LUA_TNUMBER
);
336 res
= fseeko(fp
, ofs
, opt
);
337 #elif _MSC_VER >= 1400
338 res
= _fseeki64(fp
, ofs
, opt
);
339 #elif defined(__MINGW32__)
340 res
= fseeko64(fp
, ofs
, opt
);
342 res
= fseek(fp
, (long)ofs
, opt
);
345 return luaL_fileresult(L
, 0, NULL
);
348 #elif _MSC_VER >= 1400
350 #elif defined(__MINGW32__)
353 ofs
= (int64_t)ftell(fp
);
355 setint64V(L
->top
-1, ofs
);
359 LJLIB_CF(io_method_setvbuf
)
361 FILE *fp
= io_tofile(L
)->fp
;
362 int opt
= lj_lib_checkopt(L
, 2, -1, "\4full\4line\2no");
363 size_t sz
= (size_t)lj_lib_optint(L
, 3, LUAL_BUFFERSIZE
);
364 if (opt
== 0) opt
= _IOFBF
;
365 else if (opt
== 1) opt
= _IOLBF
;
366 else if (opt
== 2) opt
= _IONBF
;
367 return luaL_fileresult(L
, setvbuf(fp
, NULL
, opt
, sz
) == 0, NULL
);
370 LJLIB_CF(io_method_lines
)
373 return io_file_lines(L
);
376 LJLIB_CF(io_method___gc
)
378 IOFileUD
*iof
= io_tofilep(L
);
379 if (iof
->fp
!= NULL
&& (iof
->type
& IOFILE_TYPE_MASK
) != IOFILE_TYPE_STDF
)
380 io_file_close(L
, iof
);
384 LJLIB_CF(io_method___tostring
)
386 IOFileUD
*iof
= io_tofilep(L
);
388 lua_pushfstring(L
, "file (%p)", iof
->fp
);
390 lua_pushliteral(L
, "file (closed)");
394 LJLIB_PUSH(top
-1) LJLIB_SET(__index
)
396 #include "lj_libdef.h"
398 /* -- I/O library functions ----------------------------------------------- */
400 #define LJLIB_MODULE_io
402 LJLIB_PUSH(top
-2) LJLIB_SET(!) /* Set environment. */
406 const char *fname
= strdata(lj_lib_checkstr(L
, 1));
407 GCstr
*s
= lj_lib_optstr(L
, 2);
408 const char *mode
= s
? strdata(s
) : "r";
409 IOFileUD
*iof
= io_file_new(L
);
410 iof
->fp
= fopen(fname
, mode
);
411 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, fname
);
416 #if LJ_TARGET_POSIX || LJ_TARGET_WINDOWS
417 const char *fname
= strdata(lj_lib_checkstr(L
, 1));
418 GCstr
*s
= lj_lib_optstr(L
, 2);
419 const char *mode
= s
? strdata(s
) : "r";
420 IOFileUD
*iof
= io_file_new(L
);
421 iof
->type
= IOFILE_TYPE_PIPE
;
424 iof
->fp
= popen(fname
, mode
);
426 iof
->fp
= _popen(fname
, mode
);
428 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, fname
);
430 return luaL_error(L
, LUA_QL("popen") " not supported");
436 IOFileUD
*iof
= io_file_new(L
);
437 #if LJ_TARGET_PS3 || LJ_TARGET_PS4 || LJ_TARGET_PSVITA
438 iof
->fp
= NULL
; errno
= ENOSYS
;
442 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, NULL
);
447 return lj_cf_io_method_close(L
);
452 return io_file_read(L
, io_stdfile(L
, GCROOT_IO_INPUT
), 0);
455 LJLIB_CF(io_write
) LJLIB_REC(io_write GCROOT_IO_OUTPUT
)
457 return io_file_write(L
, io_stdfile(L
, GCROOT_IO_OUTPUT
), 0);
460 LJLIB_CF(io_flush
) LJLIB_REC(io_flush GCROOT_IO_OUTPUT
)
462 return luaL_fileresult(L
, fflush(io_stdfile(L
, GCROOT_IO_OUTPUT
)) == 0, NULL
);
465 static int io_std_getset(lua_State
*L
, ptrdiff_t id
, const char *mode
)
467 if (L
->base
< L
->top
&& !tvisnil(L
->base
)) {
468 if (tvisudata(L
->base
)) {
472 io_file_open(L
, mode
);
474 /* NOBARRIER: The standard I/O handles are GC roots. */
475 setgcref(G(L
)->gcroot
[id
], gcV(L
->top
-1));
477 setudataV(L
, L
->top
++, IOSTDF_UD(L
, id
));
484 return io_std_getset(L
, GCROOT_IO_INPUT
, "r");
489 return io_std_getset(L
, GCROOT_IO_OUTPUT
, "w");
494 if (L
->base
== L
->top
) setnilV(L
->top
++);
495 if (!tvisnil(L
->base
)) { /* io.lines(fname) */
496 IOFileUD
*iof
= io_file_open(L
, "r");
497 iof
->type
= IOFILE_TYPE_FILE
|IOFILE_FLAG_CLOSE
;
499 setudataV(L
, L
->base
, udataV(L
->top
));
500 } else { /* io.lines() iterates over stdin. */
501 setudataV(L
, L
->base
, IOSTDF_UD(L
, GCROOT_IO_INPUT
));
503 return io_file_lines(L
);
508 cTValue
*o
= lj_lib_checkany(L
, 1);
509 if (!(tvisudata(o
) && udataV(o
)->udtype
== UDTYPE_IO_FILE
))
511 else if (((IOFileUD
*)uddata(udataV(o
)))->fp
!= NULL
)
512 lua_pushliteral(L
, "file");
514 lua_pushliteral(L
, "closed file");
518 #include "lj_libdef.h"
520 /* ------------------------------------------------------------------------ */
522 static GCobj
*io_std_new(lua_State
*L
, FILE *fp
, const char *name
)
524 IOFileUD
*iof
= (IOFileUD
*)lua_newuserdata(L
, sizeof(IOFileUD
));
525 GCudata
*ud
= udataV(L
->top
-1);
526 ud
->udtype
= UDTYPE_IO_FILE
;
527 /* NOBARRIER: The GCudata is new (marked white). */
528 setgcref(ud
->metatable
, gcV(L
->top
-3));
530 iof
->type
= IOFILE_TYPE_STDF
;
531 lua_setfield(L
, -2, name
);
535 LUALIB_API
int luaopen_io(lua_State
*L
)
537 LJ_LIB_REG(L
, NULL
, io_method
);
538 copyTV(L
, L
->top
, L
->top
-1); L
->top
++;
539 lua_setfield(L
, LUA_REGISTRYINDEX
, LUA_FILEHANDLE
);
540 LJ_LIB_REG(L
, LUA_IOLIBNAME
, io
);
541 setgcref(G(L
)->gcroot
[GCROOT_IO_INPUT
], io_std_new(L
, stdin
, "stdin"));
542 setgcref(G(L
)->gcroot
[GCROOT_IO_OUTPUT
], io_std_new(L
, stdout
, "stdout"));
543 io_std_new(L
, stderr
, "stderr");