解決網路資料粘包的問題
一個網路上來的資料包之後,我們必需將收到的資料包整理成為一個一個的完成的資料包.這裡寫了一個程式碼來搞定這個問題的.
首先我們需要有一個記憶體緩衝區.還有我們已經使用了這個緩衝區的大小的標誌.
void getData(const char* buff,int nSize)
{
char *pCheckBuff=(char*)buff; // 檢查的緩衝區
int nCountSize=nSize; // 緩衝區的大小
bool bBuff=false; // 是否使用本地緩衝區
// 如果以前有斷的資料包
if (nUsedSize_!=0)
{
memcpy(buf_+nUsedSize_,buff,nSize);
nUsedSize_+=nSize;
bBuff=true;
nCountSize=nUsedSize_;
}
// 開始對 pCheckBuff 記憶體塊進行解包
// 當處理完成,或是資料包的長度沒有這麼長的時候退出
int offset=0;
int leave=nCountSize;
while (1)
{
leave=nCountSize-offset;
// 得到資料包頭
if (leave>DATA_HEAD_SIZE)
{
DataHead* pHead=(DataHead*)pCheckBuff+offset;
// 檢查資料包是否正確
if (check_head(pHead))
{
// 檢查資料體是否完整
if (pHead->nSize<=leave)
{
ACE_Message_Block *new_mb = blockTash_.get_free_block();
new_mb->copy(pCheckBuff+offset,pHead->nSize);
//new_mb->wr_ptr(pHead->nSize);
((Acceptor*)pAcceptor_)->on_user_data(this->nIndex_,*new_mb,pHead->nSize);
//
// ((Acceptor*)pAcceptor_)->on_user_data(this->nIndex_,mb,result.bytes_transferred ());
offset+=pHead->nSize;
}
// 不完整的資料包
else
{
goto lable1;
}
}
// 錯誤的資料包.將把這一次的資料全部丟失
else
{
nUsedSize_=0;
leave=0;
break;
}
}
// 沒有完整的資料包頭
else
{
goto lable1;
}
}
return ;
// 儲存資料
// 移動緩衝區的資料
lable1:
// 在本地的緩衝區中
if (leave!=0)
{
memcpy(buf_,pCheckBuff+offset,leave);
nUsedSize_=leave;
}
return ;
}