From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Wed, 30 Apr 2014 16:09:31 +0000 (+0100)
Subject: Increase HTTP conn refcount during task processing.
X-Git-Tag: 0.7.0~257
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7cc961501dfae75d2c389e95375de43d2332e444;p=thirdparty%2Frspamd.git

Increase HTTP conn refcount during task processing.
---

diff --git a/src/webui.c b/src/webui.c
index 685e176043..aa009f9e56 100644
--- a/src/webui.c
+++ b/src/webui.c
@@ -795,10 +795,12 @@ rspamd_webui_learn_fin_task (void *ud)
 
 	if (!learn_task_spam (session->cl, task, session->is_spam, &err)) {
 		rspamd_controller_send_error (conn_ent, 500 + err->code, err->message);
+		rspamd_http_connection_unref (conn_ent->conn);
 		return TRUE;
 	}
 	/* Successful learn */
 	rspamd_controller_send_string (conn_ent, "{\"success\":true}");
+	rspamd_http_connection_unref (conn_ent->conn);
 
 	return TRUE;
 }
@@ -813,6 +815,7 @@ rspamd_webui_check_fin_task (void *ud)
 	task->http_conn = conn_ent->conn;
 	rspamd_protocol_write_reply (task);
 	conn_ent->is_reply = TRUE;
+	rspamd_http_connection_unref (conn_ent->conn);
 
 	return TRUE;
 }
@@ -856,6 +859,7 @@ rspamd_webui_handle_learn_common (struct rspamd_http_connection_entry *conn_ent,
 	task->resolver = ctx->resolver;
 	task->ev_base = ctx->ev_base;
 
+	rspamd_http_connection_ref (conn_ent->conn);
 	task->s = new_async_session (session->pool, rspamd_webui_learn_fin_task, NULL,
 			rspamd_task_free_hard, task);
 	task->s->wanna_die = TRUE;
@@ -936,6 +940,7 @@ rspamd_webui_handle_scan (struct rspamd_http_connection_entry *conn_ent,
 	task->resolver = ctx->resolver;
 	task->ev_base = ctx->ev_base;
 
+	rspamd_http_connection_ref (conn_ent->conn);
 	task->s = new_async_session (session->pool, rspamd_webui_check_fin_task, NULL,
 			rspamd_task_free_hard, task);
 	task->s->wanna_die = TRUE;