Winsock Overlapped IO模型（一）

// NetWork.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

typedef enum _OPERATION_INFO
{
	OP_NULL,
	OP_READ,
	OP_WRITE
}OPERATIONINFO;

typedef struct _OVERLAPPED_INFO_
{	
public:
	_OVERLAPPED_INFO_()
	{
		Clean();
	}
	~_OVERLAPPED_INFO_()
	{
		Clean();
	}
protected:
	void Clean()
	{
		ol.hEvent = NULL;
		sock = INVALID_SOCKET;
		memset(&addr, 0, sizeof(addr));
		addr.sin_addr.S_un.S_addr = INADDR_ANY;
		addr.sin_family = AF_INET;
		addr.sin_port = htons(0);
		wsaBuf.buf = szBuf;
		wsaBuf.len = MAX_BUFFER;
		memset(szBuf, 0, sizeof(szBuf));
		nOpType = OP_NULL;
	}
public:
	WSAOVERLAPPED ol;
	SOCKET sock;
	SOCKADDR_IN addr;
	WSABUF wsaBuf;
	char szBuf[MAX_BUFFER];
	OPERATIONINFO nOpType;
}OVERLAPPEDINFO, *POVERLAPPEDINFO;

WSAEVENT g_event[WSA_MAXIMUM_WAIT_EVENTS] = {0};
OVERLAPPEDINFO* g_ol[WSA_MAXIMUM_WAIT_EVENTS] = {0};
int g_nIndex = 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();
		return EXIT_FAILURE;
	}
	SOCKET sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(INVALID_SOCKET == sListen)
	{
		printf("socket failed with error code: %d/n", WSAGetLastError());
		WSACleanup();
		return EXIT_FAILURE;
	}
	SOCKADDR_IN addr = {0};
	addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
	addr.sin_family = AF_INET;
	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;
		WSACleanup();
		return EXIT_FAILURE;
	}
	if(SOCKET_ERROR == listen(sListen, 5))
	{
		printf("listen failed with error code: %d/n", WSAGetLastError());
		closesocket(sListen);
		sListen = INVALID_SOCKET;
		WSACleanup();
		return EXIT_FAILURE;
	}
	HANDLE hThread = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, NULL, 0, NULL);
	if(NULL == hThread)
	{
		printf("_beginthreadex failed with error code: %d/n", GetLastError());
		closesocket(sListen);
		sListen = INVALID_SOCKET;
		WSACleanup();
		return EXIT_FAILURE;
	}
	
	printf("Server start listen and wait for client to connect .../n");

	SOCKADDR_IN remote = {0};
	int len = sizeof(remote);

	while(TRUE)
	{
		if(g_nIndex >= WSA_MAXIMUM_WAIT_EVENTS)
		{
			printf("Too many connect, wait for disconnect .../n");
			continue;
		}
		SOCKET sNew = accept(sListen, (LPSOCKADDR)&remote, &len);
		if(INVALID_SOCKET == sNew)
		{
			printf("accept failed with error code: %d/n", WSAGetLastError());
			continue;
		}
		printf("<IP: %s, Port: %d> client come in./n", inet_ntoa(remote.sin_addr), ntohs(remote.sin_port));
		DWORD dwTrans = 0;
		DWORD dwFlags = 0;
		g_ol[g_nIndex] = new OVERLAPPEDINFO;
		g_ol[g_nIndex]->sock = sNew;
		memcpy(&g_ol[g_nIndex]->addr, &remote, sizeof(remote));
		g_event[g_nIndex] = g_ol[g_nIndex]->ol.hEvent = WSACreateEvent();
		g_ol[g_nIndex]->nOpType = OP_READ;
		if(SOCKET_ERROR == WSARecv(g_ol[g_nIndex]->sock, &(g_ol[g_nIndex]->wsaBuf), 1, &dwTrans, &dwFlags, (LPWSAOVERLAPPED)&(g_ol[g_nIndex]->ol), NULL))
		{
			if(ERROR_IO_PENDING != WSAGetLastError())
			{
				printf("WSARecv failed with error code: %d/n", WSAGetLastError());
				closesocket(g_ol[g_nIndex]->sock);
				delete g_ol[g_nIndex];
				g_ol[g_nIndex] = NULL;
				continue;
			}
		}
		g_nIndex++;
	}
	// End of thread and exit process
	g_bExitThread = TRUE; // Exit thread.
	WaitForSingleObject(hThread, INFINITE);
	CloseHandle(hThread);

	WSACleanup();
	return 0;
}

