www.pudn.com > javasnmpv3.rar > snmpSet.cpp
/*_############################################################################
_##
_## snmpSet.cpp
_##
_## SNMP++v3.2.22
_## -----------------------------------------------
_## Copyright (c) 2001-2007 Jochen Katz, Frank Fock
_##
_## This software is based on SNMP++2.6 from Hewlett Packard:
_##
_## 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 and Jochen Katz make 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.
_##
_## Stuttgart, Germany, Wed May 2 23:22:30 CEST 2007
_##
_##########################################################################*/
/*
snmpSet.cpp
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.
Peter E. Mellquist
*/
char snmpset_cpp_version[]="@(#) SNMP++ $Id: snmpSet.cpp 264 2006-06-16 20:53:15Z fock $";
#include "snmp_pp/snmp_pp.h"
#include
#include
#ifdef WIN32
#define strcasecmp stricmp
#endif
#ifdef SNMP_PP_NAMESPACE
using namespace Snmp_pp;
#endif
#if (__GNUC__ > 2)
#include
using std::cerr;
using std::cout;
using std::cin;
using std::endl;
using std::flush;
#else
#include
#endif
// determine the smi type and get a value from the user
int determine_vb( SmiUINT32 val, Vb &vb)
{
char buffer[255];
if (val == sNMP_SYNTAX_NOSUCHINSTANCE)
{
cout << "Instance does not exists but can be created.\n";
cout << "Please enter one of the following types:\n\n";
cout << "Integer: " << sNMP_SYNTAX_INT << "\n";
cout << "Bits: " << sNMP_SYNTAX_BITS << "\n";
cout << "STRING: " << sNMP_SYNTAX_OCTETS << "\n";
cout << "Oid: " << sNMP_SYNTAX_OID << "\n";
cout << "IpAddress:" << sNMP_SYNTAX_IPADDR << "\n\n";
cout << "Please choose value type: ";
cin >> val;
vb.set_syntax(val);
}
cout << "Value Type is ";
switch (val) {
// octet string
case sNMP_SYNTAX_OCTETS:
{
cout << "Octet String\n";
cout << "Please enter new value: ";
cin >> buffer;
OctetStr octetstr( buffer);
if ( octetstr.valid()) {
vb.set_value( octetstr);
return TRUE;
}
else {
cout << "Invalid OctetStr\n";
return FALSE;
}
}
// IP Address
case sNMP_SYNTAX_IPADDR:
{
cout << "IP Address\n";
cout << "Please enter new value: ";
cin >> buffer;
IpAddress ipaddress( buffer);
if ( ipaddress.valid()) {
vb.set_value( ipaddress);
return TRUE;
}
else {
cout << "Invalid IP Address\n";
return FALSE;
}
}
// Oid
case sNMP_SYNTAX_OID:
{
cout << "Oid\n";
cout << "Please enter new value: ";
cin >> buffer;
Oid oid( buffer);
if ( oid.valid()) {
vb.set_value( oid);
return TRUE;
}
else {
cout << "Invalid Oid\n";
return FALSE;
}
}
// TimeTicks
case sNMP_SYNTAX_TIMETICKS:
{
cout << "TimeTicks\n";
cout << "Please enter new value: ";
cin >> buffer;
unsigned long i;
i = atol( buffer);
TimeTicks timeticks( i);
if ( timeticks.valid()) {
vb.set_value( timeticks);
return TRUE;
}
else {
cout << "Invalid TimeTicks\n";
return FALSE;
}
}
// Gauge32
case sNMP_SYNTAX_GAUGE32:
{
cout << "Gauge32\n";
cout << "Please enter new value: ";
cin >> buffer;
unsigned long i;
i = atol( buffer);
Gauge32 gauge32(i);
if ( gauge32.valid()) {
vb.set_value( gauge32);
return TRUE;
}
else {
cout << "Invalid Gauge32\n";
return FALSE;
}
}
case sNMP_SYNTAX_CNTR32:
{
cout << "Counter32\n";
cout << "Please enter new value: ";
cin >> buffer;
unsigned long i;
i = atol( buffer);
Counter32 counter32(i);
if ( counter32.valid()) {
vb.set_value( counter32);
return TRUE;
}
else {
cout << "Invalid Counter32\n";
return FALSE;
}
}
case sNMP_SYNTAX_CNTR64:
{
cout << "Counter64\n";
cout << "Please enter value (low 32 bit): ";
cin >> buffer;
unsigned long i;
i = atol( buffer);
Counter64 counter64;
counter64.set_low(i);
cout << "Please enter value (high 32 bit): ";
cin >> buffer;
i = atol( buffer);
counter64.set_high(i);
if ( counter64.valid()) {
vb.set_value( counter64);
return TRUE;
}
else {
cout << "Invalid Counter64\n";
return FALSE;
}
}
case sNMP_SYNTAX_INT:
{
cout << "Integer\n";
cout << "Please enter new value: ";
cin >> buffer;
unsigned long i;
i = atol( buffer);
long l ;
l = ( long) i;
vb.set_value( l);
return TRUE;
}
case sNMP_SYNTAX_NOSUCHOBJECT:
{
cout << "NO SUCH OBJECT\n";
cout << "Object cannot be created.\n";
return FALSE;
}
default:
cout << "Unknown Data Type " << val << "\n";
return FALSE;
}
}
int main(int argc, char **argv)
{
//---------[ check the arg count ]----------------------------------------
if ( argc < 2) {
cout << "Usage:\n";
cout << argv[0] << " IpAddress | DNSName [Oid] [options]\n";
cout << "Oid: sysDescr object is default\n";
cout << "options: -vN , use SNMP version 1, 2 or 3, default is 1\n";
cout << " -PPort , remote port to use\n";
cout << " -CCommunity_name, specify SET community default is 'public' \n";
cout << " -GCommunity_name, specify GET community default is set community value \n";
cout << " -rN , retries default is N = 1 retry\n";
cout << " -tN , timeout in hundredths of seconds; default is N = 100\n";
#ifdef _SNMPv3
cout << " -snSecurityName, " << endl;
cout << " -slN , securityLevel to use, default N = 3 = authPriv" << endl;
cout << " -smN , securityModel to use, only default N = 3 = USM possible\n";
cout << " -cnContextName, default empty string" << endl;
cout << " -ceContextEngineID, as hex e.g. 800007E580, default empty string" << endl;
cout << " -authPROT, use authentication protocol NONE, SHA or MD5\n";
cout << " -privPROT, use privacy protocol NONE, DES, 3DESEDE, IDEA, AES128, AES192 or AES256\n";
cout << " -uaAuthPassword\n";
cout << " -upPrivPassword\n";
#endif
return 1;
}
Snmp::socket_startup(); // Initialize socket subsystem
//---------[ make a GenAddress and Oid object to retrieve ]---------------
UdpAddress address( argv[1]); // make a SNMP++ Generic address
if ( !address.valid()) { // check validity of address
cout << "Invalid Address or DNS Name, " << argv[1] << "\n";
return 1;
}
Oid oid("1.3.6.1.2.1.1.4.0"); // default is sysName
if ( argc >= 3) { // if 3 args, then use the callers Oid
if ( strstr( argv[2],"-")==0) {
oid = argv[2];
if ( !oid.valid()) { // check validity of user oid
cout << "Invalid Oid, " << argv[2] << "\n";
return 1;
}
}
}
//---------[ determine options to use ]-----------------------------------
snmp_version version=version1; // default is v1
int retries=1; // default retries is 1
int timeout=100; // default is 1 second
u_short port=161; // default snmp port is 161
OctetStr community("public"); // community name
OctetStr get_community;
#ifdef _SNMPv3
OctetStr privPassword("");
OctetStr authPassword("");
OctetStr securityName("");
int securityModel = SecurityModel_USM;
int securityLevel = SecurityLevel_authPriv;
OctetStr contextName("");
OctetStr contextEngineID("");
long authProtocol = SNMPv3_usmNoAuthProtocol;
long privProtocol = SNMPv3_usmNoPrivProtocol;
v3MP *v3_MP;
#endif
char *ptr;
for(int x=1;x5)) retries=1;
continue;
}
if ( strstr( argv[x], "-t")!=0) { // parse for timeout
ptr = argv[x]; ptr++; ptr++;
timeout = atoi( ptr);
if (( timeout < 100)||( timeout>500)) timeout=100;
continue;
}
if ( strstr( argv[x],"-C")!=0) {
ptr = argv[x]; ptr++; ptr++;
community = ptr;
continue;
}
if ( strstr( argv[x],"-G")!=0) {
ptr = argv[x]; ptr++; ptr++;
get_community = ptr;
continue;
}
if ( strstr( argv[x],"-P")!=0) {
ptr = argv[x]; ptr++; ptr++;
sscanf(ptr, "%hu", &port);
continue;
}
#ifdef _SNMPv3
if ( strstr( argv[x],"-v3")!= 0) {
version = version3;
continue;
}
if ( strstr( argv[x],"-auth") != 0) {
ptr = argv[x]; ptr+=5;
if (strcasecmp(ptr, "SHA") == 0)
authProtocol = SNMP_AUTHPROTOCOL_HMACSHA;
else if (strcasecmp(ptr, "MD5") == 0)
authProtocol = SNMP_AUTHPROTOCOL_HMACMD5;
else
authProtocol = SNMP_AUTHPROTOCOL_NONE;
continue;
}
if ( strstr( argv[x],"-priv") != 0) {
ptr = argv[x]; ptr+=5;
if (strcasecmp(ptr, "DES") == 0)
privProtocol = SNMP_PRIVPROTOCOL_DES;
else if (strcasecmp(ptr, "3DESEDE") == 0)
privProtocol = SNMP_PRIVPROTOCOL_3DESEDE;
else if (strcasecmp(ptr, "IDEA") == 0)
privProtocol = SNMP_PRIVPROTOCOL_IDEA;
else if (strcasecmp(ptr, "AES128") == 0)
privProtocol = SNMP_PRIVPROTOCOL_AES128;
else if (strcasecmp(ptr, "AES192") == 0)
privProtocol = SNMP_PRIVPROTOCOL_AES192;
else if (strcasecmp(ptr, "AES256") == 0)
privProtocol = SNMP_PRIVPROTOCOL_AES256;
else
privProtocol = SNMP_PRIVPROTOCOL_NONE;
printf("\n\nPrivProt : %ld\n", privProtocol);
continue;
}
if ( strstr( argv[x],"-sn")!=0) {
ptr = argv[x]; ptr+=3;
securityName = ptr;
continue;
}
if ( strstr( argv[x], "-sl")!=0) {
ptr = argv[x]; ptr+=3;
securityLevel = atoi( ptr);
if (( securityLevel < SecurityLevel_noAuthNoPriv) ||
( securityLevel > SecurityLevel_authPriv))
securityLevel = SecurityLevel_authPriv;
continue;
}
if ( strstr( argv[x], "-sm")!=0) {
ptr = argv[x]; ptr+=3;
securityModel = atoi( ptr);
if (( securityModel < SecurityModel_v1) ||
( securityModel > SecurityModel_USM))
securityModel = SecurityModel_USM;
continue;
}
if ( strstr( argv[x],"-cn")!=0) {
ptr = argv[x]; ptr+=3;
contextName = ptr;
continue;
}
if ( strstr( argv[x],"-ce")!=0) {
ptr = argv[x]; ptr+=3;
contextEngineID = OctetStr::from_hex_string(ptr);
continue;
}
if ( strstr( argv[x],"-ua")!=0) {
ptr = argv[x]; ptr+=3;
authPassword = ptr;
continue;
}
if ( strstr( argv[x],"-up")!=0) {
ptr = argv[x]; ptr+=3;
privPassword = ptr;
continue;
}
#endif
}
if (get_community.len() == 0)
get_community = community;
//----------[ create a SNMP++ session ]-----------------------------------
int status;
// bind to any port and use IPv6 if needed
Snmp snmp(status, 0, (address.get_ip_version() == Address::version_ipv6));
if ( status != SNMP_CLASS_SUCCESS) {
cout << "SNMP++ Session Create Fail, " << snmp.error_msg(status) << "\n";
return 1;
}
//---------[ init SnmpV3 ]--------------------------------------------
#ifdef _SNMPv3
if (version == version3) {
char *engineId = "snmpSet";
char *filename = "snmpv3_boot_counter";
unsigned int snmpEngineBoots = 0;
int status;
status = getBootCounter(filename, engineId, snmpEngineBoots);
if ((status != SNMPv3_OK) && (status < SNMPv3_FILEOPEN_ERROR))
{
cout << "Error loading snmpEngineBoots counter: " << status << endl;
return 1;
}
snmpEngineBoots++;
status = saveBootCounter(filename, engineId, snmpEngineBoots);
if (status != SNMPv3_OK)
{
cout << "Error saving snmpEngineBoots counter: " << status << endl;
return 1;
}
int construct_status;
v3_MP = new v3MP(engineId, snmpEngineBoots, construct_status);
USM *usm = v3_MP->get_usm();
usm->add_usm_user(securityName,
authProtocol, privProtocol,
authPassword, privPassword);
}
else
{
// MUST create a dummy v3MP object if _SNMPv3 is enabled!
int construct_status;
v3_MP = new v3MP("dummy", 0, construct_status);
}
#endif
//--------[ build up SNMP++ object needed ]-------------------------------
Pdu pdu; // construct a Pdu object
Vb vb; // construct a Vb object
vb.set_oid( oid); // set the Oid portion of the Vb
pdu += vb; // add the vb to the Pdu
address.set_port(port);
CTarget ctarget( address); // make a target using the address
#ifdef _SNMPv3
UTarget utarget( address);
if (version == version3) {
utarget.set_version( version); // set the SNMP version SNMPV1 or V2 or V3
utarget.set_retry( retries); // set the number of auto retries
utarget.set_timeout( timeout); // set timeout
utarget.set_security_model( securityModel);
utarget.set_security_name( securityName);
pdu.set_security_level( securityLevel);
pdu.set_context_name (contextName);
pdu.set_context_engine_id(contextEngineID);
}
else {
#endif
ctarget.set_version( version); // set the SNMP version SNMPV1 or V2
ctarget.set_retry( retries); // set the number of auto retries
ctarget.set_timeout( timeout); // set timeout
ctarget.set_readcommunity( get_community); // set the read community name
ctarget.set_writecommunity( community);// set the write community name
#ifdef _SNMPv3
}
#endif
//-------[ issue the request, blocked mode ]-----------------------------
cout << "SNMP++ Set to " << argv[1] << " SNMPV"
#ifdef _SNMPv3
<< ((version==version3) ? (version) : (version+1))
#else
<< (version+1)
#endif
<< " Retries=" << retries
<< " Timeout=" << timeout * 10 <<"ms";
#ifdef _SNMPv3
if (version == version3)
cout << endl
<< "securityName= " << securityName.get_printable()
<< ", securityLevel= " << securityLevel
<< ", securityModel= " << securityModel << endl
<< "contextName= " << contextName.get_printable()
<< ", contextEngineID= " << contextEngineID.get_printable()
<< endl;
else
#endif
cout << " SET-community=" << community.get_printable()
<< " GET-community=" << get_community.get_printable() << endl << flush;
SnmpTarget *target;
#ifdef _SNMPv3
if (version == version3)
target = &utarget;
else
#endif
target = &ctarget;
// first get the variabel to determine its type
if (( status = snmp.get( pdu,*target))== SNMP_CLASS_SUCCESS) {
pdu.get_vb( vb,0);
cout << "Oid = " << vb.get_printable_oid() << endl
<< "Current Value = " << vb.get_printable_value() << endl;
#ifdef _SNMPv3
if (pdu.get_type() == REPORT_MSG) {
cout << "Received a reportPdu: "
<< snmp.error_msg( vb.get_printable_oid())
<< endl
<< vb.get_printable_oid() << " = "
<< vb.get_printable_value() << endl;
return -5;
}
#endif
if ( determine_vb(vb.get_syntax(), vb)) {
// do the Set
Pdu setpdu;
#ifdef _SNMPv3
setpdu.set_security_level(securityLevel);
setpdu.set_context_name (contextName);
setpdu.set_context_engine_id(contextEngineID);
#endif
vb.set_oid( oid); // use the same oid as the inquire
setpdu += vb;
status = snmp.set( setpdu, *target);
cout << "Set Status = " << snmp.error_msg( status) << "\n";
}
}
else
cout << "SNMP++ Set Error, " << snmp.error_msg( status) << "\n";
Snmp::socket_cleanup(); // Shut down socket subsystem
}