IOCP C++ TCP客户端

Question

IOCP C++ TCP客户端

6

我有些困难实现TCP IOCP客户端。我已经在Mac OSX上实现了kqueue，所以在Windows上寻找类似的东西，我的理解是IOCP是最接近的。主要问题是GetCompetetionStatus从未返回并且总是超时。我假设我在创建监视句柄时错过了什么，但不确定是什么。这是我目前的进展：

我的连接例程：（为了清晰起见删除了一些错误处理）

struct sockaddr_in server;
struct hostent *hp;
SOCKET sckfd;
WSADATA wsaData;

int iResult = WSAStartup( MAKEWORD(2,2), &wsaData );


if ((hp = gethostbyname(host)) == NULL)
    return NULL;
WSASocket(AF_INET,SOCK_STREAM,0,NULL,0,WSA_FLAG_OVERLAPPED)
if ((sckfd = WSASocket(AF_INET,SOCK_STREAM,0, NULL, 0, WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET)
{
    printf("Error at socket(): Socket\n");
    WSACleanup();
    return NULL;
}

server.sin_family = AF_INET;
server.sin_port = htons(port);
server.sin_addr = *((struct in_addr *)hp->h_addr);
memset(&(server.sin_zero), 0, 8);

//non zero means non blocking. 0 is blocking.
u_long iMode = -1;
iResult = ioctlsocket(sckfd, FIONBIO, &iMode);
if (iResult != NO_ERROR)
    printf("ioctlsocket failed with error: %ld\n", iResult);


HANDLE hNewIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, ulKey, 0);
CreateIoCompletionPort((HANDLE)sckfd, hNewIOCP , ulKey, 0);

connect(sckfd, (struct sockaddr *)&server, sizeof(struct sockaddr));

//WSAConnect(sckfd, (struct sockaddr *)&server, sizeof(struct sockaddr),NULL,NULL,NULL,NULL);

return sckfd;

这是发送程序：（为了清晰起见，还删除了一些错误处理）

IOPortConnect(int ServerSocket,int timeout,string& data){

char buf[BUFSIZE];
strcpy(buf,data.c_str());
WSABUF buffer = { BUFSIZE,buf };
DWORD bytes_recvd;
int r;
ULONG_PTR ulKey = 0;
OVERLAPPED overlapped;
 OVERLAPPED* pov = NULL;
HANDLE port;

HANDLE hNewIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, ulKey, 0);
CreateIoCompletionPort((HANDLE)ServerSocket, hNewIOCP , ulKey, 0);


BOOL get = GetQueuedCompletionStatus(hNewIOCP,&bytes_recvd,&ulKey,&pov,timeout*1000);

if(!get)
    printf("waiton server failed. Error: %d\n",WSAGetLastError());
if(!pov)
    printf("waiton server failed. Error: %d\n",WSAGetLastError());

port = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, (u_long)0, 0);

SecureZeroMemory((PVOID) & overlapped, sizeof (WSAOVERLAPPED));

r = WSASend(ServerSocket, &buffer, 1, &bytes_recvd, NULL, &overlapped, NULL);
printf("WSA returned: %d WSALastError: %d\n",r,WSAGetLastError());
if(r != 0)
{
    printf("WSASend failed %d\n",GetLastError());
    printf("Bytes transfered: %d\n",bytes_recvd);
}
if (WSAGetLastError() == WSA_IO_PENDING)
    printf("we are async.\n");
CreateIoCompletionPort(port, &overlapped.hEvent,ulKey, 0);

BOOL test = GetQueuedCompletionStatus(port,&bytes_recvd,&ulKey,&pov,timeout*1000); 

CloseHandle(port);
return true;

任何见解都将不胜感激。

- daltoniam

你实际上有检查CreateIoCompletionPort的返回值吗？ - John Dibling

在每个函数的CreateIoCompletionPort之后放置GetLastError会返回错误87（参数不正确）和错误6（无效句柄）。我不确定为什么它们返回不正确，因为它们都是有效的句柄（ServerSocket和端口）。 - daltoniam

2个回答

1

Chris已经处理了大部分问题，你可能已经看过很多示例代码，但是...

我有一些免费的IOCP代码，可以在这里找到：http://www.serverframework.com/products---the-free-framework.html

此外，该页面还链接了我的几篇CodeProject文章。

- Len Holgate

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Chris Becke · Accepted Answer

您正在将同一套接字与多个IOCompletionPorts关联。我确定这是无效的。在您的IOPortConnect函数（您在其中执行写操作）中，您调用CreateIOCompletionPort 4次，并传递一个shot句柄。

我的建议：

1. 创建单个IOCompletion Port（最终，您将将众多套接字与其关联）。 2. 通过调用CreateThread创建工作线程池，每个线程都会通过调用循环中的GetQueuedCompletionStatus来阻塞IOCompletionPort句柄。 3. 创建一个或多个WSA_OVERLAPPED套接字，并将每个套接字与IOCompletionPort关联。 4. 使用需要OVERLAPPED *的WSA套接字函数来触发重叠操作。 5. 当工作线程使用传递给它的OVERLAPPED *从GetQueuedCompletionStatus返回时，处理已发布请求的完成。

注意：WSASend返回0和SOCKET_ERROR，WSAGetLastError（）作为代码返回WSA_IO_PENDING，以指示您将获得到达GetQueuedCompletionStatus的IO完成数据包。任何其他错误代码都意味着您应立即将其视为IO操作未排队，因此不会再次回调。

注意2：传递给WSASend（或其他函数）的OVERLAPPED *是从GetQueuedCompletionStatus返回的OVERLAPPED *。您可以利用这一点来使用调用传递更多上下文信息：

struct MYOVERLAPPED {
  OVERLAPPED ovl;
};
MYOVERLAPPED ctx;
WSASend(...,&ctx.ovl);
...
OVERLAPPED* pov;
if(GetQueuedCompletionStatus(...,&pov,...)){
  MYOVERLAPPED* pCtx = (MYOVERLAPPED*)pov;