确定准备好进行recv()处理的字节数

Question

我可以使用

select()

来确定对

recv()

的调用是否会阻塞，但是一旦我确定有字节要读取，有没有办法在实际读取之前查询当前可用的 多少字节打电话

recv()

？

Answer 1

如果您的操作系统提供了它（大多数操作系统都提供），您可以使用 ioctl(..,FIONREAD,..):

int get_n_readable_bytes(int fd) {
    int n = -1;
    if (ioctl(fd, FIONREAD, &n) < 0) {
        perror("ioctl failed");
        return -1;
    }
    return n;
}

Windows 提供了一个类似的 ioctlsocket(..,FIONREAD,..)，它需要一个指向 unsigned long 的指针：

unsigned long get_n_readable_bytes(SOCKET sock) {
    unsigned long n = -1;
   if (ioctlsocket(sock, FIONREAD, &n) < 0) {
       /* look in WSAGetLastError() for the error code */
       return 0;
   }
   return n;
}

ioctl 调用应该适用于套接字和其他一些 fd，但不是所有 fd。我相信它可以在您可能使用的几乎任何免费的类 UNIX 操作系统上与 TCP 套接字配合良好。对于 UDP 套接字，它的语义有点不同：对于它们来说，它告诉您下一个数据报中的字节数。

Windows 上的 ioctlsocket 调用（显然）仅适用于套接字。

Answer 2

不，协议需要确定这一点。例如：

• 如果您使用固定大小的消息，那么您知道您需要读取 X 字节。
• 您可以读取指示要读取 X 字节的消息标头。
• 您可以阅读直到找到终止字符/序列。