IOCP AcceptEx 在连接时未创建完成

Question

我目前正在尝试一些用于套接字编程的新库 (IOCP)。我偶然发现了AcceptEx 启用异步连接的功能。

正如文档所说：

AcceptEx 函数使用重叠 I/O，与 accept 函数不同。如果您的应用程序使用 AcceptEx，它可以用相对较少的线程数为大量客户端提供服务。与所有重叠的 Windows 函数一样，Windows 事件或完成端口都可以用作完成通知机制。

但是当客户端连接时，我没有收到任何完成。但是，当客户端发送数据时，我确实会完成..

这是我的代码：

DWORD dwBytes;
GUID GuidAcceptEx = WSAID_ACCEPTEX;
int iResult = WSAIoctl(m_hSocket, SIO_GET_EXTENSION_FUNCTION_POINTER,
    &GuidAcceptEx, sizeof (GuidAcceptEx), 
    &m_lpfnAcceptEx, sizeof (m_lpfnAcceptEx), 
    &dwBytes, NULL, NULL);

if (iResult == SOCKET_ERROR)
{
    CloseSocket();
}

然后：

WSAOVERLAPPED olOverlap;
memset(&olOverlap, 0, sizeof (olOverlap));
char lpOutputBuf[1024];
int outBufLen = 1024;
DWORD dwBytes;

BOOL bRet = m_lpfnAcceptEx( m_hSocket, hSocket, lpOutputBuf,
             outBufLen - ((sizeof (sockaddr_in) + 16) * 2),
             sizeof (sockaddr_in) + 16, sizeof (sockaddr_in) + 16, 
             &dwBytes, &olOverlap);
if ( bRet == FALSE )
{
    DWORD dwRet = WSAGetLastError();
    if( dwRet != WSA_IO_PENDING )
    {
        return dwRet;
    }
}

有什么建议可以接收完成吗？

编辑： 我在 m_lpfnAcceptEx() 之后将 hSocket 绑定到完成端口

Answer 1

首先，当完成发生时，您在调用AcceptEx() 上方的堆栈上声明的WSAOVERLAPPED 和数据缓冲区将不存在（除非您在同一个函数中调用GetQueuedCompletionStatus()，这将有点奇怪）。您需要动态分配它们或将它们池化。

其次，您声明在调用AcceptEx() 后将套接字与完成端口相关联。那是错误的。您需要在致电AcceptEx() 之前完成这些事情。

1. 创建一个设置了WSA_FLAG_OVERLAPPED 的套接字。
2. 将其绑定到您要收听的地址。
3. 用你想要的积压来调用它。
4. 使用侦听套接字动态加载AcceptEx() 并调用WSAIoctl（不是绝对必要的，您显示的代码应该可以工作，但是这样您可以确保您从同一个底层winsock 提供程序获取您的侦听套接字，并且它支持 AcceptEx()。
5. 以与加载 AcceptEx() 相同的方式加载 GetAcceptExSockaddrs() - 一旦接受完成，您将需要它。
6. 将侦听套接字关联到您的 IOCP。

现在您可以使用侦听套接字和您创建的新“接受”套接字发布多个 AcceptEx() 调用，如下所示：

1. 创建一个设置了WSA_FLAG_OVERLAPPED 的套接字。
2. 将套接字关联到您的 IOCP。

如上所述，您需要确保缓冲区和 OVERLAPPED 在每次调用时都是唯一的，并且持续到完成为止。

完成后，您必须执行以下操作....

1. 使用侦听套接字作为数据在接受的套接字上调用setsockopt() 和SO_UPDATE_ACCEPT_CONTEXT...
2. 使用 GetAcceptExSockaddrs() 取消阻止您的地址。
3. 处理任何数据（如果您在缓冲区中为数据分配了足够的空间）。

请注意，按照设计，AcceptEx() 可用于接受新连接并在一次操作中从该连接返回初始数据（这会在您知道您总是需要一些数据之前，这会导致性能稍好一些）开始做事，但如果你想防御拒绝服务攻击，这种攻击可以通过连接而不是发送数据来发起——我写过这个here）。

如果您不希望AcceptEx() 等待数据到达，那么只需提供一个数据缓冲区，该缓冲区仅足以返回地址，并将 0 作为“缓冲区大小”传递。这将导致AcceptEx() 像重叠的accept() 一样运行，并在建立连接后立即返回。

请注意，Martin James 对您的问题的最初评论实际上就是您正在寻找的答案。不要传outBufLen - ((sizeof (sockaddr_in) + 16) * 2)，传0。