Python中socket接收的习惯用法

Question

我有一些在C语言中使用Berkeley套接字API进行套接字编程的经验。一般来说，任何套接字编程都需要一种策略，使接收套接字知道它应该接收多少数据。这可以使用头长度字段或分隔符字符来完成。通常，我更喜欢包含长度的头字段。在

当然，我们还需要知道length头字段本身的大小，它只是一个固定的大小值，必须由发送方和接收方共同商定。在C语言中，这很容易实现，因为本机整数类型是固定大小的，并且是二进制格式，所以您可以这样说：

uint16_t bytes_to_receive;
recv(sock, &bytes_to_receive, sizeof(bytes_to_receive), 0);
bytes_to_receive = ntohs(bytes_to_receive);
// Now receive 'bytes_to_receive' bytes...

但是使用Python套接字是如何实现这种习惯用法的呢？在Python中，整数是对象，pickled整数是可变长度的字节数组。所以我们不能使用pickled integer作为长度头字段，因为我们不能确定它的大小（以字节为单位）。在

当然，我总是可以发送一个包含二进制整数的已知大小的字节数组，比如b'\x05\x00'，以little-endian格式创建一个值为5的16位二进制整数，但这似乎不是正确的方法。在

那么，这通常是如何用Python实现的呢？在

Answer 1

您可以使用struct模块将Python整数与字符串/字节数组进行转换。只需读取与类型头大小相对应的字节数，并用struct模块对其进行转换，就可以开始了。（注意：编码/解码时一定要使用右端标志）

Answer 2

^{}模块提供了getsizeof()函数，该函数以字节为单位返回对象的大小（使用objects__sizeof__方法）。如果使用自定义对象，则需要仔细测试__sizeof__实现，但听起来这对于标准类型应该可以正常工作。在

或者，也可以将数据序列化为pickle或json，并计算字符串中的字符数，尽管这可能会导致性能损失。在

使用这两种方法，如果要传输可变长度的数据，首先传输大小，然后使用该值确定要读取多少数据。在

其他注意事项：

• 如果还没有，还需要阅读sockets的api文档。在
• 请注意，复合类型（如列表）需要额外的空间，因此：

    >>> import sys
    >>> a = [1,3,4]
    >>> sys.getsizeof(a)
    96
    >>> l = 0
    >>> for i in a:
    ...     l += sys.getsizeof(i)
    ... 
    >>> print l
    72
    >>>

Answer 3

ctypes module可以为示例中使用的C类型uint16提供{}：

>>> import ctypes
>>> ctypes.sizeof(ctypes.c_uint16)
2