trezor-firmware/core/src/apps/misc/get_firmware_hash.py

from typing import TYPE_CHECKING

from trezor import ui, wire, workflow
from trezor.crypto.hashlib import sha256
from trezor.messages import FirmwareHash, GetFirmwareHash
from trezor.ui.layouts import draw_simple_text
from trezor.utils import DISABLE_ANIMATION, firmware_hash

if TYPE_CHECKING:
    from trezor.wire import Context


async def get_firmware_hash(ctx: Context, msg: GetFirmwareHash) -> FirmwareHash:
    workflow.close_others()
    draw_simple_text("Please wait")

    try:
        hash = firmware_hash(msg.challenge, _render_progress)
    except ValueError as e:
        raise wire.DataError(str(e))
    hash = sha256(msg.challenge or b"").digest()
    return FirmwareHash(hash=hash)


def _render_progress(progress: int, total: int) -> None:
    if not DISABLE_ANIMATION:
        p = 1000 * progress // total
        ui.display.loader(p, False, 18, ui.WHITE, ui.BG)
        ui.refresh()