多线程 IOCP 客户端问题

Question

我正在编写一个使用 IO 完成端口的多线程客户端。

我创建并连接设置了 WSA_FLAG_OVERLAPPED 属性的套接字。

if ((m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
{
    throw std::exception("Failed to create socket.");
}

if (WSAConnectByName(m_socket, L"server.com", L"80", &localAddressLength, reinterpret_cast<sockaddr*>(&localAddress), &remoteAddressLength, &remoteAddress, NULL, NULL) == FALSE)
{
    throw std::exception("Failed to connect.");
}

我将 IO 完成端口与套接字关联起来。

if ((m_hIOCP = CreateIoCompletionPort(reinterpret_cast<HANDLE>(m_socket), m_hIOCP, NULL, 8)) == NULL)
{
    throw std::exception("Failed to create IOCP object.");
}

一切似乎都很顺利，直到我尝试通过套接字发送一些数据。

SocketData* socketData = new SocketData;
socketData->hEvent = 0;

DWORD bytesSent = 0;
if (WSASend(m_socket, socketData->SetBuffer(socketData->GenerateLoginRequestHeader()), 1, &bytesSent, NULL, reinterpret_cast<OVERLAPPED*>(socketData), NULL) == SOCKET_ERROR && WSAGetLastError() != WSA_IO_PENDING)
{
    throw std::exception("Failed to send data.");
}

WSASend 立即返回，而不是返回 SOCKET_ERROR 并将最后一个错误设置为 WSA_IO_PENDING。

我需要 IO 挂起，并在我的线程函数（也是我的工作线程）中处理它的完成。

unsigned int __stdcall MyClass::WorkerThread(void* lpThis)
{

}

我以前曾经这样做过，但我不知道在这种情况下出了什么问题，我非常感谢任何帮助我解决这个问题的努力。

Answer 1

除非你这样做，否则这不是问题。

只要您不调用 SetFileCompletionNotificationModes() 并设置标志以在成功时跳过完成端口处理，然后即使 WSARecv（或其他）返回 SUCCESS，IO 完成数据包也会排队到 IOCP，就像返回 ERROR_IO_PENDING 一样。因此，您不需要对非错误返回情况进行特殊处理。

请参阅 http://support.microsoft.com/default.aspx ?scid=kb;en-us;Q192800 了解详细信息。

Answer 2

首先，将调用分解为更清晰的逻辑：

int nRet = WSASend(m_socket, socketData->SetBuffer(socketData->GenerateLoginRequestHeader()), 1, NULL, NULL, reinterpret_cast<OVERLAPPED*>(socketData), NULL);
if (nRet == SOCKET_ERROR)
{
    if ((WSAGetLastError()) == WSA_IO_PENDING)
        nRet = 0; // ok
    else
        throw std::exception("Failed to send data."); // failed
}

此外，正如您在我的代码中看到的，您不应该根据 WSASend：

如果该参数使用 NULL
lpOverlapped 参数不为 NULL
避免潜在的错误结果。

除此之外，您对 WSASend() 的调用看起来不错。