snmputil.java

用java实现对snmp v1

				scopedPDU.setContextEngineID(_contextEngineID);
			}
			if (_contextName != null) {
				scopedPDU.setContextName(_contextName);
			}
		} else {
			request = new PDU();
		}
		request.setType(_pduType);
		return request;
	}

	private Vector getVariableBinding(String varbind) {
		Vector v = new Vector(varbind.length());
		String oid = null;
		char type = 'i';
		String value = null;
		int equal = varbind.indexOf("={");
		if (equal > 0) {
			oid = varbind.substring(0, equal);
			type = varbind.charAt(equal+2);
			value = varbind.substring(varbind.indexOf('}')+1);
		}else{
			v.add(new VariableBinding(new OID(varbind)));
			return v;
		}

		VariableBinding vb = new VariableBinding(new OID(oid));
		if (value != null) {
			Variable variable;
			switch (type) {
				case 'i':
					variable = new Integer32(Integer.parseInt(value));
					break;
				case 'u':
					variable = new UnsignedInteger32(Long.parseLong(value));
					break;
				case 's':
					variable = new OctetString(value);
					break;
				case 'x':
					variable = OctetString.fromString(value, ':', 16);
					break;
				case 'd':
					variable = OctetString.fromString(value, '.', 10);
					break;
				case 'b':
					variable = OctetString.fromString(value, ' ', 2);
					break;
				case 'n':
					variable = new Null();
					break;
				case 'o':
					variable = new OID(value);
					break;
				case 't':
					variable = new TimeTicks(Long.parseLong(value));
					break;
				case 'a':
					variable = new IpAddress(value);
					break;
				default:
					throw new IllegalArgumentException("Variable type "+type+
					       " not supported");
			}
			vb.setVariable(variable);
		}
		v.add(vb);
		return v;
	  }


	  private static PDU walk(Snmp snmp, PDU request, Target target)
		throws IOException {
		request.setNonRepeaters(0);
		OID rootOID = request.get(0).getOid();
		PDU response = null;
		int objects = 0;
		int requests = 0;
		long startTime = System.currentTimeMillis();
		do {
			requests++;
			ResponseEvent responseEvent = _snmp.send(request, target);
			response = responseEvent.getResponse();
			if (response != null) {
				objects += response.size();
			}
		}
		while (!processWalk(response, request, rootOID));

		System.out.println();
		System.out.println("Total requests sent:    "+requests);
		System.out.println("Total objects received: "+objects);
		System.out.println("Total walk time:        "+
			       (System.currentTimeMillis()-startTime)+" milliseconds");
		return response;
	  }

	  private static boolean processWalk(PDU response, PDU request, OID rootOID) {
		if ((response == null) || (response.getErrorStatus() != 0)
			|| (response.getType() == PDU.REPORT)) {
			return true;
		}
		boolean finished = false;
		OID lastOID = request.get(0).getOid();
		for (int i=0; (!finished) && (i<response.size()); i++) {
			VariableBinding vb = response.get(i);
			if ((vb.getOid() == null) || (vb.getOid().size() < rootOID.size())
				|| (rootOID.leftMostCompare(rootOID.size(), vb.getOid()) != 0)) {
				finished = true;
			} else if (Null.isExceptionSyntax(vb.getVariable().getSyntax())) {
				System.out.println(vb.toString());
				finished = true;
			} else if (vb.getOid().compareTo(lastOID) <= 0) {
				System.out.println("Variable received is not lexicographic successor of requested one:");
				System.out.println(vb.toString() + " <= "+lastOID);
				finished = true;
			} else {
				System.out.println(vb.toString());
				lastOID = vb.getOid();
			}
		}
		if (response.size() == 0) {
			finished = true;
		}
		if (!finished) {
			VariableBinding next = response.get(response.size()-1);
			next.setVariable(new Null());
			request.set(0, next);
			request.setRequestID(new Integer32(0));
		}
		return finished;
	}

	public void initReceiver(String host) {
		Address address = new UdpAddress(host + "/" + _port);
		try {
			_transport = new DefaultUdpTransportMapping((UdpAddress) address);
		} catch(IOException ioex) {
			System.out.println("Unable to bind to local IP and port: " + ioex);
			System.exit(-1);
		}

		_threadPool = ThreadPool.create(this.getClass().getName(), _numThreads);

		MessageDispatcher mtDispatcher =
			new MultiThreadedMessageDispatcher(_threadPool, new MessageDispatcherImpl());

		// add message processing models
		mtDispatcher.addMessageProcessingModel(new MPv1());
		mtDispatcher.addMessageProcessingModel(new MPv2c());

		// add all security protocols
		SecurityProtocols.getInstance().addDefaultProtocols();

		_snmp = new Snmp(mtDispatcher, _transport);
		if(_snmp != null){
			_snmp.addCommandResponder(this);
		} else {
			System.out.println("Unable to create Target object");
			System.exit(-1);
		}

		if(_version == SnmpConstants.version3) {
			mtDispatcher.addMessageProcessingModel(new MPv3());

			MPv3 mpv3 =
				(MPv3)_snmp.getMessageProcessingModel(MessageProcessingModel.MPv3);

			USM usm = new USM(SecurityProtocols.getInstance(),
				new OctetString(mpv3.createLocalEngineID()), 0);

			SecurityModels.getInstance().addSecurityModel(usm);

			if (_authoritativeEngineID != null) {
				_snmp.setLocalEngine(_authoritativeEngineID.getValue(), 0, 0);
			}

			this.addUsmUser(_snmp);
		}
	}

	public synchronized void listen() {
		try {
			_transport.listen();
		} catch(IOException ioex) {
			System.out.println("Unable to listen: " + ioex);
			System.exit(-1);
		}

		System.out.println("Waiting for traps..");
		try {
			this.wait();//Wait for traps to come in
		} catch (InterruptedException ex) {
			System.out.println("Interrupted while waiting for traps: " + ex);
			System.exit(-1);
		}
	}

	public synchronized void processPdu(CommandResponderEvent e) {
		PDU command = e.getPDU();
		if (command != null) {
			System.out.println(command.toString());
			if ((command.getType() != PDU.TRAP) &&
				(command.getType() != PDU.V1TRAP) &&
				(command.getType() != PDU.REPORT) &&
				(command.getType() != PDU.RESPONSE)) {
				command.setErrorIndex(0);
				command.setErrorStatus(0);
				command.setType(PDU.RESPONSE);
				StatusInformation statusInformation = new StatusInformation();
				StateReference ref = e.getStateReference();
				try {
					e.getMessageDispatcher().returnResponsePdu(e.
						getMessageProcessingModel(),
						e.getSecurityModel(),
						e.getSecurityName(),
						e.getSecurityLevel(),
						command,
						e.getMaxSizeResponsePDU(),
						ref,
						statusInformation);
				}catch (MessageException ex) {
					System.err.println("Error while sending response: "+ex.getMessage());
				}
			}
		}
	}

	public void sendAndProcessResponse() {
		try {
			PDU response = this.send();
			if ((getPduType() == PDU.TRAP) ||
				(getPduType() == PDU.REPORT) ||
				(getPduType() == PDU.V1TRAP) ||
				(getPduType() == PDU.RESPONSE)) {
				System.out.println(PDU.getTypeString(getPduType()) +
					" sent successfully");
			} else if (response == null) {
				System.out.println("Request timed out.");
			} else if (response.getType() == PDU.REPORT) {
				printReport(response);
			} else if (getOperation() == WALK) {
				System.out.println("End of walked subtree '"+
					((VariableBinding)getVariableBindings().get(0)).getOid()+
					"' reached at:");
				printVariableBindings(response);
			} else {
				System.out.println("Received something strange: requestID=" +
					response.getRequestID() +
					", errorIndex=" +
					response.getErrorIndex() + ", " +
					"errorStatus=" + response.getErrorStatusText()+
					"("+response.getErrorStatus()+")");
				printVariableBindings(response);
			}
		} catch (IOException ex) {
			System.err.println("Error while trying to send request: " +
				ex.getMessage());
				ex.printStackTrace();
		}
	}

	private void checkTrapVariables(Vector vbs) {
		if ((_pduType == PDU.INFORM) || (_pduType == PDU.TRAP)) {
			if ((vbs.size() == 0) ||
				((vbs.size() > 1) &&
				(!((VariableBinding) vbs.get(0)).getOid().equals(SnmpConstants.
				sysUpTime)))) {
				vbs.add(0, new VariableBinding(SnmpConstants.sysUpTime, _sysUpTime));
			}
			if ((vbs.size() == 1) || ((vbs.size() > 2) &&
				(!((VariableBinding) vbs.get(1)).getOid().equals(SnmpConstants.
				snmpTrapOID)))) {
				vbs.add(1, new VariableBinding(SnmpConstants.snmpTrapOID, _trapOID));
			}
		}
	}
}