IO 完成端口：WSARecv() 如何工作？

Question

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

但是，这些 IO 完成端口有两个问题。

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

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

(2) 如果 WSARecv() 被注释掉
下面的代码片段，程序
仍然接收数据。那是什么
意思是？这是否意味着我不需要
到底要调用 WSARecv() 吗？

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

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

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