BG

TcpPackClient.h

@@ -0,0 +1,126 @@
+/*
+ * Copyright: JessMA Open Source (ldcsaa@gmail.com)
+ *
+ * Author	: Bruce Liang
+ * Website	: https://github.com/ldcsaa
+ * Project	: https://github.com/ldcsaa/HP-Socket
+ * Blog		: http://www.cnblogs.com/ldcsaa
+ * Wiki		: http://www.oschina.net/p/hp-socket
+ * QQ Group	: 44636872, 75375912
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+#pragma once
+
+#include "TcpClient.h"
+#include "MiscHelper.h"
+
+template<class T> class CTcpPackClientT : public IPackClient, public T
+{
+	using __super = T;
+	using __super::SetLastError;
+	using __super::m_itPool;
+
+public:
+	using __super::Stop;
+	using __super::Wait;
+	using __super::GetState;
+
+public:
+	virtual BOOL SendPackets(const WSABUF pBuffers[], int iCount)
+	{
+		int iNewCount = iCount + 1;
+		unique_ptr<WSABUF[]> buffers(new WSABUF[iNewCount]);
+
+		DWORD dwHeader;
+		if(!::AddPackHeader(pBuffers, iCount, buffers, m_dwMaxPackSize, m_usHeaderFlag, dwHeader))
+			return FALSE;
+
+		return __super::SendPackets(buffers.get(), iNewCount);
+	}
+
+protected:
+	virtual EnHandleResult DoFireReceive(ITcpClient* pSender, const BYTE* pData, int iLength)
+	{
+		return ParsePack(this, &m_pkInfo, &m_lsBuffer, (CTcpPackClientT*)pSender, m_dwMaxPackSize, m_usHeaderFlag, pData, iLength);
+	}
+
+	virtual BOOL BeforeUnpause()
+	{
+		return (ParsePack(this, &m_pkInfo, &m_lsBuffer, (CTcpPackClientT*)this, m_dwMaxPackSize, m_usHeaderFlag) != HR_ERROR);
+	}
+
+	virtual BOOL CheckParams()
+	{
+		if	((m_dwMaxPackSize > 0 && m_dwMaxPackSize <= TCP_PACK_MAX_SIZE_LIMIT)	&&
+			(m_usHeaderFlag >= 0 && m_usHeaderFlag <= TCP_PACK_HEADER_FLAG_LIMIT)	)
+			return __super::CheckParams();
+
+		SetLastError(SE_INVALID_PARAM, __FUNCTION__, ERROR_INVALID_PARAMETER);
+		return FALSE;
+	}
+
+	virtual void Reset()
+	{
+		m_lsBuffer.Clear();
+		m_pkInfo.Reset();
+
+		__super::Reset();
+	}
+
+public:
+	virtual void SetMaxPackSize		(DWORD dwMaxPackSize)		{ENSURE_HAS_STOPPED(); m_dwMaxPackSize = dwMaxPackSize;}
+	virtual void SetPackHeaderFlag	(USHORT usPackHeaderFlag)	{ENSURE_HAS_STOPPED(); m_usHeaderFlag  = usPackHeaderFlag;}
+	virtual DWORD GetMaxPackSize	()							{return m_dwMaxPackSize;}
+	virtual USHORT GetPackHeaderFlag()							{return m_usHeaderFlag;}
+
+private:
+	EnHandleResult DoFireSuperReceive(ITcpClient* pSender, const BYTE* pData, int iLength)
+		{return __super::DoFireReceive(pSender, pData, iLength);}
+
+	friend EnHandleResult ParsePack<>	(CTcpPackClientT* pThis, TPackInfo<TItemListEx>* pInfo, TItemListEx* pBuffer, CTcpPackClientT* pSocket,
+										DWORD dwMaxPackSize, USHORT usPackHeaderFlag);
+
+public:
+	CTcpPackClientT(ITcpClientListener* pListener)
+	: T					(pListener)
+	, m_dwMaxPackSize	(TCP_PACK_DEFAULT_MAX_SIZE)
+	, m_usHeaderFlag	(TCP_PACK_DEFAULT_HEADER_FLAG)
+	, m_pkInfo			(nullptr)
+	, m_lsBuffer		(m_itPool)
+	{
+
+	}
+
+	virtual ~CTcpPackClientT()
+	{
+		ENSURE_STOP();
+	}
+
+private:
+	DWORD	m_dwMaxPackSize;
+	USHORT	m_usHeaderFlag;
+
+	TPackInfo<TItemListEx>	m_pkInfo;
+	TItemListEx				m_lsBuffer;
+};
+
+typedef CTcpPackClientT<CTcpClient> CTcpPackClient;
+
+#ifdef _SSL_SUPPORT
+
+#include "SSLClient.h"
+typedef CTcpPackClientT<CSSLClient> CSSLPackClient;
+
+#endif