/* $Id: snmpbulk.src,v 1.5.2.8 2009/10/28 12:28:09 prathika Exp $ */
/*
 * @(#)snmpbulk.java
 * Copyright (c) 1996-2009 ZOHO Corp. All Rights Reserved.
 * Please read the associated COPYRIGHTS file for more details.
 */

/**
 * This is an example program to explain how to write an application to do
 * the basic SNMP operation GETBULK using com.adventnet.snmp.snmp2 package of
 * WebNMSSNMP2 api.
 * The user could run this application by giving any one of the following usage.
 *  
 * Note: The values of non-repeaters max-repetitions should be the last arguments
 *  
 * java snmpbulk [options] hostname oid [oid] ... non-repeaters max-repetitions
 *
 * v2c request:
 * java snmpbulk [-d] [-c community] [-p port] [-t timeout] [-r retries] host OID [OID] ... nonRepeaters maxRepetions
 * e.g. 
 * java snmpbulk -p 161 -c public adventnet 1.2.0 1.2.0 1.5.0 1 2
 *
 * v3 request:
 * java snmpbulk [-d] [-v version(v2,v3)] [-c community] [-p port] [-r retries] [-t timeout] [-u user] [-a auth_protocol] [-w auth_password] [-s priv_password] [-i contextName] host OID [OID] ... nonRepeaters maxRepetions
 * e.g.
 * java snmpbulk -v v3 -c public -p 161 -u advent -a MD5 -w authPass localhost 1.2.0 1.3.0 1.4.0 1 2
 * 
 * If the oid is not starting with a dot (.) it will be prefixed by .1.3.6.1.2.1 .
 * So the entire OID of 1.1.0 will become .1.3.6.1.2.1.1.1.0 . You can also
 * give the entire OID .
 * 
 * If the mib is loaded you can also give string oids in the following formats
 * .iso.org.dod.internet.mgmt.mib-2.system.sysDescr.0 or system.sysDescr.0 or
 * sysDescr.0 .
 *
 * Options:
 * [-d]                - Debug output. By default off.
 * [-c] <community>    - community String. By default "public".
 * [-p] <port>         - remote port no. By default 161.
 * [-t] <Timeout>      - Timeout. By default 5000ms.
 * [-r] <Retries>      - Retries. By default 0. 
 * [-v] <version>      - version(v2 / v3).
 * [-u] <username>     - The v3 principal/userName
 * [-a] <autProtocol>  - The authProtocol(MD5/SHA). Mandatory if authPassword is specified
 * [-pp] <privProtocol>- The privProtocol(DES/AES-128/AES-192/AES-256/3DES)
 * [-w] <authPassword> - The authentication password.
 * [-s] <privPassword> - The privacy protocol password. Must be accompanied with auth password and authProtocol fields.
 * [-n] <contextName>  - The contextName to be used for the v3 pdu.
 * [-i] <contextID>    - The contextID to be used for the v3 pdu.
 * <host> Mandatory    - The RemoteHost (agent).Format (string without double qoutes/IpAddress).
 * <OID>  Mandatory    - Give only one Object Identifier.
 * <nonRepeaters>      - non-repeaters. Mandatory
 * <maxRepetions>      - max-repetitions. Mandatory
 */

import java.lang.*;
import java.util.*;
import java.net.*;
import com.adventnet.snmp.snmp2.*;
import com.adventnet.snmp.snmp2.usm.*;

public class snmpbulk
{


