From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Wed, 30 Apr 2014 22:33:42 +0000 (-0700)
Subject: Save status for HTTP errors.
X-Git-Tag: 0.7.0~248
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4d0870aee42ede6ecced8ddbbb6ab3626db3d96f;p=thirdparty%2Frspamd.git

Save status for HTTP errors.
---

diff --git a/src/libutil/http.c b/src/libutil/http.c
index 26dc679007..1b17b59577 100644
--- a/src/libutil/http.c
+++ b/src/libutil/http.c
@@ -392,6 +392,24 @@ rspamd_http_on_url (http_parser* parser, const gchar *at, size_t length)
 	return 0;
 }
 
+static gint
+rspamd_http_on_status (http_parser* parser, const gchar *at, size_t length)
+{
+	struct rspamd_http_connection *conn = (struct rspamd_http_connection *)parser->data;
+	struct rspamd_http_connection_private *priv;
+
+	priv = conn->priv;
+
+	if (parser->status_code != 200) {
+		if (priv->msg->status == NULL) {
+			priv->msg->status = g_string_sized_new (128);
+		}
+		g_string_append_len (priv->msg->status, at, length);
+	}
+
+	return 0;
+}
+
 static gint
 rspamd_http_on_header_field (http_parser* parser, const gchar *at, size_t length)
 {
@@ -664,6 +682,7 @@ rspamd_http_connection_new (rspamd_http_body_handler_t body_handler,
 	http_parser_init (&priv->parser, type == RSPAMD_HTTP_SERVER ? HTTP_REQUEST : HTTP_RESPONSE);
 	priv->parser.data = new;
 	priv->parser_cb.on_url = rspamd_http_on_url;
+	priv->parser_cb.on_status = rspamd_http_on_status;
 	priv->parser_cb.on_header_field = rspamd_http_on_header_field;
 	priv->parser_cb.on_header_value = rspamd_http_on_header_value;
 	priv->parser_cb.on_headers_complete = rspamd_http_on_headers_complete;
@@ -898,6 +917,7 @@ rspamd_http_new_message (enum http_parser_type type)
 	new->headers = NULL;
 	new->date = 0;
 	new->body = NULL;
+	new->status = NULL;
 	new->type = type;
 	new->method = HTTP_GET;
 
@@ -920,6 +940,9 @@ rspamd_http_message_free (struct rspamd_http_message *msg)
 	if (msg->url != NULL) {
 		g_string_free (msg->url, TRUE);
 	}
+	if (msg->status != NULL) {
+		g_string_free (msg->status, TRUE);
+	}
 	g_slice_free1 (sizeof (struct rspamd_http_message), msg);
 }
 
diff --git a/src/libutil/http.h b/src/libutil/http.h
index a761356164..695f238962 100644
--- a/src/libutil/http.h
+++ b/src/libutil/http.h
@@ -53,6 +53,7 @@ struct rspamd_http_header {
  */
 struct rspamd_http_message {
 	GString *url;
+	GString *status;
 	struct rspamd_http_header *headers;
 	GString *body;
 	enum http_parser_type type;