zhujimajiang_redis/俱乐部/Source/ServerControl/KernelEngine/TCPNetworkEngine.cpp

#include "StdAfx.h"
#include "TraceService.h" //
#include "TCPNetworkEngine.h"
#include "../../MessageDef/Pb_Socket.pb.h"
//////////////////////////////////////////////////////////////////////////
//宏定义

#define TIME_BREAK_READY			9000L								//中断时间
#define TIME_BREAK_CONNECT			4000L								//中断时间
#define TIME_DETECT_SOCKET			20000L								//监测时间

//////////////////////////////////////////////////////////////////////////

//动作定义
#define QUEUE_SEND_REQUEST			1									//发送标识
#define QUEUE_SAFE_CLOSE			2									//安全关闭
#define QUEUE_ALLOW_BATCH			3									//允许群发
#define QUEUE_DETECT_SOCKET			4									//检测连接

//#define MY_ENCRYPT

//发送请求结构
struct tagSendDataRequest
{
    WORD							wMainCmdID;							//主命令码
    WORD							wSubCmdID;							//子命令码
    DWORD							dwSocketID;							//连接索引
    WORD							wDataSize;							//数据大小
    BYTE							cbSendBuf[SOCKET_TCP_BUFFER];			//发送缓冲
};

//设置群发
struct tagAllowBatchSend
{
    DWORD							dwSocketID;							//连接索引
    BYTE							cbAllow;							//允许标志
    BYTE							cbMask;
};

//安全关闭
struct tagSafeCloseSocket
{
    DWORD							dwSocketID;							//连接索引
};

//////////////////////////////////////////////////////////////////////////

//构造函数
COverLapped::COverLapped(enOperationType OperationType) : m_OperationType(OperationType)
{
    memset(&m_WSABuffer, 0, sizeof(m_WSABuffer));
    memset(&m_OverLapped, 0, sizeof(m_OverLapped));
}

//析构函数
COverLapped::~COverLapped()
{
}

//////////////////////////////////////////////////////////////////////////

//构造函数
COverLappedSend::COverLappedSend() : COverLapped(OperationType_Send)
{
    m_WSABuffer.len = 0;
    m_WSABuffer.buf = (char *)m_cbBuffer;
}

//析构函数
COverLappedSend::~COverLappedSend()
{
}

//////////////////////////////////////////////////////////////////////////

//构造函数
CServerSocketItem::CServerSocketItem(WORD wIndex, IServerSocketItemSink * pIServerSocketItemSink)
: m_wIndex(wIndex), m_pIServerSocketItemSink(pIServerSocketItemSink)
{
    m_wRountID = 1;
    m_wRecvSize = 0;
    m_cbSendRound = 0;
    m_cbRecvRound = 0;
    m_bNotify = true;
    m_bRecvIng = false;
    m_bCloseIng = false;
    m_bAllowBatch = true;
	m_bForeground = true;//websocket新增，……………………
    m_cbSendMask = 0xff;
    m_dwSendXorKey = 0;
    m_dwRecvXorKey = 0;
    m_dwClientAddr = 0;
    m_dwConnectTime = 0;
    m_dwSendTickCount = 0;
    m_dwRecvTickCount = 0;
    m_dwSendPacketCount = 0;
    m_dwRecvPacketCount = 0;
	m_dwPort = 0;
    m_hSocket = INVALID_SOCKET;
	ZeroMemory(m_szIp, sizeof(TCHAR) * 72);
	m_connectType = CHECKING;//websocket新增，……………………
}

//析够函数
CServerSocketItem::~CServerSocketItem(void)
{
    //删除空闲重叠 IO
    INT_PTR iCount = m_OverLappedSendFree.GetCount();
    for (INT_PTR i = 0; i < iCount; i++) delete m_OverLappedSendFree[i];
    m_OverLappedSendFree.RemoveAll();

    //删除活动重叠 IO
    iCount = m_OverLappedSendActive.GetCount();
    for (INT_PTR i = 0; i < iCount; i++) delete m_OverLappedSendActive[i];
    m_OverLappedSendActive.RemoveAll();

    return;
}

//随机映射
WORD CServerSocketItem::SeedRandMap(WORD wSeed)
{
    DWORD dwHold = wSeed;
    return (WORD)((dwHold = dwHold * 241103L + 2533101L) >> 16);
}

//映射发送数据
BYTE CServerSocketItem::MapSendByte(BYTE const cbData)
{
    BYTE cbMap;
    cbMap = g_SendByteMap[cbData];
    return cbMap;
}

//映射发送数据
BYTE CServerSocketItem::MapSendByteEx(BYTE const cbData)
{
	BYTE cbMap;
	cbMap = g_SendByteMapNew[cbData];
	return cbMap;
}
//映射接收数据
BYTE CServerSocketItem::MapRecvByte(BYTE const cbData)
{
    BYTE cbMap;
    cbMap = g_RecvByteMap[cbData];
    return cbMap;
}
//映射接收数据
BYTE CServerSocketItem::MapRecvByteEx(BYTE const cbData)
{
	BYTE cbMap;
	cbMap = g_RecvByteMapNew[cbData];
	return cbMap;
}

//获取发送结构
COverLappedSend * CServerSocketItem::GetSendOverLapped()
{
    //寻找空闲结构
    COverLappedSend * pOverLappedSend = NULL;
    INT_PTR nFreeCount = m_OverLappedSendFree.GetCount();
    if (nFreeCount > 0)
    {
        pOverLappedSend = m_OverLappedSendFree[nFreeCount - 1];
        ASSERT(pOverLappedSend != NULL);
        m_OverLappedSendFree.RemoveAt(nFreeCount - 1);
        m_OverLappedSendActive.Add(pOverLappedSend);
        return pOverLappedSend;
    }

    //新建发送结构
    try
    {
        pOverLappedSend = new COverLappedSend;
        ASSERT(pOverLappedSend != NULL);
        m_OverLappedSendActive.Add(pOverLappedSend);
        return pOverLappedSend;
    }
    catch (...) {}
    return NULL;
}

//绑定对象
DWORD CServerSocketItem::Attach(SOCKET hSocket, DWORD dwClientAddr, DWORD dwPort)
{
    //效验数据
    ASSERT(dwClientAddr != 0);
    ASSERT(m_bRecvIng == false);
    ASSERT(IsValidSocket() == false);
    ASSERT(hSocket != INVALID_SOCKET);

    //设置变量
    m_bNotify = false;
    m_bRecvIng = false;
    m_bCloseIng = false;
	m_bForeground = true;//websocket新增，……………………
    m_hSocket = hSocket;
    m_dwClientAddr = dwClientAddr;
    m_dwRecvTickCount = GetTickCount();
    m_dwConnectTime = (DWORD)time(NULL);
	//m_dwPort = dwPort;

	//CString strIP = _T("");
	//WORD add1, add2, add3, add4;

	//add1 = (WORD)(dwClientAddr & 255);
	//add2 = (WORD)((dwClientAddr >> 8) & 255);
	//add3 = (WORD)((dwClientAddr >> 16) & 255);
	//add4 = (WORD)((dwClientAddr >> 24) & 255);
	//strIP.Format(_T("%d.%d.%d.%d"), add1, add2, add3, add4);

	//CString csIpPort;
	//csIpPort.Format(_T("%s:%d"), strIP, htons(dwPort));
	//StringCchCopy(m_szIp, sizeof(TCHAR) * 72, csIpPort.GetBuffer());
    //发送通知
    m_pIServerSocketItemSink->OnSocketAcceptEvent(this);

    return GetSocketID();
}

//发送函数
bool CServerSocketItem::SendData(WORD wMainCmdID, WORD wSubCmdID, WORD wRountID)
{
    //效验状态
    if (m_bCloseIng == true) return false;
    if (m_wRountID != wRountID) return false;
    if (m_dwRecvPacketCount == 0) return false;
    if (IsValidSocket() == false) return false;

    //寻找发送结构
    COverLappedSend * pOverLappedSend = GetSendOverLapped();
    ASSERT(pOverLappedSend != NULL);
    if (pOverLappedSend == NULL) return false;

    //构造数据
    TCP_Head * pHead = (TCP_Head *)pOverLappedSend->m_cbBuffer;
    pHead->CommandInfo.wMainCmdID = wMainCmdID;
    pHead->CommandInfo.wSubCmdID = wSubCmdID;
	WORD wSendSize = EncryptBuffer(pOverLappedSend->m_cbBuffer, sizeof(TCP_Head), sizeof(pOverLappedSend->m_cbBuffer), pHead->TCPInfo.cbDataKind);
    pOverLappedSend->m_WSABuffer.len = wSendSize;

    //发送数据
    if (m_OverLappedSendActive.GetCount() == 1)
    {
        DWORD dwThancferred = 0;
        int iRetCode = WSASend(m_hSocket, &pOverLappedSend->m_WSABuffer, 1, &dwThancferred, 0, &pOverLappedSend->m_OverLapped, NULL);
        if ((iRetCode == SOCKET_ERROR) && (WSAGetLastError() != WSA_IO_PENDING))
        {
            OnSendCompleted(pOverLappedSend, 0L);
            return false;
        }
    }

    return true;
}

