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));

         }

     }

 }

运行测试

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

秒客网

IO复用之select实现

相关文章