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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 进程与线程 > 查看源码
DownloadFtp.cpp
资源名称：NetDownMTR.rar　[点击查看]
上传用户：oadesign
上传日期：2013-12-25
资源大小：265k
文件大小：9k
源码类别：

进程与线程

开发平台：

Visual C++

DownloadFtp.cpp：源码内容
  1. // DownloadFtp.cpp: implementation of the CDownloadFtp class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "NetDownMTR.h"
  7. #include "DownloadFtp.h"
  9. #ifdef _DEBUG
  10. #undef THIS_FILE
  11. static char THIS_FILE[]=__FILE__;
  12. #define new DEBUG_NEW
  13. #endif
  15. void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR );
  17. //////////////////////////////////////////////////////////////////////
  18. // Construction/Destruction
  19. //////////////////////////////////////////////////////////////////////
  21. CDownloadFtp::CDownloadFtp()
  22. {
  23. }
  25. CDownloadFtp::~CDownloadFtp()
  26. {
  27. }
  29. //
  30. // 创建一个 FTP 的数据通道连接
  31. //
  32. BOOL CDownloadFtp::CreateFTPDataConnect(CSocketClient &SocketClient)
  33. {
  34. ASSERT ( SocketClient.m_hSocket == INVALID_SOCKET );
  35. // 设被动模式
  36. if ( !m_SocketClient.SendString ( "PASVrn" ) ) return FALSE;
  37. CString csResponseStr;
  38. if ( !m_SocketClient.GetResponse ( 227, &csResponseStr ) )
  39. return FALSE;
  40. CString csPasvIP;
  41. USHORT uPasvPort = 0;
  42. if ( !m_SocketClient.GetIPAndPortByPasvString ( csResponseStr, csPasvIP, uPasvPort ) )
  43. return FALSE;
  45. // 创建数据通道连接
  46. if ( !SocketClient.Connect ( csPasvIP, uPasvPort ) )
  47. return FALSE;
  49. return TRUE;
  50. }
  52. //
  53. // 连接需要下载的服务器
  54. //
  55. BOOL CDownloadFtp::Connect ()
  56. {
  57. if ( !CDownloadPub::Connect () )
  58. return FALSE;
  60. int nResponseCode = m_SocketClient.GetResponse ();
  61. if ( nResponseCode != 220 )
  62. {
  63. Log ( L_WARNING, "Connect to [%s:%u] failed", m_csServer, m_nPort );
  64. }
  66. // 登录
  67. if ( m_csUsername.IsEmpty () )
  68. m_csUsername = "anonymous";
  69. if ( !m_SocketClient.SendString ( "USER %srn", m_csUsername ) ) return FALSE;
  70. nResponseCode = m_SocketClient.GetResponse ();
  71. // 需要密码
  72. if ( nResponseCode == 331 )
  73. {
  74. if ( !m_SocketClient.SendString ( "PASS %srn", m_csPassword ) ) return FALSE;
  75. nResponseCode = m_SocketClient.GetResponse ();
  76. }
  78. // 登录失败
  79. if ( nResponseCode != 230 )
  80. {
  81. Log ( L_ERROR, "Login failed" );
  82. return FALSE;
  83. }
  85. return TRUE;
  86. }
  88. BOOL CDownloadFtp::DownloadOnce()
  89. {
  90. // 不需要下载了
  91. int nWillDownloadSize = Get_WillDownloadSize(); // 本次应该下载的字节数
  92. int nDownloadedSize = Get_DownloadedSize (); // 已下载字节数
  93. if ( nWillDownloadSize > 0 && nDownloadedSize >= nWillDownloadSize )
  94. return DownloadEnd(TRUE);
  96. if ( !CDownloadPub::DownloadOnce () )
  97. return DownloadEnd(FALSE);
  99. // 设置接收数据类型为 二进制模式
  100. if ( !m_SocketClient.SendString ( "TYPE Irn" ) )
  101. return DownloadEnd(FALSE);
  102. if ( !m_SocketClient.GetResponse ( 200 ) )
  103. return DownloadEnd(FALSE);
  104. SLEEP_RETURN_Down ( 0 );
  106. // 创建数据通道连接
  107. CSocketClient SocketClient;
  108. if ( !CreateFTPDataConnect ( SocketClient ) )
  109. return DownloadEnd(FALSE);
  110. SLEEP_RETURN_Down ( 0 );
  112. // 设定下载文件的起始位置
  113. int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 开始位置
  114. if ( !m_SocketClient.SendString ( "REST %drn", nWillDownloadStartPos+nDownloadedSize ) )
  115. return DownloadEnd(FALSE);
  116. if ( !m_SocketClient.GetResponse ( 350 ) )
  117. return DownloadEnd(FALSE);
  119. // 提交下载文件的请求
  120. if ( !m_SocketClient.SendString ( "RETR %srn", m_csObject ) )
  121. return DownloadEnd(FALSE);
  122. if ( !m_SocketClient.GetResponse ( 150 ) )
  123. return DownloadEnd(FALSE);
  124. SLEEP_RETURN_Down ( 0 );
  126. // 从数据通道读取数据，并保存到文件中
  127. BOOL bRet = RecvDataAndSaveToFile(SocketClient);
  128. SocketClient.Disconnect ();
  129. return DownloadEnd ( bRet );
  131. }
  133. //
  134. // 获取远程站点信息，如：是否支持断点续传、要下载的文件大小和创建时间等
  135. //
  136. BOOL CDownloadFtp::GetRemoteSiteInfo()
  137. {
  138. BOOL bRet = FALSE;
  139. CSocketClient SocketClient;
  140. CString csReadData;
  141. char szRecvBuf[NET_BUFFER_SIZE] = {0};
  142. int nReadSize = 0;
  143. CString csOneLine;
  144. int i;
  146. if ( !CDownloadPub::GetRemoteSiteInfo() )
  147. goto Finished;
  149. if ( !m_SocketClient.Is_Connected () && !Connect () )
  150. goto Finished;
  152. // 设置接收数据类型为 ASCII
  153. if ( !m_SocketClient.SendString ( "TYPE Arn" ) ) goto Finished;
  154. if ( !m_SocketClient.GetResponse ( 200 ) )
  155. goto Finished;
  156. SLEEP_RETURN ( 0 );
  158. // 判断是否支持断点续传
  159. if ( !m_SocketClient.SendString ( "REST 1rn" ) ) goto Finished;
  160. m_bSupportResume = ( m_SocketClient.GetResponse () == 350 );
  161. TRACE ( "是否支持断点续传：%sn", m_bSupportResume ? "支持":"不支持" );
  162. if ( m_bSupportResume )
  163. {
  164. if ( !m_SocketClient.SendString ( "REST 0rn" ) ) goto Finished;
  165. VERIFY ( m_SocketClient.GetResponse ( 350 ) );
  166. }
  167. SLEEP_RETURN ( 0 );
  169. // 创建数据通道连接
  170. if ( !CreateFTPDataConnect ( SocketClient ) )
  171. goto Finished;
  172. // 发送列举文件命令
  173. if ( !m_SocketClient.SendString ( "LIST %srn", m_csObject ) )
  174. goto Finished;
  175. if ( !m_SocketClient.GetResponse ( 150 ) )
  176. goto Finished;
  177. SLEEP_RETURN ( 0 );
  179. // 从数据通道读取文件列表信息，直到命令通道返回 "226" 回应码，注意：这里接收数据用非阻塞型的。
  180. for ( i=0; ;i++ )
  181. {
  182. memset ( szRecvBuf, 0, sizeof(szRecvBuf) );
  183. nReadSize = SocketClient.Receive ( szRecvBuf, sizeof(szRecvBuf), FALSE );
  184. if ( nReadSize < 0 )
  185. {
  186. Log ( L_WARNING, "Receive file list info failed" );
  187. break;
  188. }
  189. csReadData += szRecvBuf;
  190. int nResponseCode = m_SocketClient.GetResponse ( (CString*)NULL, FALSE );
  191. if ( nResponseCode == -1 ) goto Finished;
  192. else if ( nResponseCode == 0 )
  193. {
  194. SLEEP_RETURN ( 100 );
  195. }
  196. else if ( nResponseCode == 226 )
  197. {
  198. break;
  199. }
  200. else goto Finished;
  201. SLEEP_RETURN ( 0 );
  202. }
  204. for ( i=0; !csReadData.IsEmpty() ; i++ )
  205. {
  206. csOneLine = GetOneLine ( csReadData );
  207. if ( !csOneLine.IsEmpty() )
  208. {
  209. ParseFileInfoStr ( csOneLine );
  210. }
  211. }
  212. bRet = TRUE;
  214. Finished:
  215. m_SocketClient.Disconnect ();
  216. return bRet;
  217. }
  219. //
  220. // 从 "-rw-rw-rw-   1 user     group    37979686 Mar  9 13:39 FTP-Callgle 1.4.0_20060309.cab" 字符串中分析
  221. // 出文件信息
  222. //
  223. void CDownloadFtp::ParseFileInfoStr(CString &csFileInfoStr)
  224. {
  225. csFileInfoStr.TrimLeft (); csFileInfoStr.TrimRight ();
  226. if ( csFileInfoStr.IsEmpty() ) return;
  227. BOOL bLastCharIsSpace = ( csFileInfoStr[0]==' ' );
  228. int nSpaceCount = 0;
  229. CString csFileTime1, csFileTime2, csFileTime3;
  230. CString csNodeStr;
  231. CString csFileName;
  232. for ( int i=0; i<csFileInfoStr.GetLength(); i++ )
  233. {
  234. if ( csFileInfoStr[i]==' ' )
  235. {
  236. if ( !bLastCharIsSpace )
  237. {
  238. csNodeStr = csFileInfoStr.Mid ( i );
  239. csNodeStr.TrimLeft (); csNodeStr.TrimRight ();
  240. nSpaceCount ++;
  241. int nFindPos = csNodeStr.Find ( " ", 0 );
  242. if ( nFindPos < 0 ) nFindPos = csNodeStr.GetLength() - 1;
  243. csNodeStr = csNodeStr.Left ( nFindPos );
  244. // 文件大小
  245. if ( nSpaceCount == 4 )
  246. {
  247. if ( m_nIndex == -1 ) // 主线程才需要获取文件大小的信息
  248. {
  249. m_nFileTotalSize = (int)atoi(csNodeStr);
  250. DownloadNotify ( m_nIndex, NOTIFY_TYPE_GOT_REMOTE_FILESIZE, (LPVOID)m_nFileTotalSize, m_pDownloadMTR );
  251. int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 开始位置
  252. int nWillDownloadSize = Get_WillDownloadSize(); // 本次应该下载的字节数
  253. int nDownloadedSize = Get_DownloadedSize (); // 已下载字节数
  254. if ( m_nFileTotalSize > 0 && nWillDownloadSize-nDownloadedSize > m_nFileTotalSize )
  255. Set_WillDownloadSize ( m_nFileTotalSize-nDownloadedSize );
  256. }
  257. }
  258. // 文件时间第一节
  259. else if ( nSpaceCount == 5 )
  260. {
  261. csFileTime1 = csNodeStr;
  262. }
  263. // 文件时间第二节
  264. else if ( nSpaceCount == 6 )
  265. {
  266. csFileTime2 = csNodeStr;
  267. }
  268. // 文件时间第三节
  269. else if ( nSpaceCount == 7 )
  270. {
  271. csFileTime3 = csNodeStr;
  272. }
  273. else if ( nSpaceCount > 7 )
  274. {
  275. csFileName = csFileInfoStr.Mid ( i );
  276. csFileName.TrimLeft(); csFileName.TrimRight();
  277. break;
  278. }
  279. }
  280. bLastCharIsSpace = TRUE;
  281. }
  282. else
  283. {
  284. bLastCharIsSpace = FALSE;
  285. }
  286. }
  288. GetFileTimeInfo ( csFileTime1, csFileTime2, csFileTime3 );
  289. }
  291. void CDownloadFtp::GetFileTimeInfo(CString csFileTime1, CString csFileTime2, CString csFileTime3)
  292. {
  293. if ( csFileTime3.IsEmpty() ) return;
  294. CString csYear, csMonth, csDay, csHour, csMinute, csSecond;
  295. int nMonth = GetMouthByShortStr ( csFileTime1 );
  296. ASSERT ( nMonth >= 1 && nMonth <= 12 );
  297. csMonth.Format ( "%02d", nMonth );
  299. COleDateTime tOleNow = COleDateTime::GetCurrentTime ();
  300. csDay.Format ( "%02d", atoi(csFileTime2) );
  301. csSecond = "00";
  302. // 带年的日期，如：Aug 21 2006
  303. if ( csFileTime3.Find ( ":", 0 ) < 0 )
  304. {
  305. csYear = csFileTime3;
  306. csHour = "00";
  307. csMinute = "00";
  308. }
  309. // 不带年的日期，如：Feb 21 01:11
  310. else
  311. {
  312. csYear.Format ( "%04d", tOleNow.GetYear() );
  313. int nPos = csFileTime3.Find ( ":", 0 );
  314. if ( nPos < 0 ) nPos = csFileTime3.GetLength() - 1;
  315. csHour = csFileTime3.Left ( nPos );
  316. csMinute = csFileTime3.Mid ( nPos+1 );
  317. }
  319. CString csFileTimeInfo;
  320. csFileTimeInfo.Format ( "%s-%s-%s %s:%s:%s", csYear, csMonth, csDay, csHour, csMinute, csSecond );
  321. ConvertStrToCTime ( csFileTimeInfo.GetBuffer(0), m_TimeLastModified );
  322. }