1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 15:38:11 +00:00

Implemented test of pin exponential backoff

This commit is contained in:
slush0 2013-10-11 03:51:45 +02:00
parent b8ddd0279a
commit 121e189643
3 changed files with 43 additions and 3 deletions

View File

@ -1,3 +1,4 @@
import time
import unittest
import common
@ -10,7 +11,7 @@ class TestProtectCall(common.TrezorTest):
entropy_len = 10
entropy = self.client.get_entropy(entropy_len)
self.assertEqual(len(entropy), entropy_len)
def test_no_protection(self):
self.client.load_device(seed=self.mnemonic1, pin='')
@ -30,6 +31,37 @@ class TestProtectCall(common.TrezorTest):
def test_cancelled_pin(self):
self.client.setup_debuglink(button=True, pin_correct=-1) # PIN cancel
self.assertRaises(PinException, self._some_protected_call)
def test_exponential_backoff_with_reboot(self):
self.client.setup_debuglink(button=True, pin_correct=False)
def test_backoff(attempts, start):
expected = 1.8 ** attempts
got = time.time() - start
msg = "Pin delay expected to be at least %s seconds, got %s" % (expected, got)
print msg
self.assertLessEqual(expected, got, msg)
for attempt in range(1, 6):
start = time.time()
self.assertRaises(PinException, self._some_protected_call)
test_backoff(attempt, start)
# Unplug Trezor now
self.client.debuglink.stop()
self.client.close()
# Give it some time to reboot (it may take some time on RPi)
boot_delay = 5
start = time.time()
time.sleep(boot_delay)
# Connect to Trezor again
self.setUp()
print "Expected reboot time %s seconds" % (1.8 ** attempt)
print "Rebooted in %s seconds" % (time.time() - start)
self.assertLessEqual(1.8 ** attempt, time.time() - start, "Bootup took less than expected!")
if __name__ == '__main__':
unittest.main()

View File

@ -41,7 +41,12 @@ class TrezorClient(object):
def init_device(self):
self.master_public_key = None
self.features = self.call(proto.Initialize())
def close(self):
self.transport.close()
if self.debuglink:
self.debuglink.transport.close()
def get_master_public_key(self):
if self.master_public_key:
return self.master_public_key

View File

@ -45,4 +45,7 @@ class DebugLink(object):
self.press_button(True)
def press_no(self):
self.press_button(False)
self.press_button(False)
def stop(self):
self.transport.write(proto.DebugLinkStop())