sendv2trap.java.txt
上传用户:aonuowh
上传日期:2021-05-23
资源大小:35390k
文件大小:10k
源码类别:

SNMP编程

开发平台:

C/C++

  1. /* $Id: sendv2trap.src,v 1.5 2002/09/09 05:36:28 tonyjpaul Exp $ */
  2. /*
  3.  * @(#)sendtrap.java
  4.  * Copyright (c) 1996-2003 AdventNet, Inc. All Rights Reserved.
  5.  * Please read the associated COPYRIGHTS file for more details.
  6.  */
  7. /** 
  8.  *  Send SNMP trap based on command line arguments.  Loads MIBs 
  9.  *  as specified, and converts to/from names for loaded MIB data. 
  10.  *  Since variable types are not input, MIBs have to be loaded for
  11.  *  any variables being used in trap PDU.
  12.  *
  13.  * This is an example program to explain how to write an application to send a
  14.  * Trap message using com.adventnet.snmp.beans package of AdventNetSNMP2 api.
  15.  * The user could run this application by giving any one of the following usage.
  16.  *  
  17.  * Note: Refer README file  
  18.  *
  19.  * If the oid is not starting with a dot (.) it will be prefixed by .1.3.6.1.2.1 .
  20.  * So the entire OID of 1.1.0 will become .1.3.6.1.2.1.1.1.0 . You can also
  21.  * give the entire OID .
  22.  * 
  23.  * If the mib is loaded you can also give string oids in the following formats
  24.  * .iso.org.dod.internet.mgmt.mib-2.system.sysDescr.0 or system.sysDescr.0 or
  25.  * sysDescr.0 .
  26.  *
  27.  * [-d]                - Debug output. By default off.
  28.  * [-c] <community>    - community String. By default "public".
  29.  * [-p] <port>         - remote port no. By default 161.
  30.  * [-m] <mibs>         - The mibs to be loaded. 
  31.  * [-v] <version>      - version(v2 / v3). By default v2.
  32.  * [-e] <engineID>     - EngineID for this entity.
  33.  * [-u] <username>     - The v3 principal/userName
  34.  * [-a] <autProtocol>  - The authProtocol(MD5/SHA). Mandatory if authPassword is specified
  35.  * [-w] <authPassword> - The authentication password.
  36.  * [-s] <privPassword> - The privacy protocol password. Must be accompanied with auth password and authProtocol fields.
  37.  * [-i] <contextName>  - Context Name.
  38.  * <host> Mandatory    - The RemoteHost (agent).Format (string without double qoutes/IpAddress).
  39.  * <timeticks> Mandatory - the value of object sysUpTime when the event occurred
  40.  * <OID-value> Mandatory - Object Identifier  
  41.  * <OID>  Mandatory    - Give multiple no. of Object Identifiers.
  42.  */
  43. import java.lang.*;
  44. import java.util.*;
  45. import java.net.*;
  46. import com.adventnet.snmp.beans.*;
  47. import com.adventnet.snmp.snmp2.SnmpException;
  48. import com.adventnet.snmp.snmp2.SnmpOID;
  49. import com.adventnet.snmp.snmp2.usm.*;
  50. public class sendv2trap {
  51.     private static final int ENGINEID = 4;
  52.     private static final int VERSION = 5;
  53.     private static final int USER_NAME = 6;
  54.     private static final int AUTH_PROTOCOL = 7;
  55.     private static final int AUTH_PASSWORD = 8;
  56.     private static final int PRIV_PASSWORD = 9;
  57.     private static final int CONTEXT_NAME = 10;
  58.     
  59.     private static final int COMMUNITY = 1;
  60.     private static final int PORT = 2;
  61.     private static final int MIBS = 3;
  62.     private static final int DEBUG = 0;
  63.     private static final int AGENTADDRESS = 11;
  64.     public static void main(String args[]) {
  65.     // Take care of getting options
  66.     String usage = "sendv2trap [-d Debug][-v version(v2,v3)] [-c community] [-p port] [-e engineID(1234.../0x1234...)] [-u user] [-a auth_protocol(MD5/SHA)] [-w auth_password] [-s priv_password] [-i contextName] [-g agent-address] [-m MIB_file] host TimeTicksvalue trapOID [OID value] ...";
  67.     String options[] = { "-d", "-c", "-p", "-m", "-e", "-v", "-u", "-a", "-w", "-s", "-i" ,"-g"};
  68.     String values[] = { "None", null, null, null,null, null, null, null, null, null, null,null};
  69. String id = new String("");
  70.     long upTime=0;
  71.     ParseOptions opt = new ParseOptions(args,options,values, usage);
  72.     if (opt.remArgs.length<3) opt.usage_error();
  73.     // Use an SNMP target bean to perform SNMP operations
  74.     SnmpTarget target = new SnmpTarget();
  75.  //To load MIBs from compiled file
  76.  target.setLoadFromCompiledMibs(true);
  77.     // Set Debug mode
  78.     if (values[DEBUG].equals("Set")) target.setDebug(true);
  79.     target.setTargetHost( opt.remArgs[0] );  // set destination hostname
  80.     
  81.     //Assign the port for sending the trap.
  82.     target.setTargetPort(162);
  83.     if(values[PORT]!=null) {
  84.         try {
  85.             target.setTargetPort(Integer.parseInt(values[PORT]));
  86.         }
  87.         catch(NumberFormatException e) {
  88.             System.err.println("Sending trap to port 162");
  89.         }
  90.     }
  91.     try {
  92.         upTime = Long.parseLong(opt.remArgs[1]);
  93.     } catch (NumberFormatException ex) {
  94.         System.err.println("Invalid Integer Argument for upTime "+ex);
  95.         opt.usage_error();
  96.     }
  97.     
  98.     target.setSnmpVersion(target.VERSION2C);
  99.     if (values[VERSION] != null) // set the version if specified
  100.         if(values[VERSION].equals("v3"))
  101.             target.setSnmpVersion(target.VERSION3);
  102.     if(target.getSnmpVersion()== target.VERSION3) {
  103. if(values[ENGINEID] != null){
  104. id =  values[ENGINEID];
  105. if(id.startsWith("0x") || id.startsWith("0X"))
  106. id = new String(gethexValue(values[ENGINEID]));
  107. }
  108.         if (values[USER_NAME] != null) {
  109.             target.setPrincipal(values[USER_NAME]);
  110.         
  111.         if ((values[AUTH_PROTOCOL] != null) && (values[AUTH_PASSWORD] != null)) {
  112.             if(values[AUTH_PROTOCOL].equals("SHA"))
  113.                 target.setAuthProtocol(target.SHA_AUTH);
  114.             else if(values[AUTH_PROTOCOL].equals("MD5"))
  115.                 target.setAuthProtocol(target.MD5_AUTH);
  116.             else
  117.                 target.setAuthProtocol(target.NO_AUTH);            
  118.             target.setAuthPassword(values[AUTH_PASSWORD]);
  119.             if (values[PRIV_PASSWORD] != null) 
  120.                 target.setPrivPassword(values[PRIV_PASSWORD]);
  121.         }
  122.         else if ((values[AUTH_PROTOCOL] != null) || (values[AUTH_PASSWORD] != null) || (values[PRIV_PASSWORD] != null)) {
  123.             opt.usage_error();
  124.         }
  125.         }
  126.         else {
  127.             System.err.println("UserName Missing");
  128.             opt.usage_error();
  129.         }
  130. // Configure the USM entries on this entity.
  131. createUSMTable(target.getPrincipal().getBytes(),id.getBytes(),
  132. target.getAuthProtocol(),target.getAuthPassword(),
  133. target.getPrivPassword(), target);
  134.     }
  135. String community="";
  136. int extraVarCount=0;
  137. String agentAddress="";
  138. if(values[COMMUNITY] != null)
  139. {
  140.         target.setCommunity( values[COMMUNITY]);
  141. community=values[COMMUNITY];
  142. extraVarCount++;
  143. }
  144.     if (values[MIBS] != null) try {  // Load the MIB files 
  145.         System.err.println("Loading MIBs: "+values[MIBS]);
  146.         target.loadMibs(values[MIBS]);
  147.         System.err.println("Done.");
  148.     } catch (Exception ex) {
  149.         System.err.println("Error loading MIBs: "+ex);
  150. System.exit(1);
  151.     }
  152. if (values[AGENTADDRESS] != null) 
  153. {  // Load the MIB files 
  154. agentAddress=values[AGENTADDRESS];
  155. extraVarCount++;
  156.     }
  157.     // Put together OID and variable value lists from command line
  158.     String oids[] = null, var_values[] = null;  // trap oids and values
  159.     int num_varbinds = 0;
  160.     for (int i=3;i<opt.remArgs.length;i+=2) { // add Variable Bindings
  161.         if (opt.remArgs.length < i+2) //need "{OID type value}"
  162.            opt.usage_error(); 
  163.         num_varbinds++;
  164.     }
  165.  oids = new String[num_varbinds+extraVarCount];
  166.     var_values = new String[num_varbinds+extraVarCount];
  167. if(!agentAddress.equals(""))
  168. {
  169. oids[0]=".1.3.6.1.6.3.18.1.3.0";
  170. var_values[0]=agentAddress;
  171. if(target.getMibOperations()!=null)
  172. {
  173. if(target.getMibOperations().getMibNode(new SnmpOID(".1.3.6.1.6.3.18.1.3.0"))==null)
  174. {
  175. System.out.println("Error : SNMP-TARGET and SNMP-COMMUNITY mibs are not loaded");
  176. System.exit(0);
  177. }
  178. }
  179. else
  180. {
  181. System.out.println("Error : SNMP-TARGET and SNMP-COMMUNITY mibs are not loaded");
  182. System.exit(0);
  183. }
  184. }
  185. if(!community.equals(""))
  186. {
  187. oids[1]=".1.3.6.1.6.3.18.1.4.0";
  188. var_values[1]=community;
  189. if(target.getMibOperations()!=null)
  190. {
  191. if(target.getMibOperations().getMibNode(new SnmpOID(".1.3.6.1.6.3.18.1.4.0"))==null)
  192. {
  193. System.out.println("Error : SNMP-TARGET and SNMP-COMMUNITY mibs are not loaded");
  194. System.exit(0);
  195. }
  196. }
  197. else
  198. {
  199. System.out.println("Error : SNMP-TARGET and SNMP-COMMUNITY mibs are not loaded");
  200. System.exit(0);
  201. }
  202. }
  203.     for (int i=0;i<num_varbinds;i++) { // add Variable Bindings
  204.         oids[i+extraVarCount] = opt.remArgs[(2*i)+3];
  205.         var_values[i+extraVarCount] = opt.remArgs[(2*i)+4];
  206.      }
  207.     try {  // use SnmpTarget methods to send trap w/ specified OIDs/values
  208.         target.setObjectIDList(oids);
  209.         target.snmpSendNotification(upTime, opt.remArgs[2], var_values);
  210.         // allow time to send trap before exiting
  211.         Thread.sleep(500);
  212.     } catch (Exception e) {
  213.         System.err.println("Error Sending Trap: "+e.getMessage());
  214.     }
  215.     
  216.     if (target.getErrorCode() != -1) {
  217.      System.out.println(target.getErrorString());
  218.     }
  219.     System.exit(0);
  220.     
  221.     }
  222.     public static void createUSMTable(byte[] name, byte[] engineID, 
  223. int authProtocol, String authPassword,
  224. String privPassword, SnmpTarget target)
  225.     {
  226. byte level = 0;
  227.     
  228. USMUserTable uut = target.getUSMTable();
  229. USMUserEntry entry = new USMUserEntry(name, engineID);
  230. entry.setAuthProtocol(authProtocol);
  231. if ((authProtocol != USMUserEntry.NO_AUTH) && (authPassword != null))
  232. {
  233. byte[] authKey = USMUtils.password_to_key(authProtocol, 
  234. authPassword.getBytes(), 
  235. authPassword.getBytes().length,
  236. engineID);
  237. entry.setAuthKey(authKey);
  238. level = 1;
  239.             
  240. if ((privPassword != null)&&(privPassword.length()>0))
  241. {
  242. byte[] tempKey = USMUtils.password_to_key(authProtocol, 
  243. privPassword.getBytes(),
  244. privPassword.getBytes().length,
  245. engineID);
  246. byte privKey[]=new byte[16];
  247. System.arraycopy(tempKey,0,privKey,0,16);
  248. entry.setPrivKey(privKey);
  249. level |= 2;
  250. }
  251. }
  252.     
  253. entry.setSecurityLevel(level);
  254. uut.addEntry(entry);
  255. target.setSnmpEngineID(engineID);
  256. }
  257.     private static byte[] gethexValue(String value)
  258.     {
  259.         byte temp;
  260.         byte[] Key=new byte[value.length()/2 - 1];
  261.         String ss,str;
  262.         ss = value.substring(2);
  263.         for(int i = 0; i < ss.length(); i+=2)
  264.         {
  265.             str = ss.substring(i,i+2);
  266.             temp = (byte)Integer.parseInt(str,16);
  267.             Key[i/2] = temp;
  268.         }
  269.         return Key;    
  270.     }
  271. }