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 网络编程>第六章。