10.POLL函数

poll函数和select一样都是用于I/O复用，他们提供的功能也一样，只不过是poll想对于流设备，提供一些额外的功能。在SVR3时候只局限于使用流设备，但这个SVR4以后，poll可以用于各种设备。

 

#include "poll.h"

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

 

和之前的select一样的功能，这里简单介绍下它的API，然后修改下之前曾经出现过的聊天程序，用poll替换掉之前的select函数。

 

参数timeout很好理解，用于限时作用，单位为毫秒，如果设定为负值，则永远等待，为0则立即返回不等待，大于0则就是等待指定时间（当然如果系统不提供如此精确的等待时间，则向上取最近值）。

 

参数struct polifd * fdarray则是用来确定阻塞的设备（也就是描述字）。结构定义如下：

struct polifd {

  int fd;

  short events;

  short revents;

};

fd为等待的描述字，当fd为-1时候表示该组项被忽略。events为等待的事件，他可以取一下几种值：POLLIN（普通或优先级带的数据可读），POLLRDNORM（普通数据可读），POLLRDBAND(优先级带数据可读), POLLPRI(高优先级数据可读).所谓的优先级数据是流设备中概念，在网络套接口优先级数据指的是TCP的带外数据（这个就不展开讨论）。POLLOUT（普通数据可写），POLLWRNORM（普通数据可写），POLLWRBAND(优先级带数据可写).这些就是events设置的值。而revents则是用来返回结果，所以以上的参数都可以是revents返回的结果，除以上以外，它还有可能是下面几种返回结果：POLLERR(返回结果), POLLHUP(发生挂起),POLLNVAL(描述字不是一个打开的文件)。

 

nfds则指示的是fdarray数组的长度，fdarray是由进程打开文件的描述字决定，但平常我们并不关心所有打开的描述字，比如下面的程序，我们只关心标准输入文件0号（键盘）和我们连接的套接口是否有数据到来。所以用nfds标识我们关心的描述字个数。

 

关于返回值，返回负值则表示出错（和之前select一样，会被信号中断阻塞过程，而导致函数错误返回），0则表示超时。如果是大于0，则表示准备好的描述字个数。

 

说起来很抽象，不过结合下面的代码和解释我相信很容易理解。

 

 1 #include "/programe/net/head.h"
 2 #include "stdio.h"
 3 #include "stdlib.h"
 4 #include "string.h"
 5 #include "unistd.h"
 6 #include "sys/wait.h"
 7 #include "sys/select.h"
 8 #include "sys/poll.h"
 9
10 #define MAXSIZE 100
11
12 int main(int argc, char ** argv) {
13         int sockfd;
14         struct sockaddr_in serv_socket;
15         char end_flag[] = "EOF";
16         char end_word[] = "no words\n";
17         int send_flag = 1;
18         int recv_flag = 1;
19
20         sockfd = socket(AF_INET, SOCK_STREAM, 0);
21         bzero(&serv_socket, sizeof(struct sockaddr_in));
22         serv_socket.sin_family = AF_INET;
23         serv_socket.sin_port = htons(atoi(argv[1]));
24         serv_socket.sin_addr.s_addr = htonl(INADDR_ANY);
25
26         bind(sockfd, (struct sockaddr *)&serv_socket, sizeof(serv_socket));
27         listen(sockfd, 10);
28         int connfd = accept(sockfd, (struct sockaddr *)NULL, NULL);
29         for(;;) {
30                 struct pollfd fdarray[10];
31                 char send[MAXSIZE + 1], recv[MAXSIZE + 1];
32
33                 int i;
34                 for(i = 0; i < 10; i++)
35                         fdarray[i].fd = -1;
36
37                 fdarray[0].fd = 0;
38                 fdarray[0].events = POLLIN;
39
40                 fdarray[1].fd = connfd;
41                 fdarray[1].events = POLLIN;
42
43                 if(!send_flag && !recv_flag)
44                         break;
45                 int flag = poll(fdarray, 2, -1);
46
47                 if(flag < 0) {
48                         printf("system message:some error happen when waiting for input\n");
49                         continue;
50                 }
51
52                 if(send_flag) {
53                         if(fdarray[0].revents & POLLIN) {
54                                 int n = read(0, send, MAXSIZE);
55                                 if(!strncmp(end_flag, send, 3)) {
56                                         write(connfd, end_word, strlen(end_word));
57                                         shutdown(connfd, SHUT_WR);
58                                         send_flag = 0;
59                                 } else {

60                                         write(connfd, send, n);
61                                 }
62                         }
63                 }
64
65                 if(recv_flag) {
66                         if(fdarray[1].revents & POLLIN) {
67                                 int n = read(connfd, recv, MAXSIZE);
68                                 if(!n) {
69                                         printf("client closed\n");
70                                         break;
71                                 }
72                                 if(!strncmp(end_word, recv, 8)) {
73                                         printf("client message:%s", recv);
74                                         shutdown(connfd, SHUT_RD);
75                                         recv_flag = 0;
76                                 } else {
77                                         recv[n] = '\0';
78                                         printf("client message:%s", recv);
79                                 }
80                         }
81                 }
82
83         }
84         shutdown(connfd, SHUT_RDWR);
85 }

 

代码有点长了，不过我们只关心部分代码，之前那些链接的代码看的很多了。

从第30行开始，我们创建了描述字数组，这里用10，其实我们并不知道此时进程打开了多少文件，10只是个人随便选的。然后用for循环将整个数组的描述设置成-1，也就是不关心。然后将0组项设置成标准输入文件0号，然后关心的事件为POLLIN（这个值的意思之前有介绍）。然后将网络套接口描述字放入2号组项，并同样设置关心事件。伺候当从poll返回就开始检测是什么事件发生。代码很简单，结合说明很容易看懂，至于client代码和server差不多，就不贴出了。