unsigned __stdcall ThreadProc(LPVOID lParam)
{
	while(!g_bExitThread)
	{
		DWORD dwTrans = 0;
		DWORD dwFlags = 0;
		DWORD dwIndex = WSAWaitForMultipleEvents(g_nIndex, g_event, FALSE, MAX_TIMEOUT, FALSE);
		if(WSA_WAIT_FAILED == dwIndex || WSA_WAIT_TIMEOUT == dwIndex)
		{
			continue;
		}
		
		dwIndex -= WSA_WAIT_EVENT_0;
		for(int i = dwIndex; i < g_nIndex; i++)
		{
			dwIndex = WSAWaitForMultipleEvents(1, &g_event[dwIndex], TRUE, INFINITE, FALSE);
			if(WSA_WAIT_FAILED == dwIndex)
			{
				continue;
			}
			dwIndex -= WSA_WAIT_EVENT_0;
			WSAResetEvent(g_event[dwIndex]);
			BOOL bRet = WSAGetOverlappedResult(g_ol[dwIndex]->sock, &(g_ol[dwIndex]->ol), &dwTrans, FALSE, &dwFlags);
			if(bRet)
			{
				if(dwTrans > 0)
				{
					switch(g_ol[dwIndex]->nOpType)
					{
					case OP_READ:
						{
							printf("recv client <%s:%d> data: %s/n", inet_ntoa(g_ol[dwIndex]->addr.sin_addr), ntohs(g_ol[dwIndex]->addr.sin_port), g_ol[dwIndex]->szBuf);
							g_ol[dwIndex]->nOpType = OP_WRITE;
							memset(&(g_ol[dwIndex]->ol), 0, sizeof(g_ol[dwIndex]->ol));
							if(SOCKET_ERROR == WSASend(g_ol[dwIndex]->sock, &(g_ol[dwIndex]->wsaBuf), 1, &dwTrans, dwFlags, &(g_ol[dwIndex]->ol), NULL))
							{
								if(ERROR_IO_PENDING != WSAGetLastError())
								{
									printf("WSASend failed with error code: %d/n", WSAGetLastError());
									continue;
								}
							}
						}
						break;

					case OP_WRITE:
						{
							g_ol[dwIndex]->nOpType = OP_READ;
							memset(&(g_ol[dwIndex]->ol), 0, sizeof(g_ol[dwIndex]->ol));
							memset(g_ol[dwIndex]->szBuf, 0, sizeof(g_ol[dwIndex]->szBuf));
							dwFlags = 0;
							g_ol[dwIndex]->wsaBuf.buf = g_ol[dwIndex]->szBuf;
							dwTrans = g_ol[dwIndex]->wsaBuf.len = MAX_BUFFER;
							if(SOCKET_ERROR == WSARecv(g_ol[dwIndex]->sock, &(g_ol[dwIndex]->wsaBuf), 1, &dwTrans, &dwFlags, &(g_ol[dwIndex]->ol), NULL))
							{
								if(ERROR_IO_PENDING != WSAGetLastError())
								{
									printf("WSARecv failed with error code: %d/n", WSAGetLastError());
									continue;
								}
							}
						}
						break;

					default:
						break;
					}				
				}
				else
				{
					printf("client <%s : %d> leave./n", inet_ntoa(g_ol[dwIndex]->addr.sin_addr), ntohs(g_ol[dwIndex]->addr.sin_port));
					closesocket(g_ol[dwIndex]->sock);
					WSACloseEvent(g_event[dwIndex]);
					delete g_ol[dwIndex];
					// Rebuild array
					for(int i= dwIndex; i<g_nIndex-1; i++)
					{
						g_event[i] = g_event[i+1];
						g_ol[i] = g_ol[i+1];
					}
					g_nIndex--;
				}
			}
			else
			{
				printf("WSAGetOverlappedResult failed with error: %d/n", WSAGetLastError());
			}
		}
	}
	printf("Server to exit ... /n");

	return 0;
}