Client向Server send数据,返回 WSAEWOULDBLOCK 错误

本文介绍如何处理Winsock异常WSAEWOULDBLOCK,包括其含义及应对策略,通过等待FD_WRITE事件来确保数据发送的正确性和效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参见MSDN中的一段:

The FD_WRITE network event is handled slightly differently. An FD_WRITE network event is recorded when a socket is first connected with connect/ WSAConnect or accepted with accept/ WSAAccept, and then after a send fails with WSAEWOULDBLOCK and buffer space becomes available. Therefore, an application can assume that sends are possible starting from the first FD_WRITE network event setting and lasting until a send returns WSAEWOULDBLOCK. After such a failure the application will find out that sends are again possible when an FD_WRITE network event is recorded and the associated event object is set.

参考http://blog.youkuaiyun.com/strikebone/article/details/1765839这篇文章,如下解释:

首先,Winsock 异常 WSAEWOULDBLOCK (WSAGetLastError) 的意思是 Output Buffer 已经满了,无法再写入数据。确切的说它其实不算是个错误,出现这种异常的绝大部分时候其实都不存在 Output Buffer 已满情况,而是处于一种“忙”的状态,而这种“忙”的状态还很大程度上是由于接收方造成的。意思就是你要发送的对象,对方收的没你发的快或者对方的接受缓冲区已被填满,所以就返回你一个“忙”的标志,而这时你再发多少数据都没任何意义,所以系统就返回 WSAEWOULDBLOCK 异常通知你。

那么,该怎么办呢?网上有很多朋友的做法是遇到这种情况就 Sleep 一段时间,一般短暂停顿后 Output Buffer 就空出来了,那就又可以继续发送了。推荐的方法:根据 MSDN 文档所示,当出现 WSAEWOULDBLOCK 异常后直到空出 Output Buffer 时,系统会发送一个 FD_WRITE 给发送方。我们完全可以在等收到 FD_WRITE 消息后再重新发送从出现异常开始的数据包即可(该包需要全部重新发送)。

至此,该问题结案。最后顺便提一下:FD_WRITE 消息会在至少三钟情况下出现,而上面只是其中的一种,所以我建议给 Socket 做个标志判断以便于规范性。

客户端可能的实现如下:

 

