get.cpp

HP公司的SNMP++的Win32版本源码

/*============================================================================
  Copyright (c) 1996
  Hewlett-Packard Company

  ATTENTION: USE OF THIS SOFTWARE IS SUBJECT TO THE FOLLOWING TERMS.
  Permission to use, copy, modify, distribute and/or sell this software 
  and/or its documentation is hereby granted without fee. User agrees 
  to display the above copyright notice and this license notice in all 
  copies of the software and any documentation of the software. User 
  agrees to assume all liability for the use of the software; Hewlett-Packard 
  makes no representations about the suitability of this software for any 
  purpose. It is provided "AS-IS" without warranty of any kind,either express 
  or implied. User hereby grants a royalty-free license to any and all 
  derivatives based upon this software code base. 

=============================================================================*/

#include "stdafx.h"
#include "browser.h"
#include "Get.h"
#include "db_cls.h"
#include "tarfact.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// Get dialog


Get::Get(CWnd* pParent /*=NULL*/)
	: CDialog(Get::IDD, pParent)
{
	//{{AFX_DATA_INIT(Get)
	m_mgmt_proto = _T("");
	m_net_proto = _T("");
	m_read_community = _T("");
	m_write_community = _T("");
	m_timeouts = _T("");
	m_retries = _T("");
	m_oid = _T("");
	m_output = _T("");
	//}}AFX_DATA_INIT

	int cstatus = Create(IDD, pParent);

	// create a snmp++ object
	int status;
	snmp = new Snmp( status);
	if ( status != SNMP_CLASS_SUCCESS)
	{
		AfxMessageBox("Unable To Create Snmp Object!");
		snmp = NULL;
	}
	
    // load up the target list
    CComboBox * target_cb = ( CComboBox *) GetDlgItem( IDC_TARGETS);
    target_cb->ResetContent();

    // get the db file name from the ini file
    CString filename = theApp.GetProfileString( BROWSER_VALUE,DB_NAME,DEF_DB_NAME);

	// access the target database
    Db target_db;
	TargetDb_Rec db_rec;
    target_db.set_attributtes( filename, sizeof( TargetDb_Rec));
    int nr = target_db.get_num_recs();

	if ( nr==0)
		AfxMessageBox("You Have No Targets Defined\n Please Define a Target");

	// for all records, get and display
    for (int i=1;i<=nr;i++)
    {
       if ((status = target_db.retrieve((long int) (i-1), &db_rec))!= DB_OK)
       {
          MessageBox("Unable to Read Target DB Record", ERR_MSG,MB_ICONSTOP);
          return;
       }

	   CString target_name;
	   if ( strcmp( db_rec.alias,"") ==0)
         target_cb->AddString((char*)  db_rec.key);
	   else
	   {
		 target_name = db_rec.alias;
		 target_name += " @ ";
		 target_name += db_rec.key;
		 target_cb->AddString(target_name);
	   }

       if ( i==1)
       {
          target_cb->SetCurSel(0);
          load_target_attribs( db_rec);
       }
	}
}


// load the current target view
void Get::load_target_attribs( TargetDb_Rec db_rec)
{
   char buffer[20];
   m_read_community = db_rec.read_community;
   m_write_community = db_rec.write_community;
   sprintf( buffer,"%d",db_rec.retries);
   m_retries = buffer;
   sprintf( buffer,"%d",db_rec.timeout);
   m_timeouts = buffer;
   m_mgmt_proto = (db_rec.snmp_type == SNMPV1) ? V1 :V2C;
   m_net_proto =  (db_rec.address_type == IP_TYPE) ? "IP" : "IPX";
   UpdateData( FALSE);
};


void Get::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(Get)
	DDX_Text(pDX, IDC_MGMTPROTO, m_mgmt_proto);
	DDX_Text(pDX, IDC_NETPROTO, m_net_proto);
	DDX_Text(pDX, IDC_READ_COMMUNITY, m_read_community);
	DDX_Text(pDX, IDC_WRITE_COMMUNITY, m_write_community);
	DDX_Text(pDX, IDC_TIMEOUTS, m_timeouts);
	DDX_Text(pDX, IDC_RETRIES, m_retries);
	DDX_Text(pDX, IDC_OID, m_oid);
	DDV_MaxChars(pDX, m_oid, 200);
	DDX_Text(pDX, IDC_OUTPUT, m_output);
	DDV_MaxChars(pDX, m_output, 1000);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(Get, CDialog)
	//{{AFX_MSG_MAP(Get)
	ON_CBN_SELCHANGE(IDC_TARGETS, OnSelchangeTargets)
	ON_WM_DESTROY()
	ON_BN_CLICKED(IDC_ADD, OnAdd)
	ON_BN_CLICKED(IDC_REMOVE, OnRemove)
	ON_NOTIFY(NM_DBLCLK, IDC_MIBTREE, OnDblclkMibtree)
	ON_BN_CLICKED(IDC_GET, OnGet)
	ON_BN_CLICKED(IDC_ADDOID, OnAddoid)
	ON_BN_CLICKED(IDC_STOP, OnStop)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// Get message handlers