//发送函数
bool CServerSocketItem::SendData(void * pData, WORD wDataSize, WORD wMainCmdID, WORD wSubCmdID, WORD wRountID)
{
    //效验参数
    ASSERT(wDataSize <= SOCKET_TCP_BUFFER);

    //效验状态
    if (m_bCloseIng == true) return false;
    if (m_wRountID != wRountID) return false;
    if (m_dwRecvPacketCount == 0) return false;
    if (IsValidSocket() == false) return false;
    if (wDataSize > SOCKET_TCP_BUFFER) return false;

    //寻找发送结构
    COverLappedSend * pOverLappedSend = GetSendOverLapped();
    ASSERT(pOverLappedSend != NULL);
    if (pOverLappedSend == NULL) return false;

    //构造数据
    TCP_Head * pHead = (TCP_Head *)pOverLappedSend->m_cbBuffer;
    pHead->CommandInfo.wMainCmdID = wMainCmdID;
    pHead->CommandInfo.wSubCmdID = wSubCmdID;
    if (wDataSize > 0)
    {
        ASSERT(pData != NULL);
        memcpy(pHead + 1, pData, wDataSize);
    }
	WORD wSendSize = EncryptBuffer(pOverLappedSend->m_cbBuffer, sizeof(TCP_Head) + wDataSize, sizeof(pOverLappedSend->m_cbBuffer), pHead->TCPInfo.cbDataKind);
    pOverLappedSend->m_WSABuffer.len = wSendSize;

    //发送数据
    if (m_OverLappedSendActive.GetCount() == 1)
    {
        DWORD dwThancferred = 0;
        int iRetCode = WSASend(m_hSocket, &pOverLappedSend->m_WSABuffer, 1, &dwThancferred, 0, &pOverLappedSend->m_OverLapped, NULL);
        if ((iRetCode == SOCKET_ERROR) && (WSAGetLastError() != WSA_IO_PENDING))
        {
            OnSendCompleted(pOverLappedSend, 0L);
            return false;
        }
    }

    return true;
}

//投递接收
bool CServerSocketItem::RecvData()
{
    //效验变量
    ASSERT(m_bRecvIng == false);
    ASSERT(m_hSocket != INVALID_SOCKET);

    //判断关闭
    if (m_bCloseIng == true)
    {
        if (m_OverLappedSendActive.GetCount() == 0) CloseSocket(m_wRountID);
        return false;
    }

    //接收数据
    m_bRecvIng = true;
    DWORD dwThancferred = 0, dwFlags = 0;
    int iRetCode = WSARecv(m_hSocket, &m_OverLappedRecv.m_WSABuffer, 1, &dwThancferred, &dwFlags, &m_OverLappedRecv.m_OverLapped, NULL);
    if ((iRetCode == SOCKET_ERROR) && (WSAGetLastError() != WSA_IO_PENDING))
    {
        m_bRecvIng = false;
        CloseSocket(m_wRountID);
        return false;
    }

    return true;
}

//关闭连接
bool CServerSocketItem::CloseSocket(WORD wRountID)
{
    //状态判断
    if (m_wRountID != wRountID) return false;

    //关闭连接
    if (m_hSocket != INVALID_SOCKET)
    {
        closesocket(m_hSocket);
        m_hSocket = INVALID_SOCKET;
    }

    //判断关闭
    if ((m_bRecvIng == false) && (m_OverLappedSendActive.GetCount() == 0)) OnCloseCompleted();

    return true;
}

//设置关闭
bool CServerSocketItem::ShutDownSocket(WORD wRountID)
{
    return true;

    //状态判断
    if (m_wRountID != wRountID) return false;
    if (m_hSocket == INVALID_SOCKET) return false;

    //设置变量
    if (m_bCloseIng == false)
    {
        m_bCloseIng = true;
        //if (m_OverLappedSendActive.GetCount() == 0) CloseSocket(wRountID);
    }

    return true;
}

//允许群发
bool CServerSocketItem::AllowBatchSend(WORD wRountID, bool bAllowBatch, BYTE cbBatchMask)
{
    //状态判断
    if (m_wRountID != wRountID) return false;
    if (m_hSocket == INVALID_SOCKET) return false;

    //设置变量
    m_bAllowBatch = bAllowBatch;
    m_cbSendMask = cbBatchMask;

    return true;
}

//重置变量
void CServerSocketItem::ResetSocketData()
{
    //效验状态
    ASSERT(m_hSocket == INVALID_SOCKET);

    //重置数据
    m_wRountID++;
    m_wRecvSize = 0;
    m_cbSendRound = 0;
    m_cbRecvRound = 0;
    m_dwSendXorKey = 0;
    m_dwRecvXorKey = 0;
    m_dwClientAddr = 0;
    m_dwConnectTime = 0;
    m_dwSendTickCount = 0;
    m_dwRecvTickCount = 0;
    m_dwSendPacketCount = 0;
    m_dwRecvPacketCount = 0;
    m_MyEncrypt.Reset();

    //状态变量
    m_bNotify = true;
    m_bRecvIng = false;
    m_bCloseIng = false;
    m_bAllowBatch = true;
	m_bForeground = true;//websocket新增，……………………
    m_cbSendMask = 0xff;
    m_OverLappedSendFree.Append(m_OverLappedSendActive);
    m_OverLappedSendActive.RemoveAll();
	m_connectType = CHECKING;//websocket新增，……………………
    return;
}

//发送完成函数
bool CServerSocketItem::OnSendCompleted(COverLappedSend * pOverLappedSend, DWORD dwThancferred)
{
    //效验变量
    ASSERT(pOverLappedSend != NULL);
    ASSERT(m_OverLappedSendActive.GetCount() > 0);
    ASSERT(pOverLappedSend == m_OverLappedSendActive[0]);

    //释放发送结构
    m_OverLappedSendActive.RemoveAt(0);
    m_OverLappedSendFree.Add(pOverLappedSend);

    //设置变量
    if (dwThancferred != 0) m_dwSendTickCount = GetTickCount();

    //判断关闭
    if (m_hSocket == INVALID_SOCKET)
    {
        m_OverLappedSendFree.Append(m_OverLappedSendActive);
        m_OverLappedSendActive.RemoveAll();
        CloseSocket(m_wRountID);
        return true;
    }

    //继续发送数据
    if (m_OverLappedSendActive.GetCount() > 0)
    {
        DWORD dwThancferred = 0;
        pOverLappedSend = m_OverLappedSendActive[0];
        ASSERT(pOverLappedSend != NULL);
        int iRetCode = WSASend(m_hSocket, &pOverLappedSend->m_WSABuffer, 1, &dwThancferred, 0, &pOverLappedSend->m_OverLapped, NULL);
        if ((iRetCode == SOCKET_ERROR) && (WSAGetLastError() != WSA_IO_PENDING))
        {
            m_OverLappedSendFree.Append(m_OverLappedSendActive);
            m_OverLappedSendActive.RemoveAll();
            CloseSocket(m_wRountID);
            return false;
        }
        return true;
    }

    //判断关闭
    if (m_bCloseIng == true)
        CloseSocket(m_wRountID);

    return true;
}

