From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Mon, 2 Aug 2021 20:32:05 +0000 (+0100)
Subject: [Minor] Lua: Add rspamd_util.strequal_caseless_utf8 routine
X-Git-Tag: 3.0~59
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=4854ab4df2cb13caf052b02512cfc93364e861d5;p=rspamd.git

[Minor] Lua: Add rspamd_util.strequal_caseless_utf8 routine
---

diff --git a/src/lua/lua_util.c b/src/lua/lua_util.c
index 93235d262..8683f4801 100644
--- a/src/lua/lua_util.c
+++ b/src/lua/lua_util.c
@@ -260,6 +260,18 @@ LUA_FUNCTION_DEF (util, strcasecmp_ascii);
  */
 LUA_FUNCTION_DEF (util, strequal_caseless);
 
+
+/***
+ * @function util.strequal_caseless(str1, str2)
+ * Compares two utf8 strings regardless of their case. Return `true` if `str1` is
+ * equal to `str2`
+ * @param {string} str1 utf8 encoded string
+ * @param {string} str2 utf8 encoded string
+ * @return {bool} result of comparison
+ */
+LUA_FUNCTION_DEF (util, strequal_caseless_utf8);
+
+
 /***
  * @function util.get_ticks()
  * Returns current number of ticks as floating point number
@@ -673,6 +685,7 @@ static const struct luaL_reg utillib_f[] = {
 	LUA_INTERFACE_DEF (util, lower_utf8),
 	LUA_INTERFACE_DEF (util, strcasecmp_ascii),
 	LUA_INTERFACE_DEF (util, strequal_caseless),
+	LUA_INTERFACE_DEF (util, strequal_caseless_utf8),
 	LUA_INTERFACE_DEF (util, get_ticks),
 	LUA_INTERFACE_DEF (util, get_time),
 	LUA_INTERFACE_DEF (util, time_to_string),
@@ -1641,6 +1654,34 @@ lua_util_strequal_caseless (lua_State *L)
 	return 1;
 }
 
+static gint
+lua_util_strequal_caseless_utf8 (lua_State *L)
+{
+	LUA_TRACE_POINT;
+	struct rspamd_lua_text *t1, *t2;
+	gint ret = -1;
+
+	t1 = lua_check_text_or_string (L, 1);
+	t2 = lua_check_text_or_string (L, 2);
+
+	if (t1 && t2) {
+
+		if (t1->len == t2->len) {
+			ret = rspamd_utf8_strcmp_sizes(t1->start, t1->len, t2->start, t2->len);
+		}
+		else {
+			ret = t1->len - t2->len;
+		}
+	}
+	else {
+		return luaL_error (L, "invalid arguments");
+	}
+
+	lua_pushboolean (L, (ret == 0) ? true : false);
+
+	return 1;
+}
+
 static gint
 lua_util_get_ticks (lua_State *L)
 {