7、I/O多路复用之poll：poll函数详解、poll与select对比、poll事件类型、实战：基于poll的聊天服务器

好，我们继续往下走。上一章我们把select扒了个底朝天，它的优点和槽点大家心里都有数了。今天聊的poll，说白了就是select的“改良版”。我当年刚接触网络编程时，先学的select，后来切换到poll，第一感觉就是——嗯，终于不用再数那个1024了。

7.1 poll函数详解

poll的核心数据结构是struct pollfd。它比select的fd_set要直观得多。你想想看，select需要三个位图，还要反复FD_SET、FD_CLR，代码写起来像在拼积木。poll直接给你一个数组，每个元素管一个fd，清晰明了。

#include <poll.h>

struct pollfd {
    int   fd;         // 要监视的文件描述符
    short events;     // 感兴趣的事件（输入参数）
    short revents;    // 实际发生的事件（输出参数）
};

int poll(struct pollfd *fds, nfds_t nfds, int timeout);

参数解释一下：

  • fds：pollfd数组的首地址。你可以动态分配，也可以静态声明。我个人习惯用vector动态管理，方便扩容。
  • nfds：数组中有多少个有效元素。注意不是最大fd值+1，而是实际监视的fd数量。这一点比select省心多了。
  • timeout：超时时间，单位毫秒。传-1表示阻塞等待，传0表示立即返回（轮询模式）。

返回值：成功时返回就绪的fd数量；超时返回0；出错返回-1。

核心区别：poll不需要每次调用前重新构造监视集合。你只需要维护好pollfd数组，events字段填好，poll返回后检查revents即可。select每次都要重新FD_ZERO、FD_SET，那叫一个麻烦。

7.2 poll事件类型

poll支持的事件类型比select丰富。select只能判断“可读、可写、异常”，poll则细分了很多。我列个表，大家一目了然：

事件宏 含义 使用场景
POLLIN 有数据可读 普通TCP连接有数据到达
POLLPRI 有紧急数据可读 带外数据（OOB）
POLLOUT 可写 发送缓冲区有空闲空间
POLLERR 发生错误 连接异常断开
POLLHUP 对方关闭连接 收到FIN包
POLLNVAL fd未打开 fd已经关闭或无效

注意，POLLERR、POLLHUP、POLLNVAL这三个事件是自动监视的，你不需要在events中设置。它们会直接出现在revents里。我曾经踩过一个坑：以为POLLHUP需要手动注册，结果漏掉了，导致对方断开后我这边一直傻等——嗯，后来查文档才明白。

7.3 poll与select对比

直接上对比表，这样最清楚：

对比维度 select poll
数据结构 fd_set位图（固定大小） pollfd数组（动态大小）
最大fd数 通常1024（受FD_SETSIZE限制） 无上限（受系统内存限制）
事件注册方式 每次调用前重新设置位图 维护数组，events字段不变
就绪fd遍历 遍历整个位图（0~maxfd） 遍历数组（0~nfds-1）
事件类型 读、写、异常三种 POLLIN/OUT/ERR/HUP等
可移植性 几乎所有平台 POSIX标准，Linux/Unix
性能（大量fd） O(maxfd)，效率低 O(nfds)，相对好一些

说白了，poll就是select的“去痛点”升级版。它解决了最大fd限制的问题，也避免了每次重新构造监视集合的麻烦。但poll也不是完美的——它仍然需要遍历整个数组来查找就绪的fd。当连接数达到几万时，这个遍历开销就很可观了。这也是为什么后来出现了epoll。

我的建议：如果你的连接数在几百以内，用select或poll差别不大。但如果连接数上千，或者你需要处理大量短连接，建议直接上epoll。poll更适合中等规模的场景，比如几百个长连接。

7.4 实战：基于poll的聊天服务器

好，理论说完了，我们来写一个真正的聊天服务器。这个服务器支持多个客户端同时连接，任何一个客户端发消息，服务器都会广播给所有其他客户端。

先画个图，看看整体流程：

基于poll的聊天服务器架构 poll() 主循环 遍历 pollfd 数组 监听fd就绪 accept() 新连接 加入pollfd数组 客户端fd就绪 recv() 读取数据 广播给其他客户端

