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.            /*nready为poll函数调用返回可做动作的个数*/
  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 网络编程>第六章。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值