poll

2020-09-12

使用poll函数实现I/O复用. poll允许工作在任何描述符中，poll提供的功能与select函数类似。

函数原型如下：

1 2	#include <poll.h> int poll(struct pollfd *fdarray,unsigned long nfds,int timeout);

返回：若有就绪描述符则为其数目，若超时返回0，出错返回-1

第一个参数是指向一个结构体数组第一个元素的指针。每个元素都是一个pollfd结构，用于指定测试某个给定描述符fd的条件

struct pollfd{
 int fd;                  //需要检测的文件描述符
 short events;            //请求的事件 
 short revents;           //返回的事件
};

下表说明了能够作为events和revents的常量

结构体数组中元素的个数是由nfds参数指定。

timeout 参数指定poll函数返回前等待多长时间。它是一个指定应等待毫秒数的正值。取值如下表：

timeout值说明-1永远等待，直到有描述符就绪0立即返回，不阻塞进程>0等待指定的毫秒数

使用poll函数建立的服务器端如下：

//
// Created by silver on 2020/8/23.
//
#include <stdio.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/unistd.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <errno.h>
#include <string.h>
#include <sys/wait.h>
#include <sys/select.h>
#include <poll.h>
#include <sys/stropts.h>
#define MAXLINE 4096
#define PORT 9873
#define LISTQUE 1024
#define OPEN_MAX 256
int main(int argc,char* argv[])
{
    int listenfd,confd,sockfd;
    struct pollfd client[OPEN_MAX];
    int nready;
    int maxcount = -1;
    int count = 0;
    char buf[MAXLINE];
    struct sockaddr_in SerAddr,Cliaddr;
    socklen_t Clilen = sizeof(Cliaddr);
    //initializer
    bzero(&SerAddr,sizeof(SerAddr));
    bzero(&Cliaddr,Clilen);
    bzero(buf,MAXLINE);
    listenfd = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
    SerAddr.sin_family = AF_INET;
    SerAddr.sin_port = htons(PORT);
    SerAddr.sin_addr.s_addr = htonl(INADDR_ANY);

    if (bind(listenfd,(struct sockaddr*)&SerAddr,sizeof(SerAddr)) < 0)
    {
        perror("Bind error");
    }
    if (listen(listenfd,LISTQUE) < 0)
    {
        perror("listen error");
    }
    client[0].fd = listenfd;
    client[0].events = POLLRDNORM;
    for (count = 1; count < OPEN_MAX; count++) {
        client[count].fd = -1;
    }
    maxcount = 0;
    ssize_t  n;
    fputs("server waiting....",stdout);
    fflush(stdout);
    while(true)
    {
        nready = poll(client,maxcount +1,-1);

        if (client[0].revents & POLLRDNORM) // new client connection
        {
            confd = accept(listenfd,(struct sockaddr*)&Cliaddr,&Clilen);
            printf("client address: %s\n",inet_ntoa(Cliaddr.sin_addr));
            if (confd < 0){ perror("accept error");}

            for (count = 1;  count<OPEN_MAX ; count++)
            {
                if (client[count].fd < 0)
                {
                    client[count].fd = confd;
                    break;
                }
            }
            if (count == OPEN_MAX) {printf("too many clients");exit(-1);}

            client[count].events = POLLRDNORM;
            if (count > maxcount) maxcount = count;
            if (--nready <= 0) continue; //no more readable descriptions
        }
        for (count = 1; count <=maxcount ; count++)
        {
            if ((sockfd = client[maxcount].fd) < 0){ continue;}
            if(client[count].revents & (POLLRDNORM | POLLERR))
            {
                n = read(sockfd,buf,MAXLINE);
                if (n < 0) {
                    if (errno == ECONNRESET) {
                        close(sockfd);
                        client[count].fd = -1;
                    } else {
                        printf("read error");
                        exit(-1);
                    }
                } else if (n == 0){
                    close(sockfd);
                    client[count].fd = -1;
                } else{
                    writen(sockfd,buf,n);
                }
            }
            if (--nready <= 0) break; //no more readable descriptions
        }
    }
    close(listenfd);
    return 0;
}