Procházet zdrojové kódy

Fix crashing when payload size is greater than 125

Changli Gao před 8 roky
rodič
revize
43cea0206a
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  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