Do not close connection until write is complete on failed auth PONG

fixes fluent#1834
arcivanov · Jan 27, 2018 · 1d8b3cf · 1d8b3cf
1 parent b3d51c2
commit 1d8b3cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/fluent/plugin/in_forward.rb b/lib/fluent/plugin/in_forward.rb
@@ -199,8 +199,8 @@ def handle_connection(conn)
         when :pingpong
           success, reason_or_salt, shared_key = check_ping(msg, conn.remote_addr, user_auth_salt, nonce)
           unless success
+            conn.on(:write_complete) { |c| c.close }
             send_data.call(serializer, generate_pong(false, reason_or_salt, nonce, shared_key))
-            conn.close
             next
           end
           send_data.call(serializer, generate_pong(true, reason_or_salt, nonce, shared_key))