c++ - IO 完成端口 : How does WSARecv() work?

Question

我想使用工作线程池和 IO 完成端口编写一个服务器。服务器应该在多个客户端之间处理和转发消息。 “每个客户”数据位于 ClientContext 类中。此类实例之间的数据使用工作线程进行交换。我认为这是一个典型的场景。

但是，我对那些 IO 完成端口有两个问题。

(1) 第一个问题是服务器基本上从客户端接收数据，但我不知道是否收到了完整的消息。事实上，WSAGetLastError() 总是返回 WSARecv() 仍在挂起。我试图用 WaitForMultipleObjects() 等待事件 OVERLAPPED.hEvent。但是，它会永远阻塞，即 WSARecv() 在我的程序中永远不会完成。我的目标是绝对确保在进一步处理开始之前已收到整个消息。我的消息在其 header 中有一个“消息长度”字段，但我真的不知道如何将它与 IOCP 函数参数一起使用。

(2) 如果在下面的代码片段中注释掉WSARecv()，程序仍然接收数据。那是什么意思？这是否意味着我根本不需要调用 WSARecv()？我无法通过这些 IO 完成端口获得确定性行为。感谢您的帮助!

while(WaitForSingleObject(module_com->m_shutdown_event, 0)!= WAIT_OBJECT_0)
{

    dequeue_result = GetQueuedCompletionStatus(module_com->m_h_io_completion_port,
                                               &transfered_bytes,
                                               (LPDWORD)&lp_completion_key,
                                               &p_ol,
                                               INFINITE);
     if (lp_completion_key == NULL)
     {
         //Shutting down
         break;
     }

     //Get client context
     current_context = (ClientContext *)lp_completion_key;

     //IOCP error
     if(dequeue_result == FALSE)
     {
         //... do some error handling...
     }
     else
     {   
         // 'per client' data
         thread_state = current_context->GetState();
         wsa_recv_buf = current_context->GetWSABUFPtr();

         // 'per call' data
         this_overlapped = current_context->GetOVERLAPPEDPtr();
     }

     while(thread_state != STATE_DONE)
     {
         switch(thread_state)
         {
         case STATE_INIT:

             //Check if completion packet has been posted by internal function or by WSARecv(), WSASend()
             if(transfered_bytes > 0)
             {
                 dwFlags = 0;
                 transf_now = 0;
                 transf_result = WSARecv(current_context->GetSocket(),
                                         wsa_recv_buf,
                                         1,
                                         &transf_now,
                                         &dwFlags,
                                         this_overlapped,
                                         NULL);

                 if (SOCKET_ERROR == transf_result && WSAGetLastError() != WSA_IO_PENDING)
                 {   
                     //...error handling...
                     break;
                 }

                 // put received message into a message queue

             }
             else // (transfered_bytes == 0)
             {
                 // Another context passed data to this context
                 // and notified it via PostQueuedCompletionStatus().
             }
             break;
         }
     }
 }

Answer 1

(1) The first problem is that the server basically receives data from clients but I never know if a complete message was received.

您的 recv 调用可以返回从 1 个字节到整个“消息”的任何位置。您需要包含逻辑，当它有足够的数据来计算完整“消息”的长度时，然后在您实际拥有完整的“消息”时计算出来。虽然您没有足够的数据，但您可以使用相同的内存缓冲区重新发出 recv 调用，但使用更新的 WSABUF 结构指向您已经接收的数据的末尾。通过这种方式，您可以在缓冲区中累积完整的消息，而无需在每次 recv 调用完成后复制数据。

(2) If WSARecv() is commented out in the code snippet below, the program still receives data. What does that mean? Does it mean that I don't need to call WSARecv() at all?

我想这只是意味着您的代码中存在错误...

请注意，从可伸缩性的角度来看，不在重叠结构中使用事件而是将套接字与 IOCP 相关联并允许将完成发布到处理完成的线程池是“更好的”。

我有一个免费的 IOCP 客户端/服务器框架，可从 here 获得这可能会给你一些提示；以及关于 CodeProject 的一系列文章(第一篇在这里:http://www.codeproject.com/KB/IP/jbsocketserver1.aspx)，我在其中处理整个“读取完整消息”问题(参见“字节流分块”)。