Python socket中socket.recv()的陷阱

问题描述

在Python中使用socket库的socket.recv()函数时，实际接收的数据大小可能会小于指定的大小，这是因为该函数并不保证能完全接收指定大小的数据。
为了解决这一问题，我们可以像socket.sendall()一样使用循环来保证接收到的数据大小与指定的大小一致。

具体代码

import socket

def recvall(self, length):
    data = b''
    while len(data) < length: #循环接收数据
        data += self.recv(length - len(data))
    return data
socket.socket.recvall = recvall #给socket类添加recvall方法