Commit 6b58ddf8 authored by Vicent Marti's avatar Vicent Marti

lua: Proper style for main.c

parent b3dd6731
...@@ -29,149 +29,142 @@ ...@@ -29,149 +29,142 @@
#include <string.h> #include <string.h>
#include <unistd.h> #include <unistd.h>
#include "lua.h"
#include "lauxlib.h" #include "lauxlib.h"
#include "lua.h"
#include "lualib.h" #include "lualib.h"
static lua_State *globalL = NULL; static lua_State *globalL = NULL;
static const char *progname = NULL; static const char *progname = NULL;
static void lstop(lua_State *L, lua_Debug *ar) static void lstop(lua_State *L, lua_Debug *ar) {
{ (void)ar; /* unused arg. */
(void)ar; /* unused arg. */ lua_sethook(L, NULL, 0, 0);
lua_sethook(L, NULL, 0, 0); luaL_error(L, "interrupted!");
luaL_error(L, "interrupted!");
} }
static void laction(int i) static void laction(int i) {
{ signal(i, SIG_DFL);
signal(i, SIG_DFL); lua_sethook(globalL, lstop, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT, 1);
lua_sethook(globalL, lstop, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT, 1);
} }
static void l_message(const char *pname, const char *msg) static void l_message(const char *pname, const char *msg) {
{ if (pname)
if (pname) fprintf(stderr, "%s: ", pname); fprintf(stderr, "%s: ", pname);
fprintf(stderr, "%s\n", msg); fprintf(stderr, "%s\n", msg);
fflush(stderr); fflush(stderr);
} }
static int report(lua_State *L, int status) static int report(lua_State *L, int status) {
{ if (status && !lua_isnil(L, -1)) {
if (status && !lua_isnil(L, -1)) { const char *msg = lua_tostring(L, -1);
const char *msg = lua_tostring(L, -1); if (msg == NULL)
if (msg == NULL) msg = "(error object is not a string)"; msg = "(error object is not a string)";
l_message(progname, msg); l_message(progname, msg);
lua_pop(L, 1); lua_pop(L, 1);
} }
return status; return status;
} }
static int traceback(lua_State *L) static int traceback(lua_State *L) {
{ if (!lua_isstring(L, 1)) /* 'message' not a string? */
if (!lua_isstring(L, 1)) /* 'message' not a string? */ return 1; /* keep it intact */
return 1; /* keep it intact */ lua_getfield(L, LUA_GLOBALSINDEX, "debug");
lua_getfield(L, LUA_GLOBALSINDEX, "debug"); if (!lua_istable(L, -1)) {
if (!lua_istable(L, -1)) { lua_pop(L, 1);
lua_pop(L, 1); return 1;
return 1; }
} lua_getfield(L, -1, "traceback");
lua_getfield(L, -1, "traceback"); if (!lua_isfunction(L, -1)) {
if (!lua_isfunction(L, -1)) { lua_pop(L, 2);
lua_pop(L, 2); return 1;
return 1; }
} lua_pushvalue(L, 1); /* pass error message */
lua_pushvalue(L, 1); /* pass error message */ lua_pushinteger(L, 2); /* skip this function and traceback */
lua_pushinteger(L, 2); /* skip this function and traceback */ lua_call(L, 2, 1); /* call debug.traceback */
lua_call(L, 2, 1); /* call debug.traceback */ return 1;
return 1;
} }
static int docall(lua_State *L, int narg, int clear) static int docall(lua_State *L, int narg, int clear) {
{ int status;
int status; int base = lua_gettop(L) - narg; /* function index */
int base = lua_gettop(L) - narg; /* function index */ lua_pushcfunction(L, traceback); /* push traceback function */
lua_pushcfunction(L, traceback); /* push traceback function */ lua_insert(L, base); /* put it under chunk and args */
lua_insert(L, base); /* put it under chunk and args */ signal(SIGINT, laction);
signal(SIGINT, laction); status = lua_pcall(L, narg, (clear ? 0 : LUA_MULTRET), base);
status = lua_pcall(L, narg, (clear ? 0 : LUA_MULTRET), base); signal(SIGINT, SIG_DFL);
signal(SIGINT, SIG_DFL); lua_remove(L, base); /* remove traceback function */
lua_remove(L, base); /* remove traceback function */ /* force a complete garbage collection in case of errors */
/* force a complete garbage collection in case of errors */ if (status != 0)
if (status != 0) lua_gc(L, LUA_GCCOLLECT, 0); lua_gc(L, LUA_GCCOLLECT, 0);
return status; return status;
} }
static int dolibrary(lua_State *L, const char *name, int clear) static int dolibrary(lua_State *L, const char *name, int clear) {
{ lua_getglobal(L, "require");
lua_getglobal(L, "require"); lua_pushstring(L, name);
lua_pushstring(L, name); return report(L, docall(L, 1, clear));
return report(L, docall(L, 1, clear));
} }
struct Smain { struct Smain {
int argc; int argc;
char **argv; char **argv;
int status; int status;
}; };
static void pushargv(lua_State *L, char **argv, int argc, int offset) static void pushargv(lua_State *L, char **argv, int argc, int offset) {
{ int i, j;
int i, j; lua_createtable(L, argc, 0);
lua_createtable(L, argc, 0); for (i = offset, j = 1; i < argc; i++, j++) {
for (i = offset, j = 1; i < argc; i++, j++) { lua_pushstring(L, argv[i]);
lua_pushstring(L, argv[i]); lua_rawseti(L, -2, j);
lua_rawseti(L, -2, j); }
}
} }
static int pmain(lua_State *L) static int pmain(lua_State *L) {
{ struct Smain *s = (struct Smain *)lua_touserdata(L, 1);
struct Smain *s = (struct Smain *)lua_touserdata(L, 1); globalL = L;
globalL = L;
lua_gc(L, LUA_GCSTOP, 0); lua_gc(L, LUA_GCSTOP, 0);
luaL_openlibs(L); luaL_openlibs(L);
lua_gc(L, LUA_GCRESTART, 0); lua_gc(L, LUA_GCRESTART, 0);
s->status = dolibrary(L, "bcc", 0); s->status = dolibrary(L, "bcc", 0);
if (s->status) if (s->status)
return 0; return 0;
lua_pushstring(L, progname); lua_pushstring(L, progname);
lua_setglobal(L, "BCC_STANDALONE"); lua_setglobal(L, "BCC_STANDALONE");
pushargv(L, s->argv, s->argc, 1); pushargv(L, s->argv, s->argc, 1);
lua_setglobal(L, "arg"); lua_setglobal(L, "arg");
s->status = report(L, docall(L, 0, 1)); s->status = report(L, docall(L, 0, 1));
return 0; return 0;
} }
int main(int argc, char **argv) int main(int argc, char **argv) {
{ int status;
int status; struct Smain s;
struct Smain s; lua_State *L = lua_open(); /* create state */
lua_State *L = lua_open(); /* 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;
} }
if (geteuid() != 0) { if (geteuid() != 0) {
l_message(argv[0], "bcc-lua must be ran as root"); l_message(argv[0], "bcc-lua must be ran as root");
return EXIT_FAILURE; return EXIT_FAILURE;
} }
progname = argv[0]; progname = argv[0];
s.argc = argc; s.argc = argc;
s.argv = argv; s.argv = argv;
s.status = 0; s.status = 0;
status = lua_cpcall(L, &pmain, &s); status = lua_cpcall(L, &pmain, &s);
report(L, status); report(L, status);
lua_close(L); lua_close(L);
return (status || s.status) ? EXIT_FAILURE : EXIT_SUCCESS; return (status || s.status) ? EXIT_FAILURE : EXIT_SUCCESS;
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment