SOCKET IOCP DEMO代码

学习这个IOCP的时候走了不少的弯路呀...其中最弯的就是我选分不清楚重叠IO与IOCP..我汗...

 

所以在此之前呢我就先把重叠IO给先学了再学的IOCP..搞清楚了WINDOWS的异步机制...其实就是WSA开头的那些函数...

 

所谓异步.大家也可以想像成另启一个线程一样...只是过呢.这是是WINDOWS内核的问题..我只要调用这个WSA开头的函数就OK啦...

 

这当然是我理解的...呵呵...异步就是把这个工作放到一边儿去执行...就想你看电影一样,,网速比较慢..你把电脑开启,去吃饭去了..等你把

 

饭吃完电影就可以看咯这样子....然后你是不是就要点击播放播放这个电影呢...这就是你取得结果...取得结果是一个动作....而你吃饭是相当于

 

一个一直等待的过程....这又要交给WINDOWS的一个函数处理...GetQueuedCompletionStatus()   就是这家伙...它是一直阻塞的...

 

阻塞就是等到.WSA函数把工作做完.它就立马去取...就像排队买雪糕一样...哈哈.....

 

帖代码...

 

// IOCP.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include <stdlib.h> #include <WINSOCK2.H> typedef struct _PER_IO_DATA { OVERLAPPED Overlapped; SOCKET csocket; WSABUF DataBuf; char buffer[1024]; unsigned long len; unsigned long OperationType; unsigned long dwFlags; }PER_IO_DATA, *LPPER_IO_DATA; DWORD WINAPI CommandThread(LPVOID lparam); DWORD WINAPI WorkThread(LPVOID lparam); int main(int argc, char* argv[]) { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested=MAKEWORD(2,2); err=WSAStartup(wVersionRequested,&wsaData); if (err!=0) { printf("WSAStartup errors!/n"); return 0; } if (LOBYTE(wsaData.wVersion)!=2||HIBYTE(wsaData.wVersion)!=2) { WSACleanup(); return 0; } SOCKET sockSvr=WSASocket(AF_INET, SOCK_STREAM, 0, 0, 0, WSA_FLAG_OVERLAPPED ); if(INVALID_SOCKET==sockSvr) { printf("WSASocket errors!/n"); return 0; } SOCKADDR_IN addrSvr; ZeroMemory(&addrSvr, sizeof(SOCKADDR_IN)); addrSvr.sin_family = AF_INET; addrSvr.sin_port = htons(5000); addrSvr.sin_addr.S_un.S_addr = htonl(INADDR_ANY); int nRet = bind(sockSvr, (SOCKADDR*)&addrSvr, sizeof(SOCKADDR)); if(SOCKET_ERROR == nRet) { printf("bind errors!/n"); return 0; } nRet = listen(sockSvr, 500);//500:max number of connect request if(SOCKET_ERROR == nRet) { printf("listen errors!/n"); return 0; } HANDLE hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0 ); if(!hCompletionPort) { printf("CreateIoCompletionPort errors!/n"); return 0; } SYSTEM_INFO SystemInfo; UINT i = 0; DWORD dwThreadID = 0; GetSystemInfo(&SystemInfo); for(i=0; i<SystemInfo.dwNumberOfProcessors*2; i++) { HANDLE ThreadHandle = CreateThread( NULL, 0, WorkThread, hCompletionPort, 0, &dwThreadID ); if(ThreadHandle==NULL) { printf("CreateThread() failed with error %d/n", GetLastError()); return 0; } CloseHandle(ThreadHandle); } HANDLE ct = CreateThread( NULL, 0, CommandThread, hCompletionPort, 0, &dwThreadID ); if(ct==NULL) { printf("CreateThread() failed with error %d/n", GetLastError()); return 0; } CloseHandle(ct); while(TRUE) { SOCKADDR_IN addrAccept; ZeroMemory(&addrAccept, sizeof(SOCKADDR_IN)); int nSockLen = sizeof(SOCKADDR); SOCKET sockAccept = WSAAccept(sockSvr, (SOCKADDR*)&addrAccept, &nSockLen, 0, 0 ); if(INVALID_SOCKET == sockSvr) { printf("WSAAccept errors!/n"); return 0; } printf("connect/n"); if(NULL==CreateIoCompletionPort((HANDLE)sockAccept, hCompletionPort, (DWORD)sockSvr, 0)) { printf("CreateIoCompletionPort errors!/n"); return 0; } LPPER_IO_DATA pOlp = new PER_IO_DATA; ZeroMemory(pOlp, sizeof(PER_IO_DATA)); pOlp->csocket = sockAccept; pOlp->dwFlags = 0; pOlp->DataBuf.buf = pOlp->buffer; pOlp->DataBuf.len = 1024; pOlp->OperationType = FD_READ; nRet = WSARecv(pOlp->csocket, &pOlp->DataBuf, 1, &pOlp->len, &(pOlp->dwFlags), &(pOlp->Overlapped), NULL ); if(SOCKET_ERROR == nRet) { if(WSAGetLastError() != ERROR_IO_PENDING) { printf("WSARecv() failed with error %d/n", WSAGetLastError()); return 0; } } } return 0; } DWORD WINAPI CommandThread(LPVOID lparam) { HANDLE hc=(HANDLE)lparam; PER_IO_DATA ol; char command[10]; int i=0; SYSTEM_INFO SystemInfo; ol.OperationType=10; GetSystemInfo(&SystemInfo); while (1) { scanf("%s",&command); if(0==strcmp(command,"exit")) { for(i=0; i<SystemInfo.dwNumberOfProcessors*2; i++) { PostQueuedCompletionStatus(hc,0,0,(LPOVERLAPPED)&ol); Sleep(1000); } Sleep(1000); exit(0); } } } DWORD WINAPI WorkThread(LPVOID lparam) { HANDLE hCompletionPort = (HANDLE) lparam; void* re; DWORD berByte; OVERLAPPED* pOl; LPPER_IO_DATA ppol; while(TRUE) { BOOL bRet = GetQueuedCompletionStatus(hCompletionPort, &berByte, (LPDWORD)&re, (LPOVERLAPPED*)&pOl, INFINITE ); ppol=(LPPER_IO_DATA)pOl; if(bRet!=false) { switch (ppol->OperationType) { case FD_READ: if(berByte!=0) { printf("%s/n",ppol->DataBuf.buf); WSARecv(ppol->csocket, &ppol->DataBuf, 1, &ppol->len, &(ppol->dwFlags), &(ppol->Overlapped), NULL ); } else { printf("close socket ...../n"); closesocket(ppol->csocket); free(ppol); } case 10: printf("close socket ...../n"); closesocket(ppol->csocket); printf("exit thread ...../n"); ExitThread(0); default: printf("no data!!/n"); } } } return 0; }

 

代码我没写注释...主要是WINDOWS的这个异步机制...

 

懂了看什么代码都一样的....呵呵....我的只是个DEMO而已.....

 

2010-06-04  14:39:21