判断非阻塞SOCKET是否已经断开的几种方法

注意要区分不同操作系统分别进行测试, 包括WINDOWS, LINUX和UNIX会各有不同

在WINDOWS下比较简单,可以使用FD_CLOSE事件判断SOCKET是否已经断开
  1. bool IsSocketClosed(SOCKET clientSocket)  
  2. {  
  3.  bool ret = false;  
  4.  HANDLE closeEvent = WSACreateEvent();  
  5.  WSAEventSelect(clientSocket, closeEvent, FD_CLOSE);  
  6.   
  7.  DWORD dwRet = WaitForSingleObject(closeEvent, 0);  
  8.    
  9.  if(dwRet == WSA_WAIT_EVENT_0)  
  10.   ret = true;  
  11.  else if(dwRet == WSA_WAIT_TIMEOUT)  
  12.   ret = false;  
  13.   
  14.  WSACloseEvent(closeEvent);  
  15.  return ret;  
  16. }  
 

在UNIX/LINUX下,非阻塞模式SOCKET可以采用recv+MSG_PEEK的方式进行判断,其中MSG_PEEK保证了仅仅进行状态判断,而不影响数据接收

对于主动关闭的SOCKET, recv返回-1,而且errno被置为9(#define EBADF   9 /* Bad file number */)

或104 (#define ECONNRESET 104 /* Connection reset by peer */)

对于被动关闭的SOCKET,recv返回0,而且errno被置为11(#define EWOULDBLOCK EAGAIN /* Operation would block */)

对正常的SOCKET, 如果有接收数据,则返回>0, 否则返回-1,而且errno被置为11(#define EWOULDBLOCK EAGAIN /* Operation would block */)

因此对于简单的状态判断(不过多考虑异常情况),
    recv返回>0,   正常
    返回-1,而且errno被置为11  正常
    其它情况    关闭
 
  1. #include <errno.h>  
  2.   
  3. bool IsSocketClosed(int clientSocket)  
  4. {  
  5.  char buff[32];  
  6.  int recvBytes = recv(clientSocket, buff, sizeof(buff), MSG_PEEK);  
  7.    
  8.  int sockErr = errno;  
  9.    
  10.  //cout << "In close function, recv " << recvBytes << " bytes, err " << sockErr << endl;  
  11.    
  12.  if( recvBytes > 0) //Get data  
  13.   return false;  
  14.    
  15.  if( (recvBytes == -1) && (sockErr == EWOULDBLOCK) ) //No receive data  
  16.   return false;  
  17.      
  18.  return true;  
  19. }  
 

此外,所有操作系统上还可以通过TCP_KEEPLIVE实行心跳检测
Logo

更多推荐