《UNIX网络编程 卷1》 笔记： 使用poll函数的单进程TCP回射服务器程序

本节我们使用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 1024int 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;} elseerr_sys("read error");} else if (n == 0) { /*客户关闭连接*/Close(sockfd);client[i].fd = -1;} elseWriten(sockfd, buf, n);if (--nready <= 0)break;}}}}