linux 多路复用---poll

合集 - 网络编程(16)

1.linux socket地址2023-10-112.linux IP 地址转换2023-10-113.linux TCP 通信流程 套接字函数 socket() bind() listen() accept() connect()2023-10-124.linux TCP通信实现 服务器端 - 客户端2023-10-125.linux TCP 三次握手2023-10-126.linux TCP滑动窗口 四次挥手2023-10-227.linux 多进程实现并发服务器 多线程并发2023-10-248.linux TCP状态转换 半关闭 shutdown netstat 端口复用 setsockopt2023-10-249.linux IO多路复用 select poll epoll2023-10-25

10.linux 多路复用---poll2023-10-30

11.linux 多路复用---epoll2023-10-3112.linux UDP 通信2023-11-0313.linux 本地套接字通信2023-11-0314.linux Web服务器以及HTTP协议2023-11-0815.linux 服务器编程基本框架和两种高效事件的处理模式2023-11-1016.linux 线程池 EPOLLONESHOT事件2023-11-13

收起

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 结构体数组, 这是一个需要检测的文件描述符的集合
            - nfds: 这个是第一个参数数组中组后一个游戏元素的下标 + 1
            - timeout: 阻塞时长
                0: 不阻塞
                -1: 阻塞,当检测到需要检测的文件描述符有变化,解除阻塞
                >0: 阻塞的时长
        - 返回值:
            -1: 失败
            >0(n): 成功, n表示检测到集合中有n个文件描述符发生变化
*/

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

客户端不变