From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Mon, 2 Dec 2019 15:42:25 +0000 (+0000)
Subject: [Minor] Lua_task: Allow to get a raw timestamp
X-Git-Tag: 2.3~262
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=946c6d5af333c78914a37ada3ace683c1b782988;p=rspamd.git

[Minor] Lua_task: Allow to get a raw timestamp
---

diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c
index aff6d8499..26ad15dec 100644
--- a/src/lua/lua_task.c
+++ b/src/lua/lua_task.c
@@ -4889,14 +4889,19 @@ lua_task_get_timeval (lua_State *L)
 	struct timeval tv;
 
 	if (task != NULL) {
-		double_to_tv (task->task_timestamp, &tv);
-		lua_createtable (L, 0, 2);
-		lua_pushstring (L, "tv_sec");
-		lua_pushinteger (L, (lua_Integer)tv.tv_sec);
-		lua_settable (L, -3);
-		lua_pushstring (L, "tv_usec");
-		lua_pushinteger (L, (lua_Integer)tv.tv_usec);
-		lua_settable (L, -3);
+		if (lua_isboolean (L, 2) && !!lua_toboolean (L, 2)) {
+			lua_pushnumber (L, task->task_timestamp);
+		}
+		else {
+			double_to_tv (task->task_timestamp, &tv);
+			lua_createtable (L, 0, 2);
+			lua_pushstring (L, "tv_sec");
+			lua_pushinteger (L, (lua_Integer) tv.tv_sec);
+			lua_settable (L, -3);
+			lua_pushstring (L, "tv_usec");
+			lua_pushinteger (L, (lua_Integer) tv.tv_usec);
+			lua_settable (L, -3);
+		}
 	}
 	else {
 		return luaL_error (L, "invalid arguments");