trezor-firmware/core/src/apps/homescreen/lockscreen.py

import storage.cache
from trezor import loop, res, ui, wire

from . import HomescreenBase


async def lockscreen() -> None:
    from apps.common.request_pin import can_lock_device
    from apps.base import unlock_device

    # Only show the lockscreen UI if the device can in fact be locked.
    if can_lock_device():
        await Lockscreen()
    # Otherwise proceed directly to unlock() call. If the device is already unlocked,
    # it should be a no-op storage-wise, but it resets the internal configuration
    # to an unlocked state.
    try:
        await unlock_device()
    except wire.PinCancelled:
        pass


class Lockscreen(HomescreenBase):
    BACKLIGHT_LEVEL = ui.BACKLIGHT_LOW
    RENDER_SLEEP = loop.SLEEP_FOREVER
    RENDER_INDICATOR = storage.cache.LOCKSCREEN_ON

    def __init__(self, bootscreen: bool = False) -> None:
        if bootscreen:
            self.BACKLIGHT_LEVEL = ui.BACKLIGHT_NORMAL
            self.lock_label = "Not connected"
            self.tap_label = "Tap to connect"
        else:
            self.lock_label = "Locked"
            self.tap_label = "Tap to unlock"

        super().__init__()

    def do_render(self) -> None:
        # homescreen with label text on top
        ui.display.text_center(
            ui.WIDTH // 2, 35, self.label, ui.BOLD, ui.TITLE_GREY, ui.BG
        )
        ui.display.avatar(48, 48, self.get_image(), ui.WHITE, ui.BLACK)

        # lock bar
        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, self.lock_label, ui.BOLD, ui.TITLE_GREY, ui.BG
        )

        # "tap to unlock"
        ui.display.text_center(
            ui.WIDTH // 2 + 10, 220, self.tap_label, ui.BOLD, ui.TITLE_GREY, ui.BG
        )
        ui.display.icon(45, 202, res.load(ui.ICON_CLICK), ui.TITLE_GREY, ui.BG)

    def on_touch_end(self, _x: int, _y: int) -> None:
        raise ui.Result(None)
fix(core): prevent flickering when homescreen does not need to redraw 2021-04-06 12:31:03 +00:00			`import storage.cache`
feat(core): hold homescreen to lock 2021-01-18 20:18:56 +00:00			`from trezor import loop, res, ui, wire`
core: refactor homescreen app, include lockscreen in it 2020-04-14 10:21:32 +00:00
			`from . import HomescreenBase`


core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`async def lockscreen() -> None:`
			`from apps.common.request_pin import can_lock_device`
			`from apps.base import unlock_device`

core: only unlock storage if it is locked (solves determinism issue in tests) 2020-05-28 09:57:04 +00:00			`# Only show the lockscreen UI if the device can in fact be locked.`
core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`if can_lock_device():`
			`await Lockscreen()`
core: only unlock storage if it is locked (solves determinism issue in tests) 2020-05-28 09:57:04 +00:00			`# Otherwise proceed directly to unlock() call. If the device is already unlocked,`
			`# it should be a no-op storage-wise, but it resets the internal configuration`
			`# to an unlocked state.`
core/lockscreen: ignore exception when user taps "unlock" and then cancels 2020-05-21 10:53:11 +00:00			`try:`
			`await unlock_device()`
			`except wire.PinCancelled:`
			`pass`
core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00

core: refactor homescreen app, include lockscreen in it 2020-04-14 10:21:32 +00:00			`class Lockscreen(HomescreenBase):`
core: simplify homescreen and lockscreen implementations 2020-06-01 10:07:57 +00:00			`BACKLIGHT_LEVEL = ui.BACKLIGHT_LOW`
feat(core): hold homescreen to lock 2021-01-18 20:18:56 +00:00			`RENDER_SLEEP = loop.SLEEP_FOREVER`
fix(core): prevent flickering when homescreen does not need to redraw 2021-04-06 12:31:03 +00:00			`RENDER_INDICATOR = storage.cache.LOCKSCREEN_ON`
core: simplify homescreen and lockscreen implementations 2020-06-01 10:07:57 +00:00
core: fix rendering issues in homescreens 2020-06-02 13:23:11 +00:00			`def __init__(self, bootscreen: bool = False) -> None:`
			`if bootscreen:`
			`self.BACKLIGHT_LEVEL = ui.BACKLIGHT_NORMAL`
			`self.lock_label = "Not connected"`
			`self.tap_label = "Tap to connect"`
			`else:`
			`self.lock_label = "Locked"`
			`self.tap_label = "Tap to unlock"`

core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`super().__init__()`

fix(core): prevent flickering when homescreen does not need to redraw 2021-04-06 12:31:03 +00:00			`def do_render(self) -> None:`
core: fix rendering issues in homescreens 2020-06-02 13:23:11 +00:00			`# homescreen with label text on top`
			`ui.display.text_center(`
			`ui.WIDTH // 2, 35, self.label, ui.BOLD, ui.TITLE_GREY, ui.BG`
			`)`
refactor(core/homescreen): load homescreen image on-demand 2021-04-16 10:16:10 +00:00			`ui.display.avatar(48, 48, self.get_image(), ui.WHITE, ui.BLACK)`
core: fix rendering issues in homescreens 2020-06-02 13:23:11 +00:00
			`# lock bar`
core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`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, self.lock_label, ui.BOLD, ui.TITLE_GREY, ui.BG`
			`)`

core: fix rendering issues in homescreens 2020-06-02 13:23:11 +00:00			`# "tap to unlock"`
core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`ui.display.text_center(`
core/boot: modify initial lockscreen label 2020-04-24 10:02:01 +00:00			`ui.WIDTH // 2 + 10, 220, self.tap_label, ui.BOLD, ui.TITLE_GREY, ui.BG`
core: consider lockscreen to be a separate homescreen this involves some changes to the workflow defaults: * workflow.start_default() takes no arguments * workflow.set_default() (originally replace_default) configures the default that will be started by next call to start_default(). The intended usecase is to set_default() first and then start it separately. * apps.base.set_homescreen() factors out the logic originally in main.py, that decides which homescreen should be launched. This uses set_default() call. start_default() is then used explicitly in main.py 2020-04-22 08:58:24 +00:00			`)`
			`ui.display.icon(45, 202, res.load(ui.ICON_CLICK), ui.TITLE_GREY, ui.BG)`

feat(core): hold homescreen to lock 2021-01-18 20:18:56 +00:00			`def on_touch_end(self, _x: int, _y: int) -> None:`
core: refactor homescreen app, include lockscreen in it 2020-04-14 10:21:32 +00:00			`raise ui.Result(None)`