CSocket文件传输 分段传输的关键代码

分段传输的关键代码

 

struct FILEINFO
{
 int fileLength; //文件长度
 char fileName[100]; //文件名
};

　　

 

服务器：

CFileDialog fileDialog(TRUE); //选择文件
 if (fileDialog.DoModal()==IDOK)
 {
       CString filename = fileDialog.GetPathName(); //文件路径
        CString name = fileDialog.GetFileName(); //文件名字

  //发送文件
  CFile file;
  if (file.Open(filename,CFile::modeReadWrite|CFile::typeBinary))
  { //打开文件


   FILEINFO myFileInfo; //文件结构
   int length = file.GetLength(); //获取文件的长度
   m_CtrlProgressSend.SetRange32(0,length); //设置进度条大小
   m_CtrlProgressSend.SetPos(0);
   myFileInfo.fileLength = length; //赋值文件长度
   strcpy (myFileInfo.fileName,name.GetBuffer(name.GetLength())); //拷贝文件名
   name.ReleaseBuffer(); //释放buffer

   int iBufSize = 1024*5; //设置文件缓冲区长度
   int iSize = iBufSize ; //一次读取的长度
   LPBYTE pBuf= new BYTE [iBufSize]; //定义缓冲区


   m_sockClient.Send(&myFileInfo, sizeof (myFileInfo)); //发送文件头信息

   //发送文件
   int iNnmByte;
   UINT uiTotal = 0 ;
   while (uiTotal<length)
   {
    if (length-uiTotal < iBufSize) //如果发送小于1024*5 ，则发送剩余那些....
         iSize = length-uiTotal;
    iSize = file.Read(pBuf,iSize); //读取文件到缓冲区

    iNnmByte = m_sockClient.Send(pBuf,iSize); //发送剩下iSize-iCount的文件
    uiTotal += iNnmByte; //uiTotal总发送量
    m_CtrlProgressSend.SetPos(uiTotal);

   }

  }
  m_info.ReplaceSel( "发送文件完毕\r\n" );
  file.Close();
 }

 m_CtrlProgressSend.SetPos(0);
}

　　

 

 

客户端

 

FILEINFO myFileInfo;
m_client.Receive(&myFileInfo, sizeof (myFileInfo)); //接受文件信息

m_progress.SetRange32(0,myFileInfo.fileLength); //设置进度


CString str;
str.Format(myFileInfo.fileName); //文件名字

CFile file;
if (file.Open(str,CFile::modeCreate|CFile::modeWrite))
{
    UINT uiLength;
    uiLength = myFileInfo.fileLength ; //文件长度

    int iBufSize = 1024*5;
    int iSize = iBufSize ;
    LPBYTE pBuf = new BYTE [iBufSize];
    int iNumByte;
    UINT uiTotal = 0 ;


    while (uiTotal < uiLength)
    {
        if (uiLength-uiTotal<iBufSize)
            iSize = uiLength - uiTotal ;

        iNumByte = m_client.Receive(pBuf,iSize);
        file.Write(pBuf,iNumByte); //写入文件
        uiTotal+=iNumByte;
        m_progress.SetPos(uiTotal);


    }

    if (pBuf)
    {
        delete []  pBuf;
        pBuf = NULL;
    }
}
file.Close();
m_info.ReplaceSel( "接受文件成功\r\n" );
m_progress.SetPos(0);