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 }