mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-02-10 14:42:44 +00:00
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
from trezor import config, utils, wire
|
|
from trezor.messages import MessageType
|
|
from trezor.messages.Features import Features
|
|
from trezor.messages.Success import Success
|
|
from trezor.wire import protobuf_workflow, register
|
|
|
|
from apps.common import cache, storage
|
|
|
|
|
|
def get_features():
|
|
f = Features()
|
|
f.vendor = 'trezor.io'
|
|
f.language = 'english'
|
|
f.major_version = utils.symbol('VERSION_MAJOR')
|
|
f.minor_version = utils.symbol('VERSION_MINOR')
|
|
f.patch_version = utils.symbol('VERSION_PATCH')
|
|
f.revision = utils.symbol('GITREV')
|
|
f.model = utils.model()
|
|
if f.model == 'EMU':
|
|
f.model = 'T' # emulator currently emulates model T
|
|
f.device_id = storage.get_device_id()
|
|
f.label = storage.get_label()
|
|
f.initialized = storage.is_initialized()
|
|
f.pin_protection = config.has_pin()
|
|
f.pin_cached = config.has_pin()
|
|
f.passphrase_protection = storage.has_passphrase()
|
|
f.passphrase_cached = cache.has_passphrase()
|
|
f.needs_backup = storage.needs_backup()
|
|
f.unfinished_backup = storage.unfinished_backup()
|
|
f.flags = storage.get_flags()
|
|
return f
|
|
|
|
|
|
async def handle_Initialize(ctx, msg):
|
|
if msg.state is None or msg.state != cache.get_state(prev_state=bytes(msg.state)):
|
|
cache.clear(msg.skip_passphrase)
|
|
return get_features()
|
|
|
|
|
|
async def handle_GetFeatures(ctx, msg):
|
|
return get_features()
|
|
|
|
|
|
async def handle_Cancel(ctx, msg):
|
|
raise wire.ActionCancelled('Cancelled')
|
|
|
|
|
|
async def handle_ClearSession(ctx, msg):
|
|
cache.clear()
|
|
return Success(message='Session cleared')
|
|
|
|
|
|
async def handle_Ping(ctx, msg):
|
|
if msg.button_protection:
|
|
from apps.common.confirm import require_confirm
|
|
from trezor.messages.ButtonRequestType import ProtectCall
|
|
from trezor.ui.text import Text
|
|
await require_confirm(ctx, Text('Confirm'), ProtectCall)
|
|
if msg.passphrase_protection:
|
|
from apps.common.request_passphrase import protect_by_passphrase
|
|
await protect_by_passphrase(ctx)
|
|
return Success(message=msg.message)
|
|
|
|
|
|
def boot():
|
|
register(MessageType.Initialize, protobuf_workflow, handle_Initialize)
|
|
register(MessageType.GetFeatures, protobuf_workflow, handle_GetFeatures)
|
|
register(MessageType.Cancel, protobuf_workflow, handle_Cancel)
|
|
register(MessageType.ClearSession, protobuf_workflow, handle_ClearSession)
|
|
register(MessageType.Ping, protobuf_workflow, handle_Ping)
|