From a016c576b915fc5f05cf0d1c1aeda58c57c7ed51 Mon Sep 17 00:00:00 2001 From: Martin Milata Date: Fri, 14 Mar 2025 21:41:01 +0100 Subject: [PATCH] fixup! chore(core): adapt trezorlib transports to session based [no changelog] --- python/src/trezorlib/transport/thp/protocol_v2.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/src/trezorlib/transport/thp/protocol_v2.py b/python/src/trezorlib/transport/thp/protocol_v2.py index 160054b30e..c9585fa231 100644 --- a/python/src/trezorlib/transport/thp/protocol_v2.py +++ b/python/src/trezorlib/transport/thp/protocol_v2.py @@ -70,7 +70,9 @@ class ProtocolV2Channel(Channel): message_type, message_data = self.mapping.encode(message) self.session_id: int = DEFAULT_SESSION_ID self._encrypt_and_write(DEFAULT_SESSION_ID, message_type, message_data) - _ = self._read_until_valid_crc_check() # TODO check ACK + header, _payload = self._read_until_valid_crc_check() + if not header.is_ack(): + raise exceptions.TrezorException("ACK expected") _, msg_type, msg_data = self.read_and_decrypt() features = self.mapping.decode(msg_type, msg_data) if not isinstance(features, messages.Features):