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