C 判断socket 是否已经断开

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

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

在WINDOWS下比较简单，可以使用FD_CLOSE事件判断SOCKET是否已经断开

view plain copy to clipboard print ?

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  正常
    其它情况    关闭
 

view plain copy to clipboard print ?

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实行心跳检测