static BOOL ComWrite(PSFTHREAD pSFThread, size_t length, int type, size_t bufsize, void *buf, WCHAR *fname)
{
    BOOL bRet;

    PSENDLOG_PRIVATE_DATA ses = (PSENDLOG_PRIVATE_DATA)pSFThread->privateData;
    PSMON_ENV pEnv = pSFThread->pEnv;

    DWORD dwEvent;
    WSANETWORKEVENTS nEvents;
    int nRet;
    size_t send_size = 0;
    size_t read_size = 0;
    HANDLE fd;
    char *fileBuf = NULL;
    DWORD fileSize;
    DWORD rBytes;

    HANDLE hEvents[] = {
        GET_EXIT_EVENT(pEnv),
        ses->socEvent
    };

    // Parameter check
    if(type >= SFCOM_INPUTTYPE_TAILNUM){
        return FALSE;
    }
    if(bufsize == 0){
        bufsize = length;
    }

    if(type == SFCOM_INPUTTYPE_FILE){
        fd = CreateFile(fname,
                        GENERIC_READ | GENERIC_WRITE,
                        FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
                        NULL,
                        OPEN_EXISTING,
                        FILE_ATTRIBUTE_NORMAL,
                        NULL);
        if(fd == INVALID_HANDLE_VALUE){
            SFERRLOG();
            return FALSE;
        }
        fileSize = GetFileSize(fd, NULL);
        if(fileSize == -1){
            goto Cleanup;
        }else if(fileSize == 0){
            // File empty
            goto Cleanup;
        }
        fileBuf = (char*)malloc(fileSize);
        if(fileBuf == NULL){
            goto Cleanup;
        }
        bRet = ReadFile(fd, fileBuf, fileSize, &rBytes, NULL);
        if(!bRet || rBytes != fileSize){
            goto Cleanup;
        }
        buf = fileBuf;
    }

    size_t length_remained = length;
    while(1){
        nRet = send(ses->soc, (char*)buf + send_size, (int)min(length_remained, SFCOM_TCP_SEND_MAX_SIZE), 0);
        if(nRet == SOCKET_ERROR){
            if(WSAGetLastError() != WSAEWOULDBLOCK){
                // Error
                SFCOMDBGOUT("Error occured:%d\n", WSAGetLastError());
                goto Cleanup;
            }
            // WSAEWOULDBLOCK WaitForMultipleObjects FD_WRITE
            SFCOMDBGOUT("ComWrite send WSAEWOULDBLOCK");
        }else{
            if(nRet <= length_remained) length_remained -= nRet;

            send_size += nRet;
            SFCOMDBGOUT("send %dbytes(total %dbytes)\n", nRet, send_size);

            if(send_size == length){
                return TRUE;
            }else if(send_size < length){
                continue;
            }
        }

        dwEvent = WaitForMultipleObjects(2, hEvents, FALSE, INFINITE);
        if(dwEvent == WAIT_OBJECT_0){
            // Exit thread event.
            ses->exitEventFlag = TRUE;
            goto Cleanup;
        }else if(dwEvent == WAIT_OBJECT_0 + 1){
            // Socket event.
            nRet = WSAEnumNetworkEvents(ses->soc, ses->socEvent, &nEvents);
            if(nRet == SOCKET_ERROR){
                // Error
                SFCOMDBGOUT("ComWrite Error occured:%d\n", WSAGetLastError());
                goto Cleanup;
            }
            if(nEvents.lNetworkEvents & FD_CLOSE){
                // FD_CLOSE
                SFCOMDBGOUT("ComWrite FD_CLOSE\n");
                goto Cleanup;
            }
            if((nEvents.lNetworkEvents & FD_WRITE) && (nEvents.iErrorCode[FD_WRITE_BIT] == 0)){
                continue;
            }else{
                goto Cleanup;
            }
        }else if(dwEvent == WAIT_TIMEOUT){
            // Timeout
            SFCOMDBGOUT("ComWrite WAIT_TIMEOUT\n");
            goto Cleanup;
        }else{
            // Error
            SFCOMDBGOUT("ComWrite WAIT_FAILED\n");
            goto Cleanup;
        }
    }

    return TRUE;

Cleanup:
    if(type == SFCOM_INPUTTYPE_FILE){
        CloseHandle(fd);
        if(fileBuf != NULL){
            free(fileBuf);
        }
    }

    ComClose(ses);
    return FALSE;
}


 

