IO复用：poll函数

   poll提供的功能与select函数类似，不过在处理流设备时，它能够提供额外的信息 。

   #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成员中返回该描述符的状态。在下表中给出一些常值：

    我们将该图分为三个部分：第一个部分是处理输入的四个常值，第二个部分是处理输出的三个常值，第三个部分是处理错误的三个常值。其中，第三部分的三个常值不能在events中设置，但是当相应条件存在时就在revents中返回。

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

  INFTIM常值定义为一个负值，其定义在头文件#include<poll.h>，不过很多系统把它定义在头文件#include<sys/stropts.h>中。

   当发生错误时，poll函数的返回值为-1，若定时器到时之前没有任何描述符就绪，则返回0，否则返回就绪描述符的个数，即revents成员值非0的描述符个数。

   下面给出一个用poll实现的例子，其功能和之前的博客对于select函数实现的功能一样，客户端给服务器发送消息，服务器将消息回射到客户端。。。

服务器：

#include"../util.h"

int sock_bind(const char *ip,short port)
{
    int sockserfd;
    //创建套接字
    sockserfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockserfd == -1)
        perror("socket");
    struct sockaddr_in addrSer;
    addrSer.sin_family = AF_INET;
    addrSer.sin_port = htons(port);
    addrSer.sin_addr.s_addr = inet_addr(ip);

    int res;
    //绑定
    res = bind(sockserfd,(struct sockaddr *)&addrSer,sizeof(struct sockaddr));

    return sockserfd;
}

void handle_connection(struct pollfd *clientfds,int maxi)
{
    int n;
    char buf[256];
    for(int i = 1; i <= maxi; ++i)
    {
        if(clientfds[i].fd == -1)
            continue;
        //当客户端的描述符状态为可读时
        if(clientfds[i].revents & POLLIN)
        {
            //接收客户端的数据
            n = recv(clientfds[i].fd,buf,256,0);
            //当n小于0时说明客户端关闭
            if(n <= 0)
            {
                close(clientfds[i].fd);
                clientfds[i].fd = -1;
                continue;
            }
            printf("recv msg:>%s\n",buf);
            //将接收到的数据又进行发送
            send(clientfds[i].fd,buf,n,0);
                
        }
            
    }

}
void do_poll(int sockfd)
{
    pollfd clientfds[OPEN_SIZE];
    //0下标为服务器
    //将服务器的测试条件设置为可读
    clientfds[0].fd = sockfd;
    clientfds[0].events = POLLIN;

    //初始化描述符
    for(int i = 1; i < OPEN_SIZE; ++i)
        clientfds[i].fd = -1;

    int maxi = 0;
    int nready;
    int connfd;
    int i = 0;
    struct sockaddr_in addrCli;
    for(;;)
    {
        //nready表示就绪的描述符
        nready = poll(clientfds,maxi + 1,-1);
        if(nready == -1)
        {
            perror("poll");
            exit(1);
        }
        //当服务器的返回事件为可读
        if(clientfds[0].revents & POLLIN)
        {
            //新客户端建立连接
            socklen_t addrlen = sizeof(struct sockaddr);
            connfd = accept(sockfd,(struct sockaddr *)&addrCli,&addrlen);
               
            if(connfd == -1)
            {
                perror("accept");
                continue;
            }
            printf("accept a new client :%s:%d\n",inet_ntoa(addrCli.sin_addr),addrCli.sin_port);
            for(i = 1 ; i < OPEN_SIZE; ++i)
            {
                if(clientfds[i].fd < 0)
                {
                    clientfds[i].fd = connfd;
                    break;
                }
            }
            if(i == OPEN_SIZE)
            {
                printf("too many clients");
                exit(1);
            }
            //将客户端的测试条件设置为可读
            clientfds[i].events = POLLIN;
            if(i > maxi)
                maxi = i;
            if(--nready <= 0)
                continue;
        }
        handle_connection(clientfds,maxi);
    
    }
}
int main(int argc,char*argv[])
{
    int sockSer;
    sockSer = sock_bind(IPADDR,PORT);
    listen(sockSer,LISTENQ);
    do_poll(sockSer);
    return 0;
}

客户端：

#include"../util.h"

void handle_connection(int sockfd)
{
    pollfd fds[2];
    fds[0].fd = sockfd;
    fds[0].events = POLLIN;
    fds[1].fd = STDIN_FILENO;
    fds[1].events = POLLIN;

    int n;
    char buf[256];
    for(;;)
    {
        poll(fds,2,-1);
        //当服务器可读时
        if(fds[0].revents & POLLIN)
        {
            //接收服务器的消息
            n = recv(sockfd,buf,256,0);
            //如果n等于0，说明服务器关闭
            if(n <= 0)
            {
                printf("server is closed.\n");
                close(sockfd);
            }
            //将buf的内容显示在屏幕上
            write(STDIN_FILENO,buf,n);
        }
        //当标准输入输出可读时
        if(fds[1].revents & POLLIN)
        {
            //将屏幕的内容读到buf中
            n = read(STDIN_FILENO,buf,256);
            //当n等于0时，继续循环
            if(n == 0)
                continue;
            //向服务器发送数据
            write(sockfd,buf,n);
        }
    }
}

int main(int argc,char*argv[])
{
    int sockCli;
    //创建套接字
    sockCli = socket(AF_INET,SOCK_STREAM,0);
    if(sockCli == -1)
        perror("sockCli");

    struct sockaddr_in addrSer;
    socklen_t addrlen;
    addrSer.sin_family = AF_INET;
    addrSer.sin_port = htons(PORT);
    addrSer.sin_addr.s_addr = inet_addr(IPADDR);

    addrlen = sizeof(struct sockaddr);
    //客户端进行连接
    connect(sockCli,(struct sockaddr *)&addrSer,addrlen);

    handle_connection(sockCli);
    return 0;
}

公用头文件：

#pragma once
#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<poll.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<string.h>

#define IPADDR "127.0.0.1"
#define PORT 8787
#define MAXLINE 1024
#define LISTENQ 5

#define SIZE 10
#define OPEN_SIZE 5

程序的执行结果和利用select函数的执行结果是一样的。。。