arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-16 11:28:14 +00:00
Code Issues Releases Wiki Activity

trezorctl/client: add self_test

Browse Source
This commit is contained in:
Pavol Rusnak 2017-07-03 18:49:03 +02:00
parent b9293d6bcb
commit bcf54dbe94
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
trezorctl
View File

@ -350,6 +350,10 @@ class Commands(object):
ret = self.client.set_u2f_counter(args.counter) ret = self.client.set_u2f_counter(args.counter)
return ret return ret
def self_test(self, args):
ret = self.client.self_test()
return ret
def firmware_update(self, args): def firmware_update(self, args):
if args.file: if args.file:
fp = open(args.file, 'rb').read() fp = open(args.file, 'rb').read()
@ -410,6 +414,7 @@ class Commands(object):
encrypt_keyvalue.help = 'Encrypt value by given key and path' encrypt_keyvalue.help = 'Encrypt value by given key and path'
decrypt_keyvalue.help = 'Decrypt value by given key and path' decrypt_keyvalue.help = 'Decrypt value by given key and path'
set_u2f_counter.help = 'Set U2F counter' set_u2f_counter.help = 'Set U2F counter'
self_test.help = 'Perform a self-test'
firmware_update.help = 'Upload new firmware to device (must be in bootloader mode)' firmware_update.help = 'Upload new firmware to device (must be in bootloader mode)'
clear_session.arguments= () clear_session.arguments= ()
@ -552,6 +557,8 @@ class Commands(object):
(('counter',), {'type': int}), (('counter',), {'type': int}),
) )
self_test.arguments = ()
firmware_update.arguments = ( firmware_update.arguments = (
(('-f', '--file'), {'type': str}), (('-f', '--file'), {'type': str}),
(('-u', '--url'), {'type': str}), (('-u', '--url'), {'type': str}),

8
trezorlib/client.py
View File

@ -1019,6 +1019,14 @@ class ProtocolMixin(object):
raise Exception("Unexpected message %s" % resp) raise Exception("Unexpected message %s" % resp)
@field('message')
@expect(proto.Success)
def self_test(self):
if self.features.bootloader_mode is False:
raise Exception("Device must be in bootloader mode")
return self.call(proto.SelfTest())
class TrezorClient(ProtocolMixin, TextUIMixin, BaseClient): class TrezorClient(ProtocolMixin, TextUIMixin, BaseClient):
pass pass