<aside> 💡 poll 是对 select 的改进

</aside>

select的缺点

poll函数

<aside> 💡 poll将需要检测的文件描述符封装成了一个结构体pollfd，解决了上述的缺点3、4

</aside>

参数

• fds：传入传出参数，是一个结构体数组，存放需要检测的文件描述符集合
  • pollfd
    • fd：委托内核检测的文件描述符
    • events：委托内核检测的事件
      • 常用：
        • POLLIN：有数据可读。
        • POLLOUT：写操作不会阻塞。
        • POLLERR：发生错误。
        • POLLHUP：挂起事件。
        • POLLNVAL：无效请求（指定的文件描述符没有打开）
      • 可以同时检测多个事件：POLLIN | POLLOUT
    • revents：一个位掩码，由内核返回，记录文件描述符实际发生的事件
• nfds：fds数组中最后一个有效元素的下标 + 1
• timeout：阻塞时长，单位毫秒
  • 0：不阻塞
  • -1：阻塞，只有当文件描述符发生变化才返回

  • 0：阻塞时长

返回值

• 返回 -1：表示调用失败
• 返回一个大于0的值 n ：表示检测集合中有 n 个文件描述符发生了变化

示例代码

#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <poll.h>

int main() {

    // 创建socket
    int lfd = socket(PF_INET, SOCK_STREAM, 0);
    struct sockaddr_in saddr;
    saddr.sin_port = htons(9999);
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;

    // 绑定
    bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr));

    // 监听
    listen(lfd, 8);

    // 初始化检测的文件描述符数组
    struct pollfd fds[1024];
    for(int i = 0; i < 1024; i++) {
        fds[i].fd = -1;
        fds[i].events = POLLIN;
    }
    fds[0].fd = lfd;
    int nfds = 0;

    while(1) {

        // 调用poll系统函数，让内核帮检测哪些文件描述符有数据
        int ret = poll(fds, nfds + 1, -1);
        if(ret == -1) {
            perror("poll");
            exit(-1);
        } else if(ret == 0) {
            continue;
        } else if(ret > 0) {
            // 说明检测到了有文件描述符的对应的缓冲区的数据发生了改变
            if(fds[0].revents & POLLIN) {
                // 表示有新的客户端连接进来了
                struct sockaddr_in cliaddr;
                int len = sizeof(cliaddr);
                int cfd = accept(lfd, (struct sockaddr *)&cliaddr, &len);

                // 将新的文件描述符加入到集合中
                for(int i = 1; i < 1024; i++) {
                    if(fds[i].fd == -1) {
                        fds[i].fd = cfd;
                        fds[i].events = POLLIN;
                        break;
                    }
                }

                // 更新最大的文件描述符的索引
                nfds = nfds > cfd ? nfds : cfd;
            }
						
						// 轮询
						// fds[0]为监听的文件描述符，所以从fds[1]开始遍历
            for(int i = 1; i <= nfds; i++) {
		            // 使用 按位与运算符(&) 判断是否发生事件（对位掩码进行位与操作）
                if(fds[i].revents & POLLIN) {
                    // 说明这个文件描述符对应的客户端发来了数据
                    char buf[1024] = {0};
                    int len = read(fds[i].fd, buf, sizeof(buf));
                    if(len == -1) {
                        perror("read");
                        exit(-1);
                    } else if(len == 0) {
                        printf("client closed...\\n");
                        close(fds[i].fd);
                        fds[i].fd = -1;
                    } else if(len > 0) {
                        printf("read buf = %s\\n", buf);
                        write(fds[i].fd, buf, strlen(buf) + 1);
                    }
                }
            }

        }

    }
    close(lfd);
    return 0;
}