1. 程式人生 > >解決網路資料粘包的問題

解決網路資料粘包的問題

一個網路上來的資料包之後,我們必需將收到的資料包整理成為一個一個的完成的資料包.這裡寫了一個程式碼來搞定這個問題的.
首先我們需要有一個記憶體緩衝區.還有我們已經使用了這個緩衝區的大小的標誌.


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 ;
}