下面來介紹判斷非阻塞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實行心跳檢測
