output-lua: initial packet support

Add key for storing packet pointer in the lua stack and a utility
function to retrieve it from lua callbacks.

diff --git a/src/output-lua-common.c b/src/output-lua-common.c
index ae4384ed9f70cee629e01057e317a3b41cdd5fa8..c62f19c83a33d95ae5173cffdf9e45c3882b3a84 100644 (file)
--- a/src/output-lua-common.c
+++ b/src/output-lua-common.c
@@ -89,6 +89,15 @@ void LuaPrintStack(lua_State *state) {
 }
 
 extern const char lualog_ext_key_tx;
+extern const char lualog_ext_key_p;
+
+Packet *LuaStateGetPacket(lua_State *luastate)
+{
+    lua_pushlightuserdata(luastate, (void *)&lualog_ext_key_p);
+    lua_gettable(luastate, LUA_REGISTRYINDEX);
+    void *p = lua_touserdata(luastate, -1);
+    return (Packet *)p;
+}
 
 void *LuaStateGetTX(lua_State *luastate)
 {

diff --git a/src/output-lua-common.h b/src/output-lua-common.h
index eeefbaa1702b5f96b571d2215f41764fe438c975..d5f46f64695eda27d441769dc9c5907b4bfe2bed 100644 (file)
--- a/src/output-lua-common.h
+++ b/src/output-lua-common.h
@@ -28,6 +28,7 @@
 
 void LuaPrintStack(lua_State *state);
 
+Packet *LuaStateGetPacket(lua_State *luastate);
 void *LuaStateGetTX(lua_State *luastate);
 int LuaCallbackError(lua_State *luastate, const char *msg);
 int LuaReturnStringBuffer(lua_State *luastate, const uint8_t *input, size_t input_len);

diff --git a/src/output-lua.c b/src/output-lua.c
index 8ed2a1062db6fb3b06a0b318fad30acf4807e5aa..762ef66683cab7758d05829fc2a181758dd34a71 100644 (file)
--- a/src/output-lua.c
+++ b/src/output-lua.c
@@ -69,6 +69,8 @@ typedef struct LogLuaThreadCtx_ {
 } LogLuaThreadCtx;
 
 const char lualog_ext_key_tx[] = "suricata:lualog:tx:ptr";
+/* key for p (packet) pointer */
+const char lualog_ext_key_p[] = "suricata:lualog:pkt:ptr";
 
 static int LuaTxLogger(ThreadVars *tv, void *thread_data, const Packet *p, Flow *f, void *alstate, void *txptr, uint64_t tx_id)
 {