前言
在上文中，我使用select函数实现了不为客户连接创建子进程的并发回射服务器( 点此进入 )。但其中有个细节确实有点麻烦，那就是还得设置一个client数组用来标记select监听描述符集中被设置为监听位的位。有没有方法简化这个处理呢？有！在《UNIX网络编程》第六章最后介绍了一种类似select的函数 --- poll函数，用它来实现IO复用使代码简化了不少( 起码并发回射服务器的例子是的 )。
poll函数介绍
1. 包含头文件：<poll.h>
2. 函数原型：int poll ( struct pollfd * fdarray, unsigned long nfds, int timeout);
// 监听描述数组的元素
struct pollfd {
int fd; // 监听描述符
short events; // 测试事件
short revents; // 测试结果
};
3. 说明：相比于select函数，poll函数使用了结构体数组fdarray来表示监听描述符集合，该数组元素类型如上代码所示。当我们检索这个数组，我们可以知道有哪些描述符被监听，监听测试事件是什么，测试的结果又是什么。( 而在select函数的fdset描述符集合中，无法获知某位是不是被监听的描述符，这也就是下面的代码并不需要使用client数组的原因 )。
代码实现
#include "unp.h"
// 下头文件包含宏定义OPEN_MAX
#include <limits.h> int
main(int argc, char **argv)
{
int i, maxi, listenfd, connfd, sockfd;
int nready;
ssize_t n;
char buf[MAXLINE];
socklen_t clilen;
struct pollfd client[OPEN_MAX];
struct sockaddr_in cliaddr, servaddr; listenfd = Socket(AF_INET, SOCK_STREAM, ); bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT); Bind(listenfd, (SA *) &servaddr, sizeof(servaddr)); Listen(listenfd, LISTENQ); client[].fd = listenfd;
client[].events = POLLRDNORM;
// 清空监听描述符数组
for (i = ; i < OPEN_MAX; i++)
client[i].fd = -;
// 该变量表示已连接套接字描述符的最大数量( 曾经 )
maxi = ; for ( ; ; ) {
nready = Poll(client, maxi+, INFTIM); // 如果有监听描述符检测到可读数据的信号
if (client[].revents & POLLRDNORM) {
clilen = sizeof(cliaddr);
connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);
#ifdef NOTDEF
printf("new client: %s\n", Sock_ntop((SA *) &cliaddr, clilen));
#endif // 登记刚刚获取到的已连接套接字描述符
// OPEN_MAX表示最多监听的描述符个数
for (i = ; i < OPEN_MAX; i++)
if (client[i].fd < ) {
client[i].fd = connfd;
break;
}
if (i == OPEN_MAX)
err_quit("too many clients"); // 设置监听测试事件
client[i].events = POLLRDNORM;
if (i > maxi)
maxi = i; // 如果信号已经处理完毕则自动进入下一次循环
if (--nready <= )
continue;
} for (i = ; i <= maxi; i++) {
if ( (sockfd = client[i].fd) < )
continue;
// 如果检测到可读或者发生错误的信号
if (client[i].revents & (POLLRDNORM | POLLERR)) {
if ( (n = read(sockfd, buf, MAXLINE)) < ) {
if (errno == ECONNRESET) {
#ifdef NOTDEF
printf("client[%d] aborted connection\n", i);
#endif
Close(sockfd);
client[i].fd = -;
} else
err_sys("read error");
} else if (n == ) {
#ifdef NOTDEF
printf("client[%d] closed connection\n", i);
#endif
Close(sockfd);
client[i].fd = -;
} else
Writen(sockfd, buf, n); // 如果检测到可读或者发生错误的信号
if (--nready <= )
break;
}
}
}
}
说明
如果从可移植性方面考虑或者有处理信号阻塞方面的需求( 要知道select还有个作为其升级版的pselect函数能够妥善处理信号阻塞 )，就还是得用select函数而不是poll函数。否则的话就随意了。