Function wait_for_nBytes

code/IO/tcp/client.py:35–47

(n, Sock)

import struct

def wait_for_nBytes(n, Sock):
    Bytes = b''
    received = 0
    block_size = 2048
    while received < n:
        data = Sock.recv(min(n-received, block_size))
        if not data:
            break
        Bytes += data
        received += len(data)
        if len(Bytes) == n:
            break
    return Bytes

def main():
    Host, Port = '127.0.0.1', 5006