message read length fix

2018-11-12 12:26:11 -05:00 · 2018-11-12 12:26:11 -05:00 · 6e3857c89a
commit 6e3857c89a
parent 61e11a2716
2 changed files with 3 additions and 3 deletions
--- a/muxer/mplex/muxed_connection.py
+++ b/muxer/mplex/muxed_connection.py
@ -92,9 +92,9 @@ class MuxedConn(IMuxedConn):
            data += chunk

            header, end_index = decode_uvarint(data, 0)
-            length, end_index = decode_uvarint(data, end_index + 1)
+            length, end_index = decode_uvarint(data, end_index)

-            message = data[-length:]
+            message = data[end_index:end_index + length + 1]

            # Deal with other types of messages
            flag = header & 0x07
--- a/muxer/mplex/utils.py
+++ b/muxer/mplex/utils.py
@ -22,4 +22,4 @@ def decode_uvarint(buff, index):
            break
        index += 1

-    return result, index
+    return result, index + 1