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

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

进程与线程

开发平台：

Visual C++

DownloadPub.cpp：源码内容
  1. // DownloadPub.cpp: implementation of the CDownloadPub class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "NetDownMTR.h"
  7. #include "DownloadPub.h"
  8. // for PathFindExtension () / PathFindFileName ()
  9. #include "Shlwapi.h"
  10. #pragma comment ( lib, "shlwapi.lib" )
  12. #ifdef _DEBUG
  13. #undef THIS_FILE
  14. static char THIS_FILE[]=__FILE__;
  15. #define new DEBUG_NEW
  16. #endif
  18. // 下载数据保存的临时缓冲大小
  19. #define TEMP_SAVE_BUFFER_SIZE (10*NET_BUFFER_SIZE)
  20. // 当下载的数据达到这个数的时候才保存到文件中
  21. #define WRITE_TEMP_SAVE_MIN_BYTES (TEMP_SAVE_BUFFER_SIZE / 2)
  23. void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR );
  25. //////////////////////////////////////////////////////////////////////
  26. // Construction/Destruction
  27. //////////////////////////////////////////////////////////////////////
  29. CDownloadPub::CDownloadPub()
  30. {
  31. m_hThread = NULL;
  32. m_TimeLastModified = -1;
  33. m_csDownloadUrl = _T("");
  34. m_csSaveFileName = _T("");
  36. m_Proc_SaveDownloadInfo = NULL;
  37. m_wSaveDownloadInfo_Param = NULL;
  39. m_bSupportResume = FALSE;
  40. m_nFileTotalSize = -1;
  41. m_csReferer = _T("");
  42. m_csUserAgent = _T("XieHongWei-HttpDown/2.0");
  43. m_csUsername = _T("");
  44. m_csPassword = _T("");
  45. m_csProtocolType = "http";
  46. m_csServer = _T("");
  47. m_csObject = _T("");
  48. m_csFileName = _T("");
  49. m_nPort = DEFAULT_HTTP_PORT ;
  50. m_nIndex = -1;
  51. ResetVar ();
  52. m_hEvtEndModule = ::CreateEvent ( NULL, TRUE, FALSE, NULL );
  53. m_pDownloadMTR = NULL;
  54. }
  56. void CDownloadPub::ResetVar()
  57. {
  58. Clear_Thread_Handle ();
  59. m_nWillDownloadStartPos = 0;
  60. m_nWillDownloadSize = -1;
  61. m_nDownloadedSize = 0;
  62. m_nTempSaveBytes = 0;
  63. m_bDownloadSuccess = FALSE;
  64. }
  66. CDownloadPub::~CDownloadPub()
  67. {
  68. StopDownload ();
  69. Clear_Thread_Handle ();
  70. }
  72. void CDownloadPub::StopDownload ()
  73. {
  74. if ( HANDLE_IS_VALID(m_hEvtEndModule) )
  75. {
  76. ::SetEvent ( m_hEvtEndModule );
  77. WaitForThreadEnd ( m_hThread,30*1000 );
  78. CLOSE_HANDLE ( m_hEvtEndModule );
  79. m_hEvtEndModule = NULL;
  80. Clear_Thread_Handle ();
  81. }
  82. }
  84. //
  85. // 设置认证信息
  86. //
  87. void CDownloadPub::SetAuthorization ( LPCTSTR lpszUsername, LPCTSTR lpszPassword )
  88. {
  89. if( lpszUsername != NULL )
  90. {
  91. m_csUsername  = GET_SAFE_STRING(lpszUsername);
  92. m_csPassword  = GET_SAFE_STRING(lpszPassword);
  93. }
  94. else
  95. {
  96. m_csUsername  = _T("");
  97. m_csPassword  = _T("");
  98. }
  99. }
  101. // 设置Referer
  102. void CDownloadPub::SetReferer(LPCTSTR lpszReferer)
  103. {
  104. if( lpszReferer != NULL )
  105. m_csReferer = lpszReferer;
  106. else
  107. m_csReferer = _T("");
  108. }
  110. // 设置UserAgent
  111. void CDownloadPub::SetUserAgent(LPCTSTR lpszUserAgent)
  112. {
  113. m_csUserAgent = lpszUserAgent;
  114. if( m_csUserAgent.IsEmpty())
  115. m_csUserAgent = _T("XieHongWei-HttpDown/2.0");
  116. }
  118. //
  119. // 设置保存下载信息回调函数
  120. //
  121. void CDownloadPub::Set_SaveDownloadInfo_Callback ( FUNC_SaveDownloadInfo Proc_SaveDownloadInfo, WPARAM wParam )
  122. {
  123. m_Proc_SaveDownloadInfo = Proc_SaveDownloadInfo;
  124. m_wSaveDownloadInfo_Param = wParam;
  125. }
  127. //
  128. // 下载任务的线程函数
  129. //
  130. DWORD WINAPI ThreadProc_Download(
  131.   LPVOID lpParameter   // thread data
  132. )
  133. {
  134. CDownloadPub *pDownloadPub = (CDownloadPub*)lpParameter;
  135. ASSERT ( pDownloadPub );
  136. return pDownloadPub->ThreadProc_Download ();
  137. }
  139. BOOL CDownloadPub::ThreadProc_Download()
  140. {
  141. for ( int i=0; i<RETRY_TIMES; i++ )
  142. {
  143. if ( DownloadOnce () )
  144. return DownloadEnd(TRUE);
  145. SLEEP_RETURN_Down ( 5*1000 );
  146. }
  148. return DownloadEnd(FALSE);
  149. }
  151. BOOL CDownloadPub::DownloadOnce()
  152. {
  153. // 打开文件
  154. if ( !OpenFileForSave () )
  155. return FALSE;
  157. // 连接到服务器
  158. if ( !m_SocketClient.Is_Connected () )
  159. {
  160. if ( !Connect () )
  161. return FALSE;
  162. }
  164. return TRUE;
  165. }
  167. //
  168. // 创建线程下载文件
  169. //
  170. BOOL CDownloadPub::Download (
  171. int nWillDownloadStartPos, // 要下载文件的开始位置
  172. int nWillDownloadSize, // 本次需要下载的大小，-1表示一直下载到文件尾
  173. int nDownloadedSize // 已下载的字节数，指完全写到文件中的字节数
  174. )
  175. {
  176. // 设置下载参数
  177. m_nWillDownloadStartPos = nWillDownloadStartPos;
  178. Set_WillDownloadSize ( nWillDownloadSize );
  179. if ( m_nFileTotalSize > 0 && Get_WillDownloadSize() > m_nFileTotalSize )
  180. {
  181. Set_WillDownloadSize ( m_nFileTotalSize );
  182. }
  183. Set_DownloadedSize ( nDownloadedSize );
  185. // 创建一个下载线程
  186. DWORD dwThreadId = 0;
  187. m_hThread = CreateThread ( NULL, 0, ::ThreadProc_Download, LPVOID(this), 0, &dwThreadId );
  188. if ( !HANDLE_IS_VALID(m_hThread) )
  189. {
  190. Log ( L_WARNING, "Create download thread failed" );
  191. return FALSE;
  192. }
  193. return TRUE;
  194. }
  196. //
  197. // 下载结束
  198. //
  199. BOOL CDownloadPub::DownloadEnd(BOOL bRes)
  200. {
  201. m_bDownloadSuccess = bRes;
  202. m_SocketClient.Disconnect ();
  203. if ( HANDLE_IS_VALID ( m_file.m_hFile ) )
  204. {
  205. m_file.Close ();
  206. }
  208. TRACE ( "=============== 线程。%d 下载 %s 结束n", m_nIndex, bRes?"成功":"!!! 失败 !!!" );
  210. return bRes;
  211. }
  213. BOOL CDownloadPub::Connect()
  214. {
  215. if ( !HANDLE_IS_VALID(m_hEvtEndModule) )
  216. return FALSE;
  218. if ( m_csServer.IsEmpty() )
  219. {
  220. Log ( L_WARNING, "Please set download URL" );
  221. return FALSE;
  222. }
  223. m_SocketClient.SetEventOfEndModule ( m_hEvtEndModule );
  224. // 连接到服务器
  225. if ( !m_SocketClient.Connect ( m_csServer, m_nPort ) )
  226. return FALSE;
  227. TRACE ( "对象.%d 连接到服务器 成功n", m_nIndex );
  229. return TRUE;
  230. }
  232. BOOL CDownloadPub::SetDownloadUrl(LPCTSTR lpszDownloadUrl)
  233. {
  234. if ( !lpszDownloadUrl ) return FALSE;
  235. m_csDownloadUrl = lpszDownloadUrl;
  236. // 检验要下载的URL是否为空
  237. m_csDownloadUrl.TrimLeft();
  238. m_csDownloadUrl.TrimRight();
  239. if( m_csDownloadUrl.IsEmpty() )
  240. return FALSE;
  241. // 检验要下载的URL是否有效
  242. if ( !ParseURL(m_csDownloadUrl, m_csServer, m_csObject, m_nPort, m_csProtocolType))
  243. {
  244. TRACE(_T("Failed to parse the URL: %sn"), m_csDownloadUrl);
  245. return FALSE;
  246. }
  248. return TRUE;
  249. }
  251. //
  252. // 从服务器接收数据并保存到文件中
  253. //
  254. BOOL CDownloadPub::RecvDataAndSaveToFile(CSocketClient &SocketClient,char *szTailData/*=NULL*/, int nTailSize/*=0*/)
  255. {
  256. int nDownloadedSize = Get_DownloadedSize();
  257. if ( szTailData && nTailSize > 0 )
  258. {
  259. nDownloadedSize = SaveDataToFile ( szTailData, nTailSize );
  260. if ( nDownloadedSize < 0 )
  261. {
  262. return FALSE;
  263. }
  264. }
  266. char szRecvBuf[NET_BUFFER_SIZE] = {0}, *szTempSaveBuf = new char[TEMP_SAVE_BUFFER_SIZE];
  267. if ( !szTempSaveBuf )
  268. {
  269. Log ( L_ERROR, "Allocate memory failed" );
  270. return FALSE;
  271. }
  272. m_nTempSaveBytes = 0;
  274. while ( TRUE )
  275. {
  276. BOOL bDownloadFinished = FALSE;
  277. int nReadSize = 0;
  278. int nTempSaveBytes = Get_TempSaveBytes ();
  279. int nRecvTotalBytes = nDownloadedSize+nTempSaveBytes; // 保存在文件中的字节数加上临时缓冲中的字节数就是总共接收到字节数
  280. int nWillDownloadSize = Get_WillDownloadSize ();
  281. // 从字节数判断，本次下载已经完成了
  282. if ( nWillDownloadSize > 0 && nRecvTotalBytes >= nWillDownloadSize )
  283. {
  284. bDownloadFinished = TRUE;
  285. }
  286. else
  287. {
  288. int nRecvBytesThisTimes = sizeof(szRecvBuf);
  289. if ( nWillDownloadSize > 0 )
  290. nRecvBytesThisTimes = nWillDownloadSize - nRecvTotalBytes;
  291. ASSERT ( nRecvBytesThisTimes >= 0 );
  292. nRecvBytesThisTimes = MIN ( nRecvBytesThisTimes, sizeof(szRecvBuf) );
  293. nReadSize = SocketClient.Receive ( szRecvBuf, nRecvBytesThisTimes );
  294. // 读不到数据了，所以认为下载已经完成
  295. if ( nReadSize <= 0 )
  296. {
  297. if ( nWillDownloadSize <= 0 )
  298. {
  299. bDownloadFinished = TRUE;
  300. }
  301. }
  302. else
  303. {
  304. // TRACE ( "对象.%d, 收到 %d 字节，我的任务是 %d (0x%08x)字节n",
  305. // m_nIndex, nReadSize, nWillDownloadSize , nWillDownloadSize );
  306. }
  307. }
  309. // 先将数据保存到临时缓冲中
  310. if ( nReadSize > 0 )
  311. {
  312. nReadSize = MIN ( nReadSize, TEMP_SAVE_BUFFER_SIZE-nTempSaveBytes );
  313. memcpy ( szTempSaveBuf+nTempSaveBytes, szRecvBuf, nReadSize );
  314. nTempSaveBytes += nReadSize;
  315. ASSERT ( nTempSaveBytes < TEMP_SAVE_BUFFER_SIZE );
  316. }
  317. // 当下载已完成或者收到的数据超过一定数量时才保存到文件中
  318. if ( bDownloadFinished || nTempSaveBytes >= WRITE_TEMP_SAVE_MIN_BYTES )
  319. {
  320. // 保存文件失败，下载也应该终止
  321. nDownloadedSize = SaveDataToFile ( szTempSaveBuf, nTempSaveBytes );
  322. if ( nDownloadedSize < 0 )
  323. {
  324. break;
  325. }
  326. nTempSaveBytes = 0;
  327. }
  328. Set_TempSaveBytes ( nTempSaveBytes );
  330. if ( bDownloadFinished )
  331. {
  332. ASSERT ( (nWillDownloadSize > 0 && nDownloadedSize >= nWillDownloadSize) || (nReadSize <= 0 && nWillDownloadSize <= 0) );
  333. break;
  334. }
  336. SLEEP_BREAK ( 1 );
  337. }
  339. if ( szTempSaveBuf ) delete[] szTempSaveBuf;
  340. szTempSaveBuf = NULL;
  342. BOOL bRes = FALSE;
  343. int nWillDownloadSize = Get_WillDownloadSize ();
  344. if ( nWillDownloadSize != -1 )
  345. {
  346. if ( nDownloadedSize >= nWillDownloadSize )
  347. bRes = TRUE;
  348. }
  349. else if ( nDownloadedSize > 0 )
  350. {
  351. bRes = TRUE;
  352. }
  354. return bRes;
  355. }
  357. int CDownloadPub::SaveDataToFile(char *data, int size)
  358. {
  359. ASSERT ( HANDLE_IS_VALID ( m_file.m_hFile ) );
  360. if ( !data || size < 0 )
  361. return -1;
  363. int nDownloadedSize = -1;
  364. // 保存下载的数据
  365. ASSERT ( HANDLE_IS_VALID(m_file.m_hFile) );
  366. BOOL bRet = TRUE;
  367. TRY
  368. {
  369. m_file.Write ( data, size );
  370. }
  371. CATCH( CFileException, e )
  372. {
  373. e->Delete ();
  374. bRet = FALSE;
  375. }
  376. END_CATCH
  378. if ( !bRet )
  379. {
  380. Log ( L_WARNING, "Write data to local file [%s] failed", m_file.GetFileName() );
  381. }
  382. else
  383. {
  384. nDownloadedSize = Get_DownloadedSize();
  385. // TRACE ( "对象.%d, 保存 %d(0x%08x) 字节，偏移 %d(0x%08x) 字节， 共 %d(0x%08x) 字节n",
  386. // m_nIndex, size, size, m_nWillDownloadStartPos+nDownloadedSize, m_nWillDownloadStartPos+nDownloadedSize,
  387. // nDownloadedSize + size, nDownloadedSize + size );
  388. nDownloadedSize += size;
  389. Set_DownloadedSize ( nDownloadedSize );
  390. if ( m_Proc_SaveDownloadInfo )
  391. m_Proc_SaveDownloadInfo ( m_nIndex, nDownloadedSize, size, m_wSaveDownloadInfo_Param );
  392. }
  393. if ( !bRet ) return -1;
  394. return nDownloadedSize;
  395. }
  397. BOOL CDownloadPub::GetRemoteSiteInfo()
  398. {
  399. return TRUE;
  400. }
  402. CString CDownloadPub::GetRemoteFileName()
  403. {
  404. int nPos = m_csObject.ReverseFind ( '/' );
  405. if ( nPos <= 0 ) return m_csObject;
  406. return m_csObject.Mid ( nPos+1 );
  407. }
  409. const char* g_szBase64TAB = _T( "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" );
  410. // BASE64编码
  411. int Base64Encode(LPCTSTR lpszEncoding, CString &strEncoded)
  412. {
  413. UINT m_nBase64Mask[]= { 0, 1, 3, 7, 15, 31, 63, 127, 255 };
  414. int nDigit;
  415. int nNumBits = 6;
  416. int nIndex = 0;
  417. int nInputSize;
  419. strEncoded = _T( "" );
  420. if( lpszEncoding == NULL )
  421. return 0;
  423. if( ( nInputSize = lstrlen(lpszEncoding) ) == 0 )
  424. return 0;
  426. int nBitsRemaining = 0;
  427. long lBitsStorage =0;
  428. long lScratch =0;
  429. int nBits;
  430. UCHAR c;
  432. while( nNumBits > 0 )
  433. {
  434. while( ( nBitsRemaining < nNumBits ) &&  ( nIndex < nInputSize ) ) 
  435. {
  436. c = lpszEncoding[ nIndex++ ];
  437. lBitsStorage <<= 8;
  438. lBitsStorage |= (c & 0xff);
  439. nBitsRemaining += 8;
  440. }
  441. if( nBitsRemaining < nNumBits ) 
  442. {
  443. lScratch = lBitsStorage << ( nNumBits - nBitsRemaining );
  444. nBits    = nBitsRemaining;
  445. nBitsRemaining = 0;
  446. }  
  447. else 
  448. {
  449. lScratch = lBitsStorage >> ( nBitsRemaining - nNumBits );
  450. nBits  = nNumBits;
  451. nBitsRemaining -= nNumBits;
  452. }
  453. nDigit = (int)(lScratch & m_nBase64Mask[nNumBits]);
  454. nNumBits = nBits;
  455. if( nNumBits <=0 )
  456. break;
  458. strEncoded += g_szBase64TAB[ nDigit ];
  459. }
  460. // Pad with '=' as per RFC 1521
  461. while( strEncoded.GetLength() % 4 != 0 )
  462. strEncoded += '=';
  464. return strEncoded.GetLength();
  466. }
  468. // STATIC BASE64解码
  469. int Base64Decode(LPCTSTR lpszDecoding, CString &strDecoded)
  470. {
  471. int nIndex =0;
  472. int nDigit;
  473.     int nDecode[ 256 ];
  474. int nSize;
  475. int nNumBits = 6;
  477. if( lpszDecoding == NULL )
  478. return 0;
  480. if( ( nSize = lstrlen(lpszDecoding) ) == 0 )
  481. return 0;
  483. // Build Decode Table
  484. for( int i = 0; i < 256; i++ ) 
  485. nDecode[i] = -2; // Illegal digit
  486. for( i=0; i < 64; i++ )
  487. {
  488. nDecode[ g_szBase64TAB[ i ] ] = i;
  489. nDecode[ '=' ] = -1; 
  490.     }
  492. // Clear the output buffer
  493. strDecoded = _T("");
  494. long lBitsStorage  =0;
  495. int nBitsRemaining = 0;
  496. int nScratch = 0;
  497. UCHAR c;
  499. // Decode the Input
  500. for( nIndex = 0, i = 0; nIndex < nSize; nIndex++ )
  501. {
  502. c = lpszDecoding[ nIndex ];
  504. // 忽略所有不合法的字符
  505. if( c> 0x7F)
  506. continue;
  508. nDigit = nDecode[c];
  509. if( nDigit >= 0 ) 
  510. {
  511. lBitsStorage = (lBitsStorage << nNumBits) | (nDigit & 0x3F);
  512. nBitsRemaining += nNumBits;
  513. while( nBitsRemaining > 7 ) 
  514. {
  515. nScratch = lBitsStorage >> (nBitsRemaining - 8);
  516. strDecoded += (nScratch & 0xFF);
  517. i++;
  518. nBitsRemaining -= 8;
  519. }
  520. }
  521.     }
  523. return strDecoded.GetLength();
  524. }
  526. //
  527. // 从URL里面拆分出Server、Object、协议类型等信息，其中 Object 里的值是区分大小写的，否则有些网站可能会下载不了
  528. //
  529. BOOL ParseURL(LPCTSTR lpszURL, CString &strServer, CString &strObject,USHORT& nPort, CString &csProtocolType)
  530. {
  531. if ( !lpszURL || strlen(lpszURL) < 1 ) return FALSE;
  532. CString csURL_Lower(lpszURL);
  533. csURL_Lower.TrimLeft();
  534. csURL_Lower.TrimRight();
  535. csURL_Lower.Replace ( "\", "/" );
  536. CString csURL = csURL_Lower;
  537. csURL_Lower.MakeLower ();
  539. // 清除数据
  540. strServer = _T("");
  541. strObject = _T("");
  542. nPort   = 0;
  544. int nPos = csURL_Lower.Find("://");
  545. if( nPos == -1 )
  546. {
  547. csURL_Lower.Insert ( 0, "http://" );
  548. csURL.Insert ( 0, "http://" );
  549. nPos = 4;
  550. }
  551. csProtocolType = csURL_Lower.Left ( nPos );
  553. csURL_Lower = csURL_Lower.Mid( csProtocolType.GetLength()+3 );
  554. csURL = csURL.Mid( csProtocolType.GetLength()+3 );
  555. nPos = csURL_Lower.Find('/');
  556. if ( nPos == -1 )
  557. return FALSE;
  559. strObject = csURL.Mid(nPos);
  560. CString csServerAndPort = csURL_Lower.Left(nPos);
  562. // 查找是否有端口号，站点服务器域名一般用小写
  563. nPos = csServerAndPort.Find(":");
  564. if( nPos == -1 )
  565. {
  566. strServer = csServerAndPort;
  567. nPort = DEFAULT_HTTP_PORT;
  568. if ( csProtocolType == "ftp" )
  569. nPort = DEFAULT_FTP_PORT;
  570. }
  571. else
  572. {
  573. strServer = csServerAndPort.Left( nPos );
  574. csServerAndPort   = csServerAndPort.Mid( nPos+1 );
  575. nPort   = (USHORT)_ttoi((LPCTSTR)csServerAndPort);
  576. }
  577. return TRUE;
  578. }
  580. void CDownloadPub::Set_DownloadedSize(int nDownloadedSize)
  581. {
  582. m_CSFor_DownloadedSize.Lock ();
  583. m_nDownloadedSize = nDownloadedSize;
  584. m_CSFor_DownloadedSize.Unlock ();
  585. DownloadNotify ( m_nIndex, NOTIFY_TYPE_THREAD_DOWNLOADED_SIZE, (LPVOID)nDownloadedSize, m_pDownloadMTR );
  586. }
  588. int CDownloadPub::Get_DownloadedSize()
  589. {
  590. int nDownloadedSize = 0;
  591. m_CSFor_DownloadedSize.Lock ();
  592. nDownloadedSize = m_nDownloadedSize;
  593. m_CSFor_DownloadedSize.Unlock ();
  595. return nDownloadedSize;
  596. }
  598. void CDownloadPub::Set_TempSaveBytes(int nTempSaveBytes)
  599. {
  600. m_CSFor_TempSaveBytes.Lock ();
  601. m_nTempSaveBytes = nTempSaveBytes;
  602. m_CSFor_TempSaveBytes.Unlock ();
  603. }
  605. int CDownloadPub::Get_TempSaveBytes()
  606. {
  607. int nTempSaveBytes = 0;
  608. m_CSFor_TempSaveBytes.Lock ();
  609. nTempSaveBytes = m_nTempSaveBytes;
  610. m_CSFor_TempSaveBytes.Unlock ();
  612. return nTempSaveBytes;
  613. }
  615. void CDownloadPub::Set_WillDownloadSize(int nWillDownloadSize)
  616. {
  617. m_CSFor_WillDownloadSize.Lock ();
  618. m_nWillDownloadSize = nWillDownloadSize;
  619. m_CSFor_WillDownloadSize.Unlock ();
  620. DownloadNotify ( m_nIndex, NOTIFY_TYPE_WILL_DOWNLOAD_SIZE, (LPVOID)m_nWillDownloadSize, m_pDownloadMTR );
  621. }
  623. int CDownloadPub::Get_WillDownloadSize()
  624. {
  625. int nWillDownloadSize = 0;
  626. m_CSFor_WillDownloadSize.Lock ();
  627. nWillDownloadSize = m_nWillDownloadSize;
  628. m_CSFor_WillDownloadSize.Unlock ();
  630. return nWillDownloadSize;
  631. }
  633. BOOL CDownloadPub::SetSaveFileName(LPCTSTR lpszSaveFileName)
  634. {
  635. if ( !lpszSaveFileName ) return FALSE;
  636. m_csSaveFileName = lpszSaveFileName;
  637. if ( m_csSaveFileName.IsEmpty() )
  638. return FALSE;
  639. return TRUE;
  640. }
  642. //
  643. // 获取尚未下载的字节数，写到文件中的和临时缓冲里的都算是已经下载的
  644. //
  645. int CDownloadPub::GetUndownloadBytes()
  646. {
  647. // 总共需要下载的字节数减去已经下载的字节数
  648. return Get_WillDownloadSize () - ( Get_DownloadedSize () + Get_TempSaveBytes () );
  649. }
  651. BOOL CDownloadPub::OpenFileForSave()
  652. {
  653. ASSERT ( !m_csSaveFileName.IsEmpty() );
  654. // 打开本地文件
  655. if ( HANDLE_IS_VALID ( m_file.m_hFile ) )
  656. {
  657. m_file.Close ();
  658. }
  659. BOOL bRet = FALSE;
  660. TRY
  661. {
  662. if ( m_file.Open ( m_csSaveFileName, CFile::modeCreate|CFile::modeNoTruncate|CFile::modeReadWrite|CFile::typeBinary|CFile::shareDenyNone ) )
  663. {
  664. int nWillDownloadStartPos = Get_WillDownloadStartPos ();
  665. nWillDownloadStartPos += Get_DownloadedSize ();
  666. if ( m_file.Seek ( nWillDownloadStartPos, CFile::begin ) == nWillDownloadStartPos )
  667. bRet = TRUE;
  668. }
  669. }
  670. CATCH( CFileException, e )
  671. {
  672. e->Delete ();
  673. bRet = FALSE;
  674. }
  675. END_CATCH
  676. if ( !bRet )
  677. {
  678. Log ( L_WARNING, "Open file [%s] failed. %s", m_csSaveFileName, ::hwFormatMessage ( GetLastError() ) );
  679. }
  681. return bRet;
  682. }
  684. void CDownloadPub::Clear_Thread_Handle()
  685. {
  686. CLOSE_HANDLE ( m_hThread );
  687. }
  689. // 消息通知的回调函数
  690. FUNC_DownloadNotify f_Proc_DownloadNotify = NULL;
  691. WPARAM f_wDownloadNotify_Param = NULL;
  692. //
  693. // 设置通知回调函数
  694. //
  695. void Set_DownloadNotify_Callback ( FUNC_DownloadNotify Proc_DownloadNotify, WPARAM wParam )
  696. {
  697. f_Proc_DownloadNotify = Proc_DownloadNotify;
  698. f_wDownloadNotify_Param = wParam;
  699. }
  701. void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR )
  702. {
  703. ASSERT ( pDownloadMTR );
  704. t_DownloadNotifyPara DownloadNotifyPara = {0};
  705. DownloadNotifyPara.nIndex = nIndex;
  706. DownloadNotifyPara.nNotityType = nNotityType;
  707. DownloadNotifyPara.lpNotifyData = lpNotifyData;
  708. DownloadNotifyPara.pDownloadMTR = pDownloadMTR;
  710. if ( f_Proc_DownloadNotify )
  711. f_Proc_DownloadNotify ( &DownloadNotifyPara, f_wDownloadNotify_Param );
  712. }
  714. //
  715. // 获取下载对象的文件名（带扩展名的）
  716. //
  717. CString CDownloadPub::GetDownloadObjectFileName(CString *pcsExtensionName)
  718. {
  719. ASSERT ( !m_csObject.IsEmpty() );
  720. CString csOnlyPath, csOnlyFileName, csExtensionName;
  721. if ( !PartPathAndFileAndExtensionName ( m_csObject, &csOnlyPath, &csOnlyFileName, &csExtensionName ) )
  722. return "";
  723. if ( pcsExtensionName ) *pcsExtensionName = csExtensionName;
  724. if ( !csExtensionName.IsEmpty() )
  725. {
  726. csOnlyFileName += ".";
  727. csOnlyFileName += csExtensionName;
  728. }
  729. return csOnlyFileName;
  730. }