mirror of
https://github.com/lua/lua
synced 2024-11-25 22:29:39 +03:00
new options; no more _ALERT; no more getargs
This commit is contained in:
Roberto Ierusalimschy
parent
c51aa6ee33
commit
a960e62c3e
266
lua.c
266
lua.c
@ -1,5 +1,5 @@
|
||||
/*
|
||||
** $Id: lua.c,v 1.93 2002/06/20 20:40:09 roberto Exp roberto $
|
||||
** $Id: lua.c,v 1.94 2002/06/26 16:37:39 roberto Exp roberto $
|
||||
** Lua stand-alone interpreter
|
||||
** See Copyright Notice in lua.h
|
||||
*/
|
||||
@ -45,42 +45,62 @@ static lua_State *L = NULL;
|
||||
static const char *progname;
|
||||
|
||||
|
||||
static lua_Hook old_linehook = NULL;
|
||||
static lua_Hook old_callhook = NULL;
|
||||
static lua_Hook old_hook = NULL;
|
||||
static int old_mask = 0;
|
||||
|
||||
|
||||
static void lstop (void) {
|
||||
lua_setlinehook(L, old_linehook);
|
||||
lua_setcallhook(L, old_callhook);
|
||||
luaL_error(L, "interrupted!");
|
||||
static void lstop (lua_State *l, lua_Debug *ar) {
|
||||
(void)ar; /* unused arg. */
|
||||
lua_sethook(l, old_hook, old_mask);
|
||||
luaL_error(l, "interrupted!");
|
||||
}
|
||||
|
||||
|
||||
static void laction (int i) {
|
||||
signal(i, SIG_DFL); /* if another SIGINT happens before lstop,
|
||||
terminate process (default action) */
|
||||
old_linehook = lua_setlinehook(L, (lua_Hook)lstop);
|
||||
old_callhook = lua_setcallhook(L, (lua_Hook)lstop);
|
||||
old_hook = lua_gethook(L);
|
||||
old_mask = lua_gethookmask(L);
|
||||
lua_sethook(L, lstop, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT(1));
|
||||
}
|
||||
|
||||
|
||||
static void report (int status) {
|
||||
static void print_usage (void) {
|
||||
fprintf(stderr,
|
||||
"usage: %s [options] [prog [args]].\n"
|
||||
"Available options are:\n"
|
||||
" - execute stdin as a file\n"
|
||||
" -e stat execute string `stat'\n"
|
||||
" -i enter interactive mode after executing `prog'\n"
|
||||
" -l name execute file `name'\n"
|
||||
" -v print version information\n"
|
||||
" -- stop handling arguments\n" ,
|
||||
progname);
|
||||
}
|
||||
|
||||
|
||||
static void l_message (const char *pname, const char *msg) {
|
||||
if (pname) fprintf(stderr, "%s: ", pname);
|
||||
fprintf(stderr, "%s\n", msg);
|
||||
}
|
||||
|
||||
|
||||
static void report (int status, int traceback) {
|
||||
const char *msg;
|
||||
if (status) {
|
||||
if (status == LUA_ERRRUN) {
|
||||
if (lua_isstring(L, -2) && lua_isstring(L, -1))
|
||||
lua_concat(L, 2); /* concat error message and traceback */
|
||||
else
|
||||
lua_remove(L, -2); /* lease only traceback on stack */
|
||||
}
|
||||
lua_getglobal(L, "_ALERT");
|
||||
if (lua_isfunction(L, -1)) {
|
||||
lua_pushvalue(L, -2);
|
||||
lua_pcall(L, 1, 0);
|
||||
}
|
||||
else {
|
||||
lua_pop(L, 1);
|
||||
fprintf(stderr, "%s", lua_tostring(L, -1));
|
||||
if (!traceback) lua_pop(L, 1); /* remove traceback */
|
||||
else {
|
||||
if (lua_isstring(L, -2) && lua_isstring(L, -1))
|
||||
lua_concat(L, 2); /* concat error message and traceback */
|
||||
else
|
||||
lua_remove(L, -2); /* leave only traceback on stack */
|
||||
}
|
||||
}
|
||||
msg = lua_tostring(L, -1);
|
||||
if (msg == NULL) msg = "(no message)";
|
||||
l_message(progname, msg);
|
||||
lua_pop(L, 1);
|
||||
}
|
||||
}
|
||||
|
||||
@ -97,74 +117,36 @@ static int lcall (int clear) {
|
||||
}
|
||||
|
||||
|
||||
static void print_usage (void) {
|
||||
fprintf(stderr,
|
||||
"usage: %s [options]. Available options are:\n"
|
||||
" - execute stdin as a file\n"
|
||||
" -c close Lua when exiting\n"
|
||||
" -e stat execute string `stat'\n"
|
||||
" -f name execute file `name' with remaining arguments in table `arg'\n"
|
||||
" -i enter interactive mode\n"
|
||||
" -v print version information\n"
|
||||
" a=b set global `a' to string `b'\n"
|
||||
" name execute file `name'\n",
|
||||
progname);
|
||||
}
|
||||
|
||||
|
||||
static int l_panic (lua_State *l) {
|
||||
(void)l;
|
||||
fputs("unable to recover; exiting\n", stderr);
|
||||
l_message(progname, "unable to recover; exiting");
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
static void print_version (void) {
|
||||
printf("%.80s %.80s\n", LUA_VERSION, LUA_COPYRIGHT);
|
||||
l_message(NULL, LUA_VERSION " " LUA_COPYRIGHT);
|
||||
}
|
||||
|
||||
|
||||
static void assign (const char *arg) {
|
||||
char *eq = strchr(arg, '='); /* arg is `name=value'; find the `=' */
|
||||
lua_pushlstring(L, arg, eq-arg); /* push name */
|
||||
lua_pushstring(L, eq+1); /* push value */
|
||||
lua_settable(L, LUA_GLOBALSINDEX); /* _G.name = value */
|
||||
}
|
||||
|
||||
|
||||
static void getargs (char *argv[]) {
|
||||
static void getargs (char *argv[], int n) {
|
||||
int i;
|
||||
lua_newtable(L);
|
||||
for (i=0; argv[i]; i++) {
|
||||
/* arg[i] = argv[i] */
|
||||
lua_pushnumber(L, i);
|
||||
lua_pushnumber(L, i - n);
|
||||
lua_pushstring(L, argv[i]);
|
||||
lua_rawset(L, -3);
|
||||
}
|
||||
/* arg.n = maximum index in table `arg' */
|
||||
lua_pushliteral(L, "n");
|
||||
lua_pushnumber(L, i-1);
|
||||
lua_pushnumber(L, i-n-1);
|
||||
lua_rawset(L, -3);
|
||||
}
|
||||
|
||||
|
||||
static int l_alert (lua_State *l) {
|
||||
if (progname) fprintf(stderr, "%s: ", progname);
|
||||
fprintf(stderr, "%s\n", luaL_check_string(l, 1));
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
static int l_getargs (lua_State *l) {
|
||||
char **argv = (char **)lua_touserdata(l, lua_upvalueindex(1));
|
||||
getargs(argv);
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
static int docall (int status) {
|
||||
if (status == 0) status = lcall(1);
|
||||
report(status);
|
||||
report(status, 1);
|
||||
return status;
|
||||
}
|
||||
|
||||
@ -258,20 +240,20 @@ static void manual_input (int version) {
|
||||
if (version) print_version();
|
||||
while ((status = load_string()) != -1) {
|
||||
if (status == 0) status = lcall(0);
|
||||
report(status);
|
||||
report(status, 0);
|
||||
if (status == 0 && lua_gettop(L) > 0) { /* any result to print? */
|
||||
lua_getglobal(L, "print");
|
||||
lua_insert(L, 1);
|
||||
lua_pcall(L, lua_gettop(L)-1, 0);
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
fputs("\n", stdout);
|
||||
progname = oldprogname;
|
||||
}
|
||||
|
||||
|
||||
static int handle_argv (char *argv[], int *toclose) {
|
||||
if (*argv == NULL) { /* no more arguments? */
|
||||
static int handle_argv (char *argv[], int *interactive) {
|
||||
if (argv[1] == NULL) { /* no more arguments? */
|
||||
if (isatty(0)) {
|
||||
manual_input(1);
|
||||
}
|
||||
@ -280,85 +262,80 @@ static int handle_argv (char *argv[], int *toclose) {
|
||||
}
|
||||
else { /* other arguments; loop over them */
|
||||
int i;
|
||||
for (i = 0; argv[i] != NULL; i++) {
|
||||
if (argv[i][0] != '-') { /* not an option? */
|
||||
if (strchr(argv[i], '='))
|
||||
assign(argv[i]);
|
||||
else
|
||||
if (file_input(argv[i]))
|
||||
return EXIT_FAILURE; /* stop if file fails */
|
||||
for (i = 1; argv[i] != NULL; i++) {
|
||||
if (argv[i][0] != '-') break; /* not an option? */
|
||||
switch (argv[i][1]) { /* option */
|
||||
case '-': { /* `--' */
|
||||
i++; /* skip this argument */
|
||||
goto endloop; /* stop handling arguments */
|
||||
}
|
||||
else switch (argv[i][1]) { /* option */
|
||||
case '\0': {
|
||||
file_input(NULL); /* executes stdin as a file */
|
||||
break;
|
||||
}
|
||||
case 'i': {
|
||||
manual_input(0);
|
||||
break;
|
||||
}
|
||||
case 'c': {
|
||||
*toclose = 1;
|
||||
break;
|
||||
}
|
||||
case 'v': {
|
||||
print_version();
|
||||
break;
|
||||
}
|
||||
case 'e': {
|
||||
const char *chunk = argv[i] + 2;
|
||||
if (*chunk == '\0') chunk = argv[++i];
|
||||
if (chunk == NULL) {
|
||||
print_usage();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
if (dostring(chunk, "=command line") != 0)
|
||||
return EXIT_FAILURE;
|
||||
break;
|
||||
}
|
||||
case 'f': {
|
||||
const char *filename = argv[i] + 2;
|
||||
if (*filename == '\0') filename = argv[++i];
|
||||
if (filename == NULL) {
|
||||
print_usage();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
getargs(argv+i); /* collect remaining arguments */
|
||||
lua_setglobal(L, "arg");
|
||||
return file_input(filename); /* stop scanning arguments */
|
||||
}
|
||||
case 's': {
|
||||
fprintf(stderr,
|
||||
"%s: option `-s' is deprecated (dynamic stack now)\n",
|
||||
progname);
|
||||
break;
|
||||
}
|
||||
default: {
|
||||
case '\0': {
|
||||
file_input(NULL); /* executes stdin as a file */
|
||||
break;
|
||||
}
|
||||
case 'i': {
|
||||
*interactive = 1;
|
||||
break;
|
||||
}
|
||||
case 'v': {
|
||||
print_version();
|
||||
break;
|
||||
}
|
||||
case 'e': {
|
||||
const char *chunk = argv[i] + 2;
|
||||
if (*chunk == '\0') chunk = argv[++i];
|
||||
if (chunk == NULL) {
|
||||
print_usage();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
if (dostring(chunk, "=<command line>") != 0)
|
||||
return EXIT_FAILURE;
|
||||
break;
|
||||
}
|
||||
case 'l': {
|
||||
const char *filename = argv[i] + 2;
|
||||
if (*filename == '\0') filename = argv[++i];
|
||||
if (filename == NULL) {
|
||||
print_usage();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
if (file_input(filename))
|
||||
return EXIT_FAILURE; /* stop if file fails */
|
||||
break;
|
||||
}
|
||||
case 'c': {
|
||||
l_message(progname, "option `-c' is deprecated");
|
||||
break;
|
||||
}
|
||||
case 's': {
|
||||
l_message(progname, "option `-s' is deprecated");
|
||||
break;
|
||||
}
|
||||
default: {
|
||||
print_usage();
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}
|
||||
} endloop:
|
||||
if (argv[i] != NULL) {
|
||||
const char *filename = argv[i];
|
||||
getargs(argv, i); /* collect remaining arguments */
|
||||
lua_setglobal(L, "arg");
|
||||
return file_input(filename); /* stop scanning arguments */
|
||||
}
|
||||
}
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
static void register_own (char *argv[]) {
|
||||
lua_pushudataval(L, argv);
|
||||
lua_pushcclosure(L, l_getargs, 1);
|
||||
lua_setglobal(L, "getargs");
|
||||
lua_register(L, "_ALERT", l_alert);
|
||||
}
|
||||
|
||||
|
||||
static void openstdlibs (lua_State *l) {
|
||||
lua_baselibopen(l);
|
||||
lua_tablibopen(l);
|
||||
lua_iolibopen(l);
|
||||
lua_strlibopen(l);
|
||||
lua_mathlibopen(l);
|
||||
lua_dblibopen(l);
|
||||
static int openstdlibs (lua_State *l) {
|
||||
return lua_baselibopen(l) +
|
||||
lua_tablibopen(l) +
|
||||
lua_iolibopen(l) +
|
||||
lua_strlibopen(l) +
|
||||
lua_mathlibopen(l) +
|
||||
lua_dblibopen(l) +
|
||||
0;
|
||||
}
|
||||
|
||||
|
||||
@ -374,18 +351,17 @@ static int handle_luainit (void) {
|
||||
|
||||
int main (int argc, char *argv[]) {
|
||||
int status;
|
||||
int toclose = 0;
|
||||
int interactive = 0;
|
||||
(void)argc; /* to avoid warnings */
|
||||
progname = argv[0];
|
||||
L = lua_open(); /* create state */
|
||||
lua_atpanic(L, l_panic);
|
||||
LUA_USERINIT(L); /* open libraries */
|
||||
register_own(argv); /* create own function */
|
||||
lua_pop(L, LUA_USERINIT(L)); /* open libraries, dischard any results */
|
||||
status = handle_luainit();
|
||||
if (status != 0) return status;
|
||||
status = handle_argv(argv+1, &toclose);
|
||||
if (toclose)
|
||||
lua_close(L);
|
||||
status = handle_argv(argv, &interactive);
|
||||
if (status == 0 && interactive) manual_input(0);
|
||||
lua_close(L);
|
||||
return status;
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user