#include "Communication.h" #include <iostream> CommunicationBase::CommunicationBase() : m_socket(INVALID_SOCKET), m_running(false), m_receiveThread(NULL), m_queueEvent(NULL) { InitializeCriticalSection(&m_queueCS); m_queueEvent = CreateEvent(NULL, FALSE, FALSE, NULL); } CommunicationBase::~CommunicationBase() { CloseSocket(); DeleteCriticalSection(&m_queueCS); if (m_queueEvent) CloseHandle(m_queueEvent); } void CommunicationBase::CloseSocket() { if (m_socket != INVALID_SOCKET) { m_running = false; // 关闭套接字会终止接收线程 shutdown(m_socket, SD_BOTH); closesocket(m_socket); m_socket = INVALID_SOCKET; if (m_receiveThread) { WaitForSingleObject(m_receiveThread, INFINITE); CloseHandle(m_receiveThread); m_receiveThread = NULL; } } } bool CommunicationBase::SendMessageTcp(const std::string& message) { if (m_socket == INVALID_SOCKET || !m_running) return false; const char* data = message.c_str(); int length = static_cast<int>(message.size()); int totalSent = 0; while (totalSent < length) { int sent = send(m_socket, data + totalSent, length - totalSent, 0); if (sent == SOCKET_ERROR) { std::cerr << "发送失败: " << WSAGetLastError() << std::endl; return false; } totalSent += sent; } return true; } //bool SendMessageTcp(SOCKET socket, const std::string& message) { // const char* data = message.data(); // int total = static_cast<int>(message.size()); // int sent = 0; // // while (sent < total && m_running) { // int bytes = send(socket, data + sent, total - sent, 0); // // if (bytes == SOCKET_ERROR) { // int error = WSAGetLastError(); // if (error == WSAEWOULDBLOCK) { // // 等待套接字可写 // fd_set writeSet; // FD_ZERO(&writeSet); // FD_SET(socket, &writeSet); // // timeval timeout{ 0, 10000 }; // 10ms超时 // if (select(0, NULL, &writeSet, NULL, &timeout) <= 0) { // continue; // 超时或错误 // } // } // else { // // 处理其他错误 // closesocket(socket); // return false; // } // } // else { // sent += bytes; // } // } // return sent == total; //} std::string CommunicationBase::ReceiveMessage(DWORD timeout_ms) { EnterCriticalSection(&m_queueCS); if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); WaitForSingleObject(m_queueEvent, timeout_ms); EnterCriticalSection(&m_queueCS); } if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); return ""; } std::string msg = m_messageQueue.front(); m_messageQueue.pop(); LeaveCriticalSection(&m_queueCS); return msg; } DWORD WINAPI CommunicationBase::ReceiveThreadProc(LPVOID lpParam) { CommunicationBase* pThis = reinterpret_cast<CommunicationBase*>(lpParam); char buffer[4096]; int bytesReceived; while (pThis->m_running) { bytesReceived = recv(pThis->m_socket, buffer, sizeof(buffer) - 1, 0); if (bytesReceived == SOCKET_ERROR) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(10); continue; } break; } if (bytesReceived <= 0) { break; } buffer[bytesReceived] = '\0'; std::string msg(buffer); EnterCriticalSection(&pThis->m_queueCS); pThis->m_messageQueue.push(msg); SetEvent(pThis->m_queueEvent); LeaveCriticalSection(&pThis->m_queueCS); } pThis->m_running = false; return 0; } Server::Server(int port) : m_port(port), m_acceptThread(NULL) { InitializeCriticalSection(&m_clientsCS); m_socket = INVALID_SOCKET; } Server::~Server() { Stop(); DeleteCriticalSection(&m_clientsCS); } bool Server::Start() { WSADATA wsaData; if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) { std::cerr << "WSAStartup失败" << std::endl; return false; } m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (m_socket == INVALID_SOCKET) { std::cerr << "创建套接字失败" << std::endl; WSACleanup(); return false; } sockaddr_in serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_addr.s_addr = INADDR_ANY; serverAddr.sin_port = htons(m_port); if (bind(m_socket, (sockaddr*)& serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) { std::cerr << "绑定失败: " << WSAGetLastError() << std::endl; closesocket(m_socket); WSACleanup(); return false; } if (listen(m_socket, SOMAXCONN) == SOCKET_ERROR) { std::cerr << "监听失败" << std::endl; closesocket(m_socket); WSACleanup(); return false; } // 设置非阻塞模式 u_long mode = 1; ioctlsocket(m_socket, FIONBIO, &mode); m_running = true; // 启动接受客户端线程 m_acceptThread = CreateThread(NULL, 0, AcceptThreadProc, this, 0, NULL); if (m_acceptThread == NULL) { std::cerr << "创建接受线程失败" << std::endl; closesocket(m_socket); WSACleanup(); return false; } std::cout << "服务器已启动,监听端口: " << m_port << std::endl; return true; } void Server::Stop() { m_running = false; // 关闭所有客户端套接字 EnterCriticalSection(&m_clientsCS); for (size_t i = 0; i < m_clientSockets.size(); ++i) { if (m_clientSockets[i] != INVALID_SOCKET) { shutdown(m_clientSockets[i], SD_BOTH); closesocket(m_clientSockets[i]); } } m_clientSockets.clear(); LeaveCriticalSection(&m_clientsCS); CloseSocket(); if (m_acceptThread) { WaitForSingleObject(m_acceptThread, INFINITE); CloseHandle(m_acceptThread); m_acceptThread = NULL; } WSACleanup(); std::cout << "服务器已停止" << std::endl; } DWORD WINAPI Server::AcceptThreadProc(LPVOID lpParam) { Server* pServer = reinterpret_cast<Server*>(lpParam); pServer->AcceptClients(); return 0; } DWORD WINAPI Server::ClientThreadProc(LPVOID lpParam) { ClientThreadParam* param = reinterpret_cast<ClientThreadParam*>(lpParam); param->server->HandleClient(param->clientSocket); delete param; return 0; } void Server::AcceptClients() { while (m_running) { sockaddr_in clientAddr; int addrLen = sizeof(clientAddr); SOCKET clientSocket = accept(m_socket, (sockaddr*)& clientAddr, &addrLen); if (clientSocket == INVALID_SOCKET) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(100); continue; } break; } // 设置非阻塞模式a u_long mode = 1; ioctlsocket(clientSocket, FIONBIO, &mode); EnterCriticalSection(&m_clientsCS); m_clientSockets.push_back(clientSocket); LeaveCriticalSection(&m_clientsCS); char ipStr[INET_ADDRSTRLEN]; inet_ntop(AF_INET, &clientAddr.sin_addr, ipStr, INET_ADDRSTRLEN); std::cout << "新客户端连接: " << ipStr << ":" << ntohs(clientAddr.sin_port) << std::endl; // 创建线程参数 ClientThreadParam* param = new ClientThreadParam; param->server = this; param->clientSocket = clientSocket; // 创建客户端处理线程 HANDLE hThread = CreateThread( NULL, 0, ClientThreadProc, param, 0, NULL ); if (hThread == NULL) { std::cerr << "创建客户端线程失败: " << GetLastError() << std::endl; delete param; closesocket(clientSocket); } else { CloseHandle(hThread); // 立即关闭线程句柄,不保留 } } } void Server::HandleClient(SOCKET clientSocket) { char buffer[4096]; int bytesReceived; while (m_running) { bytesReceived = recv(clientSocket, buffer, sizeof(buffer) - 1, 0); if (bytesReceived == SOCKET_ERROR) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(10); continue; } break; } if (bytesReceived <= 0) { break; } buffer[bytesReceived] = '\0'; std::string msg(buffer); // 添加到消息队列 EnterCriticalSection(&m_queueCS); m_messageQueue.push(msg); SetEvent(m_queueEvent); LeaveCriticalSection(&m_queueCS); std::cout << "收到消息: " << msg << std::endl; // 广播给其他客户端 Broadcast(msg, clientSocket); } // 移除断开连接的客户端 EnterCriticalSection(&m_clientsCS); for (std::vector<SOCKET>::iterator it = m_clientSockets.begin(); it != m_clientSockets.end(); ++it) { if (*it == clientSocket) { closesocket(clientSocket); m_clientSockets.erase(it); break; } } LeaveCriticalSection(&m_clientsCS); std::cout << "客户端断开连接" << std::endl; } void Server::StartReceiver() { // 服务器使用每个客户端单独的线程处理接收 } void Server::Broadcast(const std::string& message, SOCKET excludeSocket) { EnterCriticalSection(&m_clientsCS); for (size_t i = 0; i < m_clientSockets.size(); ++i) { if (m_clientSockets[i] != excludeSocket && m_clientSockets[i] != INVALID_SOCKET) { send(m_clientSockets[i], message.c_str(), static_cast<int>(message.size()), 0); } } LeaveCriticalSection(&m_clientsCS); } Client::Client(const std::string& ip, int port) : m_serverIP(ip), m_port(port) { m_socket = INVALID_SOCKET; } Client::~Client() { Disconnect(); } bool Client::Connect() { WSADATA wsaData; if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) { std::cerr << "WSAStartup failed" << std::endl; return false; } m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (m_socket == INVALID_SOCKET) { std::cerr << "Socket creation failed" << std::endl; WSACleanup(); return false; } sockaddr_in serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_port = htons(m_port); inet_pton(AF_INET, m_serverIP.c_str(), &serverAddr.sin_addr); if (connect(m_socket, (sockaddr*)& serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) { std::cerr << "Connection failed: " << WSAGetLastError() << std::endl; closesocket(m_socket); WSACleanup(); return false; } // 设置非阻塞模式 u_long mode = 1; ioctlsocket(m_socket, FIONBIO, &mode); m_running = true; StartReceiver(); std::cout << "Connected to server: " << m_serverIP << ":" << m_port << std::endl; return true; } void Client::Disconnect() { m_running = false; CloseSocket(); WSACleanup(); std::cout << "Disconnected from server" << std::endl; } void Client::StartReceiver() { m_receiveThread = CreateThread(NULL, 0, ReceiveThreadProc, this, 0, NULL); if (m_receiveThread == NULL) { std::cerr << "Failed to create receive thread" << std::endl; m_running = false; } } // 明确的发送接口 bool Client::Send(const std::string& message) { if (m_socket == INVALID_SOCKET || !m_running) { return false; } return SendMessageTcp(message); } // 阻塞式接收 bool Client::Receive(std::string& outMessage, DWORD timeout_ms) { outMessage = ReceiveMessage(timeout_ms); return !outMessage.empty(); } // 非阻塞式接收 bool Client::TryReceive(std::string& outMessage) { EnterCriticalSection(&m_queueCS); if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); return false; } outMessage = m_messageQueue.front(); m_messageQueue.pop(); LeaveCriticalSection(&m_queueCS); return true; } 仔细阅读代码
06-25
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值