第十七篇：IO复用之select实现

前言

       在看过前文：初探IO复用后，想必你已对IO复用这个概念有了初步但清晰的认识。

       接下来，我要在一个具体的并发客户端中实现它(基于select函数)，使得一旦服务器中的客户进程被终止的时候，客户端这边立即得到通知并返回异常。

select函数

函数原型：int select ( int maxfdp1, fd_set *readset, fd_set *writeset, fd_set *exceptset, const struct timeval *timeout )

包含头文件：sys/select.h  sys/time.h ( 两个都要包含 )

参数说明：maxfdp1表示监听描述符个数( 一般直接赋予最大描述符号的值+1 )，中间几个参数表示具体的监听描述符集( 读/写/异常描述符集，本质是位向量组 )，最后一个参数表示监听的时间限制结构体。

配套使用宏：

1. void FD_ZERO ( fd_set *fdset ) 监听描述符集清零

2. void FD_SET ( int fd, fd_set *fset ) 注册要监听的描述符

3. void FD_CLR ( int fd, fd_set *fdset ) 注销不要监听的描述符

4. int FD_ISSET ( int fd, fd_set *fdset ) 判断某个监听信号是否接收到( 一般用于在select函数调用后某个信号是否接收到判断 )

返回值：返回已经就绪的描述符的个数，如果是定时器到时则返回0，出错则返回-1。

大致用法：

1. 定义一个空的描述符集

以下部分循环：

2. 注册要监听的描述符

3. 调用select函数

4. 依次编写IO处理代码( 一般是if ( FD_ISSET(描述符), &监听描述符集 ) { } 这样的结构 )。

代码实现

       下面的代码对之前的客户端代码做了修改，增加了基于select函数的IO复用机制：

#include    "unp.h"

void
str_cli(FILE *fp, int sockfd)
{
    int            maxfdp1;
    // 定义描述符集
    fd_set        rset;
    char        sendline[MAXLINE], recvline[MAXLINE];

    // 清空描述符集
    FD_ZERO(&rset);
    for ( ; ; ) {
        // 注册要监听的两个描述符
        FD_SET(fileno(fp), &rset);
        FD_SET(sockfd, &rset);
        // 计算最大描述符并将它+1后作为select函数的第一个参数
        maxfdp1 = max(fileno(fp), sockfd) + 1;
        // 调用并阻塞于select函数
        Select(maxfdp1, &rset, NULL, NULL, NULL);

        // 读取回射以及处理TCP分节
        if (FD_ISSET(sockfd, &rset)) {    
            if (Readline(sockfd, recvline, MAXLINE) == 0)
                err_quit("str_cli: server terminated prematurely");
            Fputs(recvline, stdout);
        }

        // 处理用户输入
        if (FD_ISSET(fileno(fp), &rset)) { 
            if (Fgets(sendline, MAXLINE, fp) == NULL)
                return;        
            Writen(sockfd, sendline, strlen(sendline));
        }
    }
}

运行测试

       经过测试，发现当服务器杀死客户端子进程后，客户端这边立刻报错并退出了程序。