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

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

杀毒

开发平台：

Visual C++

Engine.cpp：源码内容
  1. #include "StdAfx.h"
  2. #include "VirusDB.h"
  3. #include "Engine.h"
  4. #include "MemFileObject.h"
  6. CEngine::CEngine(void)
  7. {
  8. m_pcVDB = NULL;
  9. m_dwObjCount = 0;
  10. m_pScanParam = NULL;
  12. m_cScanResults.nSize = sizeof(SCAN_RESULTS);
  13. m_cScanResults.dwObjCount = 0;
  14. m_cScanResults.dwTime = 0;
  15. m_cScanResults.dwRecCount = 0;
  16. m_cScanResults.pScanRecords = NULL;
  18. }
  20. CEngine::~CEngine(void)
  21. {
  22. Release();
  23. }
  25. bool CEngine::Load(IN CVirusDB* pVDB)
  26. {
  27. ASSERT(pVDB);
  28. if(pVDB==NULL)
  29. return false;
  31. m_pcVDB = pVDB;
  32. return true;
  33. }
  35. bool CEngine::Release()
  36. {
  37. ReleaseResults();
  38. return true;
  39. }
  41. PSCAN_RESULTS CEngine::Scan(IN const PSCAN_PARAM pScanParam)
  42. {
  43. m_pScanParam = pScanParam;
  45. // release previous results.
  46. ReleaseResults();
  49. DWORD dwBeginTime = GetTickCount();
  51. // deep first search and call VirusDB.Search() to detect virus.
  52. DFS(pScanParam->strPathName);
  54. DWORD dwEndTime = GetTickCount();
  55. m_cScanResults.dwTime = dwEndTime-dwBeginTime;
  57. return &m_cScanResults;
  58. }
  60. void CEngine::ReleaseResults()
  61. {
  62. PSCAN_RECORD pCurRecord = m_cScanResults.pScanRecords;
  64. m_cScanResults.pScanRecords = NULL;
  65. m_cScanResults.dwRecCount = 0;
  66. m_cScanResults.dwTime = 0;
  67. m_cScanResults.dwObjCount = 0;
  70. while( pCurRecord )
  71. {
  72. PSCAN_RECORD pTmpRecord = pCurRecord;
  73. pCurRecord = pCurRecord->pNext;
  75. if(pTmpRecord->pScanObject)
  76. delete pTmpRecord->pScanObject;
  77. delete pTmpRecord;
  78. }
  80. }
  82. // deep first search
  83. void CEngine::DFS(PCTSTR pszPathName)
  84. {
  85. if(FILE_ATTRIBUTE_DIRECTORY&GetFileAttributes(pszPathName))
  86. {
  87. CString strPath = pszPathName;
  88. WIN32_FIND_DATA FindFileData;
  89. bool bFinished = false; 
  90. HANDLE hFinder = FindFirstFile(strPath+"\*.*", &FindFileData);
  91. if(hFinder!=INVALID_HANDLE_VALUE)
  92. {
  93. while(!bFinished)
  94. {
  95. if( strcmp(FindFileData.cFileName, ".") && strcmp(FindFileData.cFileName, ".."))
  96. {
  97. DFS(strPath+"\"+FindFileData.cFileName);
  98. }
  100. if (!FindNextFile(hFinder, &FindFileData)) 
  101. bFinished = true;
  102. }
  103. FindClose(hFinder);
  104. }
  105. }
  106. else
  107. {
  108. ScanOneObject(pszPathName);
  109. }
  110. }
  112. bool CEngine::ScanOneObject(PCTSTR pszObjectName)
  113. {
  114. // debug use only
  115. // 所有对外显示应该集中在主程序
  116. printf("Scan %sn", pszObjectName);
  118. m_cScanResults.dwObjCount++;
  120. CMemFileObject cScanObj;
  121. cScanObj.m_strObjName = pszObjectName;
  122. if( !cScanObj.Open() )
  123. {
  124. // TODO: show error here.
  126. // debug use only
  127. printf("Error: "%s" open failed.n", pszObjectName);
  128. return false;
  129. }
  131. DWORD dwVID = m_pcVDB->Search(&cScanObj);
  132. if( dwVID )
  133. {
  134. PSCAN_RECORD pScanRecord = new SCAN_RECORD;
  135. if(pScanRecord)
  136. {
  137. CFileObject* pScanObj = new CFileObject(cScanObj);
  139. pScanRecord->dwVirusID = dwVID;
  140. pScanRecord->eResult = BR_WITH_VIRUS;
  141. pScanRecord->pScanObject= pScanObj;
  142. pScanRecord->pNext = m_cScanResults.pScanRecords;
  143. m_cScanResults.pScanRecords = pScanRecord;
  144. m_cScanResults.dwRecCount++;
  145. }
  146. }
  148. cScanObj.Close();
  150. return true;
  151. }