时间戳

首页 > 解决方案 > 在 linux 中关闭 udp 套接字后，recv 不返回

c++ - 在 linux 中关闭 udp 套接字后，recv 不返回

问题描述

我正在尝试使用 udp 协议编写客户端服务器应用程序，但我遇到了连接结束问题。
我打开两个套接字（一个是“服务器”，另一个是“客户端”），当服务器从客户端异步接收时，客户端向他发送一条打印到控制台的简单消息。
经过一段时间的睡眠（以确保服务器将再次调用 recv），客户端和服务器套接字将关闭。在这一点上，我预计 recv 将返回 -1 并且异步将结束。
但实际发生的是，recv 永远卡住了*。
如果就在关闭套接字之前我发送了一个空包（在代码中将 sendToMakeExit 变量设置为 true），recv 将返回该空包，并且仅在下一次调用后才返回 -1，尽管套接字在第一次调用中已关闭。

    const bool sendToMakeExit = false;
    const int port = 2000;
    const auto addr = "127.0.0.1";
    int serverSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    sockaddr_in target;
    target.sin_family = AF_INET;
    target.sin_port = htons(port);
    inet_pton(AF_INET, addr, &target.sin_addr);
    bind(serverSocket, (sockaddr *) &target, sizeof(target));
    auto readAsync = std::async(std::launch::async, [&serverSocket] {
        const int MAX_READ = 4096;
        char readBuf[MAX_READ];
        ssize_t actualRead;
        do {
            actualRead = recv(serverSocket, readBuf, MAX_READ, 0);
            if (actualRead > 0) {
                cout << readBuf << endl;
            }
        } while (actualRead != -1);
    });
    int clientSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    connect(clientSocket, (sockaddr *) &target, sizeof(target));
    this_thread::sleep_for(chrono::seconds(1));
    send(clientSocket, "test", 5, 0);
    this_thread::sleep_for(chrono::seconds(1));
    close(clientSocket);
    if (sendToMakeExit) {
        sendto(serverSocket, nullptr, 0, 0, (sockaddr *) &target, sizeof(target));
    }
    close(serverSocket);

*如果我在调试中运行此代码并在recv 意外卡住时创建新断点，recv 返回-1。
当我关闭套接字时，如何让recv返回-1？

标签： c++linuxsockets

解决方案

关闭套接字并不能保证仍在使用该套接字的另一个线程中的任何函数调用立即返回。如果您有等待数据进入的调用，例如recv(), select(), poll()，您必须向套接字发送一些数据以使这些调用返回。您在代码中执行此操作，但在收到长度为零的 UDP 数据包时实际上并没有退出：将 while 循环的结尾更改为：

} while (actualRead > 0);

但是，我建议使用一个标志变量来指示线程是否应该继续运行，如下所示：

volatile bool running = true;

auto readAsync = std::async(std::launch::async, [&serverSocket, &running] {
    ...
    while (running) {
        ...recv()...
    }
});

...
running = false;
sendto(serverSocket, ...);
readAsync.wait();
close(serverSocket);

请注意，我在关闭套接字之前添加了一条等待readAsync完成的行，以防止发生任何意外：有一个小窗口，其中套接字无效，但readAsync可能仍会调用recv()它。如果你有更多的线程，也可能发生你关闭这个线程中的套接字，另一个线程打开一个新的套接字并获得与你刚刚关闭的相同的文件描述符编号，然后readAsync线程将使用错误的套接字。

推荐阅读