arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-07-03 05:12:34 +00:00
Code Issues Releases Wiki Activity

Enabled HID transport, 'list' command enumerates HID devices

Browse Source
This commit is contained in:
slush 2013-04-01 16:59:42 +02:00
parent 5d1e1a79ae
commit 5f381bd936
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cmd.py
View File

@ -36,12 +36,14 @@ def parse_args(commands):
item.add_argument(*arg[0], **arg[1]) item.add_argument(*arg[0], **arg[1])
item.set_defaults(func=func) item.set_defaults(func=func)
item.set_defaults(cmd=cmd)
return parser.parse_args() return parser.parse_args()
def get_transport(transport_string, path): def get_transport(transport_string, path):
if transport_string == 'usb': if transport_string == 'usb':
raise NotImplemented("USB HID transport not implemented yet") from bitkeylib.transport_hid import HidTransport
return HidTransport(path)
if transport_string == 'serial': if transport_string == 'serial':
from bitkeylib.transport_serial import SerialTransport from bitkeylib.transport_serial import SerialTransport
@ -69,6 +71,10 @@ class Commands(object):
def _list_commands(cls): def _list_commands(cls):
return [ x for x in dir(cls) if not x.startswith('_') ] return [ x for x in dir(cls) if not x.startswith('_') ]
def list(self, args):
# Fake method for advertising 'list' command
pass
def get_address(self, args): def get_address(self, args):
return self.client.get_address(args.n) return self.client.get_address(args.n)
def get_entropy(self, args): def get_entropy(self, args):
@ -85,6 +91,7 @@ class Commands(object):
return self.client.load_device(seed, args.otp, args.pin, args.spv) return self.client.load_device(seed, args.otp, args.pin, args.spv)
list.help = 'List connected Trezor USB devices'
get_address.help = 'Get bitcoin address in base58 encoding' get_address.help = 'Get bitcoin address in base58 encoding'
get_entropy.help = 'Get example entropy' get_entropy.help = 'Get example entropy'
get_uuid.help = 'Get device\'s unique identifier' get_uuid.help = 'Get device\'s unique identifier'
@ -109,6 +116,16 @@ class Commands(object):
def main(): def main():
args = parse_args(Commands) args = parse_args(Commands)
if args.cmd == 'list':
from bitkeylib.transport_hid import HidTransport
devices = HidTransport.enumerate()
if args.json:
print json.dumps(devices)
else:
for dev in devices:
print dev
return
transport = get_transport(args.transport, args.path) transport = get_transport(args.transport, args.path)
if args.debug: if args.debug:
debuglink_transport = get_transport(args.debuglink_transport, args.debuglink_path) debuglink_transport = get_transport(args.debuglink_transport, args.debuglink_path)