From 6e5dfd3a56c9c0bf06070f4e9197648563c25f37 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Wed, 3 Apr 2024 17:58:38 +0200 Subject: [PATCH] Count with encryption tag for channel messages --- core/src/trezor/wire/thp/channel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/core/src/trezor/wire/thp/channel.py b/core/src/trezor/wire/thp/channel.py index b0ab21477..2898ca163 100644 --- a/core/src/trezor/wire/thp/channel.py +++ b/core/src/trezor/wire/thp/channel.py @@ -287,7 +287,9 @@ class Channel(Context): pass def _handle_channel_message(self, message_length: int, message_type: int) -> None: - buf = self.buffer[INIT_DATA_OFFSET + 3 : message_length - CHECKSUM_LENGTH] + buf = self.buffer[ + INIT_DATA_OFFSET + 3 : message_length - CHECKSUM_LENGTH - TAG_LENGTH + ] expected_type = protobuf.type_for_wire(message_type) message = message_handler.wrap_protobuf_load(buf, expected_type)