找出分帧消息读取器里的 bug

from struct import Struct

HEADER = Struct("!I")

def receive_message(sock):
    header = sock.recv(HEADER.size)
    (size,) = HEADER.unpack(header)
    payload = sock.recv(size)
    return payload.decode("utf-8")