瀏覽代碼

Code cleanup

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

+ 3 - 5
frame.py

@@ -139,19 +139,17 @@ def recvn(sock, n):
     """
     Keep receiving data from `sock' until exactly `n' bytes have been read.
     """
-    left = n
     data = ''
 
-    while left > 0:
-        received = sock.recv(left)
+    while len(data) < n:
+        received = sock.recv(n - len(data))
 
         if not len(received):
             raise SocketClosed()
 
         data += received
-        left -= len(received)
 
-    return received
+    return data
 
 
 def mask(key, original):