Browse Source

print_stack now gets a parameter (file to print);

small changes.
v5-2
Roberto Ierusalimschy 29 years ago
parent
commit
f9a9bd77e4
  1. 179
      iolib.c

179
iolib.c

@ -3,7 +3,7 @@
** Input/output library to LUA
*/
char *rcs_iolib="$Id: iolib.c,v 1.25 1995/10/23 13:53:48 roberto Exp roberto $";
char *rcs_iolib="$Id: iolib.c,v 1.26 1995/10/26 14:21:56 roberto Exp roberto $";
#include <stdio.h>
#include <ctype.h>
@ -25,6 +25,22 @@ static FILE *in=stdin, *out=stdout;
#define pclose(x) (-1)
#endif
static void str_error(char *funcname)
{
char buff[250];
sprintf(buff, "incorrect arguments to function `%s'", funcname);
lua_error(buff);
}
static char *check_and_get_string (int numArg, char *funcname)
{
lua_Object o = lua_getparam(numArg);
if (!(lua_isstring(o) || lua_isnumber(o)))
str_error(funcname);
return lua_getstring(o);
}
static void closeread (void)
{
if (in != stdin)
@ -55,34 +71,23 @@ static void closewrite (void)
*/
static void io_readfrom (void)
{
lua_Object o = lua_getparam (1);
if (o == LUA_NOOBJECT) /* restore standart input */
{
if (lua_getparam (1) == LUA_NOOBJECT)
{ /* restore standart input */
closeread();
lua_pushnumber (1);
}
else
{
if (!lua_isstring (o))
{
lua_error ("incorrect argument to function 'readfrom`");
lua_pushnumber (0);
}
else
{
char *s = lua_getstring(o);
char *s = check_and_get_string(1, "readfrom");
FILE *fp = (*s == '|') ? popen(s+1, "r") : fopen(s, "r");
if (fp == NULL)
{
lua_pushnumber (0);
}
else
{
closeread();
in = fp;
lua_pushnumber (1);
}
}
}
}
@ -97,34 +102,23 @@ static void io_readfrom (void)
*/
static void io_writeto (void)
{
lua_Object o = lua_getparam (1);
if (o == LUA_NOOBJECT) /* restore standart output */
if (lua_getparam (1) == LUA_NOOBJECT) /* restore standart output */
{
closewrite();
lua_pushnumber (1);
}
else
{
if (!lua_isstring (o))
{
lua_error ("incorrect argument to function 'writeto`");
lua_pushnumber (0);
}
else
{
char *s = lua_getstring(o);
char *s = check_and_get_string(1, "writeto");
FILE *fp = (*s == '|') ? popen(s+1,"w") : fopen(s,"w");
if (fp == NULL)
{
lua_pushnumber (0);
}
else
{
closewrite();
out = fp;
lua_pushnumber (1);
}
}
}
}
@ -140,30 +134,17 @@ static void io_writeto (void)
*/
static void io_appendto (void)
{
lua_Object o = lua_getparam (1);
if (!lua_isstring (o))
{
lua_error ("incorrect argument to function 'appendto`");
lua_pushnumber (0);
}
char *s = check_and_get_string(1, "appendto");
struct stat st;
int r = (stat(s, &st) == -1) ? 1 : 2;
FILE *fp = fopen (s, "a");
if (fp == NULL)
lua_pushnumber (0);
else
{
int r;
FILE *fp;
struct stat st;
if (stat(lua_getstring(o), &st) == -1) r = 1;
else r = 2;
fp = fopen (lua_getstring(o),"a");
if (fp == NULL)
{
lua_pushnumber (0);
}
else
{
if (out != stdout) fclose (out);
out = fp;
lua_pushnumber (r);
}
if (out != stdout) fclose (out);
out = fp;
lua_pushnumber (r);
}
}
@ -185,30 +166,17 @@ static void io_appendto (void)
static void io_read (void)
{
lua_Object o = lua_getparam (1);
if (o == LUA_NOOBJECT || !lua_isstring(o)) /* free format */
if (o == LUA_NOOBJECT) /* free format */
{
int c;
char s[256];
while (isspace(c=fgetc(in)))
;
if (c == '\"')
if (c == '\"' || c == '\'')
{
int del = c;
int n=0;
while((c = fgetc(in)) != '\"')
{
if (c == EOF)
{
lua_pushnil ();
return;
}
s[n++] = c;
}
s[n] = 0;
}
else if (c == '\'')
{
int n=0;
while((c = fgetc(in)) != '\'')
while((c = fgetc(in)) != del)
{
if (c == EOF)
{
@ -239,7 +207,7 @@ static void io_read (void)
}
else /* formatted */
{
char *e = lua_getstring(o);
char *e = check_and_get_string(1, "read");
char t;
int m=0;
while (isspace(*e)) e++;
@ -483,18 +451,7 @@ static void io_write (void)
*/
static void io_execute (void)
{
lua_Object o = lua_getparam (1);
if (o == LUA_NOOBJECT || !lua_isstring (o))
{
lua_error ("incorrect argument to function 'execute`");
lua_pushnumber (0);
}
else
{
int res = system(lua_getstring(o));
lua_pushnumber (res);
}
return;
lua_pushnumber(system(check_and_get_string(1, "execute")));
}
/*
@ -503,20 +460,10 @@ static void io_execute (void)
*/
static void io_remove (void)
{
lua_Object o = lua_getparam (1);
if (o == LUA_NOOBJECT || !lua_isstring (o))
{
lua_error ("incorrect argument to function 'execute`");
lua_pushnumber (0);
}
else
{
if (remove(lua_getstring(o)) == 0)
if (remove(check_and_get_string(1, "remove")) == 0)
lua_pushnumber (1);
else
else
lua_pushnumber (0);
}
return;
}
@ -525,15 +472,9 @@ static void io_remove (void)
*/
static void io_getenv (void)
{
lua_Object s = lua_getparam(1);
if (!lua_isstring(s))
lua_pushnil();
else
{
char *env = getenv(lua_getstring(s));
if (env == NULL) lua_pushnil();
else lua_pushstring(env);
}
char *env = getenv(check_and_get_string(1, "getenv"));
if (env == NULL) lua_pushnil();
else lua_pushstring(env);
}
/*
@ -602,26 +543,23 @@ static void io_debug (void)
}
static void print_message (void)
void lua_printstack (FILE *f)
{
lua_Object o = lua_getparam(1);
char *s = lua_isstring(o) ? lua_getstring(o) : "(no messsage)";
int level = 0;
lua_Object func;
fprintf(stderr, "lua: %s\n", s);
fprintf(stderr, "Active Stack:\n");
fprintf(f, "Active Stack:\n");
while ((func = lua_stackedfunction(level++)) != LUA_NOOBJECT)
{
char *name;
int currentline;
fprintf(stderr, "\t");
fprintf(f, "\t");
switch (*getobjname(func, &name))
{
case 'g':
fprintf(stderr, "function %s", name);
fprintf(f, "function %s", name);
break;
case 'f':
fprintf(stderr, "fallback %s", name);
fprintf(f, "fallback %s", name);
break;
default:
{
@ -629,19 +567,29 @@ static void print_message (void)
int linedefined;
lua_funcinfo(func, &filename, &linedefined);
if (linedefined == 0)
fprintf(stderr, "main of %s", filename);
fprintf(f, "main of %s", filename);
else if (linedefined < 0)
fprintf(stderr, "%s", filename);
fprintf(f, "%s", filename);
else
fprintf(stderr, "function (%s:%d)", filename, linedefined);
fprintf(f, "function (%s:%d)", filename, linedefined);
}
}
if ((currentline = lua_currentline(func)) > 0)
fprintf(stderr, " at line %d", currentline);
fprintf(stderr, "\n");
fprintf(f, " at line %d", currentline);
fprintf(f, "\n");
}
}
static void errorfb (void)
{
lua_Object o = lua_getparam(1);
char *s = lua_isstring(o) ? lua_getstring(o) : "(no messsage)";
fprintf(stderr, "lua: %s\n", s);
lua_printstack(stderr);
}
/*
** Open io library
*/
@ -661,5 +609,6 @@ void iolib_open (void)
lua_register ("beep", io_beep);
lua_register ("exit", io_exit);
lua_register ("debug", io_debug);
lua_setfallback("error", print_message);
lua_register ("print_stack", errorfb);
lua_setfallback("error", errorfb);
}

Loading…
Cancel
Save