第十七篇：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) + ;
         // 调用并阻塞于select函数
         Select(maxfdp1, &rset, NULL, NULL, NULL);

         // 读取回射以及处理TCP分节
         if (FD_ISSET(sockfd, &rset)) {
             if (Readline(sockfd, recvline, MAXLINE) == )
                 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));
         }
     }
 }

运行测试

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