mirror of
https://gitee.com/ldcsaa/HP-Socket.git
synced 2024-12-08 14:47:58 +08:00
118 lines
3.3 KiB
C++
118 lines
3.3 KiB
C++
/*
|
|
* 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
|
|
{
|
|
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
|