lyfz_repos/经典版源码/lyfzClient/branches/715/ClientTunnel.cpp

//ClientTunnel.cpp
#include "stdafx.h"
#include "ClientTunnel.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#ifdef VC60

static void netGetDataBufLenCb(IN SOCKET hSocket, IN void *pHeaderBuf, 
							   OUT unsigned long *nDataLen);

static void netHandleNetEvent(IN SOCKET hSocket, IN ETransportEvent eEvent, 
							  IN void *pDataBuf, IN unsigned long nDataLen, 
							  IN int nError, IN void *pContext);
///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
#pragma comment(lib, "Transport.lib")

CClientTunnel::CClientTunnel()
{
	m_hSocket = INVALID_SOCKET;
}

CClientTunnel::~CClientTunnel()
{
}

int CClientTunnel::net_OpenSocket(IN ETransportType eType, 									 
								  IN unsigned short usPort, 
								  IN LPTRANSPORT_EVENT eventHandler, 									 
								  void *pContext)
{
	//LOG4C((LOG_NOTICE,"CClientTunnel::net_OpenSocket"));
	TransportStatus status = TRANSPORT_OK;

	unsigned long ulHeaderLen = MESSAGE_HEADER_LEN;

	m_eventHandler = eventHandler;
	m_pContext = pContext;
	eventHandler = netHandleNetEvent;
	pContext = this;

	LPTRANSPORT_GETDATABUFLEN cbHandler = netGetDataBufLenCb;	
	SOCKET hSocket = CTransportImpl::net_OpenSocket(
		eType, 
		usPort, 
		eventHandler, 
		cbHandler, 
		ulHeaderLen, 
		pContext);

	if( INVALID_SOCKET == hSocket )
	{
		return TRANSPORT_ERROR;
	}

	m_hSocket = hSocket;

	return status;
}

void CClientTunnel::net_CloseSocket()
{
	CTransportImpl::net_CloseSocket();
}

int CClientTunnel::net_Connect(IN unsigned long ulIPValue, IN unsigned short usPort)
{
	int nResult = TRANSPORT_OK;

	nResult = CTransportImpl::net_Connect(ulIPValue, usPort);

	return nResult;
}

unsigned long CClientTunnel::net_Send(IN void *pHeader, 
									  IN void *pMessage, 
									  IN unsigned long ulDataLen)
{
	unsigned long ulSendLen = 0;
	unsigned long ulBufLen = 0;

	TMessageHeader *pSendHeader = (TMessageHeader *)pHeader;

	ulBufLen = MESSAGE_HEADER_LEN + ulDataLen;	


	//	CString ss;
	//	ss.Format("send leng:%d", ulDataLen);
	//	AfxMessageBox(ss);


	char *pSendBuf = new char[ulBufLen];
	memset(pSendBuf, 0, ulBufLen);

	TMessageHeader *pMessageHeader = (TMessageHeader *)pSendBuf;
	pMessageHeader->byVersion = 101;
	pMessageHeader->wHeaderFlag = MESSAGE_HEADER_FLAG;
	pMessageHeader->wMessageId = pSendHeader->wMessageId;
	pMessageHeader->wMessageSubId = pSendHeader->wMessageSubId;
	pMessageHeader->dwDataLen = ulDataLen;
	pMessageHeader->wReserve = 0;

	memcpy((void *)(pSendBuf+MESSAGE_HEADER_LEN), pMessage, ulDataLen);

	// convert network word 
	htons(pMessageHeader->wHeaderFlag);
	htons(pMessageHeader->wMessageId);
	htons(pMessageHeader->wMessageSubId);
	htonl(pMessageHeader->dwDataLen);
	htons(pMessageHeader->wCheckSum);	
	htonl(pMessageHeader->wReserve);

	//Send the message
	SOCKET hSocket = m_hSocket;
	ulSendLen = CTransportImpl::net_Send(hSocket, (void *)pSendBuf, ulBufLen);

	delete [] pSendBuf;
	pSendBuf = NULL;
	return ulSendLen;
}


