Просмотр исходного кода

Fix crashing when payload size is greater than 125

Changli Gao 8 лет назад
Родитель
Сommit
43cea0206a
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      frame.py

+ 2 - 2
frame.py

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