應為協議都是遵循:len type data這樣的結構,所以打包發送和接收解包可以有通用的做法,,,
//////////////////////////////////////////////////////////////////////////
//數據描述頭
struct tagDataDescribe
{
WORD wDataSize; //數據大小
WORD wDataDescribe; //數據描述
};
//////////////////////////////////////////////////////////////////////////
//數據包輔助類(多包組合成一包)
class CSendPacketHelper
{
//變量定義
protected:
WORD m_wDataSize; //數據大小
WORD m_wBufferSize; //緩沖大小
BYTE * m_pcbBuffer; //緩沖指針
//函數定義
public:
//構造函數
CSendPacketHelper(void * pcbBuffer, WORD wBufferSize)
{
m_wDataSize=0;
m_wBufferSize=wBufferSize;
m_pcbBuffer=(BYTE *)pcbBuffer;
}
//析構函數
virtual ~CSendPacketHelper() {}
//功能函數
public:
//清理數據
void CleanData() { m_wDataSize=0; }
//獲取大小
WORD GetDataSize() { return m_wDataSize; }
//獲取緩沖
void * GetDataBuffer() { return m_pcbBuffer; }
//插入數據(如果緩沖區長度不夠返回false)
// wDataSize, wDataType 實際上就是DataDescribe
bool AddPacket(void * pData, WORD wDataSize, WORD wDataType)
{
//效驗大小
ASSERT((wDataSize+sizeof(tagDataDescribe)+m_wDataSize)<=m_wBufferSize);
if ((wDataSize+sizeof(tagDataDescribe)+m_wDataSize)>m_wBufferSize) return false;
//插入數據
ASSERT(m_pcbBuffer!=NULL);
tagDataDescribe * pDataDescribe=(tagDataDescribe *)(m_pcbBuffer+m_wDataSize);
pDataDescribe->wDataSize=wDataSize;
pDataDescribe->wDataDescribe=wDataType;
//插入數據
if (wDataSize>0)
{
ASSERT(pData!=NULL);
CopyMemory(pDataDescribe+1,pData,wDataSize);
}
//設置數據
m_wDataSize+=sizeof(tagDataDescribe)+wDataSize;
return true;
}
};
/////////////////////////////////////////////////////////////////////////////////////////
//數據包輔助類
class CRecvPacketHelper
{
//變量定義
protected:
WORD m_wDataPos; //數據點
WORD m_wDataSize; //數據大小
BYTE * m_pcbBuffer; //緩沖指針
//函數定義
public:
//構造函數
CRecvPacketHelper(LPCVOID pcbBuffer, WORD wDataSize)
{
m_wDataPos=0;
m_wDataSize=wDataSize;
m_pcbBuffer=(BYTE *)pcbBuffer;
}
//析構函數
virtual ~CRecvPacketHelper() {}
//功能函數
public:
//獲取數據(返回一包數據)
// @DataDescribe:數據描述(len type)
// @return: 數據
void * GetData(tagDataDescribe & DataDescribe)
{
//效驗數據
if (m_wDataPos>=m_wDataSize)
{
ASSERT(m_wDataPos==m_wDataSize);
DataDescribe.wDataSize=0;
DataDescribe.wDataDescribe=DTP_NULL;
return NULL;
}
//獲取數據
ASSERT((m_wDataPos+sizeof(tagDataDescribe))<=m_wDataSize);
CopyMemory(&DataDescribe,m_pcbBuffer+m_wDataPos,sizeof(tagDataDescribe));
ASSERT((m_wDataPos+sizeof(tagDataDescribe)+DataDescribe.wDataSize)<=m_wDataSize);
//效驗數據
if ((m_wDataPos+sizeof(tagDataDescribe)+DataDescribe.wDataSize)>m_wDataSize)
{
DataDescribe.wDataSize=0;
DataDescribe.wDataDescribe=DTP_NULL;
return NULL;
}
//設置數據
void * pData=NULL;
if (DataDescribe.wDataSize>0) pData=m_pcbBuffer+m_wDataPos+sizeof(tagDataDescribe);
m_wDataPos+=sizeof(tagDataDescribe)+DataDescribe.wDataSize;
return pData;
};
};