* Stylify and fix util functions that was gathered from libutil code (memory allocation)

* Fix protocol and network functions
* Fix some memory allocation issues

1 files changed, 3 insertions, 0 deletions

diff --git a/main.h b/main.h
index de8969de8..2da935546 100644
--- a/main.h
+++ b/main.h
@@ -109,6 +109,7 @@ struct save_point {
 /* Control session */
 struct controller_session {
 	struct rspamd_worker *worker;
+	int sock;
 	/* Access to authorized commands */
 	int authorized;
 	memory_pool_t *session_pool;
@@ -126,10 +127,12 @@ struct worker_task {
 		WRITE_REPLY,
 		WRITE_ERROR,
 		WAIT_FILTER,
+		CLOSING_CONNECTION,
 	} state;
 	size_t content_length;
 	enum rspamd_protocol proto;
 	enum rspamd_command cmd;
+	int sock;
 	char *helo;
 	char *from;
 	char *rcpt;