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

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

防火墙与安全工具

开发平台：

Visual C++

AclSet.cpp：源码内容
  1. // AclSet.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Property.h"
  6. #include "AclSet.h"
  8. //owner add
  9. #include "GuiRes.h"
  10. #include "NetTimeSheet.h"
  11. #include "SetNet.h"
  12. #include "SetTime.h"
  14. #ifdef _DEBUG
  15. #define new DEBUG_NEW
  16. #undef THIS_FILE
  17. static char THIS_FILE[] = __FILE__;
  18. #endif
  20. //=============================================================================================
  21. // CAclSet dialog
  23. extern CAclFile m_AclFile;
  24. CNetTimeSheet *dlgNetTimeSheet = NULL;
  25. CFileDialog *dlgFile = NULL;
  27. XACL CAclSet::m_tmpAcl;
  28. BOOL CAclSet::m_EnableComboApplication = TRUE;
  29. CString CAclSet::m_sPathName;
  31. CAclSet::CAclSet(CWnd* pParent /*=NULL*/)
  32. : CDialog(CAclSet::IDD, pParent)
  33. {
  34. //{{AFX_DATA_INIT(CAclSet)
  35. //}}AFX_DATA_INIT
  36. }
  39. void CAclSet::DoDataExchange(CDataExchange* pDX)
  40. {
  41. CDialog::DoDataExchange(pDX);
  42. //{{AFX_DATA_MAP(CAclSet)
  43. DDX_Control(pDX, IDC_EDIT_MEMO, m_EditMemo);
  44. DDX_Control(pDX, IDC_EDIT_SERVICE_PORT, m_EditServicePort);
  45. DDX_Control(pDX, IDC_COMBO_DIRECTION, m_ComboDirection);
  46. DDX_Control(pDX, IDC_COMBO_ACCESS_TIME, m_ComboAccessTime);
  47. DDX_Control(pDX, IDC_COMBO_SERVICE_TYPE, m_ComboServiceType);
  48. DDX_Control(pDX, IDC_COMBO_ACTION, m_ComboAction);
  49. DDX_Control(pDX, IDC_COMBO_REMOTE_NET, m_ComboRemoteNet);
  50. DDX_Control(pDX, IDC_COMBO_APPLICATION, m_ComboApplication);
  51. //}}AFX_DATA_MAP
  52. }
  55. BEGIN_MESSAGE_MAP(CAclSet, CDialog)
  56. //{{AFX_MSG_MAP(CAclSet)
  57. ON_BN_CLICKED(IDC_BUTTON_NET, OnButtonNet)
  58. ON_BN_CLICKED(IDC_BUTTON_TIME, OnButtonTime)
  59. ON_BN_CLICKED(IDC_BUTTON_APPLICATION, OnButtonApplication)
  60. ON_CBN_SELCHANGE(IDC_COMBO_SERVICE_TYPE, OnSelchangeComboServiceType)
  61. ON_CBN_SELCHANGE(IDC_COMBO_APPLICATION, OnSelchangeComboApplication)
  62. //}}AFX_MSG_MAP
  63. END_MESSAGE_MAP()
  65. //=============================================================================================
  66. // CAclSet message handlers
  68. void CAclSet::OnButtonNet() 
  69. {
  70. ShowNetTimeSheet(1);
  71. }
  72. void CAclSet::OnButtonTime() 
  73. {
  74. ShowNetTimeSheet(2);
  75. }
  77. void CAclSet::ShowNetTimeSheet(int nID)
  78. {
  79. int tmpNetIndex = m_ComboRemoteNet.GetCurSel();
  80. int tmpTimeIndex= m_ComboAccessTime.GetCurSel();
  82. dlgNetTimeSheet = new CNetTimeSheet(GUI_NET_TIME_SHEET_CAPTION);
  83. CSetNet mSetNet;
  84. CSetTime mSetTime;
  86. dlgNetTimeSheet->AddPage(&mSetNet);
  87. dlgNetTimeSheet->AddPage(&mSetTime);
  89. if(nID == 1)
  90. {
  91. dlgNetTimeSheet->SetActivePage(&mSetNet);
  92. m_ComboRemoteNet.GetWindowText(mSetNet.m_TreeText);
  93. }
  94. else if(nID == 2)
  95. {
  96. dlgNetTimeSheet->SetActivePage(&mSetTime);
  97. m_ComboAccessTime.GetWindowText(mSetTime.m_TreeText);
  98. }
  99. else
  100. return;
  102. int iRet = dlgNetTimeSheet->DoModal();
  103. delete dlgNetTimeSheet;
  104. dlgNetTimeSheet = NULL;
  106. if(iRet == IDCANCEL)
  107. return;
  109. if(tmpNetIndex != mSetNet.iTreeIndex)
  110. m_ComboRemoteNet.SetCurSel(mSetNet.iTreeIndex);
  112. if(tmpTimeIndex != mSetTime.m_iTreeIndex)
  113. m_ComboAccessTime.SetCurSel(mSetTime.m_iTreeIndex);
  114. }
  116. BOOL CAclSet::OnInitDialog() 
  117. {
  118. CDialog::OnInitDialog();
  120. SetIcon(AfxGetApp()->LoadIcon(IDR_MAINFRAME), TRUE);
  121. InitDlgResource();
  123. if(!m_EnableComboApplication)
  124. {
  125. ModifyStyleEx(WS_EX_TOOLWINDOW, WS_EX_APPWINDOW); 
  126. ::SetWindowPos(this->m_hWnd,HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE);
  127. }
  129. return TRUE;  
  130. }
  132. //---------------------------------------------------------------------------------------------
  133. //initialize dialog's resource
  135. void CAclSet::InitDlgResource()
  136. {
  137. SetWindowText(GUI_ACL_SET_CAPTION);
  138. SetDlgItemText(IDOK ,GUI_BUTTON_OK);
  139. SetDlgItemText(IDCANCEL ,GUI_BUTTON_CANCEL);
  140. SetDlgItemText(IDC_ACL_APPLICATION_LABLE ,GUI_ACL_APPLICATION_LABLE);
  141. SetDlgItemText(IDC_ACL_REMOTE_NET_LABLE ,GUI_ACL_REMOTE_NET_LABLE);
  142. SetDlgItemText(IDC_ACL_ACCESS_TIME_LABLE ,GUI_ACL_ACCESS_TIME_LABLE);
  143. SetDlgItemText(IDC_ACL_ACTION_LABLE ,GUI_ACL_ACTION_LABLE);
  144. SetDlgItemText(IDC_ACL_SUPER_SET_LABLE ,GUI_ACL_SUPER_SET_LABLE);
  145. SetDlgItemText(IDC_ACL_DERECTION_LABLE ,GUI_ACL_DERECTION_LABLE);
  146. SetDlgItemText(IDC_ACL_SERVICE_TYPE_LABLE ,GUI_ACL_SERVICE_TYPE_LABLE);
  147. SetDlgItemText(IDC_ACL_SERVICE_PORT_LABLE ,GUI_ACL_SERVICE_PORT_LABLE);
  148. SetDlgItemText(IDC_ACL_MEMO_LABLE ,GUI_ACL_MEMO_LABLE);
  150. m_ComboRemoteNet.InsertString(0, GUI_SET_NET_TREE_0);
  151. m_ComboRemoteNet.InsertString(1, GUI_SET_NET_TREE_1);
  152. m_ComboRemoteNet.InsertString(2, GUI_SET_NET_TREE_2);
  153. m_ComboRemoteNet.InsertString(3, GUI_SET_NET_TREE_3);
  154. m_ComboRemoteNet.InsertString(4, GUI_SET_NET_TREE_4);
  156. m_ComboAccessTime.InsertString(0, GUI_SET_TIME_TREE_0);
  157. m_ComboAccessTime.InsertString(1, GUI_SET_TIME_TREE_1);
  158. m_ComboAccessTime.InsertString(2, GUI_SET_TIME_TREE_2);
  159. m_ComboAccessTime.InsertString(3, GUI_SET_TIME_TREE_3);
  160. m_ComboAccessTime.InsertString(4, GUI_SET_TIME_TREE_4);
  161. m_ComboAccessTime.InsertString(5, GUI_SET_TIME_TREE_5);
  162. m_ComboAccessTime.InsertString(6, GUI_SET_TIME_TREE_6);
  164. m_ComboAction.InsertString(0, GUI_ACL_ACTION_0);
  165. m_ComboAction.InsertString(1, GUI_ACL_ACTION_1);
  167. m_ComboDirection.InsertString(0, GUI_ACL_DERECTION_0);
  168. m_ComboDirection.InsertString(1, GUI_ACL_DERECTION_1);
  169. m_ComboDirection.InsertString(2, GUI_ACL_DERECTION_2);
  171. m_ComboServiceType.InsertString(0, GUI_ACL_SERVICE_TYPE_0);
  172. m_ComboServiceType.InsertString(1, GUI_ACL_SERVICE_TYPE_1);
  173. m_ComboServiceType.InsertString(2, GUI_ACL_SERVICE_TYPE_2);
  174. m_ComboServiceType.InsertString(3, GUI_ACL_SERVICE_TYPE_3);
  175. m_ComboServiceType.InsertString(4, GUI_ACL_SERVICE_TYPE_4);
  176. m_ComboServiceType.InsertString(5, GUI_ACL_SERVICE_TYPE_5);
  177. m_ComboServiceType.InsertString(6, GUI_ACL_SERVICE_TYPE_6);
  178. m_ComboServiceType.InsertString(7, GUI_ACL_SERVICE_TYPE_7);
  179. m_ComboServiceType.InsertString(8, GUI_ACL_SERVICE_TYPE_8);
  181. int i, n = m_AclFile.mAclHeader.ulAclCount;
  183. for(i = 0; i < n; i++)
  184. {
  185. if(m_ComboApplication.FindString(0,m_AclFile.mpAcl[i].sApplication) != CB_ERR)
  186. continue;
  187. m_ComboApplication.InsertString(m_ComboApplication.GetCount(),m_AclFile.mpAcl[i].sApplication);
  188. }
  190. if(m_EnableComboApplication)
  191. {
  192. if(_tcscmp(m_tmpAcl.sApplication, _T("")) == 0)
  193. {
  194. SetDlgItemText(IDC_ACL_BASE_SET_LABLE, GUI_ACL_BASE_SET_LABLE);
  195. }
  196. else
  197. {
  198. SetDlgItemText(IDC_ACL_BASE_SET_LABLE, m_tmpAcl.sApplication);
  199. m_ComboApplication .SetWindowText(m_tmpAcl.sApplication);
  200. }
  201. }
  202. else
  203. {
  204. CString tmpStr;
  205. tmpStr.Format("%s %s", m_sPathName, GUI_ACL_QUERY_INFO);
  206. SetDlgItemText(IDC_ACL_BASE_SET_LABLE, tmpStr);
  207. m_ComboApplication .SetWindowText(m_sPathName);
  208. }
  210. GetDlgItem(IDC_BUTTON_APPLICATION)->EnableWindow(m_EnableComboApplication);
  211. m_ComboApplication .EnableWindow(m_EnableComboApplication);
  212. m_ComboRemoteNet .SetCurSel(m_tmpAcl.bRemoteNetType);
  213. m_ComboAccessTime .SetCurSel(m_tmpAcl.bAccessTimeType);
  214. m_ComboAction .SetCurSel(m_tmpAcl.bAction);
  215. m_ComboDirection .SetCurSel(m_tmpAcl.bDirection);
  216. m_ComboServiceType .SetCurSel(m_tmpAcl.bServiceType);
  218. OnSelchangeComboServiceType();
  220. CString tmpStr;
  221. tmpStr .Format(_T("%u"), m_tmpAcl.uiServicePort);
  222. m_EditServicePort .SetLimitText(5);
  223. m_EditServicePort .SetWindowText(tmpStr);
  224. m_EditMemo .SetLimitText(50);
  225. m_EditMemo .SetWindowText(m_tmpAcl.sMemo);
  226. }
  228. void CAclSet::OnButtonApplication() 
  229. {
  230. static TCHAR BASED_CODE szFilter[] = _T("(*.exe)|*.exe||");
  232. dlgFile =  new CFileDialog(TRUE,NULL,NULL,OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST,szFilter);
  234. int iRet = dlgFile->DoModal();
  236. if(iRet == IDCANCEL)
  237. {
  238. delete dlgFile;
  239. dlgFile = NULL;
  240. return;
  241. }
  243. m_ComboApplication.SetWindowText(dlgFile->m_ofn.lpstrFile);
  245. delete  dlgFile;
  246. dlgFile = NULL;
  247. }
  249. void CAclSet::OnOK() 
  250. {
  251. CString tmpStrApp, tmpStrMemo, tmpStrPort;
  253. m_ComboApplication .GetWindowText(tmpStrApp);
  254. m_EditMemo .GetWindowText(tmpStrMemo);
  255. m_EditServicePort .GetWindowText(tmpStrPort);
  257. UINT tmpPort = _ttoi(tmpStrPort);
  259. if(tmpPort > 65535 || tmpPort < 0)
  260. {
  261. AfxMessageBox(GUI_ACL_MESSAGE_INAVALID_PORT);
  262. m_EditServicePort.SetFocus();
  263. return;
  264. }
  266. if( m_tmpAcl.bAccessTimeType== m_ComboAccessTime.GetCurSel() &&
  267. m_tmpAcl.bAction == m_ComboAction .GetCurSel() &&
  268. m_tmpAcl.bDirection == m_ComboDirection .GetCurSel() &&
  269. m_tmpAcl.bRemoteNetType == m_ComboRemoteNet .GetCurSel() &&
  270. m_tmpAcl.bServiceType == m_ComboServiceType.GetCurSel() &&
  271. m_tmpAcl.uiServicePort == tmpPort &&
  272. tmpStrApp.CompareNoCase(m_tmpAcl.sApplication) == 0 &&
  273. tmpStrMemo.Compare(m_tmpAcl.sMemo) == 0 )
  274. {
  275. CDialog::OnCancel();
  276. return;
  277. }
  279. if(tmpStrApp == "")
  280. {
  281. AfxMessageBox(GUI_ACL_MESSAGE_APP_PATH_ERROR);
  282. m_ComboApplication.SetFocus();
  283. return;
  284. }
  286. if((tmpStrApp.Right(1) == "\") || (tmpStrApp.Right(1) == ":"))
  287. {
  288. AfxMessageBox(GUI_ACL_MESSAGE_ONLY_PATH);
  289. m_ComboApplication.SetFocus();
  290. return;
  291. }
  293. if(_taccess(tmpStrApp,0) == -1)
  294. {
  295. AfxMessageBox(GUI_ACL_MESSAGE_APP_NOT_EXSITS);
  296. m_ComboApplication.SetFocus();
  297. return;
  298. }
  300. m_tmpAcl.bAccessTimeType= m_ComboAccessTime .GetCurSel();
  301. m_tmpAcl.bAction = m_ComboAction .GetCurSel();
  302. m_tmpAcl.bDirection = m_ComboDirection .GetCurSel();
  303. m_tmpAcl.bRemoteNetType = m_ComboRemoteNet .GetCurSel();
  304. m_tmpAcl.bServiceType = m_ComboServiceType.GetCurSel();
  305. m_tmpAcl.uiServicePort = tmpPort;
  307. _tcscpy(m_tmpAcl.sApplication, tmpStrApp);
  308. _tcscpy(m_tmpAcl.sMemo, tmpStrMemo);
  310. CDialog::OnOK();
  311. }
  313. void CAclSet::OnSelchangeComboServiceType() 
  314. {
  315. int iIndex = m_ComboServiceType.GetCurSel();
  316. UINT mPort = ACL_SERVICE_PORT_ALL;
  317. BOOL bEnable = TRUE;
  319. switch(iIndex)
  320. {
  321. case ACL_SERVICE_TYPE_ALL:
  322. bEnable = FALSE;
  323. break;
  325. case ACL_SERVICE_TYPE_TCP:
  326. break;
  328. case ACL_SERVICE_TYPE_UDP:
  329. break;
  331. case ACL_SERVICE_TYPE_FTP:
  332. mPort = ACL_SERVICE_PORT_FTP;
  333. bEnable = FALSE;
  334. break;
  336. case ACL_SERVICE_TYPE_TELNET:
  337. mPort = ACL_SERVICE_PORT_TELNET;
  338. bEnable = FALSE;
  339. break;
  341. case ACL_SERVICE_TYPE_HTTP:
  342. break;
  344. case ACL_SERVICE_TYPE_NNTP:
  345. mPort = ACL_SERVICE_PORT_NNTP;
  346. bEnable = FALSE;
  347. break;
  349. case ACL_SERVICE_TYPE_POP3:
  350. mPort = ACL_SERVICE_PORT_POP3;
  351. bEnable = FALSE;
  352. break;
  354. case ACL_SERVICE_TYPE_SMTP:
  355. mPort = ACL_SERVICE_PORT_SMTP;
  356. bEnable = FALSE;
  357. break;
  359. default:
  360. return;
  361. }
  363. CString tmpStr;
  364. tmpStr .Format(_T("%u"),mPort);
  365. m_EditServicePort.SetWindowText(tmpStr);
  366. m_EditServicePort.EnableWindow(bEnable);
  367. }
  369. void CAclSet::OnSelchangeComboApplication() 
  370. {
  371. CString tmpStr;
  373. m_ComboApplication.GetLBText(m_ComboApplication.GetCurSel(),tmpStr);
  374. SetDlgItemText(IDC_ACL_BASE_SET_LABLE, tmpStr);
  375. }