socket中的select函数使用


socket select()函数用于定时检查以创建的socket句柄的状态，以确定设定的socket句柄当前是否可写或有数据可读。

select函数提供了一种方法，使得程序在操作socket时（如recv操作），无需因阻塞而等待直至超时。特别是在一个线程中操作多个socket时，对多个socket逐一操作直至超时将浪费大量时间。

select机制则是同时对多个socket句柄进行监控，一旦存在可操作的socket,函数及返回并通知程序。

select函数声明：
int select(
 int nfds,
 fd_set* readfds,
 fd_set* writefds,
 fd_set* exceptfds,
 const struct timeval* timeout
);

fd_set是一个结构体，包含了一个SOCKET数组及一个整数用于表示数组中有效的SOCKET句柄数。

readfds参数，用于指示select函数：当其中包含的socket句柄可读时，函数返回。
writefds和exceptfds则分别表示其中包含的socket句柄可写或异常时，函数返回。

timeout是一个时间值结构，表示当前select函数执行时的超时时间，如果所有socket暂时都没有可操作事件发生，则函数将一直等待直至超时，如果timeout设置为NULL时，select()函数则一直等到包含的socket存在操作事件时才返回。

select函数的返回值是一个整数。
如果执行异常，比如，指定的socket句柄是非法的等等，函数将返回SOCKET_ERROR，也就是-1.
如果select函数执行超时，即在指定的timeout时间内，没有可操作的socket出现，那么返回0.
其余情况函数将返回一个大于0的整数，用于指明可操作的socket总数。

由于select只返回一个表示总数的整数，那么如果一次select多个socket句柄，或者函数执行时即指定了readfds，也指定了writefds，函数本身将无法告诉程序到底哪个socket句柄可读或可写。

所以在使用select时，尽量只指定检查一个操作状态。
不过对于同步阻塞的socket操作，writefds参数用到的机会可能会比较少。

示例：

  1. SOCKET hsock = ...  
  2.  .....  
  3.   
  4.     fd_set rds, wrs, exs;  
  5.   
  6.     FD_ZERO(&rds);     // 初始化   
  7.     FD_ZERO(&wrs);  
  8.     FD_ZERO(&exs);  
  9.   
  10.     FD_SET(hsock, &rds);    //设置socket句柄到select的FD参数中   
  11.     FD_SET(hsock, &wrs);  
  12.     FD_SET(hsock, &exs);  
  13.   
  14.     timeval timeout;  
  15.     timeout.tv_sec  = 5000;  
  16.     timeout.tv_usec = 0;  
  17.     while (int ret = select(1, NULL, &wrs, NULL, &timeout))  
  18.     {  
  19.         if ( ret == 0 ) {  
  20.             cout<<"Time Expired"<<endl;  
  21.         } else if ( ret == SOCKET_ERROR) {  
  22.             cout<<"Socket Error:"<<WSAGetLastError()<<endl;  
  23.             break;  
  24.         } else {  
  25.             cout<<"Total: "<<ret<<endl;  
  26.             break;  
  27.         }  
  28.     }  

相关内容