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

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

P2P编程

开发平台：

Visual C++

SingleDownload.h：源码内容
  1. // SingleDownload.h: interface for the CSingleDownload class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #if !defined(AFX_SINGLEDOWNLOAD_H__ED8FB68A_724C_4B21_AB2A_DE37B7932B3C__INCLUDED_)
  6. #define AFX_SINGLEDOWNLOAD_H__ED8FB68A_724C_4B21_AB2A_DE37B7932B3C__INCLUDED_
  8. #if _MSC_VER > 1000
  9. #pragma once
  10. #endif // _MSC_VER > 1000
  12. #include "Measure.h"
  14. class CDownloader;
  15. class CConnection;
  16. class CDownloaderFeedback;
  18. class CRequestsItem
  19. {
  20. public:
  21. CRequestsItem(long index, long begin, long length)
  22. {
  23. m_index = index;
  24. m_begin = begin;
  25. m_length = length;
  26. }
  27. bool operator == (CRequestsItem& other)
  28. {
  29. if (m_index == other.m_index && 
  30. m_begin == other.m_begin &&
  31. m_length == other.m_length)
  32. return true;
  33. return false;
  34. }
  36. long m_index;
  37. long m_begin;
  38. long m_length;
  39. };
  41. class CSingleDownload  
  42. {
  43. public:
  44. CSingleDownload(CDownloaderFeedback* pDownloaderFeedback,CDownloader* pDownloader, CConnection* pConnection);
  45. virtual ~CSingleDownload();
  47. void got_have_bitfield(vector<bool>& vHave);
  48. void got_have(long index);
  49. bool got_piece(long index,long begin, memstream& memPiece);
  50. void got_unchoke();
  51. void got_choke();
  52. void disconnected();
  54. bool is_snubbed();
  55. long get_rate();
  56. long get_total();
  57. bool is_interested();
  58. bool is_choked();
  60. float GetHaveCountPercent();
  61. void get_have(vector<bool>& vHave) ;
  62. long get_have_count() const;
  63. void get_new_have(vector<bool>& vHave);
  65. void download_more(vector<long>& vPieces = vector<long>());
  67. // for endgame.
  68. void cancel_endgame();
  69. void make_endgame(vector<CRequestsItem>& vRequests);
  70. void mak_endgameflush();
  71. private:
  72. void _check_interest(vector<long>& vHit);
  73. void _lost_interest(vector<long>& vHit);
  74. void _d(vector<long>&vPieces, vector<long>& vHit);
  75. void _letgo();
  77. CDownloader* m_pDownloader;
  78. CConnection* m_pConnection;
  79. bool m_bChoked;
  80. bool m_bInterested;
  81. vector<CRequestsItem> m_vActiveRequests;
  82. CMeasure m_measure;
  83. vector<bool> m_vHave;
  84. time_t m_tLast;
  86. // for endgame.
  87. void send_request_end(long index, long begin, long length);
  88. void got_have_bitfield_end(vector<bool>& vHave);
  89. void got_have_end(long index);
  90. bool got_piece_end(long index,long begin, memstream& memPiece);
  91. void got_unchoke_end();
  92. void got_choke_end();
  93. void disconnected_end();
  94. void SendCancel(long index, long begin, long length, CSingleDownload* pOwner);
  96. bool m_bEndgame;
  97. long m_lHave;
  99. // for feedback
  100. CDownloaderFeedback* m_pDownloaderFeedback;
  101. long m_lAddr;
  102. short m_sPort;
  103. vector<long> m_vNewHave;
  104. };
  107. #endif // !defined(AFX_SINGLEDOWNLOAD_H__ED8FB68A_724C_4B21_AB2A_DE37B7932B3C__INCLUDED_)