浏览代码

Fixed bug: invalid mode can crash 'io.popen'

Roberto Ierusalimschy 5 年之前
父节点
当前提交
092f66fb05
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      liolib.c

+ 2 - 0
liolib.c

@@ -277,6 +277,8 @@ static int io_popen (lua_State *L) {
   const char *filename = luaL_checkstring(L, 1);
   const char *mode = luaL_optstring(L, 2, "r");
   LStream *p = newprefile(L);
+  luaL_argcheck(L, ((mode[0] == 'r' || mode[0] == 'w') && mode[1] == '\0'),
+                   2, "invalid mode");
   p->f = l_popen(L, filename, mode);
   p->closef = &io_pclose;
   return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;