[Minor] Allow to pass socket's ownership to http connection object

2 files changed, 14 insertions, 0 deletions

diff --git a/src/libserver/http/http_connection.c b/src/libserver/http/http_connection.c
index 4e880f216..050fb4285 100644
--- a/src/libserver/http/http_connection.c
+++ b/src/libserver/http/http_connection.c
@@ -2344,6 +2344,14 @@ rspamd_http_connection_set_key (struct rspamd_http_connection *conn,
 	priv->local_key = rspamd_keypair_ref (key);
 }
 
+void
+rspamd_http_connection_own_socket (struct rspamd_http_connection *conn)
+{
+	struct rspamd_http_connection_private *priv = conn->priv;
+
+	priv->flags |= RSPAMD_HTTP_CONN_OWN_SOCKET;
+}
+
 const struct rspamd_cryptobox_pubkey*
 rspamd_http_connection_get_peer_key (struct rspamd_http_connection *conn)
 {
diff --git a/src/libserver/http/http_connection.h b/src/libserver/http/http_connection.h
index 7c901fd2a..ada98d250 100644
--- a/src/libserver/http/http_connection.h
+++ b/src/libserver/http/http_connection.h
@@ -203,6 +203,12 @@ void rspamd_http_connection_set_key (struct rspamd_http_connection *conn,
 									 struct rspamd_cryptobox_keypair *key);
 
 /**
+ * Transfer ownership on socket to an HTTP connection
+ * @param conn
+ */
+void rspamd_http_connection_own_socket (struct rspamd_http_connection *conn);
+
+/**
  * Get peer's public key
  * @param conn connection structure
  * @return pubkey structure or NULL