Winsock2套接字断开重连致应用挂起问题排查求助

问题背景 #

基于Winsock2开发的阻塞式TCP客户端/服务器应用中，当同一客户端身份从另一台设备重复连接服务器时，服务器会检测到该情况并正常断开原连接的套接字。但原连接的客户端无法感知到连接已断开，在尝试关闭自身时出现挂起现象。同一设备内重复连接断开则无此问题。

提供的核心代码：

连接函数 #

bool connectSocket(SOCKET &sock, const char *serverIp, const unsigned int serverPort) {

    sockaddr_in serverAddress;

    memset(&serverAddress, 0x00, sizeof(sockaddr_in));

    serverAddress.sin_family = AF_INET;
    serverAddress.sin_addr.s_addr = inet_addr(serverIp);
    serverAddress.sin_port = htons(serverPort);

    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if(sock == INVALID_SOCKET)
        return false;

    if(connect(sock, (SOCKADDR*) &serverAddress, sizeof(serverAddress)) == SOCKET_ERROR) {
        printf("connect socket error %u\n", WSAGetLastError());
        return false;
    }
}

断开函数 #

void disconnectSocket(SOCKET &sock) {

    if(sock != INVALID_SOCKET) {

        if(closesocket(sock) == SOCKET_ERROR)
            printf("close socket error %u\n", WSAGetLastError());

        sock = INVALID_SOCKET;
    }
}

核心原因分析 #

1. 阻塞套接字的被动感知特性：阻塞模式下，客户端只有主动发起recv/send等IO操作时，才能检测到服务器已断开连接（返回SOCKET_ERROR，错误码通常为WSAECONNRESET或WSAESHUTDOWN）。若服务器断开后，客户端未执行任何IO操作，直到调用closesocket，此时套接字可能处于TCP半关闭状态或存在未处理的IO请求，导致closesocket阻塞。
2. 跨设备网络的状态同步延迟：同一设备内的TCP断开包传递无网络延迟，套接字状态同步即时；跨设备时，网络延迟可能导致客户端未及时收到服务器发送的FIN包，无法触发连接断开的检测逻辑。
3. 代码逻辑漏洞：connectSocket函数缺少return true;语句，连接成功时返回值未定义，可能导致后续逻辑误判连接状态；disconnectSocket直接调用closesocket，未处理套接字可能存在的未完成IO操作，若有线程正阻塞在该套接字的IO调用上，closesocket无法中断阻塞，引发挂起。

解决思路 #

1. 主动检测连接状态 #

通过select或带MSG_PEEK参数的recv定期检查套接字状态，及时感知连接断开：

// 示例：用select检测可读事件（连接断开时会触发可读）
fd_set readSet;
FD_ZERO(&readSet);
FD_SET(sock, &readSet);
timeval timeout = {5, 0}; // 5秒超时
int ret = select(0, &readSet, NULL, NULL, &timeout);
if (ret > 0 && FD_ISSET(sock, &readSet)) {
    char buf[1];
    int recvRet = recv(sock, buf, 1, MSG_PEEK);
    if (recvRet == 0 || (recvRet == SOCKET_ERROR && WSAGetLastError() != WSAEWOULDBLOCK)) {
        // 连接已断开，执行断开逻辑
        disconnectSocket(sock);
    }
}

2. 优化套接字关闭流程 #

在closesocket前先调用shutdown终止TCP连接，读取剩余数据确保状态同步，避免阻塞：

void disconnectSocket(SOCKET &sock) {
    if(sock != INVALID_SOCKET) {
        // 关闭发送方向，通知服务器终止连接
        shutdown(sock, SD_SEND);
        // 读取剩余数据，确保接收FIN包并处理
        char buf[1024];
        while (recv(sock, buf, sizeof(buf), 0) > 0);
        // 关闭套接字
        if(closesocket(sock) == SOCKET_ERROR)
            printf("close socket error %u\n", WSAGetLastError());
        sock = INVALID_SOCKET;
    }
}

若存在阻塞在IO操作的线程，需调用WSACancelBlockingCall中断阻塞调用，避免关闭时挂起。

3. 修复连接函数的逻辑漏洞 #

补充connectSocket的成功返回语句，确保连接状态判断正确：

bool connectSocket(SOCKET &sock, const char *serverIp, const unsigned int serverPort) {
    // ... 原有代码 ...
    if(connect(sock, (SOCKADDR*) &serverAddress, sizeof(serverAddress)) == SOCKET_ERROR) {
        printf("connect socket error %u\n", WSAGetLastError());
        return false;
    }
    return true; // 补充成功返回
}

4. 设置IO超时避免无限阻塞 #

通过setsockopt给阻塞套接字设置读写超时，确保IO操作不会永久挂起：

bool connectSocket(SOCKET &sock, const char *serverIp, const unsigned int serverPort) {
    // ... 原有代码 ...
    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if(sock == INVALID_SOCKET)
        return false;
    // 设置接收超时为5秒
    DWORD timeout = 5000;
    setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout, sizeof(timeout));
    // 设置发送超时为5秒
    setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, (const char*)&timeout, sizeof(timeout));
    // ... 后续连接代码 ...
}

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