commit c6f7b89: [Minor] Fix some issues in URI_COUNT_ODD rule

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2022-01-11 20:25:52 +0000
URL: https://github.com/rspamd/rspamd/commit/c6f7b897d45cbe4a166334ca60d29b3b7391bd3a (HEAD -> master)

[Minor] Fix some issues in URI_COUNT_ODD rule
Issue: #4037

---
 rules/misc.lua | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/rules/misc.lua b/rules/misc.lua
index 2e5f38800..9c8cade71 100644
--- a/rules/misc.lua
+++ b/rules/misc.lua
@@ -338,10 +338,8 @@ rspamd_config.URI_COUNT_ODD = {
   callback = function (task)
     local ct = task:get_header('Content-Type')
     if (ct and ct:lower():find('^multipart/alternative')) then
-      local urls = task:get_urls() or {}
-      local nurls = fun.filter(function(url)
-        return not url:is_html_displayed()
-      end, urls):foldl(function(acc, val) return acc + val:get_count() end, 0)
+      local urls = task:get_urls_filtered(nil, {'subject', 'html_displayed', 'special'}) or {}
+      local nurls = fun.foldl(function(acc, val) return acc + val:get_count() end, 0)
 
       if nurls % 2 == 1 then
         return true, 1.0, tostring(nurls)