]> source.dussan.org Git - rspamd.git/commitdiff
[Minor] Add symbols and type heuristic
authorVsevolod Stakhov <vsevolod@highsecure.ru>
Sat, 18 May 2019 13:36:32 +0000 (14:36 +0100)
committerVsevolod Stakhov <vsevolod@highsecure.ru>
Sat, 18 May 2019 13:36:32 +0000 (14:36 +0100)
lualib/lua_maps_expressions.lua

index 38cd0f296cf4f696e92dcb493ef065e90e9011d5..f0b1f176a1998e2ab897e58e100c9a4da9890c5f 100644 (file)
@@ -129,7 +129,19 @@ local function create(cfg, obj, module_name)
           name, module_name)
     end
 
-    local map = lua_maps.map_add_from_ucl(rule.map, rule.type or 'set',
+    if not rule.type then
+      -- Guess type
+      if name:find('ip') or name:find('ipnet') then
+        rule.type = 'radix'
+      elseif name:find('regexp') or name:find('re_') then
+        rule.type = 'regexp'
+      elseif name:find('glob') then
+        rule.type = 'regexp'
+      else
+        rule.type = 'set'
+      end
+    end
+    local map = lua_maps.map_add_from_ucl(rule.map, rule.type,
         obj.description or module_name)
     if not map then
       rspamd_logger.errx(cfg, 'cannot add map for element %s in module %s',
@@ -174,6 +186,14 @@ local function create(cfg, obj, module_name)
 
   ret.expr = expr
 
+  if obj.symbol then
+    rspamd_config:register_symbol{
+      type = 'virtual,ghost',
+      name = obj.symbol,
+      score = 0.0,
+    }
+  end
+
   return ret
 end