diff --git a/trezorlib/client.py b/trezorlib/client.py index 2d62db6a1..dbb4c5424 100644 --- a/trezorlib/client.py +++ b/trezorlib/client.py @@ -337,6 +337,7 @@ class DebugLinkMixin(object): class ProtocolMixin(object): PRIME_DERIVATION_FLAG = 0x80000000 + VENDORS = ('bitcointrezor.com',) def __init__(self, *args, **kwargs): super(ProtocolMixin, self).__init__(*args, **kwargs) @@ -348,6 +349,8 @@ class ProtocolMixin(object): def init_device(self): self.features = expect(proto.Features)(self.call)(proto.Initialize()) + if str(self.features.vendor) not in self.VENDORS: + raise Exception("Unsupported device") def _get_local_entropy(self): return os.urandom(32)