//接收完成函数
bool CServerSocketItem::OnRecvCompleted(COverLappedRecv * pOverLappedRecv, DWORD dwThancferred)
{
    //效验数据
    ASSERT(m_bRecvIng == true);

    //设置变量
    m_bRecvIng = false;
    m_dwRecvTickCount = GetTickCount();

    //判断关闭
    if (m_hSocket == INVALID_SOCKET)
    {
        CloseSocket(m_wRountID);
        return true;
    }

    //接收数据
    int iRetCode = recv(m_hSocket, (char *)m_cbRecvBuf + m_wRecvSize, sizeof(m_cbRecvBuf)-m_wRecvSize, 0);
    if (iRetCode <= 0)
    {
        CloseSocket(m_wRountID);
        return true;
    }

    //接收完成
    m_wRecvSize += iRetCode;
	/*CString strTip1;
	strTip1.Format(L"m_connectType值：%d", m_connectType);
	CTraceService::TraceString(strTip1, TraceLevel_Normal);*/
	//web协议建设 websocket新增
	if (m_connectType == CHECKING) {
		if (CheckIfWinSocket() == true) {
			m_connectType = WINSOCKET;
			return RecvData();
		}
		else {
			m_connectType = HANDSHAKING;
		}
	}
	if (m_connectType == HANDSHAKING) {
		//try to handshake for websocket
		const char kDoubleCRLF[] = "\r\n\r\n";
		char* recvChar = (char*)m_cbRecvBuf;
		const char* pos = std::search(recvChar, recvChar + m_wRecvSize, kDoubleCRLF, kDoubleCRLF + 4);
		if (pos == recvChar + m_wRecvSize) {
			std::string data((char*)m_cbRecvBuf, m_wRecvSize);
			return RecvData();
		}
		zl::net::ByteBuffer handshakeBuffer;
		handshakeBuffer.write(recvChar, m_wRecvSize);
		bool success = handshake(&handshakeBuffer);
		if (success == true) {
			m_connectType = WEBSOCKET;
			m_wRecvSize = 0;
			return RecvData();
		}
		else {
			CloseSocket(m_wRountID);
			return false;
		}
	}

	if (m_connectType == WEBSOCKET)
	{
		if (HandleWebsocketRecv() == false) 
		{
			CloseSocket(m_wRountID);
			return false;
		}
		return RecvData();
	}
    BYTE cbBuffer[SOCKET_TCP_BUFFER];
    TCP_Head * pHead = (TCP_Head *)m_cbRecvBuf;
	/*CString strTip;
	strTip.Format(L"m_connectType值：%d pHead->TCPInfo.cbDataKind协议值：%d ,cbCheckCode值：%d，wPacketSize值：%d", m_connectType, pHead->TCPInfo.cbDataKind, pHead->TCPInfo.cbCheckCode, pHead->TCPInfo.wPacketSize);
	CTraceService::TraceString(strTip, TraceLevel_Normal);*/
    //处理数据
    try
    {
        while (m_wRecvSize >= sizeof(TCP_Head))
        {
            //效验数据
            WORD wPacketSize = pHead->TCPInfo.wPacketSize;
            if (wPacketSize > SOCKET_TCP_BUFFER) throw TEXT("数据包超长");
            if (wPacketSize < sizeof(TCP_Head)) throw TEXT("数据包非法");
            if (m_wRecvSize < wPacketSize) break;

            //提取数据
            CopyMemory(cbBuffer, m_cbRecvBuf, wPacketSize);
            WORD wRealySize = CrevasseBuffer(cbBuffer, wPacketSize);
            ASSERT(wRealySize >= sizeof(TCP_Head));
            m_dwRecvPacketCount++;

            //解释数据
            WORD wDataSize = wRealySize - sizeof(TCP_Head);
            void * pDataBuffer = cbBuffer + sizeof(TCP_Head);
            TCP_Command Command = ((TCP_Head *)cbBuffer)->CommandInfo;

            //内核命令
            if (Command.wMainCmdID == MDM_KN_COMMAND)
            {
                switch (Command.wSubCmdID)
                {
					case SUB_KN_DETECT_SOCKET:	//网络检测
					{
					   break;
					}
                    //default:
                    //{
                    //throw TEXT("非法命令码");
                    //}
                }
            }
            else
            {
                //消息处理
                m_pIServerSocketItemSink->OnSocketReadEvent(Command, pDataBuffer, wDataSize, this);
            }

            //删除缓存数据
            m_wRecvSize -= wPacketSize;
            MoveMemory(m_cbRecvBuf, m_cbRecvBuf + wPacketSize, m_wRecvSize);
        }
    }
    catch (...)
    {
        CloseSocket(m_wRountID);
        return false;
    }

    return RecvData();
}

//--websocket新增
bool CServerSocketItem::CheckIfWinSocket() {
	BYTE cbBuffer[SOCKET_TCP_BUFFER];
	TCP_Head * pHead = (TCP_Head *)m_cbRecvBuf;

	//处理数据
	try {
		while (m_wRecvSize >= sizeof(TCP_Head)) {
			//效验数据
			WORD wPacketSize = pHead->TCPInfo.wPacketSize;

			if (wPacketSize > SOCKET_TCP_BUFFER) {
				throw TEXT("数据包超长");
			}

			if (wPacketSize < sizeof(TCP_Head)) {
				throw TEXT("数据包非法");
			}

			//if (pHead->TCPInfo.cbVersion != SOCKET_TCP_VER) throw TEXT("数据包版本错误");
			/*if (pHead->TCPInfo.cbDataKind != SOCKET_TCP_VER) {
				throw TEXT("数据包版本错误");
			}*/

			if (m_wRecvSize < wPacketSize) {
				break;
			}

			//提取数据
			CopyMemory(cbBuffer, m_cbRecvBuf, wPacketSize);
			WORD wRealySize = CrevasseBuffer(cbBuffer, wPacketSize);
			ASSERT(wRealySize >= sizeof(TCP_Head));
			m_dwRecvPacketCount++;
			//解释数据
			WORD wDataSize = wRealySize - sizeof(TCP_Head);
			void * pDataBuffer = cbBuffer + sizeof(TCP_Head);
			TCP_Command Command = ((TCP_Head *)cbBuffer)->CommandInfo;

			//内核命令
			if (Command.wMainCmdID == MDM_KN_COMMAND) {
				switch (Command.wSubCmdID) {
				case SUB_KN_DETECT_SOCKET: {	//网络检测
					break;
				}

				default: {
					throw TEXT("非法命令码");
				}
				}
			}

			else {
				//消息处理
				m_pIServerSocketItemSink->OnSocketReadEvent(Command, pDataBuffer, wDataSize, this);
			}

			//删除缓存数据
			m_wRecvSize -= wPacketSize;
			MoveMemory(m_cbRecvBuf, m_cbRecvBuf + wPacketSize, m_wRecvSize);
		}
	}

	catch (...) {
		return false;
	}

	return true;
}
bool CServerSocketItem::handshake(zl::net::ByteBuffer* byteBuffer) 
{
	zl::net::HttpContext context;

	if (!context.parseRequest(byteBuffer) || context.request().method() != zl::net::HttpGet) { // 解析失败 或者 不是Get请求
		//LOG_WARN(_T("parse handshake error,send close header"));
		std::string msg = "HTTP/1.1 400 Bad Request\r\n\r\n";
		SendRawData(msg.c_str(), msg.size());
		return false;
	}

	assert(context.gotAll());
	zl::net::HttpRequest& req = context.request();
	std::string query = req.query();
	if (query.empty() == false) {
		query = query.substr(1);
	}

	std::string key = req.getHeader(zl::net::ws::kSecWebSocketKeyHeader);
	std::string ip = req.getHeader("X-Forwarded-For");
	if (ip.empty() == false) {
		//LOG_PRINT("real ip",ip.c_str());
		m_dwClientAddr = inet_addr(ip.c_str());
		//Attach(m_hSocket, inet_addr(ip.c_str()));
		m_pIServerSocketItemSink->OnSocketAcceptEvent(this);
	}
	std::string answer = zl::net::ws::makeHandshakeResponse(key.c_str(), req.getHeader(zl::net::ws::kSecWebSocketProtocolHeader));
	//printf ("response handshake:\n %s\n", answer.c_str());
	SendRawData(answer.c_str(), answer.size());
	return true;
}
bool CServerSocketItem::SendRawData(const char* data, int len) {
	//寻找发送结构
	COverLappedSend * pOverLappedSend = GetSendOverLapped();
	ASSERT(pOverLappedSend != NULL);

	if (pOverLappedSend == NULL) {
		return false;
	}
	int n = 0;
	for (int i = 0; i < len; i++) {
		pOverLappedSend->m_cbBuffer[i] = data[i];
		n++;
	}
	pOverLappedSend->m_cbBuffer[n] = 0;
	pOverLappedSend->m_WSABuffer.len = len;

	//发送数据
	if (m_OverLappedSendActive.GetCount() == 1) {
		DWORD dwThancferred = 0;
		int iRetCode = WSASend(m_hSocket, &pOverLappedSend->m_WSABuffer, 1, &dwThancferred, 0, &pOverLappedSend->m_OverLapped, NULL);

		if ((iRetCode == SOCKET_ERROR) && (WSAGetLastError() != WSA_IO_PENDING)) {
			OnSendCompleted(pOverLappedSend, 0L);
			return false;
		}
	}
	return true;
}
bool CServerSocketItem::HandleWebsocketRecv() {
	using namespace zl::net::ws;

	if (m_wRecvSize > SOCKET_TCP_BUFFER) {
		//CloseSocket(m_wRountID);
		//LOG_WARN(_T("error,receive data len > SOCKET_TCP_BUFFER,%d>32768"), (int)m_wRecvSize);
		return false;
	}

	try {
		while (m_wRecvSize > 0) {
			char outbuf[SOCKET_TCP_BUFFER];
			int outlen = 0;
			int frameSize = 0;
			WsFrameType type = decodeFrame((char*)m_cbRecvBuf, m_wRecvSize, outbuf, &outlen, &frameSize);

			switch (type) {
			case WS_INCOMPLETE_TEXT_FRAME:
			case WS_INCOMPLETE_BINARY_FRAME:
			case WS_INCOMPLETE_FRAME: {
				return true;
			}

			case WS_TEXT_FRAME:
			case WS_BINARY_FRAME: {
				m_dwRecvPacketCount++;
				CString strTip1;
				strTip1.Format(L"m_dwRecvPacketCount值：%d", m_dwRecvPacketCount);
				CTraceService::TraceString(strTip1, TraceLevel_Normal);
				return true;
				TCP_Command Command;
				//std::string data(outbuf, outlen);
				std::string outData(outbuf, outlen);
				//std::string data = CAES::Decrypt(outData, m_strAesKey, m_strAesKey);
				std::string data = outData;
				//Json::Reader reader;
				//Json::Value root;

				//if (reader.parse(data, root)) {
				//	Command.wMainCmdID = root["mainID"].asInt();
				//	Command.wSubCmdID = root["subID"].asInt();
				//}
				//else {
					//LOG_WARN(_T("receive's data parse json error"));
				//	throw TEXT("parse error");
				//}
				//玩家进后台的状态更新
				//m_dwKNRecvTickCount = GetTickCount();
				if (m_bForeground == false) {
					setForeground(true);
				}
				//内核命令
				if (Command.wMainCmdID == MDM_KN_COMMAND) {
					switch (Command.wSubCmdID) {
					case SUB_KN_DETECT_SOCKET: {	//网络检测
						SendData(MDM_KN_COMMAND, SUB_KN_DETECT_SOCKET, m_wRountID);
						break;
					}
					default: {
						throw TEXT("非法命令码");
					}
					}
				}
				else {
					//消息处理
					m_pIServerSocketItemSink->OnSocketReadEvent(Command, (void*)data.c_str(), data.size(), this);
				}

				break;
			}

			case WS_PING_FRAME: {
				//printf ("receive ping frame,framesize:%d\n", frameSize);
				char cbSendData[SOCKET_TCP_BUFFER];
				int encodesize = zl::net::ws::encodeFrame(zl::net::ws::WsFrameType::WS_PONG_FRAME, NULL, 0, cbSendData, SOCKET_TCP_BUFFER);
				SendRawData(cbSendData, encodesize);
				break;
			}

			case WS_PONG_FRAME: {
				// printf ("receive pong frame,framesize:%d\n", frameSize);
				break;
			}

			case WS_CLOSE_FRAME: {
				//printf ("receive close frame\n");
				//CloseSocket (m_wRountID);
				return false;
			}

			default: {
				//LOG_WARN(_T("receive unknow frame,close socket,type:%d"), (int)type);
				//CloseSocket (m_wRountID);
				return false;
			}
			}

			//删除缓存数据
			m_wRecvSize -= frameSize;
			MoveMemory(m_cbRecvBuf, m_cbRecvBuf + frameSize, m_wRecvSize);
		}
	}
	catch (...) {
		//LOG_WARN(_T("parse json error or could not find mainID,subID,close socket"));
		//CloseSocket (m_wRountID);
		return false;
	}

	return true;
}
bool CServerSocketItem::isWebSocket() {
	return m_connectType == WEBSOCKET;
}
bool CServerSocketItem::isWinSocket() {
	return m_connectType == WINSOCKET;
}
bool CServerSocketItem::sendPingFrame() {
	char cbSendData[SOCKET_TCP_BUFFER];
	int encodesize = zl::net::ws::encodeFrame(zl::net::ws::WsFrameType::WS_PING_FRAME, NULL, 0, cbSendData, SOCKET_TCP_BUFFER);
	return SendRawData(cbSendData, encodesize);
}
bool CServerSocketItem::setForeground(bool foreground) {
	m_bForeground = foreground;
	m_pIServerSocketItemSink->OnSocketForegroundEvent(this);
	return true;
}
//应答消息 --websocket新增
bool CTCPNetworkEngine::OnSocketForegroundEvent(CServerSocketItem * pServerSocketItem) {
	//效验数据
	ASSERT(pServerSocketItem != NULL);

	if (NULL == pServerSocketItem) {
		return false;
	}

	//投递消息
	if (m_bService == false) {
		return false;
	}

	m_QueueServiceEvent.PostNetworkForegroundEvent(pServerSocketItem->GetSocketID(), pServerSocketItem->isForeground());
	return true;
}

