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

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

SNMP编程

开发平台：

Visual C++

TrapView.cpp：源码内容
  1. // TrapView.cpp : implementation of the CTrapView class
  2. //
  4. #include "stdafx.h"
  5. #include "Trap.h"
  6. #include "TrapDoc.h"
  7. #include "TrapView.h"
  8. #include "MainFrm.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CTrapView
  18. IMPLEMENT_DYNCREATE(CTrapView, CFormView)
  20. BEGIN_MESSAGE_MAP(CTrapView, CFormView)
  21. //{{AFX_MSG_MAP(CTrapView)
  22. ON_COMMAND(ID_START, OnStart)
  23. ON_COMMAND(ID_END, OnEnd)
  24. ON_COMMAND(ID_FINISH, OnFinish)
  25. ON_COMMAND(ID_CONFIG, OnConfig)
  26. ON_WM_DESTROY()
  27. //}}AFX_MSG_MAP
  28. // Standard printing commands
  29. // ON_COMMAND(ID_FILE_PRINT, CFormView::OnFilePrint)
  30. // ON_COMMAND(ID_FILE_PRINT_DIRECT, CFormView::OnFilePrint)
  31. // ON_COMMAND(ID_FILE_PRINT_PREVIEW, CFormView::OnFilePrintPreview)
  32. END_MESSAGE_MAP()
  34. /////////////////////////////////////////////////////////////////////////////
  35. // CTrapView construction/destruction
  37. CTrapView::CTrapView()
  38. : CFormView(CTrapView::IDD)
  39. {
  40. //{{AFX_DATA_INIT(CTrapView)
  41. //}}AFX_DATA_INIT
  42. // TODO: add construction code here
  43.  
  44. }
  46. CTrapView::~CTrapView()
  47. {
  48. }
  50. void CTrapView::DoDataExchange(CDataExchange* pDX)
  51. {
  52. CFormView::DoDataExchange(pDX);
  53. //{{AFX_DATA_MAP(CTrapView)
  54. DDX_Control(pDX, IDC_LIST1, m_list1);
  55. DDX_Control(pDX, IDC_EDIT1, m_text);
  56. DDX_Control(pDX, IDC_TREE1, m_tree);
  57. //}}AFX_DATA_MAP
  58. }
  60. BOOL CTrapView::PreCreateWindow(CREATESTRUCT& cs)
  61. {
  62. // TODO: Modify the Window class or styles here by modifying
  63. //  the CREATESTRUCT cs
  64.     return CFormView::PreCreateWindow(cs);
  65. }
  67. void CTrapView::OnInitialUpdate()
  68. {
  69. CFormView::OnInitialUpdate();
  70. GetParentFrame()->RecalcLayout();
  71. ResizeParentToFit();
  72.  //  m_hIcon=AfxGetApp()->LoadIcon(IDI_ICON1);;
  73. //pMain->pTrapView=this;
  74. TrapMessage="";
  75. m_list1.InsertColumn(0,"  类型   ",LVCFMT_LEFT,80,0);
  76. m_list1.InsertColumn(1,"  名称（DLCI）",LVCFMT_LEFT,130,1);
  77.     m_list1.InsertColumn(2,"  状态  ",LVCFMT_LEFT,70,2);
  78. m_list1.InsertColumn(3,"  辅助信息  ",LVCFMT_LEFT,300,3);
  79. m_list1.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);
  80.     m_tree.InsertItem("被监视设备",NULL); 
  81. m_tree.SetIndent(32);
  82. m_tree.SetItemHeight(26);
  83. // m_tree.InsertItem("aaa",m_tree.GetRootItem());
  84.     m_iImageList.Create(16, 16, TRUE,1, 0);
  85. HICON hIcon;
  86. hIcon= NULL;
  87. //0,root;
  88. hIcon = (HICON)::LoadImage(::AfxGetInstanceHandle(), 
  89. MAKEINTRESOURCE(IDI_ROUTER), IMAGE_ICON, 32, 32, 0);
  90. m_iImageList.Add(hIcon);
  91. //1,closed box;
  92. hIcon = (HICON)::LoadImage(::AfxGetInstanceHandle(), 
  93. MAKEINTRESOURCE(IDI_LINE), IMAGE_ICON, 16, 16, 0);
  94. m_iImageList.Add(hIcon);
  95. hIcon = (HICON)::LoadImage(::AfxGetInstanceHandle(), 
  96. MAKEINTRESOURCE(IDI_DOWN), IMAGE_ICON, 16, 16, 0);
  97. m_iImageList.Add(hIcon);
  98.     m_tree.SetImageList (&m_iImageList,TVSIL_NORMAL);
  99. dlg.Create(IDD_DIALOG1);
  100.     Recive=true;
  101. }
  103. /////////////////////////////////////////////////////////////////////////////
  104. // CTrapView printing
  106. BOOL CTrapView::OnPreparePrinting(CPrintInfo* pInfo)
  107. {
  108. // default preparation
  109. return DoPreparePrinting(pInfo);
  110. }
  112. void CTrapView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  113. {
  114. // TODO: add extra initialization before printing
  115. }
  117. void CTrapView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  118. {
  119. // TODO: add cleanup after printing
  120. }
  122. void CTrapView::OnPrint(CDC* pDC, CPrintInfo* /*pInfo*/)
  123. {
  124. // TODO: add customized printing code here
  125. }
  127. /////////////////////////////////////////////////////////////////////////////
  128. // CTrapView diagnostics
  130. #ifdef _DEBUG
  131. void CTrapView::AssertValid() const
  132. {
  134. CFormView::AssertValid();
  135. }
  137. void CTrapView::Dump(CDumpContext& dc) const
  138. {
  139. CFormView::Dump(dc);
  140. }
  142. CTrapDoc* CTrapView::GetDocument() // non-debug version is inline
  143. {
  144. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CTrapDoc)));
  145. return (CTrapDoc*)m_pDocument;
  146. }
  147. #endif //_DEBUG
  149. /////////////////////////////////////////////////////////////////////////////
  150. // CTrapView message handlers
  151. //回调函数，当有TRAP到达时，调用该函数进行进一步处理；
  152. void callme (int reason, Snmp *snmp, Pdu &pdu, SnmpTarget &target, void *cd)
  153. {  
  154.     CString str="";
  155. CTime time=CTime::GetCurrentTime();
  156. CString time_str=time.Format("%Y-%m-%d:%H:%M:%S");
  157. CTrapView *p;
  158. p=(CTrapView *)cd;
  159. Vb nextVb;
  160. GenAddress addr;
  161. target.get_address(addr);
  162. IpAddress from(addr);
  163. CString ipadd=from.get_printable();
  164. if (ipadd=="127.0.0.1")
  165. return;
  166. //结束监听时本机发送的TRAP退出循环，不做任何处理；
  167. Oid id,ent;
  168. pdu.get_notify_id(id);
  169. pdu.get_notify_enterprise(ent);
  170. CString TrapId=id.get_printable();
  171. //   AfxMessageBox(TrapId);
  172. CString EnterPrise=ent.get_printable();
  173. //取得TRAP的ID以及enterprise ；
  174.     //AfxMessageBox(EnterPrise);
  175. CString index,name;
  176. HTREEITEM ht;
  177. pdu.get_vb(nextVb, 0);
  178. index=nextVb.get_printable_value();
  179.     pdu.get_vb(nextVb, 1);
  180.     name=nextVb.get_printable_value();
  181. if (EnterPrise=="1.3.6.1.2.1.10.32")
  182. {
  183. for (int i=0;i<p->LineArray.GetSize();i++)
  184. {
  185. if (p->LineArray[i]->Selected)
  186. {   
  187. if (p->LineArray[i]->Name==name)
  188. {
  189. ht=p->LineArray[i]->Ht;
  190. pdu.get_vb(nextVb,2);
  191. CString ud=nextVb.get_printable_value();
  192. if (ud=="2")
  193. {
  194. p->TrapMessage+=time_str+"设备："+ipadd+"中PVC："+name+"目前状态为正常!";
  195. p->TrapMessage+=p->LineArray[i]->Descr+"rn";
  196. p->m_tree.SetItemImage(ht,1,1);
  197. str=str+"设备："+ipadd+"rn";
  198. str=str+"PVC:"+name+"rn";
  199. str=str+"状态正常!";
  201. }
  202. else
  203. {
  204. p->TrapMessage+=time_str+"设备："+ipadd+"中PVC："+name+"目前状态为不可用!";
  205. p->TrapMessage+=p->LineArray[i]->Descr+"rn";
  206. p->m_tree.SetItemImage(ht,2,2);
  207.                     str=str+"设备："+ipadd+"rn";
  208. str=str+"PVC:"+name+"rn";
  209. str=str+"状态不可用!";
  210. }
  211. p->m_text.SetWindowText(p->TrapMessage); 
  212.     p->dlg.m_txt.SetWindowText(str);
  213.                     p->dlg.ShowWindow(SW_SHOW);
  214. break;
  215. }
  216. }
  217. }
  218. }
  219. else
  220. {
  221.   for (int i=0;i<p->LineArray.GetSize();i++)
  222. {
  223.  if ((p->LineArray[i]->Selected)&&(p->LineArray[i]->Type=="普通接口"))
  224. {   
  225. if (p->LineArray[i]->Index==index)
  226. {
  227. ht=p->LineArray[i]->Ht;
  228. if (TrapId=="1.3.6.1.6.3.1.1.5.4")
  229. {
  230. p->TrapMessage+=time_str+"设备："+ipadd+"中接口："+name+"目前状态为UP!";
  231. p->TrapMessage+=p->LineArray[i]->Descr+"rn";
  232. p->m_tree.SetItemImage(ht,1,1);
  233.                      str=str+"设备："+ipadd+"rn";
  234. str=str+"接口:"+name+"rn";
  235. str=str+"状态正常!";
  236. }
  237. if (TrapId=="1.3.6.1.6.3.1.1.5.3")
  238. {
  239. p->TrapMessage+=time_str+"设备："+ipadd+"中接口："+name+"目前状态为Down!";
  240. p->TrapMessage+=p->LineArray[i]->Descr+"rn";
  241. p->m_tree.SetItemImage(ht,2,2);
  242.                      str=str+"设备："+ipadd+"rn";
  243. str=str+"接口:"+name+"rn";
  244. str=str+"状态不可用!";
  245. }
  246. p->m_text.SetWindowText(p->TrapMessage); 
  247.                     p->dlg.m_txt.SetWindowText(str);
  248. p->dlg.ShowWindow(SW_SHOW);
  249. break;
  250. }
  251. }
  252. }
  253. }
  254. }
  255. //监听TRAP线程的处理函数，接受一个指针作为参数；
  256. UINT WorkerThreadProc(LPVOID Param)
  257. { 
  258.   Oid linkUp("1.3.6.1.6.3.1.1.5.4");
  259.   Oid linkDown("1.3.6.1.6.3.1.1.5.3");
  260.   CTrapView *pCV;
  261.   pCV=(CTrapView *)Param;
  262.   //将指针类型转换为CTrapView *类型指针；
  263.   int trap_port;
  264.   trap_port =162; 
  265.   //指定监听端口号；
  266.   int status; 
  267.   Snmp::socket_startup();  
  268.   Snmp snmp( status, 0);   
  269.   // 创建Snmp对象；
  270.   if ( status != SNMP_CLASS_SUCCESS)
  271.   {
  272.    AfxMessageBox("出现错误!");
  273.    return TRUE;
  274.   }
  275.    OidCollection oidc;
  276.    TargetCollection targetc;
  277.    //创建过滤TRAP的对象；
  278.    CNotifyEventQueue::set_listen_port(trap_port);
  279.    //设置端口号；
  280.    status=snmp.notify_register(oidc, targetc,callme,Param);
  281.    //注册过滤条件，回调函数，并将Param，也就是指向CTrapView
  282.    //对象的this指针，传递给回调函数，以方便在回调函数中访问
  283.    //处理CTrapView对象中的成员变量；
  284.    if (status != SNMP_CLASS_SUCCESS)
  285.    {
  286.     AfxMessageBox("出现错误!");
  287.     return TRUE;
  288.    }
  289.  while (pCV->Recive)
  290.  snmp.eventListHolder->SNMPProcessEvents();
  291.  //进入SNMPProcessEvents()函数，等待TRAP触发回调用函数；
  292.  Snmp::socket_cleanup();  
  293. AfxMessageBox("停止监视线路!");
  294. return TRUE;
  295. }
  297. void CTrapView::OnStart() 
  298. { 
  299. if (m_tree.GetCount()<3)
  300. {
  301. MessageBox("先选择线路!");
  302. return;
  303. }
  304.  m_list1.DeleteAllItems();
  305. if (Recive==false)
  306.   Recive=true;
  307.  AfxBeginThread(WorkerThreadProc,this,THREAD_PRIORITY_NORMAL,0,0,NULL);
  308.  //创建一个新线程，用来在指定端口监听TRAP；将this指针作为参数传递给新线程；
  309. CMainFrame *pF=(CMainFrame *)AfxGetMainWnd();
  310. pF->m_wndStatusBar.SetPaneText(0,"开始监视......");
  311. // MessageBox("开始监听......");
  312. }
  314. void CTrapView::OnEnd() 
  315. {
  316. if (Recive==true)
  317.   Recive=false;
  318.   Snmp::socket_startup();  
  319.   UdpAddress address("127.0.0.1:162");      
  320.     int status;
  321.    Snmp snmp(status, "0.0.0.0") ;
  322.    Pdu pdu;                               
  323.    pdu.set_notify_id("1.3.6.1");               
  324.    pdu.set_notify_enterprise( "test");     
  325.    CTarget ctarget(address);             
  326.    ctarget.set_version( version1);        
  327.    ctarget.set_readcommunity("public"); 
  328.    SnmpTarget *target=&ctarget;
  329.    status = snmp.trap(pdu,*target);
  330.    Snmp::socket_cleanup();  
  331.    //发送一个TRAP，结束监听状态；
  332. }
  334. void CTrapView::OnFinish() 
  335. {
  336.  HTREEITEM HRoot,HCurrent,Htemp;
  337.   POSITION pos;
  338.   pos=m_list1.GetFirstSelectedItemPosition();
  339.   if (pos==NULL)
  340.   {
  341.   MessageBox("没有选择要监视的线路!");
  342.   return;
  343.   }
  344.  HRoot=m_tree.GetRootItem();
  346.  HCurrent=m_tree.InsertItem(TempArray[0]->Ip,0,0,HRoot);
  348.   int nItem;
  349.   while (pos)
  350.    {
  351.       nItem =m_list1.GetNextSelectedItem(pos);
  352.   if ((nItem)<=TempArray.GetSize())
  353.   {
  354.    TempArray[nItem]->Selected=true;
  355.     
  356.        Htemp=m_tree.InsertItem(TempArray[nItem]->Name,1,1,HCurrent); 
  357.    if (TempArray[nItem]->State=="Down")
  358.    m_tree.SetItemImage(Htemp,2,2);
  359.    TempArray[nItem]->Ht=Htemp;
  360.   }
  361.   }
  362.   m_tree.Expand(HRoot,TVE_EXPAND);
  363.   for (int i=0;i<TempArray.GetSize();i++)
  364.   {
  365.    if (TempArray[i]->Selected==true)
  366.    LineArray.Add(TempArray[i]);
  367.     else
  368.        delete TempArray[i];
  369.   }
  371. }
  373. void CTrapView::OnConfig() 
  374. {
  375. //number=LineArray.GetSize();
  376. TempArray.RemoveAll();
  377. CString ip,BaseOid,community;
  378. CScan dlg;
  379. if (dlg.DoModal()==IDOK)
  380. {
  381. ip=dlg.ip1;
  382. community=dlg.m_community;
  383. }
  384. m_snmp.clear();
  385. BaseOid="1.3.6.1.2.1.2.2.1.1";
  386. m_snmp.GetTree(ip,community,BaseOid);
  387. if (m_snmp.GetCount()==0)
  388. return;
  389. BaseOid="1.3.6.1.2.1.2.2.1.2";
  390. m_snmp.GetTree(ip,community,BaseOid);
  391. BaseOid="1.3.6.1.2.1.2.2.1.8";
  392. m_snmp.GetTree(ip,community,BaseOid);
  393. int row=m_snmp.GetCount()/3;
  394. for (int i=0;i<row;i++)
  395. {
  396. pLine=new Line;
  397.         pLine->Type="普通接口";
  398. pLine->Selected=false;
  399. pLine->Index=m_snmp.GetValue(i);
  400. pLine->Name=m_snmp.GetValue(i+row);
  401. if (m_snmp.GetValue(i+2*row)=="1")
  402. pLine->State="UP";
  403. else  pLine->State="Down";
  404. pLine->Ht=NULL;
  405. pLine->Descr="";
  406. pLine->Ip=ip;
  407. TempArray.Add(pLine);
  408. }
  409. m_snmp.clear();
  410. BaseOid="1.3.6.1.2.1.4.20.1.2";
  411. m_snmp.GetTree(ip,community,BaseOid);
  412. CString temp;
  413. for (i=0;i<m_snmp.GetCount();i++)
  414. {
  415.     for (int j=0;j<TempArray.GetSize();j++)
  416. if (m_snmp.GetValue(i)==TempArray[j]->Index)
  417. {
  418. temp=m_snmp.GetOid(i);
  419. temp=temp.Right(temp.GetLength()-21);
  420. TempArray[j]->Descr="IP:"+temp;
  421. break;
  422. }
  424. }
  425. //for (int j=number;j<LineArray.GetSize();j++)
  426.     for (int j=0;j<TempArray.GetSize();j++)
  427. {
  428. BaseOid="1.3.6.1.2.1.31.1.1.1.18."+TempArray[j]->Index;
  429. m_snmp.GetOne(ip,community,BaseOid,temp);
  430. TempArray[j]->Descr+=" 备注："+temp;
  431. }
  432. //获取网络接口的信息并保存；
  433. m_snmp.clear();
  434. BaseOid="1.3.6.1.2.1.10.32.2.1.1";
  435. m_snmp.GetTree(ip,community,BaseOid);
  436. BaseOid="1.3.6.1.2.1.10.32.2.1.2";
  437. m_snmp.GetTree(ip,community,BaseOid);
  438. BaseOid="1.3.6.1.2.1.10.32.2.1.3";
  439. m_snmp.GetTree(ip,community,BaseOid);
  440. row=m_snmp.GetCount()/3;
  441. for (i=0;i<row;i++)
  442. {
  443. pLine=new Line;
  444. pLine->Selected=false;
  445. pLine->Type="帧中继PVC";
  446. pLine->Ip=ip;
  447. pLine->Ht=NULL;
  448. pLine->Index=m_snmp.GetValue(i);
  449. pLine->Name=m_snmp.GetValue(i+row);
  450. if (m_snmp.GetValue(i+2*row)=="2")
  451. pLine->State="UP";
  452. else  pLine->State="Down";
  453. BaseOid="1.3.6.1.4.1.9.9.49.1.2.2.1.2."+pLine->Index+"."+pLine->Name;
  454. m_snmp.GetOne(ip,community,BaseOid,temp);
  455. if ("2"==temp)
  456. {
  457. CString rv;
  458. BaseOid="1.3.6.1.4.1.9.9.49.1.2.2.1.3."+pLine->Index+"."+pLine->Name;
  459. m_snmp.GetOne(ip,community,BaseOid,rv);
  460. BaseOid="1.3.6.1.2.1.31.1.1.1.18."+rv;
  461. m_snmp.GetOne(ip,community,BaseOid,rv);
  462. pLine->Descr=rv;
  463. }
  464. else
  465. {
  466. CString rv;
  467. BaseOid="1.3.6.1.4.1.9.9.49.1.3.1.1.3."+pLine->Index+"."+pLine->Name;
  468. m_snmp.GetOne(ip,community,BaseOid,rv);
  469. pLine->Descr=rv;
  470. }
  471. //LineArray.Add(pLine);
  472. TempArray.Add(pLine);
  473.     
  474. }
  475. // if (pTrapView->m_list1.GetItemCount()>0)
  476.     m_list1.DeleteAllItems();
  477. for (i=0;i<TempArray.GetSize();i++)
  478. {
  479.      m_list1.InsertItem(i,TempArray[i]->Type);
  480.  m_list1.SetItemText(i,1,TempArray[i]->Name);
  481.      m_list1.SetItemText(i,2,TempArray[i]->State);
  482.      m_list1.SetItemText(i,3,TempArray[i]->Descr);
  483. }
  484. }
  486. void CTrapView::OnDestroy() 
  487. {
  488. CFormView::OnDestroy();
  489. for (int i=0;i<LineArray.GetSize();i++)
  490. {
  491. delete LineArray[i];
  492. }
  493. }