arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-05-06 00:49:20 +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 f2459189fc
commit 35d862f8b4
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.enums import FailureType
from trezor.messages import Failure 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: if __debug__ and utils.ALLOW_DEBUG_MESSAGES:
self._log("Failed to get write buffer, killing channel.") self._log("Failed to get write buffer, killing channel.")
noise_payload_len = memory_manager.encode_into_buffer( noise_payload_len = memory_manager.encode_into_buffer(
self.buffer, self.buffer,
Failure( Failure(
code=FailureType.FirmwareError, code=FailureType.FirmwareError,
message="Failed to obtain write buffer.", message="Failed to obtain write buffer.",
), ),
session_id, session_id,
) )
self.set_channel_state(ChannelState.INVALIDATED) self.set_channel_state(ChannelState.INVALIDATED)
task = self._write_and_encrypt( task = self._write_and_encrypt(
noise_payload_len=noise_payload_len, force=force, fallback=fallback noise_payload_len=noise_payload_len, force=force, fallback=fallback
) )