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 應用在客戶端實例代碼：

  1 #include <sys/socket.h>
  2 #include <netinet/in.h>
  3 #include <stdio.h>
  4 #include <error.h>
  5 #include <unistd.h>
  6 #include <string.h>
  7 #include <stdlib.h>
  8 #include <sys/wait.h>
  9 #include <signal.h>
 10 #include <sys/select.h>
 11 #include <sys/time.h>
 12 
 13 #define MAXLINE 15
 14 #define SA struct sockaddr
 15 
 16 void str_cli(FILE *fp, int sockfd)
 17 {
 18     char sendline[MAXLINE], recvline[MAXLINE],buf[MAXLINE];
 19     int writenbytes;
 20     fd_set    rset;//關心讀事件的文件描述符結合
 21     int maxfdpl, stdineof;
 22     int counts = 0, n;
 23     //集合全部位清零
 24     FD_ZERO(&rset);
 25     for(;;)
 26     {
 27         //將集合中 (fp) 描述符相應的位置開關打開，表示關心這個描述符
 28         FD_SET(fileno(fp), &rset);
 29         //將集合中 sockfd 描述符相應的位置開關打開，表示關心這個描述符
 30         FD_SET(sockfd, &rset);
 31         //設定爲最大描述符 + 1
 32         maxfdpl = (sockfd > fileno(fp) ? sockfd : fileno(fp)) + 1;
 33         //會阻塞在這裏
 34         if((counts = select(maxfdpl, &rset , NULL, NULL, NULL)) > 0)
 35         {
 36             //sockfd 描述符上是否有可讀事件發生(即服務器端發送數據過來了)
 37             if(FD_ISSET(sockfd, &rset))
 38             {
 39                 if((n = read(sockfd, buf, MAXLINE)) == 0)
 40                 {
 41                     if(stdineof = 1)
 42                         return;
 43                     else
 44                     {
 45                         printf("str_cli:server terminated prematurely!\n");
 46                         exit(0);
 47                     }            
 48                 }
 49                 if( (n = write(fileno(stdout), buf, n)) != n)
 50                 {
 51                     printf("str_cli:write() error!\n");
 52                     exit(0);
 53                 }
 54             }
 55             // fp 描述符上是否有可讀事件發生(即用戶輸入了數據)
 56             if(FD_ISSET(fileno(fp), &rset))
 57             {
 58                 
 59                 if((n = read(fileno(fp), buf, MAXLINE)) == 0)
 60                 {
 61                     stdineof = 1;
 62                     shutdown(sockfd, SHUT_WR);
 63                     FD_CLR(fileno(fp), &rset);
 64                     continue;        
 65                 }
 66                 write(sockfd, buf, n);
 67             }    
 68         }
 69     }
 70 }
 71 int main(int argc, char **argv)
 72 {
 73     int sockfd;
 74     struct sockaddr_in servaddr;
 75     if(argc != 2)
 76     {
 77         printf("useage: tcpcli <IPaddress>");
 78         exit(0);
 79     }
 80     if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 81     {
 82         printf("socket() error!");
 83         exit(0);
 84     }
 85     bzero(&servaddr, sizeof(servaddr));
 86     servaddr.sin_family = AF_INET;
 87     servaddr.sin_port = htons(9806);
 88     
 89     if(inet_pton(AF_INET, argv[1], &servaddr.sin_addr) < 0)
 90     {
 91         printf("inet_pton() error!");
 92         exit(0);
 93     }
 94     if(connect(sockfd, (SA *) &servaddr, sizeof(servaddr)) < 0)
 95     {
 96         printf("inet_pton() error!");
 97         exit(0);
 98     }
 99     str_cli(stdin, sockfd);
100     exit(0);
101 }

　　先運行下，表示鏈接正常：

　　

　　再看下進程：

　　

　　在終端運行 sudo kill -9 6073 殺死服務器端子進程，再查看進程信息：

　　　　

　　客戶端能當即知道：

　　

　　看，這就是 select 的做用咯！