fix

Author: molanjad2011

client.py

@@ -1,29 +1,50 @@
 import socket
 import threading
 
-HOST = '127.0.0.1'  # آی‌پی سرور
-PORT = '65432'
+HOST = "127.0.0.1"
+PORT = int(input("enter port: "))
 
 def receive_messages(s):
-    while True:
-        data = s.recv(1024)
-        if not data:
-            print("Server disconnected")
-            break
-        print("Server:", data.decode())
+    try:
+        while True:
+            data = s.recv(1024)
+            if not data:
+                print("Server disconnected")
+                break
+            print("Server:", data.decode())
+    except ConnectionResetError:
+        print("Server closed the connection.")
+    finally:
+        s.close()
 
 def send_messages(s):
-    while True:
-        msg = input()
-        s.sendall(msg.encode())
+    try:
+        while True:
+            msg = input("enter msg: ")
+            if msg.lower() == "exit":
+                print("Closing connection...")
+                s.close()
+                break
+            s.sendall(msg.encode())
+    except BrokenPipeError:
+        print("Cannot send message, server disconnected.")
+    finally:
+        s.close()
 
-with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-    s.connect((HOST, PORT))
-    print("Connected to server. Type messages:")
+def main():
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        s.connect((HOST, PORT))
+        print("Connected to server. Type messages (type 'exit' to quit):")
 
-    threading.Thread(target=receive_messages, args=(s,), daemon=True).start()
-    threading.Thread(target=send_messages, args=(s,), daemon=True).start()
+        recv_thread = threading.Thread(target=receive_messages, args=(s,), daemon=True)
+        send_thread = threading.Thread(target=send_messages, args=(s,), daemon=True)
 
-    while True:
-        pass
+        recv_thread.start()
+        send_thread.start()
 
+        recv_thread.join()
+        send_thread.join()
+        print("Client shutting down.")
+
+if __name__ == "__main__":
+    main()

server.py

@@ -1,34 +1,55 @@
 import socket
 import threading
+import random
 
-HOST = '127.0.0.1'
-PORT = '65432'
+HOST = "127.0.0.1"
+PORT = random.randint(1000,9999)
+print("this is a port pls enter to client: ",PORT)
 
 def receive_messages(conn):
-    while True:
-        data = conn.recv(1024)
-        if not data:
-            print("Client disconnected")
-            break
-        print("Client:", data.decode())
+    try:
+        while True:
+            data = conn.recv(1024)
+            if not data:
+                print("Client disconnected")
+                break
+            print("Client:", data.decode())
+    except ConnectionResetError:
+        print("Connection was closed abruptly.")
+    finally:
+        conn.close()
 
 def send_messages(conn):
-    while True:
-        msg = input()
-        conn.sendall(msg.encode())
+    try:
+        while True:
+            msg = input("enter msg: ")
+            if msg.lower() == "exit":
+                print("Closing connection...")
+                conn.close()
+                break
+            conn.sendall(msg.encode())
+    except BrokenPipeError:
+        print("Cannot send message, client disconnected.")
+    finally:
+        conn.close()
 
-with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-    s.bind((HOST, PORT))
-    s.listen()
-    print(f"Server listening on {HOST}:{PORT}")
-    conn, addr = s.accept()
-    print('Connected by', addr)
+def main():
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:
+        server_socket.bind((HOST, PORT))
+        server_socket.listen()
+        print(f"Server listening on {HOST}:{PORT}")
+        conn, addr = server_socket.accept()
+        print('Connected by', addr)
 
-    # دو thread جداگانه برای ارسال و دریافت پیام‌ها
-    threading.Thread(target=receive_messages, args=(conn,), daemon=True).start()
-    threading.Thread(target=send_messages, args=(conn,), daemon=True).start()
+        recv_thread = threading.Thread(target=receive_messages, args=(conn,), daemon=True)
+        send_thread = threading.Thread(target=send_messages, args=(conn,), daemon=True)
 
-    # فقط نگه داشتن سرور باز
-    while True:
-        pass
+        recv_thread.start()
+        send_thread.start()
 
+        recv_thread.join()
+        send_thread.join()
+        print("Server shutting down.")
+
+if __name__ == "__main__":
+    main()