fileinti.cpp

文件浏览功能 是一个在wince下的简单的文件访问程序

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

}