new scheme for buffers

author: Roberto Ierusalimschy <roberto@inf.puc-rio.br> 2000-09-11 14:38:42 -0300
committer: Roberto Ierusalimschy <roberto@inf.puc-rio.br> 2000-09-11 14:38:42 -0300
commit: 787a78f83e0484c9e9698189982e2f309808fae8 (patch)
tree: 0682eddf4ea5a49bf5078bac937a36f90057df57 /liolib.c
parent: 70c8a310925d6c41c3ef4f7feeae604a4c9a3a95 (diff)
download: lua-787a78f83e0484c9e9698189982e2f309808fae8.tar.gz
lua-787a78f83e0484c9e9698189982e2f309808fae8.tar.bz2
lua-787a78f83e0484c9e9698189982e2f309808fae8.zip
1 files changed, 53 insertions, 34 deletions
diff --git a/liolib.c b/liolib.c
index 885db9bd..db9f36d6 100644
--- a/liolib.c
+++ b/liolib.c
@@ -1,5 +1,5 @@
 /*
-** $Id: liolib.c,v 1.76 2000/08/31 20:23:40 roberto Exp roberto $
+** $Id: liolib.c,v 1.77 2000/09/05 19:33:32 roberto Exp $
 ** Standard I/O (and system) library
 ** See Copyright Notice in lua.h
 */
@@ -263,7 +263,7 @@ static int read_pattern (lua_State *L, FILE *f, const char *p) {
        if (c == NEED_OTHER) c = getc(f);
        m = (c==EOF) ? 0 : luaI_singlematch(c, p, ep);
        if (m) {
-          if (!inskip) luaL_addchar(L, c);
+          if (!inskip) luaL_putchar(L, c);
          c = NEED_OTHER;
        }
        switch (*ep) {
@@ -274,7 +274,7 @@ static int read_pattern (lua_State *L, FILE *f, const char *p) {
            while (m) {  /* reads the same item until it fails */
              c = getc(f);
              m = (c==EOF) ? 0 : luaI_singlematch(c, p, ep);
-              if (m && !inskip) luaL_addchar(L, c);
+              if (m && !inskip) luaL_putchar(L, c);
            }
            /* go through to continue reading the pattern */
          case '?':  /* optional */
@@ -308,53 +308,74 @@ static int read_number (lua_State *L, FILE *f) {
 }
-static void read_word (lua_State *L, FILE *f) {
+static int read_word (lua_State *L, FILE *f) {
  int c;
+  luaL_Buffer b;
+  luaL_buffinit(L, &b);
  do { c = fgetc(f); } while (isspace(c));  /* skip spaces */
  while (c != EOF && !isspace(c)) {
-    luaL_addchar(L, c);
+    luaL_putchar(&b, c);
    c = fgetc(f);
  }
  ungetc(c, f);
+  luaL_pushresult(&b);  /* close buffer */
+  return (lua_strlen(L, 1) > 0);
 }
-#define HUNK_LINE       256
-#define HUNK_FILE       BUFSIZ
 static int read_line (lua_State *L, FILE *f) {
-  int n;
+  int n = 0;
-  char *b;
+  luaL_Buffer b;
+  luaL_buffinit(L, &b);
  for (;;) {
-    b = luaL_openspace(L, HUNK_LINE);
+    char *p = luaL_prepbuffer(&b);
-    if (!fgets(b, HUNK_LINE, f)) return 0;  /* read fails */
+    if (!fgets(p, LUAL_BUFFERSIZE, f))  /* read fails? */
-    n = strlen(b);
+      break;
-    if (b[n-1] != '\n')
+    n = strlen(p);
-      luaL_addsize(L, n); 
+    if (p[n-1] != '\n')
+      luaL_addsize(&b, n); 
    else {
-      luaL_addsize(L, n-1);  /* do not add the `\n' */
+      luaL_addsize(&b, n-1);  /* do not add the `\n' */
      break;
    }
  }
-  return 1;
+  luaL_pushresult(&b);  /* close buffer */
+  return (n > 0);  /* read something? */
 }
 static void read_file (lua_State *L, FILE *f) {
-  size_t n;
+  size_t len = 0;
-  do {
+  size_t size = BUFSIZ;
-    char *b = luaL_openspace(L, HUNK_FILE);
+  char *buffer = NULL;
-    n = fread(b, sizeof(char), HUNK_FILE, f);
+  for (;;) {
-    luaL_addsize(L, n);
+    buffer = (char *)realloc(buffer, size);
-  } while (n==HUNK_FILE);
+    if (buffer == NULL)
+      lua_error(L, "not enough memory to read a file");
+    len += fread(buffer+len, sizeof(char), size-len, f);
+    if (len < size) break;  /* did not read all it could */
+    size *= 2;
+  }
+  lua_pushlstring(L, buffer, len);
+  free(buffer);
 }
 static int read_chars (lua_State *L, FILE *f, size_t n) {
-  char *b = luaL_openspace(L, n);
+  char *buffer;
-  size_t n1 = fread(b, sizeof(char), n, f);
+  size_t n1;
-  luaL_addsize(L, n1);
+  char statbuff[BUFSIZ];
-  return (n == n1);
+  if (n <= BUFSIZ)
+    buffer = statbuff;
+  else {
+    buffer = (char  *)malloc(n);
+    if (buffer == NULL)
+      lua_error(L, "not enough memory to read a file");
+  }
+  n1 = fread(buffer, sizeof(char), n, f);
+  lua_pushlstring(L, buffer, n1);
+  if (buffer != statbuff) free(buffer);
+  return (n1 > 0 || n == 0);
 }
@@ -375,9 +396,7 @@ static int io_read (lua_State *L) {
  else
    luaL_checkstack(L, lastarg-firstarg+1, "too many arguments");
  for (n = firstarg; n<=lastarg; n++) {
-    size_t l;
    int success;
-    luaL_resetbuffer(L);
    if (lua_isnumber(L, n))
      success = read_chars(L, f, (size_t)lua_tonumber(L, n));
    else {
@@ -397,8 +416,7 @@ static int io_read (lua_State *L) {
            success = 1; /* always success */
            break;
          case 'w':  /* word */
-            read_word(L, f);
+            success = read_word(L, f);
-            success = 0;  /* must read something to succeed */
            break;
          default:
            luaL_argerror(L, n, "invalid format");
@@ -406,9 +424,10 @@ static int io_read (lua_State *L) {
        }
      }
    }
-    l = luaL_getsize(L);
+    if (!success) {
-    if (!success && l==0) break;  /* read fails */
+      lua_pop(L, 1);  /* remove last result */
-    lua_pushlstring(L, luaL_buffer(L), l);
+      break;  /* read fails */
+    }
  } endloop:
  return n - firstarg;
 }
author	Roberto Ierusalimschy <roberto@inf.puc-rio.br>	2000-09-11 14:38:42 -0300
committer	Roberto Ierusalimschy <roberto@inf.puc-rio.br>	2000-09-11 14:38:42 -0300
commit	787a78f83e0484c9e9698189982e2f309808fae8 (patch)
tree	0682eddf4ea5a49bf5078bac937a36f90057df57 /liolib.c
parent	70c8a310925d6c41c3ef4f7feeae604a4c9a3a95 (diff)
download	lua-787a78f83e0484c9e9698189982e2f309808fae8.tar.gz lua-787a78f83e0484c9e9698189982e2f309808fae8.tar.bz2 lua-787a78f83e0484c9e9698189982e2f309808fae8.zip