84 lines
1.8 KiB
Python
84 lines
1.8 KiB
Python
|
#!/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()
|