MyFtpDlg.cpp
上传用户:qzzxgm
上传日期:2009-12-14
资源大小:1882k
文件大小:12k
源码类别:

书籍源码

开发平台:

Visual C++

  1. // MyFtpDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "MyFtp.h"
  5. #include "MyFtpDlg.h"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CAboutDlg dialog used for App About
  13. class CAboutDlg : public CDialog
  14. {
  15. public:
  16. CAboutDlg();
  17. // Dialog Data
  18. //{{AFX_DATA(CAboutDlg)
  19. enum { IDD = IDD_ABOUTBOX };
  20. //}}AFX_DATA
  21. // ClassWizard generated virtual function overrides
  22. //{{AFX_VIRTUAL(CAboutDlg)
  23. protected:
  24. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  25. //}}AFX_VIRTUAL
  26. // Implementation
  27. protected:
  28. //{{AFX_MSG(CAboutDlg)
  29. //}}AFX_MSG
  30. DECLARE_MESSAGE_MAP()
  31. };
  32. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  33. {
  34. //{{AFX_DATA_INIT(CAboutDlg)
  35. //}}AFX_DATA_INIT
  36. }
  37. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  38. {
  39. CDialog::DoDataExchange(pDX);
  40. //{{AFX_DATA_MAP(CAboutDlg)
  41. //}}AFX_DATA_MAP
  42. }
  43. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  44. //{{AFX_MSG_MAP(CAboutDlg)
  45. // No message handlers
  46. //}}AFX_MSG_MAP
  47. END_MESSAGE_MAP()
  48. /////////////////////////////////////////////////////////////////////////////
  49. // CMyFtpDlg dialog
  50. CMyFtpDlg::CMyFtpDlg(CWnd* pParent /*=NULL*/)
  51. : CDialog(CMyFtpDlg::IDD, pParent)
  52. {
  53. //{{AFX_DATA_INIT(CMyFtpDlg)
  54. m_strPassword = _T("");
  55. m_strServer = _T("");
  56. m_strUserName = _T("");
  57. //}}AFX_DATA_INIT
  58. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  59. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  60. m_pInetSession = NULL;
  61. m_pFtpConnection = NULL;
  62. m_pRemoteFinder = NULL;
  63. m_strLocalParentRoot = m_strRemoteParentRoot = "";
  64. }
  65. void CMyFtpDlg::DoDataExchange(CDataExchange* pDX)
  66. {
  67. CDialog::DoDataExchange(pDX);
  68. //{{AFX_DATA_MAP(CMyFtpDlg)
  69. DDX_Control(pDX, IDC_LIST_REMOTE, m_ctrlRemoteFiles);
  70. DDX_Control(pDX, IDC_LIST_LOCAL, m_ctrlLocalFiles);
  71. DDX_Text(pDX, IDC_PASSWORD, m_strPassword);
  72. DDX_Text(pDX, IDC_SERVER, m_strServer);
  73. DDX_Text(pDX, IDC_USER, m_strUserName);
  74. //}}AFX_DATA_MAP
  75. }
  76. BEGIN_MESSAGE_MAP(CMyFtpDlg, CDialog)
  77. //{{AFX_MSG_MAP(CMyFtpDlg)
  78. ON_WM_SYSCOMMAND()
  79. ON_WM_PAINT()
  80. ON_WM_QUERYDRAGICON()
  81. ON_BN_CLICKED(IDC_CONNECT, OnConnect)
  82. ON_BN_CLICKED(IDC_DOWNLOAD, OnDownload)
  83. ON_BN_CLICKED(IDC_LOCAL_UP, OnLocalUp)
  84. ON_BN_CLICKED(IDC_LOCAL_DOWND, OnLocalDownd)
  85. ON_BN_CLICKED(IDC_REMOTE_UP, OnRemoteUp)
  86. ON_BN_CLICKED(IDC_REMOTE_DOWND, OnRemoteDownd)
  87. ON_BN_CLICKED(IDC_UPLOAD, OnUpload)
  88. //}}AFX_MSG_MAP
  89. END_MESSAGE_MAP()
  90. /////////////////////////////////////////////////////////////////////////////
  91. // CMyFtpDlg message handlers
  92. BOOL CMyFtpDlg::OnInitDialog()
  93. {
  94. CDialog::OnInitDialog();
  95. // Add "About..." menu item to system menu.
  96. // IDM_ABOUTBOX must be in the system command range.
  97. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  98. ASSERT(IDM_ABOUTBOX < 0xF000);
  99. CMenu* pSysMenu = GetSystemMenu(FALSE);
  100. if (pSysMenu != NULL)
  101. {
  102. CString strAboutMenu;
  103. strAboutMenu.LoadString(IDS_ABOUTBOX);
  104. if (!strAboutMenu.IsEmpty())
  105. {
  106. pSysMenu->AppendMenu(MF_SEPARATOR);
  107. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  108. }
  109. }
  110. // Set the icon for this dialog.  The framework does this automatically
  111. //  when the application's main window is not a dialog
  112. SetIcon(m_hIcon, TRUE); // Set big icon
  113. SetIcon(m_hIcon, FALSE); // Set small icon
  114. // TODO: Add extra initialization here
  115. //初始化两个列表框控件
  116. SetListCtrlStyle(&m_ctrlLocalFiles);
  117. SetListCtrlStyle(&m_ctrlRemoteFiles);
  118. AddHeaders(&m_ctrlLocalFiles);
  119. AddHeaders(&m_ctrlRemoteFiles);
  120. //本地文件目录以c盘为根目录
  121. BrowseDir("c:",&m_ctrlLocalFiles,&m_LocalFinder,&m_arrLocalFiles);
  122. return TRUE;  // return TRUE  unless you set the focus to a control
  123. }
  124. void CMyFtpDlg::OnSysCommand(UINT nID, LPARAM lParam)
  125. {
  126. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  127. {
  128. CAboutDlg dlgAbout;
  129. dlgAbout.DoModal();
  130. }
  131. else
  132. {
  133. CDialog::OnSysCommand(nID, lParam);
  134. }
  135. }
  136. // If you add a minimize button to your dialog, you will need the code below
  137. //  to draw the icon.  For MFC applications using the document/view model,
  138. //  this is automatically done for you by the framework.
  139. void CMyFtpDlg::OnPaint() 
  140. {
  141. if (IsIconic())
  142. {
  143. CPaintDC dc(this); // device context for painting
  144. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  145. // Center icon in client rectangle
  146. int cxIcon = GetSystemMetrics(SM_CXICON);
  147. int cyIcon = GetSystemMetrics(SM_CYICON);
  148. CRect rect;
  149. GetClientRect(&rect);
  150. int x = (rect.Width() - cxIcon + 1) / 2;
  151. int y = (rect.Height() - cyIcon + 1) / 2;
  152. // Draw the icon
  153. dc.DrawIcon(x, y, m_hIcon);
  154. }
  155. else
  156. {
  157. CDialog::OnPaint();
  158. }
  159. }
  160. // The system calls this to obtain the cursor to display while the user drags
  161. //  the minimized window.
  162. HCURSOR CMyFtpDlg::OnQueryDragIcon()
  163. {
  164. return (HCURSOR) m_hIcon;
  165. }
  166. //设置列表控件CListCtrl的风格
  167. void CMyFtpDlg::SetListCtrlStyle(CListCtrl* pLstCtrl)
  168. {
  169. //获得原有风格
  170. DWORD dwStyle = GetWindowLong(pLstCtrl->m_hWnd, GWL_STYLE); 
  171. dwStyle &= ~(LVS_TYPEMASK);
  172. dwStyle &= ~(LVS_EDITLABELS);
  173. //设置新风格
  174.     SetWindowLong(pLstCtrl->m_hWnd, GWL_STYLE, dwStyle | LVS_REPORT|LVS_NOLABELWRAP|LVS_SHOWSELALWAYS);
  175. //设置扩展风格
  176. DWORD styles = LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES;
  177. ListView_SetExtendedListViewStyleEx(pLstCtrl->m_hWnd, styles, styles );
  178. }
  179. //给列表控件CListCtrl增加标题头
  180. void CMyFtpDlg::AddHeaders(CListCtrl* pLstCtrl)
  181. {
  182. TCHAR rgtsz[2][10] = {_T("文件名"), 
  183. _T("类型")};
  184. LV_COLUMN lvcolumn;
  185. CRect rect;
  186. pLstCtrl->GetWindowRect(&rect);
  187. for(int i=0;i<2;i++)
  188. {
  189. lvcolumn.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT 
  190. | LVCF_WIDTH | LVCF_ORDER;
  191. lvcolumn.fmt = LVCFMT_LEFT;
  192. lvcolumn.pszText = rgtsz[i];
  193. lvcolumn.iSubItem = i;
  194. lvcolumn.iOrder = i;
  195. if(i==0)
  196. {
  197. lvcolumn.cx = rect.Width()*2/3 ; 
  198. }
  199. else
  200. lvcolumn.cx = rect.Width()/3 ;
  201. pLstCtrl->InsertColumn(i, &lvcolumn);
  202. }
  203. }
  204. //连接ftp服务器
  205. void CMyFtpDlg::OnConnect() 
  206. {
  207. UpdateData(TRUE);
  208. //新建对话
  209. m_pInetSession=new CInternetSession(AfxGetAppName(),1,PRE_CONFIG_INTERNET_ACCESS);
  210. try 
  211. //新建连接对象
  212. m_pFtpConnection=m_pInetSession->GetFtpConnection(m_strServer,m_strUserName,
  213. m_strPassword); 
  214. catch(CInternetException *pEx) 
  215. {
  216. //获取错误
  217. TCHAR szError[1024];
  218. if(pEx->GetErrorMessage(szError,1024))
  219. AfxMessageBox(szError);
  220. else  
  221. AfxMessageBox("There was an exception");
  222. pEx->Delete();
  223. m_pFtpConnection=NULL;
  224. return;
  225. }
  226. m_pRemoteFinder = new CFtpFileFind(m_pFtpConnection);
  227. //获得服务器根目录的所有文件并在列表框中显示
  228. BrowseDir("",&m_ctrlRemoteFiles,m_pRemoteFinder,&m_arrRemoteFiles);
  229. }
  230. //获得指定目录下的所有文件,并在列表框中显示
  231. void CMyFtpDlg::BrowseDir(CString strDir,CListCtrl* pLstCtrl,CFileFind* pFinder,CArray<FILEITEM,FILEITEM&>* pFilesArr)
  232. {
  233. //清空列表框
  234. pLstCtrl->DeleteAllItems();
  235. pFilesArr->RemoveAll();
  236. int nIndex = 0;
  237. CString szDir = strDir;
  238. if(szDir.Right(1) != "\")
  239. szDir += "\";
  240. szDir += "*.*";
  241. //查找目录下的文件
  242. BOOL res = pFinder->FindFile(szDir);
  243. while(res)
  244. {
  245. res = pFinder->FindNextFile();
  246. //是目录
  247. if(pFinder->IsDirectory() && !pFinder->IsDots())
  248. {
  249. FILEITEM fileitem;
  250. fileitem.nItem = nIndex;
  251. fileitem.bDir = TRUE;
  252. fileitem.strFileName = pFinder->GetFileName();
  253. pFilesArr->Add(fileitem);
  254. AddItem(pLstCtrl,nIndex,TRUE,pFinder->GetFileName());
  255. }
  256. //是文件
  257. else if(!pFinder->IsDirectory() && !pFinder->IsDots())
  258. {
  259. FILEITEM fileitem;
  260. fileitem.nItem = nIndex;
  261. fileitem.bDir = FALSE;
  262. fileitem.strFileName = pFinder->GetFileName();
  263. pFilesArr->Add(fileitem);
  264. AddItem(pLstCtrl,nIndex,FALSE,pFinder->GetFileName());
  265. }
  266. nIndex++;
  267. }
  268. }
  269. //给列表框增加一条记录
  270. void CMyFtpDlg::AddItem(CListCtrl* pLstCtrl,int nIndex,BOOL bDir,CString strText)
  271. {
  272. LV_ITEM lvitem;
  273. int iActualItem;
  274. for(int iSubItem=0;iSubItem<2;iSubItem++)
  275. {
  276. lvitem.mask = LVIF_TEXT|(iSubItem == 0? LVIF_IMAGE : 0);
  277. lvitem.iItem = (iSubItem == 0)? nIndex : iActualItem;
  278. lvitem.iSubItem = iSubItem;
  279. switch(iSubItem)
  280. {
  281. case 0:
  282. lvitem.pszText =(LPTSTR)(LPCTSTR)strText;
  283. break;
  284. case 1:
  285. if(bDir)
  286. lvitem.pszText = "文件夹";
  287. else
  288. lvitem.pszText = "文件";
  289. break;
  290. if (iSubItem == 0)
  291. iActualItem = pLstCtrl->InsertItem(&lvitem);
  292. else
  293. pLstCtrl->SetItem(&lvitem);
  294. }
  295. }
  296. //响应“download”按钮
  297. void CMyFtpDlg::OnDownload() 
  298. {
  299. //如果没有建立连接,退出
  300. if(m_pFtpConnection==NULL)
  301. return;
  302. //下载所有选中文件
  303. POSITION pos = m_ctrlRemoteFiles.GetFirstSelectedItemPosition();
  304. if (pos == NULL)
  305. AfxMessageBox("请选择要下载的文件");
  306. else
  307. {
  308. while (pos)
  309. {
  310. int nItem = m_ctrlRemoteFiles.GetNextSelectedItem(pos);
  311. DownFile(m_arrRemoteFiles.GetAt(nItem));
  312. }
  313. }
  314. }
  315. //响应“Upload”按钮
  316. void CMyFtpDlg::OnUpload() 
  317. {
  318. //如果没有建立连接,退出
  319. if(m_pFtpConnection==NULL)
  320. return;
  321. //上传所有选中文件
  322. POSITION pos = m_ctrlLocalFiles.GetFirstSelectedItemPosition();
  323. if (pos == NULL)
  324. AfxMessageBox("请选择要上传的文件");
  325. else
  326. {
  327. while (pos)
  328. {
  329. int nItem = m_ctrlLocalFiles.GetNextSelectedItem(pos);
  330. UpFile(m_arrLocalFiles.GetAt(nItem));
  331. }
  332. }
  333. }
  334. //下载单个文件
  335. void CMyFtpDlg::DownFile(FILEITEM fileItem)
  336. {
  337. if(fileItem.bDir == TRUE)
  338. {
  339. AfxMessageBox("本程序暂时不支持下载整个文件夹,请选择文件下载");
  340. }
  341. else
  342. {
  343. //格式化文件名
  344. CString strLocalFile,strRemoteFile;
  345. strRemoteFile.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.strFileName);
  346. strLocalFile.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.strFileName);
  347. //下载
  348. if(m_pFtpConnection->GetFile(strLocalFile,strLocalFile))
  349. {
  350. CString strMsg;
  351. strMsg.Format("下载文件%s成功!",fileItem.strFileName);
  352. AfxMessageBox(strMsg);
  353. }
  354. }
  355. }
  356. //上传单个文件
  357. void CMyFtpDlg::UpFile(FILEITEM fileItem)
  358. {
  359. if(fileItem.bDir == TRUE)
  360. {
  361. AfxMessageBox("本程序暂时不支持上载整个文件夹,请选择文件上载");
  362. }
  363. else
  364. {
  365. //格式化文件名
  366. CString strLocalFile,strRemoteFile;
  367. strRemoteFile.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.strFileName);
  368. strLocalFile.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.strFileName);
  369. //上传
  370. if(m_pFtpConnection->PutFile(strLocalFile,strLocalFile))
  371. {
  372. CString strMsg;
  373. strMsg.Format("上载文件%s成功!",fileItem.strFileName);
  374. AfxMessageBox(strMsg);
  375. }
  376. }
  377. }
  378. //本地文件目录中向上一层
  379. void CMyFtpDlg::OnLocalUp() 
  380. {
  381. if(m_strLocalParentRoot=="")
  382. return;
  383. BrowseDir(m_strLocalParentRoot,&m_ctrlLocalFiles,&m_LocalFinder,&m_arrLocalFiles);
  384. }
  385. //本地文件目录中向下一层
  386. void CMyFtpDlg::OnLocalDownd() 
  387. {
  388. POSITION pos = m_ctrlLocalFiles.GetFirstSelectedItemPosition();
  389. if(pos)
  390. {
  391. int nItem = m_ctrlRemoteFiles.GetNextSelectedItem(pos);
  392. FILEITEM fileItem = m_arrLocalFiles.GetAt(nItem);
  393. if(fileItem.bDir==TRUE)
  394. {
  395. CString strDir;
  396. strDir.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.strFileName);
  397. m_strLocalParentRoot = m_LocalFinder.GetRoot();
  398. BrowseDir(strDir,&m_ctrlLocalFiles,&m_LocalFinder,&m_arrLocalFiles);
  399. }
  400. }
  401. }
  402. //远程文件目录中向上一层
  403. void CMyFtpDlg::OnRemoteUp() 
  404. {
  405. if(m_pFtpConnection==NULL)
  406. return;
  407. if(m_strRemoteParentRoot=="")
  408. return;
  409. BrowseDir(m_strRemoteParentRoot,&m_ctrlRemoteFiles,m_pRemoteFinder,&m_arrRemoteFiles);
  410. }
  411. //远程文件目录中向下一层
  412. void CMyFtpDlg::OnRemoteDownd() 
  413. {
  414. POSITION pos = m_ctrlRemoteFiles.GetFirstSelectedItemPosition();
  415. if(pos)
  416. {
  417. int nItem = m_ctrlRemoteFiles.GetNextSelectedItem(pos);
  418. FILEITEM fileItem = m_arrRemoteFiles.GetAt(nItem);
  419. if(fileItem.bDir==TRUE)
  420. {
  421. CString strDir;
  422. strDir.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.strFileName);
  423. m_strRemoteParentRoot = m_pRemoteFinder->GetRoot();
  424. BrowseDir(strDir,&m_ctrlRemoteFiles,m_pRemoteFinder,&m_arrRemoteFiles);
  425. }
  426. }
  427. }