IOCP一

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

//

#include "stdafx.h"

#include <WinSock2.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

typedef enum _OPERATION_INFO_

{

OP_NULL,

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_()

private:

ZeroMemory(&ol, sizeof(ol));

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

wsaBuf.buf = buf;

wsaBuf.len = MAX_BUFFER;

opType = OP_NULL;

WSAOVERLAPPED ol;

WSABUF wsaBuf;

char buf[MAX_BUFFER];

OPERATIONINFO opType;

}PERIODATA, *PPERIODATA;

HANDLE hThread[MAX_THREAD] = {0};

int g_nThread = 0;

BOOL g_bExitThread = FALSE;

unsigned __stdcall ThreadProc(LPVOID lParam);

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

WSADATA wsaData;

if(0 != WSAStartup(MAKEWORD(2, 2), &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 I/O Completion Port

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

if(NULL == hIOCP)

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

// 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 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;

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());

closesocket(sListen);

sListen = INVALID_SOCKET;

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

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

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

while(TRUE)

SOCKADDR_IN remote;

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

int len = sizeof(remote);

SOCKET sNew = WSAAccept(sListen, (LPSOCKADDR)&remote, &len, NULL, NULL);

if(INVALID_SOCKET == sNew)

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

printf("Client