BG
This commit is contained in:
NH
221
TcpPackServer.h
Normal file
221
TcpPackServer.h
Normal file
@@ -0,0 +1,221 @@
|
||||
/*
|
||||
* 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 "TcpServer.h"
|
||||
#include "MiscHelper.h"
|
||||
|
||||
template<class T> class CTcpPackServerT : public IPackSocket, public T
|
||||
{
|
||||
using __super = T;
|
||||
using __super::SetConnectionReserved;
|
||||
using __super::GetConnectionReserved;
|
||||
using __super::GetMaxConnectionCount;
|
||||
using __super::GetSocketBufferSize;
|
||||
using __super::GetFreeBufferObjPool;
|
||||
using __super::GetFreeBufferObjHold;
|
||||
using __super::GetFreeSocketObjLockTime;
|
||||
using __super::GetFreeSocketObjPool;
|
||||
using __super::GetFreeSocketObjHold;
|
||||
using __super::SetLastError;
|
||||
|
||||
public:
|
||||
using __super::Stop;
|
||||
using __super::Wait;
|
||||
using __super::GetState;
|
||||
|
||||
public:
|
||||
virtual BOOL SendPackets(CONNID dwConnID, 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(dwConnID, buffers.get(), iNewCount);
|
||||
}
|
||||
|
||||
protected:
|
||||
virtual EnHandleResult DoFireAccept(TSocketObj* pSocketObj)
|
||||
{
|
||||
EnHandleResult result = __super::DoFireAccept(pSocketObj);
|
||||
|
||||
if(result != HR_ERROR)
|
||||
{
|
||||
TBuffer* pBuffer = m_bfPool.PickFreeBuffer(pSocketObj->connID);
|
||||
ENSURE(SetConnectionReserved(pSocketObj, TBufferPackInfo::Construct(pBuffer)));
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
virtual EnHandleResult DoFireHandShake(TSocketObj* pSocketObj)
|
||||
{
|
||||
EnHandleResult result = __super::DoFireHandShake(pSocketObj);
|
||||
|
||||
if(result == HR_ERROR)
|
||||
ReleaseConnectionExtra(pSocketObj);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
virtual EnHandleResult DoFireReceive(TSocketObj* pSocketObj, const BYTE* pData, int iLength)
|
||||
{
|
||||
TBufferPackInfo* pInfo = nullptr;
|
||||
GetConnectionReserved(pSocketObj, (PVOID*)&pInfo);
|
||||
ASSERT(pInfo);
|
||||
|
||||
TBuffer* pBuffer = (TBuffer*)pInfo->pBuffer;
|
||||
ASSERT(pBuffer && pBuffer->IsValid());
|
||||
|
||||
return ParsePack(this, pInfo, pBuffer, pSocketObj, m_dwMaxPackSize, m_usHeaderFlag, pData, iLength);
|
||||
}
|
||||
|
||||
virtual EnHandleResult DoFireClose(TSocketObj* pSocketObj, EnSocketOperation enOperation, int iErrorCode)
|
||||
{
|
||||
EnHandleResult result = __super::DoFireClose(pSocketObj, enOperation, iErrorCode);
|
||||
|
||||
ReleaseConnectionExtra(pSocketObj);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
virtual EnHandleResult DoFireShutdown()
|
||||
{
|
||||
EnHandleResult result = __super::DoFireShutdown();
|
||||
|
||||
m_bfPool.Clear();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
virtual BOOL BeforeUnpause(TSocketObj* pSocketObj)
|
||||
{
|
||||
if(!TSocketObj::IsValid(pSocketObj))
|
||||
return FALSE;
|
||||
|
||||
if(pSocketObj->IsPaused())
|
||||
return TRUE;
|
||||
|
||||
TBufferPackInfo* pInfo = nullptr;
|
||||
GetConnectionReserved(pSocketObj, (PVOID*)&pInfo);
|
||||
ASSERT(pInfo);
|
||||
|
||||
TBuffer* pBuffer = (TBuffer*)pInfo->pBuffer;
|
||||
ASSERT(pBuffer && pBuffer->IsValid());
|
||||
|
||||
return (ParsePack(this, pInfo, pBuffer, pSocketObj, 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 PrepareStart()
|
||||
{
|
||||
__super::PrepareStart();
|
||||
|
||||
m_bfPool.SetMaxCacheSize (GetMaxConnectionCount());
|
||||
m_bfPool.SetItemCapacity (GetSocketBufferSize());
|
||||
m_bfPool.SetItemPoolSize (GetFreeBufferObjPool());
|
||||
m_bfPool.SetItemPoolHold (GetFreeBufferObjHold());
|
||||
m_bfPool.SetBufferLockTime (GetFreeSocketObjLockTime());
|
||||
m_bfPool.SetBufferPoolSize (GetFreeSocketObjPool());
|
||||
m_bfPool.SetBufferPoolHold (GetFreeSocketObjHold());
|
||||
|
||||
m_bfPool.Prepare();
|
||||
}
|
||||
|
||||
virtual void ReleaseGCSocketObj(BOOL bForce = FALSE)
|
||||
{
|
||||
__super::ReleaseGCSocketObj(bForce);
|
||||
|
||||
#ifdef USE_EXTERNAL_GC
|
||||
m_bfPool.ReleaseGCBuffer(bForce);
|
||||
#endif
|
||||
}
|
||||
|
||||
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:
|
||||
void ReleaseConnectionExtra(TSocketObj* pSocketObj)
|
||||
{
|
||||
TBufferPackInfo* pInfo = nullptr;
|
||||
GetConnectionReserved(pSocketObj, (PVOID*)&pInfo);
|
||||
|
||||
if(pInfo != nullptr)
|
||||
{
|
||||
m_bfPool.PutFreeBuffer(pInfo->pBuffer);
|
||||
TBufferPackInfo::Destruct(pInfo);
|
||||
|
||||
ENSURE(SetConnectionReserved(pSocketObj, nullptr));
|
||||
}
|
||||
}
|
||||
|
||||
EnHandleResult DoFireSuperReceive(TSocketObj* pSocketObj, const BYTE* pData, int iLength)
|
||||
{return __super::DoFireReceive(pSocketObj, pData, iLength);}
|
||||
|
||||
friend EnHandleResult ParsePack<>(CTcpPackServerT* pThis, TBufferPackInfo* pInfo, TBuffer* pBuffer, TSocketObj* pSocket, DWORD dwMaxPackSize, USHORT usPackHeaderFlag);
|
||||
|
||||
public:
|
||||
CTcpPackServerT(ITcpServerListener* pListener)
|
||||
: T (pListener)
|
||||
, m_dwMaxPackSize (TCP_PACK_DEFAULT_MAX_SIZE)
|
||||
, m_usHeaderFlag (TCP_PACK_DEFAULT_HEADER_FLAG)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
virtual ~CTcpPackServerT()
|
||||
{
|
||||
ENSURE_STOP();
|
||||
}
|
||||
|
||||
private:
|
||||
DWORD m_dwMaxPackSize;
|
||||
USHORT m_usHeaderFlag;
|
||||
|
||||
CBufferPool m_bfPool;
|
||||
};
|
||||
|
||||
typedef CTcpPackServerT<CTcpServer> CTcpPackServer;
|
||||
|
||||
#ifdef _SSL_SUPPORT
|
||||
|
||||
#include "SSLServer.h"
|
||||
typedef CTcpPackServerT<CSSLServer> CSSLPackServer;
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user