using shlex on the client

MP1/server.py

@@ -44,25 +44,24 @@ while True:
     conn, addr = server_socket.accept()
     print(f"Connected by {addr}")
 
-    command = receive_message(conn)
-    if command is None:
+    try:
+        command = receive_message(conn)
+        if command is None:
+            conn.close()
+            continue
+
+        command_args = shlex.split(command)
+
+        command_args.insert(1, "-n")
+
+        result = subprocess.run(command_args, cwd=SCRIPT_DIR, capture_output=True, text=True)
+    
+        send_message(conn, result.stdout + result.stderr)
+    
+    except Exception as e:
+        error_message = f"Error executing command: {e}"
+        send_message(conn, error_message)
+    finally:
         conn.close()
-        continue
-
-    command_args = shlex.split(command)
-
-    if not command_args or command_args[0] != "grep":
-        response = "Invalid command. Please use the 'grep' command."
-        send_message(conn, response)
-        conn.close()
-        continue
-
-    command_args.insert(1, "-n")
-
-    result = subprocess.run(command_args, cwd=SCRIPT_DIR, capture_output=True, text=True)
-   
-    send_message(conn, result.stdout + result.stderr)
-
-    conn.close()
 
 server_socket.close()