afws_client: report error on JSON data length mismatch

This commit is contained in:
Florian Agbuya 2024-08-02 14:26:53 +08:00 committed by Sébastien Bourdeauducq
parent e627aaeda0
commit 2241a32c9a
1 changed files with 2 additions and 0 deletions

View File

@ -165,6 +165,8 @@ class Client:
return reply[0], None return reply[0], None
length = int(reply[1]) length = int(reply[1])
json_bytes = await self.reader.read(length) 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 return "OK", json_bytes