From b64054c9b7cda03824e8c934bee700c3f28455a3 Mon Sep 17 00:00:00 2001
From: swee <meow@swee.codes>
Date: Thu, 20 Feb 2025 11:37:42 -0800
Subject: [PATCH] Update server.py

---
 server.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/server.py b/server.py
index 3a738cd..216bb6b 100644
--- a/server.py
+++ b/server.py
@@ -915,25 +915,24 @@ def session(connection, client, ip, isssl=False):
     except:
         print(traceback.format_exc())
 def tcp_session(sock, ip_to):
-    while True:
+    while opened:
         try:
-            while opened:
-                print("Waiting for connection...")
-                connection, client = sock.accept()
-                threading.Thread(target=session, daemon=True, args=[connection, client, ip_to]).start()
+            print("Waiting for connection...")
+            connection, client = sock.accept()
+            threading.Thread(target=session, daemon=True, args=[connection, client, ip_to]).start()
         except:
             print("Something went wrong...")
             print(traceback.format_exc())
 def ssl_session(sock, ip_to):
-    while True:
+    while opened:
         try:
-            while opened:
-                connection, client = sock.accept()
-                ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
-                ctx.load_cert_chain(ssl_cert, keyfile=ssl_pkey)
-                conn = ctx.wrap_socket(connection, server_side=True)
-                conn.do_handshake()
-                threading.Thread(target=session, daemon=True, args=[conn, client, ip_to, True]).start()
+            print("Waiting for SSL connection...")
+            connection, client = sock.accept()
+            ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
+            ctx.load_cert_chain(ssl_cert, keyfile=ssl_pkey)
+            conn = ctx.wrap_socket(connection, server_side=True)
+            conn.do_handshake()
+            threading.Thread(target=session, daemon=True, args=[conn, client, ip_to, True]).start()
         except:
             print("Something went wrong...")
             print(traceback.format_exc())