3 ** Copyright (C) 2005-2013 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 /* -- I/O file methods ---------------------------------------------------- */
283 #define LJLIB_MODULE_io_method
285 LJLIB_CF(io_method_close
)
287 IOFileUD
*iof
= L
->base
< L
->top
? io_tofile(L
) :
288 IOSTDF_IOF(L
, GCROOT_IO_OUTPUT
);
289 return io_file_close(L
, iof
);
292 LJLIB_CF(io_method_read
)
294 return io_file_read(L
, io_tofile(L
)->fp
, 1);
297 LJLIB_CF(io_method_write
) LJLIB_REC(io_write
0)
299 return io_file_write(L
, io_tofile(L
)->fp
, 1);
302 LJLIB_CF(io_method_flush
) LJLIB_REC(io_flush
0)
304 return luaL_fileresult(L
, fflush(io_tofile(L
)->fp
) == 0, NULL
);
307 LJLIB_CF(io_method_seek
)
309 FILE *fp
= io_tofile(L
)->fp
;
310 int opt
= lj_lib_checkopt(L
, 2, 1, "\3set\3cur\3end");
314 if (opt
== 0) opt
= SEEK_SET
;
315 else if (opt
== 1) opt
= SEEK_CUR
;
316 else if (opt
== 2) opt
= SEEK_END
;
320 ofs
= (int64_t)intV(o
);
322 ofs
= (int64_t)numV(o
);
323 else if (!tvisnil(o
))
324 lj_err_argt(L
, 3, LUA_TNUMBER
);
327 res
= fseeko(fp
, ofs
, opt
);
328 #elif _MSC_VER >= 1400
329 res
= _fseeki64(fp
, ofs
, opt
);
330 #elif defined(__MINGW32__)
331 res
= fseeko64(fp
, ofs
, opt
);
333 res
= fseek(fp
, (long)ofs
, opt
);
336 return luaL_fileresult(L
, 0, NULL
);
339 #elif _MSC_VER >= 1400
341 #elif defined(__MINGW32__)
344 ofs
= (int64_t)ftell(fp
);
346 setint64V(L
->top
-1, ofs
);
350 LJLIB_CF(io_method_setvbuf
)
352 FILE *fp
= io_tofile(L
)->fp
;
353 int opt
= lj_lib_checkopt(L
, 2, -1, "\4full\4line\2no");
354 size_t sz
= (size_t)lj_lib_optint(L
, 3, LUAL_BUFFERSIZE
);
355 if (opt
== 0) opt
= _IOFBF
;
356 else if (opt
== 1) opt
= _IOLBF
;
357 else if (opt
== 2) opt
= _IONBF
;
358 return luaL_fileresult(L
, setvbuf(fp
, NULL
, opt
, sz
) == 0, NULL
);
361 LJLIB_CF(io_method_lines
)
364 lua_pushcclosure(L
, io_file_iter
, (int)(L
->top
- L
->base
));
368 LJLIB_CF(io_method___gc
)
370 IOFileUD
*iof
= io_tofilep(L
);
371 if (iof
->fp
!= NULL
&& (iof
->type
& IOFILE_TYPE_MASK
) != IOFILE_TYPE_STDF
)
372 io_file_close(L
, iof
);
376 LJLIB_CF(io_method___tostring
)
378 IOFileUD
*iof
= io_tofilep(L
);
380 lua_pushfstring(L
, "file (%p)", iof
->fp
);
382 lua_pushliteral(L
, "file (closed)");
386 LJLIB_PUSH(top
-1) LJLIB_SET(__index
)
388 #include "lj_libdef.h"
390 /* -- I/O library functions ----------------------------------------------- */
392 #define LJLIB_MODULE_io
394 LJLIB_PUSH(top
-2) LJLIB_SET(!) /* Set environment. */
398 const char *fname
= strdata(lj_lib_checkstr(L
, 1));
399 GCstr
*s
= lj_lib_optstr(L
, 2);
400 const char *mode
= s
? strdata(s
) : "r";
401 IOFileUD
*iof
= io_file_new(L
);
402 iof
->fp
= fopen(fname
, mode
);
403 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, fname
);
408 #if LJ_TARGET_POSIX || LJ_TARGET_WINDOWS
409 const char *fname
= strdata(lj_lib_checkstr(L
, 1));
410 GCstr
*s
= lj_lib_optstr(L
, 2);
411 const char *mode
= s
? strdata(s
) : "r";
412 IOFileUD
*iof
= io_file_new(L
);
413 iof
->type
= IOFILE_TYPE_PIPE
;
416 iof
->fp
= popen(fname
, mode
);
418 iof
->fp
= _popen(fname
, mode
);
420 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, fname
);
422 return luaL_error(L
, LUA_QL("popen") " not supported");
428 IOFileUD
*iof
= io_file_new(L
);
430 iof
->fp
= NULL
; errno
= ENOSYS
;
434 return iof
->fp
!= NULL
? 1 : luaL_fileresult(L
, 0, NULL
);
439 return lj_cf_io_method_close(L
);
444 return io_file_read(L
, io_stdfile(L
, GCROOT_IO_INPUT
), 0);
447 LJLIB_CF(io_write
) LJLIB_REC(io_write GCROOT_IO_OUTPUT
)
449 return io_file_write(L
, io_stdfile(L
, GCROOT_IO_OUTPUT
), 0);
452 LJLIB_CF(io_flush
) LJLIB_REC(io_flush GCROOT_IO_OUTPUT
)
454 return luaL_fileresult(L
, fflush(io_stdfile(L
, GCROOT_IO_OUTPUT
)) == 0, NULL
);
457 static int io_std_getset(lua_State
*L
, ptrdiff_t id
, const char *mode
)
459 if (L
->base
< L
->top
&& !tvisnil(L
->base
)) {
460 if (tvisudata(L
->base
)) {
464 io_file_open(L
, mode
);
466 /* NOBARRIER: The standard I/O handles are GC roots. */
467 setgcref(G(L
)->gcroot
[id
], gcV(L
->top
-1));
469 setudataV(L
, L
->top
++, IOSTDF_UD(L
, id
));
476 return io_std_getset(L
, GCROOT_IO_INPUT
, "r");
481 return io_std_getset(L
, GCROOT_IO_OUTPUT
, "w");
486 if (L
->base
== L
->top
) setnilV(L
->top
++);
487 if (!tvisnil(L
->base
)) { /* io.lines(fname) */
488 IOFileUD
*iof
= io_file_open(L
, "r");
489 iof
->type
= IOFILE_TYPE_FILE
|IOFILE_FLAG_CLOSE
;
491 setudataV(L
, L
->base
, udataV(L
->top
));
492 } else { /* io.lines() iterates over stdin. */
493 setudataV(L
, L
->base
, IOSTDF_UD(L
, GCROOT_IO_INPUT
));
495 lua_pushcclosure(L
, io_file_iter
, (int)(L
->top
- L
->base
));
501 cTValue
*o
= lj_lib_checkany(L
, 1);
502 if (!(tvisudata(o
) && udataV(o
)->udtype
== UDTYPE_IO_FILE
))
504 else if (((IOFileUD
*)uddata(udataV(o
)))->fp
!= NULL
)
505 lua_pushliteral(L
, "file");
507 lua_pushliteral(L
, "closed file");
511 #include "lj_libdef.h"
513 /* ------------------------------------------------------------------------ */
515 static GCobj
*io_std_new(lua_State
*L
, FILE *fp
, const char *name
)
517 IOFileUD
*iof
= (IOFileUD
*)lua_newuserdata(L
, sizeof(IOFileUD
));
518 GCudata
*ud
= udataV(L
->top
-1);
519 ud
->udtype
= UDTYPE_IO_FILE
;
520 /* NOBARRIER: The GCudata is new (marked white). */
521 setgcref(ud
->metatable
, gcV(L
->top
-3));
523 iof
->type
= IOFILE_TYPE_STDF
;
524 lua_setfield(L
, -2, name
);
528 LUALIB_API
int luaopen_io(lua_State
*L
)
530 LJ_LIB_REG(L
, NULL
, io_method
);
531 copyTV(L
, L
->top
, L
->top
-1); L
->top
++;
532 lua_setfield(L
, LUA_REGISTRYINDEX
, LUA_FILEHANDLE
);
533 LJ_LIB_REG(L
, LUA_IOLIBNAME
, io
);
534 setgcref(G(L
)->gcroot
[GCROOT_IO_INPUT
], io_std_new(L
, stdin
, "stdin"));
535 setgcref(G(L
)->gcroot
[GCROOT_IO_OUTPUT
], io_std_new(L
, stdout
, "stdout"));
536 io_std_new(L
, stderr
, "stderr");