在Win32中,有没有一种方法可以测试套接字是否非阻塞?
在POSIX系统下,我将执行以下操作:
int is_non_blocking(int sock_fd) {
flags = fcntl(sock_fd, F_GETFL, 0);
return flags & O_NONBLOCK;
}
但是,Windows套接字不支持fcntl()。非阻塞模式是使用带有FIONBIO的ioctl的set,但似乎没有办法使用ioctl get当前的非阻塞模式。
Windows上是否有其他可用来确定套接字当前是否处于非阻塞模式的调用?
一个更长的答案是:不,但是您通常会知道它是否正确,因为它的定义相对明确。
除非所有套接字都用ioctlsocket()
显式FIONBIO
或将它们交给WSAAsyncSelect
或WSAEventSelect
,否则所有套接字都是阻塞的。后两个函数“秘密地”将套接字更改为非阻塞。
因为您知道是否已经调用了这三个函数之一,所以即使您无法查询状态,也仍然可以知道。一个明显的例外是,如果该套接字来自某个第三方库,而您不知道该套接字到底对套接字做了什么。
旁注:有趣的是,套接字可以同时被阻塞和重叠,这看起来并不直观,但是这是有道理的,因为它们来自相反的范式(准备与完成)。
以前,您可以调用WSAIsBlocking来确定。如果您要管理旧代码,则仍然可以选择。
否则,您可以在套接字API上编写一个简单的抽象层。由于默认情况下所有套接字都处于阻塞状态,因此您可以维护一个内部标志,并通过API强制所有套接字操作,以便您始终了解状态。
这里是一个跨平台的代码段,用于设置/获取阻止模式,尽管它不能完全满足您的要求:
/// @author Stephen Dunn
/// @date 10/12/15
bool set_blocking_mode(const int &socket, bool is_blocking)
{
bool ret = true;
#ifdef WIN32
/// @note windows sockets are created in blocking mode by default
// currently on windows, there is no easy way to obtain the socket's current blocking mode since WSAIsBlocking was deprecated
u_long flags = is_blocking ? 0 : 1;
ret = NO_ERROR == ioctlsocket(socket, FIONBIO, &flags);
#else
const int flags = fcntl(socket, F_GETFL, 0);
if ((flags & O_NONBLOCK) && !is_blocking) { info("set_blocking_mode(): socket was already in non-blocking mode"); return ret; }
if (!(flags & O_NONBLOCK) && is_blocking) { info("set_blocking_mode(): socket was already in blocking mode"); return ret; }
ret = 0 == fcntl(socket, F_SETFL, is_blocking ? flags ^ O_NONBLOCK : flags | O_NONBLOCK);
#endif
return ret;
}
我同意接受的答案,没有正式的方法可以确定Windows上套接字的阻塞状态。如果您从第三方获得套接字(例如,您是TLS库,并且从上层获得套接字),则无法确定套接字是否处于阻塞状态。
尽管如此,对于这个对我长期有效的问题,我有一个可行的,非正式的,有限的解决方案。
我尝试从套接字读取0字节。如果它是一个阻塞套接字,它将返回0,如果它是一个非阻塞套接字,它将返回-1,并且GetLastError等于WSAEWOULDBLOCK。
int IsBlocking(SOCKET s)
{
int r = 0;
unsigned char b[1];
r = recv(s, b, 0, 0);
if (r == 0)
return 1;
else if (r == -1 && GetLastError() == WSAEWOULDBLOCK)
return 0;
return -1; /* In case it is a connection socket (TCP) and it is not in connected state you will get here 10060 */
}
注意事项: