From 64a36cb4f8ef111b0843ae20a924a4a2575c6052 Mon Sep 17 00:00:00 2001
From: nathan <nathan@dorkbox.com>
Date: Tue, 22 Sep 2020 17:05:49 +0200
Subject: [PATCH] code polish to more straightforward logic

---
 src/dorkbox/network/handshake/ServerHandshake.kt | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/dorkbox/network/handshake/ServerHandshake.kt b/src/dorkbox/network/handshake/ServerHandshake.kt
index 155fcd64..a5b1a358 100644
--- a/src/dorkbox/network/handshake/ServerHandshake.kt
+++ b/src/dorkbox/network/handshake/ServerHandshake.kt
@@ -137,11 +137,6 @@ internal class ServerHandshake<CONNECTION : Connection>(private val logger: KLog
                                           clientAddressString: String,
                                           clientAddress: InetAddress): Boolean {
 
-        if (clientAddress.isLoopbackAddress) {
-            // we do not want to limit loopback addresses
-            return true
-        }
-
         try {
             // VALIDATE:: Check to see if there are already too many clients connected.
             if (server.connections.connectionCount() >= config.maxClientCount) {
@@ -357,7 +352,9 @@ internal class ServerHandshake<CONNECTION : Connection>(private val logger: KLog
             return
         }
 
-        if (!validateUdpConnectionInfo(server, handshakePublication, config, clientAddressString, clientAddress)) {
+        if (!clientAddress.isLoopbackAddress &&
+            !validateUdpConnectionInfo(server, handshakePublication, config, clientAddressString, clientAddress)) {
+            // we do not want to limit loopback addresses!
             return
         }