开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 网络编程 > 查看源码
FileTransferDlg.cpp
资源名称：多功能聊天程序.rar　[点击查看]
上传用户：onsales
上传日期：2010-01-31
资源大小：224k
文件大小：8k
源码类别：

网络编程

开发平台：

Visual C++

FileTransferDlg.cpp：源码内容
  1. // FileTransferDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "FileTransfer.h"
  6. #include "FileTransferDlg.h"
  7. #include "FileInfo.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CAboutDlg dialog used for App About
  18. class CAboutDlg : public CDialog
  19. {
  20. public:
  21. CAboutDlg();
  23. // Dialog Data
  24. //{{AFX_DATA(CAboutDlg)
  25. enum { IDD = IDD_ABOUTBOX };
  26. //}}AFX_DATA
  28. // ClassWizard generated virtual function overrides
  29. //{{AFX_VIRTUAL(CAboutDlg)
  30. protected:
  31. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  32. //}}AFX_VIRTUAL
  34. // Implementation
  35. protected:
  36. //{{AFX_MSG(CAboutDlg)
  37. //}}AFX_MSG
  38. DECLARE_MESSAGE_MAP()
  39. };
  41. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  42. {
  43. //{{AFX_DATA_INIT(CAboutDlg)
  44. //}}AFX_DATA_INIT
  45. }
  47. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  48. {
  49. CDialog::DoDataExchange(pDX);
  50. //{{AFX_DATA_MAP(CAboutDlg)
  51. //}}AFX_DATA_MAP
  52. }
  54. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  55. //{{AFX_MSG_MAP(CAboutDlg)
  56. // No message handlers
  57. //}}AFX_MSG_MAP
  58. END_MESSAGE_MAP()
  60. /////////////////////////////////////////////////////////////////////////////
  61. // CFileTransferDlg dialog
  63. CFileTransferDlg::CFileTransferDlg(CWnd* pParent /*=NULL*/)
  64. : CDialog(CFileTransferDlg::IDD, pParent)
  65. {
  66. //{{AFX_DATA_INIT(CFileTransferDlg)
  67. m_FileName = _T("");
  68. m_FileSize = _T("");
  69. m_Name = _T("");
  70. //}}AFX_DATA_INIT
  71. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  72. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  73. }
  75. void CFileTransferDlg::DoDataExchange(CDataExchange* pDX)
  76. {
  77. CDialog::DoDataExchange(pDX);
  78. //{{AFX_DATA_MAP(CFileTransferDlg)
  79. DDX_Control(pDX, IDOK, m_Button);
  80. DDX_Control(pDX, IDC_PROGRESS, m_Progress);
  81. DDX_Control(pDX, IDC_IPADDRESS, m_IPADDRESS);
  82. DDX_Text(pDX, IDC_FILENAME, m_FileName);
  83. DDX_Text(pDX, IDC_FILESIZE, m_FileSize);
  84. DDX_Text(pDX, IDC_NAME, m_Name);
  85. //}}AFX_DATA_MAP
  86. }
  88. BEGIN_MESSAGE_MAP(CFileTransferDlg, CDialog)
  89. //{{AFX_MSG_MAP(CFileTransferDlg)
  90. ON_WM_SYSCOMMAND()
  91. ON_WM_PAINT()
  92. ON_WM_QUERYDRAGICON()
  93. //}}AFX_MSG_MAP
  94. END_MESSAGE_MAP()
  96. /////////////////////////////////////////////////////////////////////////////
  97. // CFileTransferDlg message handlers
  99. BOOL CFileTransferDlg::OnInitDialog()
  100. {
  101. CDialog::OnInitDialog();
  103. // Add "About..." menu item to system menu.
  105. // IDM_ABOUTBOX must be in the system command range.
  106. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  107. ASSERT(IDM_ABOUTBOX < 0xF000);
  109. CMenu* pSysMenu = GetSystemMenu(FALSE);
  110. if (pSysMenu != NULL)
  111. {
  112. CString strAboutMenu;
  113. strAboutMenu.LoadString(IDS_ABOUTBOX);
  114. if (!strAboutMenu.IsEmpty())
  115. {
  116. pSysMenu->AppendMenu(MF_SEPARATOR);
  117. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  118. }
  119. }
  121. // Set the icon for this dialog.  The framework does this automatically
  122. //  when the application's main window is not a dialog
  123. SetIcon(m_hIcon, TRUE); // Set big icon
  124. SetIcon(m_hIcon, FALSE); // Set small icon
  125. CFile mfile;
  126. mfile.Open("file.txt",CFile::modeRead);
  127.     int filelong=0;
  128. filelong=mfile.GetLength();
  129. char pbuf[1000];
  130. mfile.Read( pbuf, sizeof( pbuf ) ); 
  131. //if(filelong<256) 
  132. mfile.Close();
  133. pbuf[filelong]='';
  134. IPstr="";
  135. m_Name="";
  136. int i=2;
  137. while(pbuf[i]!='#')
  138. {
  139. IPstr+=pbuf[i];
  140. i++;
  141. }
  142. i++;
  143. while(pbuf[i]!='#')
  144. {
  145. m_Name+=pbuf[i];
  146. i++;
  147. }
  148. if(pbuf[0]=='4')//说明作为服务器传送文件
  149. {
  150. sign=1;
  151. m_Button.SetWindowText("传送文件");
  152. }
  153. else {
  154. sign=0;
  155. m_Button.SetWindowText("接收文件");
  156. }
  157. // TODO: Add extra initialization here
  158. m_IPADDRESS.SetWindowText(IPstr);
  159. this->UpdateData(FALSE);
  160. return TRUE;  // return TRUE  unless you set the focus to a control
  161. }
  163. void CFileTransferDlg::OnSysCommand(UINT nID, LPARAM lParam)
  164. {
  165. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  166. {
  167. CAboutDlg dlgAbout;
  168. dlgAbout.DoModal();
  169. }
  170. else
  171. {
  172. CDialog::OnSysCommand(nID, lParam);
  173. }
  174. }
  176. // If you add a minimize button to your dialog, you will need the code below
  177. //  to draw the icon.  For MFC applications using the document/view model,
  178. //  this is automatically done for you by the framework.
  180. void CFileTransferDlg::OnPaint() 
  181. {
  182. if (IsIconic())
  183. {
  184. CPaintDC dc(this); // device context for painting
  186. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  188. // Center icon in client rectangle
  189. int cxIcon = GetSystemMetrics(SM_CXICON);
  190. int cyIcon = GetSystemMetrics(SM_CYICON);
  191. CRect rect;
  192. GetClientRect(&rect);
  193. int x = (rect.Width() - cxIcon + 1) / 2;
  194. int y = (rect.Height() - cyIcon + 1) / 2;
  196. // Draw the icon
  197. dc.DrawIcon(x, y, m_hIcon);
  198. }
  199. else
  200. {
  201. CDialog::OnPaint();
  202. }
  203. }
  205. // The system calls this to obtain the cursor to display while the user drags
  206. //  the minimized window.
  207. HCURSOR CFileTransferDlg::OnQueryDragIcon()
  208. {
  209. return (HCURSOR) m_hIcon;
  210. }
  212. void CFileTransferDlg::OnOK() 
  213. {
  214. // TODO: Add extra validation here
  215. if(sign==1)this->OnServer();
  216. else this->OnClient();
  217. CDialog::OnOK();
  218. }
  220. void CFileTransferDlg::OnServer()
  221. {
  222. CFileDialog Dlg(TRUE);
  223. if(Dlg.DoModal()!=IDOK)
  224. CDialog::OnOK();
  225. if(!MyFile.Open(Dlg.GetPathName(), CFile::modeRead | CFile::typeBinary))
  226. {
  227. MessageBox("文件不存在!",MB_OK);
  228. CDialog::OnOK();
  229. }
  230. m_FileName=MyFile.GetFileName();
  231. m_FileSize.Format("%dbytes",MyFile.GetLength());
  232. UpdateData(FALSE);
  233. CSocket sockSrvr;
  234. sockSrvr.Create(800);
  236. sockSrvr.Listen();
  237. CSocket sockRecv;
  238. sockSrvr.Accept(sockRecv);
  240. SOCKET_STREAM_FILE_INFO StreamFileInfo;
  241. WIN32_FIND_DATA             FindFileData;
  243. FindClose(FindFirstFile(Dlg.GetPathName(),&FindFileData));
  244.     memset(&StreamFileInfo,0,sizeof(SOCKET_STREAM_FILE_INFO));
  245.     strcpy(StreamFileInfo.szFileTitle,MyFile.GetFileTitle());
  247.     StreamFileInfo.dwFileAttributes     =       FindFileData.dwFileAttributes;
  248.     StreamFileInfo.ftCreationTime       =       FindFileData.ftCreationTime;
  249.     StreamFileInfo.ftLastAccessTime     =       FindFileData.ftLastAccessTime;
  250.     StreamFileInfo.ftLastWriteTime      =       FindFileData.ftLastWriteTime;
  251.     StreamFileInfo.nFileSizeHigh        =       FindFileData.nFileSizeHigh;
  252.     StreamFileInfo.nFileSizeLow         =       FindFileData.nFileSizeLow;
  254. sockRecv.Send(&StreamFileInfo,sizeof(SOCKET_STREAM_FILE_INFO));
  255. int total=0;
  256. int range=(int)StreamFileInfo.nFileSizeLow;
  257. m_Progress.SetRange(0,range);
  258. UINT dwRead=0;
  259. while(dwRead<StreamFileInfo.nFileSizeLow)
  260. {
  261. byte* data = new byte[1024];
  262. UINT dw=MyFile.Read(data, 1024);
  263. sockRecv.Send(data, dw);
  264. total+=sizeof(data);
  265. m_Progress.SetPos(total);
  266. dwRead+=dw;
  267. }
  268. MyFile.Close();
  270. sockRecv.Close();
  271. AfxMessageBox("发送完毕！");
  272. }
  274. void CFileTransferDlg::OnClient()
  275. {
  276. AfxSocketInit(NULL);
  277. CSocket sockClient;
  278. sockClient.Create();
  280. CString szIP;
  281. szIP=IPstr;
  283. if(!sockClient.Connect((LPCTSTR)szIP, 800))
  284. {
  285. AfxMessageBox("连接到对方机器失败!");
  286. return;
  287. }
  289. SOCKET_STREAM_FILE_INFO StreamFileInfo;
  290. sockClient.Receive(&StreamFileInfo,sizeof(SOCKET_STREAM_FILE_INFO));
  291. m_FileName=StreamFileInfo.szFileTitle;
  292. m_FileSize.Format("%dbytes",StreamFileInfo.nFileSizeLow);
  293. UpdateData(FALSE);
  294. CFile destFile(StreamFileInfo.szFileTitle, CFile::modeCreate | CFile::modeWrite | CFile::typeBinary);
  296. UINT dwRead = 0;
  297. int total=0;
  298. int range=(int)StreamFileInfo.nFileSizeLow;
  299. m_Progress.SetRange(0,range);
  300. while(dwRead<StreamFileInfo.nFileSizeLow)
  301. {
  302. byte* data = new byte[1024];
  303. memset(data,0,1024);
  304. UINT dw=sockClient.Receive(data, 1024);
  305. destFile.Write(data, dw);
  306. total+=sizeof(data);
  307. m_Progress.SetPos(total);
  308. dwRead+=dw;
  309. }
  311. SetFileTime((HANDLE)destFile.m_hFile,&StreamFileInfo.ftCreationTime,
  312.                 &StreamFileInfo.ftLastAccessTime,&StreamFileInfo.ftLastWriteTime);
  313. destFile.Close();
  314. SetFileAttributes(StreamFileInfo.szFileTitle,StreamFileInfo.dwFileAttributes);
  315. sockClient.Close();
  316. AfxMessageBox("接收完毕！");
  317. }