BufferMgr.h
上传用户:liguizhu
上传日期:2015-11-01
资源大小:2422k
文件大小:3k
源码类别:

P2P编程

开发平台:

Visual C++

  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. #ifndef __BUFFERMGR_H_
  20. #define __BUFFERMGR_H_
  21. #include "BaseResource.h"
  22. namespace NPLayer1 {
  23. class BufferMgr {
  24. public:
  25. // 构造函数
  26. BufferMgr();
  27. // 析构函数
  28. ~BufferMgr();
  29. // 是否已经初始化
  30. bool HasInited() {return (blockArray != NULL);};
  31. // 取得缓冲文件的大小
  32. UINT GetBufferFileSize() { return bufferFileSize; };
  33. // 取得缓冲文件名
  34. string GetBufferFileName() { return bufferFileName; };
  35. // 初始化缓冲文件,只能调用一次
  36. P2P_RETURN_TYPE Init(string fullPath, bool& bufFileModified, UINT randNum, UINT bufferSize = BUFFER_SPACE);
  37. // 取得一个未使用的块, 并将其置为已使用
  38. UINT GetEmptyIndex(UINT randNum);
  39. // 将一个块置为未使用
  40. void EraseIndex(UINT index);
  41. // 获取剩余空间大小
  42. UINT GetEmptySpace();
  43. // 取得一个块的数据
  44. P2P_RETURN_TYPE GetIndexData(UINT index, UINT offset, LPVOID data, UINT size);
  45. // 填充一个块
  46. P2P_RETURN_TYPE PutIndexData(UINT index, LPVOID data, UINT size);
  47. public:
  48. static BOOL ExCreateFile(HANDLE&, LPCTSTR lpFileName, DWORD dwDesiredAccess,
  49. DWORD dwShareMode, LPSECURITY_ATTRIBUTES lpSecurityAttributes,
  50. DWORD dwCreationDisposition, DWORD dwFlagsAndAttributes, 
  51. HANDLE hTemplateFile);
  52. static BOOL ExCreateFile(HANDLE&, LPCTSTR lpFileName, DWORD dwCreationDisposition);
  53. static UINT ExReadFile(HANDLE, LPVOID buf, UINT toBeRead);
  54. static UINT ExWriteFile(HANDLE, LPCVOID buf, UINT toBeWrite);
  55. static BOOL ExSetFilePointer(HANDLE, UINT offset);
  56. static BOOL ExSetFileSize(HANDLE, UINT size);
  57. static BOOL GetSelfModulePath(LPTSTR buf, DWORD nSize);
  58. private:
  59. enum {
  60. // 默认缓冲文件大小24MB
  61. BUFFER_SPACE = 1048576*24, 
  62. };
  63. // 缓冲文件的句柄
  64. HANDLE hBufferFile;
  65. // 缓冲文件名
  66. string bufferFileName;
  67. // 缓冲文件大小
  68. UINT bufferFileSize;
  69. // 记录缓冲文件的使用情况,按分块
  70. bool* blockArray;
  71. // 缓冲文件的分块个数
  72. UINT blockNum;
  73. };
  74. }
  75. #endif