Add chat project

python-ofensivo/11_chat_cifrado_E2E/server.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+"""
+SERVER DE CHAT CIFRADO E2E
+"""
+
+import socket as s
+import threading as th
+
+
+def client_thread(client_socket, clients, usernames):
+
+    username = client_socket.recv(1024).decode()
+    usernames[client_socket] = username
+
+    print(f"\n[+] El usuario {username} se ha conectado al chat")
+
+    for client in clients:
+        if client is not client_socket:
+            client.sendall(
+                f"\n[+] El usuario {username} se ha conectado al chat\n\n".encode()
+            )
+
+    while True:
+
+        try:
+
+            message = client_socket.recv(1024).decode()
+
+            if not message:
+                break
+
+            if message == "!usuarios":
+                client_socket.sendall(
+                    f"\n[i] Usuarios conectados: {', '.join(usernames.values())}\n\n".encode()
+                )
+                continue
+
+            for client in clients:
+                if client is not client_socket:
+                    client.sendall(message.encode())
+
+        except:
+            break
+
+    client_socket.close()
+    clients.remove(client_socket)
+    del usernames[client_socket]
+
+
+def server_program():
+
+    host = 'localhost'
+    port = 1234
+
+    server_socket = s.socket(s.AF_INET, s.SOCK_STREAM)
+    server_socket.setsockopt(s.SOL_SOCKET, s.SO_REUSEADDR, 1)  # TIME_WAIT
+    server_socket.bind((host, port))
+    server_socket.listen()
+
+    print(f"\n[+] Server está en escucha...")
+
+    clients = []
+    usernames = {}
+
+    while True:
+
+        client_socket, addr = server_socket.accept()
+        clients.append(client_socket)
+
+        print(f"\n[+] Conectado a {addr}")
+
+        thread = th.Thread(
+            target=client_thread,
+            args=(client_socket, clients, usernames)
+        )
+        thread.daemon = True
+        thread.start()
+
+    server_socket.close()
+
+
+if __name__ == '__main__':
+    server_program()