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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
TestDlg.cpp
资源名称：FTP.rar　[点击查看]
上传用户：guibin827
上传日期：2021-12-07
资源大小：165k
文件大小：11k
源码类别：

Ftp客户端

开发平台：

Visual C++

TestDlg.cpp：源码内容
  1. // TestDlg.cpp : 实现文件
  2. //
  4. #include "stdafx.h"
  5. #include "Test.h"
  6. #include "TestDlg.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #endif
  10. // 用于应用程序“关于”菜单项的 CAboutDlg 对话框
  11. class CAboutDlg : public CDialog
  12. {
  13. public:
  14. CAboutDlg();
  15. // 对话框数据
  16. enum { IDD = IDD_ABOUTBOX };
  17. protected:
  18. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持
  19. // 实现
  20. protected:
  21. DECLARE_MESSAGE_MAP()
  22. };
  23. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  24. {
  25. }
  26. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  27. {
  28. CDialog::DoDataExchange(pDX);
  29. }
  30. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  31. END_MESSAGE_MAP()
  32. // CTestDlg 对话框
  33. CTestDlg::CTestDlg(CWnd* pParent /*=NULL*/)
  34. : CDialog(CTestDlg::IDD, pParent)
  35. {
  36. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  37. m_StrPassword = _T("");
  38. m_StrFTPServer = _T("");
  39. m_StrUserName = _T("");
  40. m_pIntSession = NULL;
  41. m_pFTPConnection = NULL;
  42. m_pRemoteFinder = NULL;
  43. m_StrLocalParentRoot = m_StrRemoteParentRoot = "";
  44. }
  46. void CTestDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialog::DoDataExchange(pDX);
  49. DDX_Control(pDX, IDC_LIST_REMOTE, m_ListRemoteFiles);
  50. DDX_Control(pDX, IDC_LIST_LOCAL, m_ListLocalFiles);
  51. DDX_Text(pDX, IDC_PASSWORD, m_StrPassword);
  52. DDX_Text(pDX, IDC_SERVER, m_StrFTPServer);
  53. DDX_Text(pDX, IDC_USER, m_StrUserName);
  54. }
  56. BEGIN_MESSAGE_MAP(CTestDlg, CDialog)
  57. ON_WM_SYSCOMMAND()
  58. ON_WM_PAINT()
  59. ON_WM_QUERYDRAGICON()
  60. ON_BN_CLICKED(IDC_CONNECT, OnBtnConnectFTPServer)
  61. ON_BN_CLICKED(IDC_DOWNLOAD, OnBtnDownloadFile)
  62. ON_BN_CLICKED(IDC_LOCAL_UP, OnLocalParentFolder)
  63. ON_BN_CLICKED(IDC_LOCAL_DOWND, OnLocalChildFolder)
  64. ON_BN_CLICKED(IDC_REMOTE_UP, OnRemoteParentFolder)
  65. ON_BN_CLICKED(IDC_REMOTE_DOWND, OnRemoteChildFolder)
  66. ON_BN_CLICKED(IDC_UPLOAD, OnBtnUpLoadFile)
  67. //}}AFX_MSG_MAP
  68. END_MESSAGE_MAP()
  71. // CTestDlg 消息处理程序
  73. BOOL CTestDlg::OnInitDialog()
  74. {
  75. CDialog::OnInitDialog();
  77. // 将“关于...”菜单项添加到系统菜单中。
  79. // IDM_ABOUTBOX 必须在系统命令范围内。
  80. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  81. ASSERT(IDM_ABOUTBOX < 0xF000);
  83. CMenu* pSysMenu = GetSystemMenu(FALSE);
  84. if (pSysMenu != NULL)
  85. {
  86. CString strAboutMenu;
  87. strAboutMenu.LoadString(IDS_ABOUTBOX);
  88. if (!strAboutMenu.IsEmpty())
  89. {
  90. pSysMenu->AppendMenu(MF_SEPARATOR);
  91. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  92. }
  93. }
  95. // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
  96. //  执行此操作
  97. SetIcon(m_hIcon, TRUE); // 设置大图标
  98. SetIcon(m_hIcon, FALSE); // 设置小图标
  100. //初始化远程和本地两个列表框控件
  101. SetListCtrlStyle(&m_ListLocalFiles);
  102. SetListCtrlStyle(&m_ListRemoteFiles);
  103. AddHeaders(&m_ListLocalFiles);
  104. AddHeaders(&m_ListRemoteFiles);
  105. //本地文件夹以C盘为根目录
  106. BrowseFolder("c:",&m_ListLocalFiles,&m_LocalFinder,&m_ArrayLocalFiles);
  107. return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
  108. }
  110. void CTestDlg::OnSysCommand(UINT nID, LPARAM lParam)
  111. {
  112. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  113. {
  114. CAboutDlg dlgAbout;
  115. dlgAbout.DoModal();
  116. }
  117. else
  118. {
  119. CDialog::OnSysCommand(nID, lParam);
  120. }
  121. }
  123. // 如果向对话框添加最小化按钮，则需要下面的代码
  124. //  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
  125. //  这将由框架自动完成。
  127. void CTestDlg::OnPaint() 
  128. {
  129. if (IsIconic())
  130. {
  131. CPaintDC dc(this); // 用于绘制的设备上下文
  133. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  135. // 使图标在工作矩形中居中
  136. int cxIcon = GetSystemMetrics(SM_CXICON);
  137. int cyIcon = GetSystemMetrics(SM_CYICON);
  138. CRect rect;
  139. GetClientRect(&rect);
  140. int x = (rect.Width() - cxIcon + 1) / 2;
  141. int y = (rect.Height() - cyIcon + 1) / 2;
  143. // 绘制图标
  144. dc.DrawIcon(x, y, m_hIcon);
  145. }
  146. else
  147. {
  148. CDialog::OnPaint();
  149. }
  150. }
  152. //当用户拖动最小化窗口时系统调用此函数取得光标显示。
  153. HCURSOR CTestDlg::OnQueryDragIcon()
  154. {
  155. return static_cast<HCURSOR>(m_hIcon);
  156. }
  157. //设置列表控件CListCtrl的风格
  158. void CTestDlg::SetListCtrlStyle(CListCtrl* pLstCtrl)
  159. {
  160. //获得原有风格
  161. DWORD dwStyle = GetWindowLong(pLstCtrl->m_hWnd, GWL_STYLE); 
  162. dwStyle &= ~(LVS_TYPEMASK);
  163. dwStyle &= ~(LVS_EDITLABELS);
  164. //设置新风格
  165.     SetWindowLong(pLstCtrl->m_hWnd, GWL_STYLE, dwStyle | LVS_REPORT|LVS_NOLABELWRAP|LVS_SHOWSELALWAYS);
  166. //设置扩展风格
  167. DWORD styles = LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES;
  168. ListView_SetExtendedListViewStyleEx(pLstCtrl->m_hWnd, styles, styles );
  169. }
  170. //给列表控件CListCtrl增加标题头
  171. void CTestDlg::AddHeaders(CListCtrl* pLstCtrl)
  172. {
  173. TCHAR rgtsz[2][20] = {_T("文件/文件夹名"), 
  174. _T("类型")};
  176. LV_COLUMN lvcolumn;
  177. CRect rect;
  178. pLstCtrl->GetWindowRect(&rect);
  179. for(int i=0;i<2;i++)
  180. {
  181. lvcolumn.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT 
  182. | LVCF_WIDTH | LVCF_ORDER;
  183. lvcolumn.fmt = LVCFMT_LEFT;
  184. lvcolumn.pszText = rgtsz[i];
  185. lvcolumn.iSubItem = i;
  186. lvcolumn.iOrder = i;
  187. if(i==0)
  188. {
  189. lvcolumn.cx = rect.Width()*2/3 ; 
  190. }
  191. else
  192. lvcolumn.cx = rect.Width()/3 ;
  193. pLstCtrl->InsertColumn(i, &lvcolumn);
  194. }
  195. }
  197. //连接FTP服务器
  198. void CTestDlg::OnBtnConnectFTPServer() 
  199. {
  200. UpdateData(TRUE);
  201. //新建CInternetSession对象
  202. m_pIntSession=new CInternetSession(AfxGetAppName(),1,PRE_CONFIG_INTERNET_ACCESS);
  203. try 
  204. { 
  205. //新建连接对象
  206. m_pFTPConnection=m_pIntSession->GetFtpConnection(m_StrFTPServer,m_StrUserName,m_StrPassword); 
  207. } 
  208. catch(CInternetException *pEx) 
  209. {
  210. //获取错误信息
  211. TCHAR szError[1024];
  212. if(pEx->GetErrorMessage(szError,1024))
  213. AfxMessageBox(szError);
  214. else  
  215. AfxMessageBox("出现意外情况");
  216. pEx->Delete();
  217. m_pFTPConnection=NULL;
  218. return;
  219. }
  220. m_pRemoteFinder = new CFtpFileFind(m_pFTPConnection);
  221. //获得服务器根目录的所有文件并在列表框中显示
  222. BrowseFolder("",&m_ListRemoteFiles,m_pRemoteFinder,&m_ArrayRemoteFiles);
  223. }
  225. //获得指定目录下的所有文件，并在列表框中显示
  226. void CTestDlg::BrowseFolder(CString strDir,CListCtrl* pLstCtrl,CFileFind* pFinder,CArray<FILEITEM,FILEITEM&>* pFilesArr)
  227. {
  228. //清空列表框
  229. pLstCtrl->DeleteAllItems();
  230. pFilesArr->RemoveAll();
  231. int nIndex = 0;
  232. CString szDir = strDir;
  233. if(szDir.Right(1) != "\")
  234. szDir += "\";
  235. szDir += "*.*";
  236. //查找目录下的文件
  237. BOOL res = pFinder->FindFile(szDir);
  238. while(res)
  239. {
  240. res = pFinder->FindNextFile();
  241. //是目录
  242. if(pFinder->IsDirectory() && !pFinder->IsDots())
  243. {
  244. FILEITEM fileitem;
  245. fileitem.nItem = nIndex;
  246. fileitem.bDir = TRUE;
  247. fileitem.StrFileName = pFinder->GetFileName();
  248. pFilesArr->Add(fileitem);
  249. AddItem(pLstCtrl,nIndex,TRUE,pFinder->GetFileName());
  250. }
  251. //是文件
  252. else if(!pFinder->IsDirectory() && !pFinder->IsDots())
  253. {
  254. FILEITEM fileitem;
  255. fileitem.nItem = nIndex;
  256. fileitem.bDir = FALSE;
  257. fileitem.StrFileName = pFinder->GetFileName();
  258. pFilesArr->Add(fileitem);
  259. AddItem(pLstCtrl,nIndex,FALSE,pFinder->GetFileName());
  260. }
  261. nIndex++;
  262. }
  263. }
  264. //给列表框增加一条记录
  265. void CTestDlg::AddItem(CListCtrl* pLstCtrl,int nIndex,BOOL bDir,CString strText)
  266. {
  267. LV_ITEM lvitem;
  268. int iActualItem;
  269. for(int iSubItem=0;iSubItem<2;iSubItem++)
  270. {
  271. lvitem.mask = LVIF_TEXT|(iSubItem == 0? LVIF_IMAGE : 0);
  272. lvitem.iItem = (iSubItem == 0)? nIndex : iActualItem;
  273. lvitem.iSubItem = iSubItem;
  274. switch(iSubItem)
  275. {
  276. case 0:
  277. lvitem.pszText =(LPTSTR)(LPCTSTR)strText;
  278. break;
  279. case 1:
  280. if(bDir)
  281. lvitem.pszText = "文件夹";
  282. else
  283. lvitem.pszText = "文件";
  284. break;
  285. } 
  286. if (iSubItem == 0)
  287. iActualItem = pLstCtrl->InsertItem(&lvitem);
  288. else
  289. pLstCtrl->SetItem(&lvitem);
  290. }
  291. }
  292. //响应“下载”按钮
  293. void CTestDlg::OnBtnDownloadFile() 
  294. {
  295. //如果没有建立连接，退出
  296. if(m_pFTPConnection==NULL)
  297. {
  298. MessageBox("没有连接FTP服务器！","信息提示",MB_OK);
  299. return;
  300. }
  301. //下载所有选中文件
  302. POSITION pos = m_ListRemoteFiles.GetFirstSelectedItemPosition();
  303. if (pos == NULL)
  304. AfxMessageBox("请选择要下载的文件");
  305. else
  306. {
  307. while (pos)
  308. {
  309. int nItem = m_ListRemoteFiles.GetNextSelectedItem(pos);
  310. DownLoadFile(m_ArrayRemoteFiles.GetAt(nItem));
  311. }
  312. }
  313. }
  314. //响应“上传”按钮
  315. void CTestDlg::OnBtnUpLoadFile() 
  316. {
  317. //如果没有建立连接，退出
  318. if(m_pFTPConnection==NULL)
  319. {
  320. MessageBox("没有连接FTP服务器！","信息提示",MB_OK);
  321. return;
  322. }
  323. //上传所有选中文件
  324. POSITION pos = m_ListLocalFiles.GetFirstSelectedItemPosition();
  325. if (pos == NULL)
  326. AfxMessageBox("请选择要上传的文件");
  327. else
  328. {
  329. while (pos)
  330. {
  331. int nItem = m_ListLocalFiles.GetNextSelectedItem(pos);
  332. UpLoadFile(m_ArrayLocalFiles.GetAt(nItem));
  333. }
  334. }
  335. }
  336. //下载单个文件
  337. void CTestDlg::DownLoadFile(FILEITEM fileItem)
  338. {
  339. if(fileItem.bDir == TRUE)
  340. {
  341. AfxMessageBox("程序暂时不支持下载整个文件夹,请选择文件后下载");
  342. }
  343. else
  344. {
  345. //格式化文件名
  346. CString strLocalFile,strRemoteFile;
  347. strRemoteFile.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.StrFileName);
  348. strLocalFile.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.StrFileName);
  349. //下载
  350. if(m_pFTPConnection->GetFile(strLocalFile,strLocalFile))
  351. {
  352. CString strMsg;
  353. strMsg.Format("下载文件%s成功!",fileItem.StrFileName);
  354. AfxMessageBox(strMsg);
  355. }
  356. }
  357. }
  358. //上传单个文件
  359. void CTestDlg::UpLoadFile(FILEITEM fileItem)
  360. {
  361. if(fileItem.bDir == TRUE)
  362. {
  363. AfxMessageBox("程序暂时不支持上载整个文件夹,请选择文件后上传");
  364. }
  365. else
  366. {
  367. //格式化文件名
  368. CString strLocalFile,strRemoteFile;
  369. strRemoteFile.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.StrFileName);
  370. strLocalFile.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.StrFileName);
  371. //上传
  372. if(m_pFTPConnection->PutFile(strLocalFile,strLocalFile))
  373. {
  374. CString strMsg;
  375. strMsg.Format("上传文件%s成功!",fileItem.StrFileName);
  376. AfxMessageBox(strMsg);
  377. }
  378. }
  379. }
  381. //本地文件目录中向上一层
  382. void CTestDlg::OnLocalParentFolder() 
  383. {
  384. if(m_StrLocalParentRoot=="")
  385. return;
  386. BrowseFolder(m_StrLocalParentRoot,&m_ListLocalFiles,&m_LocalFinder,&m_ArrayLocalFiles);
  387. }
  388. //本地文件目录中向下一层
  389. void CTestDlg::OnLocalChildFolder() 
  390. {
  391. POSITION pos = m_ListLocalFiles.GetFirstSelectedItemPosition();
  392. if(pos)
  393. {
  394. int nItem = m_ListRemoteFiles.GetNextSelectedItem(pos);
  395. FILEITEM fileItem = m_ArrayLocalFiles.GetAt(nItem);
  396. if(fileItem.bDir==TRUE)
  397. {
  398. CString strDir;
  399. strDir.Format("%s\%s",m_LocalFinder.GetRoot(),fileItem.StrFileName);
  400. m_StrLocalParentRoot = m_LocalFinder.GetRoot();
  401. BrowseFolder(strDir,&m_ListLocalFiles,&m_LocalFinder,&m_ArrayLocalFiles);
  402. }
  403. }
  404. }
  405. //远程文件目录中向上一层
  406. void CTestDlg::OnRemoteParentFolder() 
  407. {
  408. if(m_pFTPConnection==NULL)
  409. return;
  410. if(m_StrRemoteParentRoot=="")
  411. return;
  412. BrowseFolder(m_StrRemoteParentRoot,&m_ListRemoteFiles,m_pRemoteFinder,&m_ArrayRemoteFiles);
  413. }
  414. //远程文件目录中向下一层
  415. void CTestDlg::OnRemoteChildFolder() 
  416. {
  417. POSITION pos = m_ListRemoteFiles.GetFirstSelectedItemPosition();
  418. if(pos)
  419. {
  420. int nItem = m_ListRemoteFiles.GetNextSelectedItem(pos);
  421. FILEITEM fileItem = m_ArrayRemoteFiles.GetAt(nItem);
  422. if(fileItem.bDir==TRUE)
  423. {
  424. CString strDir;
  425. strDir.Format("%s\%s",m_pRemoteFinder->GetRoot(),fileItem.StrFileName);
  426. m_StrRemoteParentRoot = m_pRemoteFinder->GetRoot();
  427. BrowseFolder(strDir,&m_ListRemoteFiles,m_pRemoteFinder,&m_ArrayRemoteFiles);
  428. }
  429. }
  430. }