snmp4j SnmpAgent SNMPManager示例

古城微笑少年丶 2023-10-02 13:27 52阅读 0赞

import java.io.IOException;
    import org.snmp4j.CommandResponder;
    import org.snmp4j.CommandResponderEvent;
    import org.snmp4j.CommunityTarget;
    import org.snmp4j.MessageDispatcher;
    import org.snmp4j.MessageDispatcherImpl;
    import org.snmp4j.MessageException;
    import org.snmp4j.PDU;
    import org.snmp4j.Snmp;
    import org.snmp4j.log.LogFactory;
    import org.snmp4j.mp.MPv1;
    import org.snmp4j.mp.MPv2c;
    import org.snmp4j.mp.StateReference;
    import org.snmp4j.mp.StatusInformation;
    import org.snmp4j.security.Priv3DES;
    import org.snmp4j.security.SecurityProtocols;
    import org.snmp4j.smi.Integer32;
    import org.snmp4j.smi.OID;
    import org.snmp4j.smi.OctetString;
    import org.snmp4j.smi.TcpAddress;
    import org.snmp4j.smi.TransportIpAddress;
    import org.snmp4j.smi.UdpAddress;
    import org.snmp4j.smi.VariableBinding;
    import org.snmp4j.tools.console.SnmpRequest;
    import org.snmp4j.transport.AbstractTransportMapping;
    import org.snmp4j.transport.DefaultTcpTransportMapping;
    import org.snmp4j.transport.DefaultUdpTransportMapping;
    import org.snmp4j.util.MultiThreadedMessageDispatcher;
    import org.snmp4j.util.ThreadPool;
    
    public class SnmpAgent implements CommandResponder {
    
    	public static void main(String[] args) {
    		SnmpAgent snmp4jTrapReceiver = new SnmpAgent();
    		try {
    			snmp4jTrapReceiver.listen(new UdpAddress("localhost/8161"));
    		} catch (IOException e) {
    			System.err.println("Error in Listening for Trap");
    			System.err.println("Exception Message = " + e.getMessage());
    		}
    	}
    
    	public synchronized void listen(TransportIpAddress address) throws IOException {
    		AbstractTransportMapping transport;
    		if (address instanceof TcpAddress) {
    			transport = new DefaultTcpTransportMapping((TcpAddress) address);
    		} else {
    			transport = new DefaultUdpTransportMapping((UdpAddress) address);
    		}
    
    		ThreadPool threadPool = ThreadPool.create("DispatcherPool", 10);
    		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();
    		SecurityProtocols.getInstance().addPrivacyProtocol(new Priv3DES());
    
    		// Create Target
    		CommunityTarget target = new CommunityTarget();
    		target.setCommunity(new OctetString("public"));
    
    		Snmp snmp = new Snmp(mtDispatcher, transport);
    		snmp.addCommandResponder(this);
    
    		transport.listen();
    		System.out.println("SNMP Agent Listening on " + address);
    
    		try {
    			this.wait();
    		} catch (InterruptedException ex) {
    			Thread.currentThread().interrupt();
    		}
    	}
    
    	/**
    	 * This method will be called whenever a pdu is received on the given port
    	 * specified in the listen() method
    	 */
    	public synchronized void processPdu(CommandResponderEvent cmdRespEvent) {
    		PDU pdu = cmdRespEvent.getPDU();
    		if (pdu != null) {
    
    			int pduType = pdu.getType();
    			if ((pduType != PDU.TRAP) && (pduType != PDU.V1TRAP) && (pduType != PDU.REPORT)
    					&& (pduType != PDU.RESPONSE)) {
    				pdu.setErrorIndex(0);
    				pdu.setErrorStatus(0);
    
    				// forming response for requested OID request
    				if (pdu.get(0).getOid().equals(new OID(".1.3.6.1.4.1.53427.1.4.2"))) {
    					System.out.println("requested OID is  " + pdu.get(0).getOid());
    					pdu.set(0, new VariableBinding(new OID(pdu.get(0).getOid()), new Integer32(100)));
    					// pdu.add(new VariableBinding(new OID(pdu.get(0).getOid()), new
    					// Integer32(100)));
    				}
    
    				else if (pdu.get(0).getOid().equals(new OID(".1.3.6.1.4.1.53427.1.4.3")))
    
    				{
    					System.out.println("requested OID is  " + pdu.get(0).getOid());
    					pdu.set(0, new VariableBinding(new OID(pdu.get(0).getOid()), new Integer32(200)));
    					// pdu.add(new VariableBinding(new OID(pdu.get(0).getOid()), new
    					// Integer32(200)));
    				} else {
    					pdu.clear();
    				}
    
    				pdu.setType(PDU.RESPONSE);
    
    				StatusInformation statusInformation = new StatusInformation();
    				StateReference ref = cmdRespEvent.getStateReference();
    				try {
    					System.out.println("response sent please check your SNMP manager");
    					System.out.println(cmdRespEvent.getPDU());
    					cmdRespEvent.getMessageDispatcher().returnResponsePdu(cmdRespEvent.getMessageProcessingModel(),
    							cmdRespEvent.getSecurityModel(), cmdRespEvent.getSecurityName(),
    							cmdRespEvent.getSecurityLevel(), pdu, cmdRespEvent.getMaxSizeResponsePDU(), ref,
    							statusInformation);
    				} catch (MessageException ex) {
    					System.err.println("Error while sending response: " + ex.getMessage());
    					LogFactory.getLogger(SnmpRequest.class).error(ex);
    				}
    			}
    		}
    	}
    }

import java.io.IOException;
    
    import org.snmp4j.CommunityTarget;
    import org.snmp4j.PDU;
    import org.snmp4j.Snmp;
    import org.snmp4j.Target;
    import org.snmp4j.TransportMapping;
    import org.snmp4j.event.ResponseEvent;
    import org.snmp4j.mp.SnmpConstants;
    import org.snmp4j.smi.Address;
    import org.snmp4j.smi.GenericAddress;
    import org.snmp4j.smi.OID;
    import org.snmp4j.smi.OctetString;
    import org.snmp4j.smi.VariableBinding;
    import org.snmp4j.transport.DefaultUdpTransportMapping;
    
    public class SNMPManager {
    
    	Snmp snmp = null;
    	String Agent = "udp:localhost/8161";
    	String Oid=".1.3.6.1.4.1.53427.1.4.3";
    
    	public static void main(String[] args) throws IOException {
    
    		SNMPManager client = new SNMPManager();
    		client.start();
    		String ODIValue = client.getAsString(new OID(client.Oid));
    		System.out.println("OID is "+ client.Oid);
    		System.out.println("ODIValue is "+ODIValue);
    		System.out.println("---------------------------------------------");
    	}
    
    	private void start() throws IOException {
    		TransportMapping transport = new DefaultUdpTransportMapping();
    		snmp = new Snmp(transport);
    		transport.listen();
    	}
    
    
    	public String getAsString(OID oid) throws IOException {
    		ResponseEvent event = get(new OID[] { oid });
    		return event.getResponse().get(0).getVariable().toString();
    	}
    
    	public ResponseEvent get(OID oids[]) throws IOException {
    		PDU pdu = new PDU();
    		for (OID oid : oids) {
    			pdu.add(new VariableBinding(oid));
    		}
    		pdu.setType(PDU.GET);
    		ResponseEvent event = snmp.send(pdu, getTarget(), null);
    		if(event != null) {
    			return event;
    		}
    		throw new RuntimeException("Time out occured");
    	}
    
    	private Target getTarget() {
    		Address targetAddress = GenericAddress.parse(Agent);
    		CommunityTarget target = new CommunityTarget();
    		target.setCommunity(new OctetString("public"));
    		target.setAddress(targetAddress);
    		target.setRetries(2);
    		target.setTimeout(1500);
    		target.setVersion(SnmpConstants.version2c);
    		return target;
    	}
    
    }