afws_client: report error on JSON data length mismatch

2024-12-27 04:08:27 +08:00 · 2024-08-02 14:26:53 +08:00 · 2024-08-02 14:26:53 +08:00 · 2241a32c9a
commit 2241a32c9a
parent e627aaeda0
1 changed files with 2 additions and 0 deletions
--- a/artiq/frontend/afws_client.py
+++ b/artiq/frontend/afws_client.py
@ -165,6 +165,8 @@ class Client:
            return reply[0], None
        length = int(reply[1])
        json_bytes = await self.reader.read(length)
+        if length != len(json_bytes):
+            raise ValueError(f"Received data length ({len(json_bytes)}) doesn't match expected length ({length})")
        return "OK", json_bytes