IOCP的使用

来源：互联网发布：lol刷点卷软件2016 编辑：程序博客网时间：2024/05/16 11:11

#include<WINSOCK2.H>#include<windows.h>#include<iostream.h>#pragma comment(lib,"ws2_32.lib")#define PORT 5150#define DATA_BUFFSIZE 8192//关键项typedef struct {OVERLAPPED Overlapped;WSABUFDataBuf;CHARBuffer[DATA_BUFFSIZE];DWORDBytesSEND;//发送字节数DWORDBytesRECV;//接受字节数}PER_IO_OPERATION_DATA,*LPPER_IO_OPERATION_DATA;//传给线程参数typedef struct {SOCKET Socket;}PER_HANDLE_DATA,*LPPER_HANDLE_DATA;DWORD WINAPI ServerWorkerThread(LPVOID CompletionPortID);int main(int argv,char *argvc[]){SOCKADDR_INInternetAddr;SOCKETListen;SOCKETAccept;HANDLECompletionPort;SYSTEM_INFOSystemInfo;LPPER_HANDLE_DATAPerHandleData;LPPER_IO_OPERATION_DATAPerIoData;int i;DWORDRecvBytes;DWORDFlags;DWORDThreadID;WSADATAwsaData;DWORDRet;if((Ret=WSAStartup(MAKEWORD(2,2),&wsaData))!=0){cout<<"WSAStart up failed with error "<<Ret<<endl;return 1;}if((CompletionPort=CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0))==NULL){cout<<"CreateIoCompletionPort failed with error: "<<WSAGetLastError()<<endl;return 1;}GetSystemInfo(&SystemInfo);for(i=0;i<SystemInfo.dwNumberOfProcessors*2;i++){HANDLEThreadHandle;if((ThreadHandle=CreateThread(NULL,0,ServerWorkerThread,CompletionPort,0,&ThreadID))==NULL){cout<<"CreateThread() failed with error "<<WSAGetLastError()<<endl;return 1;}CloseHandle(ThreadHandle);}cout<<"创建线程数:"<<i<<endl;if((Listen=WSASocket(AF_INET,SOCK_STREAM,0,NULL,0,WSA_FLAG_OVERLAPPED))==INVALID_SOCKET){cout<<"WSASocket() failed with error "<<WSAGetLastError()<<endl;return 1;}InternetAddr.sin_addr.s_addr=htonl(INADDR_ANY);//inet_addr("");InternetAddr.sin_family=AF_INET;InternetAddr.sin_port=htons(PORT);if(bind(Listen,(SOCKADDR*)&InternetAddr,sizeof(SOCKADDR_IN))==SOCKET_ERROR){cout<<"bind() failed with error "<<WSAGetLastError()<<endl;return 1;}if(listen(Listen,5)==SOCKET_ERROR){cout<<"listen() failed with error "<<GetLastError()<<endl;return 1;}while(true){if((Accept=WSAAccept(Listen,NULL,NULL,NULL,0))==SOCKET_ERROR){cout<<"WSAAccept() failed with error "<<WSAGetLastError()<<endl;}// 创建一个关键项用于保存这个客户端的信息，用户接收发送的重叠结构，// 还有使用到的缓冲区 if((PerHandleData=(LPPER_HANDLE_DATA)GlobalAlloc(GPTR,sizeof(PER_HANDLE_DATA)))==NULL){cout<<"GlobalAlloc() failed with error "<<GetLastError();return 1;}cout<<"socket number "<<Accept<<" connected"<<endl;PerHandleData->Socket=Accept;// 与我们的创建的那个完成端口关联起来,将关键项也与指定的一个完成端口关联if(CreateIoCompletionPort((HANDLE)Accept,CompletionPort,(DWORD)PerHandleData,0)==NULL){cout<<"CreateIoCompletionPort() failed with error "<<GetLastError()<<endl;return 1;}if((PerIoData=(LPPER_IO_OPERATION_DATA)GlobalAlloc(GPTR,sizeof(PER_IO_OPERATION_DATA)))==NULL){cout<<"GlobalAlloc() failed with error "<<GetLastError()<<endl;return 1;}ZeroMemory(&(PerIoData->Overlapped),sizeof(OVERLAPPED));PerIoData->BytesRECV=0;PerIoData->BytesSEND=0;PerIoData->DataBuf.len=DATA_BUFFSIZE;PerIoData->DataBuf.buf=PerIoData->Buffer;Flags=0;if(WSARecv(Accept,&(PerIoData->DataBuf),1,&RecvBytes,&Flags,&(PerIoData->Overlapped),NULL)==SOCKET_ERROR){if(WSAGetLastError()!=ERROR_IO_PENDING){cout<<"WSARecv() failed with error "<<WSAGetLastError()<<endl;return 1;}}}return 0;}DWORD WINAPI ServerWorkerThread(LPVOID CompletionPortID){HANDLE CompletionPort=(HANDLE)CompletionPortID;DWORDBytesTransferred;LPPER_HANDLE_DATAPerHandleData;LPPER_IO_OPERATION_DATAPerIoData;DWORDSendbytes,Recvbytes;DWORDFlags;boolGetStatus_flag;while(true){GetStatus_flag=GetQueuedCompletionStatus(CompletionPort,&BytesTransferred,(LPDWORD)&PerHandleData,(LPOVERLAPPED*)&PerIoData,INFINITE);// //如果 *lpOverlapped为空并且函数没有从完成端口取出完成包// if(PerIoData->Overlapped==NULL)// {// continue;// }// //如果 *lpOverlapped不为空并且函数从完成端口出列一个失败I/O操作的完成包// if(FALSE==GetStatus_flag&&GetLastError()!=0)// {// GlobalFree(PerHandleData);// GlobalFree(PerIoData);// continue;// }//如果关联到一个完成端口的一个socket句柄被关闭了if(FALSE==GetStatus_flag&&BytesTransferred==0){cout<<"Closing socket "<<PerHandleData->Socket<<endl;if(closesocket(PerHandleData->Socket)==SOCKET_ERROR){cout<<"closesocket() failed with error "<<WSAGetLastError()<<endl;return 0;}GlobalFree(PerHandleData);GlobalFree(PerIoData);continue;}if(false==GetStatus_flag){return 0;}//如果函数从完成端口取出一个成功I/O操作的完成包if(PerIoData->BytesRECV==0){PerIoData->BytesRECV=BytesTransferred;PerIoData->BytesSEND=0;}else{PerIoData->BytesSEND+=BytesTransferred;}if(PerIoData->BytesRECV>PerIoData->BytesSEND){ZeroMemory(&(PerIoData->Overlapped),sizeof(OVERLAPPED));PerIoData->DataBuf.buf=PerIoData->Buffer+PerIoData->BytesSEND;PerIoData->DataBuf.len=PerIoData->BytesRECV-PerIoData->BytesSEND;if(WSASend(PerHandleData->Socket,&(PerIoData->DataBuf),1,&Sendbytes,0,&(PerIoData->Overlapped),NULL)==SOCKET_ERROR){if(WSAGetLastError()!=ERROR_IO_PENDING){cout<<"WSASend() failed with error "<<WSAGetLastError()<<endl;return 0;}}cout<<"send:"<<PerIoData->DataBuf.buf<<endl;}else{PerIoData->BytesRECV=0;Flags=0;ZeroMemory(&(PerIoData->Overlapped),sizeof(OVERLAPPED));PerIoData->DataBuf.len=DATA_BUFFSIZE;PerIoData->DataBuf.buf=PerIoData->Buffer;if(WSARecv(PerHandleData->Socket,&(PerIoData->DataBuf),1,&Recvbytes,&Flags,&(PerIoData->Overlapped),NULL)==SOCKET_ERROR){if(WSAGetLastError()!=ERROR_IO_PENDING){cout<<"WSARecv() failed with error "<<WSAGetLastError()<<endl;return 0;}cout<<"recv:"<<PerIoData->DataBuf.buf<<endl;}}}}

