BTCPrivate
/
python-trezor
mirror of https://github.com/BTCPrivate/python-trezor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

transport_udp: Support TREZOR_TRANSPORT_V1

This commit is contained in:
Saleem Rashid 2017-12-16 21:39:09 +00:00 committed by Pavol Rusnak
parent de95c44ad1
commit 1193b0ee85
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trezorlib/transport_udp.py
View File

@ -18,8 +18,10 @@
from __future__ import absolute_import from __future__ import absolute_import
import os
import socket import socket
from .protocol_v1 import ProtocolV1
from .protocol_v2 import ProtocolV2 from .protocol_v2 import ProtocolV2
from .transport import Transport from .transport import Transport
@ -40,7 +42,11 @@ class UdpTransport(Transport):
host = devparts[0] host = devparts[0]
port = int(devparts[1]) if len(devparts) > 1 else UdpTransport.DEFAULT_PORT port = int(devparts[1]) if len(devparts) > 1 else UdpTransport.DEFAULT_PORT
if not protocol: if not protocol:
force_v1 = os.environ.get('TREZOR_TRANSPORT_V1', '0')
if not int(force_v1):
protocol = ProtocolV2() protocol = ProtocolV2()
else:
protocol = ProtocolV1()
self.device = (host, port) self.device = (host, port)
self.protocol = protocol self.protocol = protocol
self.socket = None self.socket = None