不为客户连接创建子进程的并发回射服务器( 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, 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;
    client[0].events = POLLRDNORM;
    // 清空监听描述符数组
    for (i = 1; i < OPEN_MAX; i++)
        client[i].fd = -1;    
    // 该变量表示已连接套接字描述符的最大数量( 曾经 )
    maxi = 0;                

    for ( ; ; ) {
        nready = Poll(client, maxi+1, INFTIM);

        // 如果有监听描述符检测到可读数据的信号
        if (client[0].revents & POLLRDNORM) {    
            clilen = sizeof(cliaddr);
            connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);
#ifdef    NOTDEF
            printf("new client: %s
", Sock_ntop((SA *) &cliaddr, clilen));
#endif

            // 登记刚刚获取到的已连接套接字描述符
            // OPEN_MAX表示最多监听的描述符个数
            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) {
#ifdef    NOTDEF
                        printf("client[%d] aborted connection
", i);
#endif
                        Close(sockfd);
                        client[i].fd = -1;
                    } else
                        err_sys("read error");
                } else if (n == 0) {
#ifdef    NOTDEF
                    printf("client[%d] closed connection
", i);
#endif
                    Close(sockfd);
                    client[i].fd = -1;
                } else
                    Writen(sockfd, buf, n);

                // 如果检测到可读或者发生错误的信号
                if (--nready <= 0)
                    break;            
            }
        }
    }
}

说明

　　如果从可移植性方面考虑或者有处理信号阻塞方面的需求( 要知道select还有个作为其升级版的pselect函数能够妥善处理信号阻塞 )，就还是得用select函数而不是poll函数。否则的话就随意了。