[Fix] Fix lua_util.str_split in case of delimiters set

diff --git a/lualib/lua_util.lua b/lualib/lua_util.lua
index 2a2a80fc5e2302d92f7bb9d6613b002280a8be7e..e8fbe2deeffcbce4a667db0ccf869deb571d78d6 100644 (file)
--- a/lualib/lua_util.lua
+++ b/lualib/lua_util.lua
@@ -46,7 +46,12 @@ local function rspamd_str_split(s, sep)
     gr = split_grammar[sep]
 
     if not gr then
-      local _sep = lpeg.P(sep)
+      local _sep
+      if type(sep) == 'string' then
+        _sep = lpeg.S(sep) -- Assume set
+      else
+        _sep = sep -- Assume lpeg object
+      end
       local elem = lpeg.C((1 - _sep)^0)
       local p = lpeg.Ct(elem * (_sep * elem)^0)
       gr = p