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

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

P2P编程

开发平台：

Visual C++

Storage.h：源码内容
  1. // StorageEx.h: interface for the CStorageEx class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #if !defined(AFX_STORAGEEX_H__EB7826B8_4D91_41DF_80BB_927B7396622C__INCLUDED_)
  6. #define AFX_STORAGEEX_H__EB7826B8_4D91_41DF_80BB_927B7396622C__INCLUDED_
  8. #if _MSC_VER > 1000
  9. #pragma once
  10. #endif // _MSC_VER > 1000
  13. #include "FileInfo.h"
  14. class CDownloaderFeedback;
  16. class CSizeEx
  17. {
  18. public :
  19. CSizeEx(const BLONG x, const BLONG y)
  20. {
  21. cx = x;
  22. cy = y;
  23. }
  24. CSizeEx()
  25. {
  26. cx = cy = 0;
  27. }
  29. BLONG cx;
  30. BLONG cy;
  31. };
  33. class CStorageEx  
  34. {
  35. class CRange
  36. {
  37. public:
  38. CRange(BLONG lbeg, BLONG lend, string strFileName)
  39. {
  40. m_lbeg = lbeg;
  41. m_lend = lend;
  42. m_strFileName = strFileName;
  44. }
  45. BLONG m_lbeg;
  46. BLONG m_lend;
  47. string m_strFileName;
  48. };
  50. public:
  51. CStorageEx();
  52. virtual ~CStorageEx();
  54. bool Create(CDownloaderFeedback* pMain, vector<CFileInfo>& vFiles, 
  55. HANDLE hevDone, long lMaxFilesOpen = 0, bool bLockFiles = true, bool bLockWhileReading = false);
  56. void Close();
  58. void GetUnNeededRanges(const vector<CSize>& vUnNeededFileInxs, vector<CSizeEx>& vUnNeededRanges) const;
  59. BLONG GetTotalLength();
  60. void SetReadOnly();
  61. bool was_preallocated(BLONG lPos, BLONG lLength);
  63. bool flush();
  64. void top_off();
  65. bool write(char *pBuf, BLONG lPos, BLONG lAmount);
  66. bool read(char *pBuf, BLONG lPos, BLONG lAmount, bool bFlushFirst = false);
  68. bool GetFilesRange(vector<long>& vRangeHoles, vector<long>& vFileOffset, long lPieceLen);
  70. private:
  71. bool _intervals(BLONG lPos, BLONG lAmount, vector<CRange>& v);
  72. FILE* _get_file_handle(string strFileName, bool bForWrite);
  73. FILE* _open(string strFileName, string strMode);
  74. void _close(string strFileName);
  75. void _lock_file(string strFileName, FILE* pfile);
  76. void _unlock_file(string strFileName, FILE* pfile);
  77. static time_t getfiletime(string strFileName);
  79. HANDLE m_hevDone;
  80. CDownloaderFeedback* m_pMain;
  82. bool m_bLockFiles;
  83. bool m_bLockWhileReading;
  84. bool m_bMaxFilesOpen;
  85. long m_lMaxFilesOpen;
  86. BLONG m_lTotalLength;
  88. vector<string> m_vHandleBuffer;
  89. vector<CRange> m_vRange;
  90. map<string, BLONG> m_mTops;
  91. map<string, BLONG> m_mSizes;
  92. map<string, time_t> m_mTimes;
  93. map<string, FILE*> m_mHandles;
  94. map<string, bool> m_mWHandles;
  97. };
  99. class CStorage : public CStorageEx
  100. {
  101. };
  103. #endif // !defined(AFX_STORAGEEX_H__EB7826B8_4D91_41DF_80BB_927B7396622C__INCLUDED_)