核心代码实现如下。我尽量写得简洁，但保留了关键逻辑：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#include <vector>

#define PORT 8888
#define MAX_CLIENTS 1024

int main() {
    int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (listen_fd < 0) {
        perror("socket");
        exit(1);
    }

    // 设置端口复用，避免服务器重启时"Address already in use"
    int opt = 1;
    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    struct sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = INADDR_ANY;

    if (bind(listen_fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
        perror("bind");
        exit(1);
    }

    if (listen(listen_fd, 5) < 0) {
        perror("listen");
        exit(1);
    }

    printf("聊天服务器启动，监听端口 %d\n", PORT);

    // 初始化pollfd数组
    std::vector<struct pollfd> fds;
    fds.push_back({listen_fd, POLLIN, 0});

    while (1) {
        int ret = poll(fds.data(), fds.size(), -1);
        if (ret < 0) {
            perror("poll");
            break;
        }

        // 遍历所有fd，检查就绪状态
        for (size_t i = 0; i < fds.size(); i++) {
            if (fds[i].revents & POLLIN) {
                if (fds[i].fd == listen_fd) {
                    // 新连接到来
                    struct sockaddr_in client_addr;
                    socklen_t len = sizeof(client_addr);
                    int conn_fd = accept(listen_fd, 
                        (struct sockaddr*)&client_addr, &len);
                    if (conn_fd < 0) {
                        perror("accept");
                        continue;
                    }
                    printf("新客户端连接: %s:%d\n",
                        inet_ntoa(client_addr.sin_addr),
                        ntohs(client_addr.sin_port));

                    fds.push_back({conn_fd, POLLIN, 0});
                } else {
                    // 客户端有数据
                    char buf[1024];
                    int n = recv(fds[i].fd, buf, sizeof(buf) - 1, 0);
                    if (n <= 0) {
                        // 连接关闭或出错
                        printf("客户端断开连接\n");
                        close(fds[i].fd);
                        fds.erase(fds.begin() + i);
                        i--;  // 注意调整索引
                    } else {
                        buf[n] = '\0';
                        printf("收到消息: %s\n", buf);

                        // 广播给其他所有客户端
                        for (size_t j = 0; j < fds.size(); j++) {
                            if (fds[j].fd != listen_fd && 
                                fds[j].fd != fds[i].fd) {
                                send(fds[j].fd, buf, n, 0);
                            }
                        }
                    }
                }
            }
        }
    }

    close(listen_fd);
    return 0;
}

注意：上面的代码在遍历过程中删除了元素（fds.erase），所以必须调整索引i--，否则会漏掉元素或越界。我曾经在早期写类似代码时忘了这一步，结果客户端断开后服务器行为变得诡异——排查了半天才发现是索引问题。

这个聊天服务器的逻辑其实很简单：

  1. 监听fd就绪时，调用accept接受新连接，把新fd加入pollfd数组。
  2. 客户端fd就绪时，调用recv读取数据。如果返回0或-1，说明对方断开，关闭fd并从数组中移除。
  3. 如果读到数据，就遍历所有客户端fd（除了自己），把消息广播出去。

你可以用telnet或者nc来测试：

# 终端1：启动服务器
./chat_server

# 终端2：连接客户端1
telnet 127.0.0.1 8888

# 终端3：连接客户端2
telnet 127.0.0.1 8888

# 在客户端1输入消息，客户端2应该能收到

一个小优化：上面的代码每次广播时都遍历整个fds数组。如果客户端数量很多（比如上千），这个遍历开销会比较大。你可以考虑维护一个单独的“客户端fd列表”，只存客户端fd，不包含监听fd。这样广播时只遍历客户端列表，效率更高。

好了，poll的内容就讲到这里。它比select好用，但还不是终点。下一章我们会聊epoll——Linux下高性能网络编程的终极武器。不过在那之前，建议你把上面的代码亲手敲一遍，跑一跑，感受一下poll的工作方式。

公众号:蓝海资料掘金营，微信deep3321
← 上一章 📖 返回目录 下一章 →