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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
Get.cpp
资源名称：hp_snmp3.zip　[点击查看]
上传用户：czjinwang
上传日期：2007-01-12
资源大小：2484k
文件大小：15k
源码类别：

SNMP编程

开发平台：

Visual C++

Get.cpp：源码内容
  1. /*============================================================================
  2.   Copyright (c) 1996
  3.   Hewlett-Packard Company
  5.   ATTENTION: USE OF THIS SOFTWARE IS SUBJECT TO THE FOLLOWING TERMS.
  6.   Permission to use, copy, modify, distribute and/or sell this software 
  7.   and/or its documentation is hereby granted without fee. User agrees 
  8.   to display the above copyright notice and this license notice in all 
  9.   copies of the software and any documentation of the software. User 
  10.   agrees to assume all liability for the use of the software; Hewlett-Packard 
  11.   makes no representations about the suitability of this software for any 
  12.   purpose. It is provided "AS-IS" without warranty of any kind,either express 
  13.   or implied. User hereby grants a royalty-free license to any and all 
  14.   derivatives based upon this software code base. 
  16. =============================================================================*/
  18. #include "stdafx.h"
  19. #include "browser.h"
  20. #include "Get.h"
  21. #include "db_cls.h"
  22. #include "tarfact.h"
  24. #ifdef _DEBUG
  25. #define new DEBUG_NEW
  26. #undef THIS_FILE
  27. static char THIS_FILE[] = __FILE__;
  28. #endif
  30. /////////////////////////////////////////////////////////////////////////////
  31. // Get dialog
  34. Get::Get(CWnd* pParent /*=NULL*/)
  35. : CDialog(Get::IDD, pParent)
  36. {
  37. //{{AFX_DATA_INIT(Get)
  38. m_mgmt_proto = _T("");
  39. m_net_proto = _T("");
  40. m_read_community = _T("");
  41. m_write_community = _T("");
  42. m_timeouts = _T("");
  43. m_retries = _T("");
  44. m_oid = _T("");
  45. m_output = _T("");
  46. //}}AFX_DATA_INIT
  48. int cstatus = Create(IDD, pParent);
  50. // create a snmp++ object
  51. int status;
  52. snmp = new Snmp( status);
  53. if ( status != SNMP_CLASS_SUCCESS)
  54. {
  55. AfxMessageBox("Unable To Create Snmp Object!");
  56. snmp = NULL;
  57. }
  59.     // load up the target list
  60.     CComboBox * target_cb = ( CComboBox *) GetDlgItem( IDC_TARGETS);
  61.     target_cb->ResetContent();
  63.     // get the db file name from the ini file
  64.     CString filename = theApp.GetProfileString( BROWSER_VALUE,DB_NAME,DEF_DB_NAME);
  66. // access the target database
  67.     Db target_db;
  68. TargetDb_Rec db_rec;
  69.     target_db.set_attributtes( filename, sizeof( TargetDb_Rec));
  70.     int nr = target_db.get_num_recs();
  72. if ( nr==0)
  73. AfxMessageBox("You Have No Targets Definedn Please Define a Target");
  75. // for all records, get and display
  76.     for (int i=1;i<=nr;i++)
  77.     {
  78.        if ((status = target_db.retrieve((long int) (i-1), &db_rec))!= DB_OK)
  79.        {
  80.           MessageBox("Unable to Read Target DB Record", ERR_MSG,MB_ICONSTOP);
  81.           return;
  82.        }
  84.    CString target_name;
  85.    if ( strcmp( db_rec.alias,"") ==0)
  86.          target_cb->AddString((char*)  db_rec.key);
  87.    else
  88.    {
  89.  target_name = db_rec.alias;
  90.  target_name += " @ ";
  91.  target_name += db_rec.key;
  92.  target_cb->AddString(target_name);
  93.    }
  95.        if ( i==1)
  96.        {
  97.           target_cb->SetCurSel(0);
  98.           load_target_attribs( db_rec);
  99.        }
  100. }
  101. }
  104. // load the current target view
  105. void Get::load_target_attribs( TargetDb_Rec db_rec)
  106. {
  107.    char buffer[20];
  108.    m_read_community = db_rec.read_community;
  109.    m_write_community = db_rec.write_community;
  110.    sprintf( buffer,"%d",db_rec.retries);
  111.    m_retries = buffer;
  112.    sprintf( buffer,"%d",db_rec.timeout);
  113.    m_timeouts = buffer;
  114.    m_mgmt_proto = (db_rec.snmp_type == SNMPV1) ? V1 :V2C;
  115.    m_net_proto =  (db_rec.address_type == IP_TYPE) ? "IP" : "IPX";
  116.    UpdateData( FALSE);
  117. };
  120. void Get::DoDataExchange(CDataExchange* pDX)
  121. {
  122. CDialog::DoDataExchange(pDX);
  123. //{{AFX_DATA_MAP(Get)
  124. DDX_Text(pDX, IDC_MGMTPROTO, m_mgmt_proto);
  125. DDX_Text(pDX, IDC_NETPROTO, m_net_proto);
  126. DDX_Text(pDX, IDC_READ_COMMUNITY, m_read_community);
  127. DDX_Text(pDX, IDC_WRITE_COMMUNITY, m_write_community);
  128. DDX_Text(pDX, IDC_TIMEOUTS, m_timeouts);
  129. DDX_Text(pDX, IDC_RETRIES, m_retries);
  130. DDX_Text(pDX, IDC_OID, m_oid);
  131. DDV_MaxChars(pDX, m_oid, 200);
  132. DDX_Text(pDX, IDC_OUTPUT, m_output);
  133. DDV_MaxChars(pDX, m_output, 1000);
  134. //}}AFX_DATA_MAP
  135. }
  138. BEGIN_MESSAGE_MAP(Get, CDialog)
  139. //{{AFX_MSG_MAP(Get)
  140. ON_CBN_SELCHANGE(IDC_TARGETS, OnSelchangeTargets)
  141. ON_WM_DESTROY()
  142. ON_BN_CLICKED(IDC_ADD, OnAdd)
  143. ON_BN_CLICKED(IDC_REMOVE, OnRemove)
  144. ON_NOTIFY(NM_DBLCLK, IDC_MIBTREE, OnDblclkMibtree)
  145. ON_BN_CLICKED(IDC_GET, OnGet)
  146. ON_BN_CLICKED(IDC_ADDOID, OnAddoid)
  147. ON_BN_CLICKED(IDC_STOP, OnStop)
  148. //}}AFX_MSG_MAP
  149. END_MESSAGE_MAP()
  151. /////////////////////////////////////////////////////////////////////////////
  152. // Get message handlers
  155. // load up the Ctree control with the MIB table
  156. void Get::load_mib_view( CTreeCtrl *tree)
  157. {
  158.  CString name;
  159.  Oid current_oid, parent_oid, next_oid,prev_parent_oid,*oid_ptr;
  160.  HTREEITEM new_node, parent_node, previous_parent;
  161.  parent_node = previous_parent = TVI_ROOT;
  162.  for (int x=0;x<MAXMIBVALS; x++)
  163.  {
  164.     name = MIBVals[x][0];   
  165. current_oid = MIBVals[x][2];   
  166. next_oid = ((x+1)==MAXMIBVALS)?"":MIBVals[x+1][2];   
  168. if ( current_oid.len() <= parent_oid.len())
  169. {
  170. parent_node = previous_parent;
  171. parent_oid = prev_parent_oid;
  172. }
  174. // add the node
  175. new_node = tree->InsertItem( name,parent_node);
  176. oid_ptr = new Oid( current_oid);
  177. if ( strcmp(MIBVals[x][1],SCALAR)==0)
  178.    (*oid_ptr)+="0";
  179. tree->SetItemData( new_node, (DWORD) oid_ptr);
  181. if ( name == "sysDescr")
  182.    tree->SelectItem( new_node);
  184. if (( current_oid.len() > parent_oid.len()) && ( current_oid.len() < next_oid.len()))
  185. {
  186.    previous_parent = parent_node;  
  187.    parent_node = new_node;         
  188.    prev_parent_oid = parent_oid;
  189.    parent_oid = current_oid;        
  190. }
  191.  };
  192. };
  195. BOOL Get::OnInitDialog() 
  196. {
  197. CDialog::OnInitDialog();
  199. // TODO: Add extra initialization here
  200. CenterWindow();
  202. // load up the MIB tree control
  203. CTreeCtrl *tree;
  204. tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
  205. load_mib_view( tree);
  207. // clear the pdu table
  208. CListBox *lb;
  209. lb= ( CListBox*) GetDlgItem( IDC_PDU);
  210. lb->ResetContent();
  212. return TRUE;  // return TRUE unless you set the focus to a control
  213.               // EXCEPTION: OCX Property Pages should return FALSE
  214. }
  217. void Get::OnSelchangeTargets() 
  218. {
  219. // TODO: Add your control notification handler code here
  220. // access the target database
  221.     Db target_db;
  222. TargetDb_Rec db_rec;
  223. char key[80];
  224. char *ptr,*address;
  225. int status;
  226.  
  227.     CComboBox * target_cb = ( CComboBox *) GetDlgItem( IDC_TARGETS);
  228. target_cb->GetLBText( target_cb->GetCurSel() , key);
  230. // trim off alias, if present
  231. address = key;
  232. ptr = key;
  233. while (*ptr != 0)
  234. {
  235. if ( *ptr == '@')
  236.    address = ptr+2;
  237.    ptr++;
  238. }
  241.     // get the db file name from the ini file
  242.     CString filename = theApp.GetProfileString( BROWSER_VALUE,DB_NAME,DEF_DB_NAME);
  243. target_db.set_attributtes( filename, sizeof( TargetDb_Rec));
  244.     strcpy( db_rec.key,address);
  245.     status = target_db.read( &db_rec);
  246.     if ( status != DB_OK)
  247.     {
  248.        MessageBox( "Unable to Read Target Record",ERR_MSG,MB_ICONSTOP);
  249.        return;
  250.     }
  251.     load_target_attribs( db_rec);
  253. }
  256. // recursive delete of CtreeCtrl Items
  257. void Get::delete_nodes( CTreeCtrl *tree, HTREEITEM node)
  258. {
  259.     HTREEITEM next_node;
  260. Oid * oid_ptr;
  262.     next_node = tree->GetChildItem( node);
  263. if ( next_node != NULL)
  264. delete_nodes( tree, next_node);
  266. next_node = tree->GetNextSiblingItem( node);
  267. if ( next_node != NULL)
  268. delete_nodes( tree, next_node);
  270. oid_ptr = (Oid *) tree->GetItemData( node);
  271. if ( oid_ptr != NULL)
  272.    delete oid_ptr;
  273. }
  276. void Get::OnDestroy() 
  277. {
  278. CDialog::OnDestroy();
  280. // TODO: Add your message handler code here
  281. // need to free up CtreeCtrl's Oids
  282. CTreeCtrl *tree;
  283. tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
  284. delete_nodes( tree, tree->GetRootItem());
  285. }
  288. void Get::OnAdd() 
  289. {
  290. // TODO: Add your control notification handler code here
  291. CTreeCtrl *tree;
  292. HTREEITEM node;
  293. Oid *oid_ptr;
  295. tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
  296. node = tree->GetSelectedItem();
  297. oid_ptr = (Oid *) tree->GetItemData( node);
  298. if ( (*oid_ptr)[ oid_ptr->len() -1 ] != 0)
  299. {
  300. AfxMessageBox("You Must Select A Scalar MIB Object");
  301. return;
  302. }
  304. Oid addoid( oid_ptr->get_printable());
  305. CListBox *lb;
  306. lb = ( CListBox*) GetDlgItem( IDC_PDU);
  307. CString scalar;
  308. scalar = tree->GetItemText( node);
  309. scalar += " ";
  310. scalar += addoid.get_printable();
  311. lb->InsertString( lb->GetCount(),scalar);
  313. }
  315. void Get::OnRemove() 
  316. {
  317. // TODO: Add your control notification handler code here
  318. CListBox *lb;
  319. int selection;
  320. lb = ( CListBox*) GetDlgItem( IDC_PDU);
  322. if((selection = lb->GetCurSel()) != LB_ERR)
  323. lb->DeleteString( selection);
  324. }
  326. void Get::OnDblclkMibtree(NMHDR* pNMHDR, LRESULT* pResult) 
  327. {
  328. // TODO: Add your control notification handler code here
  329. OnAdd();
  330. *pResult = 0;
  331. }
  333. //====================================================================
  334. // SNMP++ callback function for async events
  335. // this callback is used for all async requests 
  336. void my_getcallback( int reason,
  337.                   Snmp* snmp,
  338.                   Pdu &pdu,
  339.                   SnmpTarget &target,
  340.                   void * callback_data)
  341.                                
  342. {  
  345.    // resolve the callback data to a MFC Get Window Handle
  346.    Get * get_window = (Get *) callback_data;
  347.    CButton * continue_mode = (CButton*) get_window->GetDlgItem( IDC_CONTINUOS);
  349.    // based on the reason handle the reason
  350.    switch ( reason)
  351.    { 
  352.   // an asynchronous response has arrived
  353.       case SNMP_CLASS_ASYNC_RESPONSE:
  354.   {
  355.   get_window->display_response( pdu, target, snmp);
  356.   // look for continuos mode
  357.           if (( continue_mode->GetCheck()) && (pdu.get_error_status()== 0))
  358.   get_window->get_another();
  359.   else
  360.   get_window->enable_controls( TRUE);
  361.   }
  362.   break;
  363.      
  364.   // a timeout has occured
  365.       case SNMP_CLASS_TIMEOUT:
  366.   get_window->display_timeout( target);
  367.   get_window->enable_controls( TRUE);
  368.   break;
  369.      
  370.    } // end switch
  372. };   
  375. // get another one
  376. void Get::get_another()
  377. {
  378. OnGet();
  379. };
  382. // display the response value
  383. void Get::display_response( Pdu &pdu, SnmpTarget &target, Snmp *snmp)
  384. {
  386.    GenAddress address;   // SNMP++ generic Address
  387.    Vb vb;
  388.    int error_status = pdu.get_error_status();
  389.    char buffer[50];
  392.    // show the result status
  393.    m_output = "Status = ";
  394.    m_output += snmp->error_msg( error_status);
  395.    m_output += "rn";
  397.    // show the details if check box is selected
  398.    CButton * all_details = (CButton*) GetDlgItem( IDC_ALL_DETAILS);
  399.    if ( all_details->GetCheck())
  400.    {
  401.       target.get_address( address);
  402.       m_output += "Response from ";
  403.   m_output += address.get_printable();
  404.   m_output += "rn";
  405.       m_output += "Error Status = ";
  406.   m_output += snmp->error_msg( error_status );
  407.   m_output += "rn";
  408.       sprintf( buffer,"Error Index  = %drn", pdu.get_error_index());
  409.       m_output += buffer;
  410.   sprintf( buffer,"Request ID  = %ldrn",pdu.get_request_id());
  411.   m_output += buffer;
  412.       sprintf( buffer,"Variable Binding Count = %drn",pdu.get_vb_count());
  413.       m_output += buffer;
  414.    }
  416.    // show the Vbs
  417.    for ( int x=0;x<pdu.get_vb_count();x++)
  418.    {
  419.       pdu.get_vb( vb,x);
  420.   sprintf( buffer,"Vb #%drn",x+1);
  421.   m_output += buffer;
  422.   m_output += "  Oid   = ";
  423.   m_output += vb.get_printable_oid();
  424.   m_output += "rn";
  425.   m_output += "  Value = ";
  426.   m_output += vb.get_printable_value();
  427.   m_output += "rn";
  428.    }
  430.    // update the output display
  431.    UpdateData( FALSE);
  433. };
  435. // display a timeout message
  436. void Get::display_timeout( SnmpTarget &target)
  437. {
  438.    m_output = "";
  439.    UpdateData( FALSE);
  440.    m_output = "Request Timed Out";
  441.    UpdateData( FALSE);
  442. };
  445. // invoke a SNMP++ Get
  446. void Get::OnGet() 
  447. {
  449. //------------------------------------------------------
  450. // Here is the real SNMP++ code !
  451. // 
  452. // Algorithm:
  453. // - build a Pdu using the Oids from the Pdu list box
  454. // - create a CTarget
  455. // - send an Asynchronous Get Request
  456. // - response will come to specified callback
  457. CListBox *lb;
  458. int count;
  459. char buffer[80],*ptr;
  460. Pdu pdu;          // SNMP++ Pdu object
  463. // get the ListBox id
  464. lb = ( CListBox*) GetDlgItem( IDC_PDU);
  466. // make sure we have something to get
  467. if ( (count=lb->GetCount()) <1)
  468. {
  469. AfxMessageBox("There are No PDU Variables to Get");
  470. return;
  471. }
  473. // build up a SNMP++ Pdu with all the Vbs requested
  474. for ( int z =0;z<count;z++)
  475. {
  476.    // get the text from the listbox
  477.    lb->GetText(z,buffer);
  479.    // get rid of the text part
  480.    ptr = buffer;
  481.    while (*ptr != ' ') ptr++;
  482.    ptr++;
  484.    // make a SNMP++ Oid with the dotted notation
  485.    Oid oid( ptr);      
  487.    // make a SNMP++ Vb with the oid
  488.    Vb vb( oid);
  490.    // add the vb to the Pdu
  491.    pdu += vb;
  492. }
  495. // make a SnmpTarget using the Target_Factory
  496. CComboBox *cb = ( CComboBox *) GetDlgItem( IDC_TARGETS);
  497. char key[80];
  498.     if ( cb->GetCurSel() == CB_ERR)
  499.     {
  500.        AfxMessageBox("No Target Selected!");
  501.        return;
  502.     }
  503. cb->GetLBText( cb->GetCurSel(), key); 
  506. // get a target from the target factory
  507. SnmpTarget *target = target_factory( key);
  508. if ( target == NULL)
  509. {
  510. AfxMessageBox("Unable To find Target");
  511. return;
  512. }
  515. // Invoke a SNMP++ Get for the Pdu requested
  516. // from the target created
  517. int status = snmp->get( pdu,
  518.                             *target,
  519.                             (snmp_callback) &my_getcallback,
  520.                             (void *) this);
  521.                                         
  522.     if ( status != SNMP_CLASS_SUCCESS)
  523. AfxMessageBox( snmp->error_msg( status));
  526. // clear the output display
  527. m_output = "SNMP++ Get In Progress...";
  528. UpdateData( FALSE);
  530. // free up the target when complete
  531. delete target;
  533. // disable all controls while get is pending
  534. enable_controls( FALSE);
  535. };
  537. // Manual Oid Specification
  538. void Get::OnAddoid() 
  539. {
  540. // TODO: Add your control notification handler code here
  541. UpdateData( TRUE);
  542. Oid oid( m_oid);
  543. if ( oid.valid())
  544. {
  545. CListBox *lb;
  546.     lb = ( CListBox*) GetDlgItem( IDC_PDU);
  547.     CString item;
  548.     item = "Custom ";
  549.     item += oid.get_printable();
  550.         lb->InsertString( lb->GetCount(),item);
  551. }
  552. else
  553. AfxMessageBox("Invalid Custom MIB Object Identifier");
  554. }
  557. // enable the controls
  558. void Get::enable_controls( int state)
  559. {
  560.    CWnd *control;
  562.    control = GetDlgItem( IDC_TARGETS);
  563.    control->EnableWindow( state ? TRUE : FALSE);
  565.    control = GetDlgItem( IDC_MIBTREE);
  566.    control->EnableWindow( state ? TRUE : FALSE);
  568.    control = GetDlgItem( IDC_ADD);
  569.    control->EnableWindow( state ? TRUE : FALSE);
  571.    control = GetDlgItem( IDC_REMOVE);
  572.    control->EnableWindow( state ? TRUE : FALSE);
  574.    control = GetDlgItem( IDC_PDU);
  575.    control->EnableWindow( state ? TRUE : FALSE);
  577.    control = GetDlgItem( IDC_OID);
  578.    control->EnableWindow( state ? TRUE : FALSE);
  580.    control = GetDlgItem( IDC_ADDOID);
  581.    control->EnableWindow( state ? TRUE : FALSE);
  583.    control = GetDlgItem( IDC_ALL_DETAILS);
  584.    control->EnableWindow( state ? TRUE : FALSE);
  586.    control = GetDlgItem( IDC_CONTINUOS);
  587.    control->EnableWindow( state ? TRUE : FALSE);
  589.   
  590. };
  594. void Get::OnStop() 
  595. {
  596. // TODO: Add your control notification handler code here
  597. CButton * continue_mode = (CButton*) GetDlgItem( IDC_CONTINUOS);
  598. continue_mode->SetCheck(0);
  599. }
  601. void Get::OnCancel() 
  602. {
  603. // TODO: Add extra cleanup here
  604. DestroyWindow();
  605. }