core: RebootToBootloader skeleton

mmilata/rtb
Martin Milata 2 years ago
parent d1b86eb6b2
commit a5656b0063

@ -117,6 +117,19 @@ STATIC mp_obj_t mod_trezorutils_halt(size_t n_args, const mp_obj_t *args) {
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_trezorutils_halt_obj, 0, 1,
mod_trezorutils_halt);
/// def reboot_to_bootloader() -> None:
/// """
/// Reboots to bootloader.
/// """
STATIC mp_obj_t mod_trezorutils_reboot_to_bootloader() {
// actual reboot via trezorhal goes here:
ensure(secfalse, "reboot_to_bootloader");
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_0(mod_trezorutils_reboot_to_bootloader_obj,
mod_trezorutils_reboot_to_bootloader);
STATIC mp_obj_str_t mod_trezorutils_revision_obj = {
{&mp_type_bytes}, 0, sizeof(SCM_REVISION) - 1, (const byte *)SCM_REVISION};
@ -136,6 +149,7 @@ STATIC const mp_rom_map_elem_t mp_module_trezorutils_globals_table[] = {
{MP_ROM_QSTR(MP_QSTR_consteq), MP_ROM_PTR(&mod_trezorutils_consteq_obj)},
{MP_ROM_QSTR(MP_QSTR_memcpy), MP_ROM_PTR(&mod_trezorutils_memcpy_obj)},
{MP_ROM_QSTR(MP_QSTR_halt), MP_ROM_PTR(&mod_trezorutils_halt_obj)},
{MP_ROM_QSTR(MP_QSTR_reboot_to_bootloader), MP_ROM_PTR(&mod_trezorutils_reboot_to_bootloader_obj)},
// various built-in constants
{MP_ROM_QSTR(MP_QSTR_SCM_REVISION),
MP_ROM_PTR(&mod_trezorutils_revision_obj)},

@ -40,6 +40,13 @@ def halt(msg: str | None = None) -> None:
"""
Halts execution.
"""
# extmod/modtrezorutils/modtrezorutils.c
def reboot_to_bootloader() -> None:
"""
Reboots to bootloader.
"""
SCM_REVISION: bytes
VERSION_MAJOR: int
VERSION_MINOR: int

@ -366,6 +366,8 @@ apps.management.get_next_u2f_counter
import apps.management.get_next_u2f_counter
apps.management.get_nonce
import apps.management.get_nonce
apps.management.reboot_to_bootloader
import apps.management.reboot_to_bootloader
apps.management.recovery_device
import apps.management.recovery_device
apps.management.recovery_device.homescreen

@ -0,0 +1,13 @@
from typing import TYPE_CHECKING
from trezor import utils, wire
from trezor.messages import RebootToBootloader, Success
if TYPE_CHECKING:
from typing import NoReturn
async def reboot_to_bootloader(ctx: wire.Context, msg: RebootToBootloader) -> NoReturn:
await ctx.write(Success())
utils.reboot_to_bootloader()
raise RuntimeError

@ -49,6 +49,8 @@ def find_message_handler_module(msg_type: int) -> str:
return "apps.management.change_wipe_code"
elif msg_type == MessageType.GetNonce:
return "apps.management.get_nonce"
elif msg_type == MessageType.RebootToBootloader:
return "apps.management.reboot_to_bootloader"
if utils.MODEL == "T" and msg_type == MessageType.SdProtect:
return "apps.management.sd_protect"

@ -11,6 +11,7 @@ from trezorutils import ( # noqa: F401
consteq,
halt,
memcpy,
reboot_to_bootloader,
)
from typing import TYPE_CHECKING

Loading…
Cancel
Save