传输.txt

MFC类下的winsock多线程实现文件传输

//1：显示文件进度


//2：可以随时终止传输过程


//发送数据线程


UINT SendDataThread(LPVOID lpParam);


//接收数据线程


UINT ReceiveDataThread(LPVOID lpParam);


//发送数据按钮消息响应函数


void CTzg004Dlg::OnButtonSend() 


{


     // TODO: Add your control notification handler code here


     //初始化数据发送结束标志


     m_bSendEnd=FALSE;


     //初始化数据接收结束标志


     m_bRecEnd=FALSE;


     //更新对话框数据


     UpdateData(TRUE);


     //打开文件对话框


     CFileDialog dlg(TRUE,NULL,NULL,OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,


          "所有文件 (*.*)|*.*||");


     if(dlg.DoModal()==IDOK)


     {


         m_strFileName=dlg.GetPathName();


          //开始发送数据线程


         AfxBeginThread(SendDataThread,this,THREAD_PRIORITY_NORMAL);


     }


}


//接收数据按钮消息响应函数


void CTzg004Dlg::OnButtonReceive() 


{


     // TODO: Add your control notification handler code here


     //初始化数据发送结束标志


     m_bSendEnd=FALSE;


     //初始化数据接收结束标志


     m_bRecEnd=FALSE;


     UpdateData(TRUE);


     //开始接收数据线程 


     AfxBeginThread(ReceiveDataThread,this,THREAD_PRIORITY_NORMAL);


}


//终止发送按钮消息响应


void CTzg004Dlg::OnButtonSendEnd() 


{


     // TODO: Add your control notification handler code here


     //设置发送数据结束标志


     m_bSendEnd=TRUE;


}


//终止接收按钮消息响应


void CTzg004Dlg::OnButtonRecEnd() 


{


     // TODO: Add your control notification handler code here


     //设置接收数据结束标志


     m_bRecEnd=TRUE;


}


UINT SendDataThread(LPVOID lpParam)


{


     CTzg004Dlg *pDlg=(CTzg004Dlg *)lpParam;


     CFile file;


     if( !file.Open(pDlg->m_strFileName, CFile::modeRead) )


     {


         AfxMessageBox("打开文件出错！");


          return 0;


     }


     CSocket sockTemp;


     CString str,str1;


     sockTemp.Create(pDlg->m_iDataPort1); //得到端口号


     sockTemp.Listen(1);//只接受一个连接


     CSocket sockSend;


     //设置发送按钮禁止


     pDlg->GetDlgItem(IDC_BUTTON_SEND)->EnableWindow(FALSE);


     sockTemp.Accept(sockSend);//注意，sockTemp已交了自己的指针地址到sockSend，故不用Close


     //打开发送终止按钮


     pDlg->GetDlgItem(IDC_BUTTON_SEND_END)->EnableWindow(TRUE);


     


     


     int iBufSize = 1024 * 5;  


     int iSize = iBufSize;


     LPBYTE pBuf = new BYTE[iBufSize];     


     


     DWORD dwTemp = 0;


     BOOL bTest = sockSend.AsyncSelect(0);//由于CSocket实际是异步，将它变为同步（阻塞）方式。


     sockSend.IOCtl( FIONBIO, &dwTemp);//用IOCtl要将AsyncSelect的第一个参数为0，参看MSDN


     


     UINT uiLength = file.GetLength();


     sockSend.Send(&uiLength, 4);//传送文件大小到接收方（Client端）


     


     int iNumByte;


     UINT uiTotal = 0;


     while(uiTotal < uiLength)


     {


          int iEnd=pDlg->m_bSendEnd;


          //传送发送端状态（是否结束）


         iNumByte = sockSend.Send(&iEnd, sizeof(int));


          //发送错误


         if(iNumByte == SOCKET_ERROR)


          {


              AfxMessageBox("发送错误！");


              goto ExitLable1;


          }else if(iEnd==1)//发送端终止


          {


              AfxMessageBox("发送端终止");


              goto ExitLable1;


          }        


          //读取文件内容


         if((int)(uiLength - uiTotal) < iBufSize)


              iSize = uiLength - uiTotal;//当小于缓冲区iTEST时的处理


         iSize=file.Read(pBuf , iSize);//得到读取的字节数


          int iCount=0;


          //发送定长文件数据


         while(iCount<iSize)


          {


              iNumByte = sockSend.Send(pBuf, iSize-iCount);//注意iNumByte为实际的发送字节数，不要以iSize为准


              if(iNumByte == SOCKET_ERROR)


              {


                   AfxMessageBox("发送错误！");


                   goto ExitLable1;


              }


              iCount+=iNumByte;


              if(iCount<iSize)


              {


                   file.Seek(iSize-iCount,CFile::current);


              }


          }


          uiTotal += iCount;


          //设置发送数据进度条


         pDlg->m_CtrlProgressSend.SetPos(int(((double)uiTotal/uiLength)*100));


         str.Format("发送进度：%d%%",int(((double)uiTotal/uiLength)*100));


          //表明发送数据百分比


         pDlg->GetDlgItem(IDC_STATIC_SEND)->GetWindowText(str1);


         if(str1!=str)


              pDlg->GetDlgItem(IDC_STATIC_SEND)->SetWindowText(str); 


     }


     //发送文件成功


     AfxMessageBox("发送文件成功！");


ExitLable1:


     delete[] pBuf;


     file.Close();


     sockSend.Close();


     pDlg->m_CtrlProgressSend.SetPos(0);//恢复进度


     pDlg->GetDlgItem(IDC_BUTTON_SEND_END)->EnableWindow(FALSE);//设置发送结束按钮禁止


     pDlg->GetDlgItem(IDC_BUTTON_SEND)->EnableWindow(TRUE);//设置发送按钮正常


     pDlg->GetDlgItem(IDC_STATIC_SEND)->SetWindowText("发送进度："); //恢复提示进度


     return 0;


}



