Linux网络编程 | I/O复用之poll

---

前言

poll的原理以及使用。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、poll

poll 系统调用和 select 类似，也是在指定时间内轮询一定数量的文件描述符，以测试其中是否有就绪者。

二、API接口

int poll(struct pollfd fds, nfds_t nfds, int timeout)
/ poll 系统调用成功返回就绪文件描述符的总数，超时返回 0，失败返回-1
nfds 参数指定被监听事件集合 fds 的大小。 timeout 参数指定 poll 的超时值，单位是毫秒，timeout 为-1 时，poll 调用将永久阻塞，直到某个事件发生，timeout 为 0 时，poll 调用将立即返回。 fds 参数是一个 struct pollfd 结构类型的数组，它指定所有用户感兴趣的文件描述符上发生的可读、可写和异常等事件。pollfd 结构体定义如下：
struct pollfd
{
int fd; // 文件描述符
short events; // 注册的关注事件类型
short revents; // 实际发生的事件类型，由内核填充
};
其中，fd 成员指定文件描述符，events 成员告诉 poll 监听 fd 上的哪些事件类型。它是一系列事件的按位或，revents 成员则有内核修改，通知应用程序 fd 上实际发生了哪些事件。

poll支持的事件类型：

三、使用步骤

1.服务器端

代码如下（示例）：

#include<stdio.h>
#include<stdlib.h>
#include<assert.h>
#include<string.h>
#include<unistd.h>
#include<sys/time.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<poll.h>
#define MAX 10


void Init_fds(struct pollfd*fds)//初始化文件描述符数组
{
    assert(fds!=NULL);
    for(int i=0;i<MAX;i++)
    {
        fds[i].fd=-1;
        fds[i].events=0;
        fds[i].revents=0;
    }   
}

void Insert_fds(struct pollfd*fds,int x)//插入文件描述符
{
    assert(fds!=NULL);
    for(int i=0;i<MAX;i++)
    {
        if(fds[i].fd==-1)
        {
            fds[i].fd=x;
            fds[i].events=POLLIN;
            fds[i].revents=0;
            break;
        }
    }
}

void Del_fds(struct pollfd*fds,int x)//删除文件描述符
{
    assert(fds!=NULL);
    for(int i=0;i<MAX;i++)
    {
        if(fds[i].fd==x)
        {
            fds[i].fd=-1;
            fds[i].events=0;
            fds[i].revents=0;
            break;
        }
    }
}


int main()
{
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    assert(sockfd!=-1);

    struct sockaddr_in saddr,caddr;
    memset(&saddr,0,sizeof(saddr));
    saddr.sin_family=AF_INET;
    saddr.sin_port=htons(6000);
    saddr.sin_addr.s_addr=inet_addr("127.0.0.1");

    int res=bind(sockfd,(struct sockaddr*)&saddr,sizeof(saddr));
    assert(res!=-1);

    res=listen(sockfd,5);
    assert(res!=-1);
    
    struct pollfd fds[MAX]={0};

    Init_fds(fds);
    Insert_fds(fds,sockfd);

    while(1)
    {
        int n=poll(fds,MAX,1000);//出错
        if(-1==n)
        {
            continue;
        }
        if(0==n)//超时
        {
            printf("time out\n");
            continue;
        }
        else//有读事件产生
        {
            for(int i=0;i<MAX;i++)//遍历文件描述符数组
            {
                if(fds[i].fd==-1)
                {
                    continue;
                }
                if(fds[i].revents&POLLIN)//如果有读事件
                {
                    if(fds[i].fd==sockfd)//读事件是链接套接字
                    {
                        struct sockaddr_in caddr;
                        int len=sizeof(caddr);
                        int c=accept(sockfd,(struct sockaddr*)&caddr,&len);//调用accept处理
                        if(c<0)
                        {
                            continue;
                        }
                        else
                        {
                            printf("accept=%d\n",c);//链接成功
                            Insert_fds(fds,c);//将产生的c插入文件描述符数组中，准备下一轮的监听
                        }
                    }
                    else//如果是c上的读事件
                    {
                        char buff[128]={0};
                        int num=recv(fds[i].fd,buff,127,0);//调用recv接收数据
                        if(num<=0)
                        {
                            close(fds[i].fd);
                            Del_fds(fds,fds[i].fd);//断开后在数组中删除该问价描述符
                            printf("one client over\n");
                            continue;
                        }
                        else
                        {
                            printf("buff(%d)=%s\n",fds[i].fd,buff);
                            send(fds[i].fd,"ok",2,0);
                        }
                    }
                }
            }
        }
    }
}

2.客户端

代码如下（示例）：

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


int main()
{
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    assert(sockfd!=-1);

    struct sockaddr_in saddr;
    memset(&saddr,0,sizeof(saddr));
    saddr.sin_family=AF_INET;
    saddr.sin_port=htons(6000);
    saddr.sin_addr.s_addr=inet_addr("127.0.0.1");

    int res=connect(sockfd,(struct sockaddr*)&saddr,sizeof(saddr));
    assert(res!=-1);
    char buff[128]={0};
    printf("input: \n");
    while(1)
    {
        fgets(buff,128,stdin);
        if(strncmp(buff,"end",3)==0)
        {
            break;
        }
        send(sockfd,buff,strlen(buff),0);
        memset(&buff,0,sizeof(buff));
        recv(sockfd,buff,127,0);
        printf("buff=%s\n",buff);
    }

    close(sockfd);
    exit(0);
}

---

总结

poll与select的功能相似，但是可以检测的数组的大小可以超过1024个
poll能检测的时间比select多
poll监测数组的每一个元素是一个结构体。