//关闭完成通知
bool CServerSocketItem::OnCloseCompleted()
{
    //效验状态
    ASSERT(m_hSocket == INVALID_SOCKET);
    ASSERT(m_OverLappedSendActive.GetCount() == 0);

    //关闭事件
    ASSERT(m_bNotify == false);
    if (m_bNotify == false)
    {
        m_bNotify = true;
        m_pIServerSocketItemSink->OnSocketCloseEvent(this);
    }

    //状态变量
    ResetSocketData();

    return true;
}

//加密数据
WORD CServerSocketItem::EncryptBuffer(BYTE pcbDataBuffer[], WORD wDataSize, WORD wBufferSize, BYTE cbDataKind)
{
    WORD i = 0;
    //效验参数
    ASSERT(wDataSize >= sizeof(TCP_Head));
    ASSERT(wDataSize <= (sizeof(TCP_Head)+SOCKET_TCP_BUFFER));
    ASSERT(wBufferSize >= (wDataSize + 2 * sizeof(DWORD)));

    //填写信息头
    TCP_Head * pHead = (TCP_Head *)pcbDataBuffer;
    pHead->TCPInfo.wPacketSize = wDataSize;
	//pHead->TCPInfo.cbDataKind = DK_MAPPED;
	pHead->TCPInfo.cbDataKind = DK_MAPPED_NEW;
	
#ifdef MY_ENCRYPT
    BYTE checkCode = 0;

    for (WORD i = sizeof(TCP_Info); i < wDataSize; i++)
    {
        checkCode += pcbDataBuffer[i];
        pcbDataBuffer[i] = MapSendByteEx(pcbDataBuffer[i]);
    }
    pHead->TCPInfo.cbCheckCode = ~checkCode + 1;

    //设置变量
    m_dwSendPacketCount++;
#else
    m_MyEncrypt.EncryptBuffer(pcbDataBuffer + 4, wDataSize - 4);
#endif
    return wDataSize;
}

//解密数据
WORD CServerSocketItem::CrevasseBuffer(BYTE pcbDataBuffer[], WORD wDataSize)
{
    WORD i = 0;
    //效验参数
    ASSERT(wDataSize >= sizeof(TCP_Head));
    ASSERT(((TCP_Head *)pcbDataBuffer)->TCPInfo.wPacketSize == wDataSize);

#ifdef MY_ENCRYPT
    //效验码与字节映射
    TCP_Head * pHead = (TCP_Head *)pcbDataBuffer;
    for (i = sizeof(TCP_Info); i < wDataSize; i++)
    {
		if (pHead->TCPInfo.cbDataKind == DK_MAPPED_NEW)
		{
			pcbDataBuffer[i] = MapRecvByteEx(pcbDataBuffer[i]);
		}
		//else if (pHead->TCPInfo.cbDataKind == DK_MAPPED)
		//{
		//	pcbDataBuffer[i] = MapRecvByte(pcbDataBuffer[i]);
		//}
    }

#else
    m_MyEncrypt.DecryptBuffer(pcbDataBuffer + 4, wDataSize - 4);
#endif

    return wDataSize;
}

//////////////////////////////////////////////////////////////////////////

//构造函数
CServerSocketRSThread::CServerSocketRSThread(void)
{
    m_hCompletionPort = NULL;
}

//析构函数
CServerSocketRSThread::~CServerSocketRSThread(void)
{
}

//配置函数
bool CServerSocketRSThread::InitThread(HANDLE hCompletionPort)
{
    ASSERT(hCompletionPort != NULL);
    m_hCompletionPort = hCompletionPort;
    return true;
}

//运行函数
bool CServerSocketRSThread::OnEventThreadRun()
{
    //效验参数
    ASSERT(m_hCompletionPort != NULL);

    //变量定义
    DWORD dwThancferred = 0;
    OVERLAPPED * pOverLapped = NULL;
    COverLapped * pSocketLapped = NULL;
    CServerSocketItem * pServerSocketItem = NULL;

    //等待完成端口
    BOOL bSuccess = GetQueuedCompletionStatus(m_hCompletionPort, &dwThancferred, (PULONG_PTR)& pServerSocketItem, &pOverLapped, INFINITE);
    if ((bSuccess == FALSE) && (GetLastError() != ERROR_NETNAME_DELETED)) return false;
    if ((pServerSocketItem == NULL) && (pOverLapped == NULL)) return false;

    //处理操作
    ASSERT(pOverLapped != NULL);
    ASSERT(pServerSocketItem != NULL);
    pSocketLapped = CONTAINING_RECORD(pOverLapped, COverLapped, m_OverLapped);
    switch (pSocketLapped->GetOperationType())
    {
    case OperationType_Recv:	//SOCKET 数据读取
    {
                                    COverLappedRecv * pOverLappedRecv = (COverLappedRecv *)pSocketLapped;
                                    CWHDataLocker lock(pServerSocketItem->GetSignedLock());
                                    pServerSocketItem->OnRecvCompleted(pOverLappedRecv, dwThancferred);
                                    break;
    }
    case OperationType_Send:	//SOCKET 数据发送
    {
                                    COverLappedSend * pOverLappedSend = (COverLappedSend *)pSocketLapped;
                                    CWHDataLocker lock(pServerSocketItem->GetSignedLock());
                                    pServerSocketItem->OnSendCompleted(pOverLappedSend, dwThancferred);
                                    break;
    }
    }

    return true;
}

