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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
MainFrm.cpp
资源名称：FirewallFHK_src.zip　[点击查看]
上传用户：wlquartz
上传日期：2022-03-04
资源大小：89k
文件大小：10k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

MainFrm.cpp：源码内容
  1. // MainFrm.cpp : implementation of the CMainFrame class
  2. //
  4. #include "stdafx.h"
  5. #include "FirewallApp.h"
  7. #include "MainFrm.h"
  8. #include "RuleDlg.h"
  9. #include "FirewallAppDoc.h"
  10. #include "FirewallAppView.h"
  11. #include "SockUtil.h"
  12. #include "rules.h"
  14. #ifdef _DEBUG
  15. #define new DEBUG_NEW
  16. #undef THIS_FILE
  17. static char THIS_FILE[] = __FILE__;
  18. #endif
  20. /////////////////////////////////////////////////////////////////////////////
  21. // CMainFrame
  23. IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
  25. BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
  26. //{{AFX_MSG_MAP(CMainFrame)
  27. ON_WM_CREATE()
  28. ON_COMMAND(ID_BUTTONSTART, OnButtonstart)
  29. ON_COMMAND(ID_BUTTONADD, OnButtonadd)
  30. ON_COMMAND(ID_BUTTONDEL, OnButtondel)
  31. ON_COMMAND(ID_BUTTONDESINSTALL, OnButtondesinstall)
  32. ON_COMMAND(ID_BUTTONINSTALL, OnButtoninstall)
  33. ON_COMMAND(ID_BUTTONSTOP, OnButtonstop)
  34. ON_UPDATE_COMMAND_UI(ID_BUTTONSTART, OnUpdateButtonstart)
  35. ON_UPDATE_COMMAND_UI(ID_BUTTONSTOP, OnUpdateButtonstop)
  36. ON_COMMAND(IDMENU_ADDRULE, OnMenuAddrule)
  37. ON_COMMAND(IDMENU_DELRULE, OnMenuDelrule)
  38. ON_COMMAND(IDMENU_INSTALLRULES, OnMenuInstallrules)
  39. ON_COMMAND(IDMENU_UNINSTALLRULES, OnMenuUninstallrules)
  40. ON_COMMAND(ID_MENUSTART, OnMenustart)
  41. ON_UPDATE_COMMAND_UI(ID_MENUSTART, OnUpdateMenustart)
  42. ON_COMMAND(ID_MENUSTOP, OnMenustop)
  43. ON_UPDATE_COMMAND_UI(ID_MENUSTOP, OnUpdateMenustop)
  44. ON_COMMAND(ID_APP_EXIT, OnAppExit)
  45. ON_COMMAND(IDMENU_LOADRULES, OnLoadRules)
  46. ON_COMMAND(IDMENU_SAVERULES, OnSaveRules)
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  50. static UINT indicators[] =
  51. {
  52. ID_SEPARATOR,           // status line indicator
  53. ID_INDICATOR_CAPS,
  54. ID_INDICATOR_NUM,
  55. ID_INDICATOR_SCRL,
  56. };
  58. /////////////////////////////////////////////////////////////////////////////
  59. // CMainFrame construction/destruction
  61. CMainFrame::CMainFrame()
  62. {
  63. started = FALSE;
  65. // Cargo el driver IpFilterDriver y lo pongo en estado Start.
  66. ipFltDrv.LoadDriver("IpFilterDriver", "System32\Drivers\IpFltDrv.sys", NULL, TRUE);
  68. // No debo desregistrar el driver una vez salgo de la aplicacion.
  69. ipFltDrv.SetRemovable(FALSE);
  71. // Cargo el driver de filtrado
  72. if(filterDriver.LoadDriver("DrvFltIp", NULL, NULL, TRUE) != DRV_SUCCESS)
  73. {
  74. AfxMessageBox("Error loading the driver.");
  76. exit(-1);
  77. }
  79. }
  81. CMainFrame::~CMainFrame()
  82. {
  84. }
  86. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  87. {
  88. if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
  89. return -1;
  91. if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
  92. | CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
  93. !m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
  94. {
  95. TRACE0("Failed to create toolbarn");
  96. return -1;      // fail to create
  97. }
  99. if (!m_wndStatusBar.Create(this) ||
  100. !m_wndStatusBar.SetIndicators(indicators,
  101.   sizeof(indicators)/sizeof(UINT)))
  102. {
  103. TRACE0("Failed to create status barn");
  104. return -1;      // fail to create
  105. }
  107. // TODO: Delete these three lines if you don't want the toolbar to
  108. //  be dockable
  109. // m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
  110. // EnableDocking(CBRS_ALIGN_ANY);
  111. // DockControlBar(&m_wndToolBar);
  113. SetWindowText("Firewall FHK");
  115. return 0;
  116. }
  118. BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
  119. {
  120. if( !CFrameWnd::PreCreateWindow(cs) )
  121. return FALSE;
  124. //para quitar el untitled del titulo de la aplicacion
  125. cs.style &= ~ FWS_ADDTOTITLE;
  127. return TRUE;
  128. }
  130. /////////////////////////////////////////////////////////////////////////////
  131. // CMainFrame diagnostics
  133. #ifdef _DEBUG
  134. void CMainFrame::AssertValid() const
  135. {
  136. CFrameWnd::AssertValid();
  137. }
  139. void CMainFrame::Dump(CDumpContext& dc) const
  140. {
  141. CFrameWnd::Dump(dc);
  142. }
  144. #endif //_DEBUG
  146. /////////////////////////////////////////////////////////////////////////////
  147. // CMainFrame message handlers
  150. void CMainFrame::OnButtonstart() 
  151. {
  152. if(filterDriver.WriteIo(START_IP_HOOK, NULL, 0) != DRV_ERROR_IO)
  153. {
  154. started = TRUE;
  155. }
  157. }
  159. void CMainFrame::OnButtonstop() 
  160. {
  161. if(filterDriver.WriteIo(STOP_IP_HOOK, NULL, 0) != DRV_ERROR_IO)
  162. {
  163. started = FALSE;
  164. }
  165. }
  167. void CMainFrame::OnButtonadd() 
  168. {
  169. CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
  170. CRuleDlg dlg;
  172. // Compruebo que no he llegado al limite de numero de reglas
  173. if(doc->nRules < MAX_RULES )
  174. {
  175. if(dlg.DoModal() == IDOK)
  176. {
  177. // Primero a馻do la regla a mi lista de reglas
  178. if(doc->AddRule(dlg.srcIp, dlg.srcMask, dlg.srcPort, dlg.dstIp, dlg.dstMask, dlg.dstPort, dlg.protocol, dlg.cAction) != 0)
  179. AfxMessageBox("Error adding the rule");
  181. else
  182. {
  183. // Actualizo el control de visualizacion de reglas
  184. CFirewallAppView *view = (CFirewallAppView *)GetActiveView();
  186. view->UpdateList();
  187. }
  189. }
  190. }
  192. else
  193. AfxMessageBox("No more Rules are allowed.");
  194. }
  196. void CMainFrame::OnButtondel() 
  197. {
  198. CFirewallAppView *view = (CFirewallAppView *)GetActiveView();
  201. // Veo la posici髇 de la regla en la lista
  202. POSITION pos = view->m_rules.GetFirstSelectedItemPosition();
  203. if (pos == NULL)
  204. {
  205. AfxMessageBox("Select a Rule, please.");
  207. return;
  208. }
  210. int position;
  211. position = view->m_rules.GetNextSelectedItem(pos);
  213. CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
  214. doc->DeleteRule(position);
  217. // Actualizo la vista
  218. view->UpdateList();
  220. }
  223. void CMainFrame::OnButtoninstall() 
  224. {
  225. CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
  227. unsigned int i;
  228. DWORD result;
  230. // Le envio las reglas una a una al driver
  231. for(i=0;i<doc->nRules;i++)
  232. {
  233. result = AddFilterToFw(doc->rules[i].sourceIp, 
  234.    doc->rules[i].sourceMask,
  235.    doc->rules[i].sourcePort,
  236.    doc->rules[i].destinationIp,
  237.    doc->rules[i].destinationMask,
  238.    doc->rules[i].destinationPort,
  239.    doc->rules[i].protocol, 
  240.    doc->rules[i].action);
  243. if (!result) 
  244. break;
  245. }
  247. }
  250. void CMainFrame::OnButtondesinstall() 
  251. {
  252. if(filterDriver.WriteIo(CLEAR_FILTER, NULL, 0) == DRV_ERROR_IO)
  253. {
  254. AfxMessageBox("Error uninstalling rules.");
  255. }
  257. }
  260. void CMainFrame::OnUpdateButtonstart(CCmdUI* pCmdUI) 
  261. {
  262. if(started)
  263. pCmdUI->Enable(FALSE);
  265. else
  266. pCmdUI->Enable(TRUE);
  267. }
  269. void CMainFrame::OnUpdateButtonstop(CCmdUI* pCmdUI) 
  270. {
  271. if(started)
  272. pCmdUI->Enable(TRUE);
  274. else
  275. pCmdUI->Enable(FALSE);
  276. }
  278. void CMainFrame::OnMenuAddrule() 
  279. {
  280. OnButtonadd();
  281. }
  283. void CMainFrame::OnMenuDelrule() 
  284. {
  285. OnButtondel();
  286. }
  288. void CMainFrame::OnMenuInstallrules() 
  289. {
  290. OnButtoninstall();
  291. }
  293. void CMainFrame::OnMenuUninstallrules() 
  294. {
  295. OnButtondesinstall();
  297. }
  299. void CMainFrame::OnMenustart() 
  300. {
  301. OnButtonstart();
  302. }
  304. void CMainFrame::OnUpdateMenustart(CCmdUI* pCmdUI) 
  305. {
  306. if(started)
  307. pCmdUI->Enable(FALSE);
  309. else
  310. pCmdUI->Enable(TRUE);
  312. }
  314. void CMainFrame::OnMenustop() 
  315. {
  316. OnButtonstop();
  317. }
  319. void CMainFrame::OnUpdateMenustop(CCmdUI* pCmdUI) 
  320. {
  321. if(started)
  322. pCmdUI->Enable(TRUE);
  324. else
  325. pCmdUI->Enable(FALSE);
  327. }
  329. void CMainFrame::OnAppExit() 
  330. {
  332. }
  334. BOOL CMainFrame::AddFilterToFw(unsigned long srcIp, 
  335.    unsigned long srcMask,
  336.    unsigned short srcPort, 
  337.    unsigned long dstIp,
  338.    unsigned long dstMask,
  339.    unsigned short dstPort,
  340.    unsigned int protocol,
  341.    int action)
  342. {
  343. IPFilter pf;
  346. pf.protocol = protocol;
  348. pf.destinationIp = dstIp;
  349. pf.sourceIp  = srcIp;
  351. pf.destinationMask = dstMask;
  352. pf.sourceMask    = srcMask;
  354. pf.destinationPort = htons(dstPort);
  355. pf.sourcePort    = htons(srcPort);
  357. pf.drop = action;
  359. // Envio la regla al driver
  360. DWORD result = filterDriver.WriteIo(ADD_FILTER, &pf, sizeof(pf));
  362. if (result != DRV_SUCCESS) 
  363. {
  364. AfxMessageBox("Error adding the rule.");
  366. return FALSE;
  367. }
  369. else
  370. return TRUE;
  371. }
  373. void CMainFrame::OnLoadRules() 
  374. {
  375. CFile file;
  376. CFileException e;
  377. DWORD nRead;
  379. CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
  381. CFileDialog dg(TRUE,NULL, NULL, OFN_HIDEREADONLY | OFN_CREATEPROMPT,"Rule Files(*.rul)|*.rul|all(*.*)|*.*||", NULL);
  382. if(dg.DoModal()==IDCANCEL)
  383. return;
  385. CString nf=dg.GetPathName();
  387. if(nf.GetLength() == 0)
  388. {
  389. AfxMessageBox("This file name isn't valid");
  391. return;
  392. }
  394. if( !file.Open(nf, CFile::modeRead, &e ) )
  395. {
  396. AfxMessageBox("Error openning the file.");
  398. return;
  399. }
  401. doc->ResetRules();
  402. RuleInfo rule;
  404.     do
  405.     {
  406.   nRead = file.Read(&rule, sizeof(RuleInfo));
  408. if(nRead == 0)
  409. break;
  411. if(doc->AddRule(rule.sourceIp,
  412.         rule.sourceMask,
  413. rule.sourcePort,
  414. rule.destinationIp,
  415. rule.destinationMask,
  416. rule.destinationPort,
  417. rule.protocol,    
  418. rule.action)  != 0)
  419. {
  421. AfxMessageBox("Error adding the rule.");
  423. break;
  424. }
  426.     }while (1);
  430. // Actualizo el control de visualizacion de reglas
  431. CFirewallAppView *view = (CFirewallAppView *)GetActiveView();
  433. view->UpdateList();
  434. }
  436. void CMainFrame::OnSaveRules() 
  437. {
  438. CFirewallAppDoc *doc = (CFirewallAppDoc *)GetActiveDocument();
  440. if(doc->nRules == 0)
  441. {
  442. AfxMessageBox("There aren't rules");
  444. return;
  445. }
  448. CFileDialog dg(FALSE, NULL, NULL, OFN_HIDEREADONLY | OFN_CREATEPROMPT,"Rule Files(*.rul)|*.rul|all(*.*)|*.*||", NULL);
  449. if(dg.DoModal()==IDCANCEL)
  450. return;
  452. CString nf=dg.GetPathName();
  454. if(nf.GetLength() == 0)
  455. {
  456. AfxMessageBox("This file name isn't valid");
  458. return;
  459. }
  461. CFile file;
  462. CFileException e;
  464. if( !file.Open( nf, CFile::modeCreate | CFile::modeWrite, &e ) )
  465. {
  466. AfxMessageBox("Error openning the file.");
  468. return;
  469. }
  472. unsigned int i;
  474. // Le envio las reglas una a una al driver
  475. for(i=0;i<doc->nRules;i++)
  476. {
  477. file.Write(&doc->rules[i], sizeof(RuleInfo));
  478. }
  480. file.Close();
  481. }