| /* |
| ** $Id: lua.c,v 1.36 2000/03/30 17:19:48 roberto Exp roberto $ |
| ** Lua stand-alone interpreter |
| ** See Copyright Notice in lua.h |
| */ |
| |
| |
| #include <signal.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include "lua.h" |
| #include "luadebug.h" |
| #include "lualib.h" |
| |
| |
| #ifndef PROMPT |
| #define PROMPT "> " |
| #endif |
| |
| #ifdef _POSIX_SOURCE |
| #include <unistd.h> |
| #else |
| static int isatty (int x) { return x==0; } /* assume stdin is a tty */ |
| #endif |
| |
| typedef void (*handler)(int); /* type for signal actions */ |
| |
| static void laction (int i); |
| |
| |
| static lua_Hook old_linehook = NULL; |
| static lua_Hook old_callhook = NULL; |
| |
| |
| static handler lreset (void) { |
| return signal(SIGINT, laction); |
| } |
| |
| |
| static void lstop (void) { |
| lua_setlinehook(lua_state, old_linehook); |
| lua_setcallhook(lua_state, old_callhook); |
| lreset(); |
| lua_error("interrupted!"); |
| } |
| |
| |
| static void laction (int i) { |
| (void)i; /* to avoid warnings */ |
| signal(SIGINT, SIG_DFL); /* if another SIGINT happens before lstop, |
| terminate process (default action) */ |
| old_linehook = lua_setlinehook(lua_state, (lua_Hook)lstop); |
| old_callhook = lua_setcallhook(lua_state, (lua_Hook)lstop); |
| } |
| |
| |
| static int ldo (int (*f)(lua_State *L, const char *), const char *name) { |
| int res; |
| handler h = lreset(); |
| res = f(lua_state, name); /* dostring | dofile */ |
| signal(SIGINT, h); /* restore old action */ |
| return res; |
| } |
| |
| |
| static void print_message (void) { |
| fprintf(stderr, |
| "usage: lua [options]. Available options are:\n" |
| " - execute stdin as a file\n" |
| " -d turn debug on\n" |
| " -e stat execute string `stat'\n" |
| " -f name execute file `name' with remaining arguments in table `arg'\n" |
| " -i enter interactive mode with prompt\n" |
| " -q enter interactive mode without prompt\n" |
| " -sNUM set stack size to NUM (must be first option)\n" |
| " -v print version information\n" |
| " a=b set global `a' to string `b'\n" |
| " name execute file `name'\n" |
| ); |
| } |
| |
| |
| static void print_version (void) { |
| printf("%s %s\n", LUA_VERSION, LUA_COPYRIGHT); |
| } |
| |
| |
| static void assign (char *arg) { |
| char *eq = strchr(arg, '='); |
| *eq = '\0'; /* spilt `arg' in two strings (name & value) */ |
| lua_pushstring(eq+1); |
| lua_setglobal(arg); |
| } |
| |
| |
| static lua_Object getargs (char *argv[]) { |
| lua_Object args = lua_createtable(); |
| int i; |
| for (i=0; argv[i]; i++) { |
| /* arg[i] = argv[i] */ |
| lua_pushobject(args); lua_pushnumber(i); |
| lua_pushstring(argv[i]); lua_settable(); |
| } |
| /* arg.n = maximum index in table `arg' */ |
| lua_pushobject(args); lua_pushstring("n"); |
| lua_pushnumber(i-1); lua_settable(); |
| return args; |
| } |
| |
| |
| static void l_getargs (void) { |
| char **argv = (char **)lua_getuserdata(lua_getparam(1)); |
| lua_pushobject(getargs(argv)); |
| } |
| |
| |
| static void file_input (const char *argv) { |
| int result = ldo(lua_dofile, argv); |
| if (result) { |
| if (result == 2) { |
| fprintf(stderr, "lua: cannot execute file "); |
| perror(argv); |
| } |
| exit(1); |
| } |
| } |
| |
| |
| static void manual_input (int version, int prompt) { |
| int cont = 1; |
| if (version) print_version(); |
| while (cont) { |
| char buffer[BUFSIZ]; |
| int i = 0; |
| lua_beginblock(); |
| if (prompt) { |
| const char *s = lua_getstring(lua_getglobal("_PROMPT")); |
| if (!s) s = PROMPT; |
| fputs(s, stdout); |
| } |
| for(;;) { |
| int c = getchar(); |
| if (c == EOF) { |
| cont = 0; |
| break; |
| } |
| else if (c == '\n') { |
| if (i>0 && buffer[i-1] == '\\') |
| buffer[i-1] = '\n'; |
| else break; |
| } |
| else if (i >= BUFSIZ-1) { |
| fprintf(stderr, "lua: input line too long\n"); |
| break; |
| } |
| else buffer[i++] = (char)c; |
| } |
| buffer[i] = '\0'; |
| ldo(lua_dostring, buffer); |
| lua_endblock(); |
| } |
| printf("\n"); |
| } |
| |
| |
| int main (int argc, char *argv[]) { |
| int i = 1; |
| if (i < argc && argv[1][0] == '-' && argv[1][1] == 's') { |
| int stacksize = atoi(&argv[1][2]); |
| if (stacksize == 0) { |
| fprintf(stderr, "lua: invalid stack size ('%s')\n", &argv[1][2]); |
| exit(1); |
| } |
| i++; |
| lua_state = lua_newstate("stack", stacksize, NULL); |
| } |
| else |
| lua_state = lua_newstate(NULL); |
| lua_userinit(); |
| lua_pushuserdata(argv); |
| lua_pushcclosure(l_getargs, 1); |
| lua_setglobal("getargs"); |
| if (i >= argc) { /* no other arguments? */ |
| if (isatty(0)) { |
| manual_input(1, 1); |
| } |
| else |
| ldo(lua_dofile, NULL); /* executes stdin as a file */ |
| } |
| else for (; i<argc; i++) { |
| if (argv[i][0] == '-') { /* option? */ |
| switch (argv[i][1]) { |
| case 0: |
| ldo(lua_dofile, NULL); /* executes stdin as a file */ |
| break; |
| case 'i': |
| manual_input(0, 1); |
| break; |
| case 'q': |
| manual_input(0, 0); |
| break; |
| case 'd': |
| lua_setdebug(lua_state, 1); |
| if (i+1 >= argc) { /* last argument? */ |
| manual_input(1, 1); |
| } |
| break; |
| case 'v': |
| print_version(); |
| break; |
| case 'e': |
| i++; |
| if (i >= argc) { |
| print_message(); |
| exit(1); |
| } |
| if (ldo(lua_dostring, argv[i]) != 0) { |
| fprintf(stderr, "lua: error running argument `%s'\n", argv[i]); |
| exit(1); |
| } |
| break; |
| case 'f': |
| i++; |
| if (i >= argc) { |
| print_message(); |
| exit(1); |
| } |
| lua_pushobject(getargs(argv+i)); /* collect remaining arguments */ |
| lua_setglobal("arg"); |
| file_input(argv[i]); |
| goto endloop; /* stop scanning arguments */ |
| break; |
| case 's': |
| fprintf(stderr, "lua: stack size (`-s') must be the first option\n"); |
| exit(1); |
| default: |
| print_message(); |
| exit(1); |
| } |
| } |
| else if (strchr(argv[i], '=')) |
| assign(argv[i]); |
| else |
| file_input(argv[i]); |
| } |
| endloop: |
| #ifdef DEBUG |
| lua_close(); |
| #endif |
| return 0; |
| } |
| |