static int io_open (lua_State *L) {
const char *filename = luaL_checkstring(L, 1);
const char *mode = luaL_optstring(L, 2, "r");
LStream *p = newfile(L);
const char *md = mode; /* to traverse/check mode */
luaL_argcheck(L, lua_checkmode(md), 2, "invalid mode");
int flags = 0;
if(*(mode+1) == '+') {
flags = O_RDWR;
switch(*mode) {
case 'w':
flags |= O_TRUNC; break;
case 'a':
flags |= O_APPEND; break;
}
}
else {
switch(*mode) {
case 'r':
flags = O_RDONLY; break;
case 'w':
flags = O_WRONLY | O_TRUNC; break;
case 'a':
flags = O_WRONLY | O_APPEND; break;
}
}
if((*mode == 'w' || *mode == 'a') && !rb->file_exists(filename))
flags |= O_CREAT;
int pf = rb->open(filename, flags, 0666);
p->f = (FILE*)pf;
return (pf < 0) ? luaL_fileresult(L, 0, filename) : 1;
}
static int io_open(lua_State *L) {
const char *filename = luaL_checkstring(L, 1);
const char *mode = luaL_optstring(L, 2, "r");
LStream *p = newfile(L);
const char *md = mode; /* to traverse/check mode */
luaL_argcheck(L, lua_checkmode(md), 2, "invalid mode");
p->f = fopen(filename, mode);
return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;
}
static int io_open (lua_State *L) {
const char *filename = luaL_checkstring(L, 1);
const char *md = luaL_optstring(L, 2, "r");
LStream *p = newfile(L);
#if defined(USE_FATFS)
BYTE mode;
if (strchr(md, 'w') || strchr(md, 'a'))
mode = FA_WRITE;
else
mode = FA_READ;
FRESULT result = f_open(&p->f, filename, mode);
if (result == FR_OK && strchr(md, 'a'))
result = f_lseek(&p->f, f_size(&p->f));
return result == FR_OK ? 1 : 0;
#else
const char *mode = md; /* to traverse/check mode */
luaL_argcheck(L, lua_checkmode(mode), 2, "invalid mode");
p->f = fopen(filename, md);
return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;
#endif
}
