下面來介紹判斷非阻塞SOCKET是否已經斷開的幾種方法:
注意要區分不同操作系統分別進行測試, 包括WINDOWS, LINUX和UNIX會各有不同。
在WINDOWS下比較簡單,可以使用FD_CLOSE事件判斷SOCKET是否已經斷開
view plaincopy to clipboardprint? bool IsSocketClosed(SOCKET clientSocket) { bool ret = false; HANDLE closeEvent = WSACreateEvent(); WSAEventSelect(clientSocket, closeEvent, FD_CLOSE); DWORD dwRet = WaitForSingleObject(closeEvent, 0); if(dwRet == WSA_WAIT_EVENT_0) ret = true; else if(dwRet == WSA_WAIT_TIMEOUT) ret = false; WSACloseEvent(closeEvent); return ret; }
在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 plaincopy to clipboardprint? #include <errno.h> bool IsSocketClosed(int clientSocket) { char buff[32]; int recvBytes = recv(clientSocket, buff, sizeof(buff), MSG_PEEK); int sockErr = errno; //cout << "In close function, recv " << recvBytes << " bytes, err " << sockErr << endl; if( recvBytes > 0) //Get data return false; if( (recvBytes == -1) && (sockErr == EWOULDBLOCK) ) //No receive data return false; return true; }
此外,所有操作系統上還可以通過TCP_KEEPLIVE實行心跳檢測