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

protocol: 2/3 compat fixes

This commit is contained in:
Jan Pochyla 2017-09-04 11:44:33 +02:00
parent 3d3c2a29d0
commit 051f8e961b
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trezorlib/protocol_v1.py
View File

@ -67,7 +67,7 @@ class ProtocolV1(object):
raise Exception('Unexpected magic characters')
try:
headerlen = struct.calcsize('>HL')
(msg_type, datalen) = struct.unpack('>HL', bytes(chunk[3:3 + headerlen]))
(msg_type, datalen) = struct.unpack('>HL', chunk[3:3 + headerlen])
except:
raise Exception('Cannot parse header')

15
trezorlib/protocol_v2.py
View File

@ -31,7 +31,7 @@ class ProtocolV2(object):
def session_begin(self, transport):
chunk = struct.pack('>B', 0x03)
chunk = chunk.ljust(REPLEN, bytes([0x00]))
chunk = chunk.ljust(REPLEN, b'\x00')
transport.write_chunk(chunk)
resp = transport.read_chunk()
self.session = self.parse_session_open(resp)
@ -40,10 +40,11 @@ class ProtocolV2(object):
if not self.session:
return
chunk = struct.pack('>BL', 0x04, self.session)
chunk = chunk.ljust(REPLEN, bytes([0x00]))
chunk = chunk.ljust(REPLEN, b'\x00')
transport.write_chunk(chunk)
resp = transport.read_chunk()
if resp[0] != 0x04:
(magic, ) = struct.unpack('>B', resp[:1])
if magic != 0x04:
raise Exception('Expected session close')
self.session = None
@ -65,7 +66,7 @@ class ProtocolV2(object):
repheader = struct.pack('>BLL', 0x02, self.session, seq)
datalen = REPLEN - len(repheader)
chunk = repheader + data[:datalen]
chunk = chunk.ljust(REPLEN, bytes([0x00]))
chunk = chunk.ljust(REPLEN, b'\x00')
transport.write_chunk(chunk)
data = data[datalen:]
seq += 1
@ -95,7 +96,7 @@ class ProtocolV2(object):
def parse_first(self, chunk):
try:
headerlen = struct.calcsize('>BLLL')
(magic, session, msg_type, datalen) = struct.unpack('>BLLL', bytes(chunk[:headerlen]))
(magic, session, msg_type, datalen) = struct.unpack('>BLLL', chunk[:headerlen])
except:
raise Exception('Cannot parse header')
if magic != 0x01:
@ -107,7 +108,7 @@ class ProtocolV2(object):
def parse_next(self, chunk):
try:
headerlen = struct.calcsize('>BLL')
(magic, session, sequence) = struct.unpack('>BLL', bytes(chunk[:headerlen]))
(magic, session, sequence) = struct.unpack('>BLL', chunk[:headerlen])
except:
raise Exception('Cannot parse header')
if magic != 0x02:
@ -119,7 +120,7 @@ class ProtocolV2(object):
def parse_session_open(self, chunk):
try:
headerlen = struct.calcsize('>BL')
(magic, session) = struct.unpack('>BL', bytes(chunk[:headerlen]))
(magic, session) = struct.unpack('>BL', chunk[:headerlen])
except:
raise Exception('Cannot parse header')
if magic != 0x03: