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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
MainFrm.cpp
资源名称：VC_SNMP_NetManage.rar　[点击查看]
上传用户：hbytqc8
上传日期：2014-07-31
资源大小：527k
文件大小：14k
源码类别：

SNMP编程

开发平台：

Visual C++

MainFrm.cpp：源码内容
  1. // MainFrm.cpp : implementation of the CMainFrame class
  2. //
  3. #include "stdafx.h"
  4. #include "NetMap.h"
  5. #include "NetMapDoc.h"
  6. #include "MainFrm.h"
  7. #include "snmp_pp.h"
  8. #include "Show.h"
  9. #include "Modify.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #undef THIS_FILE
  13. static char THIS_FILE[] = __FILE__;
  14. #endif
  16. /////////////////////////////////////////////////////////////////////////////
  17. // CMainFrame
  19. IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
  21. BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
  22. //{{AFX_MSG_MAP(CMainFrame)
  23. ON_WM_CREATE()
  24. ON_COMMAND(ID_SCAN, OnScan)
  25. ON_COMMAND(ID_GRAPH, OnGraph)
  26. ON_COMMAND(ID_SHOWINFORMATION, OnShowinformation)
  27. ON_COMMAND(ID_NODE_MODIFY, OnNodeModify)
  28. //}}AFX_MSG_MAP
  29. END_MESSAGE_MAP()
  31. static UINT indicators[] =
  32. {
  33. ID_SEPARATOR,           // status line indicator
  34. ID_INDICATOR_CAPS,
  35. ID_INDICATOR_NUM,
  36. ID_INDICATOR_SCRL,
  37. };
  39. /////////////////////////////////////////////////////////////////////////////
  40. // CMainFrame construction/destruction
  42. CMainFrame::CMainFrame()
  43. {
  44. // TODO: add member initialization code here
  46. }
  48. CMainFrame::~CMainFrame()
  49. {
  50. }
  52. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  53. {
  54. if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
  55. return -1;
  57. if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
  58. | CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
  59. !m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
  60. {
  61. TRACE0("Failed to create toolbarn");
  62. return -1;      // fail to create
  63. }
  65. if (!m_wndStatusBar.Create(this) ||
  66. !m_wndStatusBar.SetIndicators(indicators,
  67.   sizeof(indicators)/sizeof(UINT)))
  68. {
  69. TRACE0("Failed to create status barn");
  70. return -1;      // fail to create
  71. }
  73. // TODO: Delete these three lines if you don't want the toolbar to
  74. //  be dockable
  75. m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
  76. EnableDocking(CBRS_ALIGN_ANY);
  77. DockControlBar(&m_wndToolBar);
  79. return 0;
  80. }
  82. BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
  83. {
  84. if( !CFrameWnd::PreCreateWindow(cs) )
  85. return FALSE;
  86. // TODO: Modify the Window class or styles here by modifying
  87. //  the CREATESTRUCT cs
  88. //cs.style=WS_MAXIMIZE;
  89. cs.x=4;
  90. cs.y=4;
  91.     cs.cx=800;
  92. cs.cy=750;
  93. return TRUE;
  94. }
  95. BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext)
  96. {
  98. BOOL Result=v.CreateStatic(this,1,2,WS_CHILD | WS_VISIBLE);
  99. ASSERT(Result);
  100. h.CreateStatic(&v,2,1,WS_CHILD | WS_VISIBLE,v.IdFromRowCol(0,1));
  101. v.CreateView(0,0,RUNTIME_CLASS(CMapView),CSize(237,100),pContext);
  102. h.CreateView(0,0,RUNTIME_CLASS(CNetMapView),CSize(400,400),pContext);
  103. h.CreateView(1,0,RUNTIME_CLASS(CVlanView),CSize(400,400),pContext);
  104. return Result;
  105. }
  106. /////////////////////////////////////////////////////////////////////////////
  107. // CMainFrame diagnostics
  109. #ifdef _DEBUG
  110. void CMainFrame::AssertValid() const
  111. {
  112. CFrameWnd::AssertValid();
  113. }
  115. void CMainFrame::Dump(CDumpContext& dc) const
  116. {
  117. CFrameWnd::Dump(dc);
  118. }
  120. #endif //_DEBUG
  122. /////////////////////////////////////////////////////////////////////////////
  123. // CMainFrame message handlers
  124. //根据text属性查找特定的节点是否存在，如找到，返回符合条件节点的句柄，
  125. //否则返回NULL；
  126. HTREEITEM CMainFrame::FindNode(HTREEITEM hSubRoot,const CString &text)
  127. {
  128.     if (hSubRoot==NULL) 
  129. return NULL;
  130. //判断是否符合条件，如是，返回该节点的举柄；
  131.     if (pListView->m_map.GetItemText(hSubRoot)==text)
  132. return hSubRoot;
  133. hSubRoot=pListView->m_map.GetChildItem(hSubRoot);
  134. do 
  135. {
  136.        HTREEITEM ht;
  137.    //递归调用FindNode；
  138.    if ((ht=FindNode(hSubRoot,text)) !=NULL)
  139.    return ht;
  141. } while ((hSubRoot=pListView->m_map.GetNextSiblingItem(hSubRoot)) !=NULL);
  142.     return NULL;
  143. }
  144. bool CMainFrame::Findport(CString &port,NodeData *p)
  145. {
  146. for (int i=0;i<(p->m_neighbor.GetSize());i++)
  147. {
  148. if (p->m_neighbor.GetAt(i).localport==port)
  149. return true;
  150. }
  151. return false;
  152. }
  153. //将CISCO格式的网络地址转换为正常格式的IP地址；
  154. CString CMainFrame::HtoIP(CString Hex)
  155. {   
  157.       
  158. char ch[4];
  159. CString ip;
  160.     if (Hex.GetLength()<11)
  161. {
  162. ip="NULL";
  163. return ip;
  164. }
  165. CString Hex1;
  166. Hex.Replace(" ","");
  167. for (int i=0;i<4;i++)
  168.     {
  169.     Hex1=Hex.Mid(i*2,2);
  170. int data=0;
  171.     if((Hex1[0]>='A') && (Hex1[0]<='F')) 
  172.     data=data+Hex1[0]-'A'+10;
  173.     if((Hex1[0]>='0') && (Hex1[0]<='9')) 
  174.     data=data+Hex1[0]-'0';
  175.     data=data*16;
  176. int data1=0;
  177.     if((Hex1[1]>='A') && (Hex1[1]<='F')) 
  178.     data1=data1+Hex1[1]-'A'+10;
  179.     if((Hex1[1]>='0') && (Hex1[1]<='9')) 
  180.     data1=data1+Hex1[1]-'0';
  181. data=data+data1;
  182. itoa(data,ch,10);
  183. ip=ip+ch+'.';
  184. }
  185.     ip=ip.Left(ip.GetLength()-1);
  186. return ip;
  187. }
  188. //扫描获取一个节点中CDP信息的函数，
  189. void CMainFrame::ScanNode(CString &add,CString &community,HTREEITEM hparent)
  190. {
  191. char ch[3];
  192. neighbor n1;
  193. CString BaseOid;
  194. NodeData *p1;
  195. m_snmp.clear();
  196. //清除原来的数据；
  197. for (int i=4;i<10;i++)
  198. {
  199.  BaseOid="1.3.6.1.4.1.9.9.23.1.2.1.1.";
  200.  itoa(i,ch,10);
  201.  BaseOid=BaseOid+ch;
  202.  if (m_snmp.GetTree(add,community,BaseOid)==false)
  203.  {
  204.  MessageBox("获取数据失败，请检查网络和SNMP配置参数!");
  205.  return;
  206.  }
  207. }
  208. //(pNodeData *)(pListView->m_map.GetItemData(hparent))->scan=true;
  209. //将CDP条目表中第4~9六个列对象的实例值取回，并存放在一个MyStringArray
  210. //对象中。关于这些列对象的具体含义请参考本章前面的内容；
  211. p1=(NodeData *)(pListView->m_map.GetItemData(hparent));
  212. p1->scan=true;
  213. //设置已经扫描标志；
  214. int row=m_snmp.GetCount()/6;
  215. CString h;
  216. int j;
  217. for (j=0; j<row;j++)
  218. {
  219. h=m_snmp.GetValue(j);
  220. h.TrimLeft();
  221. h.TrimRight();
  222. h=h.Left(11);
  223. m_snmp.SetValue(j,HtoIP(h));
  224. }
  225. //将得到的IP地址格式转换为正常的格式；
  226. for (j=0; j<row;j++)
  227. {
  228. CString remote="NULL";
  229. CString Temp_oid=m_snmp.GetOid(j);
  230. Temp_oid=Temp_oid.Mid(BaseOid.GetLength()+1,8);
  231. Temp_oid=Temp_oid.Left(Temp_oid.Find('.'));
  232. Temp_oid.TrimLeft();
  233. Temp_oid.TrimRight();
  234. Temp_oid="1.3.6.1.2.1.2.2.1.2."+Temp_oid;
  235. m_snmp.GetOne(add,community,Temp_oid,remote);
  236.     m_snmp.Add(Temp_oid,remote);
  237. }
  238. //获得本地端口名，既通过哪个本地端口获得的该条CDP信息；
  239. int pos=0;
  240. //MessageBox("scaned: "+p1->ip);
  241. for (j=0; j<row;j++)
  242. //对获得的每一份邻居节点信息进行处理。正常情况下，每个CDP条目就是
  243. //一台新发现的设备，都应该在设备树增加一个节点，并在新增的节点中
  244. //保存到本节点的连接信息。但符合下列情况的CDP条目不是新发现设备，
  245. //要做特殊处理：
  246. //   1：和父节点连接的CDP条目；
  247. //   2：和同一节点有多条连接；
  248. //   3：环路；
  249. {    
  250. HTREEITEM h;
  251.     if (p1->m_neighbor.GetSize()>0)
  252.    if (Findport(m_snmp.GetValue(j+6*row),p1))
  253.  
  254.    continue;
  255.  
  256. //和父节点连接的CDP条目，在扫描父节点时这些信息已经保存在本地
  257. //节点中，跳过；
  258.     h=FindNode(pListView->m_map.GetRootItem(),m_snmp.GetValue(j));
  259.     if (h!=NULL)
  260. {
  261.      neighbor n2;
  262.      NodeData * p2=(NodeData *)(pListView->m_map.GetItemData(h));
  263.  n2.Id=hparent;
  264.  n2.localport=m_snmp.GetValue(j+3*row);
  265.  n2.remoteport=m_snmp.GetValue(j+6*row);
  266.  p2->m_neighbor.Add(n2);
  267.      continue;
  268. }
  269. //已经存在的节点，只增加连接信息，不增加新的节点；
  270. pNodeData=new NodeData;
  271. pNodeData->ip=m_snmp.GetValue(j);
  272. pNodeData->sysDescr=m_snmp.GetValue(j+row);
  273. pNodeData->scan=false;
  274. pNodeData->router=false;
  275. pNodeData->level=p1->level+1;
  276. pNodeData->pImage=NULL;
  277. pNodeData->position=p1->position+pos;
  278. pos+=1;
  279. //对任一子节点，层次数等于父节点层次数加1；横向位置以父节点
  280. //为基础，依次累加；
  281. CString rs=m_snmp.GetValue(j+5*row);
  282. rs=rs.Mid(9,2);
  283. if ((atoi(rs)%2)==1)
  284. {
  285. pNodeData->scan=true;
  286.     pNodeData->router=true;
  287. }
  288. if (atoi(rs)>=8)
  289.        pNodeData->scan=false;  
  290. n1.Id=hparent;
  291. n1.localport=m_snmp.GetValue(j+3*row);
  292. n1.remoteport=m_snmp.GetValue(j+6*row);
  293. pNodeData->m_neighbor.Add(n1);
  294.     h=pListView->m_map.InsertItem(m_snmp.GetValue(j),hparent);
  295. pListView->m_map.SetItemData(h,(DWORD)pNodeData);
  296.     NodeArray.Add(h);
  297.  }
  299. }
  301. void CMainFrame::OnScan() 
  302. {   
  303. if (NodeArray.GetSize()>0)
  304. NodeArray.RemoveAll();
  305. if (pListView->m_map.GetCount()>0)
  306.         pListView->m_map.DeleteAllItems();
  307. //清除上次扫描数据；
  308.     NodeData *p1;
  309. CNetMapDoc *pDoc=(CNetMapDoc *)this->GetActiveDocument();
  310. if (pDoc->ip.GetLength()==0||pDoc->community.GetLength()==0)
  311. {
  312. MessageBox("设备地址,Community信息错误!");
  313. return;
  314. }
  315. CString sysoid="1.3.6.1.2.1.1.1.0";
  316. CString reply;
  317. m_snmp.GetOne(pDoc->ip,pDoc->community,sysoid,reply);
  318.     pNodeData=new NodeData;
  319. pNodeData->ip=pDoc->ip;
  320. pNodeData->sysDescr=reply;
  321. pNodeData->scan=false;
  322. pNodeData->pImage=NULL;
  323. pNodeData->level=0;
  324. pNodeData->position=0;
  325. HRoot=pListView->m_map.InsertItem(pDoc->ip);
  326.     NodeArray.Add(HRoot);
  327. pListView->m_map.SetItemData(HRoot,(DWORD)pNodeData);
  328. HCurrent=HRoot;
  329. int NodeNumber;
  330. int i;
  331. while (1)
  332.     {
  333.       NodeNumber=NodeArray.GetSize();
  334.   for(i=0;i<NodeNumber;i++)
  335.   {
  336.        p1=(NodeData *)(pListView->m_map.GetItemData(NodeArray[i]));
  337.    if (p1->scan==true) 
  338.        continue;
  339.    ScanNode(p1->ip,pDoc->community,NodeArray[i]);
  340.       }
  341.   if (NodeNumber==NodeArray.GetSize())
  342.   break;
  343. }
  344.     
  345. }
  347. void CMainFrame::OnGraph() 
  348. {
  349.  NodeData *p1;
  350.  int NodeNumber=NodeArray.GetSize();
  351.  if (NodeNumber==0) return;
  352.  //没有图形数据，退出；
  353.  p1=(NodeData *)(pListView->m_map.GetItemData(NodeArray[0]));
  354.  if (p1->pImage!=NULL) return;
  355.  //已经生成图形，退出；
  356.  int x=80;
  357.  int y=50;
  358.  //CRect cr;
  359. CBitmap bmpLoad; 
  360.  for(int i=0;i<NodeNumber;i++)
  361.  {
  362.   p1=(NodeData *)(pListView->m_map.GetItemData(NodeArray[i]));
  363.   CRect cr(x+100*p1->position,y+80*p1->level,65,35);
  364.   p1->pImage=new CMyImage();
  365.   p1->pImage->Create(NULL,WS_CHILD|SS_BITMAP|WS_VISIBLE|SS_NOTIFY,
  366.                    cr,pNetMapView,IDC_MYIMAGE+i);
  367.   HBITMAP hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP2));
  368.   if (p1->router==true)
  369.   hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP1));
  370.   p1->pImage->SetBitmap(hBitmap);
  371.   p1->pImage->MyPos.x=cr.left+33;
  372.   p1->pImage->MyPos.y=cr.top+12;
  373.   //保存生成连接线时的点坐标；
  374.   p1->pImage->HSelf=NodeArray[i];
  375.   //保存在树形控件中的举柄；
  376.  }
  377. pNetMapView->paint=true;
  378. pNetMapView->RedrawWindow();
  379. }
  381. void CMainFrame::OnShowinformation() 
  382. {
  383.  NodeData * p1=(NodeData *)(pListView->m_map.GetItemData(HSelect));
  384.  CShow dlg;
  385.  dlg.m_descr=p1->sysDescr;
  386.  CString local="";
  387.  CString remote="";
  388.  CString ip="";
  389.  for (int i=0;i<p1->m_neighbor.GetSize();i++)
  390.  {
  391.  ip+=pListView->m_map.GetItemText(p1->m_neighbor[i].Id)+"  ";
  392.  local+=p1->m_neighbor[i].localport+"  ";
  393.  remote+=p1->m_neighbor[i].remoteport+"  ";
  394.  }
  395.  dlg.m_localport=local;
  396.  dlg.m_remotelocal=remote;
  397.  dlg.m_ip=ip;
  398.  dlg.DoModal();
  399. }
  401. void CMainFrame::OnNodeModify() 
  402. {
  403. HTREEITEM h;
  404.     NodeData *p1;
  405.     NodeData *p2;
  406. CString Lip,Lport,Rip,Rport;
  407.     CModify dlg;
  408. if(dlg.DoModal()==IDOK)
  409. {
  410. Lip=dlg.m_LocalIp;
  411. Lport=dlg.m_LocalPort;
  412. Rip=dlg.m_RemoteIp;
  413. Rport=dlg.m_RemotePort;
  414. if (dlg.add==true)
  415. {
  416. h=FindNode(pListView->m_map.GetRootItem(),Lip);
  417. if (h!=NULL)
  418. {
  419. MessageBox("要增加的节点已经存在!");
  420. return;
  421. }
  422. h=FindNode(pListView->m_map.GetRootItem(),Rip);
  423. if (h==NULL)
  424. {
  425. MessageBox("上一级节点不存在!");
  426. return;
  427. }
  428. p2=(NodeData *)(pListView->m_map.GetItemData(h));
  429. p1=new NodeData;
  430. p1->ip=Lip;
  431. p1->level=p2->level+1;
  432. p1->position=p2->position;
  433. p1->sysDescr="增加的节点，IP："+Lip;
  434. if (dlg.m_router==1)
  435. p1->router=true;;
  436. p1->scan=true;
  437. neighbor n1;
  438. n1.Id=h;
  439. n1.localport=Lport;
  440. n1.remoteport=Rport; 
  441. p1->m_neighbor.Add(n1);
  442. //设置邻居节点信息；
  443. HCurrent=pListView->m_map.InsertItem(Lip,h);
  444. NodeArray.Add(HCurrent);
  445. //在tree中增加节点，父节点为制定节点；
  446. h=pListView->m_map.GetPrevSiblingItem(HCurrent);
  447. if (h!=NULL)
  448.             {
  449. p2=(NodeData *)(pListView->m_map.GetItemData(h));
  450. p1->position=p2->position+1;
  451. }
  452.      
  453. //根据前面的兄弟节点的位置信息，获取新增加节点的位置信息；
  454. pListView->m_map.SetItemData(HCurrent,(DWORD)p1);
  455. //设置data属性；
  456. p1=(NodeData *)(pListView->m_map.GetItemData(pListView->m_map.GetRootItem()));
  457.             p2=(NodeData *)pListView->m_map.GetItemData(HCurrent);
  458. if (p1->pImage!=NULL)
  459. {
  460.              CRect cr(80+100*p2->position,50+80*p2->level,65,35);
  461.              p2->pImage=new CMyImage();
  462.              p2->pImage->Create(NULL,WS_CHILD|SS_BITMAP|WS_VISIBLE|SS_NOTIFY,
  463.                    cr,pNetMapView,IDC_MYIMAGE+NodeArray.GetSize()+1);
  464.               HBITMAP hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP2));
  465.              if (p2->router==true)
  466.              hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP1));
  467.              p2->pImage->SetBitmap(hBitmap);
  468.              p2->pImage->MyPos.x=cr.left+33;
  469.              p2->pImage->MyPos.y=cr.top+12;
  470.             //保存生成连接线时的点坐标；
  471.              p2->pImage->HSelf=HCurrent;
  472. }
  474. pNetMapView->RedrawWindow();
  475.             pListView->RedrawWindow();
  476. }
  477. else
  478. {
  479. h=FindNode(pListView->m_map.GetRootItem(),Lip);
  480. if (h==NULL)
  481. {
  482. MessageBox("要合并的节点不存在!");
  483. return;
  484. }
  485. HCurrent=FindNode(pListView->m_map.GetRootItem(),Rip);
  486. if (h==NULL)
  487. {
  488. MessageBox("上一级节点不存在!");
  489. return;
  490. }
  491. if ((pListView->m_map.GetChildItem(h)) ||(pListView->m_map.GetChildItem(HCurrent)))
  492. {
  493.              MessageBox("只有叶子节点才能被合并!");
  494. return;
  496. }
  497.             p1=(NodeData *)(pListView->m_map.GetItemData(h));
  498. p2=(NodeData *)(pListView->m_map.GetItemData(HCurrent));
  499.     if ((p1->router!=true) ||(p2->router!=true))
  500.             {
  501. MessageBox("要合并的设备必须是路由设备!");
  502. return;
  503. }
  504. p2->m_neighbor.Append(p1->m_neighbor);
  505. //合并连结信息；
  506. for(int i=0;i<NodeArray.GetSize();i++)
  507. {
  508. if (NodeArray[i]==h)
  509. {
  510. NodeArray.RemoveAt(i);
  511. break;
  512. }
  513. }
  514. if (p1->pImage!=NULL) delete p1->pImage;
  515. delete p1;
  516. pListView->m_map.DeleteItem(h);
  517. pNetMapView->RedrawWindow();
  518.             pListView->RedrawWindow();
  519. }
  520. }
  521. }