linux--select
在linux中的I/O都是用文件描述符表示的，文件描述符就是一個int，比如0是標(biāo)準(zhǔn)輸入，1是標(biāo)準(zhǔn)輸出，2是標(biāo)準(zhǔn)錯誤輸出。同理，socket也是返回一個int文件描述符。
select系統(tǒng)調(diào)用是用來讓我們的程序監(jiān)視多個文件描述符(file descriptor)的狀態(tài)變化的。程序會停在select這里等待，直到被監(jiān)視的文件描述符有某一個或多個發(fā)生了狀態(tài)改變。
只要有一個描述符狀態(tài)變?yōu)榭捎?#xff0c;select會停止堵塞，繼續(xù)往下執(zhí)行，這時我們就需要用輪詢的方法查出是哪個文件描述符可用。
理解select模型的關(guān)鍵在于理解fd_set,為說明方便，取fd_set長度為1字節(jié)，fd_set中的每一bit可以對應(yīng)一個文件描述符fd。則1字節(jié)長的fd_set最大可以對應(yīng)8個fd。
（1）執(zhí)行fd_set set; FD_ZERO(&set);則set用位表示是0000,0000。
（2）若fd＝5,執(zhí)行FD_SET(fd,&set);后set變?yōu)?0000100(第5位置為1)
（3）若再加入fd＝2，fd=1,則set變?yōu)?span style="color:rgb(51,51,51); font-family:微軟雅黑,Tahoma,Verdana,Arial,Helvetica,'宋 体 ',sans-serif; font-size:14px; line-height:20px; text-align:left">01100100
（4）執(zhí)行select(6,&set,0,0,0)阻塞等待（第一個參數(shù)是最大的文件描述符+1）
（5）若fd=1,fd=2上都發(fā)生可讀事件，則select返回，此時set變?yōu)?span style="color:rgb(51,51,51); font-family:微軟雅黑,Tahoma,Verdana,Arial,Helvetica,'宋 体 ',sans-serif; font-size:14px; line-height:20px; text-align:left">01100000。注意：沒有事件發(fā)生的fd=5被清空。
注意第五條中的fd=5被清空，說明只要是不活躍的文件描述符都會在select的阻塞被結(jié)束后從原有的文件描述集合fd_set中被去除。因此在socket使用select的場景下，是可以在while輪詢中使用FD_ISSET宏來不斷監(jiān)聽服務(wù)器端生成的socket文件描述符是否活躍。
FD_ZERO(fd_set *fdset)：清空fdset與所有文件句柄的聯(lián)系。
FD_SET(int fd, fd_set *fdset)：建立文件句柄fd與fdset的聯(lián)系。
FD_CLR(int fd, fd_set *fdset)：清除文件句柄fd與fdset的聯(lián)系。
FD_ISSET(int fd, fdset *fdset)：檢查fdset聯(lián)系的文件句柄fd是否可讀寫，>0表示可讀寫。
可以采用select實現(xiàn) ??并發(fā)服務(wù)器:多路復(fù)用I/O
下面是服務(wù)器select并發(fā)版本：
http://blog.csdn.net/vagrxie/article/details/3948204
select模型屬于網(wǎng)絡(luò)的I/O復(fù)用模型，比純粹的阻塞I/O模型更具有實用性，因為可以同時等待多個描述字的就緒。
當(dāng)年學(xué)習(xí)C/C++的時候，很少碰到底層以數(shù)字標(biāo)示的描述字，只在寫文件系統(tǒng)的去嘗試各種情況，以獲得最佳效率的時候?qū)嶋H嘗試使用過一次，一直覺得那種open,write,read的文件操作方式，實在是比fopen一族函數(shù)還要低級的方式-_-!平時沒有必要使用。但是等到網(wǎng)絡(luò)編程的時候，才發(fā)現(xiàn)。。。。原來這么底層的東西，竟然也有一定的通用性，文件的描述字和網(wǎng)絡(luò)的描述字竟然是一致的-_-!不管是誰設(shè)計的，還是挺佩服的。。。。。。
???????這里僅僅是為了學(xué)習(xí)Select模型而寫的學(xué)習(xí)例子，作用是在服務(wù)器端輸出連接上的客戶端的IP(僅以數(shù)字形式)，然后將客戶端的IP以字符串的形式返回，客戶端連接服務(wù)器，并接受由服務(wù)器端返回的IP地址，然后輸出轉(zhuǎn)換為字符串形式的IP地址和數(shù)字形式的IP地址，為了區(qū)別select到正確的不同listen套接字，這里用了不同的端口，并且不同的兩個套接字響應(yīng)時以echo 1,echo 2區(qū)別。功能是很簡單的，僅僅用于學(xué)習(xí)，所以其中很多地方本來可以抽出來稱為函數(shù)的，都貪簡單，直接復(fù)制了(-_-!這里本來習(xí)慣想說Ctrl-C Ctrl-V的。。。但是發(fā)現(xiàn)自己實在Ubuntu下用vim復(fù)制的，好像和實際情況不符。。。。)
???????另外。。。。由于用的是《Unix Network Programming》一書，所以編程風(fēng)格都變得有點像書中了。。。。服務(wù)器端全是自己寫的，客戶端代碼由書中的daytime客戶端改過來的，并且發(fā)現(xiàn)書中客戶端代碼都不關(guān)閉套接字，都交由退出進(jìn)程的時候由系統(tǒng)關(guān)閉，不知道這種風(fēng)格好不好。由于學(xué)習(xí)。。。寫的是ANSI C程序，用gcc編譯-_-!
unp.h是《Unix Network Programming》源代碼中的公用頭文件，makefile可能也得注意一下，為了圖省事，我用了其源代碼中的Make.defines，因為這樣比自己寫簡單多了：），makefile就不貼了，沒有什么學(xué)習(xí)意義。
?
運(yùn)行效果如下：
客戶端運(yùn)行：
./TestSelectCli 127.0.0.1 1000
Conncet OK
127.0.0.1:16777343 Echo 1.
laptop:~/unpv1/unpv13e/MyTest$ ./TestSelectCli 127.0.0.1 1001
Conncet OK
127.0.0.1:16777343 Echo 2.
laptop:~/unpv1/unpv13e/MyTest$ ./TestSelectCli 192.168.0.138 1000
Conncet OK
192.168.0.138:2315299008 Echo 1.
laptop:~/unpv1/unpv13e/MyTest$ ./TestSelectCli 192.168.0.138 1001
Conncet OK
192.168.0.138:2315299008 Echo 2.
?
服務(wù)器端輸出：
2315299008 Echo 1.
16777343 Echo 1.
16777343 Echo 2.
2315299008 Echo 1.
2315299008 Echo 2.
?
?
服務(wù)器端源代碼：
?
??1?#include????"unp.h"
??2?
??3?
??4?void?str_echo1(int?connfd);
??5?void?str_echo2(int?connfd);
??6?
??7?int?main(int?argc,?char?**argv)
??8?{
??9?????struct?sockaddr_in cliaddr;
?10?????pid_t childpid;
?11?
?12?????/*??Bind 1000 port to listen socket 1 */
?13?????int?listenfd1 = Socket(AF_INET, SOCK_STREAM,?0);
?14?
?15?????struct?sockaddr_in servaddr;
?16?????bzero(&servaddr,?sizeof(servaddr));
?17?????servaddr.sin_family = AF_INET;
?18?????servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
?19?????servaddr.sin_port = htons(1000);
?20?
?21?????Bind(listenfd1, (SA *)&servaddr,?sizeof(servaddr));
?22?
?23?????Listen(listenfd1, LISTENQ);
?24?
?25?????/*??Bind 1001 port to listen socket 2*/
?26?????int?listenfd2 = Socket(AF_INET, SOCK_STREAM,?0);
?27?
?28?????struct?sockaddr_in servaddr2;
?29?????bzero(&servaddr2,?sizeof(servaddr2));
?30?????servaddr2.sin_family = AF_INET;
?31?????servaddr2.sin_addr.s_addr = htonl(INADDR_ANY);
?32?????servaddr2.sin_port = htons(1001);
?33?
?34?????Bind(listenfd2, (SA *)&servaddr2,?sizeof(servaddr2));
?35?
?36?????Listen(listenfd2, LISTENQ);
?37?
?38?????/*?Initialize fd_set struct */
?39?????int?maxfdp1 = max(listenfd1, listenfd2) +?1;
?40?????fd_set rset;
?41?????FD_ZERO(&rset);
?42?
?43?????/*??Select from this two listen socket */
?44?????for( ; ; )
?45?????{
?46?????????FD_SET(listenfd1, &rset);
?47?????????FD_SET(listenfd2, &rset);
?48?
?49?????????int?nready = -1;
?50?????????if( (nready = select(maxfdp1, &rset,?NULL,?NULL,NULL)) <?0)//只要有一個客戶連接，select就停止堵塞
?51?????????{
?52?????????????if(EINTR?== errno)
?53?????????????{
?54?????????????????continue;
?55?????????????}
?56?????????????else
?57?????????????{
?58?????????????????err_sys("Select error.");
?59?????????????}
?60?????????}
?61?
?62?????????/*??some one listening socket is readable.*/
?63?????????if(FD_ISSET(listenfd1, &rset))
?64?????????{
?65?????????????socklen_t len =?sizeof(cliaddr);
?66?????????????int?connfd = Accept(listenfd1, (SA *)&cliaddr, &len);
?67?
?68?????????????if(?0?== (childpid = Fork()) )
?69?????????????{
?70?????????????????/*?child process */
?71?????????????????Close(listenfd1);
?72?
?73?????????????????str_echo1(connfd);
?74?????????????????exit(0);
?75?????????????}
?76?
?77?????????????/*?parent process??*/
?78?????????????Close(connfd);
?79?
?80?????????}
?81?
?82?
?83?????????if(FD_ISSET(listenfd2, &rset))
?84?????????{
?85?????????????socklen_t len =?sizeof(cliaddr);
?86?????????????int?connfd = Accept(listenfd2, (SA *)&cliaddr, &len);
?87?
?88?????????????if(?0?== (childpid = Fork()) )
?89?????????????{
?90?????????????????/*?child process */
?91?????????????????Close(listenfd2);
?92?
?93?????????????????str_echo2(connfd);
?94?????????????????exit(0);
?95?????????????}
?96?
?97?????????????/*?parent process??*/
?98?????????????Close(connfd);
?99?
100?????????}
101?
102?????}
103?
104?????exit(0);
105?}
106?
107?void?str_echo1(int?connfd)
108?{
109?????struct?sockaddr_in clientAddr;
110?????socklen_t len =?sizeof(clientAddr);
111?
112?????if(getpeername(connfd, (SA*) &clientAddr, &len) <?0)
113?????{
114?????????return;
115?????}
116?
117?????char?lcBuffer[MAXLINE] = {0};
118?????sprintf(lcBuffer,?"%u?Echo 1.", clientAddr.sin_addr.s_addr);
119?
120?????printf("%s/n", lcBuffer);
121?
122?????Write(connfd, lcBuffer, MAXLINE);
123?}
124?
125?
126?void?str_echo2(int?connfd)
127?{
128?????struct?sockaddr_in clientAddr;
129?????socklen_t len =?sizeof(clientAddr);
130?
131?????if(getpeername(connfd, (SA*) &clientAddr, &len) <?0)
132?????{
133?????????return;
134?????}
135?
136?
137?????char?lcBuffer[MAXLINE] = {0};
138?????sprintf(lcBuffer,?"%u?Echo 2.", clientAddr.sin_addr.s_addr);
139?
140?????printf("%s/n", lcBuffer);
141?
142?????Write(connfd, lcBuffer, MAXLINE);
143?}
144?
145
?
客戶端源代碼：
?1?#include????"unp.h"
?2?
?3?int?main(int?argc,?char?**argv)
?4?{
?5?????int????????????????????sockfd, n;
?6?????char?????????????? recvline[MAXLINE +?1];
?7?????struct?sockaddr_in servaddr;
?8?
?9?????if?(argc !=?3)
10?????????err_quit("usage: a.out <IPaddress> <IPPort>");
11?
12?????int?port = atoi(argv[2]);
13?
14?????if?( (sockfd = socket(AF_INET, SOCK_STREAM,?0)) <?0)
15?????????err_sys("socket error");
16?
17?????bzero(&servaddr,?sizeof(servaddr));
18?????servaddr.sin_family = AF_INET;
19?????servaddr.sin_port?? = htons(port);
20?????if?(inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <=0)
21?????????err_quit("inet_pton error for?%s", argv[1]);
22?
23?????if?(connect(sockfd, (SA *) &servaddr,?sizeof(servaddr)) <?0)
24?????????err_sys("connect error");
25?
26?????printf("Conncet OK/n");
27?
28?????while?( (n = read(sockfd, recvline, MAXLINE)) >?0) {
29?????????recvline[n] =?0;??/*?null terminate */
30?
31?????????/*??change number string to number and to ip string */
32?????????struct?in_addr svraddr;
33?????????svraddr.s_addr = strtoul(recvline,?NULL,?10);
34?????????char?*pszsvraddr = inet_ntoa(svraddr);
35?
36?????????printf("%s:%s/n", pszsvraddr, recvline);
37?????}
38?????if?(n <?0)
39?????????err_sys("read error");
40?
41?????exit(0);
42?}
下面更簡潔的代碼會加深理解
我們要監(jiān)控readfd文件描述符數(shù)組
int use_select(int *readfd,int n)?
{?
?? fd_set my_readfd;?
?? int maxfd;?
?? int i;?
????
?? maxfd=readfd[0];?
?? for(i=1;i <n;++i) ? if(readfd[i]>maxfd) maxfd=readfd[i];?
?? while(1)?
?? {?
??????? /*?? 將所有的文件描述符加入?? */?
??????? FD_ZERO(&my_readfd);?
??????? for(i=0;i??<n;++i)?????????? FD_SET(readfd[i],*my_readfd);?
??????? /*???? 進(jìn)程阻塞???????????????? */?
??????? select(maxfd+1,& my_readfd,NULL,NULL,NULL);?
??????? /*??????? 有東西可以讀了?????? */?
??????? for(i=0;i?<n;++i)? ? ?if(FD_ISSET(readfd[i],&my_readfd))?
????????????? {?
????????????????? /* 原來是我可以讀了 */?
??????????????????????? we_read(readfd[i]);?
????????????? }?
?? }?
}
select監(jiān)控的是一個文件描述符數(shù)組，一旦至少一個文件描述符可以用了，select停止堵塞，這時程序需要一個輪詢確定具體哪個文件可以用了。
下面是一個?用select實現(xiàn)的異步聊天程序：
服務(wù)器端：
#include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <sys/types.h> #include <netinet/in.h> #include <sys/socket.h> #include <sys/wait.h> #include <unistd.h> #include <arpa/inet.h> #include <sys/time.h> #include <sys/types.h>#define MAXBUF 1024 /************關(guān)于本文檔******************************************** *filename: async-server.c *purpose: 演示網(wǎng)絡(luò)異步通訊，這是服務(wù)器端程序 *wrote by: zhoulifa(zhoulifa@163.com) 周立發(fā)(http://zhoulifa.bokee.com) Linux愛好者 Linux知識傳播者 SOHO族 開發(fā)者 最擅長C語言 *date time:2007-01-25 21:22 *Note: 任何人可以任意復(fù)制代碼并運(yùn)用這些文檔，當(dāng)然包括你的商業(yè)用途 * 但請遵循GPL *Thanks to: Google.com *Hope:希望越來越多的人貢獻(xiàn)自己的力量，為科學(xué)技術(shù)發(fā)展出力 * 科技站在巨人的肩膀上進(jìn)步更快！感謝有開源前輩的貢獻(xiàn)！ *********************************************************************/int main(int argc, char **argv) {int sockfd, new_fd;socklen_t len;struct sockaddr_in my_addr, their_addr;unsigned int myport, lisnum;char buf[MAXBUF + 1];fd_set rfds;struct timeval tv;int retval, maxfd = -1;if (argv[1])myport = atoi(argv[1]);elsemyport = 7838;if (argv[2])lisnum = atoi(argv[2]);elselisnum = 2;if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {perror("socket");exit(1);}bzero(&my_addr, sizeof(my_addr));my_addr.sin_family = PF_INET;my_addr.sin_port = htons(myport);if (argv[3])my_addr.sin_addr.s_addr = inet_addr(argv[3]);elsemy_addr.sin_addr.s_addr = INADDR_ANY;if (bind(sockfd, (struct sockaddr *) &my_addr, sizeof(struct sockaddr))== -1) {perror("bind");exit(1);}if (listen(sockfd, lisnum) == -1) {perror("listen");exit(1);}while (1) {printf("\n----等待新的連接到來開始新一輪聊天……\n");len = sizeof(struct sockaddr);if ((new_fd =accept(sockfd, (struct sockaddr *) &their_addr,&len)) == -1) {perror("accept");exit(errno);} elseprintf("server: got connection from %s, port %d, socket %d\n",inet_ntoa(their_addr.sin_addr),ntohs(their_addr.sin_port), new_fd);/* 開始處理每個新連接上的數(shù)據(jù)收發(fā) */printf("\n準(zhǔn)備就緒，可以開始聊天了……直接輸入消息回車即可發(fā)信息給對方\n");while (1) {/* 把集合清空 */FD_ZERO(&rfds);/* 把標(biāo)準(zhǔn)輸入句柄0加入到集合中 */FD_SET(0, &rfds);maxfd = 0;/* 把當(dāng)前連接句柄new_fd加入到集合中 */FD_SET(new_fd, &rfds);if (new_fd > maxfd)maxfd = new_fd;/* 設(shè)置最大等待時間 */tv.tv_sec = 1;tv.tv_usec = 0;/* 開始等待 */retval = select(maxfd + 1, &rfds, NULL, NULL, &tv);if (retval == -1) {printf("將退出，select出錯！ %s", strerror(errno));break;} else if (retval == 0) {/* printf("沒有任何消息到來，用戶也沒有按鍵，繼續(xù)等待……\n"); */continue;} else {if (FD_ISSET(0, &rfds)) {/* 用戶按鍵了，則讀取用戶輸入的內(nèi)容發(fā)送出去 */bzero(buf, MAXBUF + 1);fgets(buf, MAXBUF, stdin);if (!strncasecmp(buf, "quit", 4)) {printf("自己請求終止聊天！\n");break;}len = send(new_fd, buf, strlen(buf) - 1, 0);if (len > 0)printf("消息:%s\t發(fā)送成功，共發(fā)送了%d個字節(jié)！\n",buf, len);else {printf("消息'%s'發(fā)送失敗！錯誤代碼是%d，錯誤信息是'%s'\n",buf, errno, strerror(errno));break;}}if (FD_ISSET(new_fd, &rfds)) {/* 當(dāng)前連接的socket上有消息到來則接收對方發(fā)過來的消息并顯示 */bzero(buf, MAXBUF + 1);/* 接收客戶端的消息 */len = recv(new_fd, buf, MAXBUF, 0);if (len > 0)printf("接收消息成功:'%s'，共%d個字節(jié)的數(shù)據(jù)\n",buf, len);else {if (len < 0)printf("消息接收失敗！錯誤代碼是%d，錯誤信息是'%s'\n",errno, strerror(errno));elseprintf("對方退出了，聊天終止\n");break;}}}}close(new_fd);/* 處理每個新連接上的數(shù)據(jù)收發(fā)結(jié)束 */printf("還要和其它連接聊天嗎？(no->退出)");fflush(stdout);bzero(buf, MAXBUF + 1);fgets(buf, MAXBUF, stdin);if (!strncasecmp(buf, "no", 2)) {printf("終止聊天！\n");break;}}close(sockfd);return 0; }
用select實現(xiàn)服務(wù)器的好處是，當(dāng)客戶與其建立連接之后，客戶可能要過段時間才發(fā)送消息，如果之前普通的處理，服務(wù)器只能堵塞在recv，不能有其他操作；而用select監(jiān)控輸入文件描述符和socket描述符，只要有一個準(zhǔn)備好，select就返回，譬如，select堵塞時，只要服務(wù)器輸入消息，并回車發(fā)送時，select檢測到輸入文件描述符，就立馬返回，將消息發(fā)送給客戶端；
客戶端程序：
#include <stdio.h> #include <string.h> #include <errno.h> #include <sys/socket.h> #include <resolv.h> #include <stdlib.h> #include <netinet/in.h> #include <arpa/inet.h> #include <unistd.h> #include <sys/time.h> #include <sys/types.h>#define MAXBUF 1024 /************關(guān)于本文檔******************************************** // *filename: ssync-client.c *purpose: 演示網(wǎng)絡(luò)異步通訊，這是客戶端程序 *wrote by: zhoulifa(zhoulifa@163.com) 周立發(fā)(http://zhoulifa.bokee.com) Linux愛好者 Linux知識傳播者 SOHO族 開發(fā)者 最擅長C語言 *date time:2007-01-25 21:32 *Note: 任何人可以任意復(fù)制代碼并運(yùn)用這些文檔，當(dāng)然包括你的商業(yè)用途 * 但請遵循GPL *Thanks to: Google.com *Hope:希望越來越多的人貢獻(xiàn)自己的力量，為科學(xué)技術(shù)發(fā)展出力 * 科技站在巨人的肩膀上進(jìn)步更快！感謝有開源前輩的貢獻(xiàn)！ *********************************************************************/ int main(int argc, char **argv) {int sockfd, len;struct sockaddr_in dest;char buffer[MAXBUF + 1];fd_set rfds;struct timeval tv;int retval, maxfd = -1;if (argc != 3) {printf("參數(shù)格式錯誤！正確用法如下：\n\t\t%s IP地址 端口\n\t比如:\t%s 127.0.0.1 80\n此程序用來從某個 IP 地址的服務(wù)器某個端口接收最多 MAXBUF 個字節(jié)的消息",argv[0], argv[0]);exit(0);}/* 創(chuàng)建一個 socket 用于 tcp 通信 */if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {perror("Socket");exit(errno);}/* 初始化服務(wù)器端（對方）的地址和端口信息 */bzero(&dest, sizeof(dest));dest.sin_family = AF_INET;dest.sin_port = htons(atoi(argv[2]));if (inet_aton(argv[1], (struct in_addr *) &dest.sin_addr.s_addr) == 0) {perror(argv[1]);exit(errno);}/* 連接服務(wù)器 */if (connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {perror("Connect ");exit(errno);}printf("\n準(zhǔn)備就緒，可以開始聊天了……直接輸入消息回車即可發(fā)信息給對方\n");while (1) {/* 把集合清空 */FD_ZERO(&rfds);/* 把標(biāo)準(zhǔn)輸入句柄0加入到集合中 */FD_SET(0, &rfds);maxfd = 0;/* 把當(dāng)前連接句柄sockfd加入到集合中 */FD_SET(sockfd, &rfds);if (sockfd > maxfd)maxfd = sockfd;/* 設(shè)置最大等待時間 */tv.tv_sec = 1;tv.tv_usec = 0;/* 開始等待 */retval = select(maxfd + 1, &rfds, NULL, NULL, &tv);if (retval == -1) {printf("將退出，select出錯！ %s", strerror(errno));break;} else if (retval == 0) {/* printf("沒有任何消息到來，用戶也沒有按鍵，繼續(xù)等待……\n"); */continue;} else {if (FD_ISSET(sockfd, &rfds)) {/* 連接的socket上有消息到來則接收對方發(fā)過來的消息并顯示 */bzero(buffer, MAXBUF + 1);/* 接收對方發(fā)過來的消息，最多接收 MAXBUF 個字節(jié) */len = recv(sockfd, buffer, MAXBUF, 0);if (len > 0)printf("接收消息成功:'%s'，共%d個字節(jié)的數(shù)據(jù)\n",buffer, len);else {if (len < 0)printf("消息接收失敗！錯誤代碼是%d，錯誤信息是'%s'\n",errno, strerror(errno));elseprintf("對方退出了，聊天終止！\n");break;}}if (FD_ISSET(0, &rfds)) {/* 用戶按鍵了，則讀取用戶輸入的內(nèi)容發(fā)送出去 */bzero(buffer, MAXBUF + 1);fgets(buffer, MAXBUF, stdin);if (!strncasecmp(buffer, "quit", 4)) {printf("自己請求終止聊天！\n");break;}/* 發(fā)消息給服務(wù)器 */len = send(sockfd, buffer, strlen(buffer) - 1, 0);if (len < 0) {printf("消息'%s'發(fā)送失敗！錯誤代碼是%d，錯誤信息是'%s'\n",buffer, errno, strerror(errno));break;} elseprintf("消息：%s\t發(fā)送成功，共發(fā)送了%d個字節(jié)！\n",buffer, len);}}}/* 關(guān)閉連接 */close(sockfd);return 0; } 編譯用如下命令：gcc -Wall async-server.c -o server
gcc -Wall async-client.c -o client
運(yùn)行用如下命令：
./server 7838 1
./client 127.0.0.1 7838
下面還是一個簡單的并發(fā)處理的服務(wù)器，但是僅僅是能多路復(fù)用I/O，但是不能同時處理連接請求，因為這只有一個線程
http://www.cnblogs.com/faraway/archive/2009/03/06/1404449.html
使用select函數(shù)可以以非阻塞的方式和多個socket通信。程序只是演示select函數(shù)的使用，功能非常簡單，即使某個連接關(guān)閉以后也不會修改當(dāng)前連接數(shù)，連接數(shù)達(dá)到最大值后會終止程序。
1. 程序使用了一個數(shù)組fd_A，通信開始后把需要通信的多個socket描述符都放入此數(shù)組。
2. 首先生成一個叫sock_fd的socket描述符，用于監(jiān)聽端口。
3. 將sock_fd和數(shù)組fd_A中不為0的描述符放入select將檢查的集合fdsr。
4. 處理fdsr中可以接收數(shù)據(jù)的連接。如果是sock_fd，表明有新連接加入，將新加入連接的socket描述符放置到fd_A。
#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <errno.h>#include <string.h>#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#define MYPORT 1234 // the port users will be connecting to#define BACKLOG 5 // how many pending connections queue will hold#define BUF_SIZE 200int fd_A[BACKLOG]; // accepted connection fdint conn_amount; // current connection amountvoid showclient(){int i;printf("client amount: %d\n", conn_amount);for (i = 0; i < BACKLOG; i++) {printf("[%d]:%d ", i, fd_A[i]);}printf("\n\n");}int main(void){int sock_fd, new_fd; // listen on sock_fd, new connection on new_fdstruct sockaddr_in server_addr; // server address informationstruct sockaddr_in client_addr; // connector's address informationsocklen_t sin_size;int yes = 1;char buf[BUF_SIZE];int ret;int i;if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {perror("socket");exit(1);}if (setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) {perror("setsockopt");exit(1);}server_addr.sin_family = AF_INET; // host byte orderserver_addr.sin_port = htons(MYPORT); // short, network byte orderserver_addr.sin_addr.s_addr = INADDR_ANY; // automatically fill with my IPmemset(server_addr.sin_zero, '\0', sizeof(server_addr.sin_zero));if (bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {perror("bind");exit(1);}if (listen(sock_fd, BACKLOG) == -1) {perror("listen");exit(1);}printf("listen port %d\n", MYPORT);fd_set fdsr;int maxsock;struct timeval tv;conn_amount = 0;sin_size = sizeof(client_addr);maxsock = sock_fd;while (1) {// initialize file descriptor setFD_ZERO(&fdsr);FD_SET(sock_fd, &fdsr);// timeout settingtv.tv_sec = 30;tv.tv_usec = 0;// add active connection to fd setfor (i = 0; i < BACKLOG; i++) {if (fd_A[i] != 0) {FD_SET(fd_A[i], &fdsr);}}ret = select(maxsock + 1, &fdsr, NULL, NULL, &tv);if (ret < 0) {perror("select");break;} else if (ret == 0) {printf("timeout\n");continue;}// check every fd in the setfor (i = 0; i < conn_amount; i++) {if (FD_ISSET(fd_A[i], &fdsr)) {ret = recv(fd_A[i], buf, sizeof(buf), 0);if (ret <= 0) { // client closeprintf("client[%d] close\n", i);close(fd_A[i]);FD_CLR(fd_A[i], &fdsr);fd_A[i] = 0;} else { // receive dataif (ret < BUF_SIZE)memset(&buf[ret], '\0', 1);printf("client[%d] send:%s\n", i, buf);}}}// check whether a new connection comesif (FD_ISSET(sock_fd, &fdsr)) {new_fd = accept(sock_fd, (struct sockaddr *)&client_addr, &sin_size);if (new_fd <= 0) {perror("accept");continue;}// add to fd queueif (conn_amount < BACKLOG) {fd_A[conn_amount++] = new_fd;printf("new connection client[%d] %s:%d\n", conn_amount,inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));if (new_fd > maxsock)maxsock = new_fd;}else {printf("max connections arrive, exit\n");send(new_fd, "bye", 4, 0);close(new_fd);break;}}showclient();}// close other connectionsfor (i = 0; i < BACKLOG; i++) {if (fd_A[i] != 0) {close(fd_A[i]);}}exit(0);}
有網(wǎng)友提出了點改進(jìn)：
accept函數(shù)里面fd_A[conn_amount++] = new_fd;可以稍加改進(jìn)，按照樓主的意圖，會出現(xiàn)當(dāng)一個用戶不斷連接再斷開的情況下，當(dāng)連接次數(shù)超過maxconnection的時候，就會退出，因此fd_A[i]沒有很好的利用,不能實現(xiàn)動態(tài)管理，我建議僅將conn_amount僅作為客戶端連接數(shù)，而不是有連接就增加，當(dāng)accept成功的時候，就加1，當(dāng)recv＝0的時候就減1；建議將fd_A[conn_amount++] = new_fd;這句程序改為
for(i = 0;i < MAXCLIENT;i++)
{
if(fd[i] == 0)
{
fd[i] = new_fd;
break;
}
}
conn_amount++;
這樣就可以重復(fù)利用fd[i]的空間；
另外在recv返回值<=0的時候，加一句conn_amount++;
還有一點，超過最大連接數(shù)的時候break應(yīng)該為continue，這樣會更人性化一點，客戶端太多關(guān)閉它的請求就行了，沒必要自毀，這樣整個系統(tǒng)就可以動態(tài)與客戶端實現(xiàn)連接，
總結(jié)
以上是生活随笔為你收集整理的linux--select的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 矩阵基本概念
- 下一篇： linux脚本--输出占用内存最多的进程