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

test(core): add tests for device_secret and cred_auth_key_counter

Browse Source 
[no changelog]
This commit is contained in:
M1nd3r 2024-05-24 17:32:04 +02:00 committed by Petr Sedláček
parent 45d06f13eb
commit 5c6af8332a
1 changed files with 44 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
core/tests/test_storage.py
View File

@ -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()