snmpGet.cpp
上传用户:cnryan
上传日期:2008-12-15
资源大小:260k
文件大小:13k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*_############################################################################
  2.   _## 
  3.   _##  snmpGet.cpp  
  4.   _##
  5.   _##  SNMP++v3.2.21
  6.   _##  -----------------------------------------------
  7.   _##  Copyright (c) 2001-2006 Jochen Katz, Frank Fock
  8.   _##
  9.   _##  This software is based on SNMP++2.6 from Hewlett Packard:
  10.   _##  
  11.   _##    Copyright (c) 1996
  12.   _##    Hewlett-Packard Company
  13.   _##  
  14.   _##  ATTENTION: USE OF THIS SOFTWARE IS SUBJECT TO THE FOLLOWING TERMS.
  15.   _##  Permission to use, copy, modify, distribute and/or sell this software 
  16.   _##  and/or its documentation is hereby granted without fee. User agrees 
  17.   _##  to display the above copyright notice and this license notice in all 
  18.   _##  copies of the software and any documentation of the software. User 
  19.   _##  agrees to assume all liability for the use of the software; 
  20.   _##  Hewlett-Packard and Jochen Katz make no representations about the 
  21.   _##  suitability of this software for any purpose. It is provided 
  22.   _##  "AS-IS" without warranty of any kind, either express or implied. User 
  23.   _##  hereby grants a royalty-free license to any and all derivatives based
  24.   _##  upon this software code base. 
  25.   _##  
  26.   _##  Stuttgart, Germany, Fri Jun 16 17:48:57 CEST 2006 
  27.   _##  
  28.   _##########################################################################*/
  29. /*
  30.   snmpGet.cpp 
  31.   Copyright (c) 1996
  32.   Hewlett-Packard Company
  33.   ATTENTION: USE OF THIS SOFTWARE IS SUBJECT TO THE FOLLOWING TERMS.
  34.   Permission to use, copy, modify, distribute and/or sell this software
  35.   and/or its documentation is hereby granted without fee. User agrees
  36.   to display the above copyright notice and this license notice in all
  37.   copies of the software and any documentation of the software. User
  38.   agrees to assume all liability for the use of the software; Hewlett-Packard
  39.   makes no representations about the suitability of this software for any
  40.   purpose. It is provided "AS-IS" without warranty of any kind,either express
  41.   or implied. User hereby grants a royalty-free license to any and all
  42.   derivatives based upon this software code base.
  43.   Peter E. Mellquist
  44. */
  45. char snmpget_cpp_version[]="@(#) SNMP++ $Id: snmpGet.cpp,v 1.8 2005/10/11 18:08:04 katz Exp $";
  46. #include "snmp_pp/snmp_pp.h"
  47. #include <stdlib.h>
  48. #include <stdio.h>
  49. #ifdef SNMP_PP_NAMESPACE
  50. using namespace Snmp_pp;
  51. #endif
  52. #if (__GNUC__ > 2)
  53. #include <iostream>
  54. using std::cerr;
  55. using std::cout;
  56. using std::endl;
  57. using std::flush;
  58. #else
  59. #include <iostream.h>
  60. #endif
  61. int main(int argc, char **argv)
  62. {
  63.    //---------[ check the arg count ]----------------------------------------
  64.    if ( argc < 2) {
  65.   cout << "Usage:n";
  66.   cout << argv[0] << " IpAddress | DNSName [Oid] [options]n";
  67.   cout << "Oid: sysDescr object is defaultn";
  68.   cout << "options: -vN , use SNMP version 1, 2 or 3, default is 1n";
  69.   cout << "         -PPort , remote port to usen";
  70.   cout << "         -CCommunity_name, specify community default is 'public' n";
  71.   cout << "         -rN , retries default is N = 1 retryn";
  72.   cout << "         -tN , timeout in hundredths of seconds; default is N = 100n";
  73. #ifdef _SNMPv3
  74.           cout << "         -snSecurityName, " << endl;
  75.           cout << "         -slN , securityLevel to use, default N = 3 = authPriv" << endl;
  76.           cout << "         -smN , securityModel to use, only default N = 3 = USM possiblen";
  77.           cout << "         -cnContextName, default empty string" << endl;
  78.           cout << "         -ceContextEngineID, as hex e.g. 800007E580, default empty string" << endl;
  79.           cout << "         -authPROT, use authentication protocol NONE, SHA or MD5n";
  80.           cout << "         -privPROT, use privacy protocol NONE, DES, 3DESEDE, IDEA, AES128, AES192 or AES256n";
  81.           cout << "         -uaAuthPasswordn";
  82.           cout << "         -upPrivPasswordn";
  83. #endif
  84.   return 1;
  85.    }
  86.    Snmp::socket_startup();  // Initialize socket subsystem
  87.    //---------[ make a GenAddress and Oid object to retrieve ]---------------
  88.    UdpAddress address( argv[1]);      // make a SNMP++ Generic address
  89.    if ( !address.valid()) {           // check validity of address
  90.   cout << "Invalid Address or DNS Name, " << argv[1] << "n";
  91.   return 1;
  92.    }
  93.    Oid oid("1.3.6.1.2.1.1.1.0");      // default is sysDescr
  94.    if ( argc >= 3) {                  // if 3 args, then use the callers Oid
  95.   if ( strstr( argv[2],"-")==0) {
  96.      oid = argv[2];
  97.      if ( !oid.valid()) {            // check validity of user oid
  98.     cout << "Invalid Oid, " << argv[2] << "n";
  99.     return 1;
  100.          }
  101.       }
  102.    }
  103.    //---------[ determine options to use ]-----------------------------------
  104.    snmp_version version=version1;                  // default is v1
  105.    int retries=1;                                  // default retries is 1
  106.    int timeout=100;                                // default is 1 second
  107.    u_short port=161;                               // default snmp port is 161
  108.    OctetStr community("public");                   // community name
  109. #ifdef _SNMPv3
  110.    OctetStr privPassword("");
  111.    OctetStr authPassword("");
  112.    OctetStr securityName("");
  113.    int securityModel = SecurityModel_USM;
  114.    int securityLevel = SecurityLevel_authPriv;
  115.    OctetStr contextName("");
  116.    OctetStr contextEngineID("");
  117.    long authProtocol = SNMPv3_usmNoAuthProtocol;
  118.    long privProtocol = SNMPv3_usmNoPrivProtocol;
  119.    v3MP *v3_MP;
  120. #endif
  121.    char *ptr;
  122.    for(int x=1;x<argc;x++) {                           // parse for version
  123.      if ( strstr( argv[x],"-v2")!= 0) {
  124.        version = version2c;
  125.        continue;
  126.      }
  127.      if ( strstr( argv[x],"-r")!= 0) {                 // parse for retries
  128.        ptr = argv[x]; ptr++; ptr++;
  129.        retries = atoi(ptr);
  130.        if (( retries<0)|| (retries>5)) retries=1; 
  131.        continue;
  132.      }
  133.      if ( strstr( argv[x], "-t")!=0) {                 // parse for timeout
  134.        ptr = argv[x]; ptr++; ptr++;
  135.        timeout = atoi( ptr);
  136.        if (( timeout < 100)||( timeout>500)) timeout=100;
  137.        continue;
  138.      }
  139.      if ( strstr( argv[x],"-C")!=0) {
  140.        ptr = argv[x]; ptr++; ptr++;
  141.        community = ptr;
  142.        continue;
  143.      }
  144.      if ( strstr( argv[x],"-P")!=0) {
  145.        ptr = argv[x]; ptr++; ptr++;
  146.        sscanf(ptr, "%hu", &port);
  147.        continue;
  148.      }
  149. #ifdef _SNMPv3
  150.      if ( strstr( argv[x],"-v3")!= 0) {
  151.        version = version3;
  152.        continue;
  153.      }
  154.      if ( strstr( argv[x],"-auth") != 0) {
  155.        ptr = argv[x]; ptr+=5;
  156.        if (strcasecmp(ptr, "SHA") == 0)
  157.    authProtocol = SNMP_AUTHPROTOCOL_HMACSHA;
  158.        else if (strcasecmp(ptr, "MD5") == 0)
  159.    authProtocol = SNMP_AUTHPROTOCOL_HMACMD5;
  160.        else
  161.    authProtocol = SNMP_AUTHPROTOCOL_NONE;
  162.        continue;
  163.      }
  164.      if ( strstr( argv[x],"-priv") != 0) {
  165.        ptr = argv[x]; ptr+=5;
  166.        if (strcasecmp(ptr, "DES") == 0)
  167.    privProtocol = SNMP_PRIVPROTOCOL_DES;
  168.        else if (strcasecmp(ptr, "3DESEDE") == 0)
  169.    privProtocol = SNMP_PRIVPROTOCOL_3DESEDE;
  170.        else if (strcasecmp(ptr, "IDEA") == 0)
  171.    privProtocol = SNMP_PRIVPROTOCOL_IDEA;
  172.        else if (strcasecmp(ptr, "AES128") == 0)
  173.    privProtocol = SNMP_PRIVPROTOCOL_AES128;
  174.        else if (strcasecmp(ptr, "AES192") == 0)
  175.    privProtocol = SNMP_PRIVPROTOCOL_AES192;
  176.        else if (strcasecmp(ptr, "AES256") == 0)
  177.    privProtocol = SNMP_PRIVPROTOCOL_AES256;
  178.        else
  179.    privProtocol = SNMP_PRIVPROTOCOL_NONE;
  180.        printf("nnPrivProt : %ldn", privProtocol);
  181.        continue;
  182.      }
  183.      if ( strstr( argv[x],"-sn")!=0) {
  184.        ptr = argv[x]; ptr+=3;
  185.        securityName = ptr;
  186.        continue;
  187.       }
  188.      if ( strstr( argv[x], "-sl")!=0) {
  189.        ptr = argv[x]; ptr+=3;
  190.        securityLevel = atoi( ptr);
  191.        if (( securityLevel < SecurityLevel_noAuthNoPriv) ||
  192.            ( securityLevel > SecurityLevel_authPriv))
  193.          securityLevel = SecurityLevel_authPriv;
  194.        continue;
  195.      }
  196.      if ( strstr( argv[x], "-sm")!=0) {
  197.        ptr = argv[x]; ptr+=3;
  198.        securityModel = atoi( ptr);
  199.        if (( securityModel < SecurityModel_v1) ||
  200.            ( securityModel > SecurityModel_USM))
  201.          securityModel = SecurityModel_USM;
  202.        continue;
  203.      }
  204.      if ( strstr( argv[x],"-cn")!=0) {
  205.        ptr = argv[x]; ptr+=3;
  206.        contextName = ptr;
  207.        continue;
  208.      }
  209.      if ( strstr( argv[x],"-ce")!=0) {
  210.        ptr = argv[x]; ptr+=3;
  211.        contextEngineID = OctetStr::from_hex_string(ptr);
  212.        continue;
  213.      }
  214.      if ( strstr( argv[x],"-ua")!=0) {
  215.        ptr = argv[x]; ptr+=3;
  216.        authPassword = ptr;
  217.        continue;
  218.      }
  219.      if ( strstr( argv[x],"-up")!=0) {
  220.        ptr = argv[x]; ptr+=3;
  221.        privPassword = ptr;
  222.        continue;
  223.      }
  224. #endif
  225.   }
  226.    //----------[ create a SNMP++ session ]-----------------------------------
  227.    int status;
  228.    // bind to any port and use IPv6 if needed
  229.    Snmp snmp(status, 0, (address.get_ip_version() == Address::version_ipv6));
  230.    if ( status != SNMP_CLASS_SUCCESS) {
  231.       cout << "SNMP++ Session Create Fail, " << snmp.error_msg(status) << "n";
  232.       return 1;
  233.    }
  234.    //---------[ init SnmpV3 ]--------------------------------------------
  235. #ifdef _SNMPv3
  236.    if (version == version3) {
  237.      char *engineId = "snmpGet";
  238.      char *filename = "snmpv3_boot_counter";
  239.      unsigned int snmpEngineBoots = 0;
  240.      int status;
  241.      status = getBootCounter(filename, engineId, snmpEngineBoots);
  242.      if ((status != SNMPv3_OK) && (status < SNMPv3_FILEOPEN_ERROR))
  243.      {
  244.        cout << "Error loading snmpEngineBoots counter: " << status << endl;
  245.        return 1;
  246.      }
  247.      snmpEngineBoots++;
  248.      status = saveBootCounter(filename, engineId, snmpEngineBoots);
  249.      if (status != SNMPv3_OK)
  250.      {
  251.        cout << "Error saving snmpEngineBoots counter: " << status << endl;
  252.        return 1;
  253.      }
  254.      int construct_status;
  255.      v3_MP = new v3MP(engineId, snmpEngineBoots, construct_status);
  256.      USM *usm = v3_MP->get_usm();
  257.      usm->add_usm_user(securityName,
  258.        authProtocol, privProtocol,
  259.        authPassword, privPassword);
  260.    }
  261.    else
  262.    {
  263.      // MUST create a dummy v3MP object if _SNMPv3 is enabled!
  264.      int construct_status;
  265.      v3_MP = new v3MP("dummy", 0, construct_status);
  266.    }
  267. #endif
  268.    //--------[ build up SNMP++ object needed ]-------------------------------
  269.    Pdu pdu;                               // construct a Pdu object
  270.    Vb vb;                                 // construct a Vb object
  271.    vb.set_oid( oid);                      // set the Oid portion of the Vb
  272.    pdu += vb;                             // add the vb to the Pdu
  273.    address.set_port(port);
  274.    CTarget ctarget( address);             // make a target using the address
  275. #ifdef _SNMPv3
  276.    UTarget utarget( address);
  277.    if (version == version3) {
  278.      utarget.set_version( version);          // set the SNMP version SNMPV1 or V2 or V3
  279.      utarget.set_retry( retries);            // set the number of auto retries
  280.      utarget.set_timeout( timeout);          // set timeout
  281.      utarget.set_security_model( securityModel);
  282.      utarget.set_security_name( securityName);
  283.      pdu.set_security_level( securityLevel);
  284.      pdu.set_context_name (contextName);
  285.      pdu.set_context_engine_id(contextEngineID);
  286.    }
  287.    else {
  288. #endif
  289.      ctarget.set_version( version);         // set the SNMP version SNMPV1 or V2
  290.      ctarget.set_retry( retries);           // set the number of auto retries
  291.      ctarget.set_timeout( timeout);         // set timeout
  292.      ctarget.set_readcommunity( community); // set the read community name
  293. #ifdef _SNMPv3
  294.    }
  295. #endif
  296.    //-------[ issue the request, blocked mode ]-----------------------------
  297.    cout << "SNMP++ Get to " << argv[1] << " SNMPV" 
  298. #ifdef _SNMPv3
  299.         << ((version==version3) ? (version) : (version+1))
  300. #else
  301.         << (version+1)
  302. #endif
  303.         << " Retries=" << retries
  304.         << " Timeout=" << timeout * 10 <<"ms";
  305. #ifdef _SNMPv3
  306.    if (version == version3)
  307.      cout << endl
  308.           << "securityName= " << securityName.get_printable()
  309.           << ", securityLevel= " << securityLevel
  310.           << ", securityModel= " << securityModel << endl
  311.           << "contextName= " << contextName.get_printable()
  312.           << ", contextEngineID= " << contextEngineID.get_printable()
  313.           << endl;
  314.    else
  315. #endif
  316.      cout << " Community=" << community.get_printable() << endl << flush;
  317.    SnmpTarget *target;
  318. #ifdef _SNMPv3
  319.    if (version == version3)
  320.      target = &utarget;
  321.    else
  322. #endif
  323.      target = &ctarget;
  324.    status = snmp.get( pdu, *target);
  325.    if (status == SNMP_CLASS_SUCCESS)
  326.    {
  327.      pdu.get_vb( vb,0);
  328. #ifdef _SNMPv3
  329.      if (pdu.get_type() == REPORT_MSG) {
  330.        cout << "Received a report pdu: "
  331.             << snmp.error_msg(vb.get_printable_oid()) << endl;
  332.      }
  333. #endif
  334.      cout << "Oid = " << vb.get_printable_oid() << endl
  335.           << "Value = " << vb.get_printable_value() << endl;
  336.      if ((vb.get_syntax() == sNMP_SYNTAX_ENDOFMIBVIEW) ||
  337.          (vb.get_syntax() == sNMP_SYNTAX_NOSUCHINSTANCE) ||
  338.          (vb.get_syntax() == sNMP_SYNTAX_NOSUCHOBJECT))
  339.        cout << "Exception: " << vb.get_syntax() << " occured." << endl;
  340.    }
  341.    else
  342.    {
  343.      cout << "SNMP++ Get Error, " << snmp.error_msg( status)
  344.   << " (" << status <<")" << endl;
  345.    }
  346.    Snmp::socket_cleanup();  // Shut down socket subsystem
  347. }