登 录
// IOCP2.cpp : Defines the entry point for the console application.<br /> //</p> <p>#include "stdafx.h"<br /> #include <WinSock2.h><br /> #include <MSWSock.h><br /> #include <Windows.h><br /> #include <process.h><br /> #pragma comment(lib, "WS2_32.lib")</p> <p>#define MAX_BUFFER 256<br /> #define MAX_TIMEOUT 1000<br /> #define MAX_SOCKET 1024<br /> #define MAX_THREAD 64<br /> #define MAX_ACCEPT 5</p> <p>typedef enum _OPERATION_INFO_<br /> {<br /> OP_NULL,<br /> OP_ACCEPT,<br /> OP_READ,<br /> OP_WRITE<br /> }OPERATIONINFO;</p> <p>typedef struct _PER_HANDLE_DATA_<br /> {<br /> public:<br /> _PER_HANDLE_DATA_()<br /> {<br /> clean();<br /> }<br /> ~_PER_HANDLE_DATA_()<br /> {<br /> clean();<br /> }<br /> protected:<br /> void clean()<br /> {<br /> sock = INVALID_SOCKET;<br /> memset(&addr, 0, sizeof(addr));<br /> addr.sin_addr.S_un.S_addr = INADDR_ANY;<br /> addr.sin_port = htons(0);<br /> addr.sin_family = AF_INET;<br /> }<br /> public:<br /> SOCKET sock;<br /> SOCKADDR_IN addr;<br /> }PERHANDLEDATA, *PPERHANDLEDATA;</p> <p>typedef struct _PER_IO_DTATA_<br /> {<br /> public:<br /> _PER_IO_DTATA_()<br /> {<br /> clean();<br /> }<br /> ~_PER_IO_DTATA_()<br /> {<br /> clean();<br /> }<br /> void clean()<br /> {<br /> ZeroMemory(&ol, sizeof(ol));<br /> memset(buf, 0, sizeof(buf));<br /> sAccept = INVALID_SOCKET;<br /> sListen = INVALID_SOCKET;<br /> wsaBuf.buf = buf;<br /> wsaBuf.len = MAX_BUFFER;<br /> opType = OP_NULL;<br /> }<br /> public:<br /> WSAOVERLAPPED ol;<br /> SOCKET sAccept; // Only valid with AcceptEx<br /> SOCKET sListen; // Only valid with AcceptEx<br /> WSABUF wsaBuf;<br /> char buf[MAX_BUFFER];<br /> OPERATIONINFO opType;<br /> }PERIODATA, *PPERIODATA;</p> <p>HANDLE hThread[MAX_THREAD] = {0};<br /> PERIODATA* pAcceptData[MAX_ACCEPT] = {0};<br /> int g_nThread = 0;<br /> BOOL g_bExitThread = FALSE;<br /> LPFN_ACCEPTEX lpfnAcceptEx = NULL;<br /> LPFN_GETACCEPTEXSOCKADDRS lpfnGetAcceptExSockAddrs = NULL;<br /> GUID GuidAcceptEx = WSAID_ACCEPTEX;<br /> GUID GuidGetAcceptExSockAddrs = WSAID_GETACCEPTEXSOCKADDRS;</p> <p>unsigned __stdcall ThreadProc(LPVOID lParam);<br /> BOOL PostAccept(PERIODATA* pIoData);</p> <p>int _tmain(int argc, _TCHAR* argv[])<br /> {<br /> WORD wVersionRequested = MAKEWORD(2, 2);<br /> WSADATA wsaData;<br /> if(0 != WSAStartup(wVersionRequested, &wsaData))<br /> {<br /> printf("WSAStartup failed with error code: %d/n", GetLastError());<br /> return EXIT_FAILURE;<br /> }</p> <p> if(2 != HIBYTE(wsaData.wVersion) || 2 != LOBYTE(wsaData.wVersion))<br /> {<br /> printf("Socket version not supported./n");<br /> WSACleanup();<br /> return EXIT_FAILURE;<br /> }</p> <p> // Create IOCP<br /> HANDLE hIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0);<br /> if(NULL == hIOCP)<br /> {<br /> printf("CreateIoCompletionPort failed with error code: %d/n", WSAGetLastError());<br /> WSACleanup();<br /> return EXIT_FAILURE;<br /> }</p> <p> // Create worker thread<br /> SYSTEM_INFO si = {0};<br /> GetSystemInfo(&si);<br /> for(int i = 0; i < (int)si.dwNumberOfProcessors+2; i++)<br /> {<br /> hThread[g_nThread] = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, (LPVOID)hIOCP, 0, NULL);<br /> if(NULL == hThread[g_nThread])<br /> {<br /> printf("_beginthreadex failed with error code: %d/n", GetLastError());<br /> continue;<br /> }<br /> ++g_nThread;</p> <p> if(g_nThread > MAX_THREAD)<br /> {<br /> break;<br /> }<br /> }</p> <p> // Create listen SOCKET<br /> SOCKET sListen = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);<br /> if(INVALID_SOCKET == sListen)<br /> {<br /> printf("WSASocket failed with error code: %d/n", WSAGetLastError());<br /> goto EXIT_CODE;<br /> }</p> <p> // Associate SOCKET with IOCP<br /> if(NULL == CreateIoCompletionPort((HANDLE)sListen, hIOCP, NULL, 0))<br /> {<br /> printf("CreateIoCompletionPort failed with error code: %d/n", WSAGetLastError());<br /> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> goto EXIT_CODE;<br /> }</p> <p> // Bind SOCKET<br /> SOCKADDR_IN addr;<br /> memset(&addr, 0, sizeof(addr));<br /> addr.sin_family = AF_INET;<br /> addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");<br /> addr.sin_port = htons(5050);<br /> if(SOCKET_ERROR == bind(sListen, (LPSOCKADDR)&addr, sizeof(addr)))<br /> {<br /> printf("bind failed with error code: %d/n", WSAGetLastError());<br /> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> goto EXIT_CODE;<br /> }</p> <p> // Start Listen<br /> if(SOCKET_ERROR == listen(sListen, 200))<br /> {<br /> printf("listen failed with error code: %d/n", WSAGetLastError());<br /> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> goto EXIT_CODE;<br /> }</p> <p> printf("Server start, wait for client to connect .../n");</p> <p> DWORD dwBytes = 0;<br /> if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidAcceptEx, sizeof(GuidAcceptEx), &lpfnAcceptEx,<br /> sizeof(lpfnAcceptEx), &dwBytes, NULL, NULL))<br /> {<br /> printf("WSAIoctl failed with error code: %d/n", WSAGetLastError());<br /> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> goto EXIT_CODE;<br /> }</p> <p> if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidGetAcceptExSockAddrs,<br /> sizeof(GuidGetAcceptExSockAddrs), &lpfnGetAcceptExSockAddrs, sizeof(lpfnGetAcceptExSockAddrs),<br /> &dwBytes, NULL, NULL))<br /> {<br /> printf("WSAIoctl failed with error code: %d/n", WSAGetLastError());<br /> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> goto EXIT_CODE;<br /> }</p> <p> // Post MAX_ACCEPT accept<br /> for(int i=0; i<MAX_ACCEPT; i++)<br /> {<br /> pAcceptData[i] = new PERIODATA;<br /> pAcceptData[i]->sListen = sListen;<br /> PostAccept(pAcceptData[i]);<br /> }<br /> // After 1 hour later, Server shutdown.<br /> Sleep(1000 * 60 *60);</p> <p>EXIT_CODE:<br /> g_bExitThread = TRUE;</p> <p> PostQueuedCompletionStatus(hIOCP, 0, NULL, NULL);<br /> WaitForMultipleObjects(g_nThread, hThread, TRUE, INFINITE);<br /> for(int i = 0; i < g_nThread; i++)<br /> {<br /> CloseHandle(hThread[g_nThread]);<br /> }</p> <p> for(int i=0; i<MAX_ACCEPT; i++)<br /> {<br /> if(pAcceptData[i])<br /> {<br /> delete pAcceptData[i];<br /> pAcceptData[i] = NULL;<br /> }<br /> }</p> <p> if(INVALID_SOCKET != sListen)<br /> {<br /> closesocket(sListen);<br /> sListen = INVALID_SOCKET;<br /> }<br /> CloseHandle(hIOCP); // Close IOCP</p> <p> WSACleanup();<br /> return 0;<br /> }</p> <p>BOOL PostAccept(PERIODATA* pIoData)<br /> {<br /> if(INVALID_SOCKET == pIoData->sListen)<br /> {<br /> return FALSE;<br /> }</p> <p> DWORD dwBytes = 0;<br /> pIoData->opType = OP_ACCEPT;<br /> pIoData->sAccept = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);<br /> if(INVALID_SOCKET == pIoData->sAccept)<br /> {<br /> printf("WSASocket failed with error code: %d/n", WSAGetLastError());<br /> return FALSE;<br /> }</p> <p> if(FALSE == lpfnAcceptEx(pIoData->sListen, pIoData->sAccept, pIoData->wsaBuf.buf, pIoData->wsaBuf.len - ((sizeof(SOCKADDR_IN)+16)*2),<br /> sizeof(SOCKADDR_IN)+16, sizeof(SOCKADDR_IN)+16, &dwBytes, &(pIoData->ol)))<br /> {<br /> if(WSA_IO_PENDING != WSAGetLastError())<br /> {<br /> printf("lpfnAcceptEx failed with error code: %d/n", WSAGetLastError());</p> <p> return FALSE;<br /> }<br /> }<br /> return TRUE;<br /> }</p> <p>unsigned __stdcall ThreadProc(LPVOID lParam)<br /> {<br /> HANDLE hIOCP = (HANDLE)lParam;</p> <p> PERHANDLEDATA* pPerHandleData = NULL;<br /> PERIODATA* pPerIoData = NULL;<br /> WSAOVERLAPPED* lpOverlapped = NULL;<br /> DWORD dwTrans = 0;<br /> DWORD dwFlags = 0;<br /> while(!g_bExitThread)<br /> {<br /> BOOL bRet = GetQueuedCompletionStatus(hIOCP, &dwTrans, (PULONG_PTR)&pPerHandleData, &lpOverlapped, MAX_TIMEOUT);<br /> if(!bRet)<br /> {<br /> // Timeout and exit thread<br /> if(WAIT_TIMEOUT == GetLastError())<br /> {<br /> continue;<br /> }<br /> // Error<br /> printf("GetQueuedCompletionStatus failed with error: %d/n", GetLastError());<br /> continue;<br /> }<br /> else<br /> {<br /> pPerIoData = CONTAINING_RECORD(lpOverlapped, PERIODATA, ol);<br /> if(NULL == pPerIoData)<br /> {<br /> // Exit thread<br /> break;<br /> }</p> <p> if((0 == dwTrans) && (OP_READ == pPerIoData->opType || OP_WRITE == pPerIoData->opType))<br /> {<br /> // Client leave.<br /> printf("Client: <%s : %d> leave./n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port));<br /> closesocket(pPerHandleData->sock);<br /> delete pPerHandleData;<br /> delete pPerIoData;<br /> continue;<br /> }<br /> else<br /> {<br /> switch(pPerIoData->opType)<br /> {<br /> case OP_ACCEPT: // Accept<br /> {<br /> SOCKADDR_IN* remote = NULL;<br /> SOCKADDR_IN* local = NULL;<br /> int remoteLen = sizeof(SOCKADDR_IN);<br /> int localLen = sizeof(SOCKADDR_IN);<br /> lpfnGetAcceptExSockAddrs(pPerIoData->wsaBuf.buf, pPerIoData->wsaBuf.len - ((sizeof(SOCKADDR_IN)+16)*2),<br /> sizeof(SOCKADDR_IN)+16, sizeof(SOCKADDR_IN)+16, (LPSOCKADDR*)&local, &localLen, (LPSOCKADDR*)&remote, &remoteLen);<br /> printf("Client <%s : %d> come in./n", inet_ntoa(remote->sin_addr), ntohs(remote->sin_port));<br /> printf("Recv Data: <%s : %d> %s./n", inet_ntoa(remote->sin_addr), ntohs(remote->sin_port), pPerIoData->wsaBuf.buf);</p> <p> if(NULL != pPerHandleData)<br /> {<br /> delete pPerHandleData;<br /> pPerHandleData = NULL;<br /> }<br /> pPerHandleData = new PERHANDLEDATA;<br /> pPerHandleData->sock = pPerIoData->sAccept;</p> <p> PERHANDLEDATA* pPerHandle = new PERHANDLEDATA;<br /> pPerHandle->sock = pPerIoData->sAccept;<br /> PERIODATA* pPerIo = new PERIODATA;<br /> pPerIo->opType = OP_WRITE;<br /> strcpy_s(pPerIo->buf, MAX_BUFFER, pPerIoData->buf);<br /> DWORD dwTrans = strlen(pPerIo->buf);<br /> memcpy(&(pPerHandleData->addr), remote, sizeof(SOCKADDR_IN));<br /> // Associate with IOCP<br /> if(NULL == CreateIoCompletionPort((HANDLE)(pPerHandleData->sock), hIOCP, (ULONG_PTR)pPerHandleData, 0))<br /> {<br /> printf("CreateIoCompletionPort failed with error code: %d/n", GetLastError());<br /> closesocket(pPerHandleData->sock);<br /> delete pPerHandleData;<br /> continue;<br /> }</p> <p> // Post Accept<br /> memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));<br /> PostAccept(pPerIoData);</p> <p> // Post Receive<br /> DWORD dwFlags = 0;<br /> if(SOCKET_ERROR == WSASend(pPerHandle->sock, &(pPerIo->wsaBuf), 1,<br /> &dwTrans, dwFlags, &(pPerIo->ol), NULL))<br /> {<br /> if(WSA_IO_PENDING != WSAGetLastError())<br /> {<br /> printf("WSASend failed with error code: %d/n", WSAGetLastError());<br /> closesocket(pPerHandle->sock);<br /> delete pPerHandle;<br /> delete pPerIo;<br /> continue;<br /> }<br /> }<br /> }<br /> break;</p> <p> case OP_READ: // Read<br /> printf("recv client <%s : %d> data: %s/n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port), pPerIoData->buf);<br /> pPerIoData->opType = OP_WRITE;<br /> memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));<br /> if(SOCKET_ERROR == WSASend(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, dwFlags, &(pPerIoData->ol), NULL))<br /> {<br /> if(WSA_IO_PENDING != WSAGetLastError())<br /> {<br /> printf("WSASend failed with error code: %d./n", WSAGetLastError());<br /> continue;<br /> }<br /> }<br /> break;</p> <p> case OP_WRITE: // Write<br /> {<br /> pPerIoData->opType = OP_READ;<br /> dwFlags = 0;<br /> memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));<br /> memset(pPerIoData->buf, 0, sizeof(pPerIoData->buf));<br /> pPerIoData->wsaBuf.buf = pPerIoData->buf;<br /> dwTrans = pPerIoData->wsaBuf.len = MAX_BUFFER;<br /> if(SOCKET_ERROR == WSARecv(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, &dwFlags, &(pPerIoData->ol), NULL))<br /> {<br /> if(WSA_IO_PENDING != WSAGetLastError())<br /> {<br /> printf("WSARecv failed with error code: %d./n", WSAGetLastError());<br /> continue;<br /> }<br /> }<br /> }<br /> break;</p> <p> default:<br /> break;<br /> }<br /> }<br /> }<br /> }<br /> return 0;<br /> }
抱歉!评论已关闭.