GetQueuedCompletionStatus блокирует неограниченное время на UDP-сокете

Question

Я пытаюсь создать класс клиента/сервера UDP, который полагается на порты завершения ввода-вывода с использованием Winsock, но мне не удалось вернуть функцию GetQueuedCompletionStatus(), когда будут возвращены новые данные доступный. Вероятно, это связано с некоторыми недоразумениями с моей стороны, но примеры/документация по IOCP с UDP вместо TCP немного и далеко друг от друга.

Вот соответствующие биты моего класса сервера с ошибкой проверки удалено для краткости:

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 и никогда не вернетесь, даже если отправлено больше датаграмм. Я также попытался поместить сокет в неблокирующий режим с помощью WSAEventSelect (код для этого комментируется выше), но это не имело никакого значения.

От чтения this похожее сообщение похоже, что должно быть хотя бы одно чтение на сокете, чтобы вызвать IOCP, поэтому я добавил первый вызов WSARecvFrom из основного цикла. Надеюсь, я прав, если предположить, что код клиента не имеет значения, если сервер получает данные без IOCP, поэтому я его не разместил.

Я уверен, что я делаю что-то неправильно, но я просто не вижу его.

Answer 1

Вам нужно проверить код результата от WSARecvFrom и вызвать GetQueuedCompletionStatus только если код возврата ERROR_IO_PENDING - если это не либо операция завершена без блокировки и у вас есть данные, или произошла ошибка, но ни в одном из в этих случаях он не был отправлен на порт завершения ввода-вывода, и поэтому он никогда не будет поднят на GetQueuedCompletionStatus, и вызов будет заблокирован.

И вы не должны делать это в одном потоке. Общий подход состоит в том, чтобы иметь поток, который только опросает порт завершения ввода-вывода и вызывает некоторые обратные вызовы в объектах контекста, чтобы уведомлять о входящих/исходящих данных, а отправляющие принимающие вызовы вызывают везде, где это необходимо.