once_received.lua: Fix indentation & add exclusion for authenticated users

1 files changed, 98 insertions, 95 deletions

diff --git a/src/plugins/lua/once_received.lua b/src/plugins/lua/once_received.lua
index 46a3e1b09..6e1243545 100644
--- a/src/plugins/lua/once_received.lua
+++ b/src/plugins/lua/once_received.lua
@@ -8,88 +8,91 @@ local good_hosts = {}
 local rspamd_logger = require "rspamd_logger"
 
 local function check_quantity_received (task)
-	local function recv_dns_cb(resolver, to_resolve, results, err)
-		task:inc_dns_req()
-		if not results then
-			task:insert_result(symbol_strict, 1)
-		else
-			rspamd_logger.info(string.format('SMTP resolver failed to resolve: %s is %s', to_resolve, results[1]))
-			local i = true
-			for _,h in ipairs(bad_hosts) do
-				if string.find(results[1], h) then
-					-- Check for good hostname
-					if good_hosts then
-						for _,gh in ipairs(good_hosts) do
-							if string.find(results[1], gh) then
-								i = false
-								break
-							end
-						end
-					end
-					if i then
-						task:insert_result(symbol_strict, 1, h)
-						return
-					end
-				end
-			end
-		end
-	end
+  local function recv_dns_cb(resolver, to_resolve, results, err)
+    task:inc_dns_req()
+    if not results then
+      task:insert_result(symbol_strict, 1)
+    else
+      rspamd_logger.info(string.format('SMTP resolver failed to resolve: %s is %s', to_resolve, results[1]))
+      local i = true
+      for _,h in ipairs(bad_hosts) do
+        if string.find(results[1], h) then
+          -- Check for good hostname
+          if good_hosts then
+            for _,gh in ipairs(good_hosts) do
+              if string.find(results[1], gh) then
+                i = false
+                break
+              end
+            end
+          end
+          if i then
+            task:insert_result(symbol_strict, 1, h)
+            return
+          end
+        end
+      end
+    end
+  end
 
-	local recvh = task:get_received_headers()
-	if table.maxn(recvh) <= 1 then
-		task:insert_result(symbol, 1)
-		-- Strict checks
-		if symbol_strict then
-			local r = recvh[1]
+  if task:get_user() ~= nil then
+    return
+  end
+  local recvh = task:get_received_headers()
+  if table.maxn(recvh) <= 1 then
+    task:insert_result(symbol, 1)
+    -- Strict checks
+    if symbol_strict then
+      local r = recvh[1]
             if not r then
                 return
             end
-			-- Unresolved host
-			if not r['real_hostname'] or string.lower(r['real_hostname']) == 'unknown' or 
-				string.match(r['real_hostname'], '^%d+%.%d+%.%d+%.%d+$') then
-				
-				if r['real_ip'] and r['real_ip']:is_valid() then
-					-- Try to resolve it again
-					task:get_resolver():resolve_ptr(task:get_session(), task:get_mempool(), 
-						r['real_ip']:to_string(), recv_dns_cb)
-				else
-					task:insert_result(symbol_strict, 1)
-				end
-		    return
-			end
+      -- Unresolved host
+      if not r['real_hostname'] or string.lower(r['real_hostname']) == 'unknown' or 
+        string.match(r['real_hostname'], '^%d+%.%d+%.%d+%.%d+$') then
+        
+        if r['real_ip'] and r['real_ip']:is_valid() then
+          -- Try to resolve it again
+          task:get_resolver():resolve_ptr(task:get_session(), task:get_mempool(), 
+            r['real_ip']:to_string(), recv_dns_cb)
+        else
+          task:insert_result(symbol_strict, 1)
+        end
+        return
+      end
 
-			local i = true
-			local hn = string.lower(r['real_hostname'])
+      local i = true
+      local hn = string.lower(r['real_hostname'])
 
-			for _,h in ipairs(bad_hosts) do
-				if string.find(hn, h) then
-					-- Check for good hostname
-					if good_hosts then
-						for _,gh in ipairs(good_hosts) do
-							if string.find(hn, gh) then
-								i = false
-								break
-							end
-						end
-					end
-					if i then
-						task:insert_result(symbol_strict, 1, h)
-						return
-					end
-				end
-			end
-		end
-	end
+      for _,h in ipairs(bad_hosts) do
+        if string.find(hn, h) then
+          -- Check for good hostname
+          if good_hosts then
+            for _,gh in ipairs(good_hosts) do
+              if string.find(hn, gh) then
+                i = false
+                break
+              end
+            end
+          end
+          if i then
+            task:insert_result(symbol_strict, 1, h)
+            return
+          end
+        end
+      end
+    end
+  end
 end
 
 -- Registration
 if type(rspamd_config.get_api_version) ~= 'nil' then
-	if rspamd_config:get_api_version() >= 1 then
-		rspamd_config:register_module_option('once_received', 'symbol', 'string')
-		rspamd_config:register_module_option('once_received', 'symbol_strict', 'string')
-		rspamd_config:register_module_option('once_received', 'bad_host', 'string')
-		rspamd_config:register_module_option('once_received', 'good_host', 'string')
-	end
+  if rspamd_config:get_api_version() >= 1 then
+    rspamd_config:register_module_option('once_received', 'symbol', 'string')
+    rspamd_config:register_module_option('once_received', 'symbol_strict', 'string')
+    rspamd_config:register_module_option('once_received', 'bad_host', 'string')
+    rspamd_config:register_module_option('once_received', 'good_host', 'string')
+  end
 end
 
 -- Configuration
@@ -98,28 +101,28 @@ if opts then
     if opts['symbol'] then
         local symbol = opts['symbol']
 
-	    for n,v in pairs(opts) do
-			if n == 'symbol_strict' then
-				symbol_strict = v
-				if type(rspamd_config.get_api_version) ~= 'nil' then
-					rspamd_config:register_virtual_symbol(symbol_strict, 1.0)
-				end
-			elseif n == 'bad_host' then
-				if type(v) == 'string' then
-			    	bad_hosts[1] = v
-				else
-					bad_hosts = v
-				end
-			elseif n == 'good_host' then
-				if type(v) == 'string' then
-			    	good_hosts[1] = v
-				else
-					good_hosts = v
-				end
-		    end
-	    end
+      for n,v in pairs(opts) do
+      if n == 'symbol_strict' then
+        symbol_strict = v
+        if type(rspamd_config.get_api_version) ~= 'nil' then
+          rspamd_config:register_virtual_symbol(symbol_strict, 1.0)
+        end
+      elseif n == 'bad_host' then
+        if type(v) == 'string' then
+            bad_hosts[1] = v
+        else
+          bad_hosts = v
+        end
+      elseif n == 'good_host' then
+        if type(v) == 'string' then
+            good_hosts[1] = v
+        else
+          good_hosts = v
+        end
+        end
+      end
 
-		-- Register symbol's callback
-		rspamd_config:register_symbol(symbol, 1.0, check_quantity_received)
-	end
+    -- Register symbol's callback
+    rspamd_config:register_symbol(symbol, 1.0, check_quantity_received)
+  end
 end