arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-06 12:51:21 +00:00
Code Issues Releases Wiki Activity

core/slip39: remove partial persistance in Shamir recovery

Browse Source
This commit is contained in:
Tomas Susanka 2019-06-27 13:34:52 +02:00
parent bc70fe0604
commit 4f8022c5cd
3 changed files with 16 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/src/apps/common/mnemonic/bip39.py
View File

@ -34,7 +34,7 @@ def get_seed(secret: bytes, passphrase: str):
return seed
def get_mnemonic_count(mnemonic: str) -> int:
def get_mnemonic_threshold(mnemonic: str) -> int:
return 1

6
core/src/apps/common/mnemonic/slip39.py
View File

@ -66,7 +66,9 @@ def process_all(mnemonics: list) -> bytes:
Receives all mnemonics and processes it into pre-master secret which is usually then
stored in the storage.
"""
_, _, secret = slip39.combine_mnemonics(mnemonics)
identifier, iteration_exponent, secret = slip39.combine_mnemonics(mnemonics)
storage.set_slip39_iteration_exponent(iteration_exponent)
storage.set_slip39_identifier(identifier)
return secret
@ -86,6 +88,6 @@ def get_seed(encrypted_master_secret: bytes, passphrase: str):
return master_secret
def get_mnemonic_count(mnemonic: str) -> int:
def get_mnemonic_threshold(mnemonic: str) -> int:
_, _, _, _, _, _, threshold, _ = slip39.decode_mnemonic(mnemonic)
return threshold

23
core/src/apps/management/recovery_device.py
View File

@ -60,9 +60,8 @@ async def recovery_device(ctx, msg):
wordcount = storage.get_slip39_words_count()
mnemonic_module = mnemonic.slip39
if msg.dry_run:
dry_run_mnemonics = []
dry_run_mnemonic_count = None
mnemonic_threshold = None
mnemonics = []
secret = None
while secret is None:
@ -70,21 +69,15 @@ async def recovery_device(ctx, msg):
words = await request_mnemonic(
ctx, wordcount, mnemonic_module == mnemonic.slip39
)
if mnemonic_threshold is None:
mnemonic_threshold = mnemonic_module.get_mnemonic_threshold(words)
mnemonics.append(words)
remaining = mnemonic_threshold - len(mnemonics)
if remaining == 0:
try:
if msg.dry_run:
if dry_run_mnemonic_count is None:
dry_run_mnemonic_count = mnemonic_module.get_mnemonic_count(words)
dry_run_mnemonics.append(words)
else:
secret = mnemonic_module.process_single(words)
secret = mnemonic_module.process_all(mnemonics)
except slip39.MnemonicError as e:
raise wire.ProcessError("Mnemonic is not valid: " + str(e))
if msg.dry_run:
remaining = dry_run_mnemonic_count - len(dry_run_mnemonics)
if remaining == 0:
secret = mnemonic_module.process_all(dry_run_mnemonics)
else:
remaining = storage.get_slip39_remaining()
# show a number of remaining mnemonics for SLIP39
if secret is None and mnemonic_module == mnemonic.slip39:
await show_remaining_slip39_mnemonics(ctx, title, remaining)