unix下网络编程之I/O复用（三）

poll函数

在上文unix下网络编程之I/O复用（二）中已经介绍了select函数的相关使用，本文将介绍另一个常用的I/O复用函数poll。poll提供的功能与select类似，不过在处理流设备时，它能够提供额外的信息。

poll函数原型：

1 2 3

#include<poll.h>    int poll (struct pollfd * fdarray , unsigned long nfds , int timeout);    //返回：就需描述字的个数，0——超时，-1——出错

 第一个参数是指向一个结构数组第一个元素的指针，每个数组元素都是一个pollfd结构。如下：

1 2 3 4 5

struct pollfd {     int fd; //descriptor to check     short events; //events of interest on fd `   short revents; //events tha occurred on fd }

要测试的条件由events成员指定，函数在相应的revents成语中返回该描述字的状态。（每个描述字都有两个变量，一个为调用值，另一个为返回结果，从而避免使用值-结果参数，这与select函数是不同的）。下图列出了用于指定events标志以及测试revents标志的一些常值。

上图需要注意的是，POLLERR,POLLHUP,POLLNVAL是处理错误的描述字，因此它们也就不可以出现在input事件中，即events。poll识别三类数据：普通（normal），优先级带（priority band）和高优先级（high priority）。

对TCP和UPD而言，以下条件引起poll返回特定的revents。

1、 All regular TCP data and all UDP data is considered normal. 
2、 TCP's out-of-band data (Chapter 24) is considered priority band. 
3、 When the read half of a TCP connection is closed (e.g., a FIN is received), this is also considered normal data and a subsequent read operation will return 0. 
4、 The presence of an error for a TCP connection can be considered either normal data or an error (POLLERR). In either case, a subsequent read will return –1 with errno set to the appropriate value. This handles conditions such as the receipt of an RST or a timeout. 
5、 The availability of a new connection on a listening socket can be considered either normal data or priority data. Most implementations consider this normal data. 
6、 The completion of a nonblocking connect is considered to make a socket writable.

——《unix网络编程》第三版

参数nfds，指示结构数组中元素的个数。

参数timeout：

与select中的timeout不同，poll函数的timeout参数是一int值，表示poll函数返回前等待多长时间，它是毫秒级别的。它有三种情况的取值：1、INFTIM（一个负数值），表示永远等待，即一直阻塞。2、0，表示立即返回，非阻塞。3、>0，表示正待指定数目的毫秒数。

poll函数的返回值：

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

如果我们不再关心某个特定描述字，那么可以把与他对应的pollfd结构的fd成员设置成一个负值。poll函数将忽略这样的pollfd结构的events成员，返回时将它的revents成员的值置为0。

poll函数的通信列子：一个简单的TCP回射服务器程序

pollServer.c:使用select机制的服务器程序

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94

#include <stdio.h> #include <string.h> #include <arpa/inet.h> #include <netinet/in.h> #include <sys/socket.h> #include <poll.h>   /*环境为ubuntu10.04自带c环境，无法自动引入下列宏，所以自己写在前面了*/ #define INFTIM -1 #define POLLRDNORM  0x040       /* Normal data may be read.  */ #define POLLRDBAND  0x080       /* Priority data may be read.  */ #define POLLWRNORM  0x100       /* Writing now will not block.  */ #define POLLWRBAND  0x200       /* Priority data may be written.  */   #define MAXLINE  1024 #define OPEN_MAX  16 //一些系统会定义这些宏 #define SERV_PORT  10001   int main() {     int i , maxi ,listenfd , connfd , sockfd ;     int nready;     int n;     char buf[MAXLINE];     socklen_t clilen;     struct pollfd client[OPEN_MAX];       struct sockaddr_in cliaddr , servaddr;     listenfd = socket(AF_INET , SOCK_STREAM , 0);     memset(&servaddr,0,sizeof(servaddr));     servaddr.sin_family = AF_INET;     servaddr.sin_port = htons(SERV_PORT);     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);       bind(listenfd , (struct sockaddr *) & servaddr, sizeof(servaddr));     listen(listenfd,10);     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,INFTIM);         if (client[0].revents & POLLRDNORM)         {             clilen = sizeof(cliaddr);             connfd = accept(listenfd , (struct sockaddr *)&cliaddr, &clilen);             for(i=1;i<OPEN_MAX;i++)             {                 if(client[i].fd<0)                 {                     client[i].fd = connfd;                     client[i].events = POLLRDNORM;                     break;                 }             }             if(i==OPEN_MAX)             {                 printf("too many clients! \n");             }             if(i>maxi) maxi = i;             nready--;             if(nready<=0) continue;         }           for(i=1;i<=maxi;i++)         {             if(client[i].fd<0) continue;             sockfd = client[i].fd;             if(client[i].revents & (POLLRDNORM|POLLERR))             {                 n = read(client[i].fd,buf,MAXLINE);                 if(n<=0)                 {                     close(client[i].fd);                     client[i].fd = -1;                 }                 else                 {                     buf[n]='\0';                     printf("Socket %d said : %s\n",sockfd,buf);                     write(sockfd,buf,n); //Write back to client                 }                 nready--;                 if(nready<=0) break; //no more readable descriptors             }         }     }     return 0; }

客户端程序参考上一篇文章。

总结：

本文介绍了poll函数的原型，参数说明，注意事项以及一个简单的代码例子。在unix后续版本中，加入了epoll函数I/O复用机制，它在一定条件下更加高效，在以后的文章中，会对epoll机制再进行详细的描述。之前在学习python的时候，也接触了select和poll，但是当时了解的比较浅显，希望通过最近的学习可以对unix下I/O复用有更深入的认识。