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差不多,就不贴出了。