mirror of https://github.com/lua/lua
print_stack now gets a parameter (file to print);
small changes.
This commit is contained in:
Roberto Ierusalimschy
parent
63b8a6fd20
commit
f9a9bd77e4
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…
Reference in New Issue