From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Thu, 3 Mar 2016 16:53:00 +0000 (+0000)
Subject: [Fix] Add watcher for lua_tcp module
X-Git-Tag: 1.2.0~126
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ff3ae310006f00880f97d134c6a73ee12ca875c1;p=thirdparty%2Frspamd.git

[Fix] Add watcher for lua_tcp module
---

diff --git a/src/lua/lua_tcp.c b/src/lua/lua_tcp.c
index 93050eb2d2..a6ebb2dc2a 100644
--- a/src/lua/lua_tcp.c
+++ b/src/lua/lua_tcp.c
@@ -65,6 +65,7 @@ struct lua_tcp_cbdata {
 	struct iovec *iov;
 	GString *in;
 	gchar *stop_pattern;
+	struct rspamd_async_watcher *w;
 	struct event ev;
 	gint fd;
 	gint cbref;
@@ -113,6 +114,7 @@ lua_tcp_maybe_free (struct lua_tcp_cbdata *cbd)
 {
 	if (cbd->session) {
 		rspamd_session_remove_event (cbd->session, lua_tcp_fin, cbd);
+		rspamd_session_watcher_pop (cbd->session, cbd->w);
 	}
 	else {
 		lua_tcp_fin (cbd);
@@ -609,6 +611,8 @@ lua_tcp_request (lua_State *L)
 				(event_finalizer_t)lua_tcp_fin,
 				cbd,
 				g_quark_from_static_string ("lua tcp"));
+		cbd->w = rspamd_session_get_watcher (session);
+		rspamd_session_watcher_push (session);
 	}
 
 	if (rspamd_parse_inet_address (&cbd->addr, host, 0)) {