xref: /freebsd-src/contrib/lua/src/liolib.c (revision 0495ed398c4f64013bab2327eb13a303e1f90c13)
18e3e3a7aSWarner Losh /*
2*0495ed39SKyle Evans ** $Id: liolib.c $
38e3e3a7aSWarner Losh ** Standard I/O (and system) library
48e3e3a7aSWarner Losh ** See Copyright Notice in lua.h
58e3e3a7aSWarner Losh */
68e3e3a7aSWarner Losh 
78e3e3a7aSWarner Losh #define liolib_c
88e3e3a7aSWarner Losh #define LUA_LIB
98e3e3a7aSWarner Losh 
108e3e3a7aSWarner Losh #include "lprefix.h"
118e3e3a7aSWarner Losh 
128e3e3a7aSWarner Losh 
138e3e3a7aSWarner Losh #include <ctype.h>
148e3e3a7aSWarner Losh #include <errno.h>
158e3e3a7aSWarner Losh #include <locale.h>
168e3e3a7aSWarner Losh #include <stdio.h>
178e3e3a7aSWarner Losh #include <stdlib.h>
188e3e3a7aSWarner Losh #include <string.h>
198e3e3a7aSWarner Losh 
208e3e3a7aSWarner Losh #include "lua.h"
218e3e3a7aSWarner Losh 
228e3e3a7aSWarner Losh #include "lauxlib.h"
238e3e3a7aSWarner Losh #include "lualib.h"
248e3e3a7aSWarner Losh 
258e3e3a7aSWarner Losh 
268e3e3a7aSWarner Losh 
278e3e3a7aSWarner Losh 
288e3e3a7aSWarner Losh /*
298e3e3a7aSWarner Losh ** Change this macro to accept other modes for 'fopen' besides
308e3e3a7aSWarner Losh ** the standard ones.
318e3e3a7aSWarner Losh */
328e3e3a7aSWarner Losh #if !defined(l_checkmode)
338e3e3a7aSWarner Losh 
348e3e3a7aSWarner Losh /* accepted extensions to 'mode' in 'fopen' */
358e3e3a7aSWarner Losh #if !defined(L_MODEEXT)
368e3e3a7aSWarner Losh #define L_MODEEXT	"b"
378e3e3a7aSWarner Losh #endif
388e3e3a7aSWarner Losh 
398e3e3a7aSWarner Losh /* Check whether 'mode' matches '[rwa]%+?[L_MODEEXT]*' */
408e3e3a7aSWarner Losh static int l_checkmode (const char *mode) {
418e3e3a7aSWarner Losh   return (*mode != '\0' && strchr("rwa", *(mode++)) != NULL &&
42*0495ed39SKyle Evans          (*mode != '+' || ((void)(++mode), 1)) &&  /* skip if char is '+' */
438e3e3a7aSWarner Losh          (strspn(mode, L_MODEEXT) == strlen(mode)));  /* check extensions */
448e3e3a7aSWarner Losh }
458e3e3a7aSWarner Losh 
468e3e3a7aSWarner Losh #endif
478e3e3a7aSWarner Losh 
488e3e3a7aSWarner Losh /*
498e3e3a7aSWarner Losh ** {======================================================
508e3e3a7aSWarner Losh ** l_popen spawns a new process connected to the current
518e3e3a7aSWarner Losh ** one through the file streams.
528e3e3a7aSWarner Losh ** =======================================================
538e3e3a7aSWarner Losh */
548e3e3a7aSWarner Losh 
55*0495ed39SKyle Evans #if !defined(l_checkmodep)
56*0495ed39SKyle Evans /* By default, Lua accepts only "r" or "w" as mode */
57*0495ed39SKyle Evans #define l_checkmodep(m)	((m[0] == 'r' || m[0] == 'w') && m[1] == '\0')
58*0495ed39SKyle Evans #endif
59*0495ed39SKyle Evans 
60*0495ed39SKyle Evans 
618e3e3a7aSWarner Losh #if !defined(l_popen)		/* { */
628e3e3a7aSWarner Losh 
638e3e3a7aSWarner Losh #if defined(LUA_USE_POSIX)	/* { */
648e3e3a7aSWarner Losh 
658e3e3a7aSWarner Losh #define l_popen(L,c,m)		(fflush(NULL), popen(c,m))
668e3e3a7aSWarner Losh #define l_pclose(L,file)	(pclose(file))
678e3e3a7aSWarner Losh 
688e3e3a7aSWarner Losh #elif defined(LUA_USE_WINDOWS)	/* }{ */
698e3e3a7aSWarner Losh 
708e3e3a7aSWarner Losh #define l_popen(L,c,m)		(_popen(c,m))
718e3e3a7aSWarner Losh #define l_pclose(L,file)	(_pclose(file))
728e3e3a7aSWarner Losh 
738e3e3a7aSWarner Losh #else				/* }{ */
748e3e3a7aSWarner Losh 
758e3e3a7aSWarner Losh /* ISO C definitions */
768e3e3a7aSWarner Losh #define l_popen(L,c,m)  \
77*0495ed39SKyle Evans 	  ((void)c, (void)m, \
788e3e3a7aSWarner Losh 	  luaL_error(L, "'popen' not supported"), \
798e3e3a7aSWarner Losh 	  (FILE*)0)
808e3e3a7aSWarner Losh #define l_pclose(L,file)		((void)L, (void)file, -1)
818e3e3a7aSWarner Losh 
828e3e3a7aSWarner Losh #endif				/* } */
838e3e3a7aSWarner Losh 
848e3e3a7aSWarner Losh #endif				/* } */
858e3e3a7aSWarner Losh 
868e3e3a7aSWarner Losh /* }====================================================== */
878e3e3a7aSWarner Losh 
888e3e3a7aSWarner Losh 
898e3e3a7aSWarner Losh #if !defined(l_getc)		/* { */
908e3e3a7aSWarner Losh 
918e3e3a7aSWarner Losh #if defined(LUA_USE_POSIX)
928e3e3a7aSWarner Losh #define l_getc(f)		getc_unlocked(f)
938e3e3a7aSWarner Losh #define l_lockfile(f)		flockfile(f)
948e3e3a7aSWarner Losh #define l_unlockfile(f)		funlockfile(f)
958e3e3a7aSWarner Losh #else
968e3e3a7aSWarner Losh #define l_getc(f)		getc(f)
978e3e3a7aSWarner Losh #define l_lockfile(f)		((void)0)
988e3e3a7aSWarner Losh #define l_unlockfile(f)		((void)0)
998e3e3a7aSWarner Losh #endif
1008e3e3a7aSWarner Losh 
1018e3e3a7aSWarner Losh #endif				/* } */
1028e3e3a7aSWarner Losh 
1038e3e3a7aSWarner Losh 
1048e3e3a7aSWarner Losh /*
1058e3e3a7aSWarner Losh ** {======================================================
1068e3e3a7aSWarner Losh ** l_fseek: configuration for longer offsets
1078e3e3a7aSWarner Losh ** =======================================================
1088e3e3a7aSWarner Losh */
1098e3e3a7aSWarner Losh 
1108e3e3a7aSWarner Losh #if !defined(l_fseek)		/* { */
1118e3e3a7aSWarner Losh 
1128e3e3a7aSWarner Losh #if defined(LUA_USE_POSIX)	/* { */
1138e3e3a7aSWarner Losh 
1148e3e3a7aSWarner Losh #include <sys/types.h>
1158e3e3a7aSWarner Losh 
1168e3e3a7aSWarner Losh #define l_fseek(f,o,w)		fseeko(f,o,w)
1178e3e3a7aSWarner Losh #define l_ftell(f)		ftello(f)
1188e3e3a7aSWarner Losh #define l_seeknum		off_t
1198e3e3a7aSWarner Losh 
1208e3e3a7aSWarner Losh #elif defined(LUA_USE_WINDOWS) && !defined(_CRTIMP_TYPEINFO) \
1218e3e3a7aSWarner Losh    && defined(_MSC_VER) && (_MSC_VER >= 1400)	/* }{ */
1228e3e3a7aSWarner Losh 
1238e3e3a7aSWarner Losh /* Windows (but not DDK) and Visual C++ 2005 or higher */
1248e3e3a7aSWarner Losh #define l_fseek(f,o,w)		_fseeki64(f,o,w)
1258e3e3a7aSWarner Losh #define l_ftell(f)		_ftelli64(f)
1268e3e3a7aSWarner Losh #define l_seeknum		__int64
1278e3e3a7aSWarner Losh 
1288e3e3a7aSWarner Losh #else				/* }{ */
1298e3e3a7aSWarner Losh 
1308e3e3a7aSWarner Losh /* ISO C definitions */
1318e3e3a7aSWarner Losh #define l_fseek(f,o,w)		fseek(f,o,w)
1328e3e3a7aSWarner Losh #define l_ftell(f)		ftell(f)
1338e3e3a7aSWarner Losh #define l_seeknum		long
1348e3e3a7aSWarner Losh 
1358e3e3a7aSWarner Losh #endif				/* } */
1368e3e3a7aSWarner Losh 
1378e3e3a7aSWarner Losh #endif				/* } */
1388e3e3a7aSWarner Losh 
1398e3e3a7aSWarner Losh /* }====================================================== */
1408e3e3a7aSWarner Losh 
1418e3e3a7aSWarner Losh 
142*0495ed39SKyle Evans 
1438e3e3a7aSWarner Losh #define IO_PREFIX	"_IO_"
1448e3e3a7aSWarner Losh #define IOPREF_LEN	(sizeof(IO_PREFIX)/sizeof(char) - 1)
1458e3e3a7aSWarner Losh #define IO_INPUT	(IO_PREFIX "input")
1468e3e3a7aSWarner Losh #define IO_OUTPUT	(IO_PREFIX "output")
1478e3e3a7aSWarner Losh 
1488e3e3a7aSWarner Losh 
1498e3e3a7aSWarner Losh typedef luaL_Stream LStream;
1508e3e3a7aSWarner Losh 
1518e3e3a7aSWarner Losh 
1528e3e3a7aSWarner Losh #define tolstream(L)	((LStream *)luaL_checkudata(L, 1, LUA_FILEHANDLE))
1538e3e3a7aSWarner Losh 
1548e3e3a7aSWarner Losh #define isclosed(p)	((p)->closef == NULL)
1558e3e3a7aSWarner Losh 
1568e3e3a7aSWarner Losh 
1578e3e3a7aSWarner Losh static int io_type (lua_State *L) {
1588e3e3a7aSWarner Losh   LStream *p;
1598e3e3a7aSWarner Losh   luaL_checkany(L, 1);
1608e3e3a7aSWarner Losh   p = (LStream *)luaL_testudata(L, 1, LUA_FILEHANDLE);
1618e3e3a7aSWarner Losh   if (p == NULL)
162*0495ed39SKyle Evans     luaL_pushfail(L);  /* not a file */
1638e3e3a7aSWarner Losh   else if (isclosed(p))
1648e3e3a7aSWarner Losh     lua_pushliteral(L, "closed file");
1658e3e3a7aSWarner Losh   else
1668e3e3a7aSWarner Losh     lua_pushliteral(L, "file");
1678e3e3a7aSWarner Losh   return 1;
1688e3e3a7aSWarner Losh }
1698e3e3a7aSWarner Losh 
1708e3e3a7aSWarner Losh 
1718e3e3a7aSWarner Losh static int f_tostring (lua_State *L) {
1728e3e3a7aSWarner Losh   LStream *p = tolstream(L);
1738e3e3a7aSWarner Losh   if (isclosed(p))
1748e3e3a7aSWarner Losh     lua_pushliteral(L, "file (closed)");
1758e3e3a7aSWarner Losh   else
1768e3e3a7aSWarner Losh     lua_pushfstring(L, "file (%p)", p->f);
1778e3e3a7aSWarner Losh   return 1;
1788e3e3a7aSWarner Losh }
1798e3e3a7aSWarner Losh 
1808e3e3a7aSWarner Losh 
1818e3e3a7aSWarner Losh static FILE *tofile (lua_State *L) {
1828e3e3a7aSWarner Losh   LStream *p = tolstream(L);
1838e3e3a7aSWarner Losh   if (isclosed(p))
1848e3e3a7aSWarner Losh     luaL_error(L, "attempt to use a closed file");
1858e3e3a7aSWarner Losh   lua_assert(p->f);
1868e3e3a7aSWarner Losh   return p->f;
1878e3e3a7aSWarner Losh }
1888e3e3a7aSWarner Losh 
1898e3e3a7aSWarner Losh 
1908e3e3a7aSWarner Losh /*
1918e3e3a7aSWarner Losh ** When creating file handles, always creates a 'closed' file handle
1928e3e3a7aSWarner Losh ** before opening the actual file; so, if there is a memory error, the
1938e3e3a7aSWarner Losh ** handle is in a consistent state.
1948e3e3a7aSWarner Losh */
1958e3e3a7aSWarner Losh static LStream *newprefile (lua_State *L) {
196*0495ed39SKyle Evans   LStream *p = (LStream *)lua_newuserdatauv(L, sizeof(LStream), 0);
1978e3e3a7aSWarner Losh   p->closef = NULL;  /* mark file handle as 'closed' */
1988e3e3a7aSWarner Losh   luaL_setmetatable(L, LUA_FILEHANDLE);
1998e3e3a7aSWarner Losh   return p;
2008e3e3a7aSWarner Losh }
2018e3e3a7aSWarner Losh 
2028e3e3a7aSWarner Losh 
2038e3e3a7aSWarner Losh /*
2048e3e3a7aSWarner Losh ** Calls the 'close' function from a file handle. The 'volatile' avoids
2058e3e3a7aSWarner Losh ** a bug in some versions of the Clang compiler (e.g., clang 3.0 for
2068e3e3a7aSWarner Losh ** 32 bits).
2078e3e3a7aSWarner Losh */
2088e3e3a7aSWarner Losh static int aux_close (lua_State *L) {
2098e3e3a7aSWarner Losh   LStream *p = tolstream(L);
2108e3e3a7aSWarner Losh   volatile lua_CFunction cf = p->closef;
2118e3e3a7aSWarner Losh   p->closef = NULL;  /* mark stream as closed */
2128e3e3a7aSWarner Losh   return (*cf)(L);  /* close it */
2138e3e3a7aSWarner Losh }
2148e3e3a7aSWarner Losh 
2158e3e3a7aSWarner Losh 
216e112e9d2SKyle Evans static int f_close (lua_State *L) {
217e112e9d2SKyle Evans   tofile(L);  /* make sure argument is an open stream */
218e112e9d2SKyle Evans   return aux_close(L);
219e112e9d2SKyle Evans }
220e112e9d2SKyle Evans 
221e112e9d2SKyle Evans 
2228e3e3a7aSWarner Losh static int io_close (lua_State *L) {
2238e3e3a7aSWarner Losh   if (lua_isnone(L, 1))  /* no argument? */
224*0495ed39SKyle Evans     lua_getfield(L, LUA_REGISTRYINDEX, IO_OUTPUT);  /* use default output */
225e112e9d2SKyle Evans   return f_close(L);
2268e3e3a7aSWarner Losh }
2278e3e3a7aSWarner Losh 
2288e3e3a7aSWarner Losh 
2298e3e3a7aSWarner Losh static int f_gc (lua_State *L) {
2308e3e3a7aSWarner Losh   LStream *p = tolstream(L);
2318e3e3a7aSWarner Losh   if (!isclosed(p) && p->f != NULL)
2328e3e3a7aSWarner Losh     aux_close(L);  /* ignore closed and incompletely open files */
2338e3e3a7aSWarner Losh   return 0;
2348e3e3a7aSWarner Losh }
2358e3e3a7aSWarner Losh 
2368e3e3a7aSWarner Losh 
2378e3e3a7aSWarner Losh /*
2388e3e3a7aSWarner Losh ** function to close regular files
2398e3e3a7aSWarner Losh */
2408e3e3a7aSWarner Losh static int io_fclose (lua_State *L) {
2418e3e3a7aSWarner Losh   LStream *p = tolstream(L);
2428e3e3a7aSWarner Losh   int res = fclose(p->f);
2438e3e3a7aSWarner Losh   return luaL_fileresult(L, (res == 0), NULL);
2448e3e3a7aSWarner Losh }
2458e3e3a7aSWarner Losh 
2468e3e3a7aSWarner Losh 
2478e3e3a7aSWarner Losh static LStream *newfile (lua_State *L) {
2488e3e3a7aSWarner Losh   LStream *p = newprefile(L);
2498e3e3a7aSWarner Losh   p->f = NULL;
2508e3e3a7aSWarner Losh   p->closef = &io_fclose;
2518e3e3a7aSWarner Losh   return p;
2528e3e3a7aSWarner Losh }
2538e3e3a7aSWarner Losh 
2548e3e3a7aSWarner Losh 
2558e3e3a7aSWarner Losh static void opencheck (lua_State *L, const char *fname, const char *mode) {
2568e3e3a7aSWarner Losh   LStream *p = newfile(L);
2578e3e3a7aSWarner Losh   p->f = fopen(fname, mode);
2588e3e3a7aSWarner Losh   if (p->f == NULL)
2598e3e3a7aSWarner Losh     luaL_error(L, "cannot open file '%s' (%s)", fname, strerror(errno));
2608e3e3a7aSWarner Losh }
2618e3e3a7aSWarner Losh 
2628e3e3a7aSWarner Losh 
2638e3e3a7aSWarner Losh static int io_open (lua_State *L) {
2648e3e3a7aSWarner Losh   const char *filename = luaL_checkstring(L, 1);
2658e3e3a7aSWarner Losh   const char *mode = luaL_optstring(L, 2, "r");
2668e3e3a7aSWarner Losh   LStream *p = newfile(L);
2678e3e3a7aSWarner Losh   const char *md = mode;  /* to traverse/check mode */
2688e3e3a7aSWarner Losh   luaL_argcheck(L, l_checkmode(md), 2, "invalid mode");
2698e3e3a7aSWarner Losh   p->f = fopen(filename, mode);
2708e3e3a7aSWarner Losh   return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;
2718e3e3a7aSWarner Losh }
2728e3e3a7aSWarner Losh 
2738e3e3a7aSWarner Losh 
2748e3e3a7aSWarner Losh /*
2758e3e3a7aSWarner Losh ** function to close 'popen' files
2768e3e3a7aSWarner Losh */
2778e3e3a7aSWarner Losh static int io_pclose (lua_State *L) {
2788e3e3a7aSWarner Losh   LStream *p = tolstream(L);
279*0495ed39SKyle Evans   errno = 0;
2808e3e3a7aSWarner Losh   return luaL_execresult(L, l_pclose(L, p->f));
2818e3e3a7aSWarner Losh }
2828e3e3a7aSWarner Losh 
2838e3e3a7aSWarner Losh 
2848e3e3a7aSWarner Losh static int io_popen (lua_State *L) {
2858e3e3a7aSWarner Losh   const char *filename = luaL_checkstring(L, 1);
2868e3e3a7aSWarner Losh   const char *mode = luaL_optstring(L, 2, "r");
2878e3e3a7aSWarner Losh   LStream *p = newprefile(L);
288*0495ed39SKyle Evans   luaL_argcheck(L, l_checkmodep(mode), 2, "invalid mode");
2898e3e3a7aSWarner Losh   p->f = l_popen(L, filename, mode);
2908e3e3a7aSWarner Losh   p->closef = &io_pclose;
2918e3e3a7aSWarner Losh   return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;
2928e3e3a7aSWarner Losh }
2938e3e3a7aSWarner Losh 
2948e3e3a7aSWarner Losh 
2958e3e3a7aSWarner Losh static int io_tmpfile (lua_State *L) {
2968e3e3a7aSWarner Losh   LStream *p = newfile(L);
2978e3e3a7aSWarner Losh   p->f = tmpfile();
2988e3e3a7aSWarner Losh   return (p->f == NULL) ? luaL_fileresult(L, 0, NULL) : 1;
2998e3e3a7aSWarner Losh }
3008e3e3a7aSWarner Losh 
3018e3e3a7aSWarner Losh 
3028e3e3a7aSWarner Losh static FILE *getiofile (lua_State *L, const char *findex) {
3038e3e3a7aSWarner Losh   LStream *p;
3048e3e3a7aSWarner Losh   lua_getfield(L, LUA_REGISTRYINDEX, findex);
3058e3e3a7aSWarner Losh   p = (LStream *)lua_touserdata(L, -1);
3068e3e3a7aSWarner Losh   if (isclosed(p))
307*0495ed39SKyle Evans     luaL_error(L, "default %s file is closed", findex + IOPREF_LEN);
3088e3e3a7aSWarner Losh   return p->f;
3098e3e3a7aSWarner Losh }
3108e3e3a7aSWarner Losh 
3118e3e3a7aSWarner Losh 
3128e3e3a7aSWarner Losh static int g_iofile (lua_State *L, const char *f, const char *mode) {
3138e3e3a7aSWarner Losh   if (!lua_isnoneornil(L, 1)) {
3148e3e3a7aSWarner Losh     const char *filename = lua_tostring(L, 1);
3158e3e3a7aSWarner Losh     if (filename)
3168e3e3a7aSWarner Losh       opencheck(L, filename, mode);
3178e3e3a7aSWarner Losh     else {
3188e3e3a7aSWarner Losh       tofile(L);  /* check that it's a valid file handle */
3198e3e3a7aSWarner Losh       lua_pushvalue(L, 1);
3208e3e3a7aSWarner Losh     }
3218e3e3a7aSWarner Losh     lua_setfield(L, LUA_REGISTRYINDEX, f);
3228e3e3a7aSWarner Losh   }
3238e3e3a7aSWarner Losh   /* return current value */
3248e3e3a7aSWarner Losh   lua_getfield(L, LUA_REGISTRYINDEX, f);
3258e3e3a7aSWarner Losh   return 1;
3268e3e3a7aSWarner Losh }
3278e3e3a7aSWarner Losh 
3288e3e3a7aSWarner Losh 
3298e3e3a7aSWarner Losh static int io_input (lua_State *L) {
3308e3e3a7aSWarner Losh   return g_iofile(L, IO_INPUT, "r");
3318e3e3a7aSWarner Losh }
3328e3e3a7aSWarner Losh 
3338e3e3a7aSWarner Losh 
3348e3e3a7aSWarner Losh static int io_output (lua_State *L) {
3358e3e3a7aSWarner Losh   return g_iofile(L, IO_OUTPUT, "w");
3368e3e3a7aSWarner Losh }
3378e3e3a7aSWarner Losh 
3388e3e3a7aSWarner Losh 
3398e3e3a7aSWarner Losh static int io_readline (lua_State *L);
3408e3e3a7aSWarner Losh 
3418e3e3a7aSWarner Losh 
3428e3e3a7aSWarner Losh /*
3438e3e3a7aSWarner Losh ** maximum number of arguments to 'f:lines'/'io.lines' (it + 3 must fit
3448e3e3a7aSWarner Losh ** in the limit for upvalues of a closure)
3458e3e3a7aSWarner Losh */
3468e3e3a7aSWarner Losh #define MAXARGLINE	250
3478e3e3a7aSWarner Losh 
348*0495ed39SKyle Evans /*
349*0495ed39SKyle Evans ** Auxiliary function to create the iteration function for 'lines'.
350*0495ed39SKyle Evans ** The iteration function is a closure over 'io_readline', with
351*0495ed39SKyle Evans ** the following upvalues:
352*0495ed39SKyle Evans ** 1) The file being read (first value in the stack)
353*0495ed39SKyle Evans ** 2) the number of arguments to read
354*0495ed39SKyle Evans ** 3) a boolean, true iff file has to be closed when finished ('toclose')
355*0495ed39SKyle Evans ** *) a variable number of format arguments (rest of the stack)
356*0495ed39SKyle Evans */
3578e3e3a7aSWarner Losh static void aux_lines (lua_State *L, int toclose) {
3588e3e3a7aSWarner Losh   int n = lua_gettop(L) - 1;  /* number of arguments to read */
3598e3e3a7aSWarner Losh   luaL_argcheck(L, n <= MAXARGLINE, MAXARGLINE + 2, "too many arguments");
360*0495ed39SKyle Evans   lua_pushvalue(L, 1);  /* file */
3618e3e3a7aSWarner Losh   lua_pushinteger(L, n);  /* number of arguments to read */
3628e3e3a7aSWarner Losh   lua_pushboolean(L, toclose);  /* close/not close file when finished */
363*0495ed39SKyle Evans   lua_rotate(L, 2, 3);  /* move the three values to their positions */
3648e3e3a7aSWarner Losh   lua_pushcclosure(L, io_readline, 3 + n);
3658e3e3a7aSWarner Losh }
3668e3e3a7aSWarner Losh 
3678e3e3a7aSWarner Losh 
3688e3e3a7aSWarner Losh static int f_lines (lua_State *L) {
3698e3e3a7aSWarner Losh   tofile(L);  /* check that it's a valid file handle */
3708e3e3a7aSWarner Losh   aux_lines(L, 0);
3718e3e3a7aSWarner Losh   return 1;
3728e3e3a7aSWarner Losh }
3738e3e3a7aSWarner Losh 
3748e3e3a7aSWarner Losh 
375*0495ed39SKyle Evans /*
376*0495ed39SKyle Evans ** Return an iteration function for 'io.lines'. If file has to be
377*0495ed39SKyle Evans ** closed, also returns the file itself as a second result (to be
378*0495ed39SKyle Evans ** closed as the state at the exit of a generic for).
379*0495ed39SKyle Evans */
3808e3e3a7aSWarner Losh static int io_lines (lua_State *L) {
3818e3e3a7aSWarner Losh   int toclose;
3828e3e3a7aSWarner Losh   if (lua_isnone(L, 1)) lua_pushnil(L);  /* at least one argument */
3838e3e3a7aSWarner Losh   if (lua_isnil(L, 1)) {  /* no file name? */
3848e3e3a7aSWarner Losh     lua_getfield(L, LUA_REGISTRYINDEX, IO_INPUT);  /* get default input */
3858e3e3a7aSWarner Losh     lua_replace(L, 1);  /* put it at index 1 */
3868e3e3a7aSWarner Losh     tofile(L);  /* check that it's a valid file handle */
3878e3e3a7aSWarner Losh     toclose = 0;  /* do not close it after iteration */
3888e3e3a7aSWarner Losh   }
3898e3e3a7aSWarner Losh   else {  /* open a new file */
3908e3e3a7aSWarner Losh     const char *filename = luaL_checkstring(L, 1);
3918e3e3a7aSWarner Losh     opencheck(L, filename, "r");
3928e3e3a7aSWarner Losh     lua_replace(L, 1);  /* put file at index 1 */
3938e3e3a7aSWarner Losh     toclose = 1;  /* close it after iteration */
3948e3e3a7aSWarner Losh   }
395*0495ed39SKyle Evans   aux_lines(L, toclose);  /* push iteration function */
396*0495ed39SKyle Evans   if (toclose) {
397*0495ed39SKyle Evans     lua_pushnil(L);  /* state */
398*0495ed39SKyle Evans     lua_pushnil(L);  /* control */
399*0495ed39SKyle Evans     lua_pushvalue(L, 1);  /* file is the to-be-closed variable (4th result) */
400*0495ed39SKyle Evans     return 4;
401*0495ed39SKyle Evans   }
402*0495ed39SKyle Evans   else
4038e3e3a7aSWarner Losh     return 1;
4048e3e3a7aSWarner Losh }
4058e3e3a7aSWarner Losh 
4068e3e3a7aSWarner Losh 
4078e3e3a7aSWarner Losh /*
4088e3e3a7aSWarner Losh ** {======================================================
4098e3e3a7aSWarner Losh ** READ
4108e3e3a7aSWarner Losh ** =======================================================
4118e3e3a7aSWarner Losh */
4128e3e3a7aSWarner Losh 
4138e3e3a7aSWarner Losh 
4148e3e3a7aSWarner Losh /* maximum length of a numeral */
4158e3e3a7aSWarner Losh #if !defined (L_MAXLENNUM)
4168e3e3a7aSWarner Losh #define L_MAXLENNUM     200
4178e3e3a7aSWarner Losh #endif
4188e3e3a7aSWarner Losh 
4198e3e3a7aSWarner Losh 
4208e3e3a7aSWarner Losh /* auxiliary structure used by 'read_number' */
4218e3e3a7aSWarner Losh typedef struct {
4228e3e3a7aSWarner Losh   FILE *f;  /* file being read */
4238e3e3a7aSWarner Losh   int c;  /* current character (look ahead) */
4248e3e3a7aSWarner Losh   int n;  /* number of elements in buffer 'buff' */
4258e3e3a7aSWarner Losh   char buff[L_MAXLENNUM + 1];  /* +1 for ending '\0' */
4268e3e3a7aSWarner Losh } RN;
4278e3e3a7aSWarner Losh 
4288e3e3a7aSWarner Losh 
4298e3e3a7aSWarner Losh /*
4308e3e3a7aSWarner Losh ** Add current char to buffer (if not out of space) and read next one
4318e3e3a7aSWarner Losh */
4328e3e3a7aSWarner Losh static int nextc (RN *rn) {
4338e3e3a7aSWarner Losh   if (rn->n >= L_MAXLENNUM) {  /* buffer overflow? */
4348e3e3a7aSWarner Losh     rn->buff[0] = '\0';  /* invalidate result */
4358e3e3a7aSWarner Losh     return 0;  /* fail */
4368e3e3a7aSWarner Losh   }
4378e3e3a7aSWarner Losh   else {
4388e3e3a7aSWarner Losh     rn->buff[rn->n++] = rn->c;  /* save current char */
4398e3e3a7aSWarner Losh     rn->c = l_getc(rn->f);  /* read next one */
4408e3e3a7aSWarner Losh     return 1;
4418e3e3a7aSWarner Losh   }
4428e3e3a7aSWarner Losh }
4438e3e3a7aSWarner Losh 
4448e3e3a7aSWarner Losh 
4458e3e3a7aSWarner Losh /*
4468e3e3a7aSWarner Losh ** Accept current char if it is in 'set' (of size 2)
4478e3e3a7aSWarner Losh */
4488e3e3a7aSWarner Losh static int test2 (RN *rn, const char *set) {
4498e3e3a7aSWarner Losh   if (rn->c == set[0] || rn->c == set[1])
4508e3e3a7aSWarner Losh     return nextc(rn);
4518e3e3a7aSWarner Losh   else return 0;
4528e3e3a7aSWarner Losh }
4538e3e3a7aSWarner Losh 
4548e3e3a7aSWarner Losh 
4558e3e3a7aSWarner Losh /*
4568e3e3a7aSWarner Losh ** Read a sequence of (hex)digits
4578e3e3a7aSWarner Losh */
4588e3e3a7aSWarner Losh static int readdigits (RN *rn, int hex) {
4598e3e3a7aSWarner Losh   int count = 0;
4608e3e3a7aSWarner Losh   while ((hex ? isxdigit(rn->c) : isdigit(rn->c)) && nextc(rn))
4618e3e3a7aSWarner Losh     count++;
4628e3e3a7aSWarner Losh   return count;
4638e3e3a7aSWarner Losh }
4648e3e3a7aSWarner Losh 
4658e3e3a7aSWarner Losh 
4668e3e3a7aSWarner Losh /*
4678e3e3a7aSWarner Losh ** Read a number: first reads a valid prefix of a numeral into a buffer.
4688e3e3a7aSWarner Losh ** Then it calls 'lua_stringtonumber' to check whether the format is
469*0495ed39SKyle Evans ** correct and to convert it to a Lua number.
4708e3e3a7aSWarner Losh */
4718e3e3a7aSWarner Losh static int read_number (lua_State *L, FILE *f) {
4728e3e3a7aSWarner Losh   RN rn;
4738e3e3a7aSWarner Losh   int count = 0;
4748e3e3a7aSWarner Losh   int hex = 0;
4758e3e3a7aSWarner Losh   char decp[2];
4768e3e3a7aSWarner Losh   rn.f = f; rn.n = 0;
4778e3e3a7aSWarner Losh   decp[0] = lua_getlocaledecpoint();  /* get decimal point from locale */
4788e3e3a7aSWarner Losh   decp[1] = '.';  /* always accept a dot */
4798e3e3a7aSWarner Losh   l_lockfile(rn.f);
4808e3e3a7aSWarner Losh   do { rn.c = l_getc(rn.f); } while (isspace(rn.c));  /* skip spaces */
481*0495ed39SKyle Evans   test2(&rn, "-+");  /* optional sign */
4828e3e3a7aSWarner Losh   if (test2(&rn, "00")) {
4838e3e3a7aSWarner Losh     if (test2(&rn, "xX")) hex = 1;  /* numeral is hexadecimal */
4848e3e3a7aSWarner Losh     else count = 1;  /* count initial '0' as a valid digit */
4858e3e3a7aSWarner Losh   }
4868e3e3a7aSWarner Losh   count += readdigits(&rn, hex);  /* integral part */
4878e3e3a7aSWarner Losh   if (test2(&rn, decp))  /* decimal point? */
4888e3e3a7aSWarner Losh     count += readdigits(&rn, hex);  /* fractional part */
4898e3e3a7aSWarner Losh   if (count > 0 && test2(&rn, (hex ? "pP" : "eE"))) {  /* exponent mark? */
490*0495ed39SKyle Evans     test2(&rn, "-+");  /* exponent sign */
4918e3e3a7aSWarner Losh     readdigits(&rn, 0);  /* exponent digits */
4928e3e3a7aSWarner Losh   }
4938e3e3a7aSWarner Losh   ungetc(rn.c, rn.f);  /* unread look-ahead char */
4948e3e3a7aSWarner Losh   l_unlockfile(rn.f);
4958e3e3a7aSWarner Losh   rn.buff[rn.n] = '\0';  /* finish string */
4968e3e3a7aSWarner Losh   if (lua_stringtonumber(L, rn.buff))  /* is this a valid number? */
4978e3e3a7aSWarner Losh     return 1;  /* ok */
4988e3e3a7aSWarner Losh   else {  /* invalid format */
4998e3e3a7aSWarner Losh    lua_pushnil(L);  /* "result" to be removed */
5008e3e3a7aSWarner Losh    return 0;  /* read fails */
5018e3e3a7aSWarner Losh   }
5028e3e3a7aSWarner Losh }
5038e3e3a7aSWarner Losh 
5048e3e3a7aSWarner Losh 
5058e3e3a7aSWarner Losh static int test_eof (lua_State *L, FILE *f) {
5068e3e3a7aSWarner Losh   int c = getc(f);
5078e3e3a7aSWarner Losh   ungetc(c, f);  /* no-op when c == EOF */
5088e3e3a7aSWarner Losh   lua_pushliteral(L, "");
5098e3e3a7aSWarner Losh   return (c != EOF);
5108e3e3a7aSWarner Losh }
5118e3e3a7aSWarner Losh 
5128e3e3a7aSWarner Losh 
5138e3e3a7aSWarner Losh static int read_line (lua_State *L, FILE *f, int chop) {
5148e3e3a7aSWarner Losh   luaL_Buffer b;
515*0495ed39SKyle Evans   int c;
5168e3e3a7aSWarner Losh   luaL_buffinit(L, &b);
517*0495ed39SKyle Evans   do {  /* may need to read several chunks to get whole line */
518*0495ed39SKyle Evans     char *buff = luaL_prepbuffer(&b);  /* preallocate buffer space */
5198e3e3a7aSWarner Losh     int i = 0;
5208e3e3a7aSWarner Losh     l_lockfile(f);  /* no memory errors can happen inside the lock */
5218e3e3a7aSWarner Losh     while (i < LUAL_BUFFERSIZE && (c = l_getc(f)) != EOF && c != '\n')
522*0495ed39SKyle Evans       buff[i++] = c;  /* read up to end of line or buffer limit */
5238e3e3a7aSWarner Losh     l_unlockfile(f);
5248e3e3a7aSWarner Losh     luaL_addsize(&b, i);
525*0495ed39SKyle Evans   } while (c != EOF && c != '\n');  /* repeat until end of line */
5268e3e3a7aSWarner Losh   if (!chop && c == '\n')  /* want a newline and have one? */
5278e3e3a7aSWarner Losh     luaL_addchar(&b, c);  /* add ending newline to result */
5288e3e3a7aSWarner Losh   luaL_pushresult(&b);  /* close buffer */
5298e3e3a7aSWarner Losh   /* return ok if read something (either a newline or something else) */
5308e3e3a7aSWarner Losh   return (c == '\n' || lua_rawlen(L, -1) > 0);
5318e3e3a7aSWarner Losh }
5328e3e3a7aSWarner Losh 
5338e3e3a7aSWarner Losh 
5348e3e3a7aSWarner Losh static void read_all (lua_State *L, FILE *f) {
5358e3e3a7aSWarner Losh   size_t nr;
5368e3e3a7aSWarner Losh   luaL_Buffer b;
5378e3e3a7aSWarner Losh   luaL_buffinit(L, &b);
5388e3e3a7aSWarner Losh   do {  /* read file in chunks of LUAL_BUFFERSIZE bytes */
5398e3e3a7aSWarner Losh     char *p = luaL_prepbuffer(&b);
5408e3e3a7aSWarner Losh     nr = fread(p, sizeof(char), LUAL_BUFFERSIZE, f);
5418e3e3a7aSWarner Losh     luaL_addsize(&b, nr);
5428e3e3a7aSWarner Losh   } while (nr == LUAL_BUFFERSIZE);
5438e3e3a7aSWarner Losh   luaL_pushresult(&b);  /* close buffer */
5448e3e3a7aSWarner Losh }
5458e3e3a7aSWarner Losh 
5468e3e3a7aSWarner Losh 
5478e3e3a7aSWarner Losh static int read_chars (lua_State *L, FILE *f, size_t n) {
5488e3e3a7aSWarner Losh   size_t nr;  /* number of chars actually read */
5498e3e3a7aSWarner Losh   char *p;
5508e3e3a7aSWarner Losh   luaL_Buffer b;
5518e3e3a7aSWarner Losh   luaL_buffinit(L, &b);
5528e3e3a7aSWarner Losh   p = luaL_prepbuffsize(&b, n);  /* prepare buffer to read whole block */
5538e3e3a7aSWarner Losh   nr = fread(p, sizeof(char), n, f);  /* try to read 'n' chars */
5548e3e3a7aSWarner Losh   luaL_addsize(&b, nr);
5558e3e3a7aSWarner Losh   luaL_pushresult(&b);  /* close buffer */
5568e3e3a7aSWarner Losh   return (nr > 0);  /* true iff read something */
5578e3e3a7aSWarner Losh }
5588e3e3a7aSWarner Losh 
5598e3e3a7aSWarner Losh 
5608e3e3a7aSWarner Losh static int g_read (lua_State *L, FILE *f, int first) {
5618e3e3a7aSWarner Losh   int nargs = lua_gettop(L) - 1;
562*0495ed39SKyle Evans   int n, success;
5638e3e3a7aSWarner Losh   clearerr(f);
5648e3e3a7aSWarner Losh   if (nargs == 0) {  /* no arguments? */
5658e3e3a7aSWarner Losh     success = read_line(L, f, 1);
5668e3e3a7aSWarner Losh     n = first + 1;  /* to return 1 result */
5678e3e3a7aSWarner Losh   }
568*0495ed39SKyle Evans   else {
569*0495ed39SKyle Evans     /* ensure stack space for all results and for auxlib's buffer */
5708e3e3a7aSWarner Losh     luaL_checkstack(L, nargs+LUA_MINSTACK, "too many arguments");
5718e3e3a7aSWarner Losh     success = 1;
5728e3e3a7aSWarner Losh     for (n = first; nargs-- && success; n++) {
5738e3e3a7aSWarner Losh       if (lua_type(L, n) == LUA_TNUMBER) {
5748e3e3a7aSWarner Losh         size_t l = (size_t)luaL_checkinteger(L, n);
5758e3e3a7aSWarner Losh         success = (l == 0) ? test_eof(L, f) : read_chars(L, f, l);
5768e3e3a7aSWarner Losh       }
5778e3e3a7aSWarner Losh       else {
5788e3e3a7aSWarner Losh         const char *p = luaL_checkstring(L, n);
5798e3e3a7aSWarner Losh         if (*p == '*') p++;  /* skip optional '*' (for compatibility) */
5808e3e3a7aSWarner Losh         switch (*p) {
5818e3e3a7aSWarner Losh           case 'n':  /* number */
5828e3e3a7aSWarner Losh             success = read_number(L, f);
5838e3e3a7aSWarner Losh             break;
5848e3e3a7aSWarner Losh           case 'l':  /* line */
5858e3e3a7aSWarner Losh             success = read_line(L, f, 1);
5868e3e3a7aSWarner Losh             break;
5878e3e3a7aSWarner Losh           case 'L':  /* line with end-of-line */
5888e3e3a7aSWarner Losh             success = read_line(L, f, 0);
5898e3e3a7aSWarner Losh             break;
5908e3e3a7aSWarner Losh           case 'a':  /* file */
5918e3e3a7aSWarner Losh             read_all(L, f);  /* read entire file */
5928e3e3a7aSWarner Losh             success = 1; /* always success */
5938e3e3a7aSWarner Losh             break;
5948e3e3a7aSWarner Losh           default:
5958e3e3a7aSWarner Losh             return luaL_argerror(L, n, "invalid format");
5968e3e3a7aSWarner Losh         }
5978e3e3a7aSWarner Losh       }
5988e3e3a7aSWarner Losh     }
5998e3e3a7aSWarner Losh   }
6008e3e3a7aSWarner Losh   if (ferror(f))
6018e3e3a7aSWarner Losh     return luaL_fileresult(L, 0, NULL);
6028e3e3a7aSWarner Losh   if (!success) {
6038e3e3a7aSWarner Losh     lua_pop(L, 1);  /* remove last result */
604*0495ed39SKyle Evans     luaL_pushfail(L);  /* push nil instead */
6058e3e3a7aSWarner Losh   }
6068e3e3a7aSWarner Losh   return n - first;
6078e3e3a7aSWarner Losh }
6088e3e3a7aSWarner Losh 
6098e3e3a7aSWarner Losh 
6108e3e3a7aSWarner Losh static int io_read (lua_State *L) {
6118e3e3a7aSWarner Losh   return g_read(L, getiofile(L, IO_INPUT), 1);
6128e3e3a7aSWarner Losh }
6138e3e3a7aSWarner Losh 
6148e3e3a7aSWarner Losh 
6158e3e3a7aSWarner Losh static int f_read (lua_State *L) {
6168e3e3a7aSWarner Losh   return g_read(L, tofile(L), 2);
6178e3e3a7aSWarner Losh }
6188e3e3a7aSWarner Losh 
6198e3e3a7aSWarner Losh 
620*0495ed39SKyle Evans /*
621*0495ed39SKyle Evans ** Iteration function for 'lines'.
622*0495ed39SKyle Evans */
6238e3e3a7aSWarner Losh static int io_readline (lua_State *L) {
6248e3e3a7aSWarner Losh   LStream *p = (LStream *)lua_touserdata(L, lua_upvalueindex(1));
6258e3e3a7aSWarner Losh   int i;
6268e3e3a7aSWarner Losh   int n = (int)lua_tointeger(L, lua_upvalueindex(2));
6278e3e3a7aSWarner Losh   if (isclosed(p))  /* file is already closed? */
6288e3e3a7aSWarner Losh     return luaL_error(L, "file is already closed");
6298e3e3a7aSWarner Losh   lua_settop(L , 1);
6308e3e3a7aSWarner Losh   luaL_checkstack(L, n, "too many arguments");
6318e3e3a7aSWarner Losh   for (i = 1; i <= n; i++)  /* push arguments to 'g_read' */
6328e3e3a7aSWarner Losh     lua_pushvalue(L, lua_upvalueindex(3 + i));
6338e3e3a7aSWarner Losh   n = g_read(L, p->f, 2);  /* 'n' is number of results */
6348e3e3a7aSWarner Losh   lua_assert(n > 0);  /* should return at least a nil */
6358e3e3a7aSWarner Losh   if (lua_toboolean(L, -n))  /* read at least one value? */
6368e3e3a7aSWarner Losh     return n;  /* return them */
637*0495ed39SKyle Evans   else {  /* first result is false: EOF or error */
6388e3e3a7aSWarner Losh     if (n > 1) {  /* is there error information? */
6398e3e3a7aSWarner Losh       /* 2nd result is error message */
6408e3e3a7aSWarner Losh       return luaL_error(L, "%s", lua_tostring(L, -n + 1));
6418e3e3a7aSWarner Losh     }
6428e3e3a7aSWarner Losh     if (lua_toboolean(L, lua_upvalueindex(3))) {  /* generator created file? */
643*0495ed39SKyle Evans       lua_settop(L, 0);  /* clear stack */
644*0495ed39SKyle Evans       lua_pushvalue(L, lua_upvalueindex(1));  /* push file at index 1 */
6458e3e3a7aSWarner Losh       aux_close(L);  /* close it */
6468e3e3a7aSWarner Losh     }
6478e3e3a7aSWarner Losh     return 0;
6488e3e3a7aSWarner Losh   }
6498e3e3a7aSWarner Losh }
6508e3e3a7aSWarner Losh 
6518e3e3a7aSWarner Losh /* }====================================================== */
6528e3e3a7aSWarner Losh 
6538e3e3a7aSWarner Losh 
6548e3e3a7aSWarner Losh static int g_write (lua_State *L, FILE *f, int arg) {
6558e3e3a7aSWarner Losh   int nargs = lua_gettop(L) - arg;
6568e3e3a7aSWarner Losh   int status = 1;
6578e3e3a7aSWarner Losh   for (; nargs--; arg++) {
6588e3e3a7aSWarner Losh     if (lua_type(L, arg) == LUA_TNUMBER) {
6598e3e3a7aSWarner Losh       /* optimization: could be done exactly as for strings */
6608e3e3a7aSWarner Losh       int len = lua_isinteger(L, arg)
6618e3e3a7aSWarner Losh                 ? fprintf(f, LUA_INTEGER_FMT,
6628e3e3a7aSWarner Losh                              (LUAI_UACINT)lua_tointeger(L, arg))
6638e3e3a7aSWarner Losh                 : fprintf(f, LUA_NUMBER_FMT,
6648e3e3a7aSWarner Losh                              (LUAI_UACNUMBER)lua_tonumber(L, arg));
6658e3e3a7aSWarner Losh       status = status && (len > 0);
6668e3e3a7aSWarner Losh     }
6678e3e3a7aSWarner Losh     else {
6688e3e3a7aSWarner Losh       size_t l;
6698e3e3a7aSWarner Losh       const char *s = luaL_checklstring(L, arg, &l);
6708e3e3a7aSWarner Losh       status = status && (fwrite(s, sizeof(char), l, f) == l);
6718e3e3a7aSWarner Losh     }
6728e3e3a7aSWarner Losh   }
6738e3e3a7aSWarner Losh   if (status) return 1;  /* file handle already on stack top */
6748e3e3a7aSWarner Losh   else return luaL_fileresult(L, status, NULL);
6758e3e3a7aSWarner Losh }
6768e3e3a7aSWarner Losh 
6778e3e3a7aSWarner Losh 
6788e3e3a7aSWarner Losh static int io_write (lua_State *L) {
6798e3e3a7aSWarner Losh   return g_write(L, getiofile(L, IO_OUTPUT), 1);
6808e3e3a7aSWarner Losh }
6818e3e3a7aSWarner Losh 
6828e3e3a7aSWarner Losh 
6838e3e3a7aSWarner Losh static int f_write (lua_State *L) {
6848e3e3a7aSWarner Losh   FILE *f = tofile(L);
6858e3e3a7aSWarner Losh   lua_pushvalue(L, 1);  /* push file at the stack top (to be returned) */
6868e3e3a7aSWarner Losh   return g_write(L, f, 2);
6878e3e3a7aSWarner Losh }
6888e3e3a7aSWarner Losh 
6898e3e3a7aSWarner Losh 
6908e3e3a7aSWarner Losh static int f_seek (lua_State *L) {
6918e3e3a7aSWarner Losh   static const int mode[] = {SEEK_SET, SEEK_CUR, SEEK_END};
6928e3e3a7aSWarner Losh   static const char *const modenames[] = {"set", "cur", "end", NULL};
6938e3e3a7aSWarner Losh   FILE *f = tofile(L);
6948e3e3a7aSWarner Losh   int op = luaL_checkoption(L, 2, "cur", modenames);
6958e3e3a7aSWarner Losh   lua_Integer p3 = luaL_optinteger(L, 3, 0);
6968e3e3a7aSWarner Losh   l_seeknum offset = (l_seeknum)p3;
6978e3e3a7aSWarner Losh   luaL_argcheck(L, (lua_Integer)offset == p3, 3,
6988e3e3a7aSWarner Losh                   "not an integer in proper range");
6998e3e3a7aSWarner Losh   op = l_fseek(f, offset, mode[op]);
7008e3e3a7aSWarner Losh   if (op)
7018e3e3a7aSWarner Losh     return luaL_fileresult(L, 0, NULL);  /* error */
7028e3e3a7aSWarner Losh   else {
7038e3e3a7aSWarner Losh     lua_pushinteger(L, (lua_Integer)l_ftell(f));
7048e3e3a7aSWarner Losh     return 1;
7058e3e3a7aSWarner Losh   }
7068e3e3a7aSWarner Losh }
7078e3e3a7aSWarner Losh 
7088e3e3a7aSWarner Losh 
7098e3e3a7aSWarner Losh static int f_setvbuf (lua_State *L) {
7108e3e3a7aSWarner Losh   static const int mode[] = {_IONBF, _IOFBF, _IOLBF};
7118e3e3a7aSWarner Losh   static const char *const modenames[] = {"no", "full", "line", NULL};
7128e3e3a7aSWarner Losh   FILE *f = tofile(L);
7138e3e3a7aSWarner Losh   int op = luaL_checkoption(L, 2, NULL, modenames);
7148e3e3a7aSWarner Losh   lua_Integer sz = luaL_optinteger(L, 3, LUAL_BUFFERSIZE);
7158e3e3a7aSWarner Losh   int res = setvbuf(f, NULL, mode[op], (size_t)sz);
7168e3e3a7aSWarner Losh   return luaL_fileresult(L, res == 0, NULL);
7178e3e3a7aSWarner Losh }
7188e3e3a7aSWarner Losh 
7198e3e3a7aSWarner Losh 
7208e3e3a7aSWarner Losh 
7218e3e3a7aSWarner Losh static int io_flush (lua_State *L) {
7228e3e3a7aSWarner Losh   return luaL_fileresult(L, fflush(getiofile(L, IO_OUTPUT)) == 0, NULL);
7238e3e3a7aSWarner Losh }
7248e3e3a7aSWarner Losh 
7258e3e3a7aSWarner Losh 
7268e3e3a7aSWarner Losh static int f_flush (lua_State *L) {
7278e3e3a7aSWarner Losh   return luaL_fileresult(L, fflush(tofile(L)) == 0, NULL);
7288e3e3a7aSWarner Losh }
7298e3e3a7aSWarner Losh 
7308e3e3a7aSWarner Losh 
7318e3e3a7aSWarner Losh /*
7328e3e3a7aSWarner Losh ** functions for 'io' library
7338e3e3a7aSWarner Losh */
7348e3e3a7aSWarner Losh static const luaL_Reg iolib[] = {
7358e3e3a7aSWarner Losh   {"close", io_close},
7368e3e3a7aSWarner Losh   {"flush", io_flush},
7378e3e3a7aSWarner Losh   {"input", io_input},
7388e3e3a7aSWarner Losh   {"lines", io_lines},
7398e3e3a7aSWarner Losh   {"open", io_open},
7408e3e3a7aSWarner Losh   {"output", io_output},
7418e3e3a7aSWarner Losh   {"popen", io_popen},
7428e3e3a7aSWarner Losh   {"read", io_read},
7438e3e3a7aSWarner Losh   {"tmpfile", io_tmpfile},
7448e3e3a7aSWarner Losh   {"type", io_type},
7458e3e3a7aSWarner Losh   {"write", io_write},
7468e3e3a7aSWarner Losh   {NULL, NULL}
7478e3e3a7aSWarner Losh };
7488e3e3a7aSWarner Losh 
7498e3e3a7aSWarner Losh 
7508e3e3a7aSWarner Losh /*
7518e3e3a7aSWarner Losh ** methods for file handles
7528e3e3a7aSWarner Losh */
753*0495ed39SKyle Evans static const luaL_Reg meth[] = {
7548e3e3a7aSWarner Losh   {"read", f_read},
7558e3e3a7aSWarner Losh   {"write", f_write},
756*0495ed39SKyle Evans   {"lines", f_lines},
757*0495ed39SKyle Evans   {"flush", f_flush},
758*0495ed39SKyle Evans   {"seek", f_seek},
759*0495ed39SKyle Evans   {"close", f_close},
760*0495ed39SKyle Evans   {"setvbuf", f_setvbuf},
761*0495ed39SKyle Evans   {NULL, NULL}
762*0495ed39SKyle Evans };
763*0495ed39SKyle Evans 
764*0495ed39SKyle Evans 
765*0495ed39SKyle Evans /*
766*0495ed39SKyle Evans ** metamethods for file handles
767*0495ed39SKyle Evans */
768*0495ed39SKyle Evans static const luaL_Reg metameth[] = {
769*0495ed39SKyle Evans   {"__index", NULL},  /* place holder */
7708e3e3a7aSWarner Losh   {"__gc", f_gc},
771*0495ed39SKyle Evans   {"__close", f_gc},
7728e3e3a7aSWarner Losh   {"__tostring", f_tostring},
7738e3e3a7aSWarner Losh   {NULL, NULL}
7748e3e3a7aSWarner Losh };
7758e3e3a7aSWarner Losh 
7768e3e3a7aSWarner Losh 
7778e3e3a7aSWarner Losh static void createmeta (lua_State *L) {
778*0495ed39SKyle Evans   luaL_newmetatable(L, LUA_FILEHANDLE);  /* metatable for file handles */
779*0495ed39SKyle Evans   luaL_setfuncs(L, metameth, 0);  /* add metamethods to new metatable */
780*0495ed39SKyle Evans   luaL_newlibtable(L, meth);  /* create method table */
781*0495ed39SKyle Evans   luaL_setfuncs(L, meth, 0);  /* add file methods to method table */
782*0495ed39SKyle Evans   lua_setfield(L, -2, "__index");  /* metatable.__index = method table */
783*0495ed39SKyle Evans   lua_pop(L, 1);  /* pop metatable */
7848e3e3a7aSWarner Losh }
7858e3e3a7aSWarner Losh 
7868e3e3a7aSWarner Losh 
7878e3e3a7aSWarner Losh /*
7888e3e3a7aSWarner Losh ** function to (not) close the standard files stdin, stdout, and stderr
7898e3e3a7aSWarner Losh */
7908e3e3a7aSWarner Losh static int io_noclose (lua_State *L) {
7918e3e3a7aSWarner Losh   LStream *p = tolstream(L);
7928e3e3a7aSWarner Losh   p->closef = &io_noclose;  /* keep file opened */
793*0495ed39SKyle Evans   luaL_pushfail(L);
7948e3e3a7aSWarner Losh   lua_pushliteral(L, "cannot close standard file");
7958e3e3a7aSWarner Losh   return 2;
7968e3e3a7aSWarner Losh }
7978e3e3a7aSWarner Losh 
7988e3e3a7aSWarner Losh 
7998e3e3a7aSWarner Losh static void createstdfile (lua_State *L, FILE *f, const char *k,
8008e3e3a7aSWarner Losh                            const char *fname) {
8018e3e3a7aSWarner Losh   LStream *p = newprefile(L);
8028e3e3a7aSWarner Losh   p->f = f;
8038e3e3a7aSWarner Losh   p->closef = &io_noclose;
8048e3e3a7aSWarner Losh   if (k != NULL) {
8058e3e3a7aSWarner Losh     lua_pushvalue(L, -1);
8068e3e3a7aSWarner Losh     lua_setfield(L, LUA_REGISTRYINDEX, k);  /* add file to registry */
8078e3e3a7aSWarner Losh   }
8088e3e3a7aSWarner Losh   lua_setfield(L, -2, fname);  /* add file to module */
8098e3e3a7aSWarner Losh }
8108e3e3a7aSWarner Losh 
8118e3e3a7aSWarner Losh 
8128e3e3a7aSWarner Losh LUAMOD_API int luaopen_io (lua_State *L) {
8138e3e3a7aSWarner Losh   luaL_newlib(L, iolib);  /* new module */
8148e3e3a7aSWarner Losh   createmeta(L);
8158e3e3a7aSWarner Losh   /* create (and set) default files */
8168e3e3a7aSWarner Losh   createstdfile(L, stdin, IO_INPUT, "stdin");
8178e3e3a7aSWarner Losh   createstdfile(L, stdout, IO_OUTPUT, "stdout");
8188e3e3a7aSWarner Losh   createstdfile(L, stderr, NULL, "stderr");
8198e3e3a7aSWarner Losh   return 1;
8208e3e3a7aSWarner Losh }
8218e3e3a7aSWarner Losh 
822