mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-23 07:58:09 +00:00
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
import storage
|
|
import storage.device
|
|
import storage.sd_salt
|
|
from trezor import config, io, log, loop, res, ui, utils
|
|
from trezor.pin import pin_to_int, show_pin_timeout
|
|
|
|
from apps.common.request_pin import PinCancelled, request_pin
|
|
from apps.common.sd_salt import SdProtectCancelled, request_sd_salt
|
|
|
|
|
|
async def bootscreen() -> None:
|
|
ui.display.orientation(storage.device.get_rotation())
|
|
|
|
while True:
|
|
try:
|
|
if storage.sd_salt.is_enabled() or config.has_pin():
|
|
await lockscreen()
|
|
|
|
salt = await request_sd_salt()
|
|
|
|
if not config.has_pin():
|
|
config.unlock(pin_to_int(""), salt)
|
|
storage.init_unlocked()
|
|
return
|
|
|
|
label = "Enter your PIN"
|
|
while True:
|
|
pin = await request_pin(label, config.get_pin_rem())
|
|
if config.unlock(pin_to_int(pin), salt):
|
|
storage.init_unlocked()
|
|
return
|
|
else:
|
|
label = "Wrong PIN, enter again"
|
|
except (OSError, PinCancelled, SdProtectCancelled) as e:
|
|
if __debug__:
|
|
log.exception(__name__, e)
|
|
except BaseException as e:
|
|
if __debug__:
|
|
log.exception(__name__, e)
|
|
utils.halt(e.__class__.__name__)
|
|
|
|
|
|
async def lockscreen() -> None:
|
|
label = storage.device.get_label()
|
|
image = storage.device.get_homescreen()
|
|
if not label:
|
|
label = "My Trezor"
|
|
if not image:
|
|
image = res.load("apps/homescreen/res/bg.toif")
|
|
|
|
ui.backlight_fade(ui.BACKLIGHT_DIM)
|
|
|
|
ui.display.clear()
|
|
ui.display.avatar(48, 48, image, ui.TITLE_GREY, ui.BG)
|
|
ui.display.text_center(ui.WIDTH // 2, 35, label, ui.BOLD, ui.TITLE_GREY, ui.BG)
|
|
|
|
ui.display.bar_radius(40, 100, 160, 40, ui.TITLE_GREY, ui.BG, 4)
|
|
ui.display.bar_radius(42, 102, 156, 36, ui.BG, ui.TITLE_GREY, 4)
|
|
ui.display.text_center(ui.WIDTH // 2, 128, "Locked", ui.BOLD, ui.TITLE_GREY, ui.BG)
|
|
|
|
ui.display.text_center(
|
|
ui.WIDTH // 2 + 10, 220, "Tap to unlock", ui.BOLD, ui.TITLE_GREY, ui.BG
|
|
)
|
|
ui.display.icon(45, 202, res.load(ui.ICON_CLICK), ui.TITLE_GREY, ui.BG)
|
|
|
|
ui.backlight_fade(ui.BACKLIGHT_NORMAL)
|
|
|
|
await ui.click()
|
|
|
|
|
|
if utils.EMULATOR:
|
|
# Ensure the emulated SD card is FAT32 formatted.
|
|
sd = io.SDCard()
|
|
sd.power(True)
|
|
fs = io.FatFS()
|
|
try:
|
|
fs.mount()
|
|
except OSError:
|
|
fs.mkfs()
|
|
else:
|
|
fs.unmount()
|
|
sd.power(False)
|
|
|
|
ui.display.backlight(ui.BACKLIGHT_NONE)
|
|
ui.backlight_fade(ui.BACKLIGHT_NORMAL)
|
|
config.init(show_pin_timeout)
|
|
loop.schedule(bootscreen())
|
|
loop.run()
|