mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-22 15:38:11 +00:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import pytest
|
|
|
|
from python.src import consts
|
|
from python.src.norcow import NC_CLASSES
|
|
|
|
from . import common
|
|
|
|
|
|
@pytest.mark.parametrize("nc_class", NC_CLASSES)
|
|
def test_init_pin(nc_class):
|
|
sc, sp = common.init(nc_class, uid=b"\x00\x00\x00\x00\x00\x00")
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
sc, sp = common.init(nc_class, uid=b"\x22\x00\xDD\x00\x00\xBE")
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
|
|
@pytest.mark.parametrize("nc_class", NC_CLASSES)
|
|
def test_change_pin(nc_class):
|
|
sc, sp = common.init(nc_class, unlock=True)
|
|
for s in (sc, sp):
|
|
assert s.change_pin("", "222")
|
|
assert not s.change_pin("9999", "") # invalid PIN
|
|
assert s.unlock("222")
|
|
assert s.change_pin("222", "99999")
|
|
assert s.change_pin("99999", "Trezor")
|
|
assert s.unlock("Trezor")
|
|
assert not s.unlock("9999") # invalid PIN
|
|
assert not s.unlock("99999") # invalid old PIN
|
|
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
|
|
@pytest.mark.parametrize("nc_class", NC_CLASSES)
|
|
def test_has_pin(nc_class):
|
|
sc, sp = common.init(nc_class)
|
|
for s in (sc, sp):
|
|
assert not s.has_pin()
|
|
assert s.unlock("")
|
|
assert not s.has_pin()
|
|
assert s.change_pin("", "22")
|
|
assert s.has_pin()
|
|
assert s.change_pin("22", "")
|
|
assert not s.has_pin()
|
|
|
|
|
|
@pytest.mark.parametrize("nc_class", NC_CLASSES)
|
|
def test_wipe_after_max_pin(nc_class):
|
|
sc, sp = common.init(nc_class, unlock=True)
|
|
for s in (sc, sp):
|
|
assert s.change_pin("", "222")
|
|
assert s.unlock("222")
|
|
s.set(0x0202, b"Hello")
|
|
|
|
# try an invalid PIN MAX - 1 times
|
|
for i in range(consts.PIN_MAX_TRIES - 1):
|
|
assert not s.unlock("9999")
|
|
# this should pass
|
|
assert s.unlock("222")
|
|
assert s.get(0x0202) == b"Hello"
|
|
|
|
# try an invalid PIN MAX times, the storage should get wiped
|
|
for i in range(consts.PIN_MAX_TRIES):
|
|
assert not s.unlock("9999")
|
|
assert i == consts.PIN_MAX_TRIES - 1
|
|
# this should return False and raise an exception, the storage is wiped
|
|
assert not s.unlock("222")
|
|
with pytest.raises(RuntimeError):
|
|
assert s.get(0x0202) == b"Hello"
|
|
|
|
assert common.memory_equals(sc, sp)
|