From a9b4edd040156d875c1f26d4b8ebf8636a78b26a Mon Sep 17 00:00:00 2001 From: Aryan Singh Date: Sat, 12 Sep 2026 12:31:42 -0500 Subject: [PATCH] Update server.py --- MP1/server.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/MP1/server.py b/MP1/server.py index 3598aaa..1c5fa6f 100644 --- a/MP1/server.py +++ b/MP1/server.py @@ -20,17 +20,21 @@ def send_message(sock, message): sock.sendall(header) sock.sendall(e_data) -def receive_message(sock): - header = sock.recv(10) - if not header: - return None - message_length = int(header.decode('utf-8')) +def receive_exact(sock, num_bytes): received_data = b"" - while len(received_data) < message_length: - chunk = sock.recv(65536) + while len(received_data) < num_bytes: + chunk = sock.recv(num_bytes - len(received_data)) if not chunk: break received_data += chunk + return received_data + +def receive_message(sock): + header = receive_exact(sock, 10) + if not header: + return None + message_length = int(header.decode('utf-8')) + received_data = receive_exact(sock, message_length) return received_data.decode('utf-8') while True: