本节我们使用poll函数来实现单进程TCP回显服务器。poll函数与select函数不同,它不使用描述符集,而是使用pollfd结构体数组,pollfd结构体如下:
struct pollfd {
int fd; /* file descriptor */
short events; /* requested events */
short revents; /* returned events */
};
events表示该描述符感兴趣的事件,revents表示在该描述符上发生的事件,各种事件定义如下:
我们定义一个1024大小的pollfd结构体数组,传递给poll函数,poll函数会监听数组里所有有效的描述符。每次一个新的客户连接时,我们为其分配数组中的一项,fd设置为客户套接字描述符,events设置为POLLRDNORM,然后循环处理所有的事件。代码如下:
#include "unp.h"
#define OPEN_MAX 1024
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, 0);
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[0].fd = listenfd; /*监听服务器socket*/
client[0].events = POLLRDNORM;
for (i = 1; i < OPEN_MAX; i++) /*任何fd为负值的项都被poll忽略*/
client[i].fd = -1;
maxi = 0;
for ( ; ; ) {
nready = Poll(client, maxi + 1, -1);
if (client[0].revents & POLLRDNORM) { /*新的客户连接*/
clilen = sizeof(cliaddr);
connfd = Accept(listenfd, (SA *)&cliaddr, &clilen);
for (i = 1; i < OPEN_MAX; i++) {
if (client[i].fd < 0) {
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 <= 0)
continue;
}
for (i = 1; i <= maxi; i++) {
if ((sockfd = client[i].fd) < 0)
continue;
if (client[i].revents & (POLLRDNORM | POLLERR)) {
if ((n = read(sockfd, buf, MAXLINE)) < 0) {
if (errno == ECONNRESET) { /*客户复位连接*/
Close(sockfd);
client[i].fd = -1;
} else
err_sys("read error");
} else if (n == 0) { /*客户关闭连接*/
Close(sockfd);
client[i].fd = -1;
} else
Writen(sockfd, buf, n);
if (--nready <= 0)
break;
}
}
}
}