mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-15 10:58:09 +00:00
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
# flake8: noqa: F403,F405
|
|
from common import * # isort:skip
|
|
|
|
from storage import device
|
|
from trezor import config, utils
|
|
|
|
|
|
class TestConfig(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
config.init()
|
|
config.wipe()
|
|
|
|
def test_u2f_counter(self):
|
|
for i in range(150):
|
|
self.assertEqual(device.next_u2f_counter(), i)
|
|
device.set_u2f_counter(350)
|
|
for i in range(351, 500):
|
|
self.assertEqual(device.next_u2f_counter(), i)
|
|
device.set_u2f_counter(0)
|
|
self.assertEqual(device.next_u2f_counter(), 1)
|
|
|
|
if utils.USE_THP:
|
|
|
|
def test_cred_auth_key_counter(self):
|
|
rounds = 200
|
|
for i in range(rounds):
|
|
self.assertEqual(
|
|
device.get_cred_auth_key_counter(), i.to_bytes(4, "big")
|
|
)
|
|
device.increment_cred_auth_key_counter()
|
|
|
|
# Test get_cred_auth_key_counter does not change the counter value
|
|
self.assertEqual(
|
|
device.get_cred_auth_key_counter(), rounds.to_bytes(4, "big")
|
|
)
|
|
self.assertEqual(
|
|
device.get_cred_auth_key_counter(), rounds.to_bytes(4, "big")
|
|
)
|
|
|
|
def test_cred_auth_key_counter_overflow(self):
|
|
from storage import common
|
|
from storage.device import _CRED_AUTH_KEY_COUNTER, _NAMESPACE
|
|
|
|
common.set(_NAMESPACE, _CRED_AUTH_KEY_COUNTER, b"\xff\xff\xff\xfe")
|
|
device.increment_cred_auth_key_counter()
|
|
self.assertEqual(device.get_cred_auth_key_counter(), b"\xff\xff\xff\xff")
|
|
with self.assertRaises(AssertionError) as e:
|
|
device.increment_cred_auth_key_counter()
|
|
self.assertEqual(e.value.value, "Overflow of cred_auth_key_counter")
|
|
|
|
def test_device_secret(self):
|
|
secret1 = device.get_device_secret()
|
|
self.assertEqual(len(secret1), 16)
|
|
secret2 = device.get_device_secret()
|
|
self.assertEqual(secret1, secret2)
|
|
config.wipe()
|
|
secret3 = device.get_device_secret()
|
|
self.assertEqual(len(secret3), 16)
|
|
self.assertNotEqual(secret1, secret3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|