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

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

防火墙与安全工具

开发平台：

Visual C++

MyFireWallDlg.cpp：源码内容
  1. // MyFireWallDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "MyFireWall.h"
  6. #include "MyFireWallDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. extern CMyFireWallApp theApp;
  16. #define IDM_SHOW_WINDOW  11     //显示主界面
  18. DWORD WINAPI GetFlowRateThread(LPVOID lpParam)
  19. {
  20. CMyFireWallDlg *pMyFireWallDlg = (CMyFireWallDlg *)lpParam;
  22. HANDLE hControlDevice = PtOpenControlDevice();
  23. CIMAdapters adapters;
  24. if(!adapters.EnumAdapters(hControlDevice))
  25. return ERROR_SUCCESS;
  27. HANDLE hAdapter;
  29. while (TRUE)
  30. {
  31. PassthruStatistics PassthruStats;
  32. for(int i=0; i<adapters.m_nAdapters; i++)
  33. {
  34. // 打开下层适配器
  35. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  36. if(hAdapter != INVALID_HANDLE_VALUE)
  37. {
  38. PassthruStatistics Stats;
  39. BOOL bRet = PtQueryStatistics(hAdapter, &Stats);
  40. if(bRet == FALSE)
  41. {
  42. AfxMessageBox(L"返回值位 FALSE");
  43. ::CloseHandle(hAdapter);
  44. ::CloseHandle(hControlDevice);
  45. break;
  46. }
  47. PassthruStats.nSendFlowRate  = Stats.nSendFlowRate;
  48. PassthruStats.nRecvFlowRate  = Stats.nRecvFlowRate;
  49. PassthruStats.nMPSendPktsCt  = Stats.nMPSendPktsCt;
  50. PassthruStats.nMPSendPktsDropped = Stats.nMPSendPktsDropped;
  51. PassthruStats.nPTRcvCt  = Stats.nPTRcvCt;
  52. PassthruStats.nPTRcvDropped  = Stats.nPTRcvDropped;
  53. PassthruStats.nPTRcvPktCt  = Stats.nPTRcvPktCt;
  54. PassthruStats.nPTRcvPktDropped  = Stats.nPTRcvPktDropped;
  55. ::CloseHandle(hAdapter);
  56. }
  57. }
  58. // 
  59. //  CString strFlowRate;
  60. //  strFlowRate.Format(L"发送流量：%d kbpsrn接收流量：%d KbpsrnnMPSendPktsCt: %drnnMPSendPktsDropped: %drnnPTRcvCt: %drnnPTRcvDropped: %drnnPTRcvPktCt: %drnnPTRcvPktDropped: %d",
  61. //  PassthruStats.nSendFlowRate*8/1024, 
  62. //  PassthruStats.nRecvFlowRate*8/1024,
  63. //  PassthruStats.nMPSendPktsCt,
  64. //  PassthruStats.nMPSendPktsDropped,
  65. //  PassthruStats.nPTRcvCt,
  66. //  PassthruStats.nPTRcvDropped,
  67. //  PassthruStats.nPTRcvPktCt,
  68. //  PassthruStats.nPTRcvPktDropped
  69. //  );
  70. //  pMyFireWallDlg->m_WorkStatusPage.SetDlgItemText(IDC_FLOW_RATE, strFlowRate);
  71. // 
  73. Sleep(1000);
  74. }
  75. ::CloseHandle(hControlDevice);
  76. }
  78. /////////////////////////////////////////////////////////////////////////////
  79. // CAboutDlg dialog used for App About
  81. class CAboutDlg : public CDialog
  82. {
  83. public:
  84. CAboutDlg();
  86. // Dialog Data
  87. //{{AFX_DATA(CAboutDlg)
  88. enum { IDD = IDD_ABOUTBOX };
  89. //}}AFX_DATA
  91. // ClassWizard generated virtual function overrides
  92. //{{AFX_VIRTUAL(CAboutDlg)
  93. protected:
  94. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  95. //}}AFX_VIRTUAL
  97. // Implementation
  98. protected:
  99. //{{AFX_MSG(CAboutDlg)
  100. //}}AFX_MSG
  101. DECLARE_MESSAGE_MAP()
  102. };
  104. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  105. {
  106. //{{AFX_DATA_INIT(CAboutDlg)
  107. //}}AFX_DATA_INIT
  108. }
  110. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  111. {
  112. CDialog::DoDataExchange(pDX);
  113. //{{AFX_DATA_MAP(CAboutDlg)
  114. //}}AFX_DATA_MAP
  115. }
  117. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  118. //{{AFX_MSG_MAP(CAboutDlg)
  119. // No message handlers
  120. //}}AFX_MSG_MAP
  121. END_MESSAGE_MAP()
  123. /////////////////////////////////////////////////////////////////////////////
  124. // CMyFireWallDlg dialog
  126. CMyFireWallDlg::CMyFireWallDlg(CWnd* pParent /*=NULL*/)
  127. : CDialog(CMyFireWallDlg::IDD, pParent)
  128. {
  129. //{{AFX_DATA_INIT(CMyFireWallDlg)
  130. // NOTE: the ClassWizard will add member initialization here
  131. //}}AFX_DATA_INIT
  132. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  133. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  134. }
  136. void CMyFireWallDlg::DoDataExchange(CDataExchange* pDX)
  137. {
  138. CDialog::DoDataExchange(pDX);
  139. //{{AFX_DATA_MAP(CMyFireWallDlg)
  140. // NOTE: the ClassWizard will add DDX and DDV calls here
  141. //}}AFX_DATA_MAP
  142. }
  144. BEGIN_MESSAGE_MAP(CMyFireWallDlg, CDialog)
  145. //{{AFX_MSG_MAP(CMyFireWallDlg)
  146. ON_WM_SYSCOMMAND()
  147. ON_WM_PAINT()
  148. ON_WM_QUERYDRAGICON()
  149. ON_COMMAND(IDM_BLUE, OnSkinBlue)
  150. ON_COMMAND(IDM_GREEN, OnSkinGreen)
  151. ON_COMMAND(IDM_WHITE, OnSkinWhite)
  152. ON_COMMAND(IDM_EXIT, OnMenuExit)
  153. ON_COMMAND(IDM_LSP_RULE, OnMenuLspRule)
  154. ON_COMMAND(IDM_KER_RULE, OnMenuKerRule)
  155. ON_COMMAND(IDM_CUTDOWN_NETWORK, OnMenuCutdownNetwork)
  156. ON_COMMAND(IDM_STOP_PROTECT, OnMenuStopProtect)
  157. //}}AFX_MSG_MAP
  158. ON_MESSAGE(PM_QUERY_ACL_NOTIFY, OnQueryAcl)
  159. ON_MESSAGE(PM_SESSION_NOTIFY, OnSessionNotify)
  160. ON_MESSAGE(QUERY_DLG_DESTROY, OnQueryDlgDestroy)
  161. ON_MESSAGE(UPDATE_ACCESS_CTRL_LIST, OnUpdateAccessCtrlList)
  162. ON_MESSAGE(UPDATE_PACKET_MONITOR_LIST, OnUpdatePacketMonitorList)
  163. ON_MESSAGE(UPDATE_WEBSITE_LIST, OnUpdateWebSiteList)
  164. ON_MESSAGE(UPDATE_IPRULE_LIST, OnUpdateIPRuleList)
  165. ON_COMMAND(IDM_SHOW_WINDOW,ShowMainWindow)
  166. ON_MESSAGE(WM_TO_TRAY,OnShowTask)
  167. END_MESSAGE_MAP()
  169. /////////////////////////////////////////////////////////////////////////////
  170. // CMyFireWallDlg message handlers
  172. BOOL CMyFireWallDlg::OnInitDialog()
  173. {
  174. CDialog::OnInitDialog();
  176. // Add "About..." menu item to system menu.
  178. // IDM_ABOUTBOX must be in the system command range.
  179. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  180. ASSERT(IDM_ABOUTBOX < 0xF000);
  182. CMenu* pSysMenu = GetSystemMenu(FALSE);
  183. if (pSysMenu != NULL)
  184. {
  185. CString strAboutMenu;
  186. strAboutMenu.LoadString(IDS_ABOUTBOX);
  187. if (!strAboutMenu.IsEmpty())
  188. {
  189. pSysMenu->AppendMenu(MF_SEPARATOR);
  190. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  191. }
  192. }
  194. // Set the icon for this dialog.  The framework does this automatically
  195. //  when the application's main window is not a dialog
  196. SetIcon(m_hIcon, TRUE); // Set big icon
  197. SetIcon(m_hIcon, FALSE); // Set small icon
  199. // TODO: Add extra initialization here
  201. // 设置 DLL 对应的主对话框句柄
  202. TCHAR szPath[256];
  203. ::GetModuleFileName(NULL, szPath, 256);
  204. g_pIoControl->SetInstance(m_hWnd, szPath);
  207. // 保存主对话框句柄到一个全局变量
  208. g_hMainWnd = m_hWnd;
  210. m_pQueryDlg = NULL;
  211. m_pStartQueryDlg = NULL;
  213. m_Sheet.AddPage(&m_WorkStatusPage);
  214. m_Sheet.AddPage(&m_AppMonitorPage);
  215. m_Sheet.AddPage(&m_AccessCtrlPage);
  216. m_Sheet.AddPage(&m_PacketMonitorPage);
  217. m_Sheet.AddPage(&m_IPRulePage);
  218. m_Sheet.AddPage(&m_WebSiteSettingPage);
  219. m_Sheet.AddPage(&m_ARPRulePage);
  220. m_Sheet.Create(this,WS_CHILD|WS_VISIBLE,0);
  221. m_Sheet.ModifyStyleEx(0,WS_EX_CONTROLPARENT);
  222. m_Sheet.ModifyStyle(0,WS_TABSTOP);
  223. CRect rect;
  224. m_Sheet.GetWindowRect(&rect);
  225. ScreenToClient(&rect);
  226. m_Sheet.SetWindowPos(NULL,rect.left-4,rect.top-5,0,0,SWP_NOZORDER|SWP_NOSIZE|SWP_NOACTIVATE);
  228. m_Sheet.SetActivePage(&m_ARPRulePage);
  229. m_Sheet.SetActivePage(&m_WebSiteSettingPage);
  230. m_Sheet.SetActivePage(&m_IPRulePage);
  231. m_Sheet.SetActivePage(&m_PacketMonitorPage);
  232. m_Sheet.SetActivePage(&m_AccessCtrlPage);
  233. m_Sheet.SetActivePage(&m_AppMonitorPage);
  234. m_Sheet.SetActivePage(&m_WorkStatusPage);
  236. // 显示应用层规则到 AccessCtrlList
  237. OnUpdateAccessCtrlList(0, 0);
  238. // 显示核心层规则到 PacketMonitorList
  239. OnUpdatePacketMonitorList(0, 0);
  240. // 显示网站黑名单到 WebSiteList
  241. OnUpdateWebSiteList(0, 0);
  243. // 获得日志文件路径
  244. char szPathName[256];
  245. sprintf(szPathName, ("%s%s"), strCurrentPath, ("MyFireWall.log"));
  246. // 打开日志文件
  247. if((g_pLogFile = fopen(szPathName, "a")) == NULL)
  248. {
  249. AfxMessageBox(_T("MyFireWall.exe 打开日志文件出错！"));
  250. }
  252. return TRUE;  // return TRUE  unless you set the focus to a control
  253. }
  255. void CMyFireWallDlg::OnSysCommand(UINT nID, LPARAM lParam)
  256. {
  257. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  258. {
  259. CAboutDlg dlgAbout;
  260. dlgAbout.DoModal();
  261. }
  262. else
  263. {
  264. CDialog::OnSysCommand(nID, lParam);
  265. }
  266. }
  268. // If you add a minimize button to your dialog, you will need the code below
  269. //  to draw the icon.  For MFC applications using the document/view model,
  270. //  this is automatically done for you by the framework.
  272. void CMyFireWallDlg::OnPaint() 
  273. {
  274. // 设置 DLL 对应的主对话框句柄
  275. if (IsIconic())
  276. {
  277. CPaintDC dc(this); // device context for painting
  279. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  281. // Center icon in client rectangle
  282. int cxIcon = GetSystemMetrics(SM_CXICON);
  283. int cyIcon = GetSystemMetrics(SM_CYICON);
  284. CRect rect;
  285. GetClientRect(&rect);
  286. int x = (rect.Width() - cxIcon + 1) / 2;
  287. int y = (rect.Height() - cyIcon + 1) / 2;
  289. // Draw the icon
  290. dc.DrawIcon(x, y, m_hIcon);
  291. }
  292. else
  293. {
  294. CDialog::OnPaint();
  295. }
  296. }
  298. // The system calls this to obtain the cursor to display while the user drags
  299. //  the minimized window.
  300. HCURSOR CMyFireWallDlg::OnQueryDragIcon()
  301. {
  302. return (HCURSOR) m_hIcon;
  303. }
  305. void CMyFireWallDlg::OnSkinBlue() 
  306. {
  307. // TODO: Add your command handler code here
  308. LoadSkin("Minimized.ssk");
  310. // 保存日志文件
  311. SaveLogFile("用户将防火墙皮肤设置为蓝色。");
  312. }
  314. void CMyFireWallDlg::OnSkinGreen() 
  315. {
  316. // TODO: Add your command handler code here
  317. LoadSkin("Topax.ssk");
  319. // 保存日志文件
  320. SaveLogFile("用户将防火墙皮肤设置为绿色。");
  321. }
  323. void CMyFireWallDlg::OnSkinWhite() 
  324. {
  325. // TODO: Add your command handler code here
  326. LoadSkin("dogmax.ssk");
  328. // 保存日志文件
  329. SaveLogFile("用户将防火墙皮肤设置为白色。");
  330. }
  332. void CMyFireWallDlg::OnSessionNotify(WPARAM wParam, LPARAM lParam)
  333. {
  334. SESSION session;
  336. //获取Dll发送来的 Session
  337. g_pIoControl->GetSession(&session, wParam);
  339. //解析出消息类型
  340. int nCode = lParam;
  342. switch(nCode)
  343. {
  344. case CODE_CHANGE_SESSION: //一个会话的属性改变了
  345. m_AppMonitorPage.AddSession(&session);
  346. break;
  347. case CODE_DELETE_SESSION: //一个会话被删除
  348. m_AppMonitorPage.DeleteSession(&session, FALSE);
  349. break;
  350. case CODE_APP_EXIT: //一个应用程序退出，我们要删除此应用程序的所有会话
  351. m_AppMonitorPage.DeleteSession(&session,TRUE);
  352. break;
  353. }
  354. }
  356. void CMyFireWallDlg::OnQueryAcl(WPARAM wParam, LPARAM lParam)
  357. {
  358. CQueryDlg *pQueryDlg = new CQueryDlg;
  359. /*******************************************************************************************/
  360. CRect DlgRect,ScreenRect;
  361. SystemParametersInfo(SPI_GETWORKAREA,0,&ScreenRect,0); //检索工作区大小，工作区是指没有被任务遮盖的屏幕部分。
  363. // 传递应用程序会话
  364. g_pIoControl->GetQuerySession(&pQueryDlg->m_Session, wParam);
  366. pQueryDlg->m_nIndex = wParam;
  368. pQueryDlg->Create(IDD_QUERY_DLG);
  369. pQueryDlg->GetWindowRect(&DlgRect);
  370. pQueryDlg->SetWindowPos(&wndTopMost,ScreenRect.Width()-DlgRect.Width(),ScreenRect.Height()-DlgRect.Height(),DlgRect.Width(),DlgRect.Height(),SWP_NOSIZE);
  371. pQueryDlg->m_pNext = NULL;
  372. /*******************************************************************************************/
  374. if (m_pStartQueryDlg == NULL)
  375. {
  376. m_pStartQueryDlg = pQueryDlg;
  377. m_pQueryDlg  =  pQueryDlg;
  378. // 使用淡出效果显示对话框
  379. // AnimateWindow(dlg.m_hWnd,1000,AW_ACTIVATE|AW_CENTER|AW_BLEND);
  380. m_pStartQueryDlg->ShowWindow(SW_SHOW);
  381. }
  382. else
  383. {
  384. m_pQueryDlg->m_pNext = pQueryDlg;
  385. m_pQueryDlg = pQueryDlg;
  386. }
  387. }
  389. void CMyFireWallDlg::OnQueryDlgDestroy(WPARAM wParam, LPARAM lParam)
  390. {
  391. CQueryDlg *pTemp = m_pStartQueryDlg;
  392. m_pStartQueryDlg = m_pStartQueryDlg->m_pNext;
  393. delete pTemp;
  394. if (m_pStartQueryDlg != NULL)
  395. {
  396. // 显示下一个询问对话框
  397. m_pStartQueryDlg->ShowWindow(SW_SHOW);
  398. }
  399. }
  401. void CMyFireWallDlg::OnUpdateAccessCtrlList(WPARAM wParam, LPARAM lParam)
  402. {
  403. CString ss;
  404. ss.Format(L"DEBUG输出： ulLspRuleCount = %d",g_RuleFile.m_header.ulLspRuleCount);
  405. m_AccessCtrlPage.SetDlgItemText(IDC_DEBUG, ss);
  406. int nItem;
  407. CString strTemp;
  409. m_AccessCtrlPage.ImageList.DeleteImageList();
  411. m_AccessCtrlPage.ImageList.Create(18, 18, ILC_COLOR8, 30, 10);
  413. m_AccessCtrlPage.m_AccessCtrlList.SetImageList(&m_AccessCtrlPage.ImageList, LVSIL_SMALL);
  416. //清除原来的显示内容
  417. m_AccessCtrlPage.m_AccessCtrlList.DeleteAllItems();
  419. for(int i=0; i < (int)g_RuleFile.m_header.ulLspRuleCount; i++)
  420. {
  421. //设置应用程序的图标
  422. SHFILEINFOW stFileInfo;
  423. if( ::SHGetFileInfo(g_RuleFile.m_pLspRules[i].szApplication, 0, &stFileInfo, sizeof(stFileInfo), SHGFI_ICON)) 
  424. {
  425. m_AccessCtrlPage.ImageList.Add(stFileInfo.hIcon);
  426. }
  427. //设置应用程序名称
  428. nItem = m_AccessCtrlPage.m_AccessCtrlList.InsertItem(i, GetFileName(g_RuleFile.m_pLspRules[i].szApplication), i);
  430. //设置采取的动作
  431. if(g_RuleFile.m_pLspRules[i].ucAction == RULE_ACTION_PASS)
  432. {
  433. strTemp = L" 允许 ";
  434. }
  435. else if(g_RuleFile.m_pLspRules[i].ucAction == RULE_ACTION_DENY)
  436. {
  437. strTemp = L" 禁止 ";
  438. }
  439. m_AccessCtrlPage.m_AccessCtrlList.SetItemText(nItem, 1, strTemp);
  441. //设置协议类型
  442. switch(g_RuleFile.m_pLspRules[i].ucServiceType)
  443. {
  444. case RULE_SERVICE_TYPE_ALL:
  445. strTemp.Format(L"%s",L"所有");
  446. break;
  447. case RULE_SERVICE_TYPE_TCP:
  448. strTemp.Format(L"%s",L"TCP");
  449. break;
  450. case RULE_SERVICE_TYPE_UDP:
  451. strTemp.Format(L"%s",L"UDP");
  452. break;
  453. case RULE_SERVICE_TYPE_FTP:
  454. strTemp.Format(L"%s",L"FTP");
  455. break;
  456. case RULE_SERVICE_TYPE_TELNET:
  457. strTemp.Format(L"%s",L"TELNET");
  458. break;
  459. case RULE_SERVICE_TYPE_HTTP:
  460. strTemp.Format(L"%s",L"HTTP");
  461. break;
  462. case RULE_SERVICE_TYPE_NNTP:
  463. strTemp.Format(L"%s",L"NNTP");
  464. break;
  465. case RULE_SERVICE_TYPE_SMTP:
  466. strTemp.Format(L"%s",L"SMTP");
  467. break;
  468. }
  469. m_AccessCtrlPage.m_AccessCtrlList.SetItemText(nItem, 2, strTemp);
  471. //设置应用程序路径
  472. m_AccessCtrlPage.m_AccessCtrlList.SetItemText(nItem, 3, g_RuleFile.m_pLspRules[i].szApplication);
  474. //设置应用程序的描述
  475. m_AccessCtrlPage.m_AccessCtrlList.SetItemText(nItem, 4, g_RuleFile.m_pLspRules[i].sDemo);
  476. }
  477. }
  479. void CMyFireWallDlg::OnUpdatePacketMonitorList(WPARAM wParam, LPARAM lParam)
  480. {
  481. int nItem;
  482. WCHAR wszTemp[6];
  483. BYTE *pByte = new BYTE;
  484. CString strIP, strTemp;
  485. PPassthruFilter pItem = new PassthruFilter;
  487. //清除原来的显示内容
  488. m_PacketMonitorPage.m_PacketMonitorList.DeleteAllItems();
  490. for (int i = 0; i < (int)g_RuleFile.m_header.ulKerRuleCount; i++)
  491. {
  492. pItem = &g_RuleFile.m_pKerRules[i];
  493. // 源IP地址
  494. pByte = (BYTE *)&pItem->sourceIP;
  495. strIP.Format(L"%d.%d.%d.%d", pByte[3], pByte[2], pByte[1], pByte[0]);
  496. strTemp = (pItem->sourceIP == 0)? L"所有" : strIP;
  497. nItem = m_PacketMonitorPage.m_PacketMonitorList.InsertItem(i, strTemp);
  498. /*
  499. // 源IP掩码
  500. pByte = (BYTE *)&pItem->sourceMask;
  501. strTemp.Format(L"%d.%d.%d.%d", pByte[3], pByte[2], pByte[1], pByte[0]);
  502. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 1, strTemp);
  503. */
  504. // 源端口号
  505. strTemp = (pItem->sourcePort == 0)? L"所有" : ::_itot(pItem->sourcePort, wszTemp, 10);
  506. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 1, strTemp);
  508. // 目的IP地址
  509. pByte = (BYTE *)&pItem->destinationIP;
  510. strIP.Format(L"%d.%d.%d.%d", pByte[3], pByte[2], pByte[1], pByte[0]);
  511. strTemp = (pItem->destinationIP == 0)? L"所有" : strIP;
  512. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 2, strTemp);
  513. /*
  514. // 目的IP掩码
  515. pByte = (BYTE *)&pItem->destinationMask;
  516. strTemp.Format(L"%d.%d.%d.%d", pByte[3], pByte[2], pByte[1], pByte[0]);
  517. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 4, strTemp);
  518. */
  519. // 目的端口号
  520. strTemp = (pItem->destinationPort == 0)? L"所有" : ::_itot(pItem->destinationPort, wszTemp, 10);
  521. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 3, strTemp);
  523. // 协议
  524. switch(pItem->protocol)
  525. {
  526. case 1:
  527. strTemp = L"ICMP";
  528. break;
  529. case 6:
  530. strTemp = L"TCP";
  531. break;
  532. case 17:
  533. strTemp = L"UDP";
  534. break;
  535. default:
  536. strTemp = L"所有";
  537. }
  538. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 4, strTemp);
  540. //动作
  541. strTemp = pItem->bDrop? L"禁止" : L"允许";
  542. m_PacketMonitorPage.m_PacketMonitorList.SetItemText(nItem, 5, strTemp);
  543. }
  544. }
  547. void CMyFireWallDlg::OnUpdateWebSiteList(WPARAM wParam, LPARAM lParam)
  548. {
  549. int nItem;
  550. int nIndex = 0;
  551. BYTE *pByte = new BYTE;
  552. CString strIP, strTemp;
  553. PPassthruFilter pItem = new PassthruFilter;
  555. //清除原来的显示内容
  556. m_WebSiteSettingPage.m_WebSiteList.DeleteAllItems();
  558. for (int i = 0; i < (int)g_RuleFile.m_header.ulKerRuleCount; i++)
  559. {
  560. pItem = &g_RuleFile.m_pKerRules[i];
  561. if (_tcscmp(pItem->szWebSiteURL, L"") == 0)
  562. {
  563. continue ;
  564. }
  566. // 网站URL
  567. nItem = m_WebSiteSettingPage.m_WebSiteList.InsertItem(nIndex, pItem->szWebSiteURL);
  569. // 网站IP地址
  570. pByte = (BYTE *)&pItem->destinationIP;
  571. strIP.Format(L"%d.%d.%d.%d", pByte[3], pByte[2], pByte[1], pByte[0]);
  572. strTemp = (pItem->destinationIP == 0)? L"所有" : strIP;
  573. m_WebSiteSettingPage.m_WebSiteList.SetItemText(nItem, 1, strTemp);
  575. // 网站描述信息
  576. m_WebSiteSettingPage.m_WebSiteList.SetItemText(nItem, 2, pItem->szDescription);
  578. nIndex++;
  579. }
  580. }
  582. void CMyFireWallDlg::OnUpdateIPRuleList(WPARAM wParam, LPARAM lParam)
  583. {
  584. int nItem;
  585. CString strTemp;
  587. // 应用 CImageList 从另一种角度上增加 IPRuleList 的列宽
  588. CImageList ImageList;
  589. ImageList.Create(1, 18, ILC_COLOR8, 0, 0);
  590. m_IPRulePage.m_IPRuleList.SetImageList(&ImageList, LVSIL_SMALL);
  592. // 删除以前全部内容
  593. m_IPRulePage.m_IPRuleList.DeleteAllItems();
  595. // Ping入规则
  596. nItem = m_IPRulePage.m_IPRuleList.InsertItem(0, L"Ping入");
  597. strTemp = g_RuleFile.m_pIPRules[0].bState? L" 允许 " : L" 拒绝 ";
  598. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 1, strTemp);
  599. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 2, L"ICMP");
  600. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 3, L"任意");
  601. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 4, g_RuleFile.m_pIPRules[0].description);
  603. // Ping出规则
  604. nItem = m_IPRulePage.m_IPRuleList.InsertItem(1, L"Ping出");
  605. strTemp = g_RuleFile.m_pIPRules[1].bState? L" 允许 " : L" 拒绝 ";
  606. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 1, strTemp);
  607. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 2, L"ICMP");
  608. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 3, L"任意");
  609. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 4, g_RuleFile.m_pIPRules[1].description);
  611. // RPC(Remote Procedurce call: 远程过程调用)规则
  612. nItem = m_IPRulePage.m_IPRuleList.InsertItem(2, L"RPC(远程过程调用)");
  613. strTemp = g_RuleFile.m_pIPRules[2].bState? L" 允许 " : L" 拒绝 ";
  614. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 1, strTemp);
  615. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 2, L"TCP");
  616. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 3, L"135");
  617. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 4, g_RuleFile.m_pIPRules[2].description);
  619. // DNS(域名服务)规则
  620. nItem = m_IPRulePage.m_IPRuleList.InsertItem(3, L"DNS(域名服务)");
  621. strTemp = g_RuleFile.m_pIPRules[3].bState? L" 允许 " : L" 拒绝 ";
  622. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 1, strTemp);
  623. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 2, L"UDP");
  624. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 3, L"53");
  625. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 4, g_RuleFile.m_pIPRules[3].description);
  627. // 局域网共享规则
  628. nItem = m_IPRulePage.m_IPRuleList.InsertItem(4, L"局域网共享");
  629. strTemp = g_RuleFile.m_pIPRules[4].bState? L" 允许 " : L" 拒绝 ";
  630. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 1, strTemp);
  631. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 2, L"TCP");
  632. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 3, L"139，445");
  633. m_IPRulePage.m_IPRuleList.SetItemText(nItem, 4, g_RuleFile.m_pIPRules[4].description);
  634. }
  637. void CMyFireWallDlg::OnMenuCutdownNetwork() 
  638. {
  639. // TODO: Add your command handler code here
  640. CString strItem;
  641. GetMenu()->GetMenuString(IDM_CUTDOWN_NETWORK, strItem, MF_BYCOMMAND);
  642. if (strItem == L"断开网络")
  643. {
  644. int nWorkMode = IM_DENY_ALL;
  645. PtSetWorkMode(&nWorkMode);
  646. // 显示“连接网络”
  647. GetMenu()->GetSubMenu(0)->ModifyMenu(IDM_CUTDOWN_NETWORK, MF_BYCOMMAND, IDM_CUTDOWN_NETWORK, L"连接网络");
  648. // 在“工作状态”标签上改变核心层工作模式
  649. m_WorkStatusPage.SetDlgItemText(IDC_KER_MODE, L"禁止所有");
  650. // 保存日志文件
  651. SaveLogFile("用户断开网络。");
  652. } 
  653. else
  654. {
  655. int nWorkMode = g_RuleFile.m_header.ucKerWorkMode;
  656. PtSetWorkMode(&nWorkMode);
  657. // 显示“断开网络”
  658. GetMenu()->GetSubMenu(0)->ModifyMenu(IDM_CUTDOWN_NETWORK, MF_BYCOMMAND, IDM_CUTDOWN_NETWORK, L"断开网络");
  659. // 在“工作状态”标签上改变核心层工作模式
  660. m_WorkStatusPage.UpdateWorkMode();
  661. // 保存日志文件
  662. SaveLogFile("用户连接网络。");
  663. }
  664. }
  666. void CMyFireWallDlg::OnMenuStopProtect()
  667. {
  668. // TODO: Add your command handler code here
  669. CString strItem;
  670. GetMenu()->GetMenuString(IDM_STOP_PROTECT, strItem, MF_BYCOMMAND);
  671. if (strItem == L"停止保护")
  672. {
  673. // 应用层放行所有
  674. g_pIoControl->SetWorkMode(PF_PASS_ALL);
  675. // 驱动层放行所有
  676. int nWorkMode = IM_PASS_ALL;
  677. PtSetWorkMode(&nWorkMode);
  678. // 显示“开启保护”
  679. GetMenu()->GetSubMenu(0)->ModifyMenu(IDM_STOP_PROTECT, MF_BYCOMMAND, IDM_STOP_PROTECT, L"开启保护");
  680. // 在“工作状态”标签上改变应用层和核心层工作模式
  681. m_WorkStatusPage.SetDlgItemText(IDC_LSP_MODE, L"允许所有");
  682. m_WorkStatusPage.SetDlgItemText(IDC_KER_MODE, L"允许所有");
  683. // 保存日志文件
  684. SaveLogFile("用户停止防火墙保护。");
  685. } 
  686. else
  687. {
  688. // 应用层过滤所有
  689. g_pIoControl->SetWorkMode(g_RuleFile.m_header.ucLspWorkMode);
  690. // 驱动层过滤所有
  691. int nWorkMode = g_RuleFile.m_header.ucKerWorkMode;
  692. PtSetWorkMode(&nWorkMode);
  693. // 显示“停止保护”
  694. GetMenu()->GetSubMenu(0)->ModifyMenu(IDM_STOP_PROTECT, MF_BYCOMMAND, IDM_STOP_PROTECT, L"停止保护");
  695. // 在“工作状态”标签上还原应用层和核心层工作模式
  696. m_WorkStatusPage.UpdateWorkMode();
  697. // 保存日志文件
  698. SaveLogFile("用户开启防火墙保护。");
  699. }
  700. }
  702. void CMyFireWallDlg::OnMenuExit() 
  703. {
  704. // TODO: Add your command handler code here
  705. RemoveProvider();
  707. // 保存日志文件
  708. SaveLogFile("用户关闭防火墙。");
  710. exit(0);
  711. }
  713. void CMyFireWallDlg::OnMenuLspRule() 
  714. {
  715. // TODO: Add your command handler code here
  716. m_Sheet.SetActivePage(&m_AccessCtrlPage);
  717. }
  719. void CMyFireWallDlg::OnMenuKerRule() 
  720. {
  721. // TODO: Add your command handler code here
  722. m_Sheet.SetActivePage(&m_PacketMonitorPage);
  723. }
  725. void CMyFireWallDlg::OnCancel() 
  726. {
  727. // TODO: Add extra cleanup here
  728. nid.cbSize=(DWORD)sizeof(NOTIFYICONDATA);
  729. nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));
  730. nid.hWnd=this->m_hWnd;
  731. nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP;
  732. nid.uID=IDR_MAINFRAME;
  733. nid.uCallbackMessage=WM_TO_TRAY;
  734. wsprintf(nid.szTip, _T("%s"), _T("MyFireWall 正在保护您的网络……"));
  736. Shell_NotifyIcon(NIM_ADD,&nid);
  737. ShowWindow(SW_HIDE);
  738. }
  740. void CMyFireWallDlg::OnShowTask(WPARAM wParam, LPARAM lParam)
  741. {
  742. UINT uID; //发出该消息的图标的ID
  743. UINT uMouseMsg; //鼠标动作
  744. uID=(UINT) wParam;
  745. uMouseMsg=(UINT) lParam;
  746. if(uID!=IDR_MAINFRAME)
  747. {
  748. return;
  749. }
  750. switch(uMouseMsg)
  751. {
  752. case WM_LBUTTONDBLCLK: 
  753. ShowWindow(SW_SHOW);
  754. break;
  755. case WM_RBUTTONUP:
  756. POINT point;
  757. //得到鼠标坐标
  758. GetCursorPos(&point);
  759. //创建弹出菜单
  760. CMenu menu;
  761. menu.CreatePopupMenu();
  762. //增加菜单项“启动”，点击则发送消息WM_DESTROY给主窗口，打开主界面。
  763. menu.AppendMenu(MF_STRING, IDM_SHOW_WINDOW, _T("  启 动  "));
  764. //增加菜单项“关闭”，点击则发送消息WM_DESTROY给主窗口，将程序结束。
  765. menu.AppendMenu(MF_STRING,IDM_EXIT, _T("  关 闭  "));
  766. //将程序设置为前端窗口
  767. SetForegroundWindow();
  768. //确定弹出菜单的位置
  769. menu.TrackPopupMenu(TPM_LEFTALIGN,point.x,point.y,this);
  770. //资源回收
  771. HMENU hMenu=menu.Detach();
  772. menu.DestroyMenu();
  773. break;
  774. }
  776. return;
  777. }
  779. void CMyFireWallDlg::ShowMainWindow()
  780. {
  781. ShowWindow(SW_SHOW);
  782. }