arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-03 11:20:59 +00:00
Code Issues Releases Wiki Activity

core: support load device with SLIP-39

Browse Source 
updates #356
This commit is contained in:
Tomas Susanka 2019-07-24 13:31:48 +02:00
parent e95a97f281
commit 2b6a14cb5f
1 changed files with 28 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
core/src/apps/management/load_device.py
View File

@ -1,5 +1,5 @@
from trezor import config, wire from trezor import config, wire
from trezor.crypto import bip39 from trezor.crypto import bip39, slip39
from trezor.messages.Success import Success from trezor.messages.Success import Success
from trezor.pin import pin_to_int from trezor.pin import pin_to_int
from trezor.ui.text import Text from trezor.ui.text import Text
@ -9,14 +9,29 @@ from apps.common.confirm import require_confirm
async def load_device(ctx, msg): async def load_device(ctx, msg):
# TODO implement SLIP-39
if storage.is_initialized(): if storage.is_initialized():
raise wire.UnexpectedMessage("Already initialized") raise wire.UnexpectedMessage("Already initialized")
if msg.node is not None: if msg.node is not None:
raise wire.ProcessError("LoadDevice.node is not supported") raise wire.ProcessError("LoadDevice.node is not supported")
if not msg.skip_checksum and not bip39.check(msg.mnemonic): if not msg.mnemonics:
raise wire.ProcessError("No mnemonic provided")
word_count = len(msg.mnemonics[0].split(" "))
for m in msg.mnemonics[1:]:
if word_count != len(m.split(" ")):
raise wire.ProcessError(
"All shares are required to have the same number of words"
)
mnemonic_type = mnemonic.type_from_word_count(word_count)
if (
mnemonic_type == mnemonic.TYPE_BIP39
and not msg.skip_checksum
and not bip39.check(msg.mnemonics[0])
):
raise wire.ProcessError("Mnemonic is not valid") raise wire.ProcessError("Mnemonic is not valid")
text = Text("Loading seed") text = Text("Loading seed")
@ -24,11 +39,17 @@ async def load_device(ctx, msg):
text.normal("Continue only if you", "know what you are doing!") text.normal("Continue only if you", "know what you are doing!")
await require_confirm(ctx, text) await require_confirm(ctx, text)
if mnemonic_type == mnemonic.TYPE_BIP39:
secret = msg.mnemonics[0].encode()
elif mnemonic_type == mnemonic.TYPE_SLIP39:
identifier, iteration_exponent, secret = slip39.combine_mnemonics(msg.mnemonics)
storage.device.set_slip39_identifier(identifier)
storage.device.set_slip39_iteration_exponent(iteration_exponent)
else:
raise RuntimeError("Unknown mnemonic type")
storage.device.store_mnemonic_secret( storage.device.store_mnemonic_secret(
secret=msg.mnemonic.encode(), secret, mnemonic_type, needs_backup=True, no_backup=False
mnemonic_type=mnemonic.TYPE_BIP39,
needs_backup=True,
no_backup=False,
) )
storage.device.load_settings( storage.device.load_settings(
use_passphrase=msg.passphrase_protection, label=msg.label use_passphrase=msg.passphrase_protection, label=msg.label