    public static void main(String args[])
    {
        // Take care of getting options
        String usage =
            "\nsnmpbulk [-d] [-v version(v2,v3)] \n" +
            "[-c community] [-p port] [-r retries] \n" +
            "[-t timeout] [-u user] [-a auth_protocol] \n" +
            "[-w auth_password] [-s priv_password] \n" +
            "[-n contextName] [-i contextID] \n" +
            "[-DB_driver database_driver]\n" +
            "[-DB_url database_url]\n" +
            "[-DB_username database_username]\n" +
            "[-DB_password database_password]\n" +
						"[-pp privProtocol(DES/AES-128/AES-192/AES-256/3DES)]\n" +
						"host OID [OID] ... \n" +
            "nonRepeaters maxRepetions";
        String options[] =
        {
            "-d", "-c",  "-wc", "-p", "-r", "-t", "-m",
            "-v", "-u", "-a", "-w", "-s", "-n", "-i",
            "-DB_driver", "-DB_url", "-DB_username", "-DB_password","-pp"
        };

        String values[] =
        {
            "None", null, null, null, null, null, "None",
            null, null, null, null, null, null, null,
            null, null, null, null,null
        };
        
        ParseOptions opt = new ParseOptions(args,options,values, usage);

        // Start SNMP API
        SnmpAPI api;
        api = new SnmpAPI();
        if (values[0].equals("Set"))
        {
            api.setDebug( true );        
        }

        if (opt.remArgs.length<4)
        {
            opt.usage_error();
        }

        // Open session
        SnmpSession session = new SnmpSession(api);

        int PORT = 3;
        
        SnmpPDU pdu = new SnmpPDU();
        UDPProtocolOptions ses_opt = new UDPProtocolOptions();
        ses_opt.setRemoteHost(opt.remArgs[0]);
        if(values[PORT] != null)
        {
            try
            {
                ses_opt.setRemotePort(Integer.parseInt(values[PORT]));
            }
            catch(Exception exp)
            {
                System.out.println("Invalid port: " + values[PORT]);
                System.exit(1);
            }
        }
        pdu.setProtocolOptions(ses_opt);

        //set the values
        SetValues setVal = new SetValues(session, values);
        if(setVal.usage_error)
        {
            opt.usage_error();
        }

        // set version for default
        if(values[7] == null)
        {
            session.setVersion(SnmpAPI.SNMP_VERSION_2C);
        }

        if(session.getVersion()==SnmpAPI.SNMP_VERSION_1)
        {
            opt.usage_error();
        }

        String driver = values[14];
        String url = values[15];
        String username = values[16];
        String password = values[17];
        if(driver != null || url != null ||
            username != null || password != null)
        {
            if(session.getVersion() != 3)
            {
                System.out.println(
                    "Database option can be used only for SNMPv3.");
                System.exit(1);
            }
            if(driver == null)
            {
                System.out.println(
                    "The Database driver name should be given.");
                System.exit(1);
            }
            if(url == null)
            {
                System.out.println("The Database URL should be given.");
                System.exit(1);
            }
            try
            {
                api.setV3DatabaseFlag(true);
                api.initJdbcParams(driver, url, username, password);
            }
            catch(Exception exp)
            {
                System.out.println("Unable to Establish Database Connection.");
                System.out.println("Please check the driverName and url.");
                System.exit(1);
            }
        }

        // Build GetBulk request PDU
       // SnmpPDU pdu = new SnmpPDU();
        pdu.setCommand( api.GETBULK_REQ_MSG );

        // add OIDs
        for (int i=1;i<(opt.remArgs.length)-2;i++)
        {
            SnmpOID oid = new SnmpOID(opt.remArgs[i]);
            if (oid.toValue() == null)
            {
                System.err.println("Invalid OID argument: " + opt.remArgs[i]);
            }
            else 
            {
                pdu.addNull(oid);
            }
        }

        try
        {
            // set non-repeaters
            pdu.setErrstat(
                Integer.parseInt(opt.remArgs[(opt.remArgs.length)-2]) );
        }
        catch(NumberFormatException nfe)
        {
            System.out.println("Non-Repeaters value should be an integer. " +
                opt.remArgs[(opt.remArgs.length)-2]);
            System.exit(1);
        }
        
        try
        {
            // set max-repetitions
            pdu.setErrindex(
                Integer.parseInt(opt.remArgs[(opt.remArgs.length)-1]) );
        }
        catch(NumberFormatException nfe)
        {
            System.out.println("Non-Repeaters value should be an integer. " +
                opt.remArgs[(opt.remArgs.length)-1]);
            System.exit(1);
        }
        try
        {
            // open session
            session.open();
        } 
        catch (SnmpException e)
        {
            System.err.println("Error while opening session " +
                e.getMessage());
            System.exit(1);
        }

        if(session.getVersion()==SnmpAPI.SNMP_VERSION_3)
        {
            pdu.setUserName(setVal.userName.getBytes());
            try
            {
                USMUtils.init_v3_parameters(
                setVal.userName,
								null,
                setVal.authProtocol,
                setVal.authPassword,
                setVal.privPassword,
                ses_opt,
                session,
								false,
								setVal.privProtocol);
            }
            catch(Exception exp)
            {
                System.out.println(exp.getMessage());
                System.exit(1);
            }
            pdu.setContextName(setVal.contextName.getBytes());
            pdu.setContextID(setVal.contextID.getBytes());
        }
        SnmpPDU res_pdu = null;
        try
        {
            res_pdu = session.syncSend(pdu);
        }
        catch(SnmpException e)
        {
            System.err.println("Sending PDU: " + e.getMessage());
            System.exit(1);
        }
        // timeout
        if (res_pdu == null)
        {
            System.out.println("Request timed out to: " + opt.remArgs[0] );
            System.exit(1);
        }

        // print and exit
        String res = "Response PDU received from " + res_pdu.getProtocolOptions().getSessionId() + ".";
        if(res_pdu.getVersion() < 3)
        {
            res = res + " Community = \"" + res_pdu.getCommunity() + "\".\n";
        }
        System.out.println(res);
        // check for error
        if (res_pdu.getErrstat() != 0)
        {
            System.out.println("Error indication in response: " +
                SnmpException.exceptionString((byte)res_pdu.getErrstat()) +
                "\nErrindex: " + res_pdu.getErrindex());
        }
        else
        {
            Enumeration e = res_pdu.getVariableBindings().elements();
            while(e.hasMoreElements())
            {
                int error = 0;
                SnmpVarBind varbind = (SnmpVarBind)e.nextElement();
                // check for error index
                if ( (error = varbind.getErrindex()) != 0)
                {
                    System.out.println("Error Indication in response: " +
                        SnmpException.exceptionString((byte)error));
                }
                // print varbind
                System.out.println(varbind.toTagString() + "\n");
            }
        }

        //close session
        session.close();
        // stop api thread
        api.close();
    }

}