4

我想使用工作线程池和 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;
         }
     }
 }
4

1 回答 1

8

(1)第一个问题是服务器基本上是从客户端接收数据,但我不知道是否接收到完整的消息。

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

(2) 如果WSARecv() 在下面的代码片段中被注释掉了,程序仍然会接收到数据。这意味着什么?这是否意味着我根本不需要调用 WSARecv() ?

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

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

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

于 2009-11-20T13:06:08.637 回答