IOCP二

// IOCP2.cpp : Defines the entry point for the console application.

//

#include "stdafx.h"

#include <WinSock2.h>

#include <MSWSock.h>

#include <Windows.h>

#include <process.h>

#pragma comment(lib, "WS2_32.lib")

#define MAX_BUFFER 256

#define MAX_TIMEOUT 1000

#define MAX_SOCKET 1024

#define MAX_THREAD 64

#define MAX_ACCEPT 5

typedef enum _OPERATION_INFO_

{

OP_NULL,

OP_ACCEPT,

OP_READ,

OP_WRITE

}OPERATIONINFO;

typedef struct _PER_HANDLE_DATA_

public:

_PER_HANDLE_DATA_()

clean();

}

~_PER_HANDLE_DATA_()

protected:

void clean()

sock = INVALID_SOCKET;

memset(&addr, 0, sizeof(addr));

addr.sin_addr.S_un.S_addr = INADDR_ANY;

addr.sin_port = htons(0);

addr.sin_family = AF_INET;

SOCKET sock;

SOCKADDR_IN addr;

}PERHANDLEDATA, *PPERHANDLEDATA;

typedef struct _PER_IO_DTATA_

_PER_IO_DTATA_()

~_PER_IO_DTATA_()

ZeroMemory(&ol, sizeof(ol));

memset(buf, 0, sizeof(buf));

sAccept = INVALID_SOCKET;

sListen = INVALID_SOCKET;

wsaBuf.buf = buf;

wsaBuf.len = MAX_BUFFER;

opType = OP_NULL;

WSAOVERLAPPED ol;

SOCKET sAccept; // Only valid with AcceptEx

SOCKET sListen; // Only valid with AcceptEx

WSABUF wsaBuf;

char buf[MAX_BUFFER];

OPERATIONINFO opType;

}PERIODATA, *PPERIODATA;

HANDLE hThread[MAX_THREAD] = {0};

PERIODATA* pAcceptData[MAX_ACCEPT] = {0};

int g_nThread = 0;

BOOL g_bExitThread = FALSE;

LPFN_ACCEPTEX lpfnAcceptEx = NULL;

LPFN_GETACCEPTEXSOCKADDRS lpfnGetAcceptExSockAddrs = NULL;

GUID GuidAcceptEx = WSAID_ACCEPTEX;

GUID GuidGetAcceptExSockAddrs = WSAID_GETACCEPTEXSOCKADDRS;

unsigned __stdcall ThreadProc(LPVOID lParam);

BOOL PostAccept(PERIODATA* pIoData);

int _tmain(int argc, _TCHAR* argv[])

WORD wVersionRequested = MAKEWORD(2, 2);

WSADATA wsaData;

if(0 != WSAStartup(wVersionRequested, &wsaData))

printf("WSAStartup failed with error code: %d/n", GetLastError());

return EXIT_FAILURE;

if(2 != HIBYTE(wsaData.wVersion) || 2 != LOBYTE(wsaData.wVersion))

printf("Socket version not supported./n");

WSACleanup();

// Create IOCP

HANDLE hIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0);

if(NULL == hIOCP)

printf("CreateIoCompletionPort failed with error code: %d/n", WSAGetLastError());

// Create worker thread

SYSTEM_INFO si = {0};

GetSystemInfo(&si);

for(int i = 0; i < (int)si.dwNumberOfProcessors+2; i++)

hThread[g_nThread] = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, (LPVOID)hIOCP, 0, NULL);

if(NULL == hThread[g_nThread])

printf("_beginthreadex failed with error code: %d/n", GetLastError());

continue;

++g_nThread;

if(g_nThread > MAX_THREAD)

break;

// Create listen SOCKET

SOCKET sListen = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);

if(INVALID_SOCKET == sListen)

printf("WSASocket failed with error code: %d/n", WSAGetLastError());

goto EXIT_CODE;

