码迷，mamicode.com
首页 > Windows程序 > 详细

windows和linux套接字中的select机制浅析

时间：2014-06-25 23:51:35      阅读：404      评论：0      收藏：0      [点我收藏+]

标签：des   class   blog   http   tar   com   

先来谈谈为什么会出现select函数，也就是select是解决什么问题的？

平常使用的recv函数时阻塞的，也就是如果没有数据可读，recv就会一直阻塞在那里，这是如果有另外一个连接过来，就得一直等待，这样实时性就不是太好。

这个问题的几个解决方法：1. 使用ioctlsocket函数，将recv函数设置成非阻塞的，这样不管套接字上有没有数据都会立刻返回，可以重复调用recv函数，这种方式叫做轮询（polling），但是这样效率很是问题，因为，大多数时间实际上是无数据可读的，花费时间不断反复执行read系统调用，这样就比较浪费CPU的时间。并且循环之间的间隔不好确定。2. 使用fork，使用多进程来解决，这里终止会比较复杂（待研究）。 3.使用多线程来解决，这样避免了终止的复杂性，但却要求处理线程之间的同步，在减少复杂性方面这可能会得不偿失。4. 使用异步IO（待研究）。5. 就是本文所使用的I/O多路转接（多路复用）--其实就是在套接字阻塞和非阻塞之间做了一个均衡，我们称之为半阻塞。

经过对select的初步了解，在windows和linux下的实现小有区别，所以分开来写。这里先写windows下的select机制。

select的大概思想：将多个套接字放在一个集合里，然后统一检查这些套接字的状态（可读、可写、异常等），调用select后，会更新这些套接字的状态，然后做判断，如果套接字可读，就执行read操作。这样就巧妙地避免了阻塞，达到同时处理多个连接的目的。当然如果没有事件发生，select会一直阻塞，如果不想一直让它等待，想去处理其它事情，可以设置一个最大的等待时间。

/***********************************************************************************************************/

下面具体讲讲函数的参数，参见MSDN的解释http://msdn.microsoft.com/en-us/library/windows/desktop/ms740141(v=vs.85).aspx：

 

[cpp] view plaincopy
 
  1. int select(  
  2.   _In_     int nfds,  
  3.   _Inout_  fd_set *readfds,  
  4.   _Inout_  fd_set *writefds,  
  5.   _Inout_  fd_set *exceptfds,  
  6.   _In_     const struct timeval *timeout  
  7. );  


函数的返回值，表示准备好的套接字的个数，如果是0，则表示没有一个准备好（超时就是一种情况），如果是-1（SOCKET_ERROR），表示有错误发生，可以使用WSAGetLastError（）函数来得到错误代码，从而知道是什么错误。

 

函数的参数，第一个是输入参数nfds，表示满足条件的套接字的个数，windows下可以设置为0，因为fd_set结构体中已经包含了这个参数，这个参数已经是多余的了，之所以还存在，只是是为了与FreeBSD兼容。

第二三四参数都是输入输出参数（值-结果参数，输入和输出会不一样），表示套接字的可读、可写和异常三种状态的集合。调用select之后，如果指定套接字不可读或者不可写，就会从相应队列中清除，这样就可以判断哪些套接字可读或者可写。 

说明一下，这里的可读性是指：如果有客户的连接请求到达，套接口就是可读的，调用accept能够立即完成，而不发生阻塞；如果套接口接收队列缓冲区中的字节数大于0，调用recv或者recvfrom就不会阻塞。可写性是指，可以向套接字发送数据（套接字创建成功后，就是可写的）。当然不是套接字可写就会去发送数据，就像不是看到电话就去打电话一样，而是由打电话的需求了，才去看电话是否可打；可读就不一样了，电话响了，自然要去接电话（除非，你有事忙或者不想接，一般都是要接的)。可读已经包含了缓冲区中有数据可以读取，可写只是说明了缓冲区有空间让你写，你需不需要写就要看你有没有数据要写了.关于异常，就是指一些意外情况，自己用的比较少，以后用到了，再过来补上。

第五个参数是等待的最大时间，是一个结构体：struct timeval，它的定义是：

 

[cpp] view plaincopy
 
  1. /* 
  2. * Structure used in select() call, taken from the BSD file sys/time.h. 
  3. */  
  4. struct timeval {  
  5.         long    tv_sec;         /* seconds */  
  6.         long    tv_usec;        /* and microseconds */  
  7. };  

具体到秒和微妙，按照等待的时间长短可以分为不等待、等待一定时间、一直等待。对应的设置分别为，（0，0）是不等待，这是select是非阻塞的，（x,y）最大等待时间x秒y微妙（如果有事件就会提前返回，而不继续等待），NULL表示一直等待，直到有事件发生。这里可以将timeout分别设置成0（不阻塞）或者1微妙（阻塞很短的时间），然后观察CPU的使用率，会发现设置成非阻塞后，CPU的使用率已下载就上升到了50%左右，这样可以看出非阻塞占用CPU很多，但利用率不高。

 

/***********************************************************************************************************/

跟select配合使用的几个宏和fd_set结构体介绍：

套接字描述符为了方便管理是放在一个集合里的，这个集合是fd_set，它的具体定义是：

 

[cpp] view plaincopy
 
  1. typedef struct fd_set {  
  2.         u_int   fd_count;               /* how many are SET? */  
  3.         SOCKET  fd_array[FD_SETSIZE];   /* an array of SOCKETs */  
  4. } fd_set;  

fd_count是集合中已经设置的套接口描述符的数量。fd_array数组保存已经设置的套接口描述符，其中FD_SETSIZE的定义是：

 

 

[cpp] view plaincopy
 
  1. #ifndef FD_SETSIZE  
  2. #define FD_SETSIZE      64  
  3. #endif /* FD_SETSIZE */  

这个默认值在一般的程序中已经够用，如果需要，可以将其更改为更大的值。

 

集合的管理操作，比如元素的清空、加入、删除以及判断元素是否在集合中都是用宏来完成的。四个宏是：

 

[html] view plaincopy
 
  1. FD_ZERO(*set)  
  2. FD_SET(s, *set)  
  3. FD_ISSET(s, *set)  
  4. FD_CLR(s, *set)  

下面一一介绍这些宏的作用和定义：

 

FD_ZERO(*set)，是把集合清空（初始化为0，确切的说，是把集合中的元素个数初始化为0，并不修改描述符数组).使用集合前，必须用FD_ZERO初始化，否则集合在栈上作为自动变量分配时，fd_set分配的将是随机值，导致不可预测的问题。它的宏定义如下：

 

[cpp] view plaincopy
 
  1. #define FD_ZERO(set) (((fd_set FAR *)(set))->fd_count=0)  

FD_SET(s,*set)，向集合中加入一个套接口描述符（如果该套接口描述符s没在集合中，并且数组中已经设置的个数小于最大个数时，就把该描述符加入到集合中，集合元素个数加1）。这里是将s的值直接放入数组中。它的宏定义如下：

 

 

[cpp] view plaincopy
 
  1. #define FD_SET(fd, set) do { \  
  2.     if (((fd_set FAR *)(set))->fd_count < FD_SETSIZE) \  
  3.         ((fd_set FAR *)(set))->fd_array[((fd_set FAR *)(set))->fd_count++]=(fd);\  
  4. } while(0)  

FD_ISSET(s,*set)，检查描述符是否在集合中，如果在集合中返回非0值，否则返回0. 它的宏定义并没有给出具体实现，但实现的思路很简单，就是搜索集合，判断套接字s是否在数组中。它的宏定义是：

 

 

[cpp] view plaincopy
 
  1. #define FD_ISSET(fd, set) __WSAFDIsSet((SOCKET)(fd), (fd_set FAR *)(set))  

FD_CLR(s,*set)，从集合中移出一个套接口描述符（比如一个套接字连接中断后，就应该移除它）。实现思路是，在数组集合中找到对应的描述符，然后把后面的描述依次前移一个位置，最后把描述符的个数减1. 它的宏定义是：

 

 

