简单的使用 IOCP 模型的 TCP 服务器例子

例子中有两种类型的线程-主线程和它创建的线程。主线程创建监听套接字,创建额外的工作线程,关联IOCP,负责等待和接受到来的连接等。由主线程创建的线程负责处理I/O事件,这些线程调用GetQueuedCompletionStatus函数在完成端口对象上等待完成的 I/O操作。

GetQueuedCompletionStatus函数返回后,说明发生了如下事件之一。

(1)GetQueuedCompletionStatus调用失败,说明在此套接字上有错误发生。

(2)BytesTransferred为0说明套接字被对方关闭。注意,per-handle数据用来引用与I/O操作相关的套接字。

(3)I/O请求成功完成。通过per-I/O数据(这是程序自定义的结构)中的OperationType域查看哪个I/O请求完成了。

/
// IOCPDemo.cpp文件	

#include "../common/initsock.h"
#include <stdio.h>
#include <windows.h>

CInitSock theSock;	// 初始化Winsock库

#define BUFFER_SIZE 1024

typedef struct _PER_HANDLE_DATA		// per-handle(句柄惟一)数据
{
	SOCKET s;			// 对应的套节字句柄
	sockaddr_in addr;		// 客户方地址
} PER_HANDLE_DATA, *PPER_HANDLE_DATA;

typedef struct _PER_IO_DATA		// per-I/O(I/O惟一)数据
{
	OVERLAPPED ol;			// 重叠结构
	char buf[BUFFER_SIZE];		// 数据缓冲区
	int nOPerationType;		// 操作类型
#define OP_READ		1
#define OP_WRITE	2
#define OP_ACCPT	3
} PER_IO_DATA, *PPER_IO_DATA;

DWORD WINAPI ServerThread(LPVOID lpParam)
{
	// 得到完成端口对象句柄
	HANDLE hComplete = (HANDLE)lpParam;

	DWORD dwTrans;
	PPER_HANDLE_DATA pPerHandle;
	PPER_IO_DATA pPerIO;
	while (TRUE) 
	{
		// 4.在关联到此完成端口的所有套节字上等待I/O完成
		// 在这些I/O操作完成时,I/O系统会向完成端口对象发送一个完成通知封包。
		// 取得有事件发生的套接字信息；
		BOOL bOK = ::GetQueuedCompletionStatus(hComplete, &dwTrans, 
			(LPDWORD)&pPerHandle, (LPOVERLAPPED *)&pPerIO, WSA_INFINITE);
		
		if (!bOK)	// 在此套节字上有错误发生
		{
			::closesocket(pPerHandle->s);
			::GlobalFree(pPerHandle);
			::GlobalFree(pPerIO);
			continue;
		}
		
		// 套节字被对方关闭		
		if (dwTrans == 0 &&				
			(pPerIO->nOPerationType = OP_READ || pPerIO->nOPerationType == OP_WRITE)) 
		{
			::closesocket(pPerHandle->s);
			::GlobalFree(pPerHandle);
			::GlobalFree(pPerIO);
			continue;
		}

		// 通过per-I/O数据中的nOperationType域查看什么I/O请求完成了
		switch(pPerIO->nOPerationType) 
		{
		case OP_READ:		// 完成一个接收请求
			{
				pPerIO->buf[dwTrans] = '\0';
				printf(pPerIO->buf);

				// 继续投递接收I/O请求
				WSABUF buf;
				buf.buf = pPerIO->buf;
				buf.len = BUFFER_SIZE;
				pPerIO->nOPerationType = OP_READ;

				DWORD nFlags = 0;
				::WSARecv(pPerHandle->s, &buf, 1, &dwTrans, &nFlags, &pPerIO->ol, NULL);
			}
			break;
		case OP_WRITE:	// 本例中没有投递这些类型的I/O请求
		case OP_ACCPT:			
			break;
		}
	}
	return 0;
}


void main()
{
	int nPort = 4567;
	// 1.创建完成端口对象，创建工作线程处理完成端口对象中事件
	HANDLE hCompletion = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE, 0, 0, 0);
	::CreateThread(NULL, 0, ServerThread, (LPVOID)hCompletion, 0, 0);
	
	// 创建监听套节字，绑定到本地地址，开始监听
	SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
	SOCKADDR_IN si;
	si.sin_family = AF_INET;
	si.sin_port = ::ntohs(nPort);
	si.sin_addr.S_un.S_addr = INADDR_ANY;
	::bind(sListen, (sockaddr *)&si, sizeof(si));
	::listen(sListen, 5);

	// 循环处理到来的连接
	while (TRUE) 
	{
		// 等待接受未决的连接请求
		SOCKADDR_IN saRemote;
		int nRemoteLen = sizeof(saRemote);
		SOCKET sNew = ::accept(sListen, (sockaddr *)&saRemote, &nRemoteLen);
		
		// 2.接受网络连接之后，为它创建一个per-handle数据，
		// 并将它们(socket)关联到完成端口对象；
		PPER_HANDLE_DATA pPerHandle = (PPER_HANDLE_DATA)::GlobalAlloc(GPTR, sizeof(PER_HANDLE_DATA));
		pPerHandle->s = sNew;
		memcpy(&pPerHandle->addr, &saRemote, nRemoteLen);
		
		::CreateIoCompletionPort((HANDLE)pPerHandle->s, hCompletion, (DWORD)pPerHandle, 0);
		
		// 3.投递一个接收请求
		PPER_IO_DATA pPerIO = (PPER_IO_DATA)::GlobalAlloc(GPTR, sizeof(PER_IO_DATA));
		pPerIO->nOPerationType = OP_READ;
		WSABUF buf;
		buf.buf = pPerIO->buf;
		buf.len = BUFFER_SIZE;
		DWORD dwRecv;
		DWORD dwFlags = 0;

		::WSARecv(pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pPerIO->ol, NULL);
	}


}