UINT ReceiveDataThread(LPVOID lpParam)


{


     CTzg004Dlg *pDlg=(CTzg004Dlg *)lpParam;


     //保存文件对话框


     CFileDialog dlg(FALSE,NULL,NULL,OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,


          "所有文件 (*.*)|*.*||");


     while(dlg.DoModal()!=IDOK)


     {


         AfxMessageBox("选择文件出错，请重新选择！");


     }         


     CString str,str1,str2;


     CSocket sockRecv;


     sockRecv.Create();


     pDlg->m_CtrlIPSend.GetWindowText(str);//得到发送端IP地址


     pDlg->GetDlgItem(IDC_BUTTON_RECEIVE)->EnableWindow(FALSE);//禁止接收按钮


     while(sockRecv.Connect(str,pDlg->m_iDataPort2)==0)//连接发送方地址，若上网，可改为实际IP地址，端口要跟Server端相同。


     {


         Sleep(50);


     }


     pDlg->GetDlgItem(IDC_BUTTON_REC_END)->EnableWindow(TRUE);//打开终止接收按钮


     str2=dlg.GetPathName();//得到文件名


     CFile file;


     file.Open(str2, CFile::modeCreate | CFile::modeWrite);


     BOOL bFileFail=FALSE;


     DWORD dwTemp = 0;


     sockRecv.AsyncSelect(0);


     sockRecv.IOCtl( FIONBIO, &dwTemp);//变为阻塞方式


     


     UINT uiLength;


     sockRecv.Receive(&uiLength, 4);//接收发方（Server端）的文件大小


     int iBufSize = 1024 * 5;


     int iSize = iBufSize;


     LPBYTE pBuf = new BYTE[iBufSize];


     int iNumByte;


     UINT uiTotal = 0;


     while(uiTotal < uiLength)


     {


          int iEnd=0;


          //接收端终止


         if(pDlg->m_bRecEnd)


          {


              AfxMessageBox("接收端终止！");


              goto ExitLable2;


          }


          //接收发送端状态数据


         iNumByte=sockRecv.Receive(&iEnd, sizeof(int));


         if(iNumByte == SOCKET_ERROR)


          {


              AfxMessageBox("接收信号错误！");


              goto ExitLable2;


          }


          //发送端终止


         if(iEnd==1)


          {


              AfxMessageBox("发送端终止！");


              goto ExitLable2;


          }




         if((int)(uiLength - uiTotal) < iBufSize)


              iSize = uiLength - uiTotal;


          int iCount=0;


          //读取定长数据


         while(iCount<iSize)


          {


              iNumByte = sockRecv.Receive(pBuf, iSize-iCount);


              if(iNumByte == SOCKET_ERROR)


              {


                   AfxMessageBox("接收错误！");


                   goto ExitLable2;


              }


              iCount+=iNumByte;


              file.Write(pBuf, iNumByte);


          }


          uiTotal += iCount;//以实际接收字节为准


          //设置接收进度


         pDlg->m_CtrlProgressRec.SetPos(int(((double)uiTotal/uiLength)*100));


         str.Format("接收进度：%d%%",int(((double)uiTotal/uiLength)*100));


          //显示接收进度百分比


         pDlg->GetDlgItem(IDC_STATIC_REC)->GetWindowText(str1);


         if(str1!=str)


              pDlg->GetDlgItem(IDC_STATIC_REC)->SetWindowText(str);


     }


     //接收文件成功


     AfxMessageBox("接收文件成功！");


     bFileFail=TRUE;


ExitLable2:


     delete[] pBuf;


     file.Close();


     //文件接收失败，则删除接收文件


     if(!bFileFail)


     {


         CFile::Remove( str2 );


     }


     sockRecv.Close();


     pDlg->m_CtrlProgressRec.SetPos(0);//恢复接收进度


     //禁止终止接收按钮


     pDlg->GetDlgItem(IDC_BUTTON_REC_END)->EnableWindow(FALSE);


     //打开接收按钮


     pDlg->GetDlgItem(IDC_BUTTON_RECEIVE)->EnableWindow(TRUE);


     //恢复提示进度


     pDlg->GetDlgItem(IDC_STATIC_REC)->SetWindowText("接收进度：");


     return 0;


}