[Feature] Add lua functions for caseless comparison

diff --git a/src/lua/lua_util.c b/src/lua/lua_util.c
index 04bf76942b283bd6bf7bd548ee724a67d4668936..0707f098ccf58d5bb3dc2fa1bc8489013981c386 100644 (file)
--- a/src/lua/lua_util.c
+++ b/src/lua/lua_util.c
@@ -193,6 +193,36 @@ LUA_FUNCTION_DEF (util, parse_mail_address);
  */
 LUA_FUNCTION_DEF (util, strlen_utf8);
 
+/***
+ * @function util.strcasecmp(str1, str2)
+ * Compares two utf8 strings regardless of their case. Return value >0, 0 and <0
+ * if `str1` is more, equal or less than `str2`
+ * @param {string} str1 utf8 encoded string
+ * @param {string} str2 utf8 encoded string
+ * @return {number} result of comparison
+ */
+LUA_FUNCTION_DEF (util, strcasecmp_utf8);
+
+/***
+ * @function util.strcasecmp(str1, str2)
+ * Compares two ascii strings regardless of their case. Return value >0, 0 and <0
+ * if `str1` is more, equal or less than `str2`
+ * @param {string} str1 plain string
+ * @param {string} str2 plain string
+ * @return {number} result of comparison
+ */
+LUA_FUNCTION_DEF (util, strcasecmp_ascii);
+
+/***
+ * @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);
+
 static const struct luaL_reg utillib_f[] = {
        LUA_INTERFACE_DEF (util, create_event_base),
        LUA_INTERFACE_DEF (util, load_rspamd_config),
@@ -214,6 +244,9 @@ static const struct luaL_reg utillib_f[] = {
        LUA_INTERFACE_DEF (util, glob),
        LUA_INTERFACE_DEF (util, parse_mail_address),
        LUA_INTERFACE_DEF (util, strlen_utf8),
+       LUA_INTERFACE_DEF (util, strcasecmp_utf8),
+       LUA_INTERFACE_DEF (util, strcasecmp_ascii),
+       LUA_INTERFACE_DEF (util, strequal_caseless),
        {NULL, NULL}
 };
 
@@ -941,6 +974,87 @@ lua_util_strlen_utf8 (lua_State *L)
        return 1;
 }
 
+static gint
+lua_util_strcasecmp_utf8 (lua_State *L)
+{
+       const gchar *str1, *str2;
+       gsize len1, len2;
+       gint ret = -1;
+
+       str1 = lua_tolstring (L, 1, &len1);
+       str2 = lua_tolstring (L, 1, &len2);
+
+       if (str1 && str2) {
+
+               if (len1 == len2) {
+                       ret = rspamd_lc_cmp (str1, str2, len1);
+               }
+               else {
+                       ret = len1 - len2;
+               }
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       lua_pushnumber (L, ret);
+       return 1;
+}
+
+static gint
+lua_util_strcasecmp_ascii (lua_State *L)
+{
+       const gchar *str1, *str2;
+       gsize len1, len2;
+       gint ret = -1;
+
+       str1 = lua_tolstring (L, 1, &len1);
+       str2 = lua_tolstring (L, 1, &len2);
+
+       if (str1 && str2) {
+
+               if (len1 == len2) {
+                       ret = g_ascii_strncasecmp (str1, str2, len1);
+               }
+               else {
+                       ret = len1 - len2;
+               }
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       lua_pushnumber (L, ret);
+       return 1;
+}
+
+static gint
+lua_util_strequal_caseless (lua_State *L)
+{
+       const gchar *str1, *str2;
+       gsize len1, len2;
+       gint ret = -1;
+
+       str1 = lua_tolstring (L, 1, &len1);
+       str2 = lua_tolstring (L, 1, &len2);
+
+       if (str1 && str2) {
+
+               if (len1 == len2) {
+                       ret = rspamd_lc_cmp (str1, str2, len1);
+               }
+               else {
+                       ret = len1 - len2;
+               }
+       }
+       else {
+               return luaL_error (L, "invalid arguments");
+       }
+
+       lua_pushboolean (L, (ret == 0));
+       return 1;
+}
+
 static gint
 lua_load_util (lua_State * L)
 {