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 的作用咯！