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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
Property.cpp
资源名称：FireWall-C++.rar　[点击查看]
上传用户：kklily621
上传日期：2013-06-25
资源大小：252k
文件大小：10k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

Property.cpp：源码内容
  1. // Property.cpp : Defines the class behaviors for the application.
  2. //
  4. #include "stdafx.h"
  5. #include "Property.h"
  7. #include "GuiRes.h"
  8. #include ".MainFrameMainFrame.h"
  9. #include ".MainSheet.h"
  10. #include "AclSet.h"
  11. #include "NetTimeSheet.h"
  12. #include "NetIPAria.h"
  13. #include "Acl.h"
  14. #include "Splash.h"
  16. #ifdef _DEBUG
  17. #define new DEBUG_NEW
  18. #undef THIS_FILE
  19. static char THIS_FILE[] = __FILE__;
  20. #endif
  22. //=============================================================================================
  23. // The one and only CPropertyApp object
  25. CPropertyApp theApp;
  26. CAclFile m_AclFile;
  27. extern CMainSheet *dlg;
  28. extern CAclSet *dlgAclSet;
  29. extern CNetTimeSheet *dlgNetTimeSheet;
  30. extern CFileDialog *dlgFile;
  31. extern CNetIPAria *dlgNetIPAria;
  32. extern CSystemTray m_TrayIcon;
  34. //=============================================================================================
  35. // CPropertyApp
  37. BEGIN_MESSAGE_MAP(CPropertyApp, CWinApp)
  38. //{{AFX_MSG_MAP(CPropertyApp)
  39. ON_COMMAND(ID_APP_EXIT, OnAppExit)
  40. ON_COMMAND(ID_ABOUT, OnAbout)
  41. ON_COMMAND(ID_ACL_SET, OnAclSet)
  42. ON_COMMAND(ID_LOG_QUERY, OnLogQuery)
  43. ON_COMMAND(ID_SYSTEM_SET, OnSystemSet)
  44. ON_UPDATE_COMMAND_UI(ID_ABOUT, OnUpdateAbout)
  45. ON_UPDATE_COMMAND_UI(ID_ACL_SET, OnUpdateAclSet)
  46. ON_UPDATE_COMMAND_UI(ID_APP_EXIT, OnUpdateAppExit)
  47. ON_UPDATE_COMMAND_UI(ID_LOG_QUERY, OnUpdateLogQuery)
  48. ON_UPDATE_COMMAND_UI(ID_SYSTEM_SET, OnUpdateSystemSet)
  49. ON_COMMAND(ID_CONTROL_FRAME, OnControlFrame)
  50. ON_UPDATE_COMMAND_UI(ID_CONTROL_FRAME, OnUpdateControlFrame)
  51. ON_COMMAND(ID_PACKET_MONITOR, OnPacketMonitor)
  52. ON_UPDATE_COMMAND_UI(ID_PACKET_MONITOR, OnUpdatePacketMonitor)
  53. //}}AFX_MSG_MAP
  54. ON_COMMAND(ID_HELP, CWinApp::OnHelp)
  55. END_MESSAGE_MAP()
  57. //=============================================================================================
  58. // CPropertyApp construction
  60. CPropertyApp::CPropertyApp()
  61. {
  62. m_EnablePacketMonitor = TRUE;
  63. m_EnableQueryLog = TRUE;
  64. m_EnableAcl = TRUE;
  65. m_EnableSystemSet = TRUE;
  66. m_EnableAbout = TRUE;
  67. m_EnableExit = TRUE;
  68. m_EnableContolFrame = TRUE;
  70. m_XfIoControl = NULL;
  71. }
  73. //=============================================================================================
  74. // CPropertyApp initialization
  76. BOOL CPropertyApp::InitInstance()
  77. {
  78. TCHAR sPathName[MAX_PATH];
  79. _stprintf(sPathName, _T("%s%s"), m_AclFile.GetAppPath(), XFILTER_SERVICE_DLL_NAME);
  81. TCHAR sTemp[MAX_PATH];
  82. _stprintf(sTemp, _T("%s%s"), m_AclFile.GetAppPath(), XFILTER_HELP_FILE_NAME);
  83. theApp.m_pszHelpFilePath = _tcsdup(sTemp);
  85. if(_tcscmp(theApp.m_lpCmdLine, _T("-install")) == 0)
  86. {
  87. int iRet;
  88. if((iRet = m_Install.InstallProvider(sPathName)) != XERR_SUCCESS
  89. && iRet != XERR_PROVIDER_ALREADY_INSTALL)
  90. AfxMessageBox(GUI_ACL_MESSAGE_INSTALL_FAILED);
  91. SetAutoStart(FALSE);
  92. m_RegisterRequest.UserReg();
  93. return FALSE;
  94. }
  96. if(_tcscmp(theApp.m_lpCmdLine, _T("-remove")) == 0)
  97. {
  98. m_Install.RemoveProvider();
  99. m_Install.DeleteReg(HKEY_LOCAL_MACHINE, REG_AUTO_START_KEY, REG_AUTO_START_ITEM);
  100. return FALSE;
  101. }
  103. if(!m_Install.IsWinsock2())
  104. {
  105. AfxMessageBox(GUI_ACL_MESSAGE_NO_WINSOCK2);
  106. return FALSE;
  107. }
  109. CWnd *PrevCWnd = CWnd::FindWindow(NULL, GUI_APP_CAPTION);
  111. if (PrevCWnd != NULL) 
  112. return FALSE;
  114. if(m_AclFile.ReadAcl() != XERR_SUCCESS)
  115. {
  116. AfxMessageBox(GUI_ACL_MESSAGE_ACL_READ_ERROR);
  117. return FALSE;
  118. }
  120. if(m_AclFile.GetBit(m_AclFile.mAclHeader.bSet, 6))
  121. {
  122. CCommandLineInfo cmdInfo;
  123. ParseCommandLine(cmdInfo);
  124. CSplashWnd::EnableSplashScreen(cmdInfo.m_bShowSplash);
  125. }
  127. if(!theApp.m_RegisterRequest.InitRegister())
  128. return FALSE;
  130. if(!m_Install.IsInstalled(sPathName) && m_Install.InstallProvider(sPathName) != XERR_SUCCESS)
  131. {
  132. AfxMessageBox(GUI_ACL_MESSAGE_INSTALL_FAILED);
  133. return FALSE;
  134. }
  136. if ((m_hTcpIpDog = LoadLibrary(sPathName)) == NULL)
  137. {
  138. AfxMessageBox(GUI_ACL_MESSAGE_DLL_NOT_FOUND);
  139. return FALSE;
  140. }
  142. m_XfIoControl = (XF_IO_CONTROL)GetProcAddress(m_hTcpIpDog, _T("XfIoControl"));
  144. if (m_XfIoControl == NULL)
  145. {
  146. AfxMessageBox(GUI_ACL_MESSAGE_FUNCTION_NOT_FOUNT);
  147. return FALSE;
  148. }
  150. if(!InitDllData())
  151. {
  152. AfxMessageBox(GUI_ACL_MESSAGE_INIT_DLL_DATA_ERROR);
  153. return FALSE;
  154. }
  156. CMainFrame* pFrame = new CMainFrame;
  157. m_pMainWnd = pFrame;
  158. pFrame->Create(NULL,GUI_APP_CAPTION);
  159. pFrame->ShowWindow(SW_HIDE);
  160. pFrame->UpdateWindow();
  162. XFILTER_IO_CONTROL ioControl;
  163. ioControl.hwnd = pFrame->m_hWnd;
  164. m_XfIoControl(IO_CONTROL_SET_GUI_INSTANCE, &ioControl);
  166. return TRUE;
  167. }
  169. BOOL CPropertyApp::InitDllData()
  170. {
  171. if(m_XfIoControl == NULL)
  172. return FALSE;
  174. XFILTER_IO_CONTROL ioControl;
  176. if((ioControl.iWorkMode = CAclFile::GetBit(m_AclFile.mAclHeader.bSet, 4, 2)) != XF_QUERY_ALL)
  177. {
  178. if(m_XfIoControl(IO_CONTROL_SET_WORK_MODE, &ioControl) != XERR_SUCCESS)
  179. return FALSE;
  181. SetTrayIcon();
  182. return TRUE;
  183. }
  185. ioControl.AclFile.mAclHeader = m_AclFile.mAclHeader;
  186. ioControl.AclFile.mAclIntranetIP = m_AclFile.mAclIntranetIP;
  187. memcpy(ioControl.AclFile.mAclTime , m_AclFile.mAclTime, ACL_TIME_LENTH * ACL_HEADER_TIME_COUNT) ;
  188. memcpy(ioControl.AclFile.mpAcl , m_AclFile.mpAcl , ACL_ACL_LENTH * m_AclFile.mAclHeader.ulAclCount);
  189. memcpy(ioControl.AclFile.mpAclCustomIP , m_AclFile.mpAclCustomIP , ACL_IP_LENTH * m_AclFile.mAclHeader.ulCustomIPCount);
  190. memcpy(ioControl.AclFile.mpAclDistrustIP , m_AclFile.mpAclDistrustIP , ACL_IP_LENTH * m_AclFile.mAclHeader.ulDistrustIPCount);
  191. memcpy(ioControl.AclFile.mpAclTrustIP , m_AclFile.mpAclTrustIP , ACL_IP_LENTH * m_AclFile.mAclHeader.ulTrustIPCount);
  193. if(m_XfIoControl(IO_CONTROL_SET_ACL, &ioControl) != XERR_SUCCESS)
  194. return FALSE;
  196. SetTrayIcon();
  197. return TRUE;
  198. }
  200. void CPropertyApp::ExitApp()
  201. {
  202. if(CheckDlg()) 
  203. return;
  205. if(dlg != NULL)
  206. {
  207. if(dlg->GetDlgItem(ID_APPLY_NOW)->IsWindowEnabled())
  208. dlg->OnCancel();
  210. dlg->EndDialog(IDCANCEL);
  211. }
  213. m_pMainWnd->DestroyWindow();
  214. }
  216. /*---------------------------------------------------------------------------------------------
  217. Return Value: 
  218. TRUE: had some sub_dialogs visible 
  219. FALSE: no sub_dialogs visible
  220. */
  221. BOOL CPropertyApp::CheckDlg(BOOL ShowMessage)
  222. {
  223. if( CheckDlgVisible (dlgNetIPAria , ShowMessage)
  224. || CheckDlgVisible (dlgNetTimeSheet, ShowMessage)
  225. || CheckDlgVisible (dlgFile , ShowMessage)
  226. || CheckDlgVisible (dlgAclSet , ShowMessage)
  227. )
  228. return TRUE;
  230. return FALSE;
  231. }
  232. BOOL CPropertyApp::CheckDlgVisible(void* pdlg, BOOL ShowMessage)
  233. {
  234. if(pdlg == NULL) 
  235. return FALSE;
  237. CDialog* mdlg = (CDialog*)pdlg;
  239. if(ShowMessage)
  240. AfxMessageBox(GUI_ACL_MESSAGE_PLEASE_CLOSE_SUB_WINDOW);
  242. mdlg->SetFocus();
  244. return TRUE;
  245. }
  247. void CPropertyApp::OnAppExit() 
  248. {
  249. if(!m_EnableExit) 
  250. return;
  252. ExitApp();
  253. }
  255. void CPropertyApp::OnPacketMonitor() 
  256. {
  257. if(!m_EnablePacketMonitor)
  258. return;
  260. CMainFrame::InitMainSheet();
  261. }
  263. void CPropertyApp::OnLogQuery() 
  264. {
  265. if(!m_EnableQueryLog) 
  266. return;
  268. CMainFrame::InitMainSheet(1);
  269. }
  271. void CPropertyApp::OnAclSet() 
  272. {
  273. if(!m_EnableAcl) 
  274. return;
  276. CMainFrame::InitMainSheet(2);
  277. }
  279. void CPropertyApp::OnSystemSet() 
  280. {
  281. if(!m_EnableSystemSet) 
  282. return;
  284. CMainFrame::InitMainSheet(3);
  285. }
  287. void CPropertyApp::OnAbout() 
  288. {
  289. if(!m_EnableAbout) 
  290. return;
  292. CMainFrame::InitMainSheet(4);
  293. }
  295. void CPropertyApp::OnControlFrame() 
  296. {
  297. if(!m_EnableContolFrame) 
  298. return;
  300. if(CheckDlg(FALSE))
  301. return;
  303. CMainFrame::InitMainSheet(-1);
  304. }
  306. void CPropertyApp::OnUpdateAbout(CCmdUI* pCmdUI) 
  307. {
  308. pCmdUI->Enable(m_EnableAbout);
  309. pCmdUI->SetText(GUI_MENU_ABOUT);
  310. }
  312. void CPropertyApp::OnUpdateAclSet(CCmdUI* pCmdUI) 
  313. {
  314. pCmdUI->Enable(m_EnableAcl);
  315. pCmdUI->SetText(GUI_MENU_ACL);
  316. }
  318. void CPropertyApp::OnUpdateAppExit(CCmdUI* pCmdUI) 
  319. {
  320. pCmdUI->Enable(m_EnableExit);
  321. pCmdUI->SetText(GUI_MENU_EXIT);
  322. }
  324. void CPropertyApp::OnUpdateLogQuery(CCmdUI* pCmdUI) 
  325. {
  326. pCmdUI->Enable(m_EnableQueryLog);
  327. pCmdUI->SetText(GUI_MENU_LOG_QUERY);
  328. }
  330. void CPropertyApp::OnUpdatePacketMonitor(CCmdUI* pCmdUI) 
  331. {
  332. pCmdUI->Enable(m_EnablePacketMonitor);
  333. pCmdUI->SetText(GUI_MENU_PACKET_MONITOR);
  334. }
  336. void CPropertyApp::OnUpdateSystemSet(CCmdUI* pCmdUI) 
  337. {
  338. pCmdUI->Enable(m_EnableSystemSet);
  339. pCmdUI->SetText(GUI_MENU_SYSTEM_SET);
  340. }
  342. void CPropertyApp::OnUpdateControlFrame(CCmdUI* pCmdUI) 
  343. {
  344. pCmdUI->Enable(m_EnableContolFrame);
  345. pCmdUI->SetText(GUI_MENU_CONTROL_FRAME);
  346. }
  348. int CPropertyApp::ExitInstance() 
  349. {
  350. ODS(_T("XFILTER.EXE: ExitInstance Starting..."));
  352. m_TrayIcon.HideIcon();
  354.   if(m_XfIoControl != NULL)
  355. {
  356. XFILTER_IO_CONTROL ioControl;
  357. ioControl.iWorkMode = XF_PASS_ALL;
  358. ioControl.hwnd = NULL;
  360. m_XfIoControl(IO_CONTROL_SET_WORK_MODE, &ioControl);
  361. m_XfIoControl(IO_CONTROL_SET_GUI_INSTANCE, &ioControl);
  362. }
  364. if(m_hTcpIpDog != NULL)
  365. FreeLibrary(m_hTcpIpDog);
  367. m_hTcpIpDog = NULL;
  369. while(m_DownloadCommand.m_IsConnecting || m_RegisterRequest.m_IsConnecting)
  370. Sleep(1000);
  372. ODS(_T("XFILTER.EXE: ExitInstance Over..."));
  373. return CWinApp::ExitInstance();
  374. }
  376. BOOL CPropertyApp::PreTranslateMessage(MSG* pMsg)
  377. {
  378. // CG: The following lines were added by the Splash Screen component. if (CSplashWnd::PreTranslateAppMessage(pMsg)) return TRUE; return CWinApp::PreTranslateMessage(pMsg);
  379. }
  381. void CPropertyApp::SetAutoStart(BOOL IsCheckFromAcl)
  382. {
  383. BOOL IsCheck = TRUE;
  385. if(IsCheckFromAcl)
  386. IsCheck = m_AclFile.GetBit(m_AclFile.mAclHeader.bSet, 1);
  388. if(IsCheck)
  389. {
  390. TCHAR tmpStr[MAX_PATH];
  391. strcpy(tmpStr, m_AclFile.GetAppPath(FALSE, NULL, TRUE));
  393. m_Install.SaveReg(REG_AUTO_START_ITEM, (BYTE*)tmpStr, _tcslen(tmpStr)
  394. , HKEY_LOCAL_MACHINE, REG_AUTO_START_KEY, REG_SZ);
  395. return;
  396. }
  398. m_Install.DeleteReg(HKEY_LOCAL_MACHINE, REG_AUTO_START_KEY, REG_AUTO_START_ITEM);
  399. }
  401. void CPropertyApp::SetTrayIcon()
  402. {
  403. int iWorkMode = XF_PASS_ALL;
  405. if(m_XfIoControl != NULL)
  406. iWorkMode = m_XfIoControl(IO_CONTROL_GET_WORK_MODE, NULL);
  408. if(iWorkMode == XF_PASS_ALL)
  409. m_TrayIcon.SetIcon(IDI_SMALL_PASS);
  410. else if(iWorkMode == XF_QUERY_ALL)
  411. m_TrayIcon.SetIcon(IDI_SMALL_QUERY);
  412. else if(iWorkMode == XF_DENY_ALL)
  413. m_TrayIcon.SetIcon(IDI_SMALL_DENY);
  414. else
  415. m_TrayIcon.SetIcon(IDI_SMALL_PASS);
  416. }
  418. void CPropertyApp::WinHelp(DWORD dwData, UINT nCmd) 
  419. {
  420. HWND hHtmlHelp = HtmlHelp(
  421. theApp.m_pMainWnd->m_hWnd, 
  422. theApp.m_pszHelpFilePath, 
  423. nCmd, 
  424. dwData
  425. );
  427. if(hHtmlHelp == 0)
  428. AfxMessageBox(GUI_MESSAGE_OPEN_HELP_FAILED);
  429. }