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

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

进程与线程

开发平台：

Visual C++

DownloadMTR.cpp：源码内容
  1. // DownloadMTR.cpp: implementation of the CDownloadMTR class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "NetDownMTR.h"
  7. #include "DownloadMTR.h"
  8. #include <io.h>
  10. #ifdef _DEBUG
  11. #undef THIS_FILE
  12. static char THIS_FILE[]=__FILE__;
  13. #define new DEBUG_NEW
  14. #endif
  16. void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR );
  18. //////////////////////////////////////////////////////////////////////
  19. // Construction/Destruction
  20. //////////////////////////////////////////////////////////////////////
  22. CDownloadMTR::CDownloadMTR()
  23. : m_nThreadCount ( DEFAULT_THREAD_COUNT )
  24. , m_pDownloadPub_MTR ( NULL )
  25. , m_pDownloadPub_Info ( NULL )
  26. , m_pDownloadCellInfo ( NULL )
  27. , m_hThread ( NULL )
  28. , m_bForceDownload ( FALSE )
  29. , m_nTotalDownloadedSize_ThisTimes ( 0 )
  30. {
  31. memset ( &m_BaseDownInfo, 0, sizeof(t_BaseDownInfo) );
  32. m_hEvtEndModule = ::CreateEvent ( NULL, TRUE, FALSE, NULL );
  33. m_dwDownloadStartTime = GetTickCount();
  34. }
  36. CDownloadMTR::~CDownloadMTR()
  37. {
  38. StopDownload ();
  39. }
  41. //
  42. // 设置下载的线程数
  43. //
  44. BOOL CDownloadMTR::SetThreadCount(int nThreadCount)
  45. {
  46. if ( nThreadCount <= 0 || nThreadCount > MAX_DOWNLOAD_THREAD_COUNT )
  47. {
  48. Log ( L_WARNING, "Thread count %d is invalid. Rang [%d-%d]", nThreadCount, 1, MAX_DOWNLOAD_THREAD_COUNT );
  49. return FALSE;
  50. }
  51. if ( nThreadCount == m_nThreadCount )
  52. return TRUE;
  54. m_nThreadCount = nThreadCount;
  55. return TRUE;
  56. }
  58. //
  59. // 下载任务的线程函数
  60. //
  61. DWORD WINAPI ThreadProc_DownloadMTR(
  62.   LPVOID lpParameter   // thread data
  63. )
  64. {
  65. CDownloadMTR *pDownloadMTR = (CDownloadMTR*)lpParameter;
  66. ASSERT ( pDownloadMTR );
  67. pDownloadMTR->ThreadProc_DownloadMTR ();
  68. TRACE ( "下载任务的线程函数 执行完毕n" );
  69. return TRUE;
  70. }
  72. BOOL CDownloadMTR::ThreadProc_DownloadMTR()
  73. {
  74. // 启动多线程下载任务
  75. int nRet = StartMTRDownload ();
  76. if ( nRet == 2 ) return HandleDownloadFinished(ENUM_DOWNLOAD_RESULT_SUCCESS);
  77. if ( nRet == 0 ) return HandleDownloadFinished(ENUM_DOWNLOAD_RESULT_FAILED);
  79. // 等待所有线程下载完成
  80. ENUM_DOWNLOAD_RESULT eDownloadResult = WaitForDownloadFinished ();
  81. if ( eDownloadResult == ENUM_DOWNLOAD_RESULT_SUCCESS && !GetDownloadResult () )
  82. {
  83. eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  84. }
  86. return HandleDownloadFinished ( eDownloadResult );
  87. }
  89. //
  90. // 多线程断点续传下载一个文件
  91. //
  92. BOOL CDownloadMTR::Download (
  93. LPCTSTR lpszDownloadURL,
  94. LPCTSTR lpszSavePath,
  95. LPCTSTR lpszSaveOnlyFileName,
  96. LPCTSTR lpszUsername/*=NULL*/,
  97. LPCTSTR lpszPassword/*=NULL*/,
  98. BOOL bForceDownload/*=FALSE*/ // 如果为 TRUE 表示强制性重新下载，以下载的部分将会被删除，FALSE 表示断点续传
  99. )
  100. {
  101. if ( !HANDLE_IS_VALID(m_hEvtEndModule) )
  102. return FALSE;
  103. if ( !lpszSavePath || strlen(lpszSavePath) < 1 )
  104. return FALSE;
  105. m_csSavePath = lpszSavePath;
  106. m_csSaveOnlyFileName = GET_SAFE_STRING(lpszSaveOnlyFileName);
  107. m_bForceDownload = bForceDownload;
  109. CString csServer, csObject;
  110. USHORT nPort = 0;
  111. if ( !ParseURL ( lpszDownloadURL, csServer, csObject, nPort, m_csProtocolType ) )
  112. {
  113. Log ( L_ERROR, "Download URL [%s] invalid", lpszDownloadURL );
  114. return FALSE;
  115. }
  116. m_csDownloadURL = lpszDownloadURL;
  118. // 创建取站点信息对象
  119. if ( !( m_pDownloadPub_Info = CreateDownloadObject () ) )
  120. {
  121. Log ( L_ERROR, "Create download object failed" );
  122. return HandleDownloadFinished(ENUM_DOWNLOAD_RESULT_FAILED);
  123. }
  124. // 设置取站点信息对象的参数
  125. m_pDownloadPub_Info->SetAuthorization ( lpszUsername, lpszPassword );
  126. m_pDownloadPub_Info->m_pDownloadMTR = this;
  127. m_pDownloadPub_Info->SetDownloadUrl ( lpszDownloadURL );
  129. // 创建一个下载线程
  130. DWORD dwThreadId = 0;
  131. m_hThread = CreateThread ( NULL, 0, ::ThreadProc_DownloadMTR, LPVOID(this), 0, &dwThreadId );
  132. if ( !HANDLE_IS_VALID(m_hThread) )
  133. {
  134. Log ( L_WARNING, "Create download thread failed" );
  135. return FALSE;
  136. }
  138. return TRUE;
  139. }
  141. //
  142. // 创建下载对象
  143. //
  144. CDownloadPub* CDownloadMTR::CreateDownloadObject ( int nCount/*=1*/ )
  145. {
  146. if ( nCount < 1 ) return NULL;
  147. CDownloadPub *pDownloadPub = NULL;
  148. if ( m_csProtocolType.CompareNoCase ( "http" ) == 0 )
  149. {
  150. pDownloadPub = (CDownloadPub*)new CDownloadHttp[nCount];
  151. }
  152. else if ( m_csProtocolType.CompareNoCase ( "ftp" ) == 0 )
  153. {
  154. pDownloadPub = (CDownloadPub*)new CDownloadFtp[nCount];
  155. }
  156. else return NULL;
  158. return pDownloadPub;
  159. }
  161. //
  162. // 删除下载对象
  163. //
  164. void CDownloadMTR::DeleteDownloadObject ( CDownloadPub *pDownloadPub )
  165. {
  166. if ( m_csProtocolType.CompareNoCase ( "http" ) == 0 )
  167. {
  168. delete[] ( (CDownloadHttp*)pDownloadPub );
  169. }
  170. else if ( m_csProtocolType.CompareNoCase ( "ftp" ) == 0 )
  171. {
  172. delete[] ( (CDownloadFtp*)pDownloadPub );
  173. }
  174. else delete[] pDownloadPub;
  175. }
  177. void Callback_SaveDownloadInfo ( int nIndex, int nDownloadedSize, int nSimpleSaveSize, WPARAM wParam )
  178. {
  179. CDownloadMTR *pDownloadMTR = (CDownloadMTR*)wParam;
  180. ASSERT ( pDownloadMTR );
  181. pDownloadMTR->Callback_SaveDownloadInfo ( nIndex, nDownloadedSize, nSimpleSaveSize );
  182. }
  184. void CDownloadMTR::Callback_SaveDownloadInfo ( int nIndex, int nDownloadedSize, int nSimpleSaveSize )
  185. {
  186. if ( nIndex >= 0 && nIndex < m_nThreadCount )
  187. {
  188. m_pDownloadCellInfo[nIndex].nDownloadedSize = nDownloadedSize;
  189. if ( nDownloadedSize > 0 )
  190. {
  191. m_CSFor_DownloadedData.Lock();
  192. m_BaseDownInfo.nTotalDownloadedSize += nSimpleSaveSize;
  193. m_nTotalDownloadedSize_ThisTimes += nSimpleSaveSize;
  194. m_CSFor_DownloadedData.Unlock ();
  195. }
  196. }
  197. }
  199. //
  200. // 创建多线程下载使用的对象和数据缓冲
  201. //
  202. BOOL CDownloadMTR::CreateDownloadObjectAndDataMTR ()
  203. {
  204. DeleteDownloadObjectAndDataMTR ();
  206. ASSERT ( !m_pDownloadPub_MTR && m_pDownloadPub_Info );
  207. m_pDownloadPub_MTR = CreateDownloadObject ( m_nThreadCount );
  208. // 设置多线程下载使用的对象的参数
  209. if ( m_pDownloadPub_MTR )
  210. {
  211. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  212. {
  213. m_pDownloadPub_MTR[nIndex].m_nIndex = nIndex;
  214. m_pDownloadPub_MTR[nIndex].m_pDownloadMTR = this;
  215. m_pDownloadPub_MTR[nIndex].Set_SaveDownloadInfo_Callback ( ::Callback_SaveDownloadInfo, WPARAM(this) );
  216. m_pDownloadPub_MTR[nIndex].SetAuthorization ( m_pDownloadPub_Info->Get_UserName(), m_pDownloadPub_Info->Get_GetPassword() );
  217. m_pDownloadPub_MTR[nIndex].SetDownloadUrl ( m_pDownloadPub_Info->Get_DownloadUrl () );
  218. if ( !m_pDownloadPub_MTR[nIndex].SetSaveFileName ( GetTempFilePath() ) )
  219. return FALSE;
  220. }
  221. }
  223. // 创建多线程下载使用的数据缓冲
  224. ASSERT ( !m_pDownloadCellInfo );
  225. m_pDownloadCellInfo = new t_DownloadCellInfo[m_nThreadCount];
  226. if ( m_pDownloadCellInfo )
  227. memset ( m_pDownloadCellInfo, 0, m_nThreadCount*sizeof(t_DownloadCellInfo) );
  229. if ( m_pDownloadPub_MTR != NULL && m_pDownloadCellInfo != NULL )
  230. return TRUE;
  231. Log ( L_WARNING, "Create MTR download object or buffer failed" );
  232. return FALSE;
  234. }
  236. //
  237. // 删除多线程下载使用的对象和数据缓冲
  238. //
  239. void CDownloadMTR::DeleteDownloadObjectAndDataMTR()
  240. {
  241. if ( m_pDownloadPub_MTR )
  242. {
  243. DeleteDownloadObject ( m_pDownloadPub_MTR );
  244. m_pDownloadPub_MTR = NULL;
  245. }
  246. if ( m_pDownloadCellInfo )
  247. {
  248. delete[] m_pDownloadCellInfo;
  249. m_pDownloadCellInfo = NULL;
  250. }
  251. }
  253. //
  254. // 删除取站点信息的下载对象
  255. //
  256. void CDownloadMTR::DeleteDownloadObject_Info()
  257. {
  258. if ( m_pDownloadPub_Info )
  259. {
  260. DeleteDownloadObject ( m_pDownloadPub_Info );
  261. m_pDownloadPub_Info = NULL;
  262. }
  263. }
  265. //
  266. // 启动多线程下载，返回 0 表示失败，1表示成功，2表示不用下载了，因为该文件已经下载过了
  267. //
  268. int CDownloadMTR::StartMTRDownload ()
  269. {
  270. m_dwDownloadStartTime = GetTickCount();
  271. DownloadNotify ( -1, NOTIFY_TYPE_START_DOWNLOAD, (LPVOID)NULL, this );
  272. // 先获取站点信息
  273. ASSERT ( m_pDownloadPub_Info );
  274. if ( !m_pDownloadPub_Info->GetRemoteSiteInfo () )
  275. return 0;
  276. DbgLog ( "要下载的文件大小是：%d 字节n", m_pDownloadPub_Info->Get_FileTotalSize () );
  277. StandardSaveFileName ();
  279. // 要保存的文件是否已经存在，且大小和创建时间一致，如果不是强制性下载，则不需要再下载了。
  280. CFileStatus fileStatus;
  281. if ( CFile::GetStatus(m_csSavePathFileName,fileStatus) )
  282. {
  283. if (
  284. (
  285. fileStatus.m_mtime.GetTime() - m_pDownloadPub_Info->Get_TimeLastModified() <=2 &&
  286. m_pDownloadPub_Info->Get_TimeLastModified()-fileStatus.m_mtime.GetTime() <=2
  287. )
  288. &&
  289. fileStatus.m_size == m_pDownloadPub_Info->Get_FileTotalSize ()
  290. &&
  291. !m_bForceDownload
  292. )
  293. {
  294. return 2;
  295. }
  296. // 需要重新下载
  297. ::DeleteFile ( m_csSavePathFileName );
  298. ::DeleteFile ( GetTempFilePath() );
  299. }
  301. BOOL bMustCreateNullFile = TRUE;
  302. // 读取下载信息，如果能读到说明上次下载尚未完成
  303. if ( m_pDownloadPub_Info->Is_SupportResume() )
  304. {
  305. if ( CFile::GetStatus(GetTempFilePath(),fileStatus) &&
  306. fileStatus.m_size == m_pDownloadPub_Info->Get_FileTotalSize()+GetDownloadInfoWholeSize() )
  307. {
  308. if ( ReadDownloadInfo () )
  309. bMustCreateNullFile = FALSE;
  310. }
  311. }
  313. if ( bMustCreateNullFile )
  314. {
  315. int nFileSize = m_pDownloadPub_Info->Get_FileTotalSize();
  316. int nTempFileSize = nFileSize+GetDownloadInfoWholeSize();
  317. if ( nFileSize < 0 || !m_pDownloadPub_Info->Is_SupportResume() )
  318. nTempFileSize = 0;
  319. // 创建一个用来保存下载数据的空文件
  320. if ( !CreateNullFile ( GetTempFilePath(), nTempFileSize ) )
  321. return FALSE;
  322. }
  324. // 分配下载任务
  325. if ( !AssignDownloadTask () )
  326. {
  327. Log ( L_WARNING, "Assign task failed" );
  328. return 0;
  329. }
  331. m_dwDownloadStartTime = GetTickCount();
  332. return 1;
  333. }
  335. //
  336. // 得到临时数据保存的路径文件名
  337. //
  338. CString CDownloadMTR::GetTempFilePath ()
  339. {
  340. ASSERT ( !m_csSavePathFileName.IsEmpty () );
  341. CString csTempFileName;
  342. csTempFileName.Format ( "%s.~xhw~", m_csSavePathFileName );
  343. ::SetFileAttributes ( csTempFileName, FILE_ATTRIBUTE_HIDDEN );
  344. return csTempFileName;
  345. }
  347. //
  348. // 分配下载任务
  349. //
  350. BOOL CDownloadMTR::AssignDownloadTask()
  351. {
  352. ASSERT ( m_pDownloadPub_Info );
  353. if ( !m_pDownloadPub_Info->Is_SupportResume() )
  354. {
  355. DeleteDownloadObjectAndDataMTR ();
  356. Log ( L_WARNING, "Site [%s] not support resume download", m_pDownloadPub_Info->Get_ServerName() );
  357. }
  358. // 文件大小未知，采用单线程
  359. if ( m_pDownloadPub_Info->Get_FileTotalSize () <= 0 || !m_pDownloadPub_Info->Is_SupportResume() )
  360. {
  361. if ( m_nThreadCount != 1 )
  362. {
  363. DeleteDownloadObjectAndDataMTR ();
  364. SetThreadCount ( 1 );
  365. }
  366. }
  368. if ( !DownloadInfoIsValid() || !m_pDownloadPub_MTR || !m_pDownloadCellInfo )
  369. {
  370. if ( !CreateDownloadObjectAndDataMTR () )
  371. return FALSE;
  372. }
  374. ASSERT ( m_pDownloadPub_MTR && m_pDownloadCellInfo );
  376. // 下载任务尚未分配
  377. if ( !DownloadInfoIsValid() )
  378. {
  379. int nWillDownloadSize = -1, nWillDownloadStartPos = 0, nNoAssignSize = 0;
  380. if ( m_pDownloadPub_Info->Get_FileTotalSize () > 0 )
  381. {
  382. nWillDownloadSize = m_pDownloadPub_Info->Get_FileTotalSize () / m_nThreadCount;
  383. // 均分后剩下的部分，让第一个线程来承担下载
  384. nNoAssignSize = m_pDownloadPub_Info->Get_FileTotalSize () % m_nThreadCount;
  385. }
  387. DbgLog ( "任务分配如下：--------------------n" );
  388. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  389. {
  390. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos = nWillDownloadStartPos;
  391. m_pDownloadCellInfo[nIndex].nWillDownloadSize = nWillDownloadSize;
  392. if ( nIndex == 0 && m_pDownloadPub_Info->Get_FileTotalSize () > 0 )
  393. {
  394. m_pDownloadCellInfo[nIndex].nWillDownloadSize += nNoAssignSize;
  395. }
  397. DbgLog ( "线程.%d 从 %d(0x%08x) 下载到 %d(0x%08x) 共 %d(0x%08x) 字节n", nIndex, 
  398. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos, m_pDownloadCellInfo[nIndex].nWillDownloadStartPos,
  399. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos+m_pDownloadCellInfo[nIndex].nWillDownloadSize,
  400. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos+m_pDownloadCellInfo[nIndex].nWillDownloadSize,
  401. m_pDownloadCellInfo[nIndex].nWillDownloadSize, m_pDownloadCellInfo[nIndex].nWillDownloadSize );
  403. nWillDownloadStartPos += m_pDownloadCellInfo[nIndex].nWillDownloadSize;
  404. }
  405. }
  408. // 启动下载任务
  409. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  410. {
  411. if ( !m_pDownloadPub_MTR[nIndex].Download ( m_pDownloadCellInfo[nIndex].nWillDownloadStartPos,
  412. m_pDownloadCellInfo[nIndex].nWillDownloadSize, m_pDownloadCellInfo[nIndex].nDownloadedSize ) )
  413. return FALSE;
  414. }
  416. m_BaseDownInfo.dwThreadCount = m_nThreadCount;
  417. return TRUE;
  418. }
  420. //
  421. // 从下载信息文件中读取下载信息
  422. //
  423. BOOL CDownloadMTR::ReadDownloadInfo()
  424. {
  425. CString csTempFileName = GetTempFilePath ();
  426. BOOL bRet = FALSE;
  427. CFile file;
  428. TRY
  429. {
  430. if ( file.Open ( csTempFileName, CFile::modeCreate|CFile::modeNoTruncate|CFile::modeReadWrite|CFile::typeBinary|CFile::shareDenyNone ) )
  431. {
  432. if ( file.Seek ( -(int)sizeof(t_BaseDownInfo), CFile::end ) == (int)(file.GetLength() - sizeof(t_BaseDownInfo)) )
  433. {
  434. if ( (UINT)file.Read ( &m_BaseDownInfo, sizeof(t_BaseDownInfo) ) == sizeof(t_BaseDownInfo) )
  435. {
  436. if ( (m_BaseDownInfo.dwThreadCount > 0 && m_BaseDownInfo.dwThreadCount <= MAX_DOWNLOAD_THREAD_COUNT)&&
  437. SetThreadCount ( m_BaseDownInfo.dwThreadCount ) )
  438. {
  439. if ( CreateDownloadObjectAndDataMTR () )
  440. {
  441. if ( file.Seek ( -GetDownloadInfoWholeSize(), CFile::end ) == int(file.GetLength() - GetDownloadInfoWholeSize()) )
  442. {
  443. if ( file.Read ( m_pDownloadCellInfo, sizeof(t_DownloadCellInfo)*m_nThreadCount ) == sizeof(t_DownloadCellInfo)*m_nThreadCount )
  444. {
  445. bRet = TRUE;
  446. }
  447. else
  448. {
  449. memset ( m_pDownloadCellInfo, 0, sizeof(t_DownloadCellInfo)*m_nThreadCount );
  450. }
  451. }
  452. }
  453. }
  454. }
  455. }
  456. }
  457. }
  458. CATCH( CFileException, e )
  459. {
  460. e->Delete ();
  461. bRet = FALSE;
  462. }
  463. END_CATCH
  465. if ( HANDLE_IS_VALID(file.m_hFile) )
  466. file.Close ();
  468. return bRet;
  469. }
  471. BOOL CDownloadMTR::SaveDownloadInfo ()
  472. {
  473. if ( !m_pDownloadPub_Info->Is_SupportResume() )
  474. return TRUE;
  475. CString csTempFileName = GetTempFilePath ();
  476. BOOL bRet = FALSE;
  477. CFile file;
  478. TRY
  479. {
  480. if ( file.Open ( csTempFileName, CFile::modeCreate|CFile::modeNoTruncate|CFile::modeReadWrite|CFile::typeBinary|CFile::shareDenyNone ) )
  481. {
  482. if ( file.Seek ( -(int)sizeof(t_BaseDownInfo), CFile::end ) == (int)(file.GetLength() - sizeof(t_BaseDownInfo)) )
  483. {
  484. file.Write ( &m_BaseDownInfo, sizeof(t_BaseDownInfo) );
  485. if ( file.Seek ( -GetDownloadInfoWholeSize(), CFile::end ) == int(file.GetLength() - GetDownloadInfoWholeSize()) )
  486. {
  487. file.Write ( m_pDownloadCellInfo, m_nThreadCount*sizeof(t_DownloadCellInfo) );
  488. bRet = TRUE;
  489. }
  490. }
  491. }
  492. }
  493. CATCH( CFileException, e )
  494. {
  495. e->Delete ();
  496. bRet = FALSE;
  497. }
  498. END_CATCH
  499. if ( HANDLE_IS_VALID ( file.m_hFile ) )
  500. file.Close ();
  502. if ( !bRet ) Log ( L_WARNING, "Save download info failed. %s", hwFormatMessage ( GetLastError() ) );
  504. return bRet;
  506. }
  508. BOOL CDownloadMTR::HandleDownloadFinished(ENUM_DOWNLOAD_RESULT eDownloadResult)
  509. {
  510. CString csTempFileName;
  511. CFileStatus fileStatus;
  512. BOOL bRet = FALSE;
  513. CFile file;
  515. if ( eDownloadResult != ENUM_DOWNLOAD_RESULT_SUCCESS )
  516. {
  517. SaveDownloadInfo ();
  518. goto Finished;
  519. }
  520. csTempFileName = GetTempFilePath ();
  522. // 设置文件大小
  523. if ( m_pDownloadPub_Info->Is_SupportResume() && m_pDownloadPub_Info->Get_FileTotalSize() > 0 )
  524. {
  525. TRY
  526. {
  527. file.Open ( csTempFileName, CFile::modeCreate|CFile::modeNoTruncate|CFile::modeReadWrite|CFile::typeBinary|CFile::shareDenyNone );
  528. file.SetLength(m_pDownloadPub_Info->Get_FileTotalSize ());
  529. bRet = TRUE;
  530. }
  531. CATCH( CFileException, e )
  532. {
  533. e->Delete ();
  534. bRet = FALSE;
  535. }
  536. END_CATCH
  537. if ( HANDLE_IS_VALID(file.m_hFile) )
  538. file.Close ();
  539. if ( !bRet )
  540. {
  541. Log ( L_WARNING, "Set [%s] length failed", csTempFileName );
  542. eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  543. goto Finished;
  544. }
  545. }
  547. if ( _access(csTempFileName,04) == 0 )
  548. {
  549. // 将文件改名
  550. bRet = FALSE;
  551. DeleteFile ( m_csSavePathFileName );
  552. TRY
  553. {
  554. CFile::Rename ( csTempFileName, m_csSavePathFileName );
  555. bRet = TRUE;
  556. }
  557. CATCH( CFileException, e )
  558. {
  559. e->Delete ();
  560. bRet = FALSE;
  561. }
  562. END_CATCH
  564. if ( !bRet )
  565. {
  566. Log ( L_WARNING, "Rename [%s] failed. %s", csTempFileName, hwFormatMessage(GetLastError()) );
  567. eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  568. goto Finished;
  569. }
  571. // 设置文件属性，时间设置和服务器一致
  572. bRet = FALSE;
  573. if ( CFile::GetStatus(m_csSavePathFileName,fileStatus) )
  574. {
  575. fileStatus.m_mtime = m_pDownloadPub_Info->Get_TimeLastModified();
  576. fileStatus.m_attribute = CFile::normal;
  577. CFile::SetStatus ( m_csSavePathFileName, fileStatus );
  578. bRet = TRUE;
  579. }
  580. if ( !bRet )
  581. {
  582. Log ( L_WARNING, "Set file [%s] status failed. %s", csTempFileName, hwFormatMessage(GetLastError()) );
  583. eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  584. goto Finished;
  585. }
  586. }
  588. Finished:
  589. DownloadNotify ( -1, NOTIFY_TYPE_END_DOWNLOAD, (LPVOID)eDownloadResult, this );
  590. return bRet;
  591. }
  593. BOOL CDownloadMTR::GetDownloadResult()
  594. {
  595. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  596. {
  597. if ( !m_pDownloadPub_MTR[nIndex].Is_DownloadSuccess() )
  598. return FALSE;
  599. }
  601. return TRUE;
  602. }
  604. //
  605. // 下载信息是否有效
  606. //
  607. BOOL CDownloadMTR::DownloadInfoIsValid()
  608. {
  609. BOOL bValid = FALSE;
  610. int nIndex = 0;
  611. if ( !m_pDownloadCellInfo ) goto Invalid;
  612. if ( m_BaseDownInfo.dwThreadCount < 1 || m_BaseDownInfo.dwThreadCount > MAX_DOWNLOAD_THREAD_COUNT )
  613. goto Invalid;
  615. for ( nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  616. {
  617. if ( m_pDownloadCellInfo[nIndex].nWillDownloadSize > 0 )
  618. {
  619. bValid = TRUE;
  620. break;
  621. }
  622. }
  623. if ( !bValid ) goto Invalid;
  625. return TRUE;
  627. Invalid:
  628. if ( m_pDownloadCellInfo )
  629. memset ( m_pDownloadCellInfo, 0, m_nThreadCount*sizeof(t_DownloadCellInfo) );
  630. memset ( &m_BaseDownInfo, 0, sizeof(t_BaseDownInfo) );
  631. return FALSE;
  632. }
  634. //
  635. // 找到剩余未下载的数量最大的那个对象编号
  636. //
  637. int CDownloadMTR::GetUndownloadMaxBytes( int &nUndownloadBytes )
  638. {
  639. nUndownloadBytes = 0;
  640. int nMaxIndex = -1;
  641. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  642. {
  643. int nTempBytes = m_pDownloadPub_MTR[nIndex].GetUndownloadBytes ();
  644. if ( nUndownloadBytes < nTempBytes )
  645. {
  646. nUndownloadBytes = nTempBytes;
  647. nMaxIndex = nIndex;
  648. }
  649. }
  650. return nMaxIndex;
  651. }
  653. //
  654. // 编号为 nIndex 的对象调度任务，为下载任务最繁重的对象减轻负担
  655. //
  656. BOOL CDownloadMTR::AttemperDownloadTask(int nIndex)
  657. {
  658. ASSERT ( m_pDownloadPub_MTR && m_pDownloadCellInfo );
  659. if ( m_nThreadCount <= 1 || m_pDownloadCellInfo[nIndex].nWillDownloadSize == -1 )
  660. return FALSE;
  661. int nUndownloadBytes = 0;
  662. int nIndex_Heavy = GetUndownloadMaxBytes ( nUndownloadBytes );
  663. if ( nIndex_Heavy == -1 || nIndex_Heavy == nIndex || nUndownloadBytes < 10*1024 )
  664. return FALSE;
  665. ASSERT ( nIndex_Heavy >= 0 && nIndex_Heavy < m_nThreadCount );
  666. ASSERT ( m_pDownloadPub_MTR[nIndex_Heavy].Get_WillDownloadStartPos() == m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadStartPos );
  668. DbgLog ( "下载对象.%d 帮 %d 减轻负担n", nIndex, nIndex_Heavy );
  669. // 给空闲下载对象分配新任务
  670. m_pDownloadCellInfo[nIndex].nWillDownloadSize = nUndownloadBytes / 2;
  671. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos = m_pDownloadPub_MTR[nIndex_Heavy].Get_WillDownloadStartPos() +
  672. m_pDownloadPub_MTR[nIndex_Heavy].Get_WillDownloadSize() - m_pDownloadCellInfo[nIndex].nWillDownloadSize;
  673. m_pDownloadCellInfo[nIndex].nDownloadedSize = 0;
  674. DbgLog ( "空闲下载对象.%d 分配新任务： %d(0x%08x) - %d(0x%08x) 共 %d(0x%08x)n", nIndex, m_pDownloadCellInfo[nIndex].nWillDownloadStartPos, m_pDownloadCellInfo[nIndex].nWillDownloadStartPos,
  675. m_pDownloadCellInfo[nIndex].nWillDownloadStartPos + m_pDownloadCellInfo[nIndex].nWillDownloadSize, m_pDownloadCellInfo[nIndex].nWillDownloadStartPos + m_pDownloadCellInfo[nIndex].nWillDownloadSize,
  676. m_pDownloadCellInfo[nIndex].nWillDownloadSize, m_pDownloadCellInfo[nIndex].nWillDownloadSize );
  678. // 减轻繁忙下载对象的任务
  679. m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize -= m_pDownloadCellInfo[nIndex].nWillDownloadSize;
  680. m_pDownloadPub_MTR[nIndex_Heavy].Set_WillDownloadSize ( m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize );
  681. DbgLog ( "繁忙下载对象.%d 下载了 %d(0x%08x) 未完 %d(0x%08x) 字节，调整任务为： %d(0x%08x) - %d(0x%08x) 共 %d(0x%08x)n",
  682. nIndex_Heavy, m_pDownloadPub_MTR[nIndex_Heavy].Get_DownloadedSize(), m_pDownloadPub_MTR[nIndex_Heavy].Get_DownloadedSize(),
  683. nUndownloadBytes, nUndownloadBytes,
  684. m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadStartPos, m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadStartPos,
  685. m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadStartPos + m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize, m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadStartPos + m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize,
  686. m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize, m_pDownloadCellInfo[nIndex_Heavy].nWillDownloadSize );
  689. // 启动空闲下载对象的下载任务
  690. m_pDownloadPub_MTR[nIndex].ResetVar ();
  691. if ( !m_pDownloadPub_MTR[nIndex].Download ( m_pDownloadCellInfo[nIndex].nWillDownloadStartPos,
  692. m_pDownloadCellInfo[nIndex].nWillDownloadSize, m_pDownloadCellInfo[nIndex].nDownloadedSize ) )
  693. return FALSE;
  695. return TRUE;
  696. }
  698. //
  699. // 等待下载结束
  700. //
  701. ENUM_DOWNLOAD_RESULT CDownloadMTR::WaitForDownloadFinished()
  702. {
  703. ASSERT ( HANDLE_IS_VALID(m_hEvtEndModule) );
  704. int nCount = m_nThreadCount + 1;
  705. ENUM_DOWNLOAD_RESULT eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  706. HANDLE *lpHandles = new HANDLE[nCount];
  707. if ( !lpHandles ) goto End;
  708. while ( TRUE )
  709. {
  710. nCount = 0;
  711. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  712. {
  713. HANDLE hThread = m_pDownloadPub_MTR[nIndex].Get_Thread_Handle ();
  714. if ( HANDLE_IS_VALID(hThread) )
  715. lpHandles[nCount++] = hThread;
  716. }
  717. lpHandles[nCount++] = m_hEvtEndModule;
  719. if ( nCount == 1 )
  720. {
  721. if ( m_BaseDownInfo.nTotalDownloadedSize >= m_pDownloadPub_Info->Get_FileTotalSize() )
  722. eDownloadResult = ENUM_DOWNLOAD_RESULT_SUCCESS;
  723. else
  724. eDownloadResult = ENUM_DOWNLOAD_RESULT_FAILED;
  725. goto End;
  726. }
  727. int nRet = (int)WaitForMultipleObjects ( nCount, lpHandles, FALSE, INFINITE ) - WAIT_OBJECT_0;
  728. // 某下载对象完成任务了
  729. if ( nRet >= 0 && nRet < nCount-1 )
  730. {
  731. nIndex = FindIndexByThreadHandle ( lpHandles[nRet] );
  732. if ( ( nIndex >= 0 && nIndex < m_nThreadCount ) )
  733. {
  734. if ( !m_pDownloadPub_MTR[nIndex].Is_DownloadSuccess() ||
  735. !AttemperDownloadTask ( nIndex ) )
  736. {
  737. m_pDownloadPub_MTR[nIndex].Clear_Thread_Handle ();
  738. }
  739. }
  740. else
  741. {
  742. eDownloadResult = ENUM_DOWNLOAD_RESULT_CANCEL;
  743. goto End;
  744. }
  745. }
  746. // 模块结束
  747. else
  748. {
  749. eDownloadResult = ENUM_DOWNLOAD_RESULT_CANCEL;
  750. goto End;
  751. }
  752. }
  754. End:
  755. // 等待所有下载线程结束
  756. if ( eDownloadResult != ENUM_DOWNLOAD_RESULT_SUCCESS )
  757. {
  758. nCount = 0;
  759. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  760. {
  761. HANDLE hThread = m_pDownloadPub_MTR[nIndex].Get_Thread_Handle ();
  762. if ( HANDLE_IS_VALID(hThread) )
  763. lpHandles[nCount++] = hThread;
  764. }
  765. WaitForMultipleObjects ( nCount, lpHandles, TRUE, 500*1000 );
  766. }
  767. if ( lpHandles ) delete[] lpHandles;
  768. return eDownloadResult;
  769. }
  771. int CDownloadMTR::FindIndexByThreadHandle(HANDLE hThread)
  772. {
  773. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  774. {
  775. HANDLE hThread_Temp = m_pDownloadPub_MTR[nIndex].Get_Thread_Handle ();
  776. if ( HANDLE_IS_VALID(hThread_Temp) && hThread_Temp == hThread )
  777. return nIndex;
  778. }
  779. return -1;
  780. }
  782. int CDownloadMTR::GetDownloadInfoWholeSize()
  783. {
  784. return ( sizeof(t_DownloadCellInfo)*m_nThreadCount + sizeof(t_BaseDownInfo) );
  785. }
  787. //
  788. // 获取下载所消耗的时间（毫秒），可用来计算下载速度和推算剩余时间
  789. //
  790. DWORD CDownloadMTR::GetDownloadElapsedTime()
  791. {
  792. return (GetTickCount() - m_dwDownloadStartTime);
  793. }
  795. //
  796. // 停止下载。将所有下载线程关闭，将下载对象删除，文件关闭
  797. //
  798. void CDownloadMTR::StopDownload()
  799. {
  800. if ( HANDLE_IS_VALID(m_hEvtEndModule) )
  801. {
  802. ::SetEvent ( m_hEvtEndModule );
  803. }
  805. // 设置多线程下载使用的对象的参数
  806. if ( m_pDownloadPub_MTR )
  807. {
  808. for ( int nIndex=0; nIndex<m_nThreadCount; nIndex++ )
  809. {
  810. m_pDownloadPub_MTR[nIndex].StopDownload ();
  811. }
  812. }
  814. if ( HANDLE_IS_VALID(m_hThread) )
  815. {
  816. WaitForThreadEnd ( m_hThread,30*1000 );
  817. CLOSE_HANDLE ( m_hThread )
  818. }
  820. DeleteDownloadObjectAndDataMTR ();
  821. DeleteDownloadObject_Info ();
  823. CLOSE_HANDLE ( m_hEvtEndModule );
  824. }
  826. void CDownloadMTR::StandardSaveFileName ()
  827. {
  828. ASSERT ( m_csSavePath.GetLength() > 0 );
  829. StandardizationPathBuffer ( m_csSavePath.GetBuffer(MAX_PATH), MAX_PATH );
  830. m_csSavePath.ReleaseBuffer ();
  831. MakeSureDirectory ( m_csSavePath );
  833. char szOnlyFileName_NoExt_User[MAX_PATH] = {0};
  834. char szExtensionName_User[MAX_PATH] = {0};
  835. // 如果用户指定了新的保存文件名，就用新的。
  836. if ( m_csSaveOnlyFileName.GetLength() > 0 )
  837. {
  838. CString csFileNameByURL = GetLocalFileNameByURL ( m_csDownloadURL );
  839. if ( csFileNameByURL.CompareNoCase(m_csSaveOnlyFileName) != 0 )
  840. {
  841. PartFileAndExtensionName ( m_csSaveOnlyFileName, szOnlyFileName_NoExt_User, MAX_PATH, szExtensionName_User, MAX_PATH );
  842. }
  843. }
  845. CString csExtensionName_Remote;
  846. CString csFileName_Remote = m_pDownloadPub_Info->GetDownloadObjectFileName ( &csExtensionName_Remote );
  848. if ( strlen(szOnlyFileName_NoExt_User) > 0 )
  849. {
  850. if ( strlen(szExtensionName_User) < 1 )
  851. STRNCPY_CS ( szExtensionName_User, csExtensionName_Remote );
  852. m_csSavePathFileName.Format ( "%s%s.%s", StandardizationFileForPathName(m_csSavePath,FALSE),
  853. StandardizationFileForPathName(szOnlyFileName_NoExt_User,TRUE), StandardizationFileForPathName(szExtensionName_User,TRUE) );
  854. }
  855. else
  856. {
  857. m_csSavePathFileName.Format ( "%s%s", StandardizationFileForPathName(m_csSavePath,FALSE), StandardizationFileForPathName(csFileName_Remote,TRUE) );
  858. }
  859. }
  861. //
  862. // 根据 URL 来获取本地保存的文件名
  863. //
  864. CString CDownloadMTR::GetLocalFileNameByURL ( LPCTSTR lpszDownloadURL )
  865. {
  866. if ( !lpszDownloadURL || strlen(lpszDownloadURL) < 1 )
  867. return "";
  868. char szOnlyPath[MAX_PATH] = {0};
  869. char szOnlyFileName[MAX_PATH] = {0};
  870. if ( !PartFileAndPathByFullPath ( lpszDownloadURL, szOnlyFileName, MAX_PATH, szOnlyPath, MAX_PATH ) )
  871. return "";
  872. return szOnlyFileName;
  873. }
  875. //
  876. // 获取文件大小
  877. //
  878. int CDownloadMTR::Get_FileTotaleSize()
  879. {
  880. if ( !m_pDownloadPub_Info ) return -1;
  881. return m_pDownloadPub_Info->Get_FileTotalSize ();
  882. }
  884. //
  885. // 获取已下载的字节数
  886. //
  887. int CDownloadMTR::Get_TotalDownloadedSize()
  888. {
  889. m_CSFor_DownloadedData.Lock ();
  890. int nTotalDownloadedSize = m_BaseDownInfo.nTotalDownloadedSize;
  891. m_CSFor_DownloadedData.Unlock ();
  892. return nTotalDownloadedSize;
  893. }