added variable length receiving and sending

MP1/client.py

@@ -15,6 +15,26 @@ ALL_IPS = [
     "172.22.156.6"
 ]
 
+def send_message(sock, message):
+    e_data = message.encode('utf-8')
+    #header for the message length
+    header = str(len(e_data)).zfill(10).encode('utf-8')
+    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'))
+    received_data = b""
+    while len(received_data) < message_length:
+        chunk = sock.recv(65536)
+        if not chunk:
+            break
+        received_data += chunk
+    return received_data.decode('utf-8')
+
 def grep_all_ips(ips, cmd):
     match_count = []
 
@@ -26,8 +46,8 @@ def grep_all_ips(ips, cmd):
 
             full_cmd = cmd + f" ./machine.{i + 1}.log"
 
-            client_socket.sendall(full_cmd.encode())
-            data = client_socket.recv(65536).decode()
+            send_message(client_socket, full_cmd)
+            data = receive_message(client_socket)
 
             count = len(data.splitlines())
             match_count.append(count)