Linux程序设计之IO多路复用(API:poll)

1.使用select实现IO多路复用时存在一些缺点，比如select支持的文件描述符数量太少了，默认是1024、fds集合不能重用，因为在内核进行检测时，会清空之前设置好的1,所有每次都需要重置等等。poll的原理和select是类似的，都是让内核帮忙检测是否有文件描述符对应的进程发生了数据变化(读、写)，如果发生了变化则执行相应操作。

2.poll API介绍(图片来源于课程：登录—专业IT笔试面试备考平台_牛客网)：

 

 

 3.代码：

/*IO多路复用之poll*/
//利用IO多路复用实现一个服务器可被多个客户端连接访问

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

int main()
{
    //1.创建用于监听的socket
    int lfd = socket(AF_INET,SOCK_STREAM,0);
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;
    saddr.sin_port = htons(9999);

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

    //3.监听
    listen(lfd,128);

    struct sockaddr_in cliaddr;
    int len = sizeof(cliaddr);

    //初始化检测的文件描述符数组
    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("select");
            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);  //阻塞在此处，直到有客户端连接上来
                if(cfd == -1)   //增加这些错误的判断非常重要，可以帮助找到出现问题的地方
                {
                    perror("accept");
                    exit(-1);
                }

                //将新的文件描述符加入到集合中
                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;   //在fds中清空对应的文件描述符
                    }
                    else if(len > 0)
                    {
                        printf("read buf = %s\n",buf);
                        write(fds[i].fd,buf,strlen(buf)+1);
                    }
                }
            }
        }
    }
    close(lfd);

    return 0;
}

/*TCP/IP通信之客户端*/
#include <stdio.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>

int main()
{
    //1.创建socket套接字
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1)
    {
        perror("socket");
        exit(-1);
    }

    //2.与服务器进行连接
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    //主机字节序转为网络字节序
    inet_pton(AF_INET,"127.0.0.1",&serveraddr.sin_addr.s_addr);
    serveraddr.sin_port = htons(9999);  //客户端去访问服务端的9999端口，服务器端口是固定的，而客户端端口是随机的
    int ret = connect(fd,(struct sockaddr *)&serveraddr,sizeof(serveraddr));
    if(ret == -1)
    {
        perror("connect");
        exit(-1);
    }
    
    char buf[1024];
    int i = 0;
    while(1)
    {
        //3.通信
        //给服务器端发送数据
        // char buf[] = "hello,i am client";
        sprintf(buf,"data : %d\n",i++);
        // printf("test");
        // memset(buf,0,strlen(buf));
        // scanf("%s",buf); //scanf("%s")无法输入空格，只能获取空格和\n之前的数据
        // gets(buf);  //会产生警告
        ret = write(fd,buf,strlen(buf)+1);
        if(ret == -1)
        {
            perror("write");
            exit(-1);
        }

        //获取服务器端的数据
        char recebuf[1024] = {0};
        int len = read(fd,recebuf,sizeof(recebuf));
        if(len == -1)
        {
            perror("read");
            exit(-1);
        }
        else if(len > 0)
        {
            printf("receive server data : %s\n",recebuf);
        }
        else if(len == 0)
        {
            //服务器端断开连接
            printf("server close!!!\n");
            break;
        }
        sleep(1);
    }

    //4.关闭文件描述符
    close(fd);

    return 0;

}

4.运行结果：

5.总结 ：poll机制没有文件描述符数量的限制，并且将文件描述符的信息封装成一个结构体，记录了文件描述符、需要检测的状态和实际发生的状态，为了系统提供了更多的信息，同时，使得fds(文件描述符记录数组)可以重复使用。