From be317edbdc96159f9d232389ae046454ed92b6cc Mon Sep 17 00:00:00 2001
From: Bryan <91551702+blondon1@users.noreply.github.com>
Date: Fri, 23 Feb 2024 11:45:32 -0500
Subject: [PATCH] improved script

retry mechanism for the connect_to_server function with a delay between attempts, broader error handling for sockets-related errors, added a simple way to exit the loop and close the socket connection properly, the sending function now accepts the socket object as an argument, improving its independence and reusability.
---
 Cyber_security  projects/CLIENT.py | 78 +++++++++++++++++++-----------
 1 file changed, 50 insertions(+), 28 deletions(-)

diff --git a/Cyber_security  projects/CLIENT.py b/Cyber_security  projects/CLIENT.py
index def4a71e..1d418287 100644
--- a/Cyber_security  projects/CLIENT.py	
+++ b/Cyber_security  projects/CLIENT.py	
@@ -2,31 +2,53 @@
 
 import socket
 import threading
-
-
-def send_msg():
-	while True:
-
-		msg =input().encode()
-		s.send(msg)
-
-def recv_msg():
-	while True:
-		recevied = s.recv(1024)
-		print(recevied.decode())
-
-
-s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-print("connecting..")
-while True:
-	try:
-		s.connect("127.0.0.1",8888)
-		break
-	except CoonectionRefusedError:
-		continue
-
-print("connected....")		
-
-t1 = threading.Thread(target=send_msg)
-t1.start()
-recv_msg()
+import time
+
+def send_msg(sock):
+    while True:
+        try:
+            msg = input().encode('utf-8')  # Specify encoding
+            sock.send(msg)
+        except KeyboardInterrupt:
+            print("Exiting gracefully")
+            sock.close()
+            break
+
+def recv_msg(sock):
+    while True:
+        try:
+            received = sock.recv(1024)
+            print(received.decode('utf-8'))  # Specify decoding
+        except ConnectionResetError:
+            print("Server has closed the connection.")
+            break
+
+def connect_to_server(host, port, retries=5, delay=2):
+    """Attempt to connect to the server with retries."""
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    print("Connecting..")
+    for attempt in range(retries):
+        try:
+            s.connect((host, port))
+            print("Connected!")
+            return s
+        except socket.error as e:
+            print(f"Connection attempt {attempt+1}/{retries} failed: {e}")
+            time.sleep(delay)
+    print("Could not connect to the server.")
+    return None
+
+host = "127.0.0.1"
+port = 8888
+
+s = connect_to_server(host, port)
+if s:
+    t1 = threading.Thread(target=send_msg, args=(s,))
+    t1.start()
+
+    try:
+        recv_msg(s)
+    except KeyboardInterrupt:
+        print("Exiting...")
+    finally:
+        s.close()