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

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

防火墙与安全工具

开发平台：

Visual C++

WorkStatusPage.cpp：源码内容
  1. // WorkStatusPage.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "MyFireWall.h"
  6. #include "WorkStatusPage.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. #define MODIFY_LSP_MODE 1
  15. #define MODIFY_KER_MODE 2
  17. #define START_X      70
  18. #define START_Y      370
  19. #define UNIT_LENGTH  3
  20. #define UNIT_HEIGHT  1
  22. extern CMyFireWallApp theApp;
  24. /////////////////////////////////////////////////////////////////////////////
  25. // CWorkStatusPage property page
  27. IMPLEMENT_DYNCREATE(CWorkStatusPage, CPropertyPage)
  29. CWorkStatusPage::CWorkStatusPage() : CPropertyPage(CWorkStatusPage::IDD)
  30. {
  31. //{{AFX_DATA_INIT(CWorkStatusPage)
  32. // NOTE: the ClassWizard will add member initialization here
  33. //}}AFX_DATA_INIT
  34. }
  36. CWorkStatusPage::~CWorkStatusPage()
  37. {
  38. }
  40. void CWorkStatusPage::DoDataExchange(CDataExchange* pDX)
  41. {
  42. CPropertyPage::DoDataExchange(pDX);
  43. //{{AFX_DATA_MAP(CWorkStatusPage)
  44. // NOTE: the ClassWizard will add DDX and DDV calls here
  45. //}}AFX_DATA_MAP
  46. }
  49. BEGIN_MESSAGE_MAP(CWorkStatusPage, CPropertyPage)
  50. //{{AFX_MSG_MAP(CWorkStatusPage)
  51. ON_BN_CLICKED(IDC_LSP_MODE, OnModifyLspMode)
  52. ON_COMMAND(IDM_DENY_MODE, OnMenuDenyMode)
  53. ON_COMMAND(IDM_FILTER_MODE, OnMenuFilterMode)
  54. ON_COMMAND(IDM_PASS_MODE, OnMenuPassMode)
  55. ON_BN_CLICKED(IDC_KER_MODE, OnModifyKerMode)
  56. ON_WM_PAINT()
  57. ON_WM_TIMER()
  58. //}}AFX_MSG_MAP
  59. END_MESSAGE_MAP()
  61. /////////////////////////////////////////////////////////////////////////////
  62. // CWorkStatusPage message handlers
  64. BOOL CWorkStatusPage::OnInitDialog() 
  65. {
  66. CPropertyPage::OnInitDialog();
  68. // TODO: Add extra initialization here
  69. m_nSelected = 0; // 这是对于菜单项的操作，设为0表示开始时既不修改应用层模式(对应 1)，也不修改核心层模式(对应 2)
  70. UpdateWorkMode();
  72. for (int i=0; i<MAX_COUNTS; i++)
  73. {
  74. m_nSendFlowRate[i]  = 0;
  75. m_nRecvFlowRate[i]  = 0;
  76. m_nTotalFlowRate[i] = 0;
  77.   }
  79. SetTimer(1, 1000, 0);
  81. return TRUE;  // return TRUE unless you set the focus to a control
  82.               // EXCEPTION: OCX Property Pages should return FALSE
  83. }
  85. void CWorkStatusPage::UpdateWorkMode()
  86. {
  87. // 应用层工作模式
  88. switch (g_RuleFile.m_header.ucLspWorkMode)
  89. {
  90. case PF_PASS_ALL:
  91. SetDlgItemText(IDC_LSP_MODE, L"允许所有");
  92. break;
  93. case PF_QUERY_ALL:
  94. SetDlgItemText(IDC_LSP_MODE, L"过滤所有");
  95. break;
  96. case PF_DENY_ALL:
  97. SetDlgItemText(IDC_LSP_MODE, L"禁止所有");
  98. break;
  99. }
  101. // 核心层工作模式
  102. switch (g_RuleFile.m_header.ucKerWorkMode)
  103. {
  104. case IM_PASS_ALL:
  105. SetDlgItemText(IDC_KER_MODE, L"允许所有");
  106. break;
  107. case IM_START_FILTER:
  108. SetDlgItemText(IDC_KER_MODE, L"过滤所有");
  109. break;
  110. case IM_DENY_ALL:
  111. SetDlgItemText(IDC_KER_MODE, L"禁止所有");
  112. break;
  113. } 
  114. }
  116. void CWorkStatusPage::OnModifyLspMode() 
  117. {
  118. // TODO: Add your control notification handler code here
  119. m_nSelected = MODIFY_LSP_MODE;
  121. //创建菜单
  122. CMenu menu,*pSubMenu;
  123. menu.LoadMenu(IDR_MODE_MENU);
  124. pSubMenu=menu.GetSubMenu(0);
  126. CRect Rect;
  127. GetDlgItem(IDC_LSP_MODE)->GetWindowRect(&Rect);
  128. CPoint point(Rect.left, Rect.bottom);
  130. UINT nCheckItem;
  131. switch (g_RuleFile.m_header.ucLspWorkMode)
  132. {
  133. case PF_PASS_ALL:
  134. nCheckItem = IDM_PASS_MODE;
  135. break;
  136. case PF_QUERY_ALL:
  137. nCheckItem = IDM_FILTER_MODE;
  138. break;
  139. case PF_DENY_ALL:
  140. nCheckItem = IDM_DENY_MODE;
  141. break;
  142. }
  143. pSubMenu->CheckMenuItem(nCheckItem, MF_BYCOMMAND | MF_CHECKED);
  145. pSubMenu->TrackPopupMenu(TPM_LEFTALIGN,point.x,point.y,this);
  146. }
  148. void CWorkStatusPage::OnModifyKerMode() 
  149. {
  150. // TODO: Add your control notification handler code here
  151. m_nSelected = MODIFY_KER_MODE;
  153. //创建菜单
  154. CMenu menu,*pSubMenu;
  155. menu.LoadMenu(IDR_MODE_MENU);
  156. pSubMenu=menu.GetSubMenu(0);
  158. CRect Rect;
  159. GetDlgItem(IDC_KER_MODE)->GetWindowRect(&Rect);
  160. CPoint point(Rect.left, Rect.bottom);
  162. UINT nCheckItem;
  163. switch (g_RuleFile.m_header.ucKerWorkMode)
  164. {
  165. case IM_PASS_ALL:
  166. nCheckItem = IDM_PASS_MODE;
  167. break;
  168. case IM_START_FILTER:
  169. nCheckItem = IDM_FILTER_MODE;
  170. break;
  171. case IM_DENY_ALL:
  172. nCheckItem = IDM_DENY_MODE;
  173. break;
  174. }
  175. pSubMenu->CheckMenuItem(nCheckItem, MF_BYCOMMAND | MF_CHECKED);
  177. pSubMenu->TrackPopupMenu(TPM_LEFTALIGN,point.x,point.y,this);
  178. }
  180. void CWorkStatusPage::OnMenuPassMode() 
  181. {
  182. // TODO: Add your command handler code here
  183. if (m_nSelected == MODIFY_LSP_MODE)
  184. {
  185. g_RuleFile.m_header.ucLspWorkMode = PF_PASS_ALL;
  186. g_pIoControl->SetWorkMode(PF_PASS_ALL);
  187. SetDlgItemText(IDC_LSP_MODE, L"允许所有");
  188. // 保存文件头中的工作模式
  189. g_RuleFile.SaveRules();
  190. // 应用到DLL层
  191. theApp.ApplyLspRules();
  192. // 将该动作保存到日志文件
  193. SaveLogFile("将应用层工作模式设置为：“允许所有”。");
  194. } 
  195. else
  196. {
  197. g_RuleFile.m_header.ucKerWorkMode = IM_PASS_ALL;
  198. int nKerWorkMode = IM_PASS_ALL;
  199. PtSetWorkMode(&nKerWorkMode);
  200. SetDlgItemText(IDC_KER_MODE, L"允许所有");
  201. // 保存文件头中的工作模式
  202. g_RuleFile.SaveRules();
  203. // 应用到核心层
  204. theApp.ApplyKerRules();
  205. // 将该动作保存到日志文件
  206. SaveLogFile("将核心层工作模式设置为：“允许所有”。");
  207. }
  209. // 还原 nSelected 值
  210. m_nSelected = 0;
  211. }
  213. void CWorkStatusPage::OnMenuFilterMode() 
  214. {
  215. // TODO: Add your command handler code here
  216. if (m_nSelected == MODIFY_LSP_MODE)
  217. {
  218. g_RuleFile.m_header.ucLspWorkMode = PF_QUERY_ALL;
  219. g_pIoControl->SetWorkMode(PF_QUERY_ALL);
  220. SetDlgItemText(IDC_LSP_MODE, L"过滤所有");
  221. // 保存文件头中的工作模式
  222. g_RuleFile.SaveRules();
  223. // 应用到DLL层
  224. theApp.ApplyLspRules();
  225. // 将该动作保存到日志文件
  226. SaveLogFile("将应用层工作模式设置为：“过滤所有”。");
  227. } 
  228. else
  229. {
  230. g_RuleFile.m_header.ucKerWorkMode = IM_START_FILTER;
  231. int nKerWorkMode = IM_START_FILTER;
  232. PtSetWorkMode(&nKerWorkMode);
  233. SetDlgItemText(IDC_KER_MODE, L"过滤所有");
  234. // 保存文件头中的工作模式
  235. g_RuleFile.SaveRules();
  236. // 应用到核心层
  237. theApp.ApplyKerRules();
  238. SaveLogFile("将核心层工作模式设置为：“过滤所有”。");
  239. }
  241. // 还原 nSelected 值
  242. m_nSelected = 0;
  243. }
  245. void CWorkStatusPage::OnMenuDenyMode() 
  246. {
  247. // TODO: Add your command handler code here
  248. if (m_nSelected == MODIFY_LSP_MODE)
  249. {
  250. g_RuleFile.m_header.ucLspWorkMode = PF_DENY_ALL;
  251. g_pIoControl->SetWorkMode(PF_DENY_ALL);
  252. SetDlgItemText(IDC_LSP_MODE, L"拒绝所有");
  253. // 保存文件头中的工作模式
  254. g_RuleFile.SaveRules();
  255. // 应用到DLL层
  256. theApp.ApplyLspRules();
  257. SaveLogFile("将应用层工作模式设置为：“拒绝所有”。");
  258. } 
  259. else
  260. {
  261. g_RuleFile.m_header.ucKerWorkMode = IM_DENY_ALL;
  262. int nKerWorkMode = IM_DENY_ALL;
  263. PtSetWorkMode(&nKerWorkMode);
  264. SetDlgItemText(IDC_KER_MODE, L"拒绝所有");
  265. // 保存文件头中的工作模式
  266. g_RuleFile.SaveRules();
  267. // 应用到核心层
  268. theApp.ApplyKerRules();
  269. SaveLogFile("将核心层工作模式设置为：“拒绝所有”。");
  270. }
  272. // 还原 nSelected 值
  273. m_nSelected = 0;
  274. }
  276. void CWorkStatusPage::OnPaint() 
  277. {
  278. CPaintDC dc(this); // device context for painting
  280. // TODO: Add your message handler code here
  282. DrawFlowGraph();
  284. // Do not call CPropertyPage::OnPaint() for painting messages
  285. }
  287. void CWorkStatusPage::OnTimer(UINT nIDEvent)
  288. {
  289. // TODO: Add your message handler code here and/or call default
  291. HANDLE hControlDevice = PtOpenControlDevice();
  292. CIMAdapters adapters;
  293. if(!adapters.EnumAdapters(hControlDevice))
  294. return ;
  296. HANDLE hAdapter;
  298. PassthruStatistics PassthruStats;
  299. for(int i=0; i<adapters.m_nAdapters; i++)
  300. {
  301. // 打开下层适配器
  302. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  303. if(hAdapter != INVALID_HANDLE_VALUE)
  304. {
  305. PassthruStatistics Stats;
  306. BOOL bRet = PtQueryStatistics(hAdapter, &Stats);
  307. if(bRet == FALSE)
  308. {
  309. AfxMessageBox(L"返回值位 FALSE");
  310. ::CloseHandle(hAdapter);
  311. ::CloseHandle(hControlDevice);
  312. break;
  313. }
  314. PassthruStats.nSendFlowRate  = Stats.nSendFlowRate;
  315. PassthruStats.nRecvFlowRate  = Stats.nRecvFlowRate;
  316. PassthruStats.nMPSendPktsCt  = Stats.nMPSendPktsCt;
  317. PassthruStats.nMPSendPktsDropped = Stats.nMPSendPktsDropped;
  318. PassthruStats.nPTRcvCt  = Stats.nPTRcvCt;
  319. PassthruStats.nPTRcvDropped  = Stats.nPTRcvDropped;
  320. PassthruStats.nPTRcvPktCt  = Stats.nPTRcvPktCt;
  321. PassthruStats.nPTRcvPktDropped  = Stats.nPTRcvPktDropped;
  322. ::CloseHandle(hAdapter);
  323. }
  324. }
  326. CString strFlowRate;
  327. strFlowRate.Format(L"当前发送流速：n    %d kBpsn当前接收流速：n    %d KBpsnn发送数据包总数: n    %drn  丢弃个数: n    %dnn接收数据包总数: n    %dn  丢弃个数: n    %drn",
  328. PassthruStats.nSendFlowRate/1024, 
  329. PassthruStats.nRecvFlowRate/1024,
  330. PassthruStats.nMPSendPktsCt,
  331. PassthruStats.nMPSendPktsDropped,
  332. PassthruStats.nPTRcvCt+PassthruStats.nPTRcvPktCt,
  333. PassthruStats.nPTRcvDropped+PassthruStats.nPTRcvPktDropped
  334. );
  336. SetDlgItemText(IDC_FLOW_RATE, strFlowRate);
  338. for (i=MAX_COUNTS-2; i>=0; i--)
  339. {
  340. m_nSendFlowRate[i+1]  = m_nSendFlowRate[i];
  341. m_nRecvFlowRate[i+1]  = m_nRecvFlowRate[i];
  342. m_nTotalFlowRate[i+1] = m_nTotalFlowRate[i];
  343. }
  344. m_nSendFlowRate[0]  = PassthruStats.nSendFlowRate/1024/3;
  345. m_nRecvFlowRate[0]  = PassthruStats.nRecvFlowRate/1024/3;
  346. m_nTotalFlowRate[0] = m_nSendFlowRate[0]+m_nRecvFlowRate[0];
  347. if (m_nSendFlowRate[0]>90)
  348. {
  349. m_nSendFlowRate[0] = 90;
  350. }
  351. if (m_nRecvFlowRate[0]>90)
  352. {
  353. m_nRecvFlowRate[0] = 90;
  354. }
  355. if (m_nTotalFlowRate[0]>90)
  356. {
  357. m_nTotalFlowRate[0] = 90;
  358. }
  360. DrawFlowGraph();
  362. OnPaint();
  364. ::CloseHandle(hControlDevice);
  366. CPropertyPage::OnTimer(nIDEvent);
  367. }
  369. void CWorkStatusPage::DrawFlowGraph()
  370. {
  371. CClientDC dc(this);
  373. CRect Rect(70, 70, 370, 370);
  374. CBrush Brush(RGB(135, 206, 235)); // 天蓝
  375. dc.FillRect(&Rect, &Brush);
  377. dc.MoveTo(START_X, 70);
  378. dc.LineTo(START_X, START_Y);
  379. dc.LineTo(370, START_Y);
  380. dc.LineTo(370, 70);
  381. dc.LineTo(START_X, 70);
  383. dc.MoveTo(START_X, 170);
  384. dc.LineTo(370, 170);
  386. dc.MoveTo(START_X, 270);
  387. dc.LineTo(370, 270);
  389. dc.MoveTo(START_X, 80);
  390. dc.LineTo(START_X+5, 80);
  391. dc.MoveTo(START_X, 180);
  392. dc.LineTo(START_X+5, 180);
  393. dc.MoveTo(START_X, 280);
  394. dc.LineTo(START_X+5, 280);
  396. int  penWidth = 2;   
  397. CPen NewPen(PS_SOLID, penWidth, RGB(34, 139, 34)); // 森林绿   
  398. dc.SelectObject(&NewPen); 
  400. // 画发送流量
  401. dc.MoveTo(START_X, START_Y-200-m_nSendFlowRate[0]);
  402. for (int i=1; i<MAX_COUNTS; i++)
  403. {
  404. dc.LineTo(START_X+i*UNIT_LENGTH, START_Y-200-m_nSendFlowRate[i]*UNIT_HEIGHT);
  405. }
  407. // 画接收流量
  408. dc.MoveTo(START_X, START_Y-100-m_nRecvFlowRate[0]);
  409. for (i=1; i<MAX_COUNTS; i++)
  410. {
  411. dc.LineTo(START_X+i*UNIT_LENGTH, START_Y-100-m_nRecvFlowRate[i]*UNIT_HEIGHT);
  412. }
  414. // 画总流量
  415. dc.MoveTo(START_X, START_Y-m_nTotalFlowRate[0]);
  416. for (i=1; i<MAX_COUNTS; i++)
  417. {
  418. dc.LineTo(START_X+i*UNIT_LENGTH, START_Y-m_nTotalFlowRate[i]*UNIT_HEIGHT);
  419. }
  420. }