瀏覽代碼

Fixed parameter bug

Taddeus Kroes 13 年之前
父節點
當前提交
3f383b614a
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      frame.py

+ 2 - 2
frame.py

@@ -126,10 +126,10 @@ def receive_frame(sock):
 
 
     if mask:
     if mask:
         masking_key = recvn(sock, 4)
         masking_key = recvn(sock, 4)
-        payload = mask(masking_key, recvn(payload_len))
+        payload = mask(masking_key, recvn(sock, payload_len))
     else:
     else:
         masking_key = ''
         masking_key = ''
-        payload = recvn(payload_len)
+        payload = recvn(sock, payload_len)
 
 
     return Frame(opcode, payload, masking_key=masking_key, final=final,
     return Frame(opcode, payload, masking_key=masking_key, final=final,
                     rsv1=rsv1, rsv2=rsv2, rsv3=rsv3)
                     rsv1=rsv1, rsv2=rsv2, rsv3=rsv3)