FTP客户端上传下载实现

 1 #ifndef __ftp_client_h__
 2 #define __ftp_client_h__
 3 
 4 #include <afxinet.h>
 5 #include <iostream>
 6 #include <sstream>
 7 #include <queue>
 8 #include <boost/filesystem/operations.hpp>
 9 #include <boost/filesystem/path.hpp>
10 
11 class CFtpClient
12 {
13 public:
14     CFtpClient();
15     ~CFtpClient();
16 
17     bool Connect();
18     bool Close();
19     int Download(std::string strServerDir = "\\", std::string strFilename = "*");
20     int Upload(const std::string strLocalPath);
21 private:
22     //下载文件夹
23     int DownloadFolder();
24     //下载文件
25     int DownloadFile(std::string strServerDir, std::string strFilename);
26     //创建多级目录
27     void CreateDirectory(boost::filesystem::path path);
28 private:
29     CInternetSession* m_pInternetSession;
30     CFtpConnection* m_pFtpConnection;
31 
32     //参数配置
33     std::string m_strServerAddress;               //FTP:服务器地址
34     int m_nServerPort;                            //FTP:服务器端口号
35     std::string m_strUsername;                    //FTP:登录用户名
36     std::string m_strPassword;                    //FTP:登录密码
37     //上传目录
38     std::string m_strUploadServerDir;             //上传服务器存储目录
39     //下载目录
40     std::string m_strDownloadLocalDir;            //下载本地存储目录
41 
42     std::queue<std::string> queServerDir;
43 };
44 
45 
46 #endif

  1 // 模块名称：
  2 // 模块描述：FTP上传下载类
  3 // 开发作者：可笑痴狂
  4 // 创建日期：2015-12-04
  5 // 模块版本：1.0.0.0
  6 //------------------------------------------------------------------------------
  7 
  8 #include <string>
  9 #include <iostream>
 10 #include <sstream>
 11 #include "CFtpClient.h"
 12 #include <boost/assert.hpp>
 13 #include <boost/lexical_cast.hpp>
 14 #include <boost/foreach.hpp>
 15 
 16 
 17 CFtpClient::CFtpClient()
 18 {
 19     m_pInternetSession = NULL;
 20     m_pFtpConnection = NULL;
 21     m_strServerAddress = "127.0.0.1";
 22     m_strUsername = "123";
 23     m_strPassword = "123";
 24     m_nServerPort = 21;
 25     m_strDownloadLocalDir = "D:\\DownloadLocal";
 26     m_strUploadServerDir = "\\";
 27 }
 28 
 29 CFtpClient::~CFtpClient()
 30 {
 31     this->Close();
 32 }
 33 
 34 bool CFtpClient::Connect()
 35 {
 36     if (m_pInternetSession != NULL)
 37     {
 38         return true;
 39     }
 40 
 41     m_pInternetSession = new CInternetSession("OTC_FTP_Client");
 42     int nCnt = 1;
 43     while(nCnt++)
 44     {
 45         try
 46         {
 47             m_pFtpConnection = m_pInternetSession->GetFtpConnection(m_strServerAddress.c_str(), m_strUsername.c_str(), m_strPassword.c_str(), 
　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　m_nServerPort);
 48         }
 49         catch(CInternetException *pEx)
 50         {
 51             char szErrMsg[1024+1] = {0};
 52             if (pEx->GetErrorMessage(szErrMsg, 1024))
 53             {
 54                 std::cout << "连接FTP服务器失败, ErrMsg:" << szErrMsg << std::endl;
 55                 if (nCnt > 5)
 56                 {
 57                     return false;
 58                 }
 59                 std::cout << "正在尝试重新连接[" << nCnt-1 << "]......" << std::endl;
 60             }
 61             continue;
 62         }
 63         return true;
 64     }
 65 
 66     return true;
 67 }
 68 
 69 bool CFtpClient::Close()
 70 {
 71     if (NULL != m_pInternetSession)
 72     {
 73         m_pInternetSession->Close();
 74         delete m_pInternetSession;
 75         m_pInternetSession = NULL;
 76     }
 77     if (NULL != m_pFtpConnection)
 78     {
 79         m_pFtpConnection->Close();
 80         delete m_pFtpConnection;
 81         m_pFtpConnection = NULL;
 82     }
 83 
 84     return true;
 85 }
 86 int CFtpClient::Download(std::string strServerDir /*="\\"*/, std::string strFilename /*= "*"*/)
 87 {
 88     if ("*" == strFilename)
 89     {
 90         queServerDir.push(strServerDir);
 91         return this->DownloadFolder();
 92     }
 93     else
 94     {
 95         return this->DownloadFile(strServerDir, strFilename);
 96     }
 97 
 98 }
 99 int CFtpClient::DownloadFile(std::string strServerDir, std::string strFilename)
