From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sat, 9 Apr 2016 12:13:38 +0000 (+0100)
Subject: [Feature] Allow to get mime headers from a task
X-Git-Tag: 1.2.3~27
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=a36ee5a03288e9fabf82778a4a1a197bf0e69327;p=thirdparty%2Frspamd.git

[Feature] Allow to get mime headers from a task
---

diff --git a/src/libmime/message.c b/src/libmime/message.c
index f14c3b1857..a7a3f74391 100644
--- a/src/libmime/message.c
+++ b/src/libmime/message.c
@@ -1985,8 +1985,57 @@ rspamd_message_get_header_array (struct rspamd_task *task,
 	ret = g_ptr_array_sized_new (nelems);
 
 	LL_FOREACH (rh, cur) {
+		if (strong) {
+			if (strcmp (rh->name, field) != 0) {
+				continue;
+			}
+		}
+
 		g_ptr_array_add (ret, cur);
 	}
 
 	return ret;
 }
+
+GPtrArray *
+rspamd_message_get_mime_header_array (struct rspamd_task *task,
+		const gchar *field,
+		gboolean strong)
+{
+	GPtrArray *ret;
+	struct raw_header *rh, *cur;
+	guint nelems = 0, i;
+	struct mime_part *mp;
+
+	for (i = 0; i < task->parts->len; i ++) {
+		mp = g_ptr_array_index (task->parts, i);
+		rh = g_hash_table_lookup (mp->raw_headers, field);
+
+		if (rh == NULL) {
+			continue;
+		}
+
+		LL_FOREACH (rh, cur) {
+			nelems ++;
+		}
+	}
+
+	ret = g_ptr_array_sized_new (nelems);
+
+	for (i = 0; i < task->parts->len; i ++) {
+		mp = g_ptr_array_index (task->parts, i);
+		rh = g_hash_table_lookup (mp->raw_headers, field);
+
+		LL_FOREACH (rh, cur) {
+			if (strong) {
+				if (strcmp (rh->name, field) != 0) {
+					continue;
+				}
+			}
+
+			g_ptr_array_add (ret, cur);
+		}
+	}
+
+	return ret;
+}
diff --git a/src/libmime/message.h b/src/libmime/message.h
index 06ff78e7af..d946fa0a49 100644
--- a/src/libmime/message.h
+++ b/src/libmime/message.h
@@ -91,4 +91,8 @@ GPtrArray *rspamd_message_get_header_array (struct rspamd_task *task,
 		const gchar *field,
 		gboolean strong);
 
+GPtrArray *rspamd_message_get_mime_header_array (struct rspamd_task *task,
+		const gchar *field,
+		gboolean strong);
+
 #endif