1 files changed, 26 insertions, 30 deletions

diff --git a/liolib.c b/liolib.c
index 7dc1dab9..0884e9ac 100644
--- a/liolib.c
+++ b/liolib.c
@@ -1,5 +1,5 @@
 /*
-** $Id: liolib.c,v 2.154 2017/11/16 16:28:36 roberto Exp roberto $
+** $Id: liolib.c,v 2.155 2018/02/21 13:48:44 roberto Exp roberto $
 ** Standard I/O (and system) library
 ** See Copyright Notice in lua.h
 */
@@ -528,44 +528,40 @@ static int read_chars (lua_State *L, FILE *f, size_t n) {
 
 static int g_read (lua_State *L, FILE *f, int first) {
   int nargs = lua_gettop(L) - 1;
-  int nresults, success;
+  int n, success;
   clearerr(f);
   if (nargs == 0) {  /* no arguments? */
     success = read_line(L, f, 1);
-    nresults = 1;
+    n = first + 1;  /* to return 1 result */
   }
   else {
+    /* ensure stack space for all results and for auxlib's buffer */
+    luaL_checkstack(L, nargs+LUA_MINSTACK, "too many arguments");
     success = 1;
-    nresults = 0;
-    for (; nargs-- && success; first++) {
-      luaL_checkstack(L, LUA_MINSTACK, "too many arguments");
-      if (lua_type(L, first) == LUA_TNUMBER) {
-        size_t l = (size_t)luaL_checkinteger(L, first);
+    for (n = first; nargs-- && success; n++) {
+      if (lua_type(L, n) == LUA_TNUMBER) {
+        size_t l = (size_t)luaL_checkinteger(L, n);
         success = (l == 0) ? test_eof(L, f) : read_chars(L, f, l);
-        nresults++;
       }
       else {
-        const char *p = luaL_checkstring(L, first);
+        const char *p = luaL_checkstring(L, n);
         if (*p == '*') p++;  /* skip optional '*' (for compatibility) */
-        for (; success && *p != '\0'; p++) {
-          nresults++;
-          switch (*p) {
-            case 'n':  /* number */
-              success = read_number(L, f);
-              break;
-            case 'l':  /* line */
-              success = read_line(L, f, 1);
-              break;
-            case 'L':  /* line with end-of-line */
-              success = read_line(L, f, 0);
-              break;
-            case 'a':  /* file */
-              read_all(L, f);  /* read entire file */
-              success = 1; /* always success */
-              break;
-            default:
-              return luaL_argerror(L, first, "invalid format");
-          }
+        switch (*p) {
+          case 'n':  /* number */
+            success = read_number(L, f);
+            break;
+          case 'l':  /* line */
+            success = read_line(L, f, 1);
+            break;
+          case 'L':  /* line with end-of-line */
+            success = read_line(L, f, 0);
+            break;
+          case 'a':  /* file */
+            read_all(L, f);  /* read entire file */
+            success = 1; /* always success */
+            break;
+          default:
+            return luaL_argerror(L, n, "invalid format");
         }
       }
     }
@@ -576,7 +572,7 @@ static int g_read (lua_State *L, FILE *f, int first) {
     lua_pop(L, 1);  /* remove last result */
     lua_pushnil(L);  /* push nil instead */
   }
-  return nresults;
+  return n - first;
 }