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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 杀毒 > 查看源码
MemFileObject.cpp
资源名称：BAV.v2.rar　[点击查看]
上传用户：chaoyu
上传日期：2013-04-28
资源大小：18k
文件大小：2k
源码类别：

杀毒

开发平台：

Visual C++

MemFileObject.cpp：源码内容
  1. #include "StdAfx.h"
  2. #include "MemFileObject.h"
  4. CMemFileObject::CMemFileObject(void)
  5. {
  6. m_eObjType = BO_MEM_FILE;
  8. m_hMapFile = NULL;
  9. m_pFile = NULL;
  10. }
  12. CMemFileObject::CMemFileObject(CMemFileObject& cMemFileObj)
  13. {
  14. m_eObjType = BO_MEM_FILE;
  16. m_hFile = INVALID_HANDLE_VALUE;
  17. m_dwObjSize = cMemFileObj.m_dwObjSize;
  18. m_strObjName= cMemFileObj.m_strObjName;
  20. m_hMapFile = NULL;
  21. m_pFile = NULL;
  22. }
  24. CMemFileObject::~CMemFileObject(void)
  25. {
  26. if(m_pFile)
  27. Close();
  28. }
  31. bool CMemFileObject::Open(bool bScan)
  32. {
  33. if(!CFileObject::Open(bScan))
  34. return false;
  36. DWORD dwProtect, dwDesiredAccess;
  37. if(bScan)
  38. {
  39. dwProtect = PAGE_READONLY;
  40. dwDesiredAccess = FILE_MAP_READ;
  41. }
  42. else
  43. {
  44. dwProtect = PAGE_READWRITE;
  45. dwDesiredAccess = FILE_MAP_WRITE;
  46. }
  48. // TODO: If clean virus enlarge the file size, we need calculate and change the dwMaximumSizeLow value.
  49. m_hMapFile = ::CreateFileMapping(m_hFile, NULL, dwProtect, 0, 0, NULL);
  50. if(m_hMapFile==NULL)
  51. {
  52. Close();
  53. return false;
  54. }
  55. m_pFile = static_cast<LPBYTE>( ::MapViewOfFile(m_hMapFile, dwDesiredAccess, 0, 0, NULL) );
  56. if(m_pFile==NULL)
  57. {
  58. Close();
  59. return false;
  60. }
  62. return true;
  63. }
  65. bool CMemFileObject::Close()
  66. {
  67. if(m_pFile)
  68. {
  69. ::UnmapViewOfFile(m_pFile);
  70. m_pFile = NULL;
  71. }
  72. if(m_hMapFile)
  73. {
  74. ::CloseHandle(m_hMapFile);
  75. m_hMapFile = NULL;
  76. }
  77. return CFileObject::Close();
  78. }
  80. bool CMemFileObject::Compare(IN INT nOffset, IN INT nSize, IN LPBYTE pSignature, IN BAV_SIGN_LOGIC_OPERATION eLogicOp)
  81. {
  82. return Compare(m_pFile+nOffset, nSize, pSignature, eLogicOp);
  83. }
  85. bool CMemFileObject::Compare(IN LPBYTE pData, IN INT nSize, IN LPBYTE pSignature, IN BAV_SIGN_LOGIC_OPERATION eLogicOp)
  86. {
  87. ASSERT(m_eObjType==BO_MEM_FILE);
  88. ASSERT(nSize<=MAX_SIGNATURE_LEN);
  90. if(!IsOpened())
  91. {
  92. ASSERT(false);
  93. return false;
  94. }
  96. // Signature position exceed the file size
  97. if(pData<m_pFile || pData>(m_pFile+m_dwObjSize))
  98. {
  99. // not match
  100. return false;
  101. }
  103. int nRet = memcmp(pData, pSignature, nSize);
  104. switch(eLogicOp)
  105. {
  106. case BL_EQUAL:
  107. return nRet==0;
  108. case BL_NOT_EQUAL:
  109. return nRet!=0;
  110. default:
  111. ASSERT(false);
  112. }
  114. return false;
  115. }
  117. bool CMemFileObject::IsOpened()
  118. {
  119. return m_pFile!=NULL;
  120. }
  122. LPBYTE CMemFileObject::GetBuffer()
  123. {
  124. ASSERT(IsOpened());
  125. return m_pFile;
  126. }