PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2026-01-04 05:39:01 +00:00
Code Issues Projects Releases Wiki Activity

Apply my refactoring changes from r31989.

Browse Source
This commit is contained in:
david
2013-09-12 08:12:03 +00:00
parent ccd0c02a4c
commit 42ce2bc9cc
8 changed files with 320 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
ncat/ncat_lua.c
View File

@@ -124,16 +124,20 @@
#include "ncat.h"
#include "ncat_lua.h"
static lua_State *L;
static int last_function_number;
lua_State *luaexec_L = NULL;
int error_handler_idx = -1;
static void report(char *prefix)
void lua_report(lua_State *L, char *prefix, int panic)
{
const char *errormsg;
errormsg = lua_tostring(L, -1);
if (errormsg == NULL)
errormsg = "(error object is not a string)";
bye("%s: %s.", prefix, errormsg);
if (panic)
bye("%s: %s.", prefix, errormsg);
else
loguser("%s: %s.", prefix, errormsg);
}
static int traceback (lua_State *L)
@@ -151,32 +155,24 @@ static int traceback (lua_State *L)
return 1;
}
void lua_setup(void)
void lua_setup(char *cmdexec)
{
ncat_assert(o.cmdexec != NULL);
ncat_assert(cmdexec != NULL);
L = luaL_newstate();
luaL_openlibs(L);
lua_State **L = &luaexec_L;
if (luaL_loadfile(L, o.cmdexec) != 0)
report("Error loading the Lua script");
if (*L == NULL) {
*L = luaL_newstate();
luaL_openlibs(*L);
/* install the traceback function */
last_function_number = lua_gettop(L);
lua_pushcfunction(L, traceback);
lua_insert(L, last_function_number);
}
void lua_run(void)
{
if (lua_pcall(L, 0, 0, last_function_number) != LUA_OK && !lua_isnil(L, -1)) {
/* handle the error; the code below is taken from lua.c, Lua source code */
lua_remove(L, last_function_number);
report("Error running the Lua script");
} else {
if (o.debug)
logdebug("%s returned successfully.\n", o.cmdexec);
lua_close(L);
exit(EXIT_SUCCESS);
if (error_handler_idx == -1) {
/* install the traceback function */
error_handler_idx = lua_gettop(*L);
lua_pushcfunction(*L, traceback);
lua_insert(*L, error_handler_idx);
}
}
if (luaL_loadfile(*L, cmdexec) != 0)
lua_report(*L, "Error loading the Lua script", 1);
}