初探IOCP

今天是我第一次接触IOCP，感觉还可以，又学到了一些东西。做了几个实验，对IOCP有了初步的认识。希望自己能够深入的学习下去。现在把今天做的有关事情整理一下。IOCP实验采用的代码是从《WINDOWS网络与通信程序设计》所附的源代码修改而来的（本来想花点钱买这本书，在网上找了一下，发现有免费的电子版下载，让我省了点钱）。

 

IOCP的全称是I/O Completion Port ，中文叫做I/O完成端口。其实际上就是应用程序采用线程池处理异步I/O请求的一种机制，它可以接收多种对象句柄，如文件对象和套接字对象。

 

使用完成端口模型，最先调用的函数是：

HANDLE CreateIoCompletionPort ( HANDLE FileHandle, // handle to file HANDLE ExistingCompletionPort, // handle to I/O completion port ULONG_PTR CompletionKey, // completion key DWORD NumberOfConcurrentThreads // number of threads to execute concurrently );

这个函数在主线程要应用到两次，其作用分别是创建一个完成对象窗口和将一个或多个文件句柄关（可以是SOCKET句柄）联到I/O完成端口对象。如果是第二次用到该函数，那么当中的FileHandle为要关联的套接字句柄，ExistingCompletionPort为上面创建完成的端口对象，CompletionKey为指定句柄唯一数据，它将与FileHandle套接字句柄关联在一起。  

如：CreateIoCompletionPort((HANDLE)pPerHandle->s,hCompletion,(DWORD)pPerHandle,0);

下一个要用到的函数是：

BOOL GetQueuedCompletionStatus( HANDLE CompletionPort, // handle to completion port LPDWORD lpNumberOfBytes, // bytes transferred PULONG_PTR lpCompletionKey, // file completion key LPOVERLAPPED *lpOverlapped, // buffer DWORD dwMilliseconds // optional timeout value );

向完成端口关联套接字句柄后，便可以在套接字上投递重叠发送和接收请求处理I/O了。在这些I/O操作完成时，I/O系统会向完成端口对象发送一个完成通知封包，I/O完成端口以先进先出的方式给这些封包排队，使用GetQueuedCompletionStatus函数可以取得这些队列当中的封包，如：

BOOL bOK = GetQueuedCompletionStatus(hCompletion,&dwTrans,(LPDWORD)&pPerHandle,(LPOVERLAPPED*)&pPerIO, WSA_INFINITE);

在这些数据当中，最重要的是pPerHandle和pPerIO。lpCompletionKey参数包含了pPerHandle数据，当套接字第一次与完成端口关联时，这个数据就关联到了套接字句柄。lpOverlapped参数指向一个OVERLAPPED结构，结构后面就是pPerIO数据，这可以是工作线程完成封包时想要知道的任何信息。整个GetQueuedCompletionStatus也是工作在工作线程当中的。

总的程序如下：

#include <stdio.h> #include <Winsock2.h> #include <Windows.h> #define BUFFER_SIZE 1024 typedef struct _PER_HANDLE_DATA // per-handle数据 { SOCKET s; // 对应的套节字句柄 sockaddr_in addr; // 客户方地址 } PER_HANDLE_DATA, *PPER_HANDLE_DATA; typedef struct _PER_IO_DATA // per-I/O数据 { OVERLAPPED ol; // 重叠结构 char buf[BUFFER_SIZE]; // 数据缓冲区 int nOperationType; // 操作类型 #define OP_READ 1 #define OP_WRITE 2 #define OP_ACCEPT 3 } PER_IO_DATA, *PPER_IO_DATA; DWORD WINAPI SrvThread( LPVOID lpParameter // thread data ); void main() { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD( 2, 2 ); err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { return; } if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 2 ) { WSACleanup( ); return; } HANDLE hCompletion = CreateIoCompletionPort(INVALID_HANDLE_VALUE,0,0,0); HANDLE hThread = CreateThread(NULL,0,SrvThread,(LPVOID)hCompletion,0,0); CloseHandle(hThread); SOCKET sockListen = socket(AF_INET,SOCK_STREAM,0); SOCKADDR_IN addrLocal; addrLocal.sin_addr.S_un.S_addr = htonl(INADDR_ANY); addrLocal.sin_family = AF_INET; addrLocal.sin_port = htons(6000); bind(sockListen,(SOCKADDR*)&addrLocal,sizeof(SOCKADDR)); listen(sockListen,5); while(TRUE) { SOCKADDR_IN addrFrom; int len = sizeof(SOCKADDR); SOCKET sockRecv = accept(sockListen,(SOCKADDR*)&addrFrom,&len); PPER_HANDLE_DATA pPerHandle = (PPER_HANDLE_DATA)GlobalAlloc(GPTR,sizeof(PER_HANDLE_DATA)); pPerHandle->s = sockRecv; memcpy(&pPerHandle->addr,&addrFrom,len); CreateIoCompletionPort((HANDLE)pPerHandle->s,hCompletion,(DWORD)pPerHandle,0); PPER_IO_DATA pPerIO = (PPER_IO_DATA)GlobalAlloc(GPTR,sizeof(PER_IO_DATA)); pPerIO->nOperationType = OP_READ; WSABUF wsabuf; wsabuf.buf = pPerIO->buf; wsabuf.len = BUFFER_SIZE; DWORD dwRecv; DWORD dwFlags = 0; printf("MAIN FUN RUN!/n"); WSARecv(pPerHandle->s,&wsabuf,1,&dwRecv,&dwFlags,&pPerIO->ol,NULL); } } DWORD WINAPI SrvThread( LPVOID lpParameter // thread data ) { HANDLE hCompletion = (HANDLE)lpParameter; DWORD dwTrans; PPER_HANDLE_DATA pPerHandle; PPER_IO_DATA pPerIO; while(TRUE) { printf("THREAD FUN RUN!/n"); BOOL bOK = GetQueuedCompletionStatus(hCompletion,&dwTrans,(LPDWORD)&pPerHandle, (LPOVERLAPPED*)&pPerIO, WSA_INFINITE); if(!bOK) { closesocket(pPerHandle->s); GlobalFree(pPerHandle); GlobalFree(pPerIO); continue; } if(dwTrans == 0 && // 套节字被对方关闭 (pPerIO->nOperationType == OP_READ || pPerIO->nOperationType == OP_WRITE)) { ::closesocket(pPerHandle->s); ::GlobalFree(pPerHandle); ::GlobalFree(pPerIO); continue; } printf("THREAD2 FUN RUN!/n"); switch(pPerIO->nOperationType) { case OP_READ: { printf("THREAD3 FUN RUN!/n"); pPerIO->buf[dwTrans] = '/0'; printf(pPerIO->buf); WSABUF wsabuf; wsabuf.buf = pPerIO->buf; wsabuf.len = BUFFER_SIZE; pPerIO->nOperationType = OP_READ; DWORD nFlags = 0; WSARecv(pPerHandle->s,&wsabuf,1,&dwTrans,&nFlags,&pPerIO->ol,NULL); } break; case OP_WRITE: case OP_ACCEPT: break; } } return 1; }

程序当中加入的一些print函数是为了辅助理解整个数据接收处理过程用的。

 

大概内容就是如此，还有许多东西需要我去深入理解。