diff --git a/client.py b/client.py index ebbc24f243f3f49bd2be1422361a2ec41a0e3f8d..ef29a9548bbacb91cec8c7e7b221b4b3e39a645c 100644 --- a/client.py +++ b/client.py @@ -6,6 +6,11 @@ HOST_NAME = socket.gethostname() s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect((HOST_NAME, SERVER_PORT)) +full_msg = "" while True: msg = s.recv(8) - print(msg.decode("utf-8")) \ No newline at end of file + if len(msg) <= 0: + break + full_msg += msg.decode("utf-8") + +print(full_msg) \ No newline at end of file diff --git a/server.py b/server.py index 95217c7cfc1f3b8b6238832ebe9275c5aa83d112..95e526c30bedc5300a7c648ff7585906b9cbadec 100644 --- a/server.py +++ b/server.py @@ -1,15 +1,44 @@ import socket +# def CONSTANT SERVER_PORT = 10000 HOST_NAME = socket.gethostname() HOST_ADDRESS = socket.gethostbyname(HOST_NAME) +MSG = "Welcome to the server" + +# functions +def send_file(clientsocket, address) : + # To Change + clientsocket.send(bytes(MSG, "utf-8")) + print(f"[!] Message sent to {address}.") + clientsocket.close() + +# main s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) +s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind((HOST_NAME, SERVER_PORT)) s.listen(5) -print(f"Server listening on {HOST_ADDRESS}:{SERVER_PORT}") +print(f"Server listening on {HOST_ADDRESS}:{SERVER_PORT}\n") +# Listening client request +client_list = [] while True: clientsocket, address = s.accept() - print(f"Connection from {address} has been established!") - clientsocket.send(bytes("Welcome to the server!", "utf-8")) + print(f"[!] Client {address} found") + client_list.append((clientsocket, address)) + print(f"[?] Listen more? (y/n) ", end="") + choice = input() + if choice == "n": + break + +# Print client list +print() +print(f"{len(client_list)} client(s) found:") +for client, address in client_list: + print(f"--- {address}") +print() + +# send file to all client in the list +for client, address in client_list: + send_file(client, address) \ No newline at end of file