//////////////////////////////////////////////////////////////////////////

//构造函数
CSocketAcceptThread::CSocketAcceptThread(void)
{
    m_hCompletionPort = NULL;
    m_pTCPSocketManager = NULL;
    m_hListenSocket = INVALID_SOCKET;
}

//析构函数
CSocketAcceptThread::~CSocketAcceptThread(void)
{
}

//配置函数
bool CSocketAcceptThread::InitThread(HANDLE hCompletionPort, SOCKET hListenSocket, CTCPNetworkEngine * pTCPSocketManager)
{
    ASSERT(hCompletionPort != NULL);
    ASSERT(pTCPSocketManager != NULL);
    ASSERT(hListenSocket != INVALID_SOCKET);
    m_hListenSocket = hListenSocket;
    m_hCompletionPort = hCompletionPort;
    m_pTCPSocketManager = pTCPSocketManager;
    return true;
}

//运行函数
bool CSocketAcceptThread::OnEventThreadRun()
{
    //效验参数
    ASSERT(m_hCompletionPort != NULL);
    ASSERT(m_pTCPSocketManager != NULL);

    //设置变量
    SOCKADDR_IN	SocketAddr;
    CServerSocketItem * pServerSocketItem = NULL;
    SOCKET hConnectSocket = INVALID_SOCKET;
    int nBufferSize = sizeof(SocketAddr);

    try
    {
        //监听连接
        hConnectSocket = WSAAccept(m_hListenSocket, (SOCKADDR *)& SocketAddr, &nBufferSize, NULL, NULL);
        if (hConnectSocket == INVALID_SOCKET) return false;

		//CTraceService::TraceString(TEXT("新的房间连接"), TraceLevel_Normal);

        //获取连接
        pServerSocketItem = m_pTCPSocketManager->ActiveSocketItem();
		if (pServerSocketItem == NULL)
		{
			CTraceService::TraceString(TEXT("申请连接对象失败"), TraceLevel_Exception);
			throw TEXT("申请连接对象失败");
		}

        //激活对象
        CWHDataLocker lock(pServerSocketItem->GetSignedLock());
		pServerSocketItem->Attach(hConnectSocket, SocketAddr.sin_addr.S_un.S_addr, SocketAddr.sin_port);
        CreateIoCompletionPort((HANDLE)hConnectSocket, m_hCompletionPort, (ULONG_PTR)pServerSocketItem, 0);
        pServerSocketItem->RecvData();
    }
    catch (...)
    {
		CTraceService::TraceString(TEXT("连接失败请检查原因！！"), TraceLevel_Exception);
        //清理对象
        ASSERT(pServerSocketItem == NULL);
        if (hConnectSocket != INVALID_SOCKET)	{
            //LINGER lingerStruct;

            //lingerStruct.l_onoff = 1;
            //lingerStruct.l_linger = 0;

            //setsockopt(hConnectSocket, SOL_SOCKET, SO_LINGER, (char *)&lingerStruct, sizeof(lingerStruct));
            closesocket(hConnectSocket);
        }
    }

    return true;
}

//////////////////////////////////////////////////////////////////////////

//构造函数
CSocketDetectThread::CSocketDetectThread(void)
{
    m_dwTickCount = 0;;
    m_pTCPSocketManager = NULL;
}

//析构函数
CSocketDetectThread::~CSocketDetectThread(void)
{
}

//配置函数
bool CSocketDetectThread::InitThread(CTCPNetworkEngine * pTCPSocketManager)
{
    //效验参数
    ASSERT(pTCPSocketManager != NULL);

    //设置变量
    m_dwTickCount = 0L;
    m_pTCPSocketManager = pTCPSocketManager;

    return true;
}

//运行函数
bool CSocketDetectThread::OnEventThreadRun()
{
    //效验参数
    ASSERT(m_pTCPSocketManager != NULL);

    //设置间隔
    Sleep(500);
    m_dwTickCount += 500L;

    //检测连接
    if (m_dwTickCount > 15000L)
    {
        m_dwTickCount = 0L;
        m_pTCPSocketManager->DetectSocket();
    }

    return true;
}

//////////////////////////////////////////////////////////////////////////

//构造函数
CTCPNetworkEngine::CTCPNetworkEngine(void)
{
    m_bService = false;
    m_wListenPort = 0;
    m_dwLastDetect = 0;
    m_wMaxSocketItem = 0;
    m_hCompletionPort = NULL;
    m_hServerSocket = INVALID_SOCKET;
	
    return;
}

//析构函数
CTCPNetworkEngine::~CTCPNetworkEngine(void)
{
    //停止服务
    ConcludeService();

    //释放存储连接
    CServerSocketItem * pSocketItem = NULL;
    for (INT_PTR i = 0; i < m_StorageSocketItem.GetCount(); i++)
    {
        pSocketItem = m_StorageSocketItem[i];
        ASSERT(pSocketItem != NULL);
        SafeDelete(pSocketItem);
    }
    m_StorageSocketItem.RemoveAll();

    return;
}

//接口查询
void * CTCPNetworkEngine::QueryInterface(const IID & Guid, DWORD dwQueryVer)
{
    QUERYINTERFACE(ITCPNetworkEngine, Guid, dwQueryVer);
    QUERYINTERFACE(IQueueServiceSink, Guid, dwQueryVer);
    QUERYINTERFACE_IUNKNOWNEX(ITCPNetworkEngine, Guid, dwQueryVer);
    return NULL;
}

//设置接口
bool  CTCPNetworkEngine::SetTCPNetworkEngineEvent(IUnknownEx * pIUnknownEx)
{
    //状态判断
    if (m_bService == true)
    {
        CTraceService::TraceString(TEXT("网络引擎处于服务状态，绑定操作忽略"), TraceLevel_Exception);
        return false;
    }

    //设置接口
    if (m_QueueServiceEvent.SetQueueServiceSink(pIUnknownEx) == false)
    {
        CTraceService::TraceString(TEXT("网络引擎与触发服务绑定失败"), TraceLevel_Exception);
        return false;
    }

    return true;
}
#include <Nb30.h>
struct tagAstatInfo
{
    ADAPTER_STATUS					AdapterStatus;						//网卡状态
    NAME_BUFFER						NameBuff[16];						//名字缓冲
};
//#define LEN_NETWORK_ID 33
//网卡地址
bool GetMACAddress(TCHAR szMACAddress[LEN_NETWORK_ID])
{
    //变量定义
    HINSTANCE hInstance = NULL;

    //执行逻辑
    __try
    {
        //加载 DLL
        hInstance = LoadLibrary(TEXT("NetApi32.dll"));
        if (hInstance == NULL) __leave;

        //获取函数
        typedef BYTE __stdcall NetBiosProc(NCB * Ncb);
        NetBiosProc * pNetBiosProc = (NetBiosProc *)GetProcAddress(hInstance, "Netbios");
        if (pNetBiosProc == NULL) __leave;

        //变量定义
        NCB Ncb;
        LANA_ENUM LanaEnum;
        ZeroMemory(&Ncb, sizeof(Ncb));
        ZeroMemory(&LanaEnum, sizeof(LanaEnum));

        //枚举网卡
        Ncb.ncb_command = NCBENUM;
        Ncb.ncb_length = sizeof(LanaEnum);
        Ncb.ncb_buffer = (BYTE *)&LanaEnum;
        if ((pNetBiosProc(&Ncb) != NRC_GOODRET) || (LanaEnum.length == 0)) __leave;

        //获取地址
        if (LanaEnum.length>0)
        {
            //变量定义
            tagAstatInfo Adapter;
            ZeroMemory(&Adapter, sizeof(Adapter));

            //重置网卡
            Ncb.ncb_command = NCBRESET;
            Ncb.ncb_lana_num = LanaEnum.lana[0];
            if (pNetBiosProc(&Ncb) != NRC_GOODRET) __leave;

            //获取状态
            Ncb.ncb_command = NCBASTAT;
            Ncb.ncb_length = sizeof(Adapter);
            Ncb.ncb_buffer = (BYTE *)&Adapter;
            Ncb.ncb_lana_num = LanaEnum.lana[0];
            strcpy((char *)Ncb.ncb_callname, "*");
            if (pNetBiosProc(&Ncb) != NRC_GOODRET) __leave;

            //获取地址
            for (INT i = 0; i<6; i++)
            {
                ASSERT((i * 2)<LEN_NETWORK_ID);
                _stprintf(&szMACAddress[i * 2], TEXT("%02X"), Adapter.AdapterStatus.adapter_address[i]);
            }
        }
    }

    //结束清理
    __finally
    {
        //释放资源
        if (hInstance != NULL)
        {
            FreeLibrary(hInstance);
            hInstance = NULL;
        }

        //错误断言
        if (AbnormalTermination() == TRUE)
        {
            ASSERT(FALSE);
        }
    }

    return true;
}
//设置参数
bool  CTCPNetworkEngine::SetServiceParameter(WORD wServicePort, WORD wMaxConnect, LPCTSTR  pszCompilation)
{
    //效验状态
    if (m_bService == true)
    {
        CTraceService::TraceString(TEXT("网络引擎处于服务状态，端口绑定操作忽略"), TraceLevel_Exception);
        return false;
    }

    //判断参数
    /*if (wServicePort == 0)
    {
    CTraceService::TraceString(TEXT("网络端口错误，端口绑定操作失败"), TraceLevel_Exception);
    return false;
    }*/

    //设置变量
    m_wListenPort = wServicePort;
    m_wMaxSocketItem = wMaxConnect;
    return true;
}

