mirror of https://github.com/OISF/suricata
lua: introduce util-lua.[ch]
Shared functions for all lua parts of the engine.pull/1112/head
Victor Julien
12 years ago
parent
8bc01af581
commit
599ec36b2c
@ -0,0 +1,131 @@
|
||||
/* Copyright (C) 2014 Open Information Security Foundation
|
||||
*
|
||||
* You can copy, redistribute or modify this Program under the terms of
|
||||
* the GNU General Public License version 2 as published by the Free
|
||||
* Software Foundation.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* version 2 along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||
* 02110-1301, USA.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \file
|
||||
*
|
||||
* \author Victor Julien <victor@inliniac.net>
|
||||
*
|
||||
* Common function for Lua
|
||||
*/
|
||||
|
||||
#include "suricata-common.h"
|
||||
#include "debug.h"
|
||||
#include "detect.h"
|
||||
#include "pkt-var.h"
|
||||
#include "conf.h"
|
||||
|
||||
#include "threads.h"
|
||||
#include "threadvars.h"
|
||||
#include "tm-threads.h"
|
||||
|
||||
#include "util-print.h"
|
||||
#include "util-unittest.h"
|
||||
|
||||
#include "util-debug.h"
|
||||
|
||||
#include "output.h"
|
||||
#include "app-layer-htp.h"
|
||||
#include "app-layer.h"
|
||||
#include "app-layer-parser.h"
|
||||
#include "util-privs.h"
|
||||
#include "util-buffer.h"
|
||||
#include "util-proto-name.h"
|
||||
#include "util-logopenfile.h"
|
||||
#include "util-time.h"
|
||||
|
||||
#ifdef HAVE_LUA
|
||||
|
||||
#include <lua.h>
|
||||
#include <lualib.h>
|
||||
#include <lauxlib.h>
|
||||
|
||||
/* key for tx pointer */
|
||||
const char lua_ext_key_tx[] = "suricata:lua:tx:ptr";
|
||||
/* key for p (packet) pointer */
|
||||
const char lua_ext_key_p[] = "suricata:lua:pkt:ptr";
|
||||
|
||||
/** \brief get packet pointer from the lua state */
|
||||
Packet *LuaStateGetPacket(lua_State *luastate)
|
||||
{
|
||||
lua_pushlightuserdata(luastate, (void *)&lua_ext_key_p);
|
||||
lua_gettable(luastate, LUA_REGISTRYINDEX);
|
||||
void *p = lua_touserdata(luastate, -1);
|
||||
return (Packet *)p;
|
||||
}
|
||||
|
||||
void LuaStateSetPacket(lua_State *luastate, Packet *p)
|
||||
{
|
||||
lua_pushlightuserdata(luastate, (void *)&lua_ext_key_p);
|
||||
lua_pushlightuserdata(luastate, (void *)p);
|
||||
lua_settable(luastate, LUA_REGISTRYINDEX);
|
||||
}
|
||||
|
||||
/** \brief get tx pointer from the lua state */
|
||||
void *LuaStateGetTX(lua_State *luastate)
|
||||
{
|
||||
lua_pushlightuserdata(luastate, (void *)&lua_ext_key_tx);
|
||||
lua_gettable(luastate, LUA_REGISTRYINDEX);
|
||||
void *tx = lua_touserdata(luastate, -1);
|
||||
return tx;
|
||||
}
|
||||
|
||||
void LuaStateSetTX(lua_State *luastate, void *txptr)
|
||||
{
|
||||
lua_pushlightuserdata(luastate, (void *)&lua_ext_key_tx);
|
||||
lua_pushlightuserdata(luastate, (void *)txptr);
|
||||
lua_settable(luastate, LUA_REGISTRYINDEX);
|
||||
}
|
||||
|
||||
/** \brief dump stack from lua state to screen */
|
||||
void LuaPrintStack(lua_State *state) {
|
||||
int size = lua_gettop(state);
|
||||
int i;
|
||||
|
||||
for (i = 1; i <= size; i++) {
|
||||
int type = lua_type(state, i);
|
||||
printf("Stack size=%d, level=%d, type=%d, ", size, i, type);
|
||||
|
||||
switch (type) {
|
||||
case LUA_TFUNCTION:
|
||||
printf("function %s", lua_tostring(state, i) ? "true" : "false");
|
||||
break;
|
||||
case LUA_TBOOLEAN:
|
||||
printf("bool %s", lua_toboolean(state, i) ? "true" : "false");
|
||||
break;
|
||||
case LUA_TNUMBER:
|
||||
printf("number %g", lua_tonumber(state, i));
|
||||
break;
|
||||
case LUA_TSTRING:
|
||||
printf("string `%s'", lua_tostring(state, i));
|
||||
break;
|
||||
case LUA_TTABLE:
|
||||
printf("table `%s'", lua_tostring(state, i));
|
||||
break;
|
||||
default:
|
||||
printf("other %s", lua_typename(state, type));
|
||||
break;
|
||||
|
||||
}
|
||||
printf("\n");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
#endif /* HAVE_LUA */
|
||||
@ -0,0 +1,39 @@
|
||||
/* Copyright (C) 2014 Open Information Security Foundation
|
||||
*
|
||||
* You can copy, redistribute or modify this Program under the terms of
|
||||
* the GNU General Public License version 2 as published by the Free
|
||||
* Software Foundation.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* version 2 along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||
* 02110-1301, USA.
|
||||
*/
|
||||
|
||||
/**
|
||||
* \file
|
||||
*
|
||||
* \author Victor Julien <victor@inliniac.net>
|
||||
*/
|
||||
|
||||
#ifndef __UTIL_LUA_H__
|
||||
#define __UTIL_LUA_H__
|
||||
|
||||
#ifdef HAVE_LUA
|
||||
|
||||
Packet *LuaStateGetPacket(lua_State *luastate);
|
||||
void *LuaStateGetTX(lua_State *luastate);
|
||||
|
||||
void LuaStateSetPacket(lua_State *luastate, Packet *p);
|
||||
void LuaStateSetTX(lua_State *luastate, void *tx);
|
||||
|
||||
void LuaPrintStack(lua_State *state);
|
||||
|
||||
#endif /* HAVE_LUA */
|
||||
|
||||
#endif /* __UTIL_LUA_H__ */
|
||||
Loading…
Reference in New Issue