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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
DownloadEngine.cpp
资源名称：GGBT.rar　[点击查看]
上传用户：lds876
上传日期：2013-05-25
资源大小：567k
文件大小：4k
源码类别：

P2P编程

开发平台：

Visual C++

DownloadEngine.cpp：源码内容
  1. // DownloadEngine.cpp: implementation of the CDownloadEngine class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "testbt.h"
  7. #include "DownloadEngine.h"
  10. #include "FileDB.h"
  11. #include "Download.h"
  12. #include "CategoryBarWnd.h"
  13. #include "SetupRegister.h"
  15. #ifdef _DEBUG
  16. #undef THIS_FILE
  17. static char THIS_FILE[]=__FILE__;
  18. #define new DEBUG_NEW
  19. #endif
  21. //////////////////////////////////////////////////////////////////////
  22. // Construction/Destruction
  23. //////////////////////////////////////////////////////////////////////
  25. CDownloadEngine::CDownloadEngine()
  26. {
  27. m_pSetupRegister = 0;
  28. m_pCurrentCheckingItem = 0;
  29. m_pCategoryBarWnd = 0;
  30. }
  32. CDownloadEngine::~CDownloadEngine()
  33. {
  35. }
  37. bool CDownloadEngine::Create(CSetupRegister* pSetupRegister, CCategoryBarWnd* pCategoryBarWnd)
  38. {
  39. m_pSetupRegister = pSetupRegister;
  40. m_pCategoryBarWnd = pCategoryBarWnd;
  41. return true;
  42. }
  44. bool CDownloadEngine::StartDownload(CFileDBItem* pFileDBItem, HTREEITEM hCursorItem)
  45. {
  46. assert(!pFileDBItem->m_pDownload);
  47. pFileDBItem->m_pDownload = new CDownload();
  49. m_vTostartDownloads.push_back(CEngineItem(pFileDBItem, hCursorItem));
  51. if (m_pCurrentCheckingItem)
  52. return true;
  54. ExcuteDownload(0);
  55. return true;
  56. }
  58. //
  59. // long lparam: pointer to  the download which sends the message.
  60. //
  61. void CDownloadEngine::ExcuteDownload(long lParam)
  62. {
  63. assert(m_pSetupRegister);
  65. if (m_vTostartDownloads.size() <= 0)
  66. {
  67. m_pCurrentCheckingItem = 0;
  68. return;
  69. }
  71. // in case bad message, skip it.
  72. if (lParam != 0 && m_pCurrentCheckingItem !=0 )
  73. {
  74. if (m_pCurrentCheckingItem->m_pDownload != (CDownload*)lParam)
  75. {
  76. assert(false);
  77. return;
  78. }
  79. }
  81. HTREEITEM hTreeItem = m_vTostartDownloads[0].m_hTreeItem;
  82. CFileDBItem* pFileDBItem = m_vTostartDownloads[0].m_pFileDBItem;
  83. m_vTostartDownloads.erase(m_vTostartDownloads.begin());
  84. m_pCurrentCheckingItem = pFileDBItem;
  86. assert(pFileDBItem->m_pDownload);
  87. // pFileDBItem->m_pDownload = new CDownload();
  89. //
  90. // Run parameter setup.
  91. //
  92. if (!pFileDBItem->m_bSpecialRunParameter)
  93. {
  94. pFileDBItem->m_lUploadSpeedMode = m_pSetupRegister->m_lDownSpeed;
  95. pFileDBItem->m_lUploadSpeedMax = CSpeedParameters::m_lArrSpeedVal[m_pSetupRegister->m_lDownSpeed];
  96. pFileDBItem->m_lUploadPeerMax = CSpeedParameters::m_lArrPeerVal[m_pSetupRegister->m_lDownSpeed];
  98. pFileDBItem->m_lPeersMin = m_pSetupRegister->m_lPeersMin;
  99. pFileDBItem->m_lPeersMax = m_pSetupRegister->m_lPeersMax;
  100. pFileDBItem->m_lConnectErrorMax = m_pSetupRegister->m_lConnectErrorMax;
  101. pFileDBItem->m_lConnectIntval = m_pSetupRegister->m_lConnectIntval;
  102. pFileDBItem->m_lCompleteKeepSeedHour = m_pSetupRegister->m_lCompleteKeepSeedHour;
  103. }
  105. //
  106. // Set modified time.
  107. //
  108. time(&pFileDBItem->m_tModifiedTime);
  110. bool bCheckHash = true;
  111. if (pFileDBItem->m_fComplete >= 1 && pFileDBItem->m_vUnneededFileInx.empty())
  112. {
  113. bCheckHash = false;
  114. }
  116. if (!pFileDBItem->m_pDownload->DownloadEx(m_pSetupRegister, 
  117. formatDir(m_pCategoryBarWnd->GetDBSavePath())+ pFileDBItem->m_strTorrentFileName, 
  118. pFileDBItem->m_strFileName, bCheckHash, hTreeItem, 
  119. m_pCategoryBarWnd->GetSafeHwnd(), (long)pFileDBItem->m_pDownload, pFileDBItem))
  120. {
  121. pFileDBItem->m_bBad = true;
  122. m_pCategoryBarWnd->OnDownloadTerminated((long)hTreeItem, (long)pFileDBItem->m_pDownload);
  124. m_pCurrentCheckingItem = 0;
  125. ExcuteDownload(0);
  126. return;
  127. }
  128. }
  130. void CDownloadEngine::StopDownload(CFileDBItem* pFileDBItem)
  131. {
  132. for (int i=0; i<m_vTostartDownloads.size(); i++)
  133. {
  134. if (m_vTostartDownloads[i].m_pFileDBItem == pFileDBItem)
  135. {
  136. m_vTostartDownloads.erase(m_vTostartDownloads.begin() + i);
  137. break;
  138. }
  139. }
  141. if(m_pCurrentCheckingItem == pFileDBItem)
  142. {
  143. m_pCurrentCheckingItem = 0;
  144. ExcuteDownload(0);
  145. }
  146. }
  148. void CDownloadEngine::StopDownloadArray(const vector<CFileDBItem*>&  vpFileDBItems)
  149. {
  150. bool bExcuteCommand = false;
  151. for (int j=0; j<vpFileDBItems.size(); j++)
  152. {
  153. // erase item from vector TostartDownloads.
  154. for (int i=0; i<m_vTostartDownloads.size(); i++)
  155. {
  156. if (m_vTostartDownloads[i].m_pFileDBItem == vpFileDBItems[j])
  157. {
  158. m_vTostartDownloads.erase(m_vTostartDownloads.begin() + i);
  159. break;
  160. }
  161. }
  163. if (m_pCurrentCheckingItem == vpFileDBItems[j])
  164. bExcuteCommand = true;
  165. }
  167. if (bExcuteCommand)
  168. {
  169. m_pCurrentCheckingItem = 0;
  170. ExcuteDownload(0);
  171. }
  172. }