From 81e8e5e413ad8ba9dfb8b8f7da5ab2939dac4534 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Wed, 21 Dec 2016 14:18:42 +0000
Subject: [PATCH] [Feature] Use own code for parsing of date

---
 src/CMakeLists.txt            |  8 +++++++-
 src/libmime/smtp_parsers.h    |  2 ++
 src/lua/lua_task.c            | 16 ++++++++++++----
 src/ragel/smtp_date_parser.rl | 28 ++++++++++++++++++++++++++++
 4 files changed, 49 insertions(+), 5 deletions(-)
 create mode 100644 src/ragel/smtp_date_parser.rl

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 7254eefc2..36ab66f73 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -138,6 +138,11 @@ RAGEL_TARGET(ragel_rfc2047
 	DEPENDS ${RAGEL_DEPENDS}
 	COMPILE_FLAGS -G2
 	OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/rfc2047.rl.c)
+RAGEL_TARGET(ragel_smtp_date
+	INPUTS ${CMAKE_SOURCE_DIR}/src/ragel/smtp_date_parser.rl
+	DEPENDS ${RAGEL_DEPENDS}
+	COMPILE_FLAGS -G2
+	OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/date_parser.rl.c)
 ######################### LINK SECTION ###############################
 
 ADD_LIBRARY(rspamd-server STATIC
@@ -154,7 +159,8 @@ ADD_LIBRARY(rspamd-server STATIC
 		"${RAGEL_ragel_newlines_strip_OUTPUTS}"
 		"${RAGEL_ragel_content_type_OUTPUTS}"
 		"${RAGEL_ragel_content_disposition_OUTPUTS}"
-		"${RAGEL_ragel_rfc2047_OUTPUTS}")
+		"${RAGEL_ragel_rfc2047_OUTPUTS}"
+		"${RAGEL_ragel_smtp_date_OUTPUTS}")
 TARGET_LINK_LIBRARIES(rspamd-server rspamd-http-parser)
 TARGET_LINK_LIBRARIES(rspamd-server rspamd-cdb)
 TARGET_LINK_LIBRARIES(rspamd-server rspamd-lpeg)
diff --git a/src/libmime/smtp_parsers.h b/src/libmime/smtp_parsers.h
index 905a01f0c..3f13abb59 100644
--- a/src/libmime/smtp_parsers.h
+++ b/src/libmime/smtp_parsers.h
@@ -41,4 +41,6 @@ rspamd_rfc2047_parser (const gchar *in, gsize len, gint *pencoding,
 		const gchar **charset, gsize *charset_len,
 		const gchar **encoded, gsize *encoded_len);
 
+guint64 rspamd_parse_smtp_date (const char *data, size_t len);
+
 #endif /* SRC_LIBMIME_SMTP_PARSERS_H_ */
diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c
index 68d9b9704..96c03847c 100644
--- a/src/lua/lua_task.c
+++ b/src/lua/lua_task.c
@@ -26,6 +26,7 @@
 #include "utlist.h"
 #include "cryptobox.h"
 #include "unix-std.h"
+#include "libmime/smtp_parsers.h"
 
 /***
  * @module rspamd_task
@@ -2591,16 +2592,23 @@ lua_task_get_date (lua_State *L)
 
 			if (hdrs && hdrs->len > 0) {
 				time_t tt;
-				gint offset;
+				struct tm t;
 				struct rspamd_mime_header *h;
 
 				h = g_ptr_array_index (hdrs, 0);
-				tt = g_mime_utils_header_decode_date (h->decoded, &offset);
+				tt = rspamd_parse_smtp_date (h->raw_value, h->raw_len);
 
 				if (!gmt) {
-					tt += (offset * 60 * 60) / 100 + (offset * 60 * 60) % 100;
+					localtime_r (&tt, &t);
+#if !defined(__sun)
+					t.tm_gmtoff = 0;
+#endif
+					t.tm_isdst = 0;
+					tim = mktime (&t);
+				}
+				else {
+					tim = tt;
 				}
-				tim = tt;
 			}
 			else {
 				tim = 0.0;
diff --git a/src/ragel/smtp_date_parser.rl b/src/ragel/smtp_date_parser.rl
new file mode 100644
index 000000000..84b63f1bd
--- /dev/null
+++ b/src/ragel/smtp_date_parser.rl
@@ -0,0 +1,28 @@
+%%{
+
+  machine smtp_date_parser;
+  include smtp_date "smtp_date.rl";
+
+  main := date_time;
+}%%
+
+#include "smtp_parsers.h"
+#include "util.h"
+
+%% write data;
+
+guint64
+rspamd_parse_smtp_date (const char *data, size_t len)
+{
+  const gchar *p = data, *pe = data + len, *eof = data + len, *tmp = data;
+  struct tm tm;
+  glong tz = 0;
+  gint cs = 0;
+
+  memset (&tm, 0, sizeof (tm));
+
+  %% write init;
+  %% write exec;
+
+  return rspamd_tm_to_time (&tm, tz);
+}
\ No newline at end of file
-- 
2.39.5