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

wire: fix read_message for v1 protocol

Browse Source
This commit is contained in:
Jan Pochyla 2016-10-06 12:34:09 +02:00
parent c3ee7dd709
commit 4b0da8677a
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/trezor/wire/__init__.py
View File

@ -78,8 +78,12 @@ def setup():
async def read_message(session_id, *exp_types):
log.info(__name__, 'reading message of types %s', exp_types)
future = Future()
wire_decoder = decode_wire_stream(
_dispatch_and_build_protobuf, session_id, exp_types, future)
if session_id == SESSION_V1:
wire_decoder = decode_wire_v1_stream(
_dispatch_and_build_protobuf, session_id, exp_types, future)
else:
wire_decoder = decode_wire_stream(
_dispatch_and_build_protobuf, session_id, exp_types, future)
wire_decoder.send(None)
register_session(session_id, wire_decoder)
return await future