arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-13 11:09:01 +00:00
Code Issues Releases Wiki Activity

fix v2 transport

Browse Source
This commit is contained in:
Jan Pochyla 2016-07-13 17:41:08 +02:00
parent e00453661d
commit 0f48e15c48
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
trezorlib/transport.py
View File

@ -188,12 +188,12 @@ class TransportV2(Transport):
while len(data) < datalen:
chunk = self._read_chunk()
(session_id2, data) = self.parse_next(chunk)
(next_session_id, next_data) = self.parse_next(chunk)
if session_id != session_id2:
if next_session_id != session_id:
raise Exception("Session id mismatch")
data.extend(data)
data.extend(next_data)
# Strip padding zeros
data = data[:datalen]
@ -218,7 +218,7 @@ class TransportV2(Transport):
try:
headerlen = struct.calcsize(">L")
session_id = struct.unpack(">L", chunk[1:1 + headerlen])
(session_id,) = struct.unpack(">L", chunk[1:1 + headerlen])
except:
raise Exception("Cannot parse header")