From 9f7392f6ff87c22365acd224c9b8671bcfcf33fe Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@satoshilabs.com>
Date: Thu, 20 Mar 2025 11:56:12 +0200
Subject: [PATCH] feat(core): add `BleInterface` to session handling

Cherry-picked from b62a5aec6b28b0f9950b37e52673561c1977dbec.

[no changelog]
---
 core/src/all_modules.py          |  2 ++
 core/src/bluetooth.py            | 21 +++++++++++++++++++++
 core/src/main.py                 |  6 ++++++
 core/src/session.py              | 19 +++++++++++++++++--
 core/src/trezor/wire/__init__.py | 13 ++++---------
 5 files changed, 50 insertions(+), 11 deletions(-)
 create mode 100644 core/src/bluetooth.py

diff --git a/core/src/all_modules.py b/core/src/all_modules.py
index 33bdc38243..83a4c6668c 100644
--- a/core/src/all_modules.py
+++ b/core/src/all_modules.py
@@ -33,6 +33,8 @@ from trezor import utils
 
 all_modules
 import all_modules
+bluetooth
+import bluetooth
 boot
 import boot
 main
diff --git a/core/src/bluetooth.py b/core/src/bluetooth.py
new file mode 100644
index 0000000000..b90e21c318
--- /dev/null
+++ b/core/src/bluetooth.py
@@ -0,0 +1,21 @@
+from trezorio import BLE, ble
+
+
+# TODO: reimplement in C
+class BleInterface:
+
+    RX_PACKET_LEN = ble.RX_PACKET_LEN
+    TX_PACKET_LEN = ble.TX_PACKET_LEN
+
+    def iface_num(self) -> int:
+        return BLE
+
+    def write(self, msg: bytes) -> int:
+        return ble.write(msg)
+
+    def read(self, buffer: bytearray, offset: int = 0) -> int:
+        return ble.read(buffer, offset)
+
+
+# interface used for trezor wire protocol
+iface_ble = BleInterface()
diff --git a/core/src/main.py b/core/src/main.py
index f0c9a54b06..f6433b5fcf 100644
--- a/core/src/main.py
+++ b/core/src/main.py
@@ -49,6 +49,12 @@ import storage.device
 
 usb.bus.open(storage.device.get_device_id())
 
+
+if utils.USE_BLE:
+    from trezorio import ble
+    ble.start_comm()
+
+
 # run the endless loop
 while True:
     with unimport_manager:
diff --git a/core/src/session.py b/core/src/session.py
index 1ecbc467be..d1bb0628ab 100644
--- a/core/src/session.py
+++ b/core/src/session.py
@@ -1,9 +1,18 @@
 # isort: skip_file
+from micropython import const
+
 from trezor import log, loop, utils, wire, workflow
 
 import apps.base
 import usb
 
+_PROTOBUF_BUFFER_SIZE = const(8192)
+USB_BUFFER = bytearray(_PROTOBUF_BUFFER_SIZE)
+
+if utils.USE_BLE:
+    BLE_BUFFER = bytearray(_PROTOBUF_BUFFER_SIZE)
+
+
 apps.base.boot()
 
 if not utils.BITCOIN_ONLY and usb.ENABLE_IFACE_WEBAUTHN:
@@ -20,8 +29,14 @@ if __debug__:
 apps.base.set_homescreen()
 workflow.start_default()
 
-# initialize the wire codec
-wire.setup(usb.iface_wire)
+# initialize the wire codec over USB
+wire.setup(usb.iface_wire, USB_BUFFER)
+
+if utils.USE_BLE:
+    import bluetooth
+
+    # initialize the wire codec over BLE
+    wire.setup(bluetooth.iface_ble, BLE_BUFFER)
 
 # start the event loop
 loop.run()
diff --git a/core/src/trezor/wire/__init__.py b/core/src/trezor/wire/__init__.py
index 2662a5610a..ac4e98fd67 100644
--- a/core/src/trezor/wire/__init__.py
+++ b/core/src/trezor/wire/__init__.py
@@ -23,7 +23,6 @@ reads the message's header. When the message type is known the first handler is
 
 """
 
-from micropython import const
 from typing import TYPE_CHECKING
 
 from trezor import log, loop, protobuf, utils
@@ -37,10 +36,6 @@ from .message_handler import failure
 # other packages.
 from .errors import *  # isort:skip # noqa: F401,F403
 
-_PROTOBUF_BUFFER_SIZE = const(8192)
-
-WIRE_BUFFER = bytearray(_PROTOBUF_BUFFER_SIZE)
-
 if TYPE_CHECKING:
     from trezorio import WireInterface
     from typing import Any, Callable, Coroutine, TypeVar
@@ -52,13 +47,13 @@ if TYPE_CHECKING:
     LoadedMessageType = TypeVar("LoadedMessageType", bound=protobuf.MessageType)
 
 
-def setup(iface: WireInterface) -> None:
+def setup(iface: WireInterface, buffer: bytearray) -> None:
     """Initialize the wire stack on the provided WireInterface."""
-    loop.schedule(handle_session(iface))
+    loop.schedule(handle_session(iface, buffer))
 
 
-async def handle_session(iface: WireInterface) -> None:
-    ctx = CodecContext(iface, WIRE_BUFFER)
+async def handle_session(iface: WireInterface, buffer: bytearray) -> None:
+    ctx = CodecContext(iface, buffer)
     next_msg: protocol_common.Message | None = None
 
     # Take a mark of modules that are imported at this point, so we can