|
| 1 | +import socket |
| 2 | +import threading |
| 3 | + |
| 4 | +# Create a socket |
| 5 | +server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 6 | + |
| 7 | +# Define the host and port |
| 8 | +host = '192.168.0.11' # Listen on all available network interfaces |
| 9 | +port = 12345 |
| 10 | + |
| 11 | +# Bind the socket to the host and port |
| 12 | +server.bind((host, port)) |
| 13 | + |
| 14 | +# Listen for incoming connections |
| 15 | +server.listen() |
| 16 | + |
| 17 | +print(f"Server listening on {host}:{port}") |
| 18 | + |
| 19 | +# Store client connections and nicknames in dictionaries |
| 20 | +clients = {} |
| 21 | +nicknames = {} |
| 22 | + |
| 23 | +# Function to broadcast messages to all connected clients |
| 24 | +def broadcast(message, sender=None): |
| 25 | + for client_socket, nickname in zip(clients, nicknames): |
| 26 | + if client_socket != sender: |
| 27 | + try: |
| 28 | + client_socket.send(message.encode('utf-8')) |
| 29 | + except: |
| 30 | + remove_client(client_socket) |
| 31 | + |
| 32 | +# Function to handle client connections |
| 33 | +def handle_client(client_socket): |
| 34 | + try: |
| 35 | + # Request and store the user's nickname |
| 36 | + client_socket.send("Enter your nickname: ".encode('utf-8')) |
| 37 | + nickname = client_socket.recv(1024).decode('utf-8') |
| 38 | + nicknames[client_socket] = nickname |
| 39 | + print(f"Nickname of {nickname} is set.") |
| 40 | + |
| 41 | + # Welcome message |
| 42 | + welcome_message = f"Welcome, {nickname}!" |
| 43 | + client_socket.send(welcome_message.encode('utf-8')) |
| 44 | + |
| 45 | + while True: |
| 46 | + message = client_socket.recv(1024) |
| 47 | + if not message: |
| 48 | + remove_client(client_socket) |
| 49 | + break |
| 50 | + elif message.decode('utf-8').lower() == 'exit': |
| 51 | + remove_client(client_socket) |
| 52 | + elif message.decode('utf-8').startswith('@'): |
| 53 | + # Private message |
| 54 | + recipient = message.decode('utf-8').split(' ')[0][1:] |
| 55 | + if recipient in nicknames.values(): |
| 56 | + recipient_socket = list(nicknames.keys())[list(nicknames.values()).index(recipient)] |
| 57 | + recipient_socket.send(f"(Private) {nickname}: {message.decode('utf-8')[len(recipient) + 2:]}".encode('utf-8')) |
| 58 | + else: |
| 59 | + client_socket.send("Recipient not found.".encode('utf-8')) |
| 60 | + else: |
| 61 | + # Broadcast regular message |
| 62 | + broadcast(f"{nickname}: {message.decode('utf-8')}", sender=client_socket) |
| 63 | + except: |
| 64 | + remove_client(client_socket) |
| 65 | + |
| 66 | +# Function to remove a client from the server |
| 67 | +def remove_client(client_socket): |
| 68 | + if client_socket in clients: |
| 69 | + print(f"Connection closed with {nicknames[client_socket]}") |
| 70 | + del nicknames[client_socket] |
| 71 | + clients.pop(client_socket) |
| 72 | + client_socket.close() |
| 73 | + |
| 74 | +# Main server loop |
| 75 | +while True: |
| 76 | + client_socket, client_address = server.accept() |
| 77 | + print(f"Accepted connection from {client_address}") |
| 78 | + clients[client_socket] = client_address |
| 79 | + |
| 80 | + # Start a thread to handle the client |
| 81 | + client_thread = threading.Thread(target=handle_client, args=(client_socket,)) |
| 82 | + client_thread.start() |
| 83 | + |
0 commit comments