Merge Trezor and KeepKey functionality
parent
5932a89dc5
commit
8c6ac43cf4
@ -1,37 +0,0 @@
|
||||
''' Thin wrapper around trezorlib. '''
|
||||
|
||||
|
||||
def client():
|
||||
# pylint: disable=import-error
|
||||
from trezorlib.client import TrezorClient
|
||||
from trezorlib.transport_hid import HidTransport as TrezorHidTransport
|
||||
from trezorlib.messages_pb2 import PassphraseAck as TrezorPassphraseAck
|
||||
|
||||
from keepkeylib.client import KeepKeyClient
|
||||
from keepkeylib.transport_hid import HidTransport as KeepKeyHidTransport
|
||||
from keepkeylib.messages_pb2 import PassphraseAck as KeepKeyPassphraseAck
|
||||
|
||||
devices = list(TrezorHidTransport.enumerate())
|
||||
if len(devices) == 1:
|
||||
t = TrezorClient(TrezorHidTransport(devices[0]))
|
||||
t.callback_PassphraseRequest = lambda msg: TrezorPassphraseAck(passphrase='')
|
||||
else:
|
||||
devices = list(KeepKeyHidTransport.enumerate())
|
||||
if len(devices) != 1:
|
||||
msg = '{:d} devices found'.format(len(devices))
|
||||
raise IOError(msg)
|
||||
t = KeepKeyClient(KeepKeyHidTransport(devices[0]))
|
||||
t.callback_PassphraseRequest = lambda msg: KeepKeyPassphraseAck(passphrase='')
|
||||
|
||||
return t
|
||||
|
||||
|
||||
def trezor_identity_type(**kwargs):
|
||||
# pylint: disable=import-error
|
||||
from trezorlib.types_pb2 import IdentityType
|
||||
return IdentityType(**kwargs)
|
||||
|
||||
def keepkey_identity_type(**kwargs):
|
||||
# pylint: disable=import-error
|
||||
from keepkeylib.types_pb2 import IdentityType
|
||||
return IdentityType(**kwargs)
|
@ -0,0 +1,78 @@
|
||||
''' Thin wrapper around trezor/keepkey libraries. '''
|
||||
import binascii
|
||||
import collections
|
||||
import logging
|
||||
|
||||
import semver
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
ClientWrapper = collections.namedtuple(
|
||||
'ClientWrapper',
|
||||
['connection', 'identity_type', 'device_name'])
|
||||
|
||||
|
||||
# pylint: disable=too-many-arguments
|
||||
def _load_client(name, client_type, hid_transport,
|
||||
passphrase_ack, identity_type, required_version):
|
||||
|
||||
def empty_passphrase_handler(_):
|
||||
return passphrase_ack(passphrase='')
|
||||
|
||||
for d in hid_transport.enumerate():
|
||||
connection = client_type(hid_transport(d))
|
||||
connection.callback_PassphraseRequest = empty_passphrase_handler
|
||||
f = connection.features
|
||||
log.debug('connected to %s %s', name, f.device_id)
|
||||
log.debug('label : %s', f.label)
|
||||
log.debug('vendor : %s', f.vendor)
|
||||
current_version = '{}.{}.{}'.format(f.major_version,
|
||||
f.minor_version,
|
||||
f.patch_version)
|
||||
log.debug('version : %s', current_version)
|
||||
log.debug('revision : %s', binascii.hexlify(f.revision))
|
||||
if not semver.match(current_version, required_version):
|
||||
fmt = 'Please upgrade your {} firmware to {} version (current: {})'
|
||||
raise ValueError(fmt.format(name,
|
||||
required_version,
|
||||
current_version))
|
||||
yield ClientWrapper(connection=connection,
|
||||
identity_type=identity_type,
|
||||
device_name=name)
|
||||
|
||||
|
||||
def _load_trezor():
|
||||
# pylint: disable=import-error
|
||||
from trezorlib.client import TrezorClient
|
||||
from trezorlib.transport_hid import HidTransport
|
||||
from trezorlib.messages_pb2 import PassphraseAck
|
||||
from trezorlib.types_pb2 import IdentityType
|
||||
return _load_client(name='Trezor',
|
||||
client_type=TrezorClient,
|
||||
hid_transport=HidTransport,
|
||||
passphrase_ack=PassphraseAck,
|
||||
identity_type=IdentityType,
|
||||
required_version='>=1.3.4')
|
||||
|
||||
|
||||
def _load_keepkey():
|
||||
# pylint: disable=import-error
|
||||
from keepkeylib.client import KeepKeyClient
|
||||
from keepkeylib.transport_hid import HidTransport
|
||||
from keepkeylib.messages_pb2 import PassphraseAck
|
||||
from keepkeylib.types_pb2 import IdentityType
|
||||
return _load_client(name='KeepKey',
|
||||
client_type=KeepKeyClient,
|
||||
hid_transport=HidTransport,
|
||||
passphrase_ack=PassphraseAck,
|
||||
identity_type=IdentityType,
|
||||
required_version='>=1.0.4')
|
||||
|
||||
|
||||
def load():
|
||||
devices = list(_load_trezor()) + list(_load_keepkey())
|
||||
if len(devices) == 1:
|
||||
return devices[0]
|
||||
|
||||
msg = '{:d} devices found'.format(len(devices))
|
||||
raise IOError(msg)
|
Loading…
Reference in New Issue