commit 4234ed5: [Minor] Fix data flow

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2021-11-22 11:01:49 +0000
URL: https://github.com/rspamd/rspamd/commit/4234ed52ba22ab5e308170959e9557f8f35df273 (HEAD -> master)

[Minor] Fix data flow

---
 src/plugins/lua/mx_check.lua | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/plugins/lua/mx_check.lua b/src/plugins/lua/mx_check.lua
index a7ea7aee7..2c9fcb3e7 100644
--- a/src/plugins/lua/mx_check.lua
+++ b/src/plugins/lua/mx_check.lua
@@ -140,11 +140,13 @@ local function mx_check(task)
           mxes[name].checked = true
           mxes[name].working = true
           valid = true
+          if settings.wait_for_greeting then
+            conn:add_write(function(_)
+              conn:close()
+            end, string.format('QUIT%s', CRLF))
+          end
         end
         check_results(mxes)
-        conn:add_write(function(_)
-          conn:close()
-        end, string.format('QUIT%s', CRLF))
       end
       local function on_connect_cb(conn)
         if err then