From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Tue, 14 Jul 2015 23:22:03 +0000 (+0100)
Subject: Allow to get different types of memory pool variables.
X-Git-Tag: 1.0.0~363
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=f999cd8e8b6689ea73ba1b6d9a54ae63dde1aa86;p=thirdparty%2Frspamd.git

Allow to get different types of memory pool variables.
---

diff --git a/src/lua/lua_mempool.c b/src/lua/lua_mempool.c
index 3fe33ade63..af19c64b5e 100644
--- a/src/lua/lua_mempool.c
+++ b/src/lua/lua_mempool.c
@@ -199,11 +199,42 @@ lua_mempool_get_variable (lua_State *L)
 {
 	struct memory_pool_s *mempool = rspamd_lua_check_mempool (L, 1);
 	const gchar *var = luaL_checkstring (L, 2);
+	const gchar *type = NULL;
 	gchar *value;
 
 	if (mempool && var) {
 		value = rspamd_mempool_get_variable (mempool, var);
+
+		if (lua_gettop (L) >= 3) {
+			type = luaL_checkstring (L, 3);
+		}
+
 		if (value) {
+
+			if (type) {
+				if (g_ascii_strcasecmp (type, "double") == 0) {
+					lua_pushnumber (L, *(gdouble *)value);
+				}
+				else if (g_ascii_strcasecmp (type, "int") == 0) {
+					lua_pushnumber (L, *(gint *)value);
+				}
+				else if (g_ascii_strcasecmp (type, "int64") == 0) {
+					lua_pushnumber (L, *(gint64 *)value);
+				}
+				else if (g_ascii_strcasecmp (type, "bool") == 0) {
+					lua_pushboolean (L, *(gboolean *)value);
+				}
+				else if (g_ascii_strcasecmp (type, "string") == 0) {
+					lua_pushstring (L, (const gchar *)value);
+				}
+				else {
+					msg_err ("unknown type for get_variable: %s", type);
+					lua_pushnil (L);
+				}
+
+				return 1;
+			}
+
 			lua_pushstring (L, value);
 		}
 		else {