arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-26 16:18:22 +00:00
Code Issues Releases Wiki Activity

fix(tests): update shamir_mnemonic usage

Browse Source
This commit is contained in:
matejcik 2021-02-03 13:39:26 +01:00 committed by matejcik
parent 00c8080ad8
commit a1cd0b0949
4 changed files with 19 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/click_tests/reset.py
View File

@ -1,4 +1,4 @@
import shamir_mnemonic as shamir
from shamir_mnemonic import shamir
from trezorlib import messages
@ -72,7 +72,6 @@ def confirm_words(debug, words):
def validate_mnemonics(mnemonics, expected_ems):
# We expect these combinations to recreate the secret properly
# In case of click tests the mnemonics are always XofX so no need for combinations
ms = shamir.combine_mnemonics(mnemonics)
identifier, iteration_exponent, _, _, _ = shamir._decode_mnemonics(mnemonics)
ems = shamir._encrypt(ms, b"", iteration_exponent, identifier)
assert ems == expected_ems
groups = shamir.decode_mnemonics(mnemonics)
ems = shamir.recover_ems(groups)
assert expected_ems == ems.ciphertext

9
tests/device_tests/test_msg_resetdevice_slip39_advanced.py
View File

@ -17,7 +17,7 @@
from unittest import mock
import pytest
import shamir_mnemonic as shamir
from shamir_mnemonic import shamir
from trezorlib import device, messages as proto
from trezorlib.exceptions import TrezorFailure
@ -186,7 +186,6 @@ def validate_mnemonics(mnemonics, threshold, expected_ems):
# 3of5 shares 3of5 groups
# TODO: test all possible group+share combinations?
test_combination = mnemonics[0:3] + mnemonics[5:8] + mnemonics[10:13]
ms = shamir.combine_mnemonics(test_combination)
identifier, iteration_exponent, _, _, _ = shamir._decode_mnemonics(test_combination)
ems = shamir._encrypt(ms, b"", iteration_exponent, identifier)
assert ems == expected_ems
groups = shamir.decode_mnemonics(test_combination)
ems = shamir.recover_ems(groups)
assert expected_ems == ems.ciphertext

11
tests/device_tests/test_msg_resetdevice_slip39_basic.py
View File

@ -18,8 +18,7 @@ from itertools import combinations
from unittest import mock
import pytest
import shamir_mnemonic as shamir
from shamir_mnemonic import MnemonicError
from shamir_mnemonic import MnemonicError, shamir
from trezorlib import device, messages as proto
from trezorlib.exceptions import TrezorFailure
@ -144,11 +143,9 @@ class TestMsgResetDeviceT2:
def validate_mnemonics(mnemonics, threshold, expected_ems):
# We expect these combinations to recreate the secret properly
for test_group in combinations(mnemonics, threshold):
# TODO: HOTFIX, we should fix this properly by modifying and unifying the python-shamir-mnemonic API
ms = shamir.combine_mnemonics(test_group)
identifier, iteration_exponent, _, _, _ = shamir._decode_mnemonics(test_group)
ems = shamir._encrypt(ms, b"", iteration_exponent, identifier)
assert ems == expected_ems
groups = shamir.decode_mnemonics(test_group)
ems = shamir.recover_ems(groups)
assert expected_ems == ems.ciphertext
# We expect these combinations to raise MnemonicError
for test_group in combinations(mnemonics, threshold - 1):
with pytest.raises(

17
tests/device_tests/test_reset_backup.py
View File

@ -18,7 +18,7 @@
from unittest import mock
import pytest
import shamir_mnemonic as shamir
from shamir_mnemonic import shamir
from trezorlib import device, messages
from trezorlib.messages import BackupType, ButtonRequestType as B
@ -108,11 +108,9 @@ def backup_flow_slip39_basic(client):
)
device.backup(client)
mnemonics = mnemonics[:3]
ms = shamir.combine_mnemonics(mnemonics)
identifier, iteration_exponent, _, _, _ = shamir._decode_mnemonics(mnemonics)
secret = shamir._encrypt(ms, b"", iteration_exponent, identifier)
return secret
groups = shamir.decode_mnemonics(mnemonics[:3])
ems = shamir.recover_ems(groups)
return ems.ciphertext
def backup_flow_slip39_advanced(client):
@ -172,10 +170,9 @@ def backup_flow_slip39_advanced(client):
device.backup(client)
mnemonics = mnemonics[0:3] + mnemonics[5:8] + mnemonics[10:13]
ms = shamir.combine_mnemonics(mnemonics)
identifier, iteration_exponent, _, _, _ = shamir._decode_mnemonics(mnemonics)
secret = shamir._encrypt(ms, b"", iteration_exponent, identifier)
return secret
groups = shamir.decode_mnemonics(mnemonics)
ems = shamir.recover_ems(groups)
return ems.ciphertext
VECTORS = [