[cpp] view plaincopy
 
  1. #define FD_CLR(fd, set) do { \  
  2.     u_int __i; \  
  3.     for (__i = 0; __i < ((fd_set FAR *)(set))->fd_count ; __i++) { \  
  4.         if (((fd_set FAR *)(set))->fd_array[__i] == fd) { \  
  5.             while (__i < ((fd_set FAR *)(set))->fd_count-1) { \  
  6.                 ((fd_set FAR *)(set))->fd_array[__i] = \  
  7.                     ((fd_set FAR *)(set))->fd_array[__i+1]; \  
  8.                 __i++; \  
  9.             } \  
  10.             ((fd_set FAR *)(set))->fd_count--; \  
  11.             break; \  
  12.         } \  
  13.     } \  
  14. } while(0)  


/***********************************************************************************************************/

 

至此，一些基础的点基本就讲完了，然后给出大概流程和一个示例：

1.调用FD_ZERO来初始化套接字状态；

2.调用FD_SET将感兴趣的套接字描述符加入集合中（每次循环都要重新加入，因为select更新后，会将一些没有满足条件的套接字移除队列）；

3.设置等待时间后，调用select函数--更新套接字的状态；

4.调用FD_ISSET，来判断套接字是否有相应状态，然后做相应操作，比如，如果套接字可读，就调用recv函数去接收数据。

关键技术：套接字队列和状态的表示与处理。

server端得程序如下（套接字管理队列一个很重要的作用就是保存套接字描述符，因为accept得到的套接字描述符会覆盖掉原来的套接字描述符，而readfs中的描述符在select后会删除这些套接字描述符）：

 