100 {
101     int nRet = 0;
102     CFtpFileFind* pFtpFileFind = NULL;
103     if(!this->Connect())
104     {
105         nRet = -1;
106         goto __end;
107     }
108     else
109     {
110         boost::filesystem::path LocalPath(m_strDownloadLocalDir);
111         if (!boost::filesystem::exists(LocalPath))
112         {
113             this->CreateDirectory(LocalPath);
114         }
115         if (0 == m_pFtpConnection->SetCurrentDirectory(strServerDir.c_str()))
116         {
117             nRet = -1;
118             std::cout << "设置服务器下载目录[" << strServerDir << "]失败!ErrCode=" << GetLastError() << std::endl;
119             DWORD dw = 0;
120             char szBuf[512]={0};
121             DWORD dwLen = sizeof(szBuf)-1;
122             InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
123             std::cout << "错误信息：" << szBuf << std::endl;
124             goto __end;
125         }
126         if (pFtpFileFind == NULL)
127         {
128             pFtpFileFind = new CFtpFileFind(m_pFtpConnection);
129         }
130         int nRet = pFtpFileFind->FindFile(strFilename.c_str());
131         if(nRet)
132         {
133             nRet = pFtpFileFind->FindNextFile();
134 
135             std::string strLocalFilename = m_strDownloadLocalDir + "\\" + strFilename;
136             std::string strServerFilename = strFilename;
137             if(m_pFtpConnection->GetFile(strServerFilename.c_str(), strLocalFilename.c_str(), FALSE))
138                 std::cout << "下载文件[" << strServerFilename << "]到[" << strLocalFilename << "]成功!" << std::endl;
139             else
140             {
141                 std::cout << "下载文件[" << strServerFilename << "]到[" << strLocalFilename << "]失败!" << std::endl;
142                 char szBuf[512]={0};
143                 DWORD dw = 0;
144                 DWORD dwLen = sizeof(szBuf)-1;
145                 InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
146                 dw = GetLastError();
147                 std::cout << "错误信息：" << szBuf << std::endl;
148                 nRet = -1;
149             }
150         }
151     }
152 __end:
153     this->Close();
154     if (pFtpFileFind)
155     {
156         pFtpFileFind->Close();
157         delete pFtpFileFind;
158     }
159 
160     return nRet;
161 }
162 int CFtpClient::DownloadFolder()
163 {
164     std::string strServerDir;
165     std::string strFilename;
166     CFtpFileFind* pFtpFileFind = NULL;
167     int nRet = 0;
168 
169     if(!this->Connect())
170     {
171         nRet = -1;
172         goto __end;
173     }
174     else
175     {
176         while(!queServerDir.empty())
177         {
178             strServerDir = queServerDir.front();
179             queServerDir.pop();
180 
181             boost::filesystem::path LocalPath(m_strDownloadLocalDir);
182             if (!boost::filesystem::exists(LocalPath / strServerDir))
183             {
184                 this->CreateDirectory(LocalPath / strServerDir);
185             }
186             
187             if (0 == m_pFtpConnection->SetCurrentDirectory(strServerDir.c_str()))
188             {
189                 nRet = -1;
190                 std::cout << "设置服务器下载目录[" << strServerDir << "]失败!ErrCode=" << GetLastError() << std::endl;
191                 DWORD dw = 0;
192                 char szBuf[512]={0};
193                 DWORD dwLen = sizeof(szBuf)-1;
194                 InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
195                 std::cout << "错误信息：" << szBuf << std::endl;
196                 goto __end;
197             }
198             if (pFtpFileFind == NULL)
199             {
200                 pFtpFileFind = new CFtpFileFind(m_pFtpConnection);
201             }
202             
203             int nRet = pFtpFileFind->FindFile();
204             while(nRet)
205             {
206                 nRet = pFtpFileFind->FindNextFile();
207                 strFilename = pFtpFileFind->GetFilePath();
208                 if (pFtpFileFind->IsDirectory())
209                 {
210                     queServerDir.push(std::string(pFtpFileFind->GetFilePath()));
211                     continue;
212                 }
213 
214                 std::string strLocalFilename = m_strDownloadLocalDir + strFilename;
215                 std::string strServerFilename = strFilename;
216 
217                 if(m_pFtpConnection->GetFile(strServerFilename.c_str(), strLocalFilename.c_str(), FALSE))
218                     std::cout << "下载文件[" << strServerFilename << "]到[" << strLocalFilename << "]成功!" << std::endl;
219                 else
220                 {
221                     std::cout << "下载文件[" << strServerFilename << "]到[" << strLocalFilename << "]失败!" << std::endl;
222                     char szBuf[512]={0};
223                     DWORD dw = 0;
224                     DWORD dwLen = sizeof(szBuf)-1;
225                     InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
226                     dw = GetLastError();
227                     std::cout << "错误信息：" << szBuf << std::endl;
228                     nRet = -1;
229                 }
230             }
231         }    
232     }
233 __end:
234     this->Close();
235     if (pFtpFileFind != NULL)
236     {
237         pFtpFileFind->Close();
238         delete pFtpFileFind;
239     }
240     return nRet;
241 }
242 
243 int CFtpClient::Upload(const std::string strLocalPath)
244 {
245     int nRet = 0;
246     CFtpFileFind* pFtpFileFind = NULL;
247     const boost::filesystem::path& localPath(strLocalPath);
248 
249     if(!this->Connect())
250     {
251         nRet = -1;
252         goto __end;
253     }
254     else
255     {
256         if (0 == m_pFtpConnection->SetCurrentDirectory(m_strUploadServerDir.c_str()))
257         {
258             nRet = -1;
259             std::cout << "设置服务器上传目录[" << m_strUploadServerDir << "]失败!ErrCode=" << GetLastError() << std::endl;
260             DWORD dw = 0;
261             char szBuf[512]={0};
262             DWORD dwLen = sizeof(szBuf)-1;
263             InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
264             std::cout << "错误信息：" << szBuf << std::endl;
265             goto __end;
266         }
267 
268         if (boost::filesystem::exists(localPath))
269         {
270             if (boost::filesystem::is_directory(localPath))
271             {
272                 m_pFtpConnection->CreateDirectory(localPath.leaf().string().c_str());
273 
274                 boost::filesystem::directory_iterator itor_begin(localPath);
275                 boost::filesystem::directory_iterator itor_end;
276 
277                 std::string strServerDir = m_strUploadServerDir;
278                 for (; itor_begin != itor_end; itor_begin++)
279                 {//回溯算法
280                     m_strUploadServerDir += std::string("\\") + localPath.leaf().string();
281                     this->Upload(itor_begin->path().string());
282                     m_strUploadServerDir = strServerDir;
283                 }
284             }
285             else
286             {
287                 if(m_pFtpConnection->PutFile(localPath.string().c_str(), localPath.leaf().string().c_str(),
288                     FTP_TRANSFER_TYPE_BINARY,0))
289                     std::cout << "上传文件[" << localPath.leaf().string() << "]成功!" << std::endl;
290                 else
291                 {    
292                     DWORD dw0 = GetLastError();
293                     if (dw0)
294                     {
295                         char szBuf[512]={0};
296                         DWORD dw = 0;
297                         DWORD dwLen = sizeof(szBuf)-1;
298                         InternetGetLastResponseInfo(&dw, szBuf, &dwLen);
299                         std::cout << "上传文件[" << localPath.leaf().string() << "]失败!ErrCode:" << dw0 << "ErrMsg:" << szBuf << std::endl;
300                         goto __end;
301                     }
302                 }
303             }
304         }
305         else
306         {
307             std::cout << "指定上传文件不存在!" << std::endl;
308             nRet = -1;
309             goto __end;
310         }
311     }
312 
313 __end:
314     if (pFtpFileFind != NULL)
315     {
316         pFtpFileFind->Close();
317         delete pFtpFileFind;
318     }
319     this->Close();
320     return nRet;
321 }
322 
323 void CFtpClient::CreateDirectory(boost::filesystem::path path)
324 {
325     boost::filesystem::path::iterator itor = path.begin();
326     boost::filesystem::path pathTmp;
327     while(itor != path.end())
328     {
329         pathTmp /= (*itor).string();
330         if (!pathTmp.empty() && !boost::filesystem::exists(pathTmp))
331         {
332             boost::filesystem::create_directory(pathTmp);
333         }
334         ++itor;
335     }
336 }