mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-24 07:18:09 +00:00
test(core): add tests for device_secret and cred_auth_key_counter
[no changelog]
This commit is contained in:
parent
45d06f13eb
commit
5c6af8332a
@ -1,13 +1,15 @@
|
||||
from common import * # isort:skip
|
||||
|
||||
from storage import device
|
||||
from trezor import config
|
||||
|
||||
from trezor import config, utils
|
||||
|
||||
class TestConfig(unittest.TestCase):
|
||||
def test_counter(self):
|
||||
|
||||
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)
|
||||
@ -16,6 +18,45 @@ class TestConfig(unittest.TestCase):
|
||||
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 _NAMESPACE, _CRED_AUTH_KEY_COUNTER
|
||||
|
||||
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()
|
||||
|
Loading…
Reference in New Issue
Block a user