IO复用--poll系统调用

poll系统调用

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

#include<poll.h>
int poll(struct pollfd *fds,nfds_t nfds,int timeout);

1.fds参数是一个pollfd结构类型的数组，它指定所有我们感兴趣的文件描述符上发生的可读、可写和异常等事件，pollfd结构体的定义如下：

	struct pollfd 
	{
	   int   fd;         /* 文件描述符 */
	   short events;     /* 用户关注的事件 */
	   short revents;    /* 实际发生的事件，由内核返回 */
	};

其中，fd成员指定文件描述符，events告诉poll监听fd上的哪些事件，它是一系列事件的按位或；revents成员由内核修改，以通知应用程序上fd实际发生了那些时间，poll支持的可关注的事件类型如下：

事件	描述
POLLIN	数据可读（包括普通数据和优先数据）
POLLRDNORM	普通数据可读
POLLRDBAND	优先数据可读（linux不支持）
POLLPRI	紧急数据可读（高优先级数据可读，比如TCP带外数据）
POLLOUT	数据可写（包括普通数据和优先数据）
POLLWRNORM	普通数据可写
POLLWRBAND	优先数据可写
POLLRDHUP	TCP连接被对方关闭或对方关闭了写操作，由GNU引入
POLLERR	指定描述符发生错误
POLLHUP	指定文件描述符挂起事件（比如管道写端被关闭，读端将收到此事件）
POLLNVAL	指定描述符非法（文件描述符没有打开）

上表的POLLRDNORM、POLLRDBAND、POLLWRNORM、POLLWRBAND 由XOPEN规范定义。它们实际上是将POLLIN事件件和POLLOUT事件分得更细致，以区别对待普通数据和优先数据。但Linux并不完全支持它们。

通常，应用程序需要根据recv凋用的返回值来区分socket上接收到的是有效数据还是对方关闭连接的请求，并做相应的处理。

不过，自Linux内核2.6.17开始，GNU为poll系统凋用增加了一个POLLRDHUP事件，它在sockct上接收到对方关闭连接的请求之后触发。这为我们区分上述两种情况提供了一种更简单的方式。但使用POLLRDHUP事件时，我们需要在代码最开始处定义_GNU_SOURCE。

2.nfds参数指定被监听事件集合的大小，即数组的长度，元素的个数，用户关注的文件描述符的个数

3.timeout参数指定poll的超时时间，如果传入-1就永久阻塞。
返回值：-1表示出错，0表示超时，>0表示就绪文件描述符的个数

---

poll与select比较

1. 用户关注的事件不再是通过三个参数（可读、可写、异常）传递，即可关注的事件类型增多。
2. 内核修改的和用户关注的分开表示，每次调用前就不需要重新设置了。
3. 文件描述符不再是按位来表示，直接用int类型
   3.1 用户关注的文件描述符的值可以更大
   3.2 用户关注的文件描述符的个数由用户数组决定。

poll返回时，也是将用户关注的所有文件描述符返回。poll返回后，用户程序依旧需要循环检测哪些文件描述符就绪，所以poll检测就绪文件描述符的时间复杂度依然是O（n）。

---

使用poll实现多客户端与服务器通信

#define _GNU_SOURCE

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<assert.h>
#include<string.h>

#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<sys/types.h>
#include<poll.h>

#define SIZE 100

void Init_fds(struct pollfd *fds)
{
    int i=0;
    for(;i<SIZE;i++)
    {
        fds[i].fd=-1;
        fds[i].events=0; 
        fds[i].revents=0;   
    }
}

void Insert_fd(struct pollfd *fds,int fd,short event)
{
    int i=0;
    for(;i<SIZE;i++)
    {
        if(fds[i].fd==-1)
        {
            fds[i].fd=fd;
            fds[i].events=event;    
            break;
        }
    }
        
}

void Delete_fd(struct pollfd *fds,int fd)
{
    int i=0;
    for(;i<SIZE;i++)
    {
        if(fds[i].fd==fd)
        {
            fds[i].fd=-1;
            fds[i].events=0;
            break;    
        }
    }
}

int main()
{
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    assert(sockfd!=-1);
    
    struct sockaddr_in ser,cli;
    memset(&ser,0,sizeof(ser));
    ser.sin_family=AF_INET;
    ser.sin_port=htons(6000);
    ser.sin_addr.s_addr=inet_addr("127.0.0.1");
        
    int res=bind(sockfd,(struct sockaddr*)&ser,sizeof(ser));
    assert(res!=-1);
    
    listen(sockfd,5);
    
    struct pollfd fds[SIZE];
    Init_fds(fds);
    Insert_fd(fds,sockfd,POLLIN);

    while(1)
    {
        int n=poll(fds,SIZE,-1);
        if(n<=0)
        {
            printf("poll error\n");
            continue;    
        }
        
        int i=0;
        for(;i<SIZE;++i)
        {
            if(fds[i].fd!=-1)
            {
                int fd=fds[i].fd;
                
                if(fds[i].revents & POLLRDHUP)
                {
                    printf("%d will close\n",fd);
                    close(fd);
                    Delete_fd(fds,fd);           
                }
                else if(fds[i].revents & POLLIN)
                {
                    if(fd==sockfd)
                    {
                        int len=sizeof(cli);
                        int c=accept(fd,(struct sockaddr*)&cli,&len);
                        if(c<0)
                        {
                            continue;    
                        }
                        Insert_fd(fds,c,POLLIN | POLLRDHUP);
                    
                    }
                    else
                    {
                        char buff[128]={0};
                        recv(fd,buff,127,0);
                        printf("%d:%s\n",fd,buff);
                        send(fd,"ok",2,0);    
                    }    
                }
            }
        }
    }
}