//启动服务
bool  CTCPNetworkEngine::StartService()
{
    DWORD i = 0;
    //效验状态
    if (m_bService == true)
    {
        CTraceService::TraceString(TEXT("网络引擎重复启动，启动操作忽略"), TraceLevel_Warning);
        return true;
    }

    //判断端口
    if (m_wListenPort == 0)
    {
		m_wListenPort = PORT_GAMECYHD;
    }

    //绑定对象
    if (m_SendQueueService.SetQueueServiceSink(QUERY_ME_INTERFACE(IUnknownEx)) == false)
    {
        CTraceService::TraceString(TEXT("网络引擎与触发服务绑定失败"), TraceLevel_Exception);
        return false;
    }

    try
    {
        //获取系统信息
        SYSTEM_INFO SystemInfo;
        GetSystemInfo(&SystemInfo);
        DWORD dwThreadCount = SystemInfo.dwNumberOfProcessors;

        //建立完成端口
        m_hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, SystemInfo.dwNumberOfProcessors);
        if (m_hCompletionPort == NULL) throw TEXT("网络引擎完成端口创建失败");

        //建立监听SOCKET
        struct sockaddr_in SocketAddr;
        memset(&SocketAddr, 0, sizeof(SocketAddr));
        SocketAddr.sin_addr.s_addr = INADDR_ANY;
        SocketAddr.sin_family = AF_INET;
        SocketAddr.sin_port = htons(m_wListenPort);
		m_hServerSocket = WSASocket(AF_INET, SOCK_STREAM, /*0*/IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);
        if (m_hServerSocket == INVALID_SOCKET) throw TEXT("网络引擎监听 SOCKET 创建失败");

        BOOL bMobile = TRUE;
        for (;;){
            if (0 != bind(m_hServerSocket, (SOCKADDR*)& SocketAddr, sizeof(SocketAddr)))
            {
				//++加点提示
				CString str;
				str.Format(L"绑定端口失败port=%d", m_wListenPort);
				CTraceService::TraceString(str, TraceLevel_Warning);
                closesocket(m_hServerSocket);
            }
            else
            {
				CString str;
				str.Format(L"绑定端口成功port=%d", m_wListenPort);
				CTraceService::TraceString(str, TraceLevel_Normal);
                break;
            }
            m_hServerSocket = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED);
            if (m_hServerSocket == INVALID_SOCKET) throw TEXT("网络引擎监听 SOCKET 创建失败");
            m_wListenPort++;
            SocketAddr.sin_family = AF_INET;
            SocketAddr.sin_addr.s_addr = 0;
            SocketAddr.sin_port = htons(m_wListenPort);

        }

        //int iErrorCode = bind(m_hServerSocket, (SOCKADDR*) & SocketAddr, sizeof(SocketAddr));

        //if (iErrorCode == SOCKET_ERROR) throw TEXT("网络引擎监听端口被占用，端口绑定失败");
        int iErrorCode = listen(m_hServerSocket, 200);
        if (iErrorCode == SOCKET_ERROR) throw TEXT("网络引擎监听端口被占用，端口监听失败");

        //启动发送队列
        bool bSuccess = m_SendQueueService.StartService();
        if (bSuccess == false) throw TEXT("网络引擎发送队列服务启动失败");

        //建立读写线程
        for (i = 0; i < dwThreadCount; i++)
        {
            CServerSocketRSThread * pServerSocketRSThread = new CServerSocketRSThread();
            if (pServerSocketRSThread == NULL) throw TEXT("网络引擎读写线程服务创建失败");
            bSuccess = pServerSocketRSThread->InitThread(m_hCompletionPort);
            if (bSuccess == false) throw TEXT("网络引擎读写线程服务配置失败");
            m_SocketRSThreadArray.Add(pServerSocketRSThread);
        }

        //建立应答线程
        bSuccess = m_SocketAcceptThread.InitThread(m_hCompletionPort, m_hServerSocket, this);
        if (bSuccess == false) throw TEXT("网络引擎网络监听线程服务配置");

        //运行读写线程
        for (i = 0; i < dwThreadCount; i++)
        {
            CServerSocketRSThread * pServerSocketRSThread = m_SocketRSThreadArray[i];
            ASSERT(pServerSocketRSThread != NULL);
            bSuccess = pServerSocketRSThread->StartThread();
            if (bSuccess == false) throw TEXT("网络引擎读写线程服务启动失败");
        }

        //网络检测线程
        m_SocketDetectThread.InitThread(this);
        bSuccess = m_SocketDetectThread.StartThread();
        if (bSuccess == false) throw TEXT("网络引检测线程服务启动失败");

        //运行应答线程
        bSuccess = m_SocketAcceptThread.StartThread();
        if (bSuccess == false) throw TEXT("网络引擎监听线程服务启动失败");

        //设置变量
        m_bService = true;
    }
    catch (LPCTSTR pszError)
    {
        CTraceService::TraceString(pszError, TraceLevel_Exception);
        return false;
    }

    return true;
}

//停止服务
bool  CTCPNetworkEngine::ConcludeService()
{
    //设置变量
    m_bService = false;
    m_dwLastDetect = 0L;

    //停止检测线程
    m_SocketDetectThread.ConcludeThread(INFINITE);

    //终止应答线程
    if (m_hServerSocket != INVALID_SOCKET)
    {
        //LINGER lingerStruct;

        //lingerStruct.l_onoff = 1;
        //lingerStruct.l_linger = 0;

        //setsockopt(m_hServerSocket, SOL_SOCKET, SO_LINGER, (char *)&lingerStruct, sizeof(lingerStruct));
        closesocket(m_hServerSocket);
        m_hServerSocket = INVALID_SOCKET;
    }
    m_SocketAcceptThread.ConcludeThread(INFINITE);

    //停止发送队列
    m_SendQueueService.ConcludeService();

    //释放读写线程
    INT_PTR nCount = m_SocketRSThreadArray.GetCount(), i = 0;
    if (m_hCompletionPort != NULL)
    {
        for (i = 0; i < nCount; i++) PostQueuedCompletionStatus(m_hCompletionPort, 0, NULL, NULL);
    }
    for (i = 0; i < nCount; i++)
    {
        CServerSocketRSThread * pSocketThread = m_SocketRSThreadArray[i];
        ASSERT(pSocketThread != NULL);
        pSocketThread->ConcludeThread(INFINITE);
        SafeDelete(pSocketThread);
    }
    m_SocketRSThreadArray.RemoveAll();

    //关闭完成端口
    if (m_hCompletionPort != NULL)
    {
        CloseHandle(m_hCompletionPort);
        m_hCompletionPort = NULL;
    }

    //关闭连接
    CServerSocketItem * pServerSocketItem = NULL;
    for (i = 0; i < m_ActiveSocketItem.GetCount(); i++)
    {
        pServerSocketItem = m_ActiveSocketItem[i];
        ASSERT(pServerSocketItem != NULL);
        pServerSocketItem->CloseSocket(pServerSocketItem->GetRountID());
        pServerSocketItem->ResetSocketData();
    }
    m_FreeSocketItem.Append(m_ActiveSocketItem);
    m_ActiveSocketItem.RemoveAll();

    m_QueueServiceEvent.SetQueueServiceSink(NULL);

    return true;
}

//应答消息
bool CTCPNetworkEngine::OnSocketAcceptEvent(CServerSocketItem * pServerSocketItem)
{
    //效验数据
    ASSERT(pServerSocketItem != NULL);
    if (NULL == pServerSocketItem) return false;

    //投递消息
    if (m_bService == false) return false;
    m_QueueServiceEvent.PostNetworkAcceptEvent(pServerSocketItem->GetSocketID(), pServerSocketItem->GetClientAddr());

    return true;
}

