c++ - 从套接字接收前几个字节以确定缓冲区大小

Question

我正在用 c++ 使用 TCP/IP 和 sockets 编写一个分布式系统。

对于我的每条消息，我需要接收前 5 个字节才能知道传入消息的完整长度。

执行此操作的最佳方法是什么？

1. recv() 只有 5 个字节，然后再次 recv()。如果我选择这个，假设我会在 recv 中得到 0 或 5 个字节(也就是不写一个循环来继续尝试)是否安全？
2. 使用MSG_PEEK
3. recv() 一些较大的缓冲区大小，然后读取前 5 个字节并分配最终缓冲区。

Answer 1

您不需要知道任何事情。 TCP 是一种流协议(protocol)，在任何给定时刻，您可以获得少至一个字节或多至数兆字节的数据。使用 TCP 套接字的正确且唯一的方法是循环读取。

char buf[4096];        // or whatever

std::deque<char> data;

for (int res ; ; )
{
    res = recv(fd, buf, sizeof buf, MSG_DONTWAIT);

    if (res == -1)
    {
        if (errno == EAGAIN || errno == EWOULDBLOCK)
        {
            break;  // done reading
        }
        else
        {
            // error, break, die
        }
    }
    if (res == 0)
    {
        // socket closed, finalise, break
    }
    else
    {
        data.insert(data.end(), buf, buf + res);
    }
}

循环的唯一目的是将数据从套接字缓冲区传输到您的应用程序。然后，您的应用程序必须独立地决定队列中是否有足够的数据来尝试提取某种更高级别的应用程序消息。

例如，在您的情况下，您将检查队列的大小是否至少为 5，然后检查前五个字节，然后检查队列是否包含完整的应用程序消息。如果否，则中止；如果是，则提取整个消息并从队列的前面弹出。