flag=GetQueuedCompletionStatus(
HANDLE CompletionPort,
LPDWORD lpNumberOfBytes,
PULONG_PTR lpCompletionKey,
LPOVERLAPPED *lpOverlapped,
DWORD dwMilliseconds
);

针对:如果 *lpOverlapped为空并且函数没有从完成端口取出完成包，返回值则为0。函数则不会在lpNumberOfBytes and lpCompletionKey所指向的参数中存储信息。

if(lpOverlapped== NULL) {continue;}

针对: 如果 *lpOverlapped不为空并且函数从完成端口出列一个失败I/O操作的完成包，返回值为0。函数在指向lpNumberOfBytesTransferred, lpCompletionKey, and lpOverlapped的参数指针中存储相关信息。调用GetLastError可以得到扩展错误信息
if(FALSE == flag && GetLastError()!=0){ 归还IO句柄；continue;}

针对:如果关联到一个完成端口的一个socket句柄被关闭了，则GetQueuedCompletionStatus返回ERROR_SUCCESS,并且lpNumberOfBytes等于0 .

ERROR_SUCCESS这个值是0,即返回值是0(false)且GetLastError()等于0，与上面的错误处理代码是有分别的。
if(opType != IO_ACCEPT && dwByteTrans == 0) {做断开处理，归还IO；continue;}

针对: 如果函数从完成端口取出一个成功I/O操作的完成包，返回值为非0。函数在指向lpNumberOfBytesTransferred, lpCompletionKey, and lpOverlapped的参数中存储相关信息。
switch(){根据IO类型做相关操作}

}

现在的问题是：我需要在(FALSE == flag && GetLastError()!=0)的情况下做一些处理。导致GetQueuedCompletionStatus返回失败的因素有很多，最常见的是〖1236〗-由本地系统终止网络连接，〖1234〗-没有任何服务正在远程系统上的目标网络终结点上操作，〖64〗-指定的网络名不再可用等等。之前发生错误时简单的回还IO句柄，但有些问题由此而产生。(UDP问题:若完成键没有赋值，投递io没有进行，后期的数据会收不到)

本文来自CSDN博客，转载请标明出处：http://blog.csdn.net/xuwei2007/archive/2007/09/05/1773451.aspx