chore(python): get human-readable errors for all ThpErrorType-s in trezorlib

[no changelog]

python/src/trezorlib/transport/thp/control_byte.py

@@ -43,6 +43,10 @@ def is_ack(ctrl_byte: int) -> bool:
     return ctrl_byte & ACK_MASK == ACK_MESSAGE
 
 
+def is_error(ctrl_byte: int) -> bool:
+    return ctrl_byte == _ERROR
+
+
 def is_continuation(ctrl_byte: int) -> bool:
     return ctrl_byte & CONTINUATION_PACKET_MASK == CONTINUATION_PACKET
 

python/src/trezorlib/transport/thp/protocol_v2.py

@@ -400,11 +400,17 @@ class ProtocolV2(ProtocolAndChannel):
     def read_and_decrypt(self) -> t.Tuple[int, int, bytes]:
         header, raw_payload = self._read_until_valid_crc_check()
         if control_byte.is_ack(header.ctrl_byte):
+            # TODO fix this recursion
             return self.read_and_decrypt()
+        if control_byte.is_error(header.ctrl_byte):
+            # TODO check for different channel
+            err = _get_error_from_int(raw_payload[0])
+            raise Exception("Received ThpError: " + err)
         if not header.is_encrypted_transport():
             click.echo(
-                "Trying to decrypt not encrypted message!"
+                "Trying to decrypt not encrypted message! ("
-                + hexlify(header.to_bytes_init() + raw_payload).decode(),
+                + hexlify(header.to_bytes_init() + raw_payload).decode()
+                + ")",
                 err=True,
             )
 
@@ -467,3 +473,18 @@ class ProtocolV2(ProtocolAndChannel):
             )
             return False
         return True
+
+
+def _get_error_from_int(error_code: int) -> str:
+    # TODO FIXME improve this (ThpErrorType)
+    if error_code == 1:
+        return "TRANSPORT BUSY"
+    if error_code == 2:
+        return "UNALLOCATED CHANNEL"
+    if error_code == 3:
+        return "DECRYPTION FAILED"
+    if error_code == 4:
+        return "INVALID DATA"
+    if error_code == 5:
+        return "DEVICE_LOCKED"
+    raise Exception("Not Implemented error case")