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

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

P2P编程

开发平台：

Visual C++

testBTDoc.cpp：源码内容
  1. // testBTDoc.cpp : implementation of the CTestBTDoc class
  2. //
  4. #include "stdafx.h"
  5. #include "testBT.h"
  8. #include "testBTDoc.h"
  9. #include "TestBTView.h"
  10. #include "MainFrm.h"
  12. #include "MakeMetaFile.h"
  13. #include "DlgMakeMetafile.h"
  15. #include "Storage.h"
  16. #include "StorageWrapper.h"
  17.  
  20. #ifdef _DEBUG
  21. #define new DEBUG_NEW
  22. #undef THIS_FILE
  23. static char THIS_FILE[] = __FILE__;
  24. #endif
  26. /////////////////////////////////////////////////////////////////////////////
  27. // CTestBTDoc
  29. IMPLEMENT_DYNCREATE(CTestBTDoc, CDocument)
  31. BEGIN_MESSAGE_MAP(CTestBTDoc, CDocument)
  32. //{{AFX_MSG_MAP(CTestBTDoc)
  33. ON_COMMAND(ID_MENUITEM_MAKE_METAFILE, OnMenuitemMakeMetafile)
  34. //}}AFX_MSG_MAP
  35. ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
  36. ON_COMMAND(ID_MENUITEM_FIND_FILE, OnMenuitemFindFile)
  37. ON_COMMAND(ID_MENUITEM_FIND_NEXT, OnMenuitemFindNext)
  38. ON_COMMAND(ID_MENUITEM_SELECT_ALL, OnMenuitemSelectAll)
  39. ON_COMMAND(ID_FILE_NEW_DATABASE, OnFileNewDatabase)
  40. ON_COMMAND(ID_FILE_SAVE_AS_DATABASE, OnFileSaveAsDatabase)
  41. ON_COMMAND(ID_MENUITEM_OPEN_DATABASE, OnMenuitemOpenDatabase)
  42. ON_COMMAND(ID_FILE_SAVE_DATABASE, OnFileSaveDatabase)
  43. ON_UPDATE_COMMAND_UI_RANGE(ID_MENUITEM_STOP, ID_MENUITEM_OPEN_DOWNLOADED_DIRECTORY, OnUpdateMenuitem)
  44. ON_COMMAND_RANGE(ID_MENUITEM_STOP, ID_MENUITEM_OPEN_DOWNLOADED_DIRECTORY, OnCommandMenuitem)
  45. END_MESSAGE_MAP()
  46. //ON_UPDATE_COMMAND_UI(ID_MENUITEM_DOWNLOAD, OnUpdateMenuitem)
  47. // ON_COMMAND(ID_MENUITEM_DOWNLOAD, OnCommandMenuitem)
  49. /////////////////////////////////////////////////////////////////////////////
  50. // CTestBTDoc construction/destruction
  52. /*
  53. class temp
  54. {
  55. public :
  57. temp& operator = (const temp& other)
  58. {
  59. memcpy(szBuf, other.szBuf, 20);
  60. return *this;
  61. }
  63. bool operator == (const temp& other)  const
  64. {
  65. return memcmp(szBuf, other.szBuf, 20) == 0;
  66. return true;
  67. }
  69. bool operator < (const temp& other)  const
  70. {
  71. return memcmp(szBuf, other.szBuf, 20) < 0;
  72. }
  74. char szBuf[20];
  75. };
  76. //*/
  77. // #include "Sha.h"
  78. CTestBTDoc::CTestBTDoc()
  79. {
  80. m_iOpendFileCount = 0;
  82. /*
  84. char szText[11] = "abcdefghij";
  85. unsigned char shaPart[20] = {0};
  86. CSHA sha;
  87. sha.start();
  88. sha.update((unsigned char*)szText, 5);
  89. // sha.finish(shaPart);
  90. sha.update((unsigned char*)szText + 5, 5);
  91. sha.finish(shaPart);
  93. TRACE("rn{");
  94. for (int i=0; i<20; i++)
  95. {
  96. TRACE("%d, ", shaPart[i]);
  97. }
  98. TRACE("}rn");
  101. CSHA sha2;
  102. sha2.start();
  103. sha2.update((unsigned char*)szText, 10);
  104. sha2.finish(shaPart);
  107. TRACE("rn{");
  108. for (i=0; i<20; i++)
  109. {
  110. TRACE("%d, ", shaPart[i]);
  111. }
  112. TRACE("}rn");
  113. //*/
  115. /*
  116. temp t;
  117. memset(t.szBuf, 0, 20);
  118. vector<temp> pvPieces;
  119. for (int i=0; i<10; i++)
  120. pvPieces.push_back(t);
  121. assert(find(pvPieces.begin(), pvPieces.end(), t) != pvPieces.end());
  122. map<temp, long> mPieces;
  123. assert(mPieces.find(t) == mPieces.end());
  124. mPieces[t] = 100;
  125. assert(mPieces.begin() != mPieces.end());
  126. long l = mPieces.size();
  127. long zz = mPieces[t];
  128. zz = (*mPieces.begin()).second;
  129. temp tx = (*mPieces.begin()).first;
  130. assert(mPieces.find(t) != mPieces.end());
  131. //*/
  134. // pvPieces.push_back(szBuf);
  136. /*
  137. try
  138. {
  139. CFileTorrent fileTorrernt;
  140. if (!fileTorrernt.OpenFile("F:\Documents and Settings\道理\桌面\IEButton.torrent", "d:\za"))
  141. {
  142. assert(false);
  143. return ;
  144. }
  146. vector<CFileInfo>& vFiles = fileTorrernt.GetFileInfo();
  147. CStorageEx storage;
  148. storage.Create(0, vFiles, 0, 10, true, 0);
  150. CVal* pInfo = fileTorrernt.GetInfo();
  151. vector<char*>& vPieces = fileTorrernt.GetPieces();
  153. CStorageWrapperEx storageWrapper;
  154. storageWrapper.Create(0, 0, 0, &storage, vPieces, true, pow(2, 14), (*pInfo)["piece length"]->lVal);
  155. /*
  156. bool bpre = storage.was_preallocated(10, 1000000);
  157. long lTotal = storage.GetTotalLength();
  158. for (int i=0; i<100; i++)
  159. {
  160. storage.write("asdfdsafda", i*10, 10);
  161. }
  162. storage.SetReadOnly();
  163. storage.write("asdfdsafda", 1023667, 10);
  164. char szText[20] = {0};
  165. storage.read(szText, 1023667, 10);
  167. storage.top_off();
  168. storage.Close();
  169. //*/
  170. /*
  171. }
  172. catch (string e)
  173. {
  174. AfxMessageBox(e.data());
  175. }
  176. //*/
  178. m_bInitiated = false;
  179. }
  181. CTestBTDoc::~CTestBTDoc()
  182. {
  183. }
  185. BOOL CTestBTDoc::OnNewDocument()
  186. {
  187. if (!CDocument::OnNewDocument())
  188. return FALSE;
  190. CWnd* pwnd = AfxGetMainWnd();
  191. POSITION pos = GetFirstViewPosition();
  192. CView* pFirstView = GetNextView( pos );
  194. if (!m_bInitiated)
  195. {
  196. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  197. m_bInitiated = true;
  198. }
  200. return TRUE;
  201. }
  203. CFilesListCtrl& CTestBTDoc::GetFileListCtrl() const
  204. {
  205. POSITION Pos = GetFirstViewPosition();
  206. assert(Pos);
  207. CTestBTView* pView = (CTestBTView*)GetNextView(Pos);
  208. return pView->m_FilesList;
  209. }
  211. CCategoryBarWnd& CTestBTDoc::GetCategoryBarWnd() const 
  212. {
  213. assert(((CTestBTApp*)AfxGetApp())->m_pMainFrame);
  214. CFrameWnd* pFrame = ((CTestBTApp*)AfxGetApp())->m_pMainFrame;
  215. return ((CMainFrame*)pFrame)->m_CategoryBar.m_CategoryBarWnd;
  217. return ((CMainFrame*)AfxGetMainWnd())->m_CategoryBar.m_CategoryBarWnd;
  218. }
  220. /////////////////////////////////////////////////////////////////////////////
  221. // CTestBTDoc serialization
  223. void CTestBTDoc::Serialize(CArchive& ar)
  224. {
  225. if (ar.IsStoring())
  226. {
  227. // TODO: add storing code here
  228. }
  229. else
  230. {
  231. // TODO: add loading code here
  232. }
  233. }
  235. /////////////////////////////////////////////////////////////////////////////
  236. // CTestBTDoc diagnostics
  238. #ifdef _DEBUG
  239. void CTestBTDoc::AssertValid() const
  240. {
  241. CDocument::AssertValid();
  242. }
  244. void CTestBTDoc::Dump(CDumpContext& dc) const
  245. {
  246. CDocument::Dump(dc);
  247. }
  248. #endif //_DEBUG
  250. /////////////////////////////////////////////////////////////////////////////
  251. // CTestBTDoc commands
  253. void CTestBTDoc::MakeTorrent(CString strFilePath)
  254. {
  255. CDlgMakeMetafile dlg(this);
  256. dlg.m_strFilePath = strFilePath;
  257. dlg.DoModal();
  258. }
  260. BOOL CTestBTDoc::OnOpenDocument(LPCTSTR lpszPathName) 
  261. {
  262. DWORD dwAttr = GetFileAttributes(lpszPathName);
  263. if (dwAttr ==  0xffffffff)
  264. {
  265. AfxMessageBox(CString ("文件(") + lpszPathName + ") 不存在");
  266. return FALSE;
  267. }
  269. bool bOpenDB = false;
  270. bool bOpenTorrent = false;
  271. if (dwAttr & FILE_ATTRIBUTE_DIRECTORY)
  272. {
  273. bOpenTorrent = false;
  274. bOpenDB = false;
  275. }
  276. else
  277. {
  278. char drive[_MAX_DRIVE];
  279. char dir[_MAX_DIR];
  280. char fname[_MAX_FNAME];
  281. char ext[_MAX_EXT];
  282. _splitpath(lpszPathName, drive, dir, fname, ext);
  284. CString strTorrentExt(".torrent");
  285. CString strDBExt(".bt");
  287. if (strlen(ext))
  288. {
  289. if (!strTorrentExt.CompareNoCase(ext))
  290. {
  291. bOpenTorrent = true;
  292. }
  293. else if (!strDBExt.CompareNoCase(ext))
  294. {
  295. bOpenDB = true;
  296. }
  297. }
  299. }
  301. if (bOpenTorrent)
  302. {
  303. GetCategoryBarWnd().OpenTorrentFile(lpszPathName);
  305. if (((CMainFrame*)AfxGetMainWnd())->m_setupRegister.m_bSaveDBAuto && 
  306. ++m_iOpendFileCount >= ((CMainFrame*)AfxGetMainWnd())->m_setupRegister.m_lSaveDBIntval) 
  307. {
  308. if (!GetCategoryBarWnd().IsModifyed())
  309. {
  310. m_iOpendFileCount = 0;
  311. GetCategoryBarWnd().Save();
  312. }
  313. }
  314. }
  315. else if (bOpenDB)
  316. {
  317. if (!GetCategoryBarWnd().OpenDBFile(lpszPathName))
  318. return false;
  320. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  321. return TRUE;
  322. // AfxGetApp()->AddToRecentFileList(lpszPathName);
  323. }
  324. else
  325. {
  326. CDlgMakeMetafile dlg(this);
  327. dlg.m_strFilePath = lpszPathName;
  328. dlg.DoModal();
  329. }
  333. // if (!CDocument::OnOpenDocument(lpszPathName))
  334. // return FALSE;
  335. // return TRUE;
  336. return FALSE;
  337. }
  340. //////////////////////////////////////////////////////////////////////
  341. // find files
  343. void CTestBTDoc::OnMenuitemFindFile() 
  344. {
  345. DbgGuiLeak m_dbgGui;
  346. GetCategoryBarWnd().OnFindFile();
  347. }
  349. void CTestBTDoc::OnMenuitemFindNext() 
  350. {
  351. GetCategoryBarWnd().OnFindNext();
  352. }
  354. void CTestBTDoc::OnMenuitemSelectAll() 
  355. {
  356. GetFileListCtrl().SelectAllItems();
  357. }
  359. ////////////////////////////////////////////////////////////////////////////////
  360. // database.
  362. bool CTestBTDoc::PromptToSaveDatabase(bool bExit)
  363. {
  364. if (GetCategoryBarWnd().IsModifyed())
  365. {
  366. int iRet = AfxMessageBox("是否保存变化到无标题?"/*Save changes to Untitled?"//*/, MB_YESNOCANCEL);
  367. if (iRet == IDCANCEL)
  368. return false;
  369. else if (iRet == IDNO)
  370. {
  371. return true;
  372. }
  373. else if (iRet == IDYES)
  374. {
  375. if (!SaveAsDatabase())
  376. return false;
  377. return true;
  378. }
  379. }
  380. else
  381. {
  382. if (bExit)
  383. GetCategoryBarWnd().StopAllCompleteds();
  385. GetCategoryBarWnd().Save();
  386. }
  388. return true;
  389. }
  391. void CTestBTDoc::OnFileNewDatabase() 
  392. {
  393. if (GetCategoryBarWnd().IsDownloading())
  394. {
  395. if (AfxMessageBox("有任务正在进行中, 继续关闭吗?", MB_YESNO | MB_DEFBUTTON2) == IDNO)
  396. return ;
  397. }
  399. if (!PromptToSaveDatabase())
  400. return;
  402. GetCategoryBarWnd().NewDatabase();
  403. GetCategoryBarWnd().GetRootItem().Select();
  404. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  405. }
  407. void CTestBTDoc::OnMenuitemOpenDatabase() 
  408. {
  409. if (GetCategoryBarWnd().IsDownloading())
  410. {
  411. if (AfxMessageBox("有任务正在进行中, 继续关闭吗?", MB_YESNO|MB_DEFBUTTON2) == IDNO)
  412. return ;
  414. }
  415. if (!PromptToSaveDatabase())
  416. return;
  418.  
  419. int structsize = 76;
  420. DWORD dwVersion,dwWindowsMajorVersion,dwWindowsMinorVersion;
  421. dwVersion = GetVersion();
  422. dwWindowsMajorVersion = (DWORD)(LOBYTE(LOWORD(dwVersion)));
  423. dwWindowsMinorVersion = (DWORD)(HIBYTE(LOWORD(dwVersion)));
  424. if (dwVersion < 0x80000000)
  425. structsize =88;
  427. CFileDialog fileDialog(TRUE, 0, 0, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, "Database (*.bt)|*.bt|All Files (*.*)|*.*|");
  428. fileDialog.m_ofn.lStructSize = structsize; 
  431. int iRet = fileDialog.DoModal();
  432. if (iRet != IDOK) return ;
  433. CString strFileName = fileDialog.GetPathName();
  434. if (!GetCategoryBarWnd().OpenDBFile(strFileName.GetBuffer(0)))
  435. return;
  437. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  438. AfxGetApp()->AddToRecentFileList(strFileName);
  439. }
  441. void CTestBTDoc::OnFileSaveDatabase() 
  442. {
  443. if (GetCategoryBarWnd().IsModifyed())
  444. OnFileSaveAsDatabase();
  445. else
  446. GetCategoryBarWnd().Save();
  448. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  449. }
  451. void CTestBTDoc::OnFileSaveAsDatabase() 
  452. {
  453. SaveAsDatabase();
  454. }
  456. bool CTestBTDoc::SaveAsDatabase(bool bExit) 
  457. {
  458. CString strDataBaseFullPath = GetCategoryBarWnd().GetDatabaseFullPath().data();
  460. int structsize = 76;
  461. DWORD dwVersion,dwWindowsMajorVersion,dwWindowsMinorVersion;
  462. dwVersion = GetVersion();
  463. dwWindowsMajorVersion = (DWORD)(LOBYTE(LOWORD(dwVersion)));
  464. dwWindowsMinorVersion = (DWORD)(HIBYTE(LOWORD(dwVersion)));
  465. if (dwVersion < 0x80000000)
  466. structsize =88;
  468. CFileDialog fileDialog(FALSE, 0, strDataBaseFullPath, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, "Database (*.bt)|*.bt|");
  469. // if (!strDataBaseFullPath.IsEmpty())
  470. // fileDialog.m_ofn.lpstrFile = (char*)(LPCTSTR)strDataBaseFullPath;
  472. fileDialog.m_ofn.lStructSize = structsize; 
  474. int iRet = fileDialog.DoModal();
  475. if (iRet != IDOK)
  476. return false;
  478. if (bExit)
  479. GetCategoryBarWnd().StopAllCompleteds();
  481. CString strFileName = fileDialog.GetPathName();
  482. GetCategoryBarWnd().SaveAs(strFileName.GetBuffer(0));
  483. SetTitle(GetCategoryBarWnd().GetDatabaseName().data());
  485. AfxGetApp()->AddToRecentFileList(strFileName);
  486. return true;
  487. }
  489. BOOL CTestBTDoc::CanCloseFrame(CFrameWnd* pFrame) 
  490. {
  491. if (GetCategoryBarWnd().IsMovingFiles())
  492. {
  493. AfxMessageBox("正在移动文件， 不能关闭系统");
  494. return FALSE;
  495. }
  496. if (GetCategoryBarWnd().IsDownloading())
  497. {
  498. if (AfxMessageBox("有任务正在进行中, 继续关闭吗?", MB_YESNO|MB_DEFBUTTON2) == IDNO)
  499. return FALSE;
  500. }
  502. if (!PromptToSaveDatabase(true))
  503. return FALSE;
  505. if (pFrame && pFrame->IsWindowVisible())
  506. pFrame->SendMessage(WM_SYSCOMMAND, SC_MINIMIZE, 0);
  508. return CDocument::CanCloseFrame(pFrame);
  509. }
  511. /////////////////////////////////////////////////////////////
  512. // handle menu and toolbar message.
  514. void CTestBTDoc::OnUpdateMenuitem(CCmdUI* pCmdUI) 
  515. {
  517. switch (pCmdUI->m_nID)
  518. {
  519. case ID_MENUITEM_DOWNLOAD:
  520. case ID_MENUITEM_PAUSE:
  521. case ID_MENUITEM_STOP:
  522. case ID_MENUITEM_DEL:
  523. case ID_MENUITEM_FILE_PROPERTY:
  524. case ID_MENUITEM_OPEN_DOWNLOADED_FILE:
  525. case ID_MENUITEM_OPEN_DOWNLOADED_DIRECTORY:
  526. case ID_MENUITEM_MOVETO:
  527. case ID_MENUITEM_RUN_PARAMETERS:
  528. case ID_MENUITEM_COPY_TORRENT_FILE_PATH:
  529. {
  530. GetFileListCtrl().OnUpdateMenuitem(pCmdUI);
  531. return;
  532. }
  533. break;
  535. case ID_MENUITEM_FIND_NEXT:
  536. case ID_MENUITEM_OPEN_CATEGORY:
  537. case ID_MENUITEM_MOVETO_CATEGORY:
  538. case ID_MENUITEM_NEW_CATEGORY:
  539. case ID_MENUITEM_DEL_CATEGORY:
  540. case ID_MENUITEM_PORPERTY_CATEGORY:
  541. GetCategoryBarWnd().OnUpdateMenuitem(pCmdUI);
  542. return;
  543. break;
  546. }
  547. pCmdUI->ContinueRouting();
  548. }
  550. void CTestBTDoc::OnCommandMenuitem(UINT iID) 
  551. {
  552. switch (iID)
  553. {
  554. case ID_MENUITEM_DOWNLOAD:
  555. case ID_MENUITEM_PAUSE:
  556. case ID_MENUITEM_STOP:
  557. case ID_MENUITEM_DEL:
  558. case ID_MENUITEM_FILE_PROPERTY:
  559. case ID_MENUITEM_OPEN_DOWNLOADED_FILE:
  560. case ID_MENUITEM_OPEN_DOWNLOADED_DIRECTORY:
  561. case ID_MENUITEM_MOVETO:
  562. case ID_MENUITEM_RUN_PARAMETERS:
  563. case ID_MENUITEM_COPY_TORRENT_FILE_PATH:
  564. GetFileListCtrl().OnCommandMenuitem(iID);
  565. break;
  567. // case ID_MENUITEM_FIND_NEXT:
  568. case ID_MENUITEM_OPEN_CATEGORY:
  569. case ID_MENUITEM_MOVETO_CATEGORY:
  570. case ID_MENUITEM_NEW_CATEGORY:
  571. case ID_MENUITEM_DEL_CATEGORY:
  572. case ID_MENUITEM_PORPERTY_CATEGORY:
  573. GetCategoryBarWnd().OnCommandMenuitem(iID);
  574. break;
  575. }
  576. }
  578. void CTestBTDoc::OnMenuitemMakeMetafile() 
  579. {
  580. CDlgMakeMetafile dlg(this);
  581. dlg.DoModal();
  583. }
  585. void CTestBTDoc::OnFileOpen() 
  586. {
  587. int structsize = 76;
  588. DWORD dwVersion,dwWindowsMajorVersion,dwWindowsMinorVersion;
  589. dwVersion = GetVersion();
  590. dwWindowsMajorVersion = (DWORD)(LOBYTE(LOWORD(dwVersion)));
  591. dwWindowsMinorVersion = (DWORD)(HIBYTE(LOWORD(dwVersion)));
  592. if (dwVersion < 0x80000000)
  593. structsize =88;
  595. CFileDialog fileDialog(TRUE, 0, 0, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, "Torrent File (*.torrent)|*.torrent|All Files (*.*)|*.*|");
  596. fileDialog.m_ofn.lStructSize = structsize; 
  598. int iRet = fileDialog.DoModal();
  599. if (iRet != IDOK) return ;
  600. CString strFileName = fileDialog.GetPathName();
  601. OnOpenDocument(strFileName);
  602. }