[cpp] view plaincopy
 
  1. // server.cpp :   
  2. //程序中加入了套接字管理队列，这样管理起来更加清晰、方便，当然也可以不用这个东西  
  3.   
  4. #include "winsock.h"  
  5. #include "stdio.h"  
  6. #pragma comment (lib,"wsock32.lib")  
  7. struct socket_list{  
  8.     SOCKET MainSock;  
  9.     int num;  
  10.     SOCKET sock_array[64];  
  11. };  
  12. void init_list(socket_list *list)  
  13. {  
  14.     int i;  
  15.     list->MainSock = 0;  
  16.     list->num = 0;  
  17.     for(i = 0;i < 64;i ++){  
  18.         list->sock_array[i] = 0;  
  19.     }  
  20. }  
  21. void insert_list(SOCKET s,socket_list *list)  
  22. {  
  23.     int i;  
  24.     for(i = 0;i < 64; i++){  
  25.         if(list->sock_array[i] == 0){  
  26.             list->sock_array[i] = s;  
  27.             list->num += 1;  
  28.             break;  
  29.         }  
  30.     }  
  31. }  
  32. void delete_list(SOCKET s,socket_list *list)  
  33. {  
  34.     int i;  
  35.     for(i = 0;i < 64; i++){  
  36.         if(list->sock_array[i] == s){  
  37.             list->sock_array[i] = 0;  
  38.             list->num -= 1;  
  39.             break;  
  40.         }  
  41.     }  
  42. }  
  43. void make_fdlist(socket_list *list,fd_set *fd_list)  
  44. {  
  45.     int i;  
  46.     FD_SET(list->MainSock,fd_list);  
  47.     for(i = 0;i < 64;i++){  
  48.         if(list->sock_array[i] > 0){  
  49.             FD_SET(list->sock_array[i],fd_list);  
  50.         }  
  51.     }  
  52. }  
  53. int main(int argc, char* argv[])  
  54. {  
  55.     SOCKET s,sock;  
  56.     struct sockaddr_in ser_addr,remote_addr;  
  57.     int len;  
  58.     char buf[128];  
  59.     WSAData wsa;  
  60.     int retval;  
  61.     struct socket_list sock_list;  
  62.     fd_set readfds,writefds,exceptfds;  
  63.     timeval timeout;        //select的最多等待时间，防止一直等待  
  64.     int i;  
  65.     unsigned long arg;  
  66.   
  67.     WSAStartup(0x101,&wsa);  
  68.     s = socket(AF_INET,SOCK_STREAM,0);  
  69.     ser_addr.sin_family = AF_INET;  
  70.     ser_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);  
  71.     ser_addr.sin_port = htons(0x1234);  
  72.     bind(s,(sockaddr*)&ser_addr,sizeof(ser_addr));  
  73.   
  74.     listen(s,5);  
  75.     timeout.tv_sec = 5;     //如果套接字集合中在1s内没有数据，select就会返回，超时select返回0  
  76.     timeout.tv_usec = 0;  
  77.     init_list(&sock_list);  
  78.     FD_ZERO(&readfds);  
  79.     FD_ZERO(&writefds);  
  80.     FD_ZERO(&exceptfds);  
  81.     sock_list.MainSock = s;  
  82.     arg = 1;  
  83.     ioctlsocket(sock_list.MainSock,FIONBIO,&arg);  
  84.     while(1){  
  85.         make_fdlist(&sock_list,&readfds);  
  86.         //make_fdlist(&sock_list,&writefds);  
  87.         //make_fdlist(&sock_list,&exceptfds);  
  88.   
  89.         retval = select(0,&readfds,&writefds,&exceptfds,&timeout);     //超过这个时间，就不阻塞在这里，返回一个0值。  
  90.         if(retval == SOCKET_ERROR){  
  91.             retval = WSAGetLastError();  
  92.             break;  
  93.         }  
  94.         else if(retval == 0) {  
  95.             printf("select() is time-out! There is no data or new-connect coming!\n");  
  96.             continue;  
  97.         }  
  98.         if(FD_ISSET(sock_list.MainSock,&readfds)){  
  99.             len = sizeof(remote_addr);  
  100.             sock = accept(sock_list.MainSock,(sockaddr*)&remote_addr,&len);  
  101.             if(sock == SOCKET_ERROR)  
  102.                 continue;  
  103.             printf("accept a connection\n");  
  104.             insert_list(sock,&sock_list);  
  105.         }  
  106.         for(i = 0;i < 64;i++){  
  107.             if(sock_list.sock_array[i] == 0)  
  108.                 continue;  
  109.             sock = sock_list.sock_array[i];  
  110.             if(FD_ISSET(sock,&readfds)){  
  111.                 retval = recv(sock,buf,128,0);  
  112.                 if(retval == 0){  
  113.                     closesocket(sock);  
  114.                     printf("close a socket\n");  
  115.                     delete_list(sock,&sock_list);  
  116.                     continue;  
  117.                 }else if(retval == -1){  
  118.                     retval = WSAGetLastError();  
  119.                     if(retval == WSAEWOULDBLOCK)  
  120.                         continue;  
  121.                     closesocket(sock);  
  122.                     printf("close a socket\n");  
  123.                     delete_list(sock,&sock_list);   //连接断开后，从队列中移除该套接字  
  124.                     continue;  
  125.                 }  
  126.                 buf[retval] = 0;  
  127.                 printf("->%s\n",buf);  
  128.                 send(sock,"ACK by server",13,0);  
  129.             }  
  130.             //if(FD_ISSET(sock,&writefds)){  
  131.             //}  
  132.             //if(FD_ISSET(sock,&exceptfds)){  
  133.               
  134.         }  
  135.         FD_ZERO(&readfds);  
  136.         FD_ZERO(&writefds);  
  137.         FD_ZERO(&exceptfds);  
  138.     }  
  139.     closesocket(sock_list.MainSock);  
  140.     WSACleanup();  
  141.     return 0;  
  142. }  

 

关于linux下的select跟windows下的区别还有待学习。

 

参考书籍：

《WinSock网络编程经络》第19章

《UNIX环境高级编程》

windows和linux套接字中的select机制浅析,布布扣,bubuko.com

windows和linux套接字中的select机制浅析

标签：des   class   blog   http   tar   com   

原文地址：http://www.cnblogs.com/lidabo/p/3804411.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！