wire: fix read_message for v1 protocol

2016-10-06 12:34:09 +02:00 · 2016-10-06 12:34:09 +02:00 · 4b0da8677a
parent c3ee7dd709
commit 4b0da8677a
1 changed files with 6 additions and 2 deletions
--- a/src/trezor/wire/init.py
+++ b/src/trezor/wire/init.py
@ -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