poll函数

  poll函数起源于SVR3,最初局限于流设备。SVR4取消了这种限制,允许poll工作在任何描述字上。poll提供的功能与select类似,不过在处理流设备时,它能够提供额外的信息。

  1.#include <poll.h>

  2.

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

  4.               返回:就绪描述字的个数,0-超时,-1-出错

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

  struct pollfd{

  int fd;              //descriptor to check

  short events;    //events of interest on fd

  short revents;   //events that occurred on fd

  };

  要测试的条件由events成员指定,而返回的结果则在revents中存储。常用条件及含意说明如下:

poll函数可用的测试值

常量 说明
POLLIN 普通或优先级带数据可读
POLLRDNORM 普通数据可读
POLLRDBAND 优先级带数据可读
POLLPRI 高优先级数据可读
POLLOUT 普通数据可写
POLLWRNORM 普通数据可写
POLLWRBAND 优先级带数据可写
POLLERR 发生错误
POLLHUP 发生挂起
POLLNVAL 描述字不是一个打开的文件

 

  注意:后三个只能作为描述字的返回结果存储在revents中,而不能作为测试条件用于events中。

  第二个参数nfds是用来指定数组fdarray的长度。

  最后一个参数timeout是指定poll函数返回前等待多长时间。它的取值如下:

 

timeout值 说明
INFTIM 永远等待
0 立即返回,不阻塞进程
>0 等待指定数目的毫秒数

 

  一个使用poll的网络程序例子:

  001./**

  002.  *TCP回射服务器的服务端程序

  003.  */

  004.#include <stdio.h>

  005.#include <stdlib.h>

  006.#include <unistd.h>

  007.#include <sys/socket.h>

  008.#include <sys/types.h>

  009.#include <netinet/in.h>

  010.#include <netdb.h>

  011.#include <string.h>

  012.#include <errno.h>

  013.#include <poll.h>   //for poll

  014.

  015.#define LISTENQ 1024

  016.#define MAXLINE 1024

  017.#define OPEN_MAX 50000

  018.#define SERVER_PORT 3333

  019.

  020.#ifndef INFTIM     /*按照书上解释:POSIX规范要求INFTIM在头文件<poll.h>中定义,不过*/

  021.#define INFTIM -1  /*许多系统仍然把它定义在头文件<sys/stropts.h>中,但是经过我的测试*/

  022.#endif             /*即使都包含这两个文件,编译器也找不到,不知何解。索性自己定义了。*/

  023.

  024.int main(int argc, char *argv[])

  025.{

  026.    int i, maxi, listenfd, connfd, sockfd;

  027.    int nready;

  028.    ssize_t n;

  029.    socklen_t clilen;

  030.    struct sockaddr_in servaddr, cliaddr;

  031.    struct hostent  *hp;

  032.    char buf[BUFSIZ];

  033.    struct pollfd client[OPEN_MAX]; /*用于poll函数第一个参数的数组*/

  034.

  035.    if( argc != 2 )

  036.    {

  037.        printf("Please input %s <hostname>\n", argv[0]);

  038.        exit(1);

  039.    }

  040.

  041.    //创建socket

  042.    if( (listenfd = socket(AF_INET, SOCK_STREAM,0)) < 0 )

  043.    {

  044.        printf("Create socket error!\n");

  045.        exit(1);

  046.    }

  047.

  048.    //设置服务器地址结构

  049.    bzero(&servaddr, sizeof(servaddr));

  050.    servaddr.sin_family = AF_INET;

  051.    if( (hp = gethostbyname(argv[1])) != NULL )

  052.    {

  053.        bcopy(hp->h_addr, (struct sockaddr*)&servaddr.sin_addr, hp->h_length);

  054.    }

  055.    else if(inet_aton(argv[1], &servaddr.sin_addr) < 0 )

  056.    {

  057.        printf("Input Server IP error!\n");

  058.        exit(1);

  059.    }

  060.    servaddr.sin_port = htons(SERVER_PORT);

  061.

062.    //绑定地址

  063.    if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0 )

  064.    {

  065.        printf("IPaddress bound failure!\n");

  066.        exit(1);

  067.    }

  068.

  069.    //开始监听

  070.    listen(listenfd, LISTENQ);

  071.

  072.    client[0].fd = listenfd;         /*将数组中的第一个元素设置成监听描述字*/

  073.

  074.    client[0].events = POLLIN;  /*将测试条件设置成普通或优先级带数据可读,此处书中为POLLRDNORM,

  075.                                      但是怎么也编译不过去 ,编译器就是找不到,所以就临时改成了POLLIN这个条件,

  076.                                      希望以后能弄清楚。

  077.                                     */

  078.

  079.    for(i = 1;i < OPEN_MAX; ++i)     /*数组中的其它元素将暂时设置成不可用*/

  080.        client[i].fd = -1;

  081.    maxi = 0;

  082.

  083.    while(1)

  084.    {

  085.        nready = poll(client, maxi+1,INFTIM); //将进程阻塞在poll上

  086.        if( client[0].revents & POLLIN/*POLLRDNORM*/ ) /*先测试监听描述字*/

  087.        {

  088.            connfd = accept(listenfd,(struct sockaddr*)&servaddr, &clilen);

  089.            for(i = 1; i < OPEN_MAX; ++i)

  090.                if( client[i].fd < 0 )

  091.                {

  092.                    client[i].fd = connfd;  /*将新连接加入到测试数组中*/

  093.                    client[i].events = POLLIN;//POLLRDNORM; /*测试条件普通数据可读*/

  094.                    break;

  095.                }

  096.            if( i == OPEN_MAX )

  097.            {

  098.                printf("too many clients"); //连接的客户端太多了,都达到最大值了

  099.                exit(1);

  100.            }

  101.

  102.            if( i > maxi )

  103.                maxi = i;  //maxi记录的是数组元素的个数

  104.

  105.            if( --nready <= 0 )

  106.                continue;   //如果没有可读的描述符了,就重新监听连接

  107.        }

  108.

  109.        for(i = 1; i <= maxi; i++)  /*测试除监听描述字以后的其它连接描述字*/

  110.        {

  111.            if( (sockfd = client[i].fd) < 0) /*如果当前描述字不可用,就测试下一个*/

  112.                continue;

  113.

  114.            if(client[i].revents & (POLLIN/*POLLRDNORM*/ | POLLERR))/*如果当前描述字返回的是普通数据可读或出错条件*/

  115.            {

  116.                if( (n = read(sockfd, buf, MAXLINE)) < 0) //从套接口中读数据

  117.                {

  118.                    if( errno == ECONNRESET) //如果连接断开,就关闭连接,并设当前描述符不可用

  119.                    {

  120.                        close(sockfd);

  121.                        client[i].fd = -1;

  122.                    }

  123.                    else

  124.                        perror("read error");

  125.                }

  126.                else if(n == 0) //如果数据读取完毕,关闭连接,设置当前描述符不可用

  127.                {

  128.                    close(sockfd);

  129.                    client[i].fd = -1;

  130.                }

  131.                else

  132.                    write(sockfd, buf, n); //打印数据

  133.

  134.                if(--nready <= 0)

  135.                    break;

  136.

  137.            }

  138.        }

  139.    }

  140.

  141.    exit(0);

  142.}

  注:本章内容摘自<Unix 网络编程>第六章。

posted on 2012-03-30 16:00  一个人的天空@  阅读(4787)  评论(0编辑  收藏  举报