arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-04-21 09:39:02 +00:00
Code Issues Releases Wiki Activity

chore(core): add unexpected fallback write for small messages

Browse Source
This commit is contained in:
M1nd3r 2025-04-14 16:25:54 +02:00
parent d96ae3a6c4
commit 00836abd5f
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
core/src/trezor/wire/thp/channel.py
View File

@ -479,18 +479,29 @@ class Channel:
from trezor.enums import FailureType
from trezor.messages import Failure
if length <= len(self.buffer):
# Fallback write - Write buffer is locked, using backup buffer instead
noise_payload_len = memory_manager.encode_into_buffer(
self.buffer, msg, session_id
)
task = self._write_and_encrypt(noise_payload_len, fallback=True)
if task is not None:
await task
return
# Message cannot be written - not even in fallback mode, killing channel
if __debug__ and utils.ALLOW_DEBUG_MESSAGES:
self._log("Failed to get write buffer, killing channel.")
noise_payload_len = memory_manager.encode_into_buffer(
self.buffer,
Failure(
code=FailureType.FirmwareError,
message="Failed to obtain write buffer.",
),
session_id,
)
self.set_channel_state(ChannelState.INVALIDATED)
noise_payload_len = memory_manager.encode_into_buffer(
self.buffer,
Failure(
code=FailureType.FirmwareError,
message="Failed to obtain write buffer.",
),
session_id,
)
self.set_channel_state(ChannelState.INVALIDATED)
task = self._write_and_encrypt(
noise_payload_len=noise_payload_len, force=force, fallback=fallback
)