140-网络编程：poll

1.3 poll

1.3.1、poll API介绍

poll将需要检测的文件描述符封装成一个结构体；

#include <poll.h>
struct pollfd {
	int fd; 		/* 委托内核检测的文件描述符 */
	short events; 	/* 委托内核检测文件描述符的什么事件 */
	short revents;	 /* 文件描述符实际发生的事件 */	--实际发生设么时间，内核就是通过这个给我们返回的
};

struct pollfd myfd;
myfd.fd = 5;
myfd.events = POLLIN | POLLOUT;	--检测文件描述符的读写事件

int poll(struct pollfd *fds, nfds_t nfds, int timeout);
    - 参数：
        - fds : 是一个struct pollfd 结构体数组，这是一个需要检测的文件描述符的集合
            - 这个fds是可以重用的, 比select好，内核只会修改revents，不会修改委托检测的events
            - 没有1024个标志位的限制
        
        - nfds : 这个是第一个参数数组中最后一个有效元素的下标 + 1
        - timeout : 阻塞时长
            0 : 不阻塞
            -1 : 阻塞，当检测到需要检测的文件描述符有变化，解除阻塞
            >0 : 阻塞的时长(单位: ms)
    - 返回值:
        -1 : 失败
        >0（n）: 成功,n表示检测到集合中有n个文件描述符发生变化

events和revents取值有一些对应的宏：

一般是用来检测读事件；

有没有数据发送过来 --POLLIN

检测读写事件：

struct pollfd myfd;
myfd.fd = 5;
myfd.events = POLLIN | POLLOUT;	--检测文件描述符的读写事件

1.3.2、poll 代码

//服务端
#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);	//第三个参数-1表示阻塞
        if(ret == -1) {
            perror("poll");
            exit(-1);
        } else if(ret == 0) {
            continue;
        } else if(ret > 0) {
            // 说明检测到了有文件描述符的对应的缓冲区的数据发生了改变
            if(fds[0].revents & POLLIN) {	//fds[0].revents & POLLIN	--位运算
                //fd[0]表示监听的文件描述符，检测有没有客户端连接进来
                // 表示有新的客户端连接进来了
                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;	//新连接让内核帮忙检测读事件POLLIN
                        break;
                    }
                }

                // 更新最大的文件描述符的索引
                nfds = nfds > cfd ? nfds : cfd;
            }

            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;
}

程序运行：

运行一个客户端：

再运行一个客户端：

服务端：都可以收到，没有问题。