[Minor] Convert milter macros to HTTP headers

diff --git a/src/libserver/milter.c b/src/libserver/milter.c
index a45f19c77472d2656dafce986664bbc789496795..15835a8017fc51e9921a0a6026b17e5307a281f0 100644 (file)
--- a/src/libserver/milter.c
+++ b/src/libserver/milter.c
@@ -424,6 +424,11 @@ rspamd_milter_process_command (struct rspamd_milter_session *session,
                                        msg_debug_milter ("got macro: %T -> %T",
                                                        name_tok, value_tok);
 
+                                       if (rspamd_ftok_cstr_equal (name_tok, "{mail_host}", FALSE)) {
+                                               session->hostname = rspamd_fstring_assign (session->hostname,
+                                                               value_tok->begin, value_tok->len);
+                                       }
+
                                        cmdlen -= zero_val - pos;
                                        pos = zero_val + 1;
                                }
@@ -1085,6 +1090,51 @@ rspamd_milter_session_ref (struct rspamd_milter_session *session)
        return session;
 }
 
+#define IF_MACRO(lit) RSPAMD_FTOK_ASSIGN (&srch, (lit)); \
+       found = g_hash_table_lookup (session->macros, &srch); \
+       if (found)
+
+static void
+rspamd_milter_macro_http (struct rspamd_milter_session *session,
+               struct rspamd_http_message *msg)
+{
+       rspamd_ftok_t *found, srch;
+
+       /*
+        * We assume postfix macros here, sendmail ones might be slightly
+        * different
+        */
+
+       if (!session->macros) {
+               return;
+       }
+
+       IF_MACRO("i") {
+               rspamd_http_message_add_header_len (msg, QUEUE_ID_HEADER,
+                               found->begin, found->len);
+       }
+
+       IF_MACRO("{daemon_name}") {
+               rspamd_http_message_add_header_len (msg, MTA_TAG_HEADER,
+                               found->begin, found->len);
+       }
+
+       IF_MACRO("{v}") {
+               rspamd_http_message_add_header_len (msg, USER_AGENT_HEADER,
+                               found->begin, found->len);
+       }
+
+       IF_MACRO("{cipher}") {
+               rspamd_http_message_add_header_len (msg, TLS_CIPHER_HEADER,
+                               found->begin, found->len);
+       }
+
+       IF_MACRO("{tls_version}") {
+               rspamd_http_message_add_header_len (msg, TLS_VERSION_HEADER,
+                               found->begin, found->len);
+       }
+}
+
 struct rspamd_http_message *
 rspamd_milter_to_http (struct rspamd_milter_session *session)
 {
@@ -1125,5 +1175,7 @@ rspamd_milter_to_http (struct rspamd_milter_session *session)
                }
        }
 
+       rspamd_milter_macro_http (session, msg);
+
        return msg;
 }
\ No newline at end of file

diff --git a/src/libserver/protocol_internal.h b/src/libserver/protocol_internal.h
index ac588386b5effa77a1d672105b604dedf6522d12..7be8214620c2630c51e4fd81ac372cecbc879dac 100644 (file)
--- a/src/libserver/protocol_internal.h
+++ b/src/libserver/protocol_internal.h
@@ -75,6 +75,8 @@
 #define USER_AGENT_HEADER "User-Agent"
 #define MTA_TAG_HEADER "MTA-Tag"
 #define PROFILE_HEADER "Profile"
+#define TLS_CIPHER_HEADER "TLS-Cipher"
+#define TLS_VERSION_HEADER "TLS-Version"
 
 
 #endif //RSPAMD_PROTOCOL_INTERNAL_H