1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-03 21:18:17 +00:00
trezor-firmware/core/src/storage/common.py

80 lines
2.0 KiB
Python

from micropython import const
from trezor import config
# Namespaces:
# fmt: off
APP_DEVICE = const(0x01)
APP_RECOVERY = const(0x02)
APP_RECOVERY_SHARES = const(0x03)
APP_WEBAUTHN = const(0x04)
# fmt: on
_FALSE_BYTE = b"\x00"
_TRUE_BYTE = b"\x01"
STORAGE_VERSION_01 = b"\x01"
STORAGE_VERSION_CURRENT = b"\x02"
def set(app: int, key: int, data: bytes, public: bool = False) -> None:
config.set(app, key, data, public)
def get(app: int, key: int, public: bool = False) -> bytes | None:
return config.get(app, key, public)
def delete(
app: int, key: int, public: bool = False, writable_locked: bool = False
) -> None:
config.delete(app, key, public, writable_locked)
def set_true_or_delete(app: int, key: int, value: bool) -> None:
if value:
set_bool(app, key, value)
else:
delete(app, key)
def set_bool(app: int, key: int, value: bool, public: bool = False) -> None:
if value:
set(app, key, _TRUE_BYTE, public)
else:
set(app, key, _FALSE_BYTE, public)
def get_bool(app: int, key: int, public: bool = False) -> bool:
return get(app, key, public) == _TRUE_BYTE
def set_uint8(app: int, key: int, val: int, public: bool = False) -> None:
set(app, key, val.to_bytes(1, "big"), public)
def get_uint8(app: int, key: int, public: bool = False) -> int | None:
val = get(app, key, public)
if not val:
return None
return int.from_bytes(val, "big")
def set_uint16(app: int, key: int, val: int) -> None:
set(app, key, val.to_bytes(2, "big"))
def get_uint16(app: int, key: int) -> int | None:
val = get(app, key)
if not val:
return None
return int.from_bytes(val, "big")
def next_counter(app: int, key: int, writable_locked: bool = False) -> int:
return config.next_counter(app, key, writable_locked)
def set_counter(app: int, key: int, count: int, writable_locked: bool = False) -> None:
config.set_counter(app, key, count, writable_locked)