poll系统调用

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

#include<poll.h>

int poll(struct pollfd* fds, nfds_t nfds, int timeout);

timeout参数指定poll的超时值，单位是毫秒。当timeout为-1时，poll调用将永远阻塞，直到某个事件发生；当timeout为0时，poll调用将立即返回。

注：poll系统调用的返回值的含义与select相同，不同的是，poll系统调用通过一个结构体数组来指定关心的描述符，另外对于每一个描述符可以分别指定触发的事件，这些是通过struct pollfd结构体来实现的。

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

             {  int fd;   //指定文件描述符   short events;//关心的事件   short revents; //实际发生的事件，由内核来填充 }

其中events成员告诉poll监听fd上的哪些事件，它是一系列事件的按位或；  revents成员由内核修改，以通知应用程序fd实际发生了哪些事件。poll支持的事件类型如下表：

2、代码实现：

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

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

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

void fds_del(struct pollfd fds[], int fd)
{
    int i = 0;
    for(; i < MAXFD; 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 saddr, caddr;
    memset(&saddr,0,sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(6000);   //<1024 root 
    saddr.sin_addr.s_addr = inet_addr("127.0.0.1");

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

    listen(sockfd,5);

    struct pollfd fds[MAXFD];  //定义一个结构体数组
    fds_init(fds);//初始化数组
    
    fds_add(fds, sockfd);//把描述符添加到数组中
    while(1)
    {
        int n = poll(fds,MAXFD,5000);
        if(n == -1)
        {
            printf("poll error\n");
            continue;
        }
        else if(n == 0)
        {
            printf("time out\n");
            continue;
        }
        else
        {
            int i = 0;
            for(; i < MAXFD; i++)
            {
                if(fds[i].fd == -1)
                {
                    continue;
                }
                if(fds[i].revents & POLLIN) //判断描述符上有没有事件，前一个是系统返回给我们的
                {
                  if(fds[i].fd == sockfd)
                    {
                        int len = sizeof(caddr);
                        int c = accept(fds[i].fd,(struct sockaddr*)&caddr,&len);

                        if(c < 0)
                        {
                            continue;
                        }
                        printf("accept c = %d\n",c);
                        fds_add(fds,c);
                    }
                    else
                    {
                        char buff[128] = {0};
                        int num = recv(fds[i].fd, buff, 127,0);
                        if(num <= 0)
                        {
                            close(fds[i].fd);
                            fds_del(fds,fds[i].fd);
                            printf("one client close\n");
                        }
                        else
                        {
                            printf("recv(%d) = %s\n",fds[i].fd,buff);
                            send(fds[i].fd,"ok",2,0);
                        }
                    }
                }
                //if(fds[i].revents & POLLOUT)
            }
        }
    }
}