CString CTCPNetworkEngine::dwIP2csIP(DWORD dwIP)
{
    CString strIP = _T("");
    WORD add1, add2, add3, add4;

    add1 = (WORD)(dwIP & 255);
    add2 = (WORD)((dwIP >> 8) & 255);
    add3 = (WORD)((dwIP >> 16) & 255);
	add4 = (WORD)((dwIP >> 24) & 255);
	strIP.Format(_T("%d.%d.%d.%d"), add1, add2, add3, add4);

    return strIP;
}

//同步消息
bool CTCPNetworkEngine::OnTBDataEvent(int iCommend, DWORD dwContextID, void * pBuffer, WORD wDataSize)
{
	//投递消息
	m_QueueServiceEvent.PostChangeData(iCommend, dwContextID, pBuffer, wDataSize);
	return true;
}

//网络读取消息
bool CTCPNetworkEngine::OnSocketReadEvent(TCP_Command Command, void * pBuffer, WORD wDataSize, CServerSocketItem * pServerSocketItem)
{
    //效验数据
    ASSERT(pServerSocketItem != NULL);
    if (NULL == pServerSocketItem) return false;

    //效验状态
    if (m_bService == false) return false;

    m_QueueServiceEvent.PostNetworkReadEvent(pServerSocketItem->GetSocketID(), Command, pBuffer, wDataSize);

    return true;
}

//网络关闭消息
bool CTCPNetworkEngine::OnSocketCloseEvent(CServerSocketItem * pServerSocketItem)
{
    //效验参数
    ASSERT(pServerSocketItem != NULL);
    if (NULL == pServerSocketItem) return false;

    try
    {
        //效验状态
        if (m_bService == false) return false;

        //计算时间
        WORD wIndex = pServerSocketItem->GetIndex();
        WORD wRountID = pServerSocketItem->GetRountID();
        DWORD dwClientAddr = pServerSocketItem->GetClientAddr();
        DWORD dwConnectTime = pServerSocketItem->GetConnectTime();
        //////////////////////////////////////////////////////////////////////////,这里要调整
        m_QueueServiceEvent.PostNetworkCloseEvent(pServerSocketItem->GetSocketID(), dwClientAddr, dwConnectTime);

        //释放连接
        FreeSocketItem(pServerSocketItem);
    }
    catch (...) {}

    return true;
}

//通知回调函数（发送队列线程调用）
void  CTCPNetworkEngine::OnQueueServiceSink(WORD wIdentifier, void * pBuffer, WORD wDataSize)
{
    switch (wIdentifier)
    {
    case QUEUE_SEND_REQUEST:		//发送请求
    {
                                        //效验数据
                                        tagSendDataRequest * pSendDataRequest = (tagSendDataRequest *)pBuffer;
                                        ASSERT(wDataSize >= (sizeof(tagSendDataRequest)-sizeof(pSendDataRequest->cbSendBuf)));
                                        ASSERT(wDataSize == (pSendDataRequest->wDataSize + sizeof(tagSendDataRequest)-sizeof(pSendDataRequest->cbSendBuf)));

                                        //群发数据
                                        if (pSendDataRequest->dwSocketID == 0)
                                        {
                                            //获取活动项
                                            {
                                                CWHDataLocker lcok(m_CriticalSection);
                                                m_TempSocketItem.RemoveAll();
                                                m_TempSocketItem.Copy(m_ActiveSocketItem);
                                            }

                                            //循环发送数据
                                            CServerSocketItem * pServerSocketItem = NULL;
                                            for (INT_PTR i = 0; i < m_TempSocketItem.GetCount(); i++)
                                            {
                                                pServerSocketItem = m_TempSocketItem[i];
                                                ASSERT(pServerSocketItem != NULL);
                                                CWHDataLocker lock(pServerSocketItem->GetSignedLock());
                                                if (pServerSocketItem->IsAllowBatch())//第一次不允许？？
                                                {
                                                    pServerSocketItem->SendData(pSendDataRequest->cbSendBuf, pSendDataRequest->wDataSize, pSendDataRequest->wMainCmdID,
                                                        pSendDataRequest->wSubCmdID, pServerSocketItem->GetRountID());
												
                                                }
                                            }
                                        }
                                        else
                                        {
                                            //单项发送
                                            CServerSocketItem * pServerSocketItem = EnumSocketItem(LOWORD(pSendDataRequest->dwSocketID));
                                            CWHDataLocker lock(pServerSocketItem->GetSignedLock());

                                            pServerSocketItem->SendData(pSendDataRequest->cbSendBuf, pSendDataRequest->wDataSize, pSendDataRequest->wMainCmdID,
                                                pSendDataRequest->wSubCmdID, HIWORD(pSendDataRequest->dwSocketID));
                                        }

                                        break;
    }
    case QUEUE_SAFE_CLOSE:		//安全关闭
    {
                                    //效验数据
                                    ASSERT(wDataSize == sizeof(tagSafeCloseSocket));
                                    tagSafeCloseSocket * pSafeCloseSocket = (tagSafeCloseSocket *)pBuffer;

                                    //安全关闭
                                    try
                                    {
                                        CServerSocketItem * pServerSocketItem = EnumSocketItem(LOWORD(pSafeCloseSocket->dwSocketID));
                                        CWHDataLocker lock(pServerSocketItem->GetSignedLock());
                                        pServerSocketItem->ShutDownSocket(HIWORD(pSafeCloseSocket->dwSocketID));
                                    }
                                    catch (...)
                                    {
                                    }
                                    break;
    }
    case QUEUE_ALLOW_BATCH:		//允许群发
    {
                                    //效验数据
                                    ASSERT(wDataSize == sizeof(tagAllowBatchSend));
                                    tagAllowBatchSend * pAllowBatchSend = (tagAllowBatchSend *)pBuffer;

                                    //设置群发
                                    CServerSocketItem * pServerSocketItem = EnumSocketItem(LOWORD(pAllowBatchSend->dwSocketID));
                                    CWHDataLocker lock(pServerSocketItem->GetSignedLock());
                                    pServerSocketItem->AllowBatchSend(HIWORD(pAllowBatchSend->dwSocketID), pAllowBatchSend->cbAllow ? true : false, pAllowBatchSend->cbMask);

                                    break;
    }
    case QUEUE_DETECT_SOCKET:	//检测连接
    {
                                    //获取活动项
        {
            CWHDataLocker lcok(m_CriticalSection);
            m_TempSocketItem.RemoveAll();
            m_TempSocketItem.Copy(m_ActiveSocketItem);
        }
		//break; //田丰
                                    //构造数据
                                    CMD_KN_DetectSocket DetectSocket;
                                    ZeroMemory(&DetectSocket, sizeof(DetectSocket));

                                    //变量定义
                                    WORD wRountID = 0;
                                    DWORD dwNowTickCount = GetTickCount();
                                    DWORD dwBreakTickCount = __max(dwNowTickCount - m_dwLastDetect, TIME_BREAK_READY);

                                    //设置变量
                                    m_dwLastDetect = GetTickCount();

                                    //检测连接
                                    for (INT_PTR i = 0; i < m_TempSocketItem.GetCount(); i++)
                                    {
                                        //变量定义
                                        CServerSocketItem * pServerSocketItem = m_TempSocketItem[i];
                                        DWORD dwRecvTickCount = pServerSocketItem->GetRecvTickCount();
                                        CWHDataLocker lock(pServerSocketItem->GetSignedLock());

                                        //间隔检查
                                        if (dwRecvTickCount >= dwNowTickCount) continue;

                                        //检测连接
                                        if (pServerSocketItem->IsReadySend() == true)
                                        {
//#ifndef DEBUG
                                            if ((dwNowTickCount - dwRecvTickCount) > dwBreakTickCount)
                                            {
                                                pServerSocketItem->CloseSocket(pServerSocketItem->GetRountID());
                                                continue;
                                            }
//#endif
                                        }
                                        else
                                        {
                                            if ((dwNowTickCount - dwRecvTickCount) > TIME_BREAK_CONNECT)
                                            {
                                                pServerSocketItem->CloseSocket(pServerSocketItem->GetRountID());
                                                continue;
                                            }
                                        }

                                        //发送数据
                                        if (pServerSocketItem->IsReadySend() == true)
                                        {
                                            wRountID = pServerSocketItem->GetRountID();
                                            DetectSocket.dwSendTickCount = GetTickCount();
											Socket::PB_KN_DetectSocket pSocket;
											pSocket.set_dwsendtickcount(DetectSocket.dwSendTickCount);
											pSocket.set_dwrecvtickcount(DetectSocket.dwRecvTickCount);
											std::string pbdata = pSocket.SerializePartialAsString();
                                            pServerSocketItem->SendData((void*)pbdata.c_str(), pbdata.length(), MDM_KN_COMMAND, SUB_KN_DETECT_SOCKET, wRountID);
                                        }
                                    }

                                    break;
    }
    default:
    {
               ASSERT(FALSE);
    }
    }

    return;
}

