1 files changed, 30 insertions, 13 deletions

diff --git a/lauxlib.c b/lauxlib.c
index 820e8e93..abb28624 100644
--- a/lauxlib.c
+++ b/lauxlib.c
@@ -1,5 +1,5 @@
 /*
-** $Id: lauxlib.c,v 1.213 2010/05/18 17:32:19 roberto Exp roberto $
+** $Id: lauxlib.c,v 1.214 2010/05/31 16:34:19 roberto Exp roberto $
 ** Auxiliary functions for building Lua libraries
 ** See Copyright Notice in lua.h
 */
@@ -476,7 +476,7 @@ LUALIB_API void luaL_unref (lua_State *L, int t, int ref) {
 */
 
 typedef struct LoadF {
-  int first;  /* pre-read character */
+  int n;  /* number of pre-read characters */
   FILE *f;  /* file being read */
   char buff[LUAL_BUFFERSIZE];  /* area for reading file */
 } LoadF;
@@ -485,10 +485,9 @@ typedef struct LoadF {
 static const char *getF (lua_State *L, void *ud, size_t *size) {
   LoadF *lf = (LoadF *)ud;
   (void)L;
-  if (lf->first != EOF) {  /* first character not read yet? */
-    lf->buff[0] = (char)lf->first;  /* return it */
-    *size = 1;
-    lf->first = EOF;  /* now it has been read */
+  if (lf->n > 0) {  /* are there pre-read characters to be read? */
+    *size = lf->n;  /* return them (chars already in buffer) */
+    lf->n = 0;  /* no more pre-read characters */
   }
   else {  /* read a block from file */
     /* 'fread' can return > 0 *and* set the EOF flag. If next call to
@@ -510,6 +509,23 @@ static int errfile (lua_State *L, const char *what, int fnameindex) {
 }
 
 
+/*
+** reads the first character of file 'f' and skips its first line
+** if it starts with '#'. Returns true if it skipped the first line.
+** In any case, '*cp' has the first "valid" character of the file
+** (after the optional first-line comment).
+*/
+static int skipcomment (FILE *f, int *cp) {
+  int c = *cp = getc(f);
+  if (c == '#') {  /* first line is a comment (Unix exec. file)? */
+    while ((c = getc(f)) != EOF && c != '\n') ;  /* skip first line */
+    *cp = getc(f);  /* skip end-of-line */
+    return 1;  /* there was a comment */
+  }
+  else return 0;  /* no comment */
+}
+
+
 LUALIB_API int luaL_loadfile (lua_State *L, const char *filename) {
   LoadF lf;
   int status, readstatus;
@@ -524,16 +540,17 @@ LUALIB_API int luaL_loadfile (lua_State *L, const char *filename) {
     lf.f = fopen(filename, "r");
     if (lf.f == NULL) return errfile(L, "open", fnameindex);
   }
-  c = getc(lf.f);
-  if (c == '#') {  /* Unix exec. file? */
-    while ((c = getc(lf.f)) != EOF && c != '\n') ;  /* skip first line */
-  }
-  else if (c == LUA_SIGNATURE[0] && filename) {  /* binary file? */
+  lf.n = 0;
+  if (skipcomment(lf.f, &c))  /* read initial portion */
+    lf.buff[lf.n++] = '\n';  /* add line to correct line numbers */
+  if (c == LUA_SIGNATURE[0] && filename) {  /* binary file? */
     lf.f = freopen(filename, "rb", lf.f);  /* reopen in binary mode */
     if (lf.f == NULL) return errfile(L, "reopen", fnameindex);
-    c = getc(lf.f);  /* re-read first character */
+    lf.n = 0;
+    skipcomment(lf.f, &c);  /* re-read initial portion */
   }
-  lf.first = c;  /* 'c' is the first character of the stream */
+  if (c != EOF)
+    lf.buff[lf.n++] = c;  /* 'c' is the first character of the stream */
   status = lua_load(L, getF, &lf, lua_tostring(L, -1));
   readstatus = ferror(lf.f);
   if (filename) fclose(lf.f);  /* close file (even in case of errors) */