Windows Socket客户端连接断开后无法自动重连问题排查

问题根源 #

你的代码核心问题在于对TCP连接断开的检测逻辑有误：

• TCP是面向流的可靠协议，当服务器主动断开（比如Ctrl+C关闭netcat），客户端的send调用不会立刻返回错误——只要发送的数据还在本地缓冲区，send会正常返回发送的字节数，直到缓冲区满或者后续操作触发TCP的FIN/RST检测。
• Connect方法未处理连接失败时的socket清理，也未正确初始化fConnected状态；KeepAlive线程盲目调用Connect后直接Send，不管连接是否真的成功。

具体修复步骤 #

1. 修复Connect方法 #

连接失败时必须关闭无效socket避免资源泄漏，连接成功后才设置fConnected为true：

bool Client::Connect() {
    // 先关闭旧socket
    if (fSocket != INVALID_SOCKET) {
        closesocket(fSocket);
        fSocket = INVALID_SOCKET;
    }

    fSocket = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, nullptr, 0, WSA_FLAG_OVERLAPPED);
    if (fSocket == INVALID_SOCKET) {
        return false;
    }

    // 可选：设置非阻塞模式，避免连接超时阻塞线程
    u_long mode = 1;
    ioctlsocket(fSocket, FIONBIO, &mode);

    int result = WSAConnect(fSocket, (SOCKADDR*)&fAddress, sizeof(fAddress), nullptr, nullptr, nullptr, nullptr);
    if (result == SOCKET_ERROR) {
        int err = WSAGetLastError();
        // 非阻塞连接会返回WSAEWOULDBLOCK，这里只处理明确的连接失败
        if (err != WSAEWOULDBLOCK) {
            closesocket(fSocket);
            fSocket = INVALID_SOCKET;
            return false;
        }
    }

    fConnected = true;
    return true;
}

2. 改进KeepAlive线程逻辑 #

线程需判断Connect返回值，仅连接成功后发送消息；同时添加主动连接检测：

void Client::StartConnectionAndKeepAlive() {
    auto threadFunction = [this]() {
        while (true) {
            lock_guard<mutex> lock(fMutex); // 必须加锁保护共享变量
            if (!fConnected) {
                cout << "Attempting to reconnect..." << endl;
                if (this->Connect()) {
                    this->Send("Hello from thread!\n");
                }
            } else {
                // 用MSG_PEEK主动检测连接，不消耗接收缓冲区数据
                char buf[1];
                int result = recv(fSocket, buf, sizeof(buf), MSG_PEEK);
                // recv返回0表示对方已断开，返回错误且不是WSAEWOULDBLOCK则判定连接失效
                if (result == 0 || (result == SOCKET_ERROR && WSAGetLastError() != WSAEWOULDBLOCK)) {
                    fConnected = false;
                    cout << "Connection lost, marking for reconnect..." << endl;
                }
            }
            this_thread::sleep_for(chrono::seconds(1));
        }
    };
    fThread = new thread(threadFunction);
}

3. 修正Send方法 #

添加共享变量锁保护，处理分块发送和潜在错误：

bool Client::Send(const string& data) {
    lock_guard<mutex> lock(fMutex); // 加锁避免多线程操作冲突
    if (fSocket == INVALID_SOCKET || !fConnected) {
        return false;
    }

    int totalSent = 0;
    int remaining = data.size();
    const char* buf = data.c_str();

    while (remaining > 0) {
        int nBytesSent = send(fSocket, buf + totalSent, remaining, 0);
        if (nBytesSent == SOCKET_ERROR) {
            int err = WSAGetLastError();
            // 非阻塞模式下WSAEWOULDBLOCK可重试，这里直接判定连接失效
            if (err != WSAEWOULDBLOCK) {
                fConnected = false;
                cout << "Send failed with error: " << err << endl;
                return false;
            }
            this_thread::sleep_for(chrono::milliseconds(10));
        } else {
            totalSent += nBytesSent;
            remaining -= nBytesSent;
        }
    }

    cout << "Sent " << totalSent << " bytes: " << data << endl;
    return true;
}

4. 完善CloseConnection方法 #

确保正确关闭socket并重置状态：

void Client::CloseConnection() {
    lock_guard<mutex> lock(fMutex);
    if (fSocket != INVALID_SOCKET) {
        closesocket(fSocket);
        fSocket = INVALID_SOCKET;
    }
    fConnected = false;
}

关键知识点补充 #

• TCP连接断开检测：仅靠send无法及时检测连接断开，必须结合recv(MSG_PEEK)或者通过setsockopt启用TCP KeepAlive机制。
• 多线程同步：fConnected、fSocket等共享变量必须用std::mutex保护，否则会出现竞态条件导致逻辑混乱。
• 非阻塞socket：推荐使用非阻塞模式，避免连接或发送操作阻塞整个线程，提升程序响应性。

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