read patterns are deprecated; new option `read(n)' to read n bytes.

This commit is contained in:
Roberto Ierusalimschy 1999-10-07 17:18:36 -02:00
parent 63ccf42397
commit d435b7acd8
1 changed files with 77 additions and 28 deletions

105
liolib.c
View File

@ -1,10 +1,11 @@
/*
** $Id: liolib.c,v 1.44 1999/08/16 20:52:00 roberto Exp roberto $
** $Id: liolib.c,v 1.45 1999/09/13 19:42:02 roberto Exp roberto $
** Standard I/O (and system) library
** See Copyright Notice in lua.h
*/
#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
@ -221,6 +222,9 @@ static void io_appendto (void) {
*/
#ifdef COMPAT_READPATTERN
/*
** We cannot lookahead without need, because this can lock stdin.
** This flag signals when we need to read a next char.
@ -276,6 +280,12 @@ static int read_pattern (FILE *f, const char *p) {
return (*p == '\0');
}
#else
#define read_pattern(f,p) (lua_error("read patterns are deprecated"), 0)
#endif
static int read_number (FILE *f) {
double d;
@ -287,11 +297,21 @@ static int read_number (FILE *f) {
}
static void read_word (FILE *f) {
int c;
do { c = fgetc(f); } while isspace(c); /* skip spaces */
while (c != EOF && !isspace(c)) {
luaL_addchar(c);
c = fgetc(f);
}
ungetc(c, f);
}
#define HUNK_LINE 256
#define HUNK_FILE BUFSIZ
static int read_line (FILE *f) {
/* equivalent to: return read_pattern(f, "[^\n]*{\n}"); */
int n;
char *b;
do {
@ -306,46 +326,63 @@ static int read_line (FILE *f) {
static void read_file (FILE *f) {
/* equivalent to: return read_pattern(f, ".*"); */
int n;
do {
char *b = luaL_openspace(HUNK_FILE);
n = fread(b, sizeof(char), HUNK_FILE, f);
luaL_addsize(n);
luaL_addsize(n);
} while (n==HUNK_FILE);
}
static int read_chars (FILE *f, int n) {
char *b = luaL_openspace(n);
int n1 = fread(b, sizeof(char), n, f);
luaL_addsize(n1);
return (n == n1);
}
static void io_read (void) {
static const char *const options[] = {"*n", "*l", "*a", ".*", "*w", NULL};
int arg = FIRSTARG;
FILE *f = getfileparam(FINPUT, &arg);
const char *p = luaL_opt_string(arg++, "*l");
do { /* repeat for each part */
lua_Object op = lua_getparam(arg);
do { /* repeat for each part */
long l;
int success;
luaL_resetbuffer();
switch (luaL_findstring(p, options)) {
case 0: /* number */
if (!read_number(f)) return; /* read fails */
continue; /* number is already pushed; avoid the "pushstring" */
case 1: /* line */
success = read_line(f);
break;
case 2: case 3: /* file */
read_file(f);
success = 1; /* always success */
break;
case 4: /* word */
success = read_pattern(f, "{%s*}%S+");
break;
default:
success = read_pattern(f, p);
if (lua_isnumber(op))
success = read_chars(f, lua_getnumber(op));
else {
const char *p = luaL_opt_string(arg, "*l");
if (p[0] != '*')
success = read_pattern(f, p); /* deprecated! */
else {
switch (p[1]) {
case 'n': /* number */
if (!read_number(f)) return; /* read fails */
continue; /* number is already pushed; avoid the "pushstring" */
case 'l': /* line */
success = read_line(f);
break;
case 'a': /* file */
read_file(f);
success = 1; /* always success */
break;
case 'w': /* word */
read_word(f);
success = 0; /* must read something to succeed */
break;
default:
luaL_argerror(arg, "invalid format");
success = 0; /* to avoid warnings */
}
}
}
l = luaL_getsize();
if (!success && l==0) return; /* read fails */
lua_pushlstring(luaL_buffer(), l);
} while ((p = luaL_opt_string(arg++, NULL)) != NULL);
} while ((op = lua_getparam(++arg)) != LUA_NOOBJECT);
}
/* }====================================================== */
@ -355,10 +392,22 @@ static void io_write (void) {
int arg = FIRSTARG;
FILE *f = getfileparam(FOUTPUT, &arg);
int status = 1;
const char *s;
long l;
while ((s = luaL_opt_lstr(arg++, NULL, &l)) != NULL)
status = status && ((long)fwrite(s, sizeof(char), l, f) == l);
lua_Object o;
while ((o = lua_getparam(arg++)) != LUA_NOOBJECT) {
switch (lua_type(o)[2]) {
case 'r': { /* stRing? */
long l = lua_strlen(o);
status = status &&
((long)fwrite(lua_getstring(o), sizeof(char), l, f) == l);
break;
}
case 'm': /* nuMber? */ /* LUA_NUMBER */
/* optimization: could be done exactly as for strings */
status = status && fprintf(f, "%.16g", lua_getnumber(o)) > 0;
break;
default: luaL_argerror(arg-1, "string expected");
}
}
pushresult(status);
}