如何使用 IOCP 获取客户端真实 IP 地址和端口？

Question

我需要在服务器中获取客户端的 ip 地址和端口。 服务器使用 IOCP 写在 C++ 上，所以我不接受客户端，我创建新的套接字，然后在这个准备好的套接字上接受 (AcceptEx) 客户端。 因为它 struct sockaddr_in 是不正确的。

我怎样才能做到这一点？

谢谢！

Answer 1

AcceptEx可能会返回您的地址和对等地址（ip + TCP 套接字的端口）。

BOOL AcceptEx(
  __in   SOCKET sListenSocket,
  __in   SOCKET sAcceptSocket,
  __in   PVOID lpOutputBuffer,
  __in   DWORD dwReceiveDataLength,
  __in   DWORD dwLocalAddressLength,
  __in   DWORD dwRemoteAddressLength,
  __out  LPDWORD lpdwBytesReceived,
  __in   LPOVERLAPPED lpOverlapped
);

然后，您应该指定lpOutputBuffer指向一个大小足以容纳 2 个返回地址的缓冲区。 dwLocalAddressLength和dwRemoteAddressLength应设置为缓冲区中保留的地址大小。

根据 MSDN，单个地址（对于AcceptEx函数）所需的缓冲区大小为sizeof(SOCKADDR_IN) + 16 ：

 dwLocalAddressLength [in] The number of bytes reserved for the local address information. This value must be at least 16 bytes more than the maximum address

正在使用的传输协议的长度。

当然，缓冲区必须在 I/O 持续时间内有效。 你可以把它放在你的OVERLAPPED结构中。 像这样：

struct OverlappedAccept
    :public OVERLAPPED
{
    // some context information that you need
    // ...

    // Buffer for addresses
    struct {
        BYTE m_pLocal[sizeof(SOCKADDR_IN) + 16];
        BYTE m_pRemote[sizeof(SOCKADDR_IN) + 16];
    } m_Bufs;
};


// start accept operation
OverlappedAccept* pOver = /* ... */;

BOOL bRet = AcceptEx(
    hSockListen,
    hSockNew,
    &pOver->m_Bufs,
    0,
    sizeof(pOver->m_Bufs.m_pLocal),
    sizeof(pOver->m_Bufs.m_pRemote),
    &dwBytes,
    pOver);

I/O（成功）完成后，您可能会获得以下地址：

sockaddr *pLocal = NULL, *pRemote = NULL;
int nLocal = 0, nRemote = 0;
GetAcceptExSockAddrs(
    &pOver->m_Bufs,
    0,
    sizeof(pOver->m_Bufs.m_pLocal),
    sizeof(pOver->m_Bufs.m_pRemote),
    &pLocal,
    &nLocal,
    &pRemote,
    &nRemote);