linux/unix网络编程之 poll

转自http://www.cnblogs.com/zhuwbox/p/4222382.html

poll 与 select 很类似，都是对描述符进行遍历，查看是否有描述符就绪。如果有就返回就绪文件描述符的个数将。poll 函数如下：

　　#include <poll.h>

　　int poll(struct pollfd *fdarray, unsigned long nfds, int timeout)

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

　　struct pollfd

　　{

　　　　int fd;

　　　　short events;//关心 fd 上发生的事件

　　　　short revents;//fd 实际上上发生的事件

　　}

　　要测试的条件由 events 成员指定，函数在相应的 revents 成员中返回该描述符的状态(每个描述符都有两个变量，一个为调用值，另一个为返回结果，从而避免值-结果参数)这两个成员中的每一个都由指定某个特定条件的一位或多位组合而成。下标列出指定 events 标志以及测试 revents 标志的一些常值。

　　----------------------------+---------------------------------------+---------------------------------------+----------------------------------------+

　　　 　常量　　　　　　　  |  能做为 events 的输入吗？    |　能作为revents 的结果吗？ |　说明　　　　　　　　　　   |

　　----------------------------+---------------------------------------+---------------------------------------+----------------------------------------+

　　       POLLIN　　　　　　|　　　　能　　　　　　　　 | 　　　　　　能　　　　　　|  普通或者优先级带数据可读    |

　       　POLLRDNORM　　  |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　|  普通数据可读　　　　　　    |

　　  　POLLRDBAND          |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　|  优先级带数据可读　　　　    |

　       　POLLPRI　　　　　 |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　|  高优先级数据可读　　　　　 |

　　----------------------------+---------------------------------------+---------------------------------------+----------------------------------------+

　       　POLLOUT　　　　   |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　| 普通数据可写　　　　　　     |

　       　POLLWRNORM　　 |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　| 普通数据可写　　　　　　　 |

      　　POLLWRBAND　　  |　　　　能　　　　　　　　 |　　　　　　 能　　　　　　| 优先级带数据可写　　　　　 |

　　----------------------------+---------------------------------------+--------------------------------------+-----------------------------------------+

　       　POLLERR　　　　   |　　　　　　　　　　　　　 |　　　　　　 能　　　　　　| 发生错误　　　　　　　　　 |

　       　POLLHUP　　　　   |　　　　　　　　　　　　　 |　　　　　　 能　　　　　　| 发生挂起　　　　　　　　　 |

    　　POLLNVAL                |　　　　　　　　　　　　　 |　　　　　    能　　　　　　| 描述字不是一个打开的文件     |

　　-----------------------------+--------------------------------------+---------------------------------------+----------------------------------------+

　　第二个参数 nfds 制定数组中元素个数。第三个参数指定 poll 函数返回前等待多长时间。 INFTIM 表示永远等待， 0 代表立即返回， > 0 等待指定数目的秒数。

#include <sys/socket.h>
#include <netinet/in.h>
#include <stdio.h>
#include <error.h>
#include <errno.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <limits.h>
#include <poll.h>
#include <sys/stropts.h>
#include <signal.h>
#define MAXLINE 5
#define OPEN_MAX 1024
#define SA struct sockaddr


int main()
{
    int listenfd, connfd, sockfd, i, maxi;
    int nready;
    socklen_t clilen;
    ssize_t n;
    char buf[MAXLINE];
    struct pollfd client[OPEN_MAX];
    struct sockaddr_in servaddr, cliaddr;
    //创建监听套接字
    if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    {
        printf("socket() error!");
        exit(0);
    }
    //先要对协议地址进行清零
    bzero(&servaddr,sizeof(servaddr));
    //设置为 IPv4 or IPv6
    servaddr.sin_family = AF_INET;
    //绑定本地端口号
    servaddr.sin_port    = htons(9805);
    //任何一个 IP 地址，让内核自行选择
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    //绑定套接口到本地协议地址
    if(bind(listenfd, (SA *) &servaddr,sizeof(servaddr)) < 0)
    {
        printf("bind() error!");
        exit(0);
    }
    //服务器开始监听
    if(listen(listenfd,5) < 0)
    {
        printf("listen() error!");
        exit(0);
    }
    client[0].fd = listenfd;
    client[0].events = POLLRDNORM;//关心监听套机字的读事件
    for(i = 1; i < OPEN_MAX; ++i)
    {
        client[i].fd = -1;
    }
    maxi = 0;
    for(;;)
    {
        nready = poll(client, maxi + 1, -1);
        if(client[0].revents & POLLRDNORM)
        {
            clilen = sizeof(cliaddr);
            //accept 的后面两个参数都是值-结果参数，他们的保留的远程连接电脑的信息，如果不管新远程连接电脑的信息，可以将这两个参数设置为 NULL
            connfd = accept(listenfd, (SA *) &cliaddr, &clilen);
            if(connfd < 0)
            {
                continue;
            }
            for(i = 1; i < OPEN_MAX; ++i)
            {
                if(client[i].fd < 0)
                    client[i].fd = connfd;
                break;
            }
            if(i == OPEN_MAX)
            {
                printf("too many clients");
                exit(0);
            }
            client[i].events = POLLRDNORM;
            if(i > maxi)
            {
                maxi = i;
            }
            if(--nready <=0 )
                continue;
        }
        for(i = 1; i < OPEN_MAX; ++i)
        {
            if((sockfd = client[i].fd) < 0)
            {
                continue;
            }
            if(client[i].revents & POLLRDNORM | POLLERR)
            {
                if((n = read(sockfd, buf, MAXLINE)) < 0)
                {
                    if(errno == ECONNRESET)
                    {
                        close(sockfd);
                        client[i].fd = -1;
                    }
                    else
                    {
                        printf("read error!
");
                    }
                }
                else if(n == 0)
                {
                    close(sockfd);
                    client[i].fd = -1;
                }
                else
                {
                    write(sockfd, buf,  n);
                }
                if(--nready <= 0)
                    break;
            }
        }
    }
}