From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Thu, 17 Mar 2016 12:15:18 +0000 (+0000)
Subject: [Feature] Add MIME_BAD_ATTACHMENT rule for bad attachment types
X-Git-Tag: 1.2.0~19
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=337c39f4834384f822f73cc2a07abea1e74b9a65;p=rspamd.git

[Feature] Add MIME_BAD_ATTACHMENT rule for bad attachment types
---

diff --git a/conf/metrics.conf b/conf/metrics.conf
index 4be8325c5..fc5eeb8f5 100644
--- a/conf/metrics.conf
+++ b/conf/metrics.conf
@@ -1243,6 +1243,12 @@ metric {
             description = "Missing or unknown content-type";
             one_shot = true;
         }
+        symbol {
+            weight = 4.0;
+            name = "MIME_BAD_ATTACHMENT";
+            description = "Invalid attachement mime type";
+            one_shot = true;
+        }
     }
 
     .include(try=true; priority=1) "$LOCAL_CONFDIR/local.d/metrics.conf"
diff --git a/src/plugins/lua/mime_types.lua b/src/plugins/lua/mime_types.lua
index d2ce0d871..7cc0043a1 100644
--- a/src/plugins/lua/mime_types.lua
+++ b/src/plugins/lua/mime_types.lua
@@ -16,38 +16,63 @@ limitations under the License.
 
 -- This plugin implements mime types checks for mail messages
 local rspamd_logger = require "rspamd_logger"
+local rspamd_regexp = require "rspamd_regexp"
 
 local settings = {
   file = '',
   symbol_unknown = 'MIME_UNKNOWN',
   symbol_bad = 'MIME_BAD',
   symbol_good = 'MIME_GOOD',
+  symbol_attachment = 'MIME_BAD_ATTACHMENT',
+  regexp = false,
+  extension_map = { -- extension -> mime_type
+    html = 'text/html',
+    txt = 'text/plain',
+    pdf = 'application/pdf'
+  },
 }
 
 local map = nil
 
 local function check_mime_type(task)
   local parts = task:get_parts()
-  
+
   if parts then
     for _,p in ipairs(parts) do
       local type,subtype = p:get_type()
-      
+
       if not type then
         task:insert_result(settings['symbol_unknown'], 1.0, 'missing content type')
       else
+        -- Check for attachment
+        local filename = p:get_filename()
         local ct = string.format('%s/%s', type, subtype)
-        local v = map:get_key(ct)
-        if v then
-          local n = tonumber(v)
-          
-          if n > 0 then
-            task:insert_result(settings['symbol_bad'], n, ct)
-          elseif n < 0 then
-            task:insert_result(settings['symbol_good'], -n, ct)
+
+        if filename then
+          local ext = string.match(filename, '%.([^.]+)$')
+
+          if ext then
+            if settings['extension_map'] then
+              if ct ~= settings['extension_map'] then
+                task:insert_result(settings['symbol_attachment'], 1.0, ext)
+              end
+            end
+          end
+        end
+
+        if map then
+          local v = map:get_key(ct)
+          if v then
+            local n = tonumber(v)
+
+            if n > 0 then
+              task:insert_result(settings['symbol_bad'], n, ct)
+            elseif n < 0 then
+              task:insert_result(settings['symbol_good'], -n, ct)
+            end
+          else
+            task:insert_result(settings['symbol_unknown'], 1.0, ct)
           end
-        else
-          task:insert_result(settings['symbol_unknown'], 1.0, ct)
         end
       end
     end
@@ -59,7 +84,7 @@ if opts then
   for k,v in pairs(opts) do
     settings[k] = v
   end
-  
+
   if settings['file'] and #settings['file'] > 0 then
     map = rspamd_config:add_kv_map (settings['file'], 
       'mime types map')
@@ -72,5 +97,10 @@ if opts then
       rspamd_logger.warnx(rspamd_config, 'Cannot add mime_types: map doesn\'t exists: %1',
         settings['file'])
     end
+    local id = rspamd_config:register_callback_symbol(1.0, check_mime_type)
+    rspamd_config:register_virtual_symbol(settings['symbol_unknown'], 1.0, id)
+    rspamd_config:register_virtual_symbol(settings['symbol_bad'], 1.0, id)
+    rspamd_config:register_virtual_symbol(settings['symbol_good'], 1.0, id)
+    rspamd_config:register_virtual_symbol(settings['symbol_attachment'], 1.0, id)
   end
-end
\ No newline at end of file
+end