Jelajahi Sumber

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

Roberto Ierusalimschy 5 tahun lalu
induk
melakukan
092f66fb05
1 mengubah file dengan 2 tambahan dan 0 penghapusan
  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;