mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-02-21 20:12:02 +00:00
5c93ecd53a
This is to avoid including app-specific functionality in storage and avoid circular imports. The following policy is now in effect: modules from `storage` namespace must not import from `apps` namespace. In most files, the change only involves changing import paths. A minor refactor was needed in case of webauthn: basic get/set/delete functionality was left in storage.webauthn, and more advanced logic on top of it was moved to apps.webauthn.resident_credentials. A significant refactor was needed for sd_salt, where application (and UI) logic was tightly coupled with the IO code. This is now separated, and storage.sd_salt deals exclusively with the IO side, while the app/UI logic is implemented on top of it in apps.common.sd_salt and apps.management.sd_protect.
136 lines
3.8 KiB
Python
136 lines
3.8 KiB
Python
from micropython import const
|
|
|
|
from storage import common, recovery_shares
|
|
from trezor.crypto import slip39
|
|
|
|
# Namespace:
|
|
_NAMESPACE = common.APP_RECOVERY
|
|
|
|
# fmt: off
|
|
# Keys:
|
|
_IN_PROGRESS = const(0x00) # bool
|
|
_DRY_RUN = const(0x01) # bool
|
|
_SLIP39_IDENTIFIER = const(0x03) # bytes
|
|
_SLIP39_THRESHOLD = const(0x04) # int
|
|
_REMAINING = const(0x05) # int
|
|
_SLIP39_ITERATION_EXPONENT = const(0x06) # int
|
|
_SLIP39_GROUP_COUNT = const(0x07) # int
|
|
|
|
# Deprecated Keys:
|
|
# _WORD_COUNT = const(0x02) # int
|
|
# fmt: on
|
|
|
|
# Default values:
|
|
_DEFAULT_SLIP39_GROUP_COUNT = const(1)
|
|
|
|
|
|
if False:
|
|
from typing import List, Optional
|
|
|
|
|
|
def _require_progress() -> None:
|
|
if not is_in_progress():
|
|
raise RuntimeError
|
|
|
|
|
|
def set_in_progress(val: bool) -> None:
|
|
common.set_bool(_NAMESPACE, _IN_PROGRESS, val)
|
|
|
|
|
|
def is_in_progress() -> bool:
|
|
return common.get_bool(_NAMESPACE, _IN_PROGRESS)
|
|
|
|
|
|
def set_dry_run(val: bool) -> None:
|
|
_require_progress()
|
|
common.set_bool(_NAMESPACE, _DRY_RUN, val)
|
|
|
|
|
|
def is_dry_run() -> bool:
|
|
_require_progress()
|
|
return common.get_bool(_NAMESPACE, _DRY_RUN)
|
|
|
|
|
|
def set_slip39_identifier(identifier: int) -> None:
|
|
_require_progress()
|
|
common.set_uint16(_NAMESPACE, _SLIP39_IDENTIFIER, identifier)
|
|
|
|
|
|
def get_slip39_identifier() -> Optional[int]:
|
|
_require_progress()
|
|
return common.get_uint16(_NAMESPACE, _SLIP39_IDENTIFIER)
|
|
|
|
|
|
def set_slip39_iteration_exponent(exponent: int) -> None:
|
|
_require_progress()
|
|
common.set_uint8(_NAMESPACE, _SLIP39_ITERATION_EXPONENT, exponent)
|
|
|
|
|
|
def get_slip39_iteration_exponent() -> Optional[int]:
|
|
_require_progress()
|
|
return common.get_uint8(_NAMESPACE, _SLIP39_ITERATION_EXPONENT)
|
|
|
|
|
|
def set_slip39_group_count(group_count: int) -> None:
|
|
_require_progress()
|
|
common.set_uint8(_NAMESPACE, _SLIP39_GROUP_COUNT, group_count)
|
|
|
|
|
|
def get_slip39_group_count() -> int:
|
|
_require_progress()
|
|
return (
|
|
common.get_uint8(_NAMESPACE, _SLIP39_GROUP_COUNT) or _DEFAULT_SLIP39_GROUP_COUNT
|
|
)
|
|
|
|
|
|
def set_slip39_remaining_shares(shares_remaining: int, group_index: int) -> None:
|
|
"""
|
|
We store the remaining shares as a bytearray of length group_count.
|
|
Each byte represents share remaining for group of that group_index.
|
|
0x10 (16) was chosen as the default value because it's the max
|
|
share count for a group.
|
|
"""
|
|
_require_progress()
|
|
remaining = common.get(_NAMESPACE, _REMAINING)
|
|
group_count = get_slip39_group_count()
|
|
if not group_count:
|
|
raise RuntimeError
|
|
if remaining is None:
|
|
remaining = bytearray([slip39.MAX_SHARE_COUNT] * group_count)
|
|
remaining = bytearray(remaining)
|
|
remaining[group_index] = shares_remaining
|
|
common.set(_NAMESPACE, _REMAINING, remaining)
|
|
|
|
|
|
def get_slip39_remaining_shares(group_index: int) -> Optional[int]:
|
|
_require_progress()
|
|
remaining = common.get(_NAMESPACE, _REMAINING)
|
|
if remaining is None or remaining[group_index] == slip39.MAX_SHARE_COUNT:
|
|
return None
|
|
else:
|
|
return remaining[group_index]
|
|
|
|
|
|
def fetch_slip39_remaining_shares() -> Optional[List[int]]:
|
|
_require_progress()
|
|
remaining = common.get(_NAMESPACE, _REMAINING)
|
|
if not remaining:
|
|
return None
|
|
|
|
group_count = get_slip39_group_count()
|
|
if not group_count:
|
|
raise RuntimeError
|
|
return list(remaining[:group_count])
|
|
|
|
|
|
def end_progress() -> None:
|
|
_require_progress()
|
|
common.delete(_NAMESPACE, _IN_PROGRESS)
|
|
common.delete(_NAMESPACE, _DRY_RUN)
|
|
common.delete(_NAMESPACE, _SLIP39_IDENTIFIER)
|
|
common.delete(_NAMESPACE, _SLIP39_THRESHOLD)
|
|
common.delete(_NAMESPACE, _REMAINING)
|
|
common.delete(_NAMESPACE, _SLIP39_ITERATION_EXPONENT)
|
|
common.delete(_NAMESPACE, _SLIP39_GROUP_COUNT)
|
|
recovery_shares.delete()
|