[Fix] Various reload leak fixing

2 files changed, 37 insertions, 0 deletions

diff --git a/src/lua/lua_common.c b/src/lua/lua_common.c
index b9eabe790..7f4cea7b7 100644
--- a/src/lua/lua_common.c
+++ b/src/lua/lua_common.c
@@ -1264,3 +1264,31 @@ lua_worker_get_pid (lua_State *L)
 
 	return 1;
 }
+
+struct rspamd_lua_ref_cbdata {
+	lua_State *L;
+	gint cbref;
+};
+
+static void
+rspamd_lua_ref_dtor (gpointer p)
+{
+	struct rspamd_lua_ref_cbdata *cbdata = p;
+
+	luaL_unref (cbdata->L, LUA_REGISTRYINDEX, cbdata->cbref);
+}
+
+void
+rspamd_lua_add_ref_dtor (lua_State *L, rspamd_mempool_t *pool,
+		gint ref)
+{
+	struct rspamd_lua_ref_cbdata *cbdata;
+
+	if (ref != -1) {
+		cbdata = rspamd_mempool_alloc (pool, sizeof (*cbdata));
+		cbdata->cbref = ref;
+		cbdata->L = L;
+
+		rspamd_mempool_add_destructor (pool, rspamd_lua_ref_dtor, cbdata);
+	}
+}
diff --git a/src/lua/lua_common.h b/src/lua/lua_common.h
index dbb46c822..ecf46d32e 100644
--- a/src/lua/lua_common.h
+++ b/src/lua/lua_common.h
@@ -360,5 +360,14 @@ void lua_call_finish_script (lua_State *L, struct
 gboolean rspamd_lua_run_postloads (lua_State *L, struct rspamd_config *cfg,
 		struct event_base *ev_base, struct rspamd_worker *w);
 
+/**
+ * Adds new destructor for a local function for specific pool
+ * @param L
+ * @param pool
+ * @param ref
+ */
+void rspamd_lua_add_ref_dtor (lua_State *L, rspamd_mempool_t *pool,
+		gint ref);
+
 #endif /* WITH_LUA */
 #endif /* RSPAMD_LUA_H */