mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-22 06:18:07 +00:00
0ff7034e37
This commit accomplishes several goals: 1) it removes any upy dependencies from storage.c/storage.h 2) ui wait callback is set during config_init and storage_init, which allows to simplify the code dramatically
137 lines
4.1 KiB
Python
137 lines
4.1 KiB
Python
from common import *
|
|
|
|
from trezor.crypto import random
|
|
from trezor.pin import pin_to_int
|
|
|
|
from trezor import config
|
|
|
|
PINAPP = 0x00
|
|
PINKEY = 0x00
|
|
|
|
|
|
def random_entry():
|
|
while True:
|
|
appid, key = 1 + random.uniform(127), random.uniform(256)
|
|
if appid != PINAPP or key != PINKEY:
|
|
break
|
|
return appid, key
|
|
|
|
|
|
class TestConfig(unittest.TestCase):
|
|
|
|
def test_init(self):
|
|
config.init()
|
|
config.init()
|
|
config.init()
|
|
|
|
def test_wipe(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
config.set(1, 1, b'hello')
|
|
config.set(1, 2, b'world')
|
|
v0 = config.get(1, 1)
|
|
v1 = config.get(1, 2)
|
|
self.assertEqual(v0, b'hello')
|
|
self.assertEqual(v1, b'world')
|
|
config.wipe()
|
|
v0 = config.get(1, 1)
|
|
v1 = config.get(1, 2)
|
|
self.assertEqual(v0, bytes())
|
|
self.assertEqual(v1, bytes())
|
|
|
|
def test_lock(self):
|
|
for _ in range(128):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
appid, key = random_entry()
|
|
value = random.bytes(16)
|
|
config.set(appid, key, value)
|
|
config.init()
|
|
self.assertEqual(config.get(appid, key), bytes())
|
|
with self.assertRaises(RuntimeError):
|
|
config.set(appid, key, bytes())
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.change_pin(pin_to_int(''), pin_to_int('000')), False)
|
|
|
|
def test_public(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
|
|
appid, key = random_entry()
|
|
|
|
value32 = random.bytes(32)
|
|
config.set(appid, key, value32)
|
|
value16 = random.bytes(16)
|
|
config.set(appid, key, value16, True)
|
|
|
|
v1 = config.get(appid, key)
|
|
v2 = config.get(appid, key, True)
|
|
self.assertNotEqual(v1, v2)
|
|
self.assertEqual(v1, value32)
|
|
self.assertEqual(v2, value16)
|
|
|
|
config.init()
|
|
|
|
v1 = config.get(appid, key)
|
|
v2 = config.get(appid, key, True)
|
|
self.assertNotEqual(v1, v2)
|
|
self.assertEqual(v1, bytes())
|
|
self.assertEqual(v2, value16)
|
|
|
|
def test_change_pin(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
with self.assertRaises(RuntimeError):
|
|
config.set(PINAPP, PINKEY, b'value')
|
|
self.assertEqual(config.change_pin(pin_to_int('000'), pin_to_int('666')), False)
|
|
self.assertEqual(config.change_pin(pin_to_int(''), pin_to_int('000')), True)
|
|
self.assertEqual(config.get(PINAPP, PINKEY), bytes())
|
|
config.set(1, 1, b'value')
|
|
config.init()
|
|
self.assertEqual(config.unlock(pin_to_int('000')), True)
|
|
config.change_pin(pin_to_int('000'), pin_to_int(''))
|
|
config.init()
|
|
self.assertEqual(config.unlock(pin_to_int('000')), False)
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
self.assertEqual(config.get(1, 1), b'value')
|
|
|
|
def test_set_get(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
for _ in range(32):
|
|
appid, key = random_entry()
|
|
value = random.bytes(128)
|
|
config.set(appid, key, value)
|
|
value2 = config.get(appid, key)
|
|
self.assertEqual(value, value2)
|
|
|
|
def test_compact(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
appid, key = 1, 1
|
|
for _ in range(259):
|
|
value = random.bytes(259)
|
|
config.set(appid, key, value)
|
|
value2 = config.get(appid, key)
|
|
self.assertEqual(value, value2)
|
|
|
|
def test_get_default(self):
|
|
config.init()
|
|
config.wipe()
|
|
self.assertEqual(config.unlock(pin_to_int('')), True)
|
|
for _ in range(128):
|
|
appid, key = random_entry()
|
|
value = config.get(appid, key)
|
|
self.assertEqual(value, bytes())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|