void netGetDataBufLenCb(IN SOCKET hSocket, IN void *pHeaderBuf, 
						OUT unsigned long *nDataLen)
{
	TMessageHeader *pHeader = NULL;

	*nDataLen = 0;

	pHeader = (TMessageHeader *)pHeaderBuf;
	if( NULL == pHeader )
	{
		return;
	}

	if( MESSAGE_HEADER_FLAG != pHeader->wHeaderFlag )
	{
		return;
	}

	unsigned long dwDataBufLen = pHeader->dwDataLen;
	ntohl(dwDataBufLen);
	*nDataLen = dwDataBufLen;

	return;
}

void netHandleNetEvent(IN SOCKET hSocket, IN ETransportEvent eEvent, 
					   IN void *pDataBuf, IN unsigned long nDataLen, 
					   IN int nError, IN void *pContext)
{
	//Notify application layer
	CClientTunnel *pClientTunnel = (CClientTunnel *)pContext;
	if( NULL == pClientTunnel )
	{
		return;
	}
	void *pUserData = pClientTunnel->m_pContext;

	TransportBool bHandle = TRUE;

	if( Transport_ReadEv == eEvent )
	{
		if( nDataLen <= 0 )
		{
			return;
		}

		//提取消息头，获得消息ID
		char *pMessageBuf = (char *)pDataBuf;		
		TMessageHeader *pHeader = (TMessageHeader *)pMessageBuf;		

		//网络字节顺序的转换
		ntohs(pHeader->wHeaderFlag);
		ntohs(pHeader->wMessageId);
		ntohs(pHeader->wMessageSubId);
		ntohl(pHeader->dwDataLen);
		ntohs(pHeader->wCheckSum);
		ntohl(pHeader->wReserve);
		bHandle = (MESSAGE_HEADER_FLAG == pHeader->wHeaderFlag);
	}

	if( bHandle )
	{
		if( NULL != pClientTunnel->m_eventHandler )
		{
			pClientTunnel->m_eventHandler(hSocket, eEvent, pDataBuf, nDataLen, nError, pUserData);
		}
	}
}

#else

CClientTunnel::CClientTunnel()
{
}

CClientTunnel::~CClientTunnel()
{
}

TransportStatus CClientTunnel::Transport_Init()
{
	return CSocketHandle::InitLibrary(MAKEWORD(2,2));
}

TransportStatus CClientTunnel::Transport_UnInit()
{
	return CSocketHandle::ReleaseLibrary();
}

int CClientTunnel::net_OpenSocket(IN CONST ETransportType& eType, IN CONST unsigned short& usPort, IN LPTRANSPORT_EVENT eventHandler, void *pContext)
{
	return m_Client.net_OpenSocket(eType, usPort, eventHandler, pContext);
}

void CClientTunnel::net_CloseSocket()
{
	m_Client.net_CloseSocket();
}


/************************************************************************/
/*
函数: net_Connect
描述: 连接服务
参数:
	IN unsigned long ulIPValue,			ip地址
	IN unsigned short usPort			端口
返回:-1失败
*/
/************************************************************************/
int CClientTunnel::net_Connect(IN LPCTSTR strAddr, IN LPCTSTR strPort)
{
	return m_Client.net_Connect(strAddr,strPort);
}

int CClientTunnel::net_Connect(IN LPCTSTR strAddr, IN CONST DWORD& dwPort)
{
	TCHAR szPort[10] = {0};
	_ltot_s(dwPort, szPort, 10);

	return m_Client.net_Connect(strAddr,szPort);
}

unsigned long CClientTunnel::net_Send(IN void *pHeader, IN void *pMessage, IN CONST unsigned long& ulDataLen)
{
	return m_Client.net_Send(pHeader, pMessage, ulDataLen);
}

#endif //#ifdef VC60