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

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

防火墙与安全工具

开发平台：

Visual C++

MyFireWall.cpp：源码内容
  1. // MyFireWall.cpp : Defines the class behaviors for the application.
  2. //
  4. #include "stdafx.h"
  5. #include "MyFireWall.h"
  6. #include "MyFireWallDlg.h"
  7. #include "Winver.h"
  9. #pragma comment(lib, "Version.lib") // 提供获得文件版本的函数
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. /////////////////////////////////////////////////////////////////////////////
  18. // CMyFireWallApp
  20. BEGIN_MESSAGE_MAP(CMyFireWallApp, CWinApp)
  21. //{{AFX_MSG_MAP(CMyFireWallApp)
  22. // NOTE - the ClassWizard will add and remove mapping macros here.
  23. //    DO NOT EDIT what you see in these blocks of generated code!
  24. //}}AFX_MSG
  25. ON_COMMAND(ID_HELP, CWinApp::OnHelp)
  26. END_MESSAGE_MAP()
  28. /////////////////////////////////////////////////////////////////////////////
  29. // CMyFireWallApp construction
  31. CMyFireWallApp::CMyFireWallApp()
  32. {
  33. // TODO: add construction code here,
  34. // Place all significant initialization in InitInstance
  35. }
  37. /////////////////////////////////////////////////////////////////////////////
  38. // The one and only CMyFireWallApp object
  40. CMyFireWallApp theApp;
  42. /////////////////////////////////////////////////////////////////////////////
  43. // CMyFireWallApp initialization
  45. BOOL CMyFireWallApp::InitInstance()
  46. {
  47. // 运行一次
  48. TCHAR szModule[] = L"MyFireWall";
  49. m_hSemaphore = ::CreateSemaphore(NULL, 0, 1, szModule);
  50. if(::GetLastError() == ERROR_ALREADY_EXISTS)
  51. {
  52. AfxMessageBox(L" MyFireWall 已经在运行！");
  53. return FALSE;
  54. }
  55. // 加载过滤文件
  56. if (!g_RuleFile.LoadRules())
  57. {
  58. AfxMessageBox(L"加载配置文件时失败！");
  59. return FALSE;
  60. }
  62. // 获得防火墙当前路径
  63. TCHAR szCurrentPath[MAX_PATH];
  64. GetModuleFileName(NULL, szCurrentPath, MAX_PATH);
  65. strCurrentPath = GetFilePath(szCurrentPath);
  67. // 加载LSP
  68. if (!IsProviderInstalled())
  69. {
  70. TCHAR szPathName[256];
  72. // 注意，安装LSP需要使用完整DLL路径。这样的话，CPIOControl类在加载DLL时也应使用
  73. // 完整路径，否则CPIOControl类加载的DLL不能和作为LSP的DLL共享内存
  75. wsprintf(szPathName, _T("%s%s"), strCurrentPath, _T("MyLsp.dll"));
  77. if(!InstallProvider(szPathName))
  78. {
  79. AfxMessageBox(L" 应用层过滤安装失败！"); 
  80. return FALSE;
  81. }
  82. }
  84. //创建Dll I/O控制对象，加载Dll模块
  85. g_pIoControl = new CPIOControl;
  87. //应用文件中的数据，设置应用层和核心层的过滤规则
  88. ApplyLspRules();
  89. ApplyKerRules();
  92. //加载皮肤
  93. InitializeSkin("Topax.ssk");
  95. AfxEnableControlContainer();
  97. // Standard initialization
  98. // If you are not using these features and wish to reduce the size
  99. //  of your final executable, you should remove from the following
  100. //  the specific initialization routines you do not need.
  102. #ifdef _AFXDLL
  103. Enable3dControls(); // Call this when using MFC in a shared DLL
  104. #else
  105. Enable3dControlsStatic(); // Call this when linking to MFC statically
  106. #endif
  108. CMyFireWallDlg dlg;
  109. m_pMainWnd = &dlg;
  110. int nResponse = dlg.DoModal();
  111. if (nResponse == IDOK)
  112. {
  113. // TODO: Place code here to handle when the dialog is
  114. // dismissed with OK
  115. }
  116. else if (nResponse == IDCANCEL)
  117. {
  118. // TODO: Place code here to handle when the dialog is
  119. //  dismissed with Cancel
  120. }
  122. // Since the dialog has been closed, return FALSE so that we exit the
  123. //  application, rather than start the application's message pump.
  124. return FALSE;
  125. }
  127. int CMyFireWallApp::ExitInstance() 
  128. {
  129. // TODO: Add your specialized code here and/or call the base class
  130. if(g_pIoControl != NULL)
  131. {
  132. g_pIoControl->SetWorkMode(PF_PASS_ALL);
  133. g_pIoControl->SetInstance(NULL, L"");
  134. delete g_pIoControl;
  135. }
  136. int nKerWorkMode = IM_PASS_ALL;
  137. PtSetWorkMode(&nKerWorkMode);
  138. IMClearRules();
  140. PtSetARPRules(FALSE);
  142. ::CloseHandle(m_hSemaphore);
  144. return CWinApp::ExitInstance();
  145. }
  147. BOOL CMyFireWallApp::ApplyFileData()
  148. {
  149. // 设置应用层工作模式
  150. g_pIoControl->SetWorkMode(g_RuleFile.m_header.ucLspWorkMode);
  152. // 设置应用层规则文件
  153. g_pIoControl->SetRuleFile(&g_RuleFile.m_header, g_RuleFile.m_pLspRules);
  155. // 设置核心层工作模式
  156. int nWorkMode = g_RuleFile.m_header.ucKerWorkMode;
  157. PtSetWorkMode(&nWorkMode);
  159. // 设置核心层规则文件
  160. IMClearRules();
  161. if(g_RuleFile.m_header.ucKerWorkMode == IM_START_FILTER)
  162. {
  163. if(!IMSetRules(g_RuleFile.m_pKerRules, g_RuleFile.m_header.ulKerRuleCount))
  164. {
  165. AfxMessageBox(L" 设置核心层规则出错！n");
  166. return FALSE;
  167. }
  168. }
  170. return TRUE;
  171. }
  173. BOOL CMyFireWallApp::ApplyLspRules()
  174. {
  175. // 设置应用层工作模式
  176. g_pIoControl->SetWorkMode(g_RuleFile.m_header.ucLspWorkMode);
  178. // 设置应用层规则文件
  179. g_pIoControl->SetRuleFile(&g_RuleFile.m_header, g_RuleFile.m_pLspRules);
  181. return TRUE;
  182. }
  184. BOOL CMyFireWallApp::ApplyKerRules()
  185. {
  186. // 设置核心层工作模式
  187. int nWorkMode = g_RuleFile.m_header.ucKerWorkMode;
  188. PtSetWorkMode(&nWorkMode);
  190. // 设置核心层规则文件
  191. IMClearRules();
  192. if(g_RuleFile.m_header.ucKerWorkMode == IM_START_FILTER)
  193. {
  194. if(!IMSetRules(g_RuleFile.m_pKerRules, g_RuleFile.m_header.ulKerRuleCount))
  195. {
  196. AfxMessageBox(L" 设置核心层规则出错！n");
  197. return FALSE;
  198. }
  199. }
  201. return TRUE;
  202. }
  204. BOOL CMyFireWallApp::ApplyIPRules(int nIndex)
  205. {
  206. int nPing = (int)g_RuleFile.m_pIPRules[0].bState *10 + (int)g_RuleFile.m_pIPRules[1].bState;
  207. // 设置Ping入和Ping出
  208. PtSetPing(&nPing);
  210. PassthruFilter RuleItem;
  212. // RPC
  213. if (nIndex == -1 || nIndex == 2)
  214. {
  215. if (g_RuleFile.m_pIPRules[2].bState == TRUE)
  216. {
  217. // 如果改为“允许”的话，就删除相应的规则
  218. for (int i=0; i<g_RuleFile.m_header.ulKerRuleCount; i++)
  219. {
  220. if (_tcscmp(g_RuleFile.m_pKerRules[i].szDescription, L"RPC") == 0)
  221. {
  222. g_RuleFile.DelKerRule(i);
  223. break;
  224. }
  225. }
  226. }
  227. else
  228. {
  229. // 如果是“禁止”，就添加相应的规则
  230. RuleItem.bDrop = !g_RuleFile.m_pIPRules[2].bState;
  231. RuleItem.protocol = g_RuleFile.m_pIPRules[2].protocol; // 1：ICMP，6：TCP，17：UDP
  232. RuleItem.sourceIP   = 0;
  233. RuleItem.sourcePort = 0;
  234. RuleItem.destinationIP   = 0;
  235. RuleItem.destinationPort = g_RuleFile.m_pIPRules[2].port;
  236. _tcscpy(RuleItem.szWebSiteURL,  L"");
  237. _tcscpy(RuleItem.szDescription, L"RPC");
  238. // 添加核心层规则
  239. g_RuleFile.AddKerRules(&RuleItem, 1);
  240. }
  241. }
  243. // DNS
  244. if (nIndex == -1 || nIndex == 3)
  245. {
  246. if (g_RuleFile.m_pIPRules[3].bState == TRUE)
  247. {
  248. // 如果改为“允许”的话，就删除相应的规则
  249. for (int i=0; i<g_RuleFile.m_header.ulKerRuleCount; i++)
  250. {
  251. if (_tcscmp(g_RuleFile.m_pKerRules[i].szDescription, L"DNS") == 0)
  252. {
  253. g_RuleFile.DelKerRule(i);
  254. break;
  255. }
  256. }
  257. }
  258. else
  259. {
  260. // 如果是“禁止”，就添加相应的规则
  261. RuleItem.bDrop = !g_RuleFile.m_pIPRules[3].bState;
  262. RuleItem.protocol = g_RuleFile.m_pIPRules[3].protocol; // 1：ICMP，6：TCP，17：UDP
  263. RuleItem.sourceIP   = 0;
  264. RuleItem.sourcePort = 0;
  265. RuleItem.destinationIP   = 0;
  266. RuleItem.destinationPort = g_RuleFile.m_pIPRules[3].port;
  267. _tcscpy(RuleItem.szWebSiteURL,  L"");
  268. _tcscpy(RuleItem.szDescription, L"DNS");
  269. // 添加到核心层规则
  270. g_RuleFile.AddKerRules(&RuleItem, 1);
  271. }
  272. }
  274. // 局域网共享
  275. if (nIndex == -1 || nIndex == 4)
  276. {
  277. if (g_RuleFile.m_pIPRules[4].bState == TRUE)
  278. {
  279. // 如果改为“允许”的话，就删除相应的规则
  280. for (int i=0; i<g_RuleFile.m_header.ulKerRuleCount; i++)
  281. {
  282. if (_tcscmp(g_RuleFile.m_pKerRules[i].szDescription, L"LAN139") == 0 || _tcscmp(g_RuleFile.m_pKerRules[i].szDescription, L"LAN445") == 0)
  283. {
  284. g_RuleFile.DelKerRule(i);
  285. }
  286. }
  287. }
  288. else
  289. {
  290. // 如果是“禁止”，就添加相应的规则
  291. // 端口139
  292. RuleItem.bDrop = !g_RuleFile.m_pIPRules[4].bState;
  293. RuleItem.protocol = g_RuleFile.m_pIPRules[4].protocol; // 1：ICMP，6：TCP，17：UDP
  294. RuleItem.sourceIP   = 0;
  295. RuleItem.sourcePort = 0;
  296. RuleItem.destinationIP   = 0;
  297. RuleItem.destinationPort = 139;
  298. _tcscpy(RuleItem.szWebSiteURL,  L"");
  299. _tcscpy(RuleItem.szDescription, L"LAN139");
  300. // 添加到核心层规则
  301. g_RuleFile.AddKerRules(&RuleItem, 1);
  303. // 端口445
  304. RuleItem.bDrop = !g_RuleFile.m_pIPRules[4].bState;
  305. RuleItem.protocol = g_RuleFile.m_pIPRules[4].protocol; // 1：ICMP，6：TCP，17：UDP
  306. RuleItem.sourceIP   = 0;
  307. RuleItem.sourcePort = 0;
  308. RuleItem.destinationIP   = 0;
  309. RuleItem.destinationPort = 445;
  310. _tcscpy(RuleItem.szWebSiteURL,  L"");
  311. _tcscpy(RuleItem.szDescription, L"LAN445");
  312. // 添加到核心层规则
  313. g_RuleFile.AddKerRules(&RuleItem, 1);
  314. }
  315. }
  317. // 保存规则
  318. g_RuleFile.SaveRules();
  320. // 应用到核心层
  321. theApp.ApplyKerRules();
  323. return TRUE;
  324. }