//获取空闲对象
CServerSocketItem * CTCPNetworkEngine::ActiveSocketItem()
{
    CWHDataLocker lock(m_CriticalSection);

    //获取空闲对象
    CServerSocketItem * pServerSocketItem = NULL;
    if (m_FreeSocketItem.GetCount() > 0)
    {
        INT_PTR nItemPostion = m_FreeSocketItem.GetCount() - 1;
        pServerSocketItem = m_FreeSocketItem[nItemPostion];
        ASSERT(pServerSocketItem != NULL);
        m_FreeSocketItem.RemoveAt(nItemPostion);
        m_ActiveSocketItem.Add(pServerSocketItem);
    }

    //创建新对象
    if (pServerSocketItem == NULL)
    {
        WORD wStorageCount = (WORD)m_StorageSocketItem.GetCount();
        if (wStorageCount < m_wMaxSocketItem)
        {
            try
            {
                pServerSocketItem = new CServerSocketItem(wStorageCount, this);
                if (pServerSocketItem == NULL) return NULL;
                m_StorageSocketItem.Add(pServerSocketItem);
                m_ActiveSocketItem.Add(pServerSocketItem);
            }
            catch (...)
            {
				CString str;
				str.Format(TEXT("连接失败，目前连接数%d"), wStorageCount);
				CTraceService::TraceString(str, TraceLevel_Normal);
                return NULL;
            }
        }
    }

    return pServerSocketItem;
}

//获取连接对象
CServerSocketItem * CTCPNetworkEngine::EnumSocketItem(WORD wIndex)
{
    CWHDataLocker lock(m_CriticalSection);
    if (wIndex < m_StorageSocketItem.GetCount())
    {
        CServerSocketItem * pServerSocketItem = m_StorageSocketItem[wIndex];
        ASSERT(pServerSocketItem != NULL);
        return pServerSocketItem;
    }
    return NULL;
}

//释放连接对象
bool CTCPNetworkEngine::FreeSocketItem(CServerSocketItem * pServerSocketItem)
{
    //效验参数
    ASSERT(pServerSocketItem != NULL);

    //释放对象
    CWHDataLocker lock(m_CriticalSection);
    INT_PTR nActiveCount = m_ActiveSocketItem.GetCount();
    for (int i = 0; i < nActiveCount; i++)
    {
        if (pServerSocketItem == m_ActiveSocketItem[i])
        {
            m_ActiveSocketItem.RemoveAt(i);
            m_FreeSocketItem.Add(pServerSocketItem);
            return true;
        }
    }

    //释放失败
    ASSERT(FALSE);
    return false;
}

//检测连接
bool  CTCPNetworkEngine::DetectSocket()
{
    return m_SendQueueService.AddToQueue(QUEUE_DETECT_SOCKET, NULL, 0);
}


//发送函数
bool  CTCPNetworkEngine::SendData(DWORD dwSocketID, WORD wMainCmdID, WORD wSubCmdID)
{
	TRACE("\n\n<<<***Fun=[%s] SEND CMD[%d:%d]***\n\n", __FUNCTION__, wMainCmdID, wSubCmdID);

    //效益状态
    ASSERT(m_bService == true);
    if (m_bService == false) return false;

    //构造数据
    tagSendDataRequest SendRequest;
    SendRequest.wMainCmdID = wMainCmdID;
    SendRequest.wSubCmdID = wSubCmdID;
    SendRequest.dwSocketID = dwSocketID;
    SendRequest.wDataSize = 0;
	
    //加入发送队列
    WORD wSendSize = sizeof(SendRequest)-sizeof(SendRequest.cbSendBuf);
    return m_SendQueueService.AddToQueue(QUEUE_SEND_REQUEST, &SendRequest, wSendSize);
}

//发送函数
bool  CTCPNetworkEngine::SendData(DWORD dwSocketID, WORD wMainCmdID, WORD wSubCmdID, void * pData, WORD wDataSize)
{
	TRACE("\n\n<<<***Fun=[%s] SEND CMD[%d:%d]***\n\n", __FUNCTION__, wMainCmdID, wSubCmdID);

    //效益状态
    ASSERT(m_bService == true);
    if (m_bService == false) return false;

    //效益数据
    ASSERT((wDataSize + sizeof(TCP_Head)) <= SOCKET_TCP_BUFFER);
    if ((wDataSize + sizeof(TCP_Head)) > SOCKET_TCP_BUFFER) return false;

    //构造数据
    tagSendDataRequest SendRequest;
    SendRequest.wMainCmdID = wMainCmdID;
    SendRequest.wSubCmdID = wSubCmdID;
    SendRequest.dwSocketID = dwSocketID;
    SendRequest.wDataSize = wDataSize;

    if (wDataSize > 0)
    {
        ASSERT(pData != NULL);
        CopyMemory(SendRequest.cbSendBuf, pData, wDataSize);
    }

    //加入发送队列
    WORD wSendSize = sizeof(SendRequest)-sizeof(SendRequest.cbSendBuf) + wDataSize;
    return m_SendQueueService.AddToQueue(QUEUE_SEND_REQUEST, &SendRequest, wSendSize);
}

//批量发送
bool  CTCPNetworkEngine::SendDataBatch(WORD wMainCmdID, WORD wSubCmdID, void * pData, WORD wDataSize, BYTE cbBatchMask)
{
	TRACE("\n\n<<<***Fun=[%s] SEND CMD[%d:%d]***\n\n", __FUNCTION__, wMainCmdID, wSubCmdID);

    //效益状态
    if (m_bService == false) return false;

    //效益数据
    ASSERT((wDataSize + sizeof(TCP_Head)) <= SOCKET_TCP_BUFFER);
    if ((wDataSize + sizeof(TCP_Head)) > SOCKET_TCP_BUFFER) return false;

    //构造数据
    tagSendDataRequest SendRequest;
    SendRequest.wMainCmdID = wMainCmdID;
    SendRequest.wSubCmdID = wSubCmdID;
    SendRequest.dwSocketID = 0;
    SendRequest.wDataSize = wDataSize;

    if (wDataSize > 0)
    {
        ASSERT(pData != NULL);
        CopyMemory(SendRequest.cbSendBuf, pData, wDataSize);
    }

	

    //加入发送队列
    WORD wSendSize = sizeof(SendRequest)-sizeof(SendRequest.cbSendBuf) + wDataSize;
    return m_SendQueueService.AddToQueue(QUEUE_SEND_REQUEST, &SendRequest, wSendSize);
}

//关闭连接
bool  CTCPNetworkEngine::CloseSocket(DWORD dwSocketID)
{
    CServerSocketItem * pServerSocketItem = EnumSocketItem(LOWORD(dwSocketID));
    if (pServerSocketItem == NULL) return false;
    CWHDataLocker lock(pServerSocketItem->GetSignedLock());
    return pServerSocketItem->CloseSocket(HIWORD(dwSocketID));
}

//设置关闭
bool  CTCPNetworkEngine::ShutDownSocket(DWORD dwSocketID)
{
    tagSafeCloseSocket SafeCloseSocket;
    SafeCloseSocket.dwSocketID = dwSocketID;
    return m_SendQueueService.AddToQueue(QUEUE_SAFE_CLOSE, &SafeCloseSocket, sizeof(SafeCloseSocket));
}

//允许群发
bool  CTCPNetworkEngine::AllowBatchSend(DWORD dwSocketID, bool bAllow, BYTE cbBatchMask)
{
    tagAllowBatchSend AllowBatchSendNode;
    AllowBatchSendNode.dwSocketID = dwSocketID;
    AllowBatchSendNode.cbAllow = bAllow;
    AllowBatchSendNode.cbMask = cbBatchMask;
    return m_SendQueueService.AddToQueue(QUEUE_ALLOW_BATCH, &AllowBatchSendNode, sizeof(tagAllowBatchSend));
}

//////////////////////////////////////////////////////////////////////////

//建立对象函数
extern "C" __declspec(dllexport) void * __cdecl CreateTCPNetworkEngine(const GUID & Guid, DWORD dwInterfaceVer)
{
    //建立对象
    CTCPNetworkEngine * pTCPSocketEngine = NULL;
    try
    {
        pTCPSocketEngine = new CTCPNetworkEngine();
        if (pTCPSocketEngine == NULL) throw TEXT("创建失败");
        void * pObject = pTCPSocketEngine->QueryInterface(Guid, dwInterfaceVer);
        if (pObject == NULL) throw TEXT("接口查询失败");
        return pObject;
    }
    catch (...) {}

    //清理对象
    SafeDelete(pTCPSocketEngine);
    return NULL;
}



//////////////////////////////////////////////////////////////////////////