如何判断带超时的Winsock TCP连接是否成功建立

问题描述 #

我用以下C++代码实现TCP/IP客户端，实际使用时会循环检测直至服务器就绪。该代码在连接服务器前会按0.2、0.6、1的顺序循环输出，但成功连接后无任何输出，也无错误信息，几乎无代码分支被执行。请问如何判断TCP/IP连接已成功建立？

int connect_timeout(int & socket, struct sockaddr * name, int namelen, timeval timeout)
{
    unsigned long mode = 1;
    int result= ioctlsocket(socket, FIONBIO, &mode);
    if(NO_ERROR != result)
    {
        //std::cout << "0.0---------------" << std::endl;
        return -1;
    }
    else
    {
        //std::cout << "0.1---------------" << std::endl;
    }

    result= connect(socket, name, namelen);
    if(SOCKET_ERROR == result)
    {
        int errono = WSAGetLastError();
        if(WSAEWOULDBLOCK == errono)
        {
            std::cout << "0.2---------------" << std::endl;
            errono = 0;
        }
        else
        {
            //std::cout << "0.3---------------" << std::endl;
            mode = 0;
            ioctlsocket(socket, FIONBIO, &mode);
            return -1;
        }
    }
    else
    {
        std::cout << "0.4---------------" << std::endl;
    }

    mode = 0;
    result= ioctlsocket(socket, FIONBIO, &mode);
    if(0 < result)
    {
        //error
        std::cout << "0.5---------------" << std::endl;
        return -1;
    }
    else
    {
        std::cout << "0.6---------------" << std::endl;
    }
    
    fd_set readFd, writeFd, errFd;
    FD_ZERO(&readFd);
    FD_ZERO(&writeFd);
    FD_ZERO(&errFd);
    FD_SET(socket, &readFd);
    FD_SET(socket, &writeFd);
    FD_SET(socket, &errFd);
    int sockNum = select(socket + 1, &readFd, &writeFd, &errFd, &timeout);
    if(0 == sockNum)
    {
        //timeout
        std::cout << "1---------------" << std::endl;
        return -1;
    }
    else if(FD_ISSET(socket, &readFd) || FD_ISSET(socket, &writeFd) )
    {
        std::cout << "2---------------" << std::endl;
    }
    else
    {
        //error
        std::cout << "3---------------" << std::endl;
        return -1;
    }
    std::cout << "4---------------" << std::endl;

    return 0;
}

问题分析 #

你的代码存在两个核心逻辑错误，导致无法正确判断连接状态：

1. 过早将socket改回阻塞模式：在非阻塞connect发起后，立刻调用ioctlsocket把socket改回阻塞模式，这会导致后续的select无法正确检测连接的完成状态——阻塞模式下的socket在select中的行为和非阻塞模式完全不同，甚至可能直接阻塞住后续操作。
2. 缺少连接状态的最终验证：即使select检测到socket可读或可写，也不能直接判定连接成功——这种情况可能是连接成功，也可能是连接失败（比如服务器拒绝连接），必须通过getsockopt获取SO_ERROR选项来确认最终状态。

修复方案及正确的连接判断逻辑 #

正确的非阻塞TCP连接判断流程应该是：

1. 将socket设置为非阻塞模式。
2. 调用connect，如果直接返回成功，说明连接已建立；如果返回WSAEWOULDBLOCK，则进入等待流程。
3. 使用select监听socket的可写、错误事件，等待连接完成。
4. 当select触发事件后，通过getsockopt(socket, SOL_SOCKET, SO_ERROR, ...)获取错误码：
   • 错误码为0，说明连接成功。
   • 错误码非0，说明连接失败。
5. 确认连接成功后，再根据需要将socket改回阻塞模式。

修改后的代码示例 #

#include <winsock2.h>
#include <iostream>

int connect_timeout(int & socket, struct sockaddr * name, int namelen, timeval timeout)
{
    unsigned long mode = 1;
    int result = ioctlsocket(socket, FIONBIO, &mode);
    if (NO_ERROR != result)
    {
        return -1;
    }

    result = connect(socket, name, namelen);
    if (SOCKET_ERROR != result)
    {
        // 连接直接成功，改回阻塞模式（如果需要）
        mode = 0;
        ioctlsocket(socket, FIONBIO, &mode);
        std::cout << "连接成功（直接返回）---------------" << std::endl;
        return 0;
    }

    int err = WSAGetLastError();
    if (WSAEWOULDBLOCK != err)
    {
        // 连接直接失败，清理并返回
        mode = 0;
        ioctlsocket(socket, FIONBIO, &mode);
        return -1;
    }

    // 进入select等待连接完成
    fd_set writeFd, errFd;
    FD_ZERO(&writeFd);
    FD_ZERO(&errFd);
    FD_SET(socket, &writeFd);
    FD_SET(socket, &errFd);

    int sockNum = select(socket + 1, nullptr, &writeFd, &errFd, &timeout);
    if (0 == sockNum)
    {
        // 超时
        std::cout << "连接超时---------------" << std::endl;
        return -1;
    }
    else if (FD_ISSET(socket, &errFd))
    {
        // 错误事件触发，连接失败
        int err_code;
        int len = sizeof(err_code);
        getsockopt(socket, SOL_SOCKET, SO_ERROR, (char*)&err_code, &len);
        std::cout << "连接失败，错误码：" << err_code << std::endl;
        mode = 0;
        ioctlsocket(socket, FIONBIO, &mode);
        return -1;
    }
    else if (FD_ISSET(socket, &writeFd))
    {
        // 可写事件触发，验证连接是否成功
        int err_code;
        int len = sizeof(err_code);
        getsockopt(socket, SOL_SOCKET, SO_ERROR, (char*)&err_code, &len);
        if (0 == err_code)
        {
            // 连接成功，改回阻塞模式（如果需要）
            mode = 0;
            ioctlsocket(socket, FIONBIO, &mode);
            std::cout << "连接成功---------------" << std::endl;
            return 0;
        }
        else
        {
            std::cout << "连接失败，错误码：" << err_code << std::endl;
            mode = 0;
            ioctlsocket(socket, FIONBIO, &mode);
            return -1;
        }
    }

    // 意外分支
    mode = 0;
    ioctlsocket(socket, FIONBIO, &mode);
    return -1;
}

关键修改说明 #

• 移除了过早将socket改回阻塞模式的代码，直到确认连接成功后才进行模式切换。
• 去掉了对读事件的监听：非阻塞connect完成时，通常触发的是可写事件，读事件一般在连接失败时才会触发，但通过错误事件监听已经可以覆盖这种情况。
• 增加了getsockopt获取SO_ERROR的步骤，这是判断非阻塞connect是否成功的核心操作——即使select触发了可写事件，也必须通过这个选项确认没有错误发生。
• 明确了各个分支的输出信息，方便调试时确认连接状态。

内容的提问来源于stack exchange，提问作者taichi