Linux 网络编程二：非阻塞select

函数声明

select函数

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);
说明：
int maxfdp是一个整数值，是指集合中所有文件描述符的范围，即所有文件描述符的最大值加1。
fd_set *readfds是指向fd_set结构的指针，这个集合中应该包括文件描述符，我们是要监视这些文件描述符的读变化的，即我们关心是否可以从这些文件中读取数据了，如果这个集合中有一个文件可读，select就会返回一个大于0的值，表示有文件可读，如果没有可读的文件，则根据timeout参数再判断是否超时，若超出timeout的时间，select返回0，若发生错误返回负值。可以传入NULL值，表示不关心任何文件的读变化。 
fd_set *writefds是指向fd_set结构的指针，这个集合中应该包括文件描述符，我们是要监视这些文件描述符的写变化的，即我们关心是否可以向这些文件中写入数据了，如果这个集合中有一个文件可写，select就会返回一个大于0的值，表示有文件可写，如果没有可写的文件，则根据timeout参数再判断是否超时，若超出timeout的时间，select返回0，若发生错误返回负值。可以传入NULL值，表示不关心任何文件的写变化。 
fd_set *errorfds同上面两个参数的意图，用来监视文件错误异常。 
struct timeval* timeout是select的超时时间，这个参数至关重要，它可以使select处于三种状态，第一，若将NULL以形参传入，即不传入时间结构，就是将select置于阻塞状态，一定等到监视文件描述符集合中某个文件描述符发生变化为止；第二，若将时间值设为0秒0毫秒，就变成一个纯粹的非阻塞函数，不管文件描述符是否有变化，都立刻返回继续执行，文件无变化返回0，有变化返回一个正值；第三，timeout的值大于0，这就是等待的超时时间，即select在timeout时间内阻塞，超时时间之内有事件到来就返回了，否则在超时后不管怎样一定返回，返回值同上述。 
返回值： 
负值：select错误 正值：某些文件可读写或出错 0：等待超时，没有可读写或错误的文件 
在有了select后可以写出像样的网络程序来！举个简单的例子，就是从网络上接受数据写入一个文件中。 
说明套接字读写条件：
下列四个条件中的任何一个满足时，套接口准备好读：
（1） 套接口接收缓冲区中的数据字节数大于等于套接口接收缓冲区低潮限度的当前值。可以通过SO_REVILOAT来设置此低潮限度。
（2）连接的读这一半关闭，也就是接收了FIN的TCP连接，
（3）套接口是一个监听套接口且已完成的连接数为非0.
（4）有一个套接口错误等处理。
下列三个条件中的任一个满足时，套接口准备好写：
（1） 套接口发送缓冲区的可用空间字节娄大于等于套接口发送缓冲区低潮限度的当前值且或者（i）套接口已连接，或者（ii）套接口不要求连接。
（2）连接的写这一半关闭，对这样的套接口写操作将产生信号SIGPIEP。
（3）有一个套接口错误待处理。

描述符集

FD_ZERO(fd_set *fdset)：清空fdset与所有文件句柄的联系。 
FD_SET(int fd, fd_set *fdset)：建立文件句柄fd与fdset的联系。 
FD_CLR(int fd, fd_set *fdset)：清除文件句柄fd与fdset的联系。 
FD_ISSET(int fd, fdset *fdset)：检查fdset联系的文件句柄fd是否可读写，>0表示可读写。 

代码示例（转载）

转载自点击打开链接

服务器端

[cpp] view plain copy

1. /*使用select函数可以以非阻塞的方式和多个socket通信。程序只是演示select函数的使用，即使某个连接关闭以后也不会修改当前连接数，连接数达到最大值后会终止程序。 
2. 1. 程序使用了一个数组fd，通信开始后把需要通信的多个socket描述符都放入此数组 
3. 2. 首先生成一个叫sock_fd的socket描述符，用于监听端口。 
4. 3. 将sock_fd和数组fd中不为0的描述符放入select将检查的集合fdsr。 
5. 4. 处理fdsr中可以接收数据的连接。如果是sock_fd，表明有新连接加入，将新加入连接的socket描述符放置到fd。 */  
6. // select_server.c  
7. #include <stdio.h>  
8. #include <stdlib.h>  
9. #include <unistd.h>  
10. #include <errno.h>  
11. #include <string.h>  
12. #include <sys/types.h>  
13. #include <sys/socket.h>  
14. #include <sys/time.h>  
15. #include <netinet/in.h>  
16. #include <arpa/inet.h>  
17. #define MYPORT 1234 //连接时使用的端口  
18. #define MAXCLINE 5 //连接队列中的个数  
19. #define BUF_SIZE 200  
20. int fd[MAXCLINE]; //连接的fd  
21. int conn_amount; //当前的连接数  
22. void showclient()  
23. {  
24. int i;  
25. printf("client amount:%d\n",conn_amount);  
26. for(i=0;i<MAXCLINE;i++)  
27. {  
28. printf("[%d]:%d ",i,fd[i]);  
29. }  
30. printf("\n\n");  
31. }  
32. int main(void)  
33. {  
34. int sock_fd,new_fd; //监听套接字 连接套接字  
35. struct sockaddr_in server_addr; // 服务器的地址信息  
36. struct sockaddr_in client_addr; //客户端的地址信息  
37. socklen_t sin_size;  
38. int yes = 1;  
39. char buf[BUF_SIZE];  
40. int ret;  
41. int i;  
42. //建立sock_fd套接字  
43. if((sock_fd = socket(AF_INET,SOCK_STREAM,0))==-1)  
44. {  
45. perror("setsockopt");  
46. exit(1);  
47. }  
48. //设置套接口的选项 SO_REUSEADDR 允许在同一个端口启动服务器的多个实例  
49. // setsockopt的第二个参数SOL SOCKET 指定系统中，解释选项的级别 普通套接字  
50. if(setsockopt(sock_fd,SOL_SOCKET,SO_REUSEADDR,&yes,sizeof(int))==-1)  
51. {  
52. perror("setsockopt error \n");  
53. exit(1);  
54. }  
55. server_addr.sin_family = AF_INET; //主机字节序  
56. server_addr.sin_port = htons(MYPORT);  
57. server_addr.sin_addr.s_addr = INADDR_ANY;//通配IP  
58. memset(server_addr.sin_zero,'\0',sizeof(server_addr.sin_zero));  
59. if(bind(sock_fd,(struct sockaddr *)&server_addr,sizeof(server_addr)) == -1)  
60. {  
61. perror("bind error!\n");  
62. exit(1);  
63. }  
64. if(listen(sock_fd,MAXCLINE)==-1)  
65. {  
66. perror("listen error!\n");  
67. exit(1);  
68. }  
69. printf("listen port %d\n",MYPORT);  
70. fd_set fdsr; //文件描述符集的定义  
71. int maxsock;  
72. struct timeval tv;  
73. conn_amount =0;  
74. sin_size = sizeof(client_addr);  
75. maxsock = sock_fd;  
76. while(1)  
77. {  
78. //初始化文件描述符集合  
79. FD_ZERO(&fdsr); //清除描述符集  
80. FD_SET(sock_fd,&fdsr); //把sock_fd加入描述符集  
81. //超时的设定  
82. tv.tv_sec = 30;  
83. tv.tv_usec =0;  
84. //添加活动的连接  
85. for(i=0;i<MAXCLINE;i++)  
86. {  
87. if(fd[i]!=0)  
88. {  
89. FD_SET(fd[i],&fdsr);  
90. }  
91. }  
92. //如果文件描述符中有连接请求 会做相应的处理，实现I/O的复用 多用户的连接通讯  
93. ret = select(maxsock +1,&fdsr,NULL,NULL,&tv);  
94. if(ret <0) //没有找到有效的连接 失败  
95. {  
96. perror("select error!\n");  
97. break;  
98. }  
99. else if(ret ==0)// 指定的时间到，  
100. {  
101. printf("timeout \n");  
102. continue;  
103. }  
104. //循环判断有效的连接是否有数据到达  
105. for(i=0;i<conn_amount;i++)  
106. {  
107. if(FD_ISSET(fd[i],&fdsr))  
108. {  
109. ret = recv(fd[i],buf,sizeof(buf),0);  
110. if(ret <=0) //客户端连接关闭，清除文件描述符集中的相应的位  
111. {  
112. printf("client[%d] close\n",i);  
113. close(fd[i]);  
114. FD_CLR(fd[i],&fdsr);  
115. fd[i]=0;  
116. conn_amount--;  
117. }  
118. //否则有相应的数据发送过来 ，进行相应的处理  
119. else  
120. {  
121. if(ret <BUF_SIZE)  
122. memset(&buf[ret],'\0',1);  
123. printf("client[%d] send:%s\n",i,buf);  
124. }  
125. }  
126. }  
127. if(FD_ISSET(sock_fd,&fdsr))  
128. {  
129. new_fd = accept(sock_fd,(struct sockaddr *)&client_addr,&sin_size);  
130. if(new_fd <=0)  
131. {  
132. perror("accept error\n");  
133. continue;  
134. }  
135. //添加新的fd 到数组中 判断有效的连接数是否小于最大的连接数，如果小于的话，就把新的连接套接字加入集合  
136. if(conn_amount <MAXCLINE)  
137. {  
138. for(i=0;i< MAXCLINE;i++)  
139. {  
140. if(fd[i]==0)  
141. {  
142. fd[i] = new_fd;  
143. break;  
144. }  
145. }  
146. conn_amount++;  
147. printf("new connection client[%d]%s:%d\n",conn_amount,inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));  
148. if(new_fd > maxsock)  
149. {  
150. maxsock = new_fd;  
151. }  
152. }  
153. else  
154. {  
155. printf("max connections arrive ,exit\n");  
156. send(new_fd,"bye",4,0);  
157. close(new_fd);  
158. continue;  
159. }  
160. }  
161. showclient();  
162. }  
163. for(i=0;i<MAXCLINE;i++)  
164. {  
165. if(fd[i]!=0)  
166. {  
167. close(fd[i]);  
168. }  
169. }  
170. exit(0);  
171. }  