// Associate SOCKET with IOCP

if(NULL == CreateIoCompletionPort((HANDLE)sListen, hIOCP, NULL, 0))

if(INVALID_SOCKET != sListen)

closesocket(sListen);

// Bind SOCKET

addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");

addr.sin_port = htons(5050);

if(SOCKET_ERROR == bind(sListen, (LPSOCKADDR)&addr, sizeof(addr)))

printf("bind failed with error code: %d/n", WSAGetLastError());

// Start Listen

if(SOCKET_ERROR == listen(sListen, 200))

printf("listen failed with error code: %d/n", WSAGetLastError());

printf("Server start, wait for client to connect .../n");

DWORD dwBytes = 0;

if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidAcceptEx, sizeof(GuidAcceptEx), &lpfnAcceptEx,

sizeof(lpfnAcceptEx), &dwBytes, NULL, NULL))

printf("WSAIoctl failed with error code: %d/n", WSAGetLastError());

if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidGetAcceptExSockAddrs,

sizeof(GuidGetAcceptExSockAddrs), &lpfnGetAcceptExSockAddrs, sizeof(lpfnGetAcceptExSockAddrs),

&dwBytes, NULL, NULL))

// Post MAX_ACCEPT accept

for(int i=0; i<MAX_ACCEPT; i++)

pAcceptData[i] = new PERIODATA;

pAcceptData[i]->sListen = sListen;

PostAccept(pAcceptData[i]);

// After 1 hour later, Server shutdown.

Sleep(1000 * 60 *60);

EXIT_CODE:

g_bExitThread = TRUE;

PostQueuedCompletionStatus(hIOCP, 0, NULL, NULL);

WaitForMultipleObjects(g_nThread, hThread, TRUE, INFINITE);

for(int i = 0; i < g_nThread; i++)

CloseHandle(hThread[g_nThread]);

if(pAcceptData[i])

delete pAcceptData[i];

pAcceptData[i] = NULL;

CloseHandle(hIOCP); // Close IOCP

return 0;

BOOL PostAccept(PERIODATA* pIoData)

if(INVALID_SOCKET == pIoData->sListen)

return FALSE;

pIoData->opType = OP_ACCEPT;

pIoData->sAccept = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);

if(INVALID_SOCKET == pIoData->sAccept)

if(FALSE == lpfnAcceptEx(pIoData->sListen, pIoData->sAccept, pIoData->wsaBuf.buf, pIoData->wsaBuf.len - ((sizeof(SOCKADDR_IN)+16)*2),

sizeof(SOCKADDR_IN)+16, sizeof(SOCKADDR_IN)+16, &dwBytes, &(pIoData->ol)))

if(WSA_IO_PENDING != WSAGetLastError())

printf("lpfnAcceptEx failed with error code: %d/n", WSAGetLastError());

return TRUE;

unsigned __stdcall ThreadProc(LPVOID lParam)

HANDLE hIOCP = (HANDLE)lParam;

PERHANDLEDATA* pPerHandleData = NULL;

PERIODATA* pPerIoData = NULL;

WSAOVERLAPPED* lpOverlapped = NULL;

DWORD dwTrans = 0;

DWORD dwFlags = 0;

while(!g_bExitThread)

BOOL bRet = GetQueuedCompletionStatus(hIOCP, &dwTrans, (PULONG_PTR)&pPerHandleData, &lpOverlapped, MAX_TIMEOUT);

if(!bRet)

// Timeout and exit thread

if(WAIT_TIMEOUT == GetLastError())

// Error

printf("GetQueuedCompletionStatus failed with error: %d/n", GetLastError());

else

pPerIoData = CONTAINING_RECORD(lpOverlapped, PERIODATA, ol);

if(NULL == pPerIoData)

// Exit thread

if((0 == dwTrans) && (OP_READ == pPerIoData->opType || OP_WRITE == pPerIoData->opType))

// Client leave.

printf("Client: