www.pudn.com > netmapper.rar > MainFrm.cpp
// MainFrm.cpp : implementation of the CMainFrame class
//
#include "stdafx.h"
#include "NetMap.h"
#include "NetMapDoc.h"
#include "MainFrm.h"
#include "snmp_pp.h"
#include "Show.h"
#include "Modify.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CMainFrame
IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)
BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
//{{AFX_MSG_MAP(CMainFrame)
ON_WM_CREATE()
ON_COMMAND(ID_SCAN, OnScan)
ON_COMMAND(ID_GRAPH, OnGraph)
ON_COMMAND(ID_SHOWINFORMATION, OnShowinformation)
ON_COMMAND(ID_NODE_MODIFY, OnNodeModify)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
static UINT indicators[] =
{
ID_SEPARATOR, // status line indicator
ID_INDICATOR_CAPS,
ID_INDICATOR_NUM,
ID_INDICATOR_SCRL,
};
/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction
CMainFrame::CMainFrame()
{
// TODO: add member initialization code here
}
CMainFrame::~CMainFrame()
{
}
int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
return -1;
if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
{
TRACE0("Failed to create toolbar\n");
return -1; // fail to create
}
if (!m_wndStatusBar.Create(this) ||
!m_wndStatusBar.SetIndicators(indicators,
sizeof(indicators)/sizeof(UINT)))
{
TRACE0("Failed to create status bar\n");
return -1; // fail to create
}
// TODO: Delete these three lines if you don't want the toolbar to
// be dockable
m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
EnableDocking(CBRS_ALIGN_ANY);
DockControlBar(&m_wndToolBar);
return 0;
}
BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
if( !CFrameWnd::PreCreateWindow(cs) )
return FALSE;
// TODO: Modify the Window class or styles here by modifying
// the CREATESTRUCT cs
//cs.style=WS_MAXIMIZE;
cs.x=4;
cs.y=4;
cs.cx=800;
cs.cy=750;
return TRUE;
}
BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext)
{
BOOL Result=v.CreateStatic(this,1,2,WS_CHILD | WS_VISIBLE);
ASSERT(Result);
h.CreateStatic(&v,2,1,WS_CHILD | WS_VISIBLE,v.IdFromRowCol(0,1));
v.CreateView(0,0,RUNTIME_CLASS(CMapView),CSize(237,100),pContext);
h.CreateView(0,0,RUNTIME_CLASS(CNetMapView),CSize(400,400),pContext);
h.CreateView(1,0,RUNTIME_CLASS(CVlanView),CSize(400,400),pContext);
return Result;
}
/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics
#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
CFrameWnd::AssertValid();
}
void CMainFrame::Dump(CDumpContext& dc) const
{
CFrameWnd::Dump(dc);
}
#endif //_DEBUG
/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers
//根据text属性查找特定的节点是否存在，如找到，返回符合条件节点的句柄，
//否则返回NULL；
HTREEITEM CMainFrame::FindNode(HTREEITEM hSubRoot,const CString &text)
{
if (hSubRoot==NULL)
return NULL;
//判断是否符合条件，如是，返回该节点的举柄；
if (pListView->m_map.GetItemText(hSubRoot)==text)
return hSubRoot;
hSubRoot=pListView->m_map.GetChildItem(hSubRoot);
do
{
HTREEITEM ht;
//递归调用FindNode；
if ((ht=FindNode(hSubRoot,text)) !=NULL)
return ht;
} while ((hSubRoot=pListView->m_map.GetNextSiblingItem(hSubRoot)) !=NULL);
return NULL;
}
bool CMainFrame::Findport(CString &port,NodeData *p)
{
for (int i=0;i<(p->m_neighbor.GetSize());i++)
{
if (p->m_neighbor.GetAt(i).localport==port)
return true;
}
return false;
}
//将CISCO格式的网络地址转换为正常格式的IP地址；
CString CMainFrame::HtoIP(CString Hex)
{
char ch[4];
CString ip;
if (Hex.GetLength()<11)
{
ip="NULL";
return ip;
}
CString Hex1;
Hex.Replace(" ","");
for (int i=0;i<4;i++)
{
Hex1=Hex.Mid(i*2,2);
int data=0;
if((Hex1[0]>='A') && (Hex1[0]<='F'))
data=data+Hex1[0]-'A'+10;
if((Hex1[0]>='0') && (Hex1[0]<='9'))
data=data+Hex1[0]-'0';
data=data*16;
int data1=0;
if((Hex1[1]>='A') && (Hex1[1]<='F'))
data1=data1+Hex1[1]-'A'+10;
if((Hex1[1]>='0') && (Hex1[1]<='9'))
data1=data1+Hex1[1]-'0';
data=data+data1;
itoa(data,ch,10);
ip=ip+ch+'.';
}
ip=ip.Left(ip.GetLength()-1);
return ip;
}
//扫描获取一个节点中CDP信息的函数，
void CMainFrame::ScanNode(CString &add,CString &community,HTREEITEM hparent)
{
char ch[3];
neighbor n1;
CString BaseOid;
NodeData *p1;
m_snmp.clear();
//清除原来的数据；
for (int i=4;i<10;i++)
{
BaseOid="1.3.6.1.4.1.9.9.23.1.2.1.1.";
itoa(i,ch,10);
BaseOid=BaseOid+ch;
if (m_snmp.GetTree(add,community,BaseOid)==false)
{
MessageBox("获取数据失败，请检查网络和SNMP配置参数!");
return;
}
}
//(pNodeData *)(pListView->m_map.GetItemData(hparent))->scan=true;
//将CDP条目表中第4~9六个列对象的实例值取回，并存放在一个MyStringArray
//对象中。关于这些列对象的具体含义请参考本章前面的内容；
p1=(NodeData *)(pListView->m_map.GetItemData(hparent));
p1->scan=true;
//设置已经扫描标志；
int row=m_snmp.GetCount()/6;
CString h;
int j;
for (j=0; jip);
for (j=0; jm_neighbor.GetSize()>0)
if (Findport(m_snmp.GetValue(j+6*row),p1))
continue;
//和父节点连接的CDP条目，在扫描父节点时这些信息已经保存在本地
//节点中，跳过；
h=FindNode(pListView->m_map.GetRootItem(),m_snmp.GetValue(j));
if (h!=NULL)
{
neighbor n2;
NodeData * p2=(NodeData *)(pListView->m_map.GetItemData(h));
n2.Id=hparent;
n2.localport=m_snmp.GetValue(j+3*row);
n2.remoteport=m_snmp.GetValue(j+6*row);
p2->m_neighbor.Add(n2);
continue;
}
//已经存在的节点，只增加连接信息，不增加新的节点；
pNodeData=new NodeData;
pNodeData->ip=m_snmp.GetValue(j);
pNodeData->sysDescr=m_snmp.GetValue(j+row);
pNodeData->scan=false;
pNodeData->router=false;
pNodeData->level=p1->level+1;
pNodeData->pImage=NULL;
pNodeData->position=p1->position+pos;
pos+=1;
//对任一子节点，层次数等于父节点层次数加1；横向位置以父节点
//为基础，依次累加；
CString rs=m_snmp.GetValue(j+5*row);
rs=rs.Mid(9,2);
if ((atoi(rs)%2)==1)
{
pNodeData->scan=true;
pNodeData->router=true;
}
if (atoi(rs)>=8)
pNodeData->scan=false;
n1.Id=hparent;
n1.localport=m_snmp.GetValue(j+3*row);
n1.remoteport=m_snmp.GetValue(j+6*row);
pNodeData->m_neighbor.Add(n1);
h=pListView->m_map.InsertItem(m_snmp.GetValue(j),hparent);
pListView->m_map.SetItemData(h,(DWORD)pNodeData);
NodeArray.Add(h);
}
}
void CMainFrame::OnScan()
{
if (NodeArray.GetSize()>0)
NodeArray.RemoveAll();
if (pListView->m_map.GetCount()>0)
pListView->m_map.DeleteAllItems();
//清除上次扫描数据；
NodeData *p1;
CNetMapDoc *pDoc=(CNetMapDoc *)this->GetActiveDocument();
if (pDoc->ip.GetLength()==0||pDoc->community.GetLength()==0)
{
MessageBox("设备地址,Community信息错误!");
return;
}
CString sysoid="1.3.6.1.2.1.1.1.0";
CString reply;
m_snmp.GetOne(pDoc->ip,pDoc->community,sysoid,reply);
pNodeData=new NodeData;
pNodeData->ip=pDoc->ip;
pNodeData->sysDescr=reply;
pNodeData->scan=false;
pNodeData->pImage=NULL;
pNodeData->level=0;
pNodeData->position=0;
HRoot=pListView->m_map.InsertItem(pDoc->ip);
NodeArray.Add(HRoot);
pListView->m_map.SetItemData(HRoot,(DWORD)pNodeData);
HCurrent=HRoot;
int NodeNumber;
int i;
while (1)
{
NodeNumber=NodeArray.GetSize();
for(i=0;im_map.GetItemData(NodeArray[i]));
if (p1->scan==true)
continue;
ScanNode(p1->ip,pDoc->community,NodeArray[i]);
}
if (NodeNumber==NodeArray.GetSize())
break;
}
}
void CMainFrame::OnGraph()
{
NodeData *p1;
int NodeNumber=NodeArray.GetSize();
if (NodeNumber==0) return;
//没有图形数据，退出；
p1=(NodeData *)(pListView->m_map.GetItemData(NodeArray[0]));
if (p1->pImage!=NULL) return;
//已经生成图形，退出；
int x=80;
int y=50;
//CRect cr;
CBitmap bmpLoad;
for(int i=0;im_map.GetItemData(NodeArray[i]));
CRect cr(x+100*p1->position,y+80*p1->level,65,35);
p1->pImage=new CMyImage();
p1->pImage->Create(NULL,WS_CHILD|SS_BITMAP|WS_VISIBLE|SS_NOTIFY,
cr,pNetMapView,IDC_MYIMAGE+i);
HBITMAP hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP2));
if (p1->router==true)
hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP1));
p1->pImage->SetBitmap(hBitmap);
p1->pImage->MyPos.x=cr.left+33;
p1->pImage->MyPos.y=cr.top+12;
//保存生成连接线时的点坐标；
p1->pImage->HSelf=NodeArray[i];
//保存在树形控件中的举柄；
}
pNetMapView->paint=true;
pNetMapView->RedrawWindow();
}
void CMainFrame::OnShowinformation()
{
NodeData * p1=(NodeData *)(pListView->m_map.GetItemData(HSelect));
CShow dlg;
dlg.m_descr=p1->sysDescr;
CString local="";
CString remote="";
CString ip="";
for (int i=0;im_neighbor.GetSize();i++)
{
ip+=pListView->m_map.GetItemText(p1->m_neighbor[i].Id)+" ";
local+=p1->m_neighbor[i].localport+" ";
remote+=p1->m_neighbor[i].remoteport+" ";
}
dlg.m_localport=local;
dlg.m_remotelocal=remote;
dlg.m_ip=ip;
dlg.DoModal();
}
void CMainFrame::OnNodeModify()
{
HTREEITEM h;
NodeData *p1;
NodeData *p2;
CString Lip,Lport,Rip,Rport;
CModify dlg;
if(dlg.DoModal()==IDOK)
{
Lip=dlg.m_LocalIp;
Lport=dlg.m_LocalPort;
Rip=dlg.m_RemoteIp;
Rport=dlg.m_RemotePort;
if (dlg.add==true)
{
h=FindNode(pListView->m_map.GetRootItem(),Lip);
if (h!=NULL)
{
MessageBox("要增加的节点已经存在!");
return;
}
h=FindNode(pListView->m_map.GetRootItem(),Rip);
if (h==NULL)
{
MessageBox("上一级节点不存在!");
return;
}
p2=(NodeData *)(pListView->m_map.GetItemData(h));
p1=new NodeData;
p1->ip=Lip;
p1->level=p2->level+1;
p1->position=p2->position;
p1->sysDescr="增加的节点，IP："+Lip;
if (dlg.m_router==1)
p1->router=true;;
p1->scan=true;
neighbor n1;
n1.Id=h;
n1.localport=Lport;
n1.remoteport=Rport;
p1->m_neighbor.Add(n1);
//设置邻居节点信息；
HCurrent=pListView->m_map.InsertItem(Lip,h);
NodeArray.Add(HCurrent);
//在tree中增加节点，父节点为制定节点；
h=pListView->m_map.GetPrevSiblingItem(HCurrent);
if (h!=NULL)
{
p2=(NodeData *)(pListView->m_map.GetItemData(h));
p1->position=p2->position+1;
}
//根据前面的兄弟节点的位置信息，获取新增加节点的位置信息；
pListView->m_map.SetItemData(HCurrent,(DWORD)p1);
//设置data属性；
p1=(NodeData *)(pListView->m_map.GetItemData(pListView->m_map.GetRootItem()));
p2=(NodeData *)pListView->m_map.GetItemData(HCurrent);
if (p1->pImage!=NULL)
{
CRect cr(80+100*p2->position,50+80*p2->level,65,35);
p2->pImage=new CMyImage();
p2->pImage->Create(NULL,WS_CHILD|SS_BITMAP|WS_VISIBLE|SS_NOTIFY,
cr,pNetMapView,IDC_MYIMAGE+NodeArray.GetSize()+1);
HBITMAP hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP2));
if (p2->router==true)
hBitmap = ::LoadBitmap(AfxGetApp()->m_hInstance,MAKEINTRESOURCE(IDB_BITMAP1));
p2->pImage->SetBitmap(hBitmap);
p2->pImage->MyPos.x=cr.left+33;
p2->pImage->MyPos.y=cr.top+12;
//保存生成连接线时的点坐标；
p2->pImage->HSelf=HCurrent;
}
pNetMapView->RedrawWindow();
pListView->RedrawWindow();
}
else
{
h=FindNode(pListView->m_map.GetRootItem(),Lip);
if (h==NULL)
{
MessageBox("要合并的节点不存在!");
return;
}
HCurrent=FindNode(pListView->m_map.GetRootItem(),Rip);
if (h==NULL)
{
MessageBox("上一级节点不存在!");
return;
}
if ((pListView->m_map.GetChildItem(h)) ||(pListView->m_map.GetChildItem(HCurrent)))
{
MessageBox("只有叶子节点才能被合并!");
return;
}
p1=(NodeData *)(pListView->m_map.GetItemData(h));
p2=(NodeData *)(pListView->m_map.GetItemData(HCurrent));
if ((p1->router!=true) ||(p2->router!=true))
{
MessageBox("要合并的设备必须是路由设备!");
return;
}
p2->m_neighbor.Append(p1->m_neighbor);
//合并连结信息；
for(int i=0;ipImage!=NULL) delete p1->pImage;
delete p1;
pListView->m_map.DeleteItem(h);
pNetMapView->RedrawWindow();
pListView->RedrawWindow();
}
}
}