套接字编程，检查条目UDP缓冲区是否为空？

Question

我正在编写一个UDP客户端，它向服务器发送一个字符串，当服务器发回多个数据包时，程序的行为并不符合我的预期。我想通过 process() 一个接一个 处理任何传入数据包，直到条目缓冲区变空，但我认为存在与 的阻塞行为相关的问题接收。

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <winsock.h>

using namespace std;

void process(const char *in, int size)
{
    fprintf(stdout, "%s\n", in);
}

int main()
{
    char quack_addr[] = "127.0.0.1";
    unsigned short quack_port = 9091;

    WSAData data;
    WSAStartup(MAKEWORD(2, 2), &data);

    sockaddr_in qserver;
    qserver.sin_family = AF_INET;
    qserver.sin_addr.s_addr = inet_addr(quack_addr);
    qserver.sin_port = htons(quack_port);
    SOCKET client = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

    if (client <= 0)
    {
        fprintf(stderr, "Error - Can not create socket.\n");
        exit(1);
    }

    while (true)
    {
        const int MAX = 1024;
        char sbuf[MAX];
        char rbuf[MAX];

        fprintf(stdout, ": ");
        fgets(sbuf, MAX, stdin);
        int slen = strlen(sbuf);

        int r = sendto(client,sbuf,slen,0,(sockaddr*)&qserver,sizeof(qserver));

        // Current code:
        // int rlen = recv(client, rbuf, MAX, 0);
        // if (rlen > 0)
        // {
        //     rbuf[rlen] = 0;
        //     process(rbuf, rlen);
        // }

        // Question starts here:
        //
        // While there is data in queue do:
        // {
        //    (break if there is no data)
        //    int rlen = recv(client, rbuf, MAX, 0);
        //    rbuf[rlen] = 0;
        //    process(rbuf, rlen);
        // }   
    }

    return 0;
}

在调用 recv(...) 之前如何检查缓冲区是否为空？

问题发生在这种情况下：

1. 用户正在客户端程序中键入命令 (cmd1)。
2. 同时，服务器向客户端发送 3 个数据包（pkt1、pkt2、pkt3）。
3. 在客户端按 Enter 后，我期望收到这 3 个数据包，并且可能会收到与 cmd1 相对应的结果，以及 process() 所有他们一一。
4. 但在第 3 阶段按 Enter 后，我收到 pkt1！在向服务器发送另一个命令后，我将收到 pkt2 等等......！

我知道我的代码不足以处理这个问题，所以，我的问题是如何处理它？<​​/strong>

注意：我正在使用 netcat -L -p 9091 -u< /code> 作为 UDP 服务器

Answer 1

我认为问题（您未描述的令人不满意的行为）来自不同的来源。让我列出一些想法和评论c./之前所说的：

（1）recvfrom（）也块。但是，您想使用它。您的通信当前从环回发送和接收，这对于您的玩具程序来说很好（但是：请参见下文）。当使用recv()接收UDP数据时，你不知道是谁发送的，因为套接字从未被connect()ed。使用recvfrom() 为更严重的程序中的一些最小错误检查做好准备

(2)，因为 select() 会将程序暂停到 I/O 可用性，它只会将套接字阻塞的任何问题提升到不同的级别。但这不是问题

（3）要检查接收缓冲区是否为空，在recvfrom（）中的适当位置使用标志MSG_PEEK。它通常只用于处理稀缺内存，但它应该可以完成这项工作。

(4) 我相信您看到未更详细描述的问题的原因 1：
UDP 数据报保留消息边界。这意味着 recvfrom() 将读取构成发送的任何消息的整个数据块。但是，如果您读入的缓冲区小于读取的数据，则任何多余的内容都将被默默丢弃。因此，请确保您有一个大缓冲区（理想情况下为 65k）。

(5)原因2：
您会收到发送到环回的任何数据。如果您当前也连接到某个网络（卫星、互联网），那么您捕获的内容实际上可能来自与您预期不同的来源。所以至少在休息阶段，断开连接。

阻止应该不是问题。如果代码清晰，您的基本逻辑是：
Recvfrom()（阻塞/等待直到准备好）
过程
查看缓冲区是否为空
如果是则退出
如果没有，则返回以接收更多，

并且您当前似乎想要执行此操作。由于您不使用多线程、优化性能或类似功能，因此您不应该关心阻塞。如果您发现接收缓冲区太小，请使用

Setsockopt() for optName SO_RCVBUF增加其大小

Answer 2

使用 select()（具有适当的超时）在调用 recv() 之前检查传入数据。

类似以下内容（不可移植代码）

#include <winsock2.h>

...

/* somewhere after your sendto, or your first recv */
fd_set recv_set;
timeval tv = {1, 0}; /* one second */
FD_ZERO(&recv_set);
FD_SET(client, &recv_set);
while (select(0, &recv_set, NULL, NULL, &tv) > 0)
{
    /* recv... */
    FD_SET(client, &recv_set); /* actually redundant, since it is already set */
}

Answer 3

iPhone 有时会出现故障，不允许我发表评论。谢谢，史蒂夫。这只是继续谈话。

我认为这意味着“取消对‘问题从这里开始’的注释”。部分答案，因为这仍然取决于我的第二条评论；这或多或少比预期的要少。假设服务器要发送的三个消息已经排队，在您第一次按回车键后，您的数据包将被发送（永远不会被阻止，因为 sendto() 不会阻止 UDP），由服务器接收并且（我假设，见上面，回显并添加到 FIFO 接收缓冲区，其中您已经排队了三个消息，然后您的程序中有一个 recv() 接收第一个排队的消息，并将其打印出来。回到顶部循环，期待另一个输入并等待它（因此这不会在套接字级别上被阻止，而是当您的程序请求输入时，例如简单地“输入”），然后到达第二个最初发送的消息（由服务器）并处理再循环一次，所有三个都完成，并假设服务器回显您发送的内容，您应该开始接收您输入的消息（如果您只按回车，则当前循环可能为空）。除非你杀死它，否则不会退出。