Pārlūkot izejas kodu

Merge pull request #7 from xiaosuo/payload-size

Fix crashing when payload size is greater than 125
taddeus 8 gadi atpakaļ
vecāks
revīzija
7f4c8ee218
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 2 2
      frame.py

+ 2 - 2
frame.py

@@ -305,12 +305,12 @@ def contains_frame(data):
 
     if payload_len == 126:
         if len(data) > 4:
-            payload_len = struct.unpack('!H', data[2:4])
+            payload_len = struct.unpack('!H', data[2:4])[0]
 
         payload_start = 4
     elif payload_len == 127:
         if len(data) > 12:
-            payload_len = struct.unpack('!Q', data[4:12])
+            payload_len = struct.unpack('!Q', data[4:12])[0]
 
         payload_start = 12