arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-05-06 08:59:15 +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

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

@ -479,6 +479,17 @@ 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.")