tcprunner.cpp

Socketlib: 一个轻量级的C++ 封装Socket C API 网络编程框架。 它简化了Socket异步事件分派、进程间Socket通信的并发OO网络应用和服务的开发。 目前

//**********************************************************************
//
// Copyright (C) 2005-2007 Zhang bao yuan([e-mail]).
// All rights reserved.
//
// This copy of Socketlib is licensed to you under the terms described 
// in the LICENSE.txt file included in this distribution.
//
//**********************************************************************

#include ".\tcprunner.h"
#include "SocketLib.h"
#include "SocketSource.h"
#include "SocketEvent.h"
namespace SL
{

CTcpRunner::CTcpRunner(void)
{
}

CTcpRunner::~CTcpRunner(void)
{
	Close();
}

bool CTcpRunner::Close()
{
	if (NULL == m_hCompletionPort)
		return false;

	//使工作线程退出
	for (int i=0; i<m_nWorkThreadNum; i++)
	{
		//发出退出消息
		::PostQueuedCompletionStatus(m_hCompletionPort,-1,-1,NULL);
	}
	m_threadGroup.Stop(500);
	m_threadGroup.Kill();

	CloseHandle(m_hCompletionPort);
	m_hCompletionPort = NULL;

	return true;
}

bool CTcpRunner::Open(unsigned short nConcurrencyThreadNum, unsigned short nWorkThreadNum)
{
	Close();

	if (nWorkThreadNum > 0)
		m_nWorkThreadNum = nWorkThreadNum;

	if ( nConcurrencyThreadNum <= nWorkThreadNum )
		m_nConcurrencyThreadNum = nConcurrencyThreadNum;
	else
		m_nConcurrencyThreadNum = 0;

	//创建完成端口句柄
	m_hCompletionPort = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,m_nConcurrencyThreadNum);
	if (INVALID_HANDLE_VALUE == m_hCompletionPort)
	{
#ifdef SHOW_PRINTFINFO
		printf("Create iocompletionPort failure!\r\n");
#endif
		return false;
	}

    //启动Socket工作线程组
	m_threadGroup.Start(CTcpRunner::Run, this, m_nWorkThreadNum);

	return true;
}

unsigned int WINAPI CTcpRunner::Run(LPVOID lParam)
{
	CTcpRunner *pRun		= (CTcpRunner*)lParam;
	HANDLE hCompletionPort	= pRun->m_hCompletionPort;

	int		nRet;
	DWORD	nByteTransferred;
	PPER_HANDLE_DATA	   pPerHandle;
	PPER_IO_OPERATION_BASE pPerIoBase;

	for (;;)
	{
		nByteTransferred = 0;
		pPerHandle = NULL;
		pPerIoBase = NULL;
		BOOL bSuccess = ::GetQueuedCompletionStatus(hCompletionPort,
												&nByteTransferred,
												(LPDWORD)&pPerHandle,
												(LPOVERLAPPED*)&pPerIoBase,
												INFINITE);
		//退出信号到达，退出线程
		if ( -1==nByteTransferred ) 
		{
			return 1;
		}
		if (NULL == pPerHandle)
			continue;

		//客户机已经断开连接或者连接出现错误
		if ( 0 == nByteTransferred )
		{
			pPerHandle->pSocketSource->Disconnect(pPerHandle->hSocket);
			continue;
		}
		if ( (NULL == pPerIoBase) || (!bSuccess) )  //出现其它异常情况
			continue;

		if (pPerIoBase->OperType == RECV_POSTED)
		{//数据接收

			//消息处理
			PPER_IO_OPERATION_DATA pPerIoData =(PPER_IO_OPERATION_DATA)pPerIoBase;
			//pPerIoData->szDataBuffer[nByteTransferred] = 0;
			//nRet = pPerHandle->pSocketSource->Recv(pPerHandle->hSocket, pPerHandle->pAttachInfo, 
			//	pPerIoData->szDataBuffer, nByteTransferred);
			nRet = pPerHandle->pSocketSource->m_pSocketEvent->OnRecv(pPerHandle->pSocketSource,
				pPerHandle->hSocket, pPerHandle->pAttachInfo, pPerIoData->szDataBuffer, nByteTransferred);
			if (nRet >= 0)  //小于0，表示在此Socket上不再接收数据
				pPerHandle->pSocketSource->PostRecv(pPerHandle->hSocket, pPerIoData);

		}
		else if (pPerIoBase->OperType == SEND_POSTED)
		{//数据发送完成
			//pPerHandle->pSocketSource->PostSend(PerHandle->hSocket);
			continue;
		}
	}

	return 0;

}

}