mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-12 01:18:14 +00:00
119 lines
3.1 KiB
Python
119 lines
3.1 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import storage.device as storage_device
|
|
from trezor import utils
|
|
|
|
if TYPE_CHECKING:
|
|
from trezor.enums import BackupType
|
|
from trezor.ui.layouts.common import ProgressLayout
|
|
|
|
|
|
def get() -> tuple[bytes | None, BackupType]:
|
|
return get_secret(), get_type()
|
|
|
|
|
|
def get_secret() -> bytes | None:
|
|
return storage_device.get_mnemonic_secret()
|
|
|
|
|
|
def get_type() -> BackupType:
|
|
return storage_device.get_backup_type()
|
|
|
|
|
|
def is_bip39() -> bool:
|
|
"""
|
|
If False then SLIP-39 (either Basic or Advanced).
|
|
Other invalid values are checked directly in storage.
|
|
"""
|
|
from trezor.enums import BackupType
|
|
|
|
return get_type() == BackupType.Bip39
|
|
|
|
|
|
def get_seed(passphrase: str = "", progress_bar: bool = True) -> bytes:
|
|
mnemonic_secret = get_secret()
|
|
if mnemonic_secret is None:
|
|
raise ValueError # Mnemonic not set
|
|
|
|
render_func = None
|
|
if progress_bar and not utils.DISABLE_ANIMATION:
|
|
_start_progress()
|
|
render_func = _render_progress
|
|
|
|
if is_bip39():
|
|
from trezor.crypto import bip39
|
|
|
|
seed = bip39.seed(mnemonic_secret.decode(), passphrase, render_func)
|
|
|
|
else: # SLIP-39
|
|
from trezor.crypto import slip39
|
|
|
|
identifier = storage_device.get_slip39_identifier()
|
|
iteration_exponent = storage_device.get_slip39_iteration_exponent()
|
|
if identifier is None or iteration_exponent is None:
|
|
# Identifier or exponent expected but not found
|
|
raise RuntimeError
|
|
seed = slip39.decrypt(
|
|
mnemonic_secret,
|
|
passphrase.encode(),
|
|
iteration_exponent,
|
|
identifier,
|
|
render_func,
|
|
)
|
|
|
|
return seed
|
|
|
|
|
|
if not utils.BITCOIN_ONLY:
|
|
|
|
def derive_cardano_icarus(
|
|
passphrase: str = "",
|
|
trezor_derivation: bool = True,
|
|
progress_bar: bool = True,
|
|
) -> bytes:
|
|
if not is_bip39():
|
|
raise ValueError # should not be called for SLIP-39
|
|
|
|
mnemonic_secret = get_secret()
|
|
if mnemonic_secret is None:
|
|
raise ValueError("Mnemonic not set")
|
|
|
|
render_func = None
|
|
if progress_bar and not utils.DISABLE_ANIMATION:
|
|
_start_progress()
|
|
render_func = _render_progress
|
|
|
|
from trezor.crypto import cardano
|
|
|
|
seed = cardano.derive_icarus(
|
|
mnemonic_secret.decode(), passphrase, trezor_derivation, render_func
|
|
)
|
|
_finish_progress()
|
|
return seed
|
|
|
|
|
|
_progress_obj: ProgressLayout | None = None
|
|
|
|
|
|
def _start_progress() -> None:
|
|
from trezor import workflow
|
|
from trezor.ui.layouts.progress import progress
|
|
|
|
global _progress_obj
|
|
|
|
# Because we are drawing to the screen manually, without a layout, we
|
|
# should make sure that no other layout is running.
|
|
workflow.close_others()
|
|
_progress_obj = progress()
|
|
|
|
|
|
def _render_progress(progress: int, total: int) -> None:
|
|
global _progress_obj
|
|
if _progress_obj is not None:
|
|
_progress_obj.report(1000 * progress // total)
|
|
|
|
|
|
def _finish_progress() -> None:
|
|
global _progress_obj
|
|
_progress_obj = None
|