tree: reformat

Done automatically by clang-format
This commit is contained in:
Jeremy Baxter 2024-03-12 12:21:39 +13:00
parent 669eb3fd28
commit 30a1097bb0
8 changed files with 311 additions and 278 deletions

View file

@ -47,7 +47,7 @@ callisto_openall(lua_State *L)
/* for each Callisto library except extra */ /* for each Callisto library except extra */
for (lib = loadedlibs; lib->func; lib++) { for (lib = loadedlibs; lib->func; lib++) {
lua_newtable(L); /* make a new table for the library */ lua_newtable(L); /* make a new table for the library */
lib->func(L); /* load library */ lib->func(L); /* load library */
lua_setglobal(L, lib->name); lua_setglobal(L, lib->name);
} }
/* inject extra into the global environment */ /* inject extra into the global environment */

532
csto.c
View file

@ -4,10 +4,10 @@
* Copyright (c) 2023-2024 Jeremy Baxter. * Copyright (c) 2023-2024 Jeremy Baxter.
*/ */
#include <signal.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <signal.h>
#include <unistd.h> #include <unistd.h>
#include <lua/lauxlib.h> #include <lua/lauxlib.h>
@ -17,145 +17,153 @@
#include "callisto.h" #include "callisto.h"
#if !defined(LUA_PROGNAME) #if !defined(LUA_PROGNAME)
#define LUA_PROGNAME "csto" #define LUA_PROGNAME "csto"
#endif #endif
#if !defined(LUA_INIT_VAR) #if !defined(LUA_INIT_VAR)
#define LUA_INIT_VAR "LUA_INIT" #define LUA_INIT_VAR "LUA_INIT"
#endif #endif
#define LUA_INITVARVERSION LUA_INIT_VAR LUA_VERSUFFIX #define LUA_INITVARVERSION LUA_INIT_VAR LUA_VERSUFFIX
static lua_State *globalL = NULL; static lua_State *globalL = NULL;
static const char *progname = LUA_PROGNAME; static const char *progname = LUA_PROGNAME;
/* /*
* Use 'sigaction' when available. * Use 'sigaction' when available.
*/ */
static void setsignal (int sig, void (*handler)(int)) { static void
setsignal(int sig, void (*handler)(int))
{
struct sigaction sa; struct sigaction sa;
sa.sa_handler = handler; sa.sa_handler = handler;
sa.sa_flags = 0; sa.sa_flags = 0;
sigemptyset(&sa.sa_mask); /* do not mask any signal */ sigemptyset(&sa.sa_mask); /* do not mask any signal */
sigaction(sig, &sa, NULL); sigaction(sig, &sa, NULL);
} }
/* /*
* Hook set by signal function to stop the interpreter. * Hook set by signal function to stop the interpreter.
*/ */
static void lstop (lua_State *L, lua_Debug *ar) { static void
(void)ar; /* unused arg. */ lstop(lua_State *L, lua_Debug *ar)
lua_sethook(L, NULL, 0, 0); /* reset hook */ {
(void)ar; /* unused arg. */
lua_sethook(L, NULL, 0, 0); /* reset hook */
luaL_error(L, "interrupted!"); luaL_error(L, "interrupted!");
} }
/* /*
* Function to be called at a C signal. Because a C signal cannot * Function to be called at a C signal. Because a C signal cannot
* just change a Lua state (as there is no proper synchronization), * just change a Lua state (as there is no proper synchronization),
* this function only sets a hook that, when called, will stop the * this function only sets a hook that, when called, will stop the
* interpreter. * interpreter.
*/ */
static void laction (int i) { static void
laction(int i)
{
int flag = LUA_MASKCALL | LUA_MASKRET | LUA_MASKLINE | LUA_MASKCOUNT; int flag = LUA_MASKCALL | LUA_MASKRET | LUA_MASKLINE | LUA_MASKCOUNT;
setsignal(i, SIG_DFL); /* if another SIGINT happens, terminate process */ setsignal(i, SIG_DFL); /* if another SIGINT happens, terminate process */
lua_sethook(globalL, lstop, flag, 1); lua_sethook(globalL, lstop, flag, 1);
} }
static void
static void print_usage (const char *badoption) { print_usage(const char *badoption)
{
lua_writestringerror("%s: ", progname); lua_writestringerror("%s: ", progname);
if (badoption[1] == 'e' || badoption[1] == 'l') if (badoption[1] == 'e' || badoption[1] == 'l')
lua_writestringerror("'%s' needs argument\n", badoption); lua_writestringerror("'%s' needs argument\n", badoption);
else else
lua_writestringerror("unrecognized option '%s'\n", badoption); lua_writestringerror("unrecognized option '%s'\n", badoption);
lua_writestringerror( lua_writestringerror(
"usage: %s [-EivW] [-e stat] [-l mod|g=mod] [script [args]]\n" "usage: %s [-EivW] [-e stat] [-l mod|g=mod] [script [args]]\n"
"available options are:\n" "available options are:\n"
" -e stat execute string 'stat'\n" " -e stat execute string 'stat'\n"
" -i enter interactive mode after executing 'script'\n" " -i enter interactive mode after executing 'script'\n"
" -l mod require library 'mod' into global 'mod'\n" " -l mod require library 'mod' into global 'mod'\n"
" -l g=mod require library 'mod' into global 'g'\n" " -l g=mod require library 'mod' into global 'g'\n"
" -v show version information\n" " -v show version information\n"
" -E ignore environment variables\n" " -E ignore environment variables\n"
" -W turn warnings on\n" " -W turn warnings on\n"
" -- stop handling options\n" " -- stop handling options\n"
" - stop handling options and execute stdin\n" " - stop handling options and execute stdin\n",
, progname);
progname);
} }
/* /*
* Prints an error message, adding the program name in front of it * Prints an error message, adding the program name in front of it
* (if present) * (if present)
*/ */
static void l_message (const char *pname, const char *msg) { static void
if (pname) lua_writestringerror("%s: ", pname); l_message(const char *pname, const char *msg)
{
if (pname)
lua_writestringerror("%s: ", pname);
lua_writestringerror("%s\n", msg); lua_writestringerror("%s\n", msg);
} }
/* /*
* Check whether 'status' is not OK and, if so, prints the error * Check whether 'status' is not OK and, if so, prints the error
* message on the top of the stack. It assumes that the error object * message on the top of the stack. It assumes that the error object
* is a string, as it was either generated by Lua or by 'msghandler'. * is a string, as it was either generated by Lua or by 'msghandler'.
*/ */
static int report (lua_State *L, int status) { static int
report(lua_State *L, int status)
{
if (status != LUA_OK) { if (status != LUA_OK) {
const char *msg = lua_tostring(L, -1); const char *msg = lua_tostring(L, -1);
l_message(progname, msg); l_message(progname, msg);
lua_pop(L, 1); /* remove message */ lua_pop(L, 1); /* remove message */
} }
return status; return status;
} }
/* /*
* Message handler used to run all chunks * Message handler used to run all chunks
*/ */
static int msghandler (lua_State *L) { static int
msghandler(lua_State *L)
{
const char *msg = lua_tostring(L, 1); const char *msg = lua_tostring(L, 1);
if (msg == NULL) { /* is error object not a string? */ if (msg == NULL) { /* is error object not a string? */
if (luaL_callmeta(L, 1, "__tostring") && /* does it have a metamethod */ if (luaL_callmeta(L, 1, "__tostring") && /* does it have a metamethod */
lua_type(L, -1) == LUA_TSTRING) /* that produces a string? */ lua_type(L, -1) == LUA_TSTRING) /* that produces a string? */
return 1; /* that is the message */ return 1; /* that is the message */
else else
msg = lua_pushfstring(L, "(error object is a %s value)", msg = lua_pushfstring(L, "(error object is a %s value)",
luaL_typename(L, 1)); luaL_typename(L, 1));
} }
luaL_traceback(L, L, msg, 1); /* append a standard traceback */ luaL_traceback(L, L, msg, 1); /* append a standard traceback */
return 1; /* return the traceback */ return 1; /* return the traceback */
} }
/* /*
* Interface to 'lua_pcall', which sets appropriate message function * Interface to 'lua_pcall', which sets appropriate message function
* and C-signal handler. Used to run all chunks. * and C-signal handler. Used to run all chunks.
*/ */
static int docall (lua_State *L, int narg, int nres) { static int
docall(lua_State *L, int narg, int nres)
{
int status; int status;
int base = lua_gettop(L) - narg; /* function index */ int base = lua_gettop(L) - narg; /* function index */
lua_pushcfunction(L, msghandler); /* push message handler */ lua_pushcfunction(L, msghandler); /* push message handler */
lua_insert(L, base); /* put it under function and args */ lua_insert(L, base); /* put it under function and args */
globalL = L; /* to be available to 'laction' */ globalL = L; /* to be available to 'laction' */
setsignal(SIGINT, laction); /* set C-signal handler */ setsignal(SIGINT, laction); /* set C-signal handler */
status = lua_pcall(L, narg, nres, base); status = lua_pcall(L, narg, nres, base);
setsignal(SIGINT, SIG_DFL); /* reset C-signal handler */ setsignal(SIGINT, SIG_DFL); /* reset C-signal handler */
lua_remove(L, base); /* remove message handler from the stack */ lua_remove(L, base); /* remove message handler from the stack */
return status; return status;
} }
static void
static void print_version (void) { print_version(void)
{
lua_writestring(CALLISTO_COPYRIGHT, strlen(CALLISTO_COPYRIGHT)); lua_writestring(CALLISTO_COPYRIGHT, strlen(CALLISTO_COPYRIGHT));
lua_writeline(); lua_writeline();
} }
/* /*
* Create the 'arg' table, which stores all arguments from the * Create the 'arg' table, which stores all arguments from the
* command line ('argv'). It should be aligned so that, at index 0, * command line ('argv'). It should be aligned so that, at index 0,
@ -164,10 +172,13 @@ static void print_version (void) {
* other arguments (before the script name) go to negative indices. * other arguments (before the script name) go to negative indices.
* If there is no script name, assume interpreter's name as base. * If there is no script name, assume interpreter's name as base.
*/ */
static void createargtable (lua_State *L, char **argv, int argc, int script) { static void
createargtable(lua_State *L, char **argv, int argc, int script)
{
int i, narg; int i, narg;
if (script == argc) script = 0; /* no script name? */ if (script == argc)
narg = argc - (script + 1); /* number of positive indices */ script = 0; /* no script name? */
narg = argc - (script + 1); /* number of positive indices */
lua_createtable(L, narg, script + 1); lua_createtable(L, narg, script + 1);
for (i = 0; i < argc; i++) { for (i = 0; i < argc; i++) {
lua_pushstring(L, argv[i]); lua_pushstring(L, argv[i]);
@ -176,48 +187,54 @@ static void createargtable (lua_State *L, char **argv, int argc, int script) {
lua_setglobal(L, "arg"); lua_setglobal(L, "arg");
} }
static int
static int dochunk (lua_State *L, int status) { dochunk(lua_State *L, int status)
if (status == LUA_OK) status = docall(L, 0, 0); {
if (status == LUA_OK)
status = docall(L, 0, 0);
return report(L, status); return report(L, status);
} }
static int
static int dofile (lua_State *L, const char *name) { dofile(lua_State *L, const char *name)
{
return dochunk(L, luaL_loadfile(L, name)); return dochunk(L, luaL_loadfile(L, name));
} }
static int
static int dostring (lua_State *L, const char *s, const char *name) { dostring(lua_State *L, const char *s, const char *name)
{
return dochunk(L, luaL_loadbuffer(L, s, strlen(s), name)); return dochunk(L, luaL_loadbuffer(L, s, strlen(s), name));
} }
/* /*
* Receives 'globname[=modname]' and runs 'globname = require(modname)'. * Receives 'globname[=modname]' and runs 'globname = require(modname)'.
*/ */
static int dolibrary (lua_State *L, char *globname) { static int
dolibrary(lua_State *L, char *globname)
{
int status; int status;
char *modname = strchr(globname, '='); char *modname = strchr(globname, '=');
if (modname == NULL) /* no explicit name? */ if (modname == NULL) /* no explicit name? */
modname = globname; /* module name is equal to global name */ modname = globname; /* module name is equal to global name */
else { else {
*modname = '\0'; /* global name ends here */ *modname = '\0'; /* global name ends here */
modname++; /* module name starts after the '=' */ modname++; /* module name starts after the '=' */
} }
lua_getglobal(L, "require"); lua_getglobal(L, "require");
lua_pushstring(L, modname); lua_pushstring(L, modname);
status = docall(L, 1, 1); /* call 'require(modname)' */ status = docall(L, 1, 1); /* call 'require(modname)' */
if (status == LUA_OK) if (status == LUA_OK)
lua_setglobal(L, globname); /* globname = require(modname) */ lua_setglobal(L, globname); /* globname = require(modname) */
return report(L, status); return report(L, status);
} }
/* /*
* Push on the stack the contents of table 'arg' from 1 to #arg * Push on the stack the contents of table 'arg' from 1 to #arg
*/ */
static int pushargs (lua_State *L) { static int
pushargs(lua_State *L)
{
int i, n; int i, n;
if (lua_getglobal(L, "arg") != LUA_TTABLE) if (lua_getglobal(L, "arg") != LUA_TTABLE)
luaL_error(L, "'arg' is not a table"); luaL_error(L, "'arg' is not a table");
@ -225,32 +242,31 @@ static int pushargs (lua_State *L) {
luaL_checkstack(L, n + 3, "too many arguments to script"); luaL_checkstack(L, n + 3, "too many arguments to script");
for (i = 1; i <= n; i++) for (i = 1; i <= n; i++)
lua_rawgeti(L, -i, i); lua_rawgeti(L, -i, i);
lua_remove(L, -i); /* remove table from the stack */ lua_remove(L, -i); /* remove table from the stack */
return n; return n;
} }
static int
static int handle_script (lua_State *L, char **argv) { handle_script(lua_State *L, char **argv)
{
int status; int status;
const char *fname = argv[0]; const char *fname = argv[0];
if (strcmp(fname, "-") == 0 && strcmp(argv[-1], "--") != 0) if (strcmp(fname, "-") == 0 && strcmp(argv[-1], "--") != 0)
fname = NULL; /* stdin */ fname = NULL; /* stdin */
status = luaL_loadfile(L, fname); status = luaL_loadfile(L, fname);
if (status == LUA_OK) { if (status == LUA_OK) {
int n = pushargs(L); /* push arguments to script */ int n = pushargs(L); /* push arguments to script */
status = docall(L, n, LUA_MULTRET); status = docall(L, n, LUA_MULTRET);
} }
return report(L, status); return report(L, status);
} }
/* bits of various argument indicators in 'args' */ /* bits of various argument indicators in 'args' */
#define has_error 1 /* bad option */ #define has_error 1 /* bad option */
#define has_i 2 /* -i */ #define has_i 2 /* -i */
#define has_v 4 /* -v */ #define has_v 4 /* -v */
#define has_e 8 /* -e */ #define has_e 8 /* -e */
#define has_E 16 /* -E */ #define has_E 16 /* -E */
/* /*
* Traverses all arguments from 'argv', returning a mask with those * Traverses all arguments from 'argv', returning a mask with those
@ -258,101 +274,107 @@ static int handle_script (lua_State *L, char **argv) {
* any invalid argument). 'first' returns the first not-handled argument * any invalid argument). 'first' returns the first not-handled argument
* (either the script name or a bad argument in case of error). * (either the script name or a bad argument in case of error).
*/ */
static int collectargs (char **argv, int *first) { static int
collectargs(char **argv, int *first)
{
int args = 0; int args = 0;
int i; int i;
for (i = 1; argv[i] != NULL; i++) { for (i = 1; argv[i] != NULL; i++) {
*first = i; *first = i;
if (argv[i][0] != '-') /* not an option? */ if (argv[i][0] != '-') /* not an option? */
return args; /* stop handling options */ return args; /* stop handling options */
switch (argv[i][1]) { /* else check option */ switch (argv[i][1]) { /* else check option */
case '-': /* '--' */ case '-': /* '--' */
if (argv[i][2] != '\0') /* extra characters after '--'? */ if (argv[i][2] != '\0') /* extra characters after '--'? */
return has_error; /* invalid option */ return has_error; /* invalid option */
*first = i + 1; *first = i + 1;
return args; return args;
case '\0': /* '-' */ case '\0': /* '-' */
return args; /* script "name" is '-' */ return args; /* script "name" is '-' */
case 'E': case 'E':
if (argv[i][2] != '\0') /* extra characters? */ if (argv[i][2] != '\0') /* extra characters? */
return has_error; /* invalid option */ return has_error; /* invalid option */
args |= has_E; args |= has_E;
break; break;
case 'W': case 'W':
if (argv[i][2] != '\0') /* extra characters? */ if (argv[i][2] != '\0') /* extra characters? */
return has_error; /* invalid option */ return has_error; /* invalid option */
break; break;
case 'i': case 'i':
args |= has_i; /* (-i implies -v) *//* FALLTHROUGH */ args |= has_i; /* (-i implies -v) */ /* FALLTHROUGH */
case 'v': case 'v':
if (argv[i][2] != '\0') /* extra characters? */ if (argv[i][2] != '\0') /* extra characters? */
return has_error; /* invalid option */ return has_error; /* invalid option */
args |= has_v; args |= has_v;
break; break;
case 'e': case 'e':
args |= has_e; /* FALLTHROUGH */ args |= has_e; /* FALLTHROUGH */
case 'l': /* both options need an argument */ case 'l': /* both options need an argument */
if (argv[i][2] == '\0') { /* no concatenated argument? */ if (argv[i][2] == '\0') { /* no concatenated argument? */
i++; /* try next 'argv' */ i++; /* try next 'argv' */
if (argv[i] == NULL || argv[i][0] == '-') if (argv[i] == NULL || argv[i][0] == '-')
return has_error; /* no next argument or it is another option */ return has_error; /* no next argument or it is another option */
} }
break; break;
default: /* invalid option */ default: /* invalid option */
return has_error; return has_error;
} }
} }
*first = i; /* no script name */ *first = i; /* no script name */
return args; return args;
} }
/* /*
* Processes options 'e' and 'l', which involve running Lua code, and * Processes options 'e' and 'l', which involve running Lua code, and
* 'W', which also affects the state. * 'W', which also affects the state.
* Returns 0 if some code raises an error. * Returns 0 if some code raises an error.
*/ */
static int runargs (lua_State *L, char **argv, int n) { static int
runargs(lua_State *L, char **argv, int n)
{
int i; int i;
for (i = 1; i < n; i++) { for (i = 1; i < n; i++) {
int option = argv[i][1]; int option = argv[i][1];
lua_assert(argv[i][0] == '-'); /* already checked */ lua_assert(argv[i][0] == '-'); /* already checked */
switch (option) { switch (option) {
case 'e': case 'l': { case 'e':
int status; case 'l': {
char *extra = argv[i] + 2; /* both options need an argument */ int status;
if (*extra == '\0') extra = argv[++i]; char *extra = argv[i] + 2; /* both options need an argument */
lua_assert(extra != NULL); if (*extra == '\0')
status = (option == 'e') extra = argv[++i];
? dostring(L, extra, "=(command line)") lua_assert(extra != NULL);
: dolibrary(L, extra); status = (option == 'e') ? dostring(L, extra, "=(command line)")
if (status != LUA_OK) return 0; : dolibrary(L, extra);
break; if (status != LUA_OK)
} return 0;
case 'W': break;
lua_warning(L, "@on", 0); /* warnings on */ }
break; case 'W':
lua_warning(L, "@on", 0); /* warnings on */
break;
} }
} }
return 1; return 1;
} }
static int
static int handle_luainit (lua_State *L) { handle_luainit(lua_State *L)
{
const char *name = "=" LUA_INITVARVERSION; const char *name = "=" LUA_INITVARVERSION;
const char *init = getenv(name + 1); const char *init = getenv(name + 1);
if (init == NULL) { if (init == NULL) {
name = "=" LUA_INIT_VAR; name = "=" LUA_INIT_VAR;
init = getenv(name + 1); /* try alternative name */ init = getenv(name + 1); /* try alternative name */
} }
if (init == NULL) return LUA_OK; if (init == NULL)
return LUA_OK;
else if (init[0] == '@') else if (init[0] == '@')
return dofile(L, init+1); return dofile(L, init + 1);
else else
return dostring(L, init, name); return dostring(L, init, name);
} }
/* /*
* {================================================================== * {==================================================================
* Read-Eval-Print Loop (REPL) * Read-Eval-Print Loop (REPL)
@ -360,25 +382,23 @@ static int handle_luainit (lua_State *L) {
*/ */
#if !defined(LUA_PROMPT) #if !defined(LUA_PROMPT)
#define LUA_PROMPT "> " #define LUA_PROMPT "> "
#define LUA_PROMPT2 ">> " #define LUA_PROMPT2 ">> "
#endif #endif
#if !defined(LUA_MAXINPUT) #if !defined(LUA_MAXINPUT)
#define LUA_MAXINPUT 512 #define LUA_MAXINPUT 512
#endif #endif
/* /*
* lua_stdin_is_tty detects whether the standard input is a 'tty' (that * lua_stdin_is_tty detects whether the standard input is a 'tty' (that
* is, whether we're running lua interactively). * is, whether we're running lua interactively).
*/ */
#if !defined(lua_stdin_is_tty) /* { */ #if !defined(lua_stdin_is_tty) /* { */
#define lua_stdin_is_tty() isatty(0) #define lua_stdin_is_tty() isatty(0)
#endif /* } */
#endif /* } */
/* /*
* lua_readline defines how to show a prompt and then read a line from * lua_readline defines how to show a prompt and then read a line from
@ -386,18 +406,18 @@ static int handle_luainit (lua_State *L) {
* lua_saveline defines how to "save" a read line in a "history". * lua_saveline defines how to "save" a read line in a "history".
* lua_freeline defines how to free a line read by lua_readline. * lua_freeline defines how to free a line read by lua_readline.
*/ */
#if !defined(lua_readline) /* { */ #if !defined(lua_readline) /* { */
#if defined(LUA_USE_READLINE) /* { */ #if defined(LUA_USE_READLINE) /* { */
#include <readline/readline.h>
#include <readline/history.h> #include <readline/history.h>
#define lua_initreadline(L) ((void)L, rl_readline_name="lua") #include <readline/readline.h>
#define lua_readline(L,b,p) ((void)L, ((b)=readline(p)) != NULL) #define lua_initreadline(L) ((void)L, rl_readline_name = "lua")
#define lua_saveline(L,line) ((void)L, add_history(line)) #define lua_readline(L, b, p) ((void)L, ((b) = readline(p)) != NULL)
#define lua_freeline(L,b) ((void)L, free(b)) #define lua_saveline(L, line) ((void)L, add_history(line))
#define lua_freeline(L, b) ((void)L, free(b))
#else /* }{ */ #else /* }{ */
/* clang-format off */ /* clang-format off */
@ -410,37 +430,39 @@ static int handle_luainit (lua_State *L) {
/* clang-format on */ /* clang-format on */
#endif /* } */ #endif /* } */
#endif /* } */
#endif /* } */
/* /*
* Return the string to be used as a prompt by the interpreter. Leave * Return the string to be used as a prompt by the interpreter. Leave
* the string (or nil, if using the default value) on the stack, to keep * the string (or nil, if using the default value) on the stack, to keep
* it anchored. * it anchored.
*/ */
static const char *get_prompt (lua_State *L, int firstline) { static const char *
get_prompt(lua_State *L, int firstline)
{
if (lua_getglobal(L, firstline ? "_PROMPT" : "_PROMPT2") == LUA_TNIL) if (lua_getglobal(L, firstline ? "_PROMPT" : "_PROMPT2") == LUA_TNIL)
return (firstline ? LUA_PROMPT : LUA_PROMPT2); /* use the default */ return (firstline ? LUA_PROMPT : LUA_PROMPT2); /* use the default */
else { /* apply 'tostring' over the value */ else { /* apply 'tostring' over the value */
const char *p = luaL_tolstring(L, -1, NULL); const char *p = luaL_tolstring(L, -1, NULL);
lua_remove(L, -2); /* remove original value */ lua_remove(L, -2); /* remove original value */
return p; return p;
} }
} }
/* mark in error messages for incomplete statements */ /* mark in error messages for incomplete statements */
#define EOFMARK "<eof>" #define EOFMARK "<eof>"
#define marklen (sizeof(EOFMARK)/sizeof(char) - 1) #define marklen (sizeof(EOFMARK) / sizeof(char) - 1)
/* /*
* Check whether 'status' signals a syntax error and the error * Check whether 'status' signals a syntax error and the error
* message at the top of the stack ends with the above mark for * message at the top of the stack ends with the above mark for
* incomplete statements. * incomplete statements.
*/ */
static int incomplete (lua_State *L, int status) { static int
incomplete(lua_State *L, int status)
{
if (status == LUA_ERRSYNTAX) { if (status == LUA_ERRSYNTAX) {
size_t lmsg; size_t lmsg;
const char *msg = lua_tolstring(L, -1, &lmsg); const char *msg = lua_tolstring(L, -1, &lmsg);
@ -449,14 +471,15 @@ static int incomplete (lua_State *L, int status) {
return 1; return 1;
} }
} }
return 0; /* else... */ return 0; /* else... */
} }
/* /*
* Prompt the user, read a line, and push it into the Lua stack. * Prompt the user, read a line, and push it into the Lua stack.
*/ */
static int pushline (lua_State *L, int firstline) { static int
pushline(lua_State *L, int firstline)
{
char buffer[LUA_MAXINPUT]; char buffer[LUA_MAXINPUT];
char *b = buffer; char *b = buffer;
size_t l; size_t l;
@ -464,175 +487,184 @@ static int pushline (lua_State *L, int firstline) {
int readstatus = lua_readline(L, b, prmt); int readstatus = lua_readline(L, b, prmt);
if (readstatus == 0) if (readstatus == 0)
return 0; /* no input (prompt will be popped by caller) */ return 0; /* no input (prompt will be popped by caller) */
lua_pop(L, 1); /* remove prompt */ lua_pop(L, 1); /* remove prompt */
l = strlen(b); l = strlen(b);
if (l > 0 && b[l-1] == '\n') /* line ends with newline? */ if (l > 0 && b[l - 1] == '\n') /* line ends with newline? */
b[--l] = '\0'; /* remove it */ b[--l] = '\0'; /* remove it */
if (firstline && b[0] == '=') /* for compatibility with 5.2, ... */ if (firstline && b[0] == '=') /* for compatibility with 5.2, ... */
lua_pushfstring(L, "return %s", b + 1); /* change '=' to 'return' */ lua_pushfstring(L, "return %s", b + 1); /* change '=' to 'return' */
else else
lua_pushlstring(L, b, l); lua_pushlstring(L, b, l);
lua_freeline(L, b); lua_freeline(L, b);
return 1; return 1;
} }
/* /*
* Try to compile line on the stack as 'return <line>;'; on return, stack * Try to compile line on the stack as 'return <line>;'; on return, stack
* has either compiled chunk or original line (if compilation failed). * has either compiled chunk or original line (if compilation failed).
*/ */
static int addreturn (lua_State *L) { static int
const char *line = lua_tostring(L, -1); /* original line */ addreturn(lua_State *L)
{
const char *line = lua_tostring(L, -1); /* original line */
const char *retline = lua_pushfstring(L, "return %s;", line); const char *retline = lua_pushfstring(L, "return %s;", line);
int status = luaL_loadbuffer(L, retline, strlen(retline), "=stdin"); int status = luaL_loadbuffer(L, retline, strlen(retline), "=stdin");
if (status == LUA_OK) { if (status == LUA_OK) {
lua_remove(L, -2); /* remove modified line */ lua_remove(L, -2); /* remove modified line */
if (line[0] != '\0') /* non empty? */ if (line[0] != '\0') /* non empty? */
lua_saveline(L, line); /* keep history */ lua_saveline(L, line); /* keep history */
} } else
else lua_pop(L, 2); /* pop result from 'luaL_loadbuffer' and modified line */
lua_pop(L, 2); /* pop result from 'luaL_loadbuffer' and modified line */
return status; return status;
} }
/* /*
* Read multiple lines until a complete Lua statement * Read multiple lines until a complete Lua statement
*/ */
static int multiline (lua_State *L) { static int
for (;;) { /* repeat until gets a complete statement */ multiline(lua_State *L)
{
for (;;) { /* repeat until gets a complete statement */
size_t len; size_t len;
const char *line = lua_tolstring(L, 1, &len); /* get what it has */ const char *line = lua_tolstring(L, 1, &len); /* get what it has */
int status = luaL_loadbuffer(L, line, len, "=stdin"); /* try it */ int status = luaL_loadbuffer(L, line, len, "=stdin"); /* try it */
if (!incomplete(L, status) || !pushline(L, 0)) { if (!incomplete(L, status) || !pushline(L, 0)) {
lua_saveline(L, line); /* keep history */ lua_saveline(L, line); /* keep history */
return status; /* cannot or should not try to add continuation line */ return status; /* cannot or should not try to add continuation line */
} }
lua_pushliteral(L, "\n"); /* add newline... */ lua_pushliteral(L, "\n"); /* add newline... */
lua_insert(L, -2); /* ...between the two lines */ lua_insert(L, -2); /* ...between the two lines */
lua_concat(L, 3); /* join them */ lua_concat(L, 3); /* join them */
} }
} }
/* /*
* Read a line and try to load (compile) it first as an expression (by * Read a line and try to load (compile) it first as an expression (by
* adding "return " in front of it) and second as a statement. Return * adding "return " in front of it) and second as a statement. Return
* the final status of load/call with the resulting function (if any) * the final status of load/call with the resulting function (if any)
* in the top of the stack. * in the top of the stack.
*/ */
static int loadline (lua_State *L) { static int
loadline(lua_State *L)
{
int status; int status;
lua_settop(L, 0); lua_settop(L, 0);
if (!pushline(L, 1)) if (!pushline(L, 1))
return -1; /* no input */ return -1; /* no input */
if ((status = addreturn(L)) != LUA_OK) /* 'return ...' did not work? */ if ((status = addreturn(L)) != LUA_OK) /* 'return ...' did not work? */
status = multiline(L); /* try as command, maybe with continuation lines */ status = multiline(L); /* try as command, maybe with continuation lines */
lua_remove(L, 1); /* remove line from the stack */ lua_remove(L, 1); /* remove line from the stack */
lua_assert(lua_gettop(L) == 1); lua_assert(lua_gettop(L) == 1);
return status; return status;
} }
/* /*
* Prints (calling the Lua 'print' function) any values on the stack * Prints (calling the Lua 'print' function) any values on the stack
*/ */
static void l_print (lua_State *L) { static void
l_print(lua_State *L)
{
int n = lua_gettop(L); int n = lua_gettop(L);
if (n > 0) { /* any result to be printed? */ if (n > 0) { /* any result to be printed? */
luaL_checkstack(L, LUA_MINSTACK, "too many results to print"); luaL_checkstack(L, LUA_MINSTACK, "too many results to print");
lua_getglobal(L, "print"); lua_getglobal(L, "print");
lua_insert(L, 1); lua_insert(L, 1);
if (lua_pcall(L, n, 0, 0) != LUA_OK) if (lua_pcall(L, n, 0, 0) != LUA_OK)
l_message(progname, lua_pushfstring(L, "error calling 'print' (%s)", l_message(progname,
lua_tostring(L, -1))); lua_pushfstring(L, "error calling 'print' (%s)",
lua_tostring(L, -1)));
} }
} }
/* /*
* Do the REPL: repeatedly read (load) a line, evaluate (call) it, and * Do the REPL: repeatedly read (load) a line, evaluate (call) it, and
* print any results. * print any results.
*/ */
static void doREPL (lua_State *L) { static void
doREPL(lua_State *L)
{
int status; int status;
const char *oldprogname = progname; const char *oldprogname = progname;
progname = NULL; /* no 'progname' on errors in interactive mode */ progname = NULL; /* no 'progname' on errors in interactive mode */
lua_initreadline(L); lua_initreadline(L);
while ((status = loadline(L)) != -1) { while ((status = loadline(L)) != -1) {
if (status == LUA_OK) if (status == LUA_OK)
status = docall(L, 0, LUA_MULTRET); status = docall(L, 0, LUA_MULTRET);
if (status == LUA_OK) l_print(L); if (status == LUA_OK)
else report(L, status); l_print(L);
else
report(L, status);
} }
lua_settop(L, 0); /* clear stack */ lua_settop(L, 0); /* clear stack */
lua_writeline(); lua_writeline();
progname = oldprogname; progname = oldprogname;
} }
/* }================================================================== */ /* }================================================================== */
/* /*
* Main body of stand-alone interpreter (to be called in protected mode). * Main body of stand-alone interpreter (to be called in protected mode).
* Reads the options and handles them all. * Reads the options and handles them all.
*/ */
static int pmain (lua_State *L) { static int
pmain(lua_State *L)
{
int argc = (int)lua_tointeger(L, 1); int argc = (int)lua_tointeger(L, 1);
char **argv = (char **)lua_touserdata(L, 2); char **argv = (char **)lua_touserdata(L, 2);
int script; int script;
int args = collectargs(argv, &script); int args = collectargs(argv, &script);
luaL_checkversion(L); /* check that interpreter has correct version */ luaL_checkversion(L); /* check that interpreter has correct version */
if (argv[0] && argv[0][0]) progname = argv[0]; if (argv[0] && argv[0][0])
if (args == has_error) { /* bad arg? */ progname = argv[0];
print_usage(argv[script]); /* 'script' has index of bad arg. */ if (args == has_error) { /* bad arg? */
print_usage(argv[script]); /* 'script' has index of bad arg. */
return 0; return 0;
} }
if (args & has_v) /* option '-v'? */ if (args & has_v) /* option '-v'? */
print_version(); print_version();
if (args & has_E) { /* option '-E'? */ if (args & has_E) { /* option '-E'? */
lua_pushboolean(L, 1); /* signal for libraries to ignore env. vars. */ lua_pushboolean(L, 1); /* signal for libraries to ignore env. vars. */
lua_setfield(L, LUA_REGISTRYINDEX, "LUA_NOENV"); lua_setfield(L, LUA_REGISTRYINDEX, "LUA_NOENV");
} }
createargtable(L, argv, argc, script); /* create table 'arg' */ createargtable(L, argv, argc, script); /* create table 'arg' */
lua_gc(L, LUA_GCGEN, 0, 0); /* GC in generational mode */ lua_gc(L, LUA_GCGEN, 0, 0); /* GC in generational mode */
if (!(args & has_E)) { /* no option '-E'? */ if (!(args & has_E)) { /* no option '-E'? */
if (handle_luainit(L) != LUA_OK) /* run LUA_INIT */ if (handle_luainit(L) != LUA_OK) /* run LUA_INIT */
return 0; /* error running LUA_INIT */ return 0; /* error running LUA_INIT */
} }
if (!runargs(L, argv, script)) /* execute arguments -e and -l */ if (!runargs(L, argv, script)) /* execute arguments -e and -l */
return 0; /* something failed */ return 0; /* something failed */
if (script < argc && /* execute main script (if there is one) */ if (script < argc && /* execute main script (if there is one) */
handle_script(L, argv + script) != LUA_OK) handle_script(L, argv + script) != LUA_OK)
return 0; return 0;
if (args & has_i) /* -i option? */ if (args & has_i) /* -i option? */
doREPL(L); /* do read-eval-print loop */ doREPL(L); /* do read-eval-print loop */
else if (script == argc && !(args & (has_e | has_v))) { /* no arguments? */ else if (script == argc && !(args & (has_e | has_v))) { /* no arguments? */
if (lua_stdin_is_tty()) { /* running in interactive mode? */ if (lua_stdin_is_tty()) { /* running in interactive mode? */
print_version(); print_version();
doREPL(L); /* do read-eval-print loop */ doREPL(L); /* do read-eval-print loop */
} } else
else dofile(L, NULL); /* executes stdin as a file */ dofile(L, NULL); /* executes stdin as a file */
} }
lua_pushboolean(L, 1); /* signal no errors */ lua_pushboolean(L, 1); /* signal no errors */
return 1; return 1;
} }
int
int main (int argc, char **argv) { main(int argc, char **argv)
{
int status, result; int status, result;
lua_State *L = callisto_newstate(); /* create state */ lua_State *L = callisto_newstate(); /* create state */
if (L == NULL) { if (L == NULL) {
l_message(argv[0], "cannot create state: not enough memory"); l_message(argv[0], "cannot create state: not enough memory");
return EXIT_FAILURE; return EXIT_FAILURE;
} }
lua_pushcfunction(L, &pmain); /* to call 'pmain' in protected mode */ lua_pushcfunction(L, &pmain); /* to call 'pmain' in protected mode */
lua_pushinteger(L, argc); /* 1st argument */ lua_pushinteger(L, argc); /* 1st argument */
lua_pushlightuserdata(L, argv); /* 2nd argument */ lua_pushlightuserdata(L, argv); /* 2nd argument */
status = lua_pcall(L, 2, 1, 0); /* do the call */ status = lua_pcall(L, 2, 1, 0); /* do the call */
result = lua_toboolean(L, -1); /* get result */ result = lua_toboolean(L, -1); /* get result */
report(L, status); report(L, status);
lua_close(L); lua_close(L);
return (result && status == LUA_OK) ? EXIT_SUCCESS : EXIT_FAILURE; return (result && status == LUA_OK) ? EXIT_SUCCESS : EXIT_FAILURE;
} }

14
lcl.c
View file

@ -29,7 +29,8 @@ fmesgshift(lua_State *L, int fd, int shift)
lua_getglobal(L, "arg"); lua_getglobal(L, "arg");
if (!lua_istable(L, -1)) { if (!lua_istable(L, -1)) {
luaL_error(L, "arg table not present/inaccessible; cannot get script name"); luaL_error(L,
"arg table not present/inaccessible; cannot get script name");
return; return;
} }
@ -44,9 +45,9 @@ fmesgshift(lua_State *L, int fd, int shift)
lua_getglobal(L, "string"); lua_getglobal(L, "string");
lua_getfield(L, -1, "format"); lua_getfield(L, -1, "format");
paramc = lua_gettop(L); /* get parameter count */ paramc = lua_gettop(L); /* get parameter count */
for (i = 1 + shift; i <= paramc; i++) { /* for every parameter */ for (i = 1 + shift; i <= paramc; i++) { /* for every parameter */
lua_pushvalue(L, i); /* push argument */ lua_pushvalue(L, i); /* push argument */
} }
lua_call(L, paramc - shift, 1); lua_call(L, paramc - shift, 1);
@ -240,13 +241,14 @@ cl_options(lua_State *L)
argv[argc] = NULL; argv[argc] = NULL;
for (i = 0; i < argc; i++) { /* for every argument */ for (i = 0; i < argc; i++) { /* for every argument */
lua_pushinteger(L, i); /* push argv index */ lua_pushinteger(L, i); /* push argv index */
lua_gettable(L, 1); /* push argv[i] */ lua_gettable(L, 1); /* push argv[i] */
argv[i] = (char *)luaL_checkstring(L, -1); argv[i] = (char *)luaL_checkstring(L, -1);
} }
if (optstring[0] == ':') if (optstring[0] == ':')
return luaL_argerror(L, 2, "option string must not start with a colon (:)"); return luaL_argerror(L, 2,
"option string must not start with a colon (:)");
strprepend(optstring, ":"); strprepend(optstring, ":");

View file

@ -68,7 +68,7 @@ environ_newindex(lua_State *L)
if (lua_isnil(L, 3)) { if (lua_isnil(L, 3)) {
ret = unsetenv(variable); /* remove variable from environ */ ret = unsetenv(variable); /* remove variable from environ */
if (ret == 0) /* did unsetenv succeed? */ if (ret == 0) /* did unsetenv succeed? */
return 0; return 0;
/* if unsetenv didn't succeed: /* if unsetenv didn't succeed:
@ -77,18 +77,18 @@ environ_newindex(lua_State *L)
} }
value = luaL_checkstring(L, 3); value = luaL_checkstring(L, 3);
ret = setenv(variable, value, 1); ret = setenv(variable, value, 1);
if (ret == 0) /* did setenv succeed? */ if (ret == 0) /* did setenv succeed? */
return 0; return 0;
switch (errno) { switch (errno) {
case EINVAL: case EINVAL:
return luaL_error(L, "invalid input string"); return luaL_error(L, "invalid input string");
break; break;
case ENOMEM: case ENOMEM:
return luaL_error(L, "insufficient memory"); return luaL_error(L, "insufficient memory");
break; break;
} }
return 0; return 0;

8
lfs.c
View file

@ -49,7 +49,7 @@ fs_basename(lua_State *L)
char *path; /* parameter 1 (string) */ char *path; /* parameter 1 (string) */
path = strndup(luaL_checkstring(L, 1), lua_rawlen(L, 1)); path = strndup(luaL_checkstring(L, 1), lua_rawlen(L, 1));
ret = basename(path); ret = basename(path);
if (ret == NULL) /* failed? */ if (ret == NULL) /* failed? */
return lfail(L); return lfail(L);
@ -147,7 +147,7 @@ fs_dirname(lua_State *L)
char *path; /* parameter 1 (string) */ char *path; /* parameter 1 (string) */
path = strndup(luaL_checkstring(L, 1), lua_rawlen(L, 1)); path = strndup(luaL_checkstring(L, 1), lua_rawlen(L, 1));
ret = dirname(path); ret = dirname(path);
if (ret == NULL) /* failed? */ if (ret == NULL) /* failed? */
return lfail(L); return lfail(L);
@ -367,9 +367,9 @@ fs_move(lua_State *L)
const char *src; /* parameter 1 (string) */ const char *src; /* parameter 1 (string) */
const char *dest; /* parameter 2 (string) */ const char *dest; /* parameter 2 (string) */
src = luaL_checkstring(L, 1); src = luaL_checkstring(L, 1);
dest = luaL_checkstring(L, 2); dest = luaL_checkstring(L, 2);
ret = rename(src, dest); /* move file */ ret = rename(src, dest); /* move file */
if (ret == 0) { /* check for success */ if (ret == 0) { /* check for success */
lua_pushboolean(L, 1); lua_pushboolean(L, 1);

View file

@ -217,7 +217,6 @@ local t = json.decode(j)
* sparse; *>1*: use ratio * sparse; *>1*: use ratio
*/ */
int luaopen_cjson(lua_State *); int luaopen_cjson(lua_State *);
int int

View file

@ -19,7 +19,7 @@
#include "util.h" #include "util.h"
#define PID_MAX 8 /* rounded to the nearest even number */ #define PID_MAX 8 /* rounded to the nearest even number */
#define PROCESS_MAX 256 #define PROCESS_MAX 256
/* clang-format off */ /* clang-format off */

14
util.c
View file

@ -65,7 +65,7 @@ strbcat(char *dst, const char *src, size_t dsize)
n = dsize - dlen; n = dsize - dlen;
if (n-- == 0) if (n-- == 0)
return(dlen + strlen(src)); return (dlen + strlen(src));
while (*src != '\0') { while (*src != '\0') {
if (n != 0) { if (n != 0) {
*dst++ = *src; *dst++ = *src;
@ -75,7 +75,7 @@ strbcat(char *dst, const char *src, size_t dsize)
} }
*dst = '\0'; *dst = '\0';
return(dlen + (src - osrc)); /* count does not include NUL */ return (dlen + (src - osrc)); /* count does not include NUL */
} }
/* /*
@ -100,12 +100,12 @@ strbcpy(char *dst, const char *src, size_t dsize)
/* Not enough room in dst, add NUL and traverse rest of src. */ /* Not enough room in dst, add NUL and traverse rest of src. */
if (nleft == 0) { if (nleft == 0) {
if (dsize != 0) if (dsize != 0)
*dst = '\0'; /* NUL-terminate dst */ *dst = '\0'; /* NUL-terminate dst */
while (*src++) while (*src++)
; ;
} }
return(src - osrc - 1); /* count does not include NUL */ return (src - osrc - 1); /* count does not include NUL */
} }
/* /*
@ -114,7 +114,7 @@ strbcpy(char *dst, const char *src, size_t dsize)
void void
strprepend(char *s, const char *t) strprepend(char *s, const char *t)
{ {
size_t len = strlen(t); size_t len = strlen(t);
memmove(s + len, s, strlen(s) + 1); memmove(s + len, s, strlen(s) + 1);
memcpy(s, t, len); memcpy(s, t, len);
} }