在Win32中,有没有一种方法可以测试套接字是否非阻塞?

问题描述 投票:10回答:3

在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上是否有其他可用来确定套接字当前是否处于非阻塞模式的调用?

windows winapi sockets winsock nonblocking
3个回答
7
投票

一个更长的答案是:不,但是您通常会知道它是否正确,因为它的定义相对明确。

除非所有套接字都用ioctlsocket()显式FIONBIO或将它们交给WSAAsyncSelectWSAEventSelect,否则所有套接字都是阻塞的。后两个函数“秘密地”将套接字更改为非阻塞。

因为您知道是否已经调用了这三个函数之一,所以即使您无法查询状态,也仍然可以知道。一个明显的例外是,如果该套接字来自某个第三方库,而您不知道该套接字到底对套接字做了什么。

旁注:有趣的是,套接字可以同时被阻塞和重叠,这看起来并不直观,但是这是有道理的,因为它们来自相反的范式(准备与完成)。


5
投票

以前,您可以调用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;
}

0
投票

我同意接受的答案,没有正式的方法可以确定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 */
}

注意事项:

  • 使用UDP套接字
  • 与连接的TCP套接字一起使用
  • 不适用于未连接的TCP套接字
© www.soinside.com 2019 - 2024. All rights reserved.