add check for null input to handle client disconnection (#214)

fungover · Feb 14, 2024 · df5635c · df5635c
2 parents 7d6601d + aa34821
commit df5635c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/java/org/fungover/haze/Main.java b/src/main/java/org/fungover/haze/Main.java
@@ -13,7 +13,6 @@
 import java.util.List;
 
 
-
 public class Main {
     static boolean serverOpen = true;
     static Logger logger = LogManager.getLogger(Main.class);
@@ -50,7 +49,13 @@ public static void main(String[] args) {
 
                             String firstReading = input.readLine();
 
-                            readInputStream(input,inputList,firstReading);
+                            if (firstReading == null) {
+                                logger.info("client has closed the connection");
+                                client.close();
+                                return;
+                            }
+
+                            readInputStream(input, inputList, firstReading);
 
                             clientAuthenticated = Auth.authenticateClient(auth, isPasswordSet, client, inputList, clientAuthenticated);