elua lib: add elua_state_from_lua_get

1 files changed, 16 insertions, 0 deletions

diff --git a/src/lib/elua/elua.c b/src/lib/elua/elua.c
index a6bb3914cd..bdd5606fed 100644
--- a/src/lib/elua/elua.c
+++ b/src/lib/elua/elua.c
@@ -72,6 +72,8 @@ elua_state_new(void)
    ret = malloc(sizeof(Elua_State));
    ret->luastate = L;
    luaL_openlibs(L);
+   lua_pushlightuserdata(L, ret);
+   lua_setfield(L, LUA_REGISTRYINDEX, "elua_ptr");
    return ret;
 }
 
@@ -83,6 +85,20 @@ elua_state_free(Elua_State *state)
    free(state);
 }
 
+EAPI Elua_State *
+elua_state_from_lua_get(lua_State *L)
+{
+   lua_getfield(L, LUA_REGISTRYINDEX, "elua_ptr");
+   if (!lua_isnil(L, -1))
+     {
+        void *st = lua_touserdata(L, -1);
+        lua_pop(L, 1);
+        return (Elua_State *)st;
+     }
+   lua_pop(L, 1);
+   return NULL;
+}
+
 static void
 _elua_errmsg(const char *pname, const char *msg)
 {