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

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

杀毒

开发平台：

Visual C++

FileObject.cpp：源码内容
  1. #include "StdAfx.h"
  2. #include ".fileobject.h"
  4. CFileObject::CFileObject(void)
  5. {
  6. m_eObjType = BO_PHY_FILE;
  7. m_hFile = INVALID_HANDLE_VALUE;
  8. m_dwObjSize = 0;
  9. }
  11. CFileObject::CFileObject(CFileObject& cFileObj)
  12. {
  13. m_eObjType = BO_PHY_FILE;
  14. m_hFile = INVALID_HANDLE_VALUE;
  15. m_dwObjSize = cFileObj.m_dwObjSize;
  16. m_strObjName= cFileObj.m_strObjName;
  17. }
  19. CFileObject::~CFileObject(void)
  20. {
  21. if(m_hFile!=INVALID_HANDLE_VALUE)
  22. Close();
  23. }
  26. bool CFileObject::SetObjectName(CString* pstrName)
  27. {
  28. ASSERT(pstrName);
  29. if(pstrName==NULL)
  30. return false;
  32. m_strObjName = *pstrName;
  33. return true;
  34. }
  36. PCSTR CFileObject::GetObjectName()
  37. {
  38. return (PCSTR)m_strObjName;
  39. }
  41. bool CFileObject::Open(bool bScan)
  42. {
  43. DWORD dwDesiredAccess, dwShareMode;
  44. if(bScan)
  45. {
  46. dwDesiredAccess = GENERIC_READ;
  47. dwShareMode = FILE_SHARE_WRITE;
  48. }
  49. else
  50. {
  51. dwDesiredAccess = GENERIC_WRITE;
  52. dwShareMode = FILE_SHARE_DELETE;
  53. }
  54. m_hFile = ::CreateFile(m_strObjName, dwDesiredAccess, dwShareMode, NULL, OPEN_EXISTING, 0, NULL);
  55. m_dwObjSize = GetFileSize(m_hFile, NULL);
  56. return m_hFile != INVALID_HANDLE_VALUE;
  57. }
  59. bool CFileObject::Close()
  60. {
  61. bool bRet = ::CloseHandle(m_hFile)==TRUE;
  62. m_hFile = INVALID_HANDLE_VALUE;
  63. return bRet;
  64. }
  66. bool CFileObject::Compare(IN INT nOffset, IN INT dwSize, IN LPBYTE pSignature, IN BAV_SIGN_LOGIC_OPERATION eLogicOp)
  67. {
  68. ASSERT(m_eObjType==BO_PHY_FILE);
  69. ASSERT(dwSize<=MAX_SIGNATURE_LEN);
  71. if(!IsOpened())
  72. {
  73. ASSERT(false);
  74. return false;
  75. }
  77. // Signature position exceed the file size
  78. if( (nOffset+dwSize)>(INT)m_dwObjSize )
  79. return false;
  81. if(INVALID_SET_FILE_POINTER==::SetFilePointer(m_hFile, nOffset, NULL, FILE_BEGIN))
  82. return false;
  84. // read and compare
  85. BYTE fbuf[MAX_SIGNATURE_LEN];
  86. DWORD dwReadBytes;
  87. if( ::ReadFile(m_hFile, fbuf, dwSize, &dwReadBytes, NULL) && dwReadBytes==dwSize)
  88. {
  89. int nRet = memcmp(fbuf, pSignature, dwSize);
  90. switch(eLogicOp)
  91. {
  92. case BL_EQUAL:
  93. return nRet==0;
  94. case BL_NOT_EQUAL:
  95. return nRet!=0;
  96. default:
  97. ASSERT(false);
  98. }
  99. }
  100. return false;
  101. }
  103. bool CFileObject::Compare(IN LPBYTE pData, IN INT nSize, IN LPBYTE pSignature, IN BAV_SIGN_LOGIC_OPERATION eLogicOp)
  104. {
  105. // CFileObject not support this
  106. ASSERT(false);
  107. return false;
  108. }
  110. bool CFileObject::IsOpened()
  111. {
  112. return m_hFile!=INVALID_HANDLE_VALUE;
  113. }
  115. BAV_OBJ_TYPE CFileObject::GetObjectType()
  116. {
  117. return m_eObjType;
  118. }
  120. DWORD CFileObject::GetObjectSize()
  121. {
  122. return m_dwObjSize;
  123. }