客户端

[cpp] view plain copy

1. //客户端的一个简单的实现，只是为了证实一下，服务器端程序的正确性  
2. //select_client.c  
3. #include <stdio.h>  
4. #include <stdlib.h>  
5. #include <errno.h>  
6. #include <string.h>  
7. #include <netdb.h>  
8. #include <sys/types.h>  
9. #include <netinet/in.h>  
10. #include <sys/socket.h>  
11. #include <sys/types.h>  
12. #define MAXDATASIZE 100  
13. #define SERVPORT 1234  
14. #define MAXLINE 1024  
15. int main(int argc,char *argv[])  
16. {  
17. int sockfd,sendbytes;  
18. // char send[MAXLINE];  
19. char send[MAXLINE];  
20. char buf[MAXDATASIZE];  
21. struct hostent *host;  
22. struct sockaddr_in serv_addr;  
23. if(argc <2)  
24. {  
25. fprintf(stderr,"Please enter the server's hostname\n");  
26. exit(1);  
27. }  
28. if((host = gethostbyname(argv[1])) == NULL)  
29. {  
30. perror("gethostbyname");  
31. exit(1);  
32. }  
33. if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1)  
34. {  
35. perror("socket error \n");  
36. exit(1);  
37. }  
38. serv_addr.sin_family = AF_INET;  
39. serv_addr.sin_port = htons(SERVPORT);  
40. serv_addr.sin_addr = *((struct in_addr *)host->h_addr);  
41. bzero(&(serv_addr.sin_zero),8);  
42. if(connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(struct sockaddr)) ==-1)  
43. {  
44. perror("connect \n");  
45. exit(1);  
46. }  
47. while(fgets(send,1024,stdin)!=NULL)  
48. {  
49. if((sendbytes = write(sockfd,send,100)) ==-1)  
50. {  
51. perror("send error \n");  
52. exit(1);  
53. }  
54. }  
55. close(sockfd);  
56. }