snmpWalk.cpp
上传用户:czjinwang
上传日期:2007-01-12
资源大小:2484k
文件大小:6k
源码类别:

SNMP编程

开发平台:

Visual C++

  1. /*
  2.   snmpWalk.cpp 
  3.   version 2.8
  4.   
  5.   Copyright (c) 1996
  6.   Hewlett-Packard Company
  7.   ATTENTION: USE OF THIS SOFTWARE IS SUBJECT TO THE FOLLOWING TERMS.
  8.   Permission to use, copy, modify, distribute and/or sell this software
  9.   and/or its documentation is hereby granted without fee. User agrees
  10.   to display the above copyright notice and this license notice in all
  11.   copies of the software and any documentation of the software. User
  12.   agrees to assume all liability for the use of the software; Hewlett-Packard
  13.   makes no representations about the suitability of this software for any
  14.   purpose. It is provided "AS-IS" without warranty of any kind,either express
  15.   or implied. User hereby grants a royalty-free license to any and all
  16.   derivatives based upon this software code base.
  17.   Peter E. Mellquist
  18. */
  19. #include "snmp_pp.h"
  20. #include <iostream.h>
  21. #include <stdlib.h>
  22. #define BULK_MAX 20
  23. int main( int argc, char **argv)  {
  24.    int requests=0;        // keep track of # of requests
  25.    int objects=0;
  26.    //---------[ check the arg count ]----------------------------------------
  27.    if ( argc < 2) {
  28.   cout << "Usage:n";
  29.   cout << "snmpWalk Address | DNSName [StartOid] [options]n";
  30.   cout << "StartOid: sysDescr object is defaultn";
  31.   cout << "options: -v1 , use SNMPV1, defaultn";
  32.   cout << "         -v2 , use SNMPV2n";
  33.   cout << "         -cCommunity_name, specify community default is 'public' n";
  34.   cout << "         -rN , retries default is N = 1 retryn";
  35.   cout << "         -tN , timeout in hundredths-seconds default is N = 100 = 1 secondn";
  36.   return 0;
  37.    }
  38.    //---------[ make a GenAddress and Oid object to retrieve ]---------------
  39.    GenAddress address( argv[1]);      // make a SNMP++ Generic address
  40.    if ( !address.valid()) {           // check validity of address
  41.   cout << "Invalid Address or DNS Name, " << argv[1] << "n";
  42.   return 0;
  43.    }
  44.    Oid oid("1");                      // defualt is beginning of MIB 
  45.    if ( argc >= 3) {                  // if 3 args, then use the callers Oid
  46.   if ( strstr( argv[2],"-")==0) {
  47.      oid = argv[2];
  48.      if ( !oid.valid()) {            // check validity of user oid
  49.     cout << "Invalid Oid, " << argv[2] << "n";
  50.     return 0;
  51.          }
  52.       }
  53.    }
  54.    //---------[ determine options to use ]-----------------------------------
  55.    snmp_version version=version1;                       // default is v1
  56.    int retries=1;                                       // default retries is 1
  57.    int timeout=100;                                     // default is 1 second
  58.    OctetStr community("public");                        // read community
  59.    char *ptr;
  60.    for(int x=1;x<argc;x++) {                           // parse for version
  61.       if ( strstr( argv[x],"-v2")!= 0)   
  62.          version = version2c;
  63.       if ( strstr( argv[x],"-r")!= 0) {                 // parse for retries
  64.          ptr = argv[x]; ptr++; ptr++;
  65.  retries = atoi(ptr);
  66.  if (( retries<1)|| (retries>5)) retries=1; 
  67.       }
  68.   if ( strstr( argv[x], "-t")!=0) {                 // parse for timeout
  69.  ptr = argv[x]; ptr++; ptr++; 
  70.  timeout = atoi( ptr);
  71.  if (( timeout < 100)||( timeout>500)) timeout=100;
  72.       }
  73.   if ( strstr( argv[x],"-c")!=0) {
  74.           ptr = argv[x]; ptr++; ptr++;
  75.   community = ptr;
  76.   }
  77.    }
  78.    //----------[ create a SNMP++ session ]-----------------------------------
  79.    int status; 
  80.    Snmp snmp( status);                // check construction status
  81.    if ( status != SNMP_CLASS_SUCCESS) {
  82.       cout << "SNMP++ Session Create Fail, " << snmp.error_msg(status) << "n";
  83.       return 0;
  84.    }
  85.    //--------[ build up SNMP++ object needed ]-------------------------------
  86.    Pdu pdu;                              // construct a Pdu object
  87.    Vb vb;                                // construct a Vb object
  88.    vb.set_oid( oid);                     // set the Oid portion of the Vb
  89.    pdu += vb;                            // add the vb to the Pdu
  90.    CTarget target( address);             // make a target using the address
  91.    target.set_version( version);         // set the SNMP version SNMPV1 or V2
  92.    target.set_retry( retries);           // set the number of auto retries
  93.    target.set_timeout( timeout);         // set timeout
  94.    target.set_readcommunity( community); // set the read community name
  95.  
  96.    //-------[ issue the request, blocked mode ]-----------------------------
  97.    cout << "SNMP++ snmpWalk to " << argv[1] << " SNMPV" << (version+1) << " Retries=" << retries;
  98.    cout << " Timeout=" << timeout <<"ms " << "Community=" << community.get_printable() << "n";
  99.    while (( status = snmp.get_bulk( pdu,target,0,BULK_MAX))== SNMP_CLASS_SUCCESS) {
  100.   requests++;
  101.   for ( int z=0;z<pdu.get_vb_count(); z++) {
  102.      pdu.get_vb( vb,z);
  103.  objects++;
  104.  // look for var bind exception, applies to v2 only   
  105.  if ( vb.get_syntax() != sNMP_SYNTAX_ENDOFMIBVIEW) {
  106. cout << "Syntax = " << vb.get_syntax() << "n";
  107.         cout << "Oid = " << vb.get_printable_oid() << "n";
  108.         cout << "Value = " << vb.get_printable_value() << "nn";
  109.          }
  110.  else {
  111. cout << "End of MIB Reachedn";
  112.             cout << "Total # of Requests = " << requests << "n";
  113. cout << "Total # of Objects  = " << objects  << "n";
  114. return 0;
  115.          }
  116.       }
  117.       // last vb becomes seed of next rquest
  118.   Pdu newpdu;
  119.   newpdu += vb;
  120.   pdu = newpdu;
  121.    }
  122.    if ( status != SNMP_ERROR_NO_SUCH_NAME)
  123.       cout << "SNMP++ snmpWalk Error, ";
  124.    if ( status == SNMP_CLASS_ERR_STATUS_SET) 
  125.   status = pdu.get_error_status();
  126.    cout << snmp.error_msg( status) << "n";
  127.    cout << "Total # of Requests = " << requests << "n";
  128.    cout << "Total # of Objects  = " << objects  << "n";
  129.    return 0;
  130. }  // end Walk