// load up the Ctree control with the MIB table
void Get::load_mib_view( CTreeCtrl *tree)
{
	 CString name;
	 Oid current_oid, parent_oid, next_oid,prev_parent_oid,*oid_ptr;
	 HTREEITEM new_node, parent_node, previous_parent;
	 parent_node = previous_parent = TVI_ROOT;
	 for (int x=0;x<MAXMIBVALS; x++)
	 {
	    name = MIBVals[x][0];								  
		current_oid = MIBVals[x][2];						  
		next_oid = ((x+1)==MAXMIBVALS)?"":MIBVals[x+1][2];	  

		if ( current_oid.len() <= parent_oid.len())
		{
			parent_node = previous_parent;
			parent_oid = prev_parent_oid;
		}

		// add the node
		new_node = tree->InsertItem( name,parent_node);
		oid_ptr = new Oid( current_oid);
		if ( strcmp(MIBVals[x][1],SCALAR)==0)
		   (*oid_ptr)+="0";
		tree->SetItemData( new_node, (DWORD) oid_ptr);

		if ( name == "sysDescr")
		   tree->SelectItem( new_node);

		if (( current_oid.len() > parent_oid.len()) && ( current_oid.len() < next_oid.len()))
		{
		   previous_parent = parent_node;  
		   parent_node = new_node;         
		   prev_parent_oid = parent_oid;
		   parent_oid = current_oid;        
		}
	 };
};


BOOL Get::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	CenterWindow();

	// load up the MIB tree control
	CTreeCtrl *tree;
	tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
	load_mib_view( tree);

	// clear the pdu table
	CListBox *lb;
	lb= ( CListBox*) GetDlgItem( IDC_PDU);
	lb->ResetContent();	

	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}


void Get::OnSelchangeTargets() 
{
	// TODO: Add your control notification handler code here
	// access the target database
    Db target_db;
	TargetDb_Rec db_rec;
	char key[80];
	char *ptr,*address;
	int status;
 
    CComboBox * target_cb = ( CComboBox *) GetDlgItem( IDC_TARGETS);
	target_cb->GetLBText( target_cb->GetCurSel() , key);

	// trim off alias, if present
	address = key;
	ptr = key;
	while (*ptr != 0)
	{
		if ( *ptr == '@')
		   address = ptr+2;
	   ptr++;
	}


    // get the db file name from the ini file
    CString filename = theApp.GetProfileString( BROWSER_VALUE,DB_NAME,DEF_DB_NAME);
	target_db.set_attributtes( filename, sizeof( TargetDb_Rec));
    strcpy( db_rec.key,address);
    status = target_db.read( &db_rec);
    if ( status != DB_OK)
    {
       MessageBox( "Unable to Read Target Record",ERR_MSG,MB_ICONSTOP);
       return;
    }
    load_target_attribs( db_rec);

}


// recursive delete of CtreeCtrl Items
void Get::delete_nodes( CTreeCtrl *tree, HTREEITEM node)
{
    HTREEITEM next_node;
	Oid * oid_ptr;

    next_node = tree->GetChildItem( node);
	if ( next_node != NULL)
		delete_nodes( tree, next_node);
	
	next_node = tree->GetNextSiblingItem( node);
	if ( next_node != NULL)
		delete_nodes( tree, next_node);
	
	oid_ptr = (Oid *) tree->GetItemData( node);
	if ( oid_ptr != NULL)
	   delete oid_ptr;
}


void Get::OnDestroy() 
{
	CDialog::OnDestroy();
	
	// TODO: Add your message handler code here
	// need to free up CtreeCtrl's Oids
	CTreeCtrl *tree;
	tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
	delete_nodes( tree, tree->GetRootItem());
}


void Get::OnAdd() 
{
	// TODO: Add your control notification handler code here
	CTreeCtrl *tree;
	HTREEITEM node;
	Oid *oid_ptr;

	tree = (CTreeCtrl *) GetDlgItem( IDC_MIBTREE);
	node = tree->GetSelectedItem();
	oid_ptr = (Oid *) tree->GetItemData( node);
	if ( (*oid_ptr)[ oid_ptr->len() -1 ] != 0)
	{
		AfxMessageBox("You Must Select A Scalar MIB Object");
		return;
	}