linux 下 select 编程

　　linux 下的 select 知识点 unp 的第六章已经描述的很清楚，我们这里简单的说下 select 的作用，并给出 select 的客户端实例。我们知道 select 是IO 多路复用的一个最简单支持，poll 和 epoll 是 select 的升级版。在 UNIX 网络编程第五章读书笔记 我们遇到这样一个问题：当客户端阻塞在 fgets() 等待客户输入的时候，服务器端断开连接。而客户端却不能及时知道，只有在客户输入完毕并发送到服务器的时候才知道连接已经断开，但是此时可能已经过了很长时间了。如果我们想及时知道服务器断开连接怎么办呢？

　　我们知道不管是 fgets() 等待客户输入还是 read() 从套接口读取数据，都是 IO 操作。我们不能阻塞在某个 IO 操作中一个，这样其他 IO 操作会无法进行，即使其他 IO 操作上有数据了我们也无法及时读取。select 的原理是这样的：我们将这些 IO 操作所要操作的文件描述符放到一起(比如一个数组中)，然后阻塞在 select() 函数上，为什么要阻塞在这里呢？其实这时的 select 实在不停的遍历这个数组，查看其中的文件描述符上是否可读/可写，一旦可读/可写，select 返回，停止阻塞。然后我们对可读/可写的文件描述如做相应的操作即可。下面是 select 函数的原型：

　　int select(nfds, readfds, writefds, exceptfds, timeout)

　　nfds 是指定 select() 要遍历的最大文件描述符 + 1，readfds 就是放文件描述的数组，这个数组里面关心的是该数组中文件描述符的读事件，wretefds 也是放文件描述符的数组，这个数组里面关心的是该数组中文件描述符的写事件，exceptfds 也是放文件描述符的数组，这个数组关心的是该数组中文件描述符的出错事件。timeout 是 select 阻塞的时间。如果设置为 空指针，那么将永远阻塞下去直到某个描述符有事件发生(就绪)。否则的话就会在阻塞由 timeout 指定的时间后返回，无论关心的文件描述符是否有事件发生。select 返回有事件发生的文件描述符个数，失败返回 -1，超时返回 0！

　　下面是 select 应用在客户端实例代码：

 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <stdio.h>
 #include <error.h>
 #include <unistd.h>
 #include <string.h>
 #include <stdlib.h>
 #include <sys/wait.h>
 #include <signal.h>
 #include <sys/select.h>
 #include <sys/time.h>

 #define MAXLINE 15
 #define SA struct sockaddr

 void str_cli(FILE *fp, int sockfd)
 {
     char sendline[MAXLINE], recvline[MAXLINE],buf[MAXLINE];
     int writenbytes;
     fd_set    rset;//关心读事件的文件描述符结合
     int maxfdpl, stdineof;
     int counts = , n;
     //集合所有位清零
     FD_ZERO(&rset);
     for(;;)
     {
         //将集合中 (fp) 描述符相应的位置开关打开，表示关心这个描述符
         FD_SET(fileno(fp), &rset);
         //将集合中 sockfd 描述符相应的位置开关打开，表示关心这个描述符
         FD_SET(sockfd, &rset);
         //设定为最大描述符 + 1
         maxfdpl = (sockfd > fileno(fp) ? sockfd : fileno(fp)) + ;
         //会阻塞在这里
         if((counts = select(maxfdpl, &rset , NULL, NULL, NULL)) > )
         {
             //sockfd 描述符上是否有可读事件发生(即服务器端发送数据过来了)
             if(FD_ISSET(sockfd, &rset))
             {
                 if((n = read(sockfd, buf, MAXLINE)) == )
                 {
                     if(stdineof = )
                         return;
                     else
                     {
                         printf("str_cli:server terminated prematurely!\n");
                         exit();
                     }
                 }
                 if( (n = write(fileno(stdout), buf, n)) != n)
                 {
                     printf("str_cli:write() error!\n");
                     exit();
                 }
             }
             // fp 描述符上是否有可读事件发生(即用户输入了数据)
             if(FD_ISSET(fileno(fp), &rset))
             {

                 if((n = read(fileno(fp), buf, MAXLINE)) == )
                 {
                     stdineof = ;
                     shutdown(sockfd, SHUT_WR);
                     FD_CLR(fileno(fp), &rset);
                     continue;
                 }
                 write(sockfd, buf, n);
             }
         }
     }
 }
 int main(int argc, char **argv)
 {
     int sockfd;
     struct sockaddr_in servaddr;
     if(argc != )
     {
         printf("useage: tcpcli <IPaddress>");
         exit();
     }
     if((sockfd = socket(AF_INET, SOCK_STREAM, )) < )
     {
         printf("socket() error!");
         exit();
     }
     bzero(&servaddr, sizeof(servaddr));
     servaddr.sin_family = AF_INET;
     servaddr.sin_port = htons();

     if(inet_pton(AF_INET, argv[], &servaddr.sin_addr) < )
     {
         printf("inet_pton() error!");
         exit();
     }
     if(connect(sockfd, (SA *) &servaddr, sizeof(servaddr)) < )
     {
         printf("inet_pton() error!");
         exit();
     }
     str_cli(stdin, sockfd);
     exit();
 }

　　先运行下，表示连接正常：

　　

　　再看下进程：

　　

　　在终端运行 sudo kill -9 6073 杀死服务器端子进程，再查看进程信息：

　　　　

　　客户端能立即知道：

　　

　　看，这就是 select 的作用咯！