GetQueuedCompletionStatus 在 UDP 套接字上无限期阻塞
GetQueuedCompletionStatus blocking indefinitely on UDP socket
我正在尝试使用 Winsock 创建依赖于 IO 完成端口的 UDP client/server class,但我无法将 GetQueuedCompletionStatus() 函数获取到 return 当有新数据可用时。这可能是由于我的一些误解,但是 examples/documentation 在 IOCP 上使用 UDP 而不是 TCP 的情况很少见。
这是我的服务器 class 的相关位,为简洁起见删除了错误检查:
Server::Server()
{
m_iocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
}
void Server::StartReceiving()
{
StopReceiving();
m_iocp = CreateIoCompletionPort((HANDLE)m_receiveSocket, m_iocp, (DWORD)this, 0);
//WSAEVENT event = WSACreateEvent();
//WSAEventSelect(m_receiveSocket, event, FD_ACCEPT | FD_CLOSE);
// Start up worker thread for receiving data
m_receiveThread.reset(new std::thread(&Server::ReceiveWorkerThread, this));
}
void Server::Host(const std::string& port)
{
if (!m_initialized)
Initialize();
addrinfo hints = {};
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_DGRAM;
hints.ai_protocol = IPPROTO_UDP;
hints.ai_flags = AI_PASSIVE;
// Resolve the server address and port
const char* portStr = port.empty() ? kDefaultPort.c_str() : port.c_str();
int result;
AddressInfo addressInfo = AddressInfo::Create(nullptr, portStr, &hints, result); // Calls getaddrinfo()
m_receiveSocket = WSASocket(addressInfo.GetFamily(), addressInfo.GetSocketType(), addressInfo.GetProtocol(), nullptr, 0, WSA_FLAG_OVERLAPPED);
// Bind receiving socket
result = bind(m_receiveSocket, addressInfo.GetSocketAddress(), addressInfo.GetAddressLength());
StartReceiving();
}
void Server::ReceiveWorkerThread()
{
SOCKADDR_IN senderAddr;
int senderAddrSize = sizeof(senderAddr);
DWORD bytesTransferred = 0;
OVERLAPPED* pOverlapped = nullptr;
WSABUF wsaBuf = { (ULONG)m_buffer.GetWriteBufferSize(), m_buffer.GetWriteBufferPointer() };
DWORD flags = 0;
DWORD bytesReceived;
int result = WSARecvFrom(m_receiveSocket, &wsaBuf, 1, &bytesReceived, &flags, (sockaddr*)&senderAddr, &senderAddrSize, pOverlapped, nullptr);
// Process packets until signaled to exit
while (true)
{
DWORD context = 0;
BOOL success = GetQueuedCompletionStatus(
m_iocp,
&bytesTransferred,
&context,
&pOverlapped,
INFINITE);
wsaBuf.len = (ULONG)m_buffer.GetWriteBufferSize();
wsaBuf.buf = m_buffer.GetWriteBufferPointer();
flags = 0;
result = WSARecvFrom(m_receiveSocket, &wsaBuf, 1, &bytesReceived, &flags, (sockaddr*)&senderAddr, &senderAddrSize, pOverlapped, nullptr);
// Code to process packet would go here
if (m_exiting.load() == true)
break; // Kill worker thread
}
}
当我的客户端向服务器发送数据时,第一个 WSARecvFrom 正确地获取了数据,但是服务器阻止了对 GetQueuedCompletionStatus 的调用并且从不 returns,即使发送了更多的数据报。我也尝试过使用 WSAEventSelect 将套接字置于非阻塞模式(代码在上面有评论),但没有任何区别。
从阅读 this 类似 post 听起来好像至少需要在套接字上进行一次读取才能触发 IOCP,这就是为什么我在主循环之外添加了对 WSARecvFrom 的第一次调用.希望如果服务器在没有 IOCP 的情况下接收数据,客户端代码是无关紧要的假设是正确的,所以我没有 post 编辑它。
我确定我做错了什么,但我只是没有看到它。
您需要检查 WSARecvFrom
的结果代码,并仅在 return 代码为 ERROR_IO_PENDING
时调用 GetQueuedCompletionStatus
- 如果不是,则操作完成阻塞并且你有数据,或者有一个错误,但在任何这些情况下它都没有发布到 I/O 完成端口因此它永远不会被 GetQueuedCompletionStatus
拾取并且调用将堵塞。
并且您不应该在一个线程中执行此操作。常见的方法是让一个线程只轮询 I/O 完成端口,并在上下文对象上调用一些回调来通知 incoming/outgoing 数据,并在需要的地方调用发送接收调用。
我正在尝试使用 Winsock 创建依赖于 IO 完成端口的 UDP client/server class,但我无法将 GetQueuedCompletionStatus() 函数获取到 return 当有新数据可用时。这可能是由于我的一些误解,但是 examples/documentation 在 IOCP 上使用 UDP 而不是 TCP 的情况很少见。
这是我的服务器 class 的相关位,为简洁起见删除了错误检查:
Server::Server()
{
m_iocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
}
void Server::StartReceiving()
{
StopReceiving();
m_iocp = CreateIoCompletionPort((HANDLE)m_receiveSocket, m_iocp, (DWORD)this, 0);
//WSAEVENT event = WSACreateEvent();
//WSAEventSelect(m_receiveSocket, event, FD_ACCEPT | FD_CLOSE);
// Start up worker thread for receiving data
m_receiveThread.reset(new std::thread(&Server::ReceiveWorkerThread, this));
}
void Server::Host(const std::string& port)
{
if (!m_initialized)
Initialize();
addrinfo hints = {};
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_DGRAM;
hints.ai_protocol = IPPROTO_UDP;
hints.ai_flags = AI_PASSIVE;
// Resolve the server address and port
const char* portStr = port.empty() ? kDefaultPort.c_str() : port.c_str();
int result;
AddressInfo addressInfo = AddressInfo::Create(nullptr, portStr, &hints, result); // Calls getaddrinfo()
m_receiveSocket = WSASocket(addressInfo.GetFamily(), addressInfo.GetSocketType(), addressInfo.GetProtocol(), nullptr, 0, WSA_FLAG_OVERLAPPED);
// Bind receiving socket
result = bind(m_receiveSocket, addressInfo.GetSocketAddress(), addressInfo.GetAddressLength());
StartReceiving();
}
void Server::ReceiveWorkerThread()
{
SOCKADDR_IN senderAddr;
int senderAddrSize = sizeof(senderAddr);
DWORD bytesTransferred = 0;
OVERLAPPED* pOverlapped = nullptr;
WSABUF wsaBuf = { (ULONG)m_buffer.GetWriteBufferSize(), m_buffer.GetWriteBufferPointer() };
DWORD flags = 0;
DWORD bytesReceived;
int result = WSARecvFrom(m_receiveSocket, &wsaBuf, 1, &bytesReceived, &flags, (sockaddr*)&senderAddr, &senderAddrSize, pOverlapped, nullptr);
// Process packets until signaled to exit
while (true)
{
DWORD context = 0;
BOOL success = GetQueuedCompletionStatus(
m_iocp,
&bytesTransferred,
&context,
&pOverlapped,
INFINITE);
wsaBuf.len = (ULONG)m_buffer.GetWriteBufferSize();
wsaBuf.buf = m_buffer.GetWriteBufferPointer();
flags = 0;
result = WSARecvFrom(m_receiveSocket, &wsaBuf, 1, &bytesReceived, &flags, (sockaddr*)&senderAddr, &senderAddrSize, pOverlapped, nullptr);
// Code to process packet would go here
if (m_exiting.load() == true)
break; // Kill worker thread
}
}
当我的客户端向服务器发送数据时,第一个 WSARecvFrom 正确地获取了数据,但是服务器阻止了对 GetQueuedCompletionStatus 的调用并且从不 returns,即使发送了更多的数据报。我也尝试过使用 WSAEventSelect 将套接字置于非阻塞模式(代码在上面有评论),但没有任何区别。
从阅读 this 类似 post 听起来好像至少需要在套接字上进行一次读取才能触发 IOCP,这就是为什么我在主循环之外添加了对 WSARecvFrom 的第一次调用.希望如果服务器在没有 IOCP 的情况下接收数据,客户端代码是无关紧要的假设是正确的,所以我没有 post 编辑它。
我确定我做错了什么,但我只是没有看到它。
您需要检查 WSARecvFrom
的结果代码,并仅在 return 代码为 ERROR_IO_PENDING
时调用 GetQueuedCompletionStatus
- 如果不是,则操作完成阻塞并且你有数据,或者有一个错误,但在任何这些情况下它都没有发布到 I/O 完成端口因此它永远不会被 GetQueuedCompletionStatus
拾取并且调用将堵塞。
并且您不应该在一个线程中执行此操作。常见的方法是让一个线程只轮询 I/O 完成端口,并在上下文对象上调用一些回调来通知 incoming/outgoing 数据,并在需要的地方调用发送接收调用。