Question

UDP客户端的原始代码如下所示：

while (getline(cin, buf)) {
        // c.getConnd() return connected fd
        write(c.getConnfd(), buf.c_str(), buf.size());
        memset(cbuf, 0, sizeof(cbuf));
        read(c.getConnfd(), cbuf, 1024);
        LOG(DEBUG) << "read message: " << cbuf;
    }

它工作正常，因此服务器（使用recvfrom和sendto）和套接字绑定都不会导致此问题。

为了学习IO multiplex，我编写了一个名为IOMultiplexingUtility的类，它允许用户将fd与相应的可调用对象组合在一起。设置它们后，它进入一个无限循环（找到一个可读的fd - 做一些动作 - 再次等待）：

class IOMultiplexingUtility {
public:
    IOMultiplexingUtility() {
        FD_ZERO(&socketSet);
    }

    void addFd(int fd, std::function<void(int)> action) {
        maxfd = fd + 1;
        FD_SET(fd, &socketSet);
        fdVec[fd] = action;
    }

    void start() {
        auto readySet = socketSet;
        select(maxfd + 1, &readySet, NULL, NULL, NULL);
        for (auto &pair: fdVec)
            if (FD_ISSET(pair.first, &readySet)) {
                if (pair.second != nullptr)pair.second(pair.first);
                else if (defaultAction) defaultAction(pair.first);
            }
    }

private:
    fd_set socketSet;
    int maxfd;
    std::unordered_map<int, std::function<void(int)>> fdVec;
};

修改后的UDP客户端代码如下：

io.addFd(c.getConnfd(), [&c](int)mutable -> void {
    char cbuf[1024];
    memset(cbuf, 0, sizeof(cbuf));
    read(c.getConnfd(), cbuf, 1024);
    cout << cbuf << endl;
});

io.addFd(fileno(stdin), [&c](int)mutable -> void {
    string buf;
    getline(cin, buf);
    write(c.getConnfd(), buf.c_str(), buf.size());
});

while (true) io.start();

发生错误（我输入文本，服务器成功接收和发送，但永远不会输出cbuf）我无法找到解决方法。任何想法都表示赞赏。

Answer 1

似乎是addFd中的逻辑缺陷。应该是类似maxfd = std::max(fd, maxfd)的内容，以及在构造函数中将maxfd初始化为类似INT_MIN的内容。

选择和UDP协议组合时无法接收数据

1 个答案: