moninj: fix read of incomplete data (#1729)

pull/1794/head
Sebastien Bourdeauducq 2021-07-22 17:59:21 +08:00
parent b29d17747f
commit 0a29b75f8e
1 changed files with 2 additions and 2 deletions

View File

@ -74,11 +74,11 @@ class CommMonInj:
if not ty:
return
if ty == b"\x00":
payload = await self._reader.read(9)
payload = await self._reader.readexactly(9)
channel, probe, value = struct.unpack(">lbl", payload)
self.monitor_cb(channel, probe, value)
elif ty == b"\x01":
payload = await self._reader.read(6)
payload = await self._reader.readexactly(6)
channel, override, value = struct.unpack(">lbb", payload)
self.injection_status_cb(channel, override, value)
else: