不为客户连接创建子进程的并发回射服务器( poll实现 )

前言

　　在上文中，我使用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函数。否则的话就随意了。