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

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

防火墙与安全工具

开发平台：

Visual C++

SetNet.cpp：源码内容
  1. // SetNet.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Property.h"
  6. #include "SetNet.h"
  8. //onwer add
  9. #include "GuiRes.h"
  10. #include "NetIPAria.h"
  11. #include "NetTimeSheet.h"
  13. #ifdef _DEBUG
  14. #define new DEBUG_NEW
  15. #undef THIS_FILE
  16. static char THIS_FILE[] = __FILE__;
  17. #endif
  19. //=============================================================================================
  20. // CSetNet property page
  22. extern CAclFile m_AclFile;
  23. XACL_IP m_tmpIP;
  24. XACL_IP* m_pIP;
  25. CNetTimeSheet* mNetTimeSheet;
  26. CNetIPAria *dlgNetIPAria = NULL;
  28. IMPLEMENT_DYNCREATE(CSetNet, CPropertyPage)
  30. CSetNet::CSetNet() : CPropertyPage(CSetNet::IDD)
  31. {
  32. iListIndex = 0;
  33. iTreeIndex = 0;
  34. m_TreeText = GUI_SET_NET_TREE_0;
  35. }
  37. CSetNet::~CSetNet()
  38. {
  39. }
  41. void CSetNet::DoDataExchange(CDataExchange* pDX)
  42. {
  43. CPropertyPage::DoDataExchange(pDX);
  44. //{{AFX_DATA_MAP(CSetNet)
  45. DDX_Control(pDX, IDC_SET_NET_TREE, m_TreeSetNet);
  46. DDX_Control(pDX, IDC_SET_NET_LIST, m_ListSetNet);
  47. //}}AFX_DATA_MAP
  48. }
  51. BEGIN_MESSAGE_MAP(CSetNet, CPropertyPage)
  52. //{{AFX_MSG_MAP(CSetNet)
  53. ON_BN_CLICKED(IDC_SET_NET_BUTTON_ADD, OnSetNetButtonAdd)
  54. ON_NOTIFY(TVN_SELCHANGED, IDC_SET_NET_TREE, OnSelchangedSetNetTree)
  55. ON_NOTIFY(LVN_ITEMCHANGED, IDC_SET_NET_LIST,OnItemchangedSetNetList)
  56. ON_BN_CLICKED(IDC_SET_NET_BUTTON_EDIT, OnSetNetButtonEdit)
  57. ON_NOTIFY(NM_DBLCLK, IDC_SET_NET_LIST, OnDblclkSetNetList)
  58. ON_BN_CLICKED(IDC_SET_NET_BUTTON_DELETE, OnSetNetButtonDelete)
  59. ON_NOTIFY(NM_CLICK, IDC_SET_NET_LIST, OnClickSetNetList)
  60. //}}AFX_MSG_MAP
  61. END_MESSAGE_MAP()
  63. //=============================================================================================
  64. // CSetNet message handlers
  66. BOOL CSetNet::OnInitDialog() 
  67. {
  68. CPropertyPage::OnInitDialog();
  70. InitDlgResource();
  71. mNetTimeSheet = (CNetTimeSheet*)this->GetOwner();
  73. return TRUE;  
  74. }
  76. //---------------------------------------------------------------------------------------------
  77. //initialize dialog's resource
  79. void CSetNet::InitDlgResource()
  80. {
  81. SetDlgItemText(IDC_SET_NET_NET_LABLE ,GUI_SET_NET_NET_LABLE);
  82. SetDlgItemText(IDC_SET_NET_BUTTON_ADD ,GUI_SET_NET_BUTTON_ADD);
  83. SetDlgItemText(IDC_SET_NET_BUTTON_EDIT ,GUI_SET_NET_BUTTON_EDIT);
  84. SetDlgItemText(IDC_SET_NET_BUTTON_DELETE,GUI_SET_NET_BUTTON_DELETE);
  86. m_TreeSetNet.InsertItem(GUI_SET_NET_TREE_0);
  87. m_TreeSetNet.InsertItem(GUI_SET_NET_TREE_1);
  88. m_TreeSetNet.InsertItem(GUI_SET_NET_TREE_2);
  89. m_TreeSetNet.InsertItem(GUI_SET_NET_TREE_3);
  90. m_TreeSetNet.InsertItem(GUI_SET_NET_TREE_4);
  92. InitList(&m_ListSetNet);
  93. SelectTreeItem(m_TreeText);
  94. }
  95. void CSetNet::InitList(CListCtrl *mList)
  96. {
  97. mList->SetExtendedStyle (LVS_EX_GRIDLINES  | LVS_EX_FULLROWSELECT);
  98. mList->InsertColumn (0, GUI_SET_NET_LIST_START_IP, LVCFMT_LEFT,143);
  99. mList->InsertColumn (1, GUI_SET_NET_LIST_END_IP , LVCFMT_LEFT,143);
  100. }
  102. void CSetNet::OnSelchangedSetNetTree(NMHDR* pNMHDR, LRESULT* pResult) 
  103. {
  104. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  106. SelectTreeItemOperate();
  108. *pResult = 0;
  109. }
  111. int CSetNet::SelectTreeItemOperate()
  112. {
  113. DWORD i = 0;
  114. CString sNetText= m_TreeSetNet.GetItemText(m_TreeSetNet.GetSelectedItem());
  115. iTreeIndex = NetTextToIndex(sNetText);
  117. m_ListSetNet.DeleteAllItems();
  119. switch(iTreeIndex)
  120. {
  121. case 0:
  122. EnableButton(FALSE);
  123. m_ListSetNet.InsertItem(0,GUI_NET_IP_ARIA_ALL);
  124. m_ListSetNet.SetItemText(0,1,GUI_NET_IP_ARIA_ALL);
  125. break;
  127. case 1:
  128. EnableButton(FALSE);
  129. ListAddIP(&m_AclFile.mAclIntranetIP, m_AclFile.mAclHeader.ulIntranetIPCount);
  130. break;
  132. case 2:
  133. EnableButton(TRUE);
  134. ListAddIP(m_AclFile.mpAclDistrustIP, m_AclFile.mAclHeader.ulDistrustIPCount);
  135. break;
  137. case 3:
  138. EnableButton(TRUE);
  139. ListAddIP(m_AclFile.mpAclTrustIP, m_AclFile.mAclHeader.ulTrustIPCount);
  140. break;
  142. case 4:
  143. EnableButton(TRUE);
  144. ListAddIP(m_AclFile.mpAclCustomIP, m_AclFile.mAclHeader.ulCustomIPCount);
  145. break;
  147. default:
  148. break;
  149. }
  151. m_ListSetNet.SetItemState(0,LVIS_SELECTED,LVIS_SELECTED);
  153. return 0;
  154. }
  156. int CSetNet::ListAddIP(void* mpAclIP,DWORD mCount)
  157. {
  158. DWORD i = 0;
  160. for(i = 0; i < mCount; i++)
  161. ListAddOneIP(i, mpAclIP);
  163. return 0;
  164. }
  166. int CSetNet::ListAddOneIP(DWORD iIndex, void* mpAclIP, BOOL isSelected)
  167. {
  168. XACL_IP *mAclIP = (XACL_IP*)mpAclIP;
  170. if(isSelected)
  171. {
  172. m_ListSetNet.InsertItem(iIndex, m_AclFile.DIPToSIP(&mAclIP->ulStartIP));
  173. m_ListSetNet.SetItemText(iIndex, 1, m_AclFile.DIPToSIP(&mAclIP->ulEndIP));
  174. m_ListSetNet.EnsureVisible(iIndex, TRUE);
  175. m_ListSetNet.SetItemState(iIndex, LVIS_SELECTED,LVIS_SELECTED);
  176. }
  177. else
  178. {
  179. m_ListSetNet.InsertItem(iIndex, m_AclFile.DIPToSIP(&mAclIP[iIndex].ulStartIP));
  180. m_ListSetNet.SetItemText(iIndex, 1, m_AclFile.DIPToSIP(&mAclIP[iIndex].ulEndIP));
  181. }
  183. return 0;
  184. }
  186. int CSetNet::EnableButton(BOOL bEnable,BOOL bEnableEx)
  187. {
  188. GetDlgItem(IDC_SET_NET_BUTTON_ADD) ->EnableWindow(bEnable);
  189. GetDlgItem(IDC_SET_NET_BUTTON_EDIT) ->EnableWindow(bEnableEx);
  190. GetDlgItem(IDC_SET_NET_BUTTON_DELETE) ->EnableWindow(bEnableEx);
  192. return 0;
  193. }
  195. int CSetNet::NetTextToIndex(CString sNetText)
  196. {
  197. if(sNetText.Compare(GUI_SET_NET_TREE_0) == 0) return 0;
  198. if(sNetText.Compare(GUI_SET_NET_TREE_1) == 0) return 1;
  199. if(sNetText.Compare(GUI_SET_NET_TREE_2) == 0) return 2;
  200. if(sNetText.Compare(GUI_SET_NET_TREE_3) == 0) return 3;
  201. if(sNetText.Compare(GUI_SET_NET_TREE_4) == 0) return 4;
  203. return -1;
  204. }
  206. int CSetNet::SelectTreeItem(CString sNetText)
  207. {
  208. int i = 0 , iItemCount = m_TreeSetNet.GetCount();
  209. HTREEITEM mItem = m_TreeSetNet.GetFirstVisibleItem();
  211. if(sNetText.Compare(m_TreeSetNet.GetItemText(mItem)) == 0)
  212. return i;
  214. for(i = 1; i < iItemCount; i++)
  215. {
  216. mItem = m_TreeSetNet.GetNextSiblingItem(mItem);
  218. if(sNetText.Compare(m_TreeSetNet.GetItemText(mItem)) == 0)
  219. {
  220. m_TreeSetNet.SelectItem(mItem);
  221. return i;
  222. }
  223. }
  225. return -1;
  226. }
  228. void CSetNet::OnItemchangedSetNetList(NMHDR* pNMHDR, LRESULT* pResult) 
  229. {
  230. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  231. *pResult = 0;
  233. if((iListIndex = pNMListView->iItem) == -1)
  234. return;
  236. if(iTreeIndex != 0 && iTreeIndex != 1)
  237. EnableButton(TRUE,TRUE);
  239. if(iTreeIndex == 2)
  240. m_pIP = &m_AclFile.mpAclDistrustIP[iListIndex];
  241. else if(iTreeIndex == 3)
  242. m_pIP = &m_AclFile.mpAclTrustIP[iListIndex];
  243. else if(iTreeIndex == 4)
  244. m_pIP = &m_AclFile.mpAclCustomIP[iListIndex];
  246. return;
  247. }
  249. void CSetNet::OnSetNetButtonAdd() 
  250. {
  251. if(!GetDlgItem(IDC_SET_NET_BUTTON_ADD)->IsWindowEnabled())
  252. return;
  254. if(MAX_IP_ARIA != 0 && m_ListSetNet.GetItemCount() >= MAX_IP_ARIA)
  255. {
  256. AfxMessageBox(GUI_ACL_MESSAGE_MAX_IP_ARIA);
  257. return;
  258. }
  260. dlgNetIPAria = new CNetIPAria;
  261. dlgNetIPAria->iState= 0; //add
  262. int iRet = dlgNetIPAria->DoModal();
  263. m_tmpIP.ulStartIP = dlgNetIPAria->ulStartIP;
  264. m_tmpIP.ulEndIP = dlgNetIPAria->ulEndIP;
  266. delete dlgNetIPAria;
  267. dlgNetIPAria = NULL;
  269. if(iRet == IDCANCEL)
  270. return;
  272. if(m_AclFile.AddAcl(&m_tmpIP,1,iTreeIndex) != XERR_SUCCESS)
  273. {
  274. AfxMessageBox(GUI_ACL_MESSAGE_ADD_ACL_IP_ERROR);
  275. return;
  276. }
  278. ListAddOneIP(m_ListSetNet.GetItemCount(),&m_tmpIP,TRUE);
  279. mNetTimeSheet->EnableButton(TRUE);
  280. }
  282. void CSetNet::OnSetNetButtonEdit() 
  283. {
  284. if(!GetDlgItem(IDC_SET_NET_BUTTON_EDIT)->IsWindowEnabled())
  285. return;
  287. if(iListIndex < 0)
  288. return;
  290. dlgNetIPAria = new CNetIPAria;
  291. dlgNetIPAria->iState = 1; //edit
  292. dlgNetIPAria->ulStartIP = m_pIP->ulStartIP ;
  293. dlgNetIPAria->ulEndIP = m_pIP->ulEndIP ;
  295. int iRet = dlgNetIPAria->DoModal();
  296. m_pIP->ulStartIP= dlgNetIPAria->ulStartIP;
  297. m_pIP->ulEndIP = dlgNetIPAria->ulEndIP;
  298. delete dlgNetIPAria;
  299. dlgNetIPAria = NULL;
  301. if(iRet == IDCANCEL)
  302. return;
  304. m_ListSetNet.SetItemText(iListIndex, 0, m_AclFile.DIPToSIP(&m_pIP->ulStartIP));
  305. m_ListSetNet.SetItemText(iListIndex, 1, m_AclFile.DIPToSIP(&m_pIP->ulEndIP));
  306. mNetTimeSheet->EnableButton(TRUE);
  307. }
  309. void CSetNet::OnDblclkSetNetList(NMHDR* pNMHDR, LRESULT* pResult) 
  310. {
  311. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  313. if((iListIndex = pNMListView->iItem) != -1)
  314. OnSetNetButtonEdit();
  316. *pResult = 0;
  317. }
  319. void CSetNet::OnSetNetButtonDelete() 
  320. {
  321. if(iListIndex < 0)
  322. return;
  324. int tmpIndex = iListIndex;
  326. m_AclFile .DelAcl(iListIndex, iTreeIndex);
  327. m_ListSetNet.DeleteItem(iListIndex);
  328. mNetTimeSheet->EnableButton(TRUE);
  330. if(m_ListSetNet.GetItemCount() <= 0)
  331. {
  332. EnableButton(TRUE,FALSE);
  333. return;
  334. }
  336. if(tmpIndex == m_ListSetNet.GetItemCount())
  337. tmpIndex -- ;
  339. m_ListSetNet.SetItemState(tmpIndex,LVIS_SELECTED,LVIS_SELECTED);
  340. }
  342. void CSetNet::OnClickSetNetList(NMHDR* pNMHDR, LRESULT* pResult) 
  343. {
  344. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  346. if((iListIndex = pNMListView->iItem) == -1)
  347. EnableButton(iTreeIndex > 1 ? TRUE : FALSE, FALSE);
  349. *pResult = 0;
  350. }