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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
PRuleFile.cpp
资源名称：MyFireWall.rar　[点击查看]
上传用户：zhuzhu0204
上传日期：2020-07-13
资源大小：13165k
文件大小：10k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

PRuleFile.cpp：源码内容
  1. //////////////////////////////////////////////////
  2. // PRuleFile.cpp
  4. #include "stdafx.h"
  6. #include "PRuleFile.h"
  9. CPRuleFile::CPRuleFile()
  10. {
  11. // 获取规则文件的完整路径
  12. TCHAR *p;
  13. ::GetFullPathName(RULE_FILE_NAME, MAX_PATH, m_szPathName, &p);
  15. m_hFile = INVALID_HANDLE_VALUE;
  17. // 为过滤规则预申请内存空间
  18. m_nLspMaxCount = 50;
  19. m_nKerMaxCount = 50;
  20. m_nIPMaxCount  = 10;
  21. m_pLspRules = new RULE_ITEM[m_nLspMaxCount];
  22. m_pKerRules = new PassthruFilter[m_nKerMaxCount];
  23. m_pIPRules  = new IPRule[m_nIPMaxCount];
  24. m_bLoad = FALSE;
  25. }
  27. CPRuleFile::~CPRuleFile()
  28. {
  29. if(m_hFile != INVALID_HANDLE_VALUE)
  30. ::CloseHandle(m_hFile);
  31. delete[] m_pLspRules;
  32. delete[] m_pKerRules;
  33. delete[] m_pIPRules;
  34. }
  37. void CPRuleFile::InitFileData()
  38. {
  39. // 初始化文件头
  40. wcscpy(m_header.szSignature, RULE_HEADER_SIGNATURE);
  41. m_header.ulHeaderLength = sizeof(m_header);
  43. m_header.ucMajorVer = RULE_HEADER_MAJOR;
  44. m_header.ucMinorVer = RULE_HEADER_MINOR;
  46. m_header.dwVersion = RULE_HEADER_VERSION;
  48. wcscpy(m_header.szWebURL, RULE_HEADER_WEB_URL); 
  50. wcscpy(m_header.szEmail, RULE_HEADER_EMAIL);
  52. m_header.ulLspRuleCount = 0;
  53. m_header.ulKerRuleCount = 0;
  54. m_header.ulIPRuleCount  = 0;
  56. m_header.ucLspWorkMode = PF_QUERY_ALL;
  57. m_header.ucKerWorkMode = IM_START_FILTER;
  59. m_header.bAutoStart = FALSE;
  60. m_header.bIsIPRuleSaved = FALSE;
  61. }
  63. // 将规则写入指定文件
  64. BOOL CPRuleFile::WriteRules(TCHAR *pszPathName) 
  65. {
  66. DWORD dw;
  67. if(m_hFile != INVALID_HANDLE_VALUE)
  68. {
  69. ::CloseHandle(m_hFile);
  70. }
  71. // 打开文件
  72. m_hFile = ::CreateFile(pszPathName, GENERIC_WRITE, 
  73. 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); 
  74. if(m_hFile != INVALID_HANDLE_VALUE)
  75. {
  76. // 写文件头
  77. ::WriteFile(m_hFile, &m_header, sizeof(m_header), &dw, NULL);
  78. // 写应用层过滤规则
  79. if(m_header.ulLspRuleCount > 0)
  80. {
  81. ::WriteFile(m_hFile, 
  82. m_pLspRules, m_header.ulLspRuleCount * sizeof(RULE_ITEM), &dw, NULL);
  83. }
  84. // 写核心层过滤规则
  85. if(m_header.ulKerRuleCount > 0)
  86. {
  87. ::WriteFile(m_hFile, 
  88. m_pKerRules, m_header.ulKerRuleCount * sizeof(PassthruFilter), &dw, NULL);
  89. }
  90. // 写IP规则
  91. if(m_header.ulIPRuleCount > 0)
  92. {
  93. ::WriteFile(m_hFile, 
  94. m_pIPRules, m_header.ulIPRuleCount * sizeof(IPRule), &dw, NULL);
  95. }
  97. ::CloseHandle(m_hFile);
  98. m_hFile = INVALID_HANDLE_VALUE;
  99. return TRUE;
  100. }
  101. return FALSE;
  102. }
  104. BOOL CPRuleFile::OpenFile()
  105. {
  106. // 首先保证文件已经存在
  107. if(::GetFileAttributes(m_szPathName) == -1)
  108. {
  109. InitFileData();
  110. if(!WriteRules(m_szPathName))
  111. return FALSE;
  112. }
  113. // 如果没有关闭，就关闭
  114. if(m_hFile != INVALID_HANDLE_VALUE)
  115. ::CloseHandle(m_hFile);
  117. // 以只读方式打开文件
  118. m_hFile = ::CreateFile(m_szPathName, GENERIC_READ, 
  119. FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); 
  121. return m_hFile != INVALID_HANDLE_VALUE;
  122. }
  124. ///////////////////////////////////////////////////////////////////////////////////////
  125. // 接口成员
  127. BOOL CPRuleFile::LoadRules()
  128. {
  129. // 先打开文件
  130. if((!OpenFile()) || (::SetFilePointer(m_hFile, 0, NULL, FILE_BEGIN) == -1))
  131. {
  132. return FALSE;
  133. }
  135. // 从文件中读取数据
  136. DWORD dw = 0;
  137. do
  138. {
  139. // 读文件头
  140. ::ReadFile(m_hFile, &m_header, sizeof(m_header), &dw, NULL);
  141. // 如果签名不正确，退出
  142. if((dw != sizeof(m_header)) || 
  143. (_tcscmp(m_header.szSignature, RULE_HEADER_SIGNATURE) != 0))
  144. {
  145. AfxMessageBox(L"签名出错！");
  146. break;
  147. }
  149. // 读应用层过滤规则
  150. if(m_header.ulLspRuleCount > 0)
  151. {
  152. if(m_header.ulLspRuleCount > (ULONG)m_nLspMaxCount)
  153. {
  154. m_nLspMaxCount = m_header.ulLspRuleCount;
  155. delete[] m_pLspRules;
  156. m_pLspRules = new RULE_ITEM[m_nLspMaxCount];
  157. }
  158. if(!::ReadFile(m_hFile, m_pLspRules, 
  159. m_header.ulLspRuleCount * sizeof(RULE_ITEM), &dw, NULL))
  160. break;
  161. }
  163. // 读核心层过滤规则
  164. if(m_header.ulKerRuleCount > 0)
  165. {
  166. if(m_header.ulKerRuleCount > (ULONG)m_nKerMaxCount)
  167. {
  168. m_nKerMaxCount = m_header.ulKerRuleCount;
  169. delete[] m_pKerRules;
  170. m_pKerRules = new PassthruFilter[m_nKerMaxCount];
  171. }
  172. if(!::ReadFile(m_hFile, m_pKerRules, 
  173. m_header.ulKerRuleCount * sizeof(PassthruFilter), &dw, NULL))
  174. break;
  175. }
  177. // 读IP规则
  178. if(m_header.ulIPRuleCount > 0)
  179. {
  180. if(m_header.ulIPRuleCount > (ULONG)m_nIPMaxCount)
  181. {
  182. m_nIPMaxCount = m_header.ulIPRuleCount;
  183. delete[] m_pIPRules;
  184. m_pIPRules = new IPRule[m_nIPMaxCount];
  185. }
  186. if(!::ReadFile(m_hFile, m_pIPRules, 
  187. m_header.ulIPRuleCount * sizeof(IPRule), &dw, NULL))
  188. break;
  189. }
  191. m_bLoad = TRUE;
  192. }
  193. while(FALSE);
  195. ::CloseHandle(m_hFile);
  196. m_hFile = INVALID_HANDLE_VALUE;
  198. return m_bLoad;
  199. }
  201. BOOL CPRuleFile::SaveRules()
  202. {
  203. // 如果没有加载，退出
  204. if(!m_bLoad)
  205. return FALSE;
  206. // 保存规则
  207. return WriteRules(m_szPathName);
  208. }
  210. BOOL CPRuleFile::AddLspRules(RULE_ITEM *pItem, int nCount)
  211. {
  212. if((pItem == NULL) || !m_bLoad)
  213. return FALSE;
  215. // 首先保证有足够大的内存空间
  216. if(m_header.ulLspRuleCount + nCount > (ULONG)m_nLspMaxCount)
  217. {
  218. m_nLspMaxCount = 2*(m_header.ulLspRuleCount + nCount);
  220. RULE_ITEM *pTmp = new RULE_ITEM[m_header.ulLspRuleCount];
  221. memcpy(pTmp, m_pLspRules, m_header.ulLspRuleCount);
  223. delete[] m_pLspRules;
  224. m_pLspRules = new RULE_ITEM[m_nLspMaxCount];
  225. memcpy(m_pLspRules, pTmp, m_header.ulLspRuleCount);
  226. delete[] pTmp;
  227. }
  229. // 添加规则
  230. memcpy(m_pLspRules + m_header.ulLspRuleCount, pItem, nCount * sizeof(RULE_ITEM));
  231. m_header.ulLspRuleCount += nCount;
  232. return TRUE;
  233. }
  235. BOOL CPRuleFile::AddKerRules(PassthruFilter *pItem, int nCount)
  236. {
  237. if((pItem == NULL) || !m_bLoad)
  238. return FALSE;
  240. // 首先保证有足够大的内存空间
  241. if(m_header.ulKerRuleCount + nCount > (ULONG)m_nKerMaxCount)
  242. {
  243. m_nKerMaxCount = 2*(m_header.ulKerRuleCount + nCount);
  245. PassthruFilter *pTmp = new PassthruFilter[m_header.ulKerRuleCount];
  246. memcpy(pTmp, m_pKerRules, m_header.ulKerRuleCount);
  248. delete[] m_pKerRules;
  249. m_pKerRules = new PassthruFilter[m_nKerMaxCount];
  250. memcpy(m_pKerRules, pTmp, m_header.ulKerRuleCount);
  251. delete[] pTmp;
  252. }
  253. // 添加规则
  254. memcpy(m_pKerRules + m_header.ulKerRuleCount, pItem, nCount * sizeof(PassthruFilter));
  255. m_header.ulKerRuleCount += nCount;
  256. return TRUE;
  257. }
  259. BOOL CPRuleFile::AddIPRules(IPRule *pItem, int nCount)
  260. {
  261. if((pItem == NULL) || !m_bLoad)
  262. return FALSE;
  264. // 首先保证有足够大的内存空间
  265. if(m_header.ulIPRuleCount + nCount > (ULONG)m_nIPMaxCount)
  266. {
  267. m_nIPMaxCount = 2*(m_header.ulIPRuleCount + nCount);
  269. IPRule *pTmp = new IPRule[m_header.ulIPRuleCount];
  270. memcpy(pTmp, m_pIPRules, m_header.ulIPRuleCount);
  272. delete[] m_pIPRules;
  273. m_pIPRules = new IPRule[m_nIPMaxCount];
  274. memcpy(m_pIPRules, pTmp, m_header.ulIPRuleCount);
  275. delete[] pTmp;
  276. }
  277. // 添加规则
  278. memcpy(m_pIPRules + m_header.ulIPRuleCount, pItem, nCount * sizeof(IPRule));
  279. m_header.ulIPRuleCount += nCount;
  280. return TRUE;
  281. }
  283. BOOL CPRuleFile::DelLspRule(int nIndex)
  284. {
  285. if(((ULONG)nIndex >= m_header.ulLspRuleCount) || !m_bLoad)
  286. return FALSE;
  287. // 删除一个成员
  288. memcpy(&m_pLspRules[nIndex], 
  289. &m_pLspRules[nIndex + 1], (m_header.ulLspRuleCount - nIndex) * sizeof(RULE_ITEM));
  291. m_header.ulLspRuleCount --;
  292. return TRUE;
  293. }
  295. BOOL CPRuleFile::DelKerRule(int nIndex)
  296. {
  297. if(((ULONG)nIndex >= m_header.ulKerRuleCount) || !m_bLoad)
  298. return FALSE;
  299. // 删除一个成员
  300. memcpy(&m_pKerRules[nIndex], 
  301. &m_pKerRules[nIndex + 1], (m_header.ulKerRuleCount - nIndex) * sizeof(PassthruFilter));
  303. m_header.ulKerRuleCount --;
  304. return TRUE;
  305. }
  307. BOOL CPRuleFile::ModifyLspRule(TCHAR *szAppPath)
  308. {
  309. BOOL bRet;
  311. // 先打开文件
  312. if((!OpenFile()) || (::SetFilePointer(m_hFile, 0, NULL, FILE_BEGIN) == -1))
  313. {
  314. return FALSE;
  315. }
  317. // 从文件中读取数据
  318. DWORD dw = 0;
  319. do
  320. {
  321. // 读文件头
  322. ::ReadFile(m_hFile, &m_header, sizeof(m_header), &dw, NULL);
  323. // 如果签名不正确，退出
  324. if((dw != sizeof(m_header)) || 
  325. (_tcscmp(m_header.szSignature, RULE_HEADER_SIGNATURE) != 0))
  326. {
  327. AfxMessageBox(L"签名出错！");
  328. break;
  329. }
  331. // 读应用层过滤规则
  332. if(m_header.ulLspRuleCount > 0)
  333. {
  334. if(m_header.ulLspRuleCount > (ULONG)m_nLspMaxCount)
  335. {
  336. m_nLspMaxCount = m_header.ulLspRuleCount;
  337. delete[] m_pLspRules;
  338. m_pLspRules = new RULE_ITEM[m_nLspMaxCount];
  339. }
  340. if(!::ReadFile(m_hFile, m_pLspRules, m_header.ulLspRuleCount * sizeof(RULE_ITEM), &dw, NULL))
  341. {
  342. break;
  343. }
  344. }
  346. // 查找与应用程序相关的过滤规则
  347. for (int i = 0; i < (int)m_header.ulLspRuleCount; i++)
  348. {
  349. if (_tcscmp(m_pLspRules[i].szApplication,szAppPath) == 0)
  350. {
  351. m_pLspRules[i].ucAction = 1 - m_pLspRules[i].ucAction; // 改变应用程序的状态
  352. SaveRules(); // 保存修改后的过滤规则
  353. break;
  354. }
  355. }
  357. if (i >= (int)m_header.ulLspRuleCount)
  358. {
  359. AfxMessageBox(L"没有找到与该程序相关的过滤规则！");
  360. bRet = FALSE;
  361. }
  362. else
  363. {
  364. bRet = TRUE;
  365. }
  367. }
  368. while(FALSE);
  370. ::CloseHandle(m_hFile);
  371. m_hFile = INVALID_HANDLE_VALUE;
  373. return bRet;
  374. }
  378. BOOL CPRuleFile::DelKerRuleByURL(TCHAR *szURL)
  379. {
  380. // 先打开文件
  381. if((!OpenFile()) || (::SetFilePointer(m_hFile, 0, NULL, FILE_BEGIN) == -1))
  382. {
  383. return FALSE;
  384. }
  386. // 从文件中读取数据
  387. DWORD dw = 0;
  388. BOOL  bRet = FALSE;
  389. do
  390. {
  391. // 读文件头
  392. ::ReadFile(m_hFile, &m_header, sizeof(m_header), &dw, NULL);
  393. // 如果签名不正确，退出
  394. if((dw != sizeof(m_header)) || 
  395. (_tcscmp(m_header.szSignature, RULE_HEADER_SIGNATURE) != 0))
  396. {
  397. AfxMessageBox(L"签名出错！");
  398. break;
  399. }
  401. // 读应用层过滤规则
  402. if(m_header.ulLspRuleCount > 0)
  403. {
  404. if(m_header.ulLspRuleCount > (ULONG)m_nLspMaxCount)
  405. {
  406. m_nLspMaxCount = m_header.ulLspRuleCount;
  407. delete[] m_pLspRules;
  408. m_pLspRules = new RULE_ITEM[m_nLspMaxCount];
  409. }
  410. if(!::ReadFile(m_hFile, m_pLspRules, 
  411. m_header.ulLspRuleCount * sizeof(RULE_ITEM), &dw, NULL))
  412. break;
  413. }
  415. // 读核心层过滤规则
  416. if(m_header.ulKerRuleCount > 0)
  417. {
  418. if(m_header.ulKerRuleCount > (ULONG)m_nKerMaxCount)
  419. {
  420. m_nKerMaxCount = m_header.ulKerRuleCount;
  421. delete[] m_pKerRules;
  422. m_pKerRules = new PassthruFilter[m_nKerMaxCount];
  423. }
  424. if(!::ReadFile(m_hFile, m_pKerRules, 
  425. m_header.ulKerRuleCount * sizeof(PassthruFilter), &dw, NULL))
  426. break;
  427. }
  429. // 从核心层规则中找到与该URL匹配的规则
  430. for (int i=0; i<(int)m_header.ulKerRuleCount; i++)
  431. {
  432. if (_tcscmp(m_pKerRules[i].szWebSiteURL, szURL) == 0)
  433. {
  434. DelKerRule(i);
  435. i--; // 因为规则减少了1个，所以计数要减1。
  436. }
  437. }
  439. // 保存修改后的规则文件
  440. SaveRules();
  442. bRet = TRUE;
  443. }
  444. while(FALSE);
  446. ::CloseHandle(m_hFile);
  447. m_hFile = INVALID_HANDLE_VALUE;
  449. return bRet;
  450. }