Java实现Snmp

Java实现Snmp

本文基于Java实现了snmp同步发送消息，及模拟了snmp的代理端(虚拟设备)，对接收的snmp消息进行回复。直接上代码。

snmp客户端代码：

/** * Snmp采集器 * * @author Fan * */
public class SnmpCollector { 
   

	private final short targetPort = 161;
	private final short listenPort = 162;
	private Snmp snmp;
	private String targetIp;

	public SnmpCollector(String targetIp) { 
   
		this.targetIp = targetIp;
		init();
	}

	private void init() { 
   
		try { 
   
			// 使用本地IP和162端口号, 选择UDP发送方式
			this.snmp = new Snmp(new DefaultUdpTransportMapping(new UdpAddress(listenPort)));
			// 添加自定义消息处理
			snmp.addCommandResponder(new TripResponser());
			snmp.listen();
		} catch (IOException e) { 
   
			e.printStackTrace();
		}
	}

	/** * 发送snmp消息 * @param pdu * @return * @throws IOException */
	public ResponseEvent send(PDU pdu) throws IOException { 
   
		CommunityTarget target = new CommunityTarget();
		target.setCommunity(new OctetString("public"));//设置团体名
		target.setVersion(SnmpConstants.version2c);//设置SNMP的版本
		target.setTimeout(TimeUnit.SECONDS.toMillis(2));
		try { 
   
			target.setAddress(new UdpAddress(InetAddress.getByName(targetIp), targetPort));
		} catch (UnknownHostException e1) { 
   
			e1.printStackTrace();
		}
		return this.snmp.send(pdu, target);
	}

	/** * SNMP trip处理类 * @author Fan * */
	private class TripResponser implements CommandResponder { 
   

		@Override
		public void processPdu(CommandResponderEvent event) { 
   
			System.out.println("reveice the trip!!!");
		}

	}
}

snmp代理端(虚拟设备端)代码：

/** * Snmp代理类 * * @author Fan * */
public class SnmpAgent { 
   

	private final short targetPort = 162;// 发送端口号
	private final short listenPort = 161;// 本地监听端口号
	private Snmp snmp;

	public SnmpAgent() { 
   
		init();
	}

	/** * 初始化snmp */
	private void init() { 
   
		UdpAddress udpAddress = null;
		try { 
   
			// 使用本地IP和161端口号
			udpAddress = new UdpAddress(InetAddress.getLocalHost(), listenPort);
		} catch (UnknownHostException e1) { 
   
			e1.printStackTrace();
		}
		try { 
   
			// 选择UDP发送方式
			this.snmp = new Snmp(new DefaultUdpTransportMapping(udpAddress));
			// 设置snmp的消息处理模型 MPv2c代表SNMPv2c（基于社区的SNMPv2）
			snmp.getMessageDispatcher().addMessageProcessingModel(new MPv2c());
			// 添加自定义消息处理
			snmp.addCommandResponder(new ReceiveResponder());
			snmp.listen();
		} catch (IOException e) { 
   
			e.printStackTrace();
		}
	}

	/** * 多线程接收消息 */
	private void init2() { 
   
		ThreadPool pool = ThreadPool.create("snmpReceiver", 1);
		MultiThreadedMessageDispatcher dispatcher = new MultiThreadedMessageDispatcher(pool,
				new MessageDispatcherImpl());
		try { 
   
			DefaultUdpTransportMapping transportMapping = new DefaultUdpTransportMapping(
					new UdpAddress(InetAddress.getLocalHost(), listenPort));
			this.snmp = new Snmp(dispatcher, transportMapping);
			snmp.getMessageDispatcher().addMessageProcessingModel(new MPv2c());
			this.snmp.listen();
			this.snmp.addCommandResponder(new ReceiveResponder());

		} catch (UnknownHostException e) { 
   
			e.printStackTrace();
		} catch (IOException e) { 
   
			e.printStackTrace();
		}
	}

	/** * 自定义消息处理类 * * @author Fan * */
	private class ReceiveResponder implements CommandResponder { 
   

		@Override
		public void processPdu(CommandResponderEvent event) { 
   
			System.out.println("收到对方消息：" + event.toString());
			PDU pdu = event.getPDU();
			if (pdu == null) { 
   
				return;
			}
			if (pdu.getType() == PDU.GET) { 
   
				doGetResponse(pdu.getVariableBindings());
			}

			pdu.setType(PDU.RESPONSE);
			StatusInformation statusInfo = new StatusInformation();
			StateReference reference = event.getStateReference();
			try { 
   
				// 响应snmp消息
				event.getMessageDispatcher().returnResponsePdu(event.getMessageProcessingModel(),
						event.getSecurityModel(), event.getSecurityName(), event.getSecurityLevel(), pdu,
						event.getMaxSizeResponsePDU(), reference, statusInfo);
			} catch (MessageException e) { 
   
				e.printStackTrace();
			}
		}

		/** * 对接收的消息设置返回值 * * @param bindings */
		private void doGetResponse(Vector<? extends VariableBinding> bindings) { 
   
			for (VariableBinding varBin : bindings) { 
   
				System.out.println("接收到的OID是：" + varBin.getOid().toString());
				varBin.setVariable(new Integer32(1));
				System.out.println("给客户端回复的值是：" + varBin.getVariable().toInt());
			}
		}

	}

	public ResponseEvent send(PDU pdu) throws IOException { 
   
		CommunityTarget target = new CommunityTarget();
		target.setCommunity(new OctetString("public"));
		target.setVersion(SnmpConstants.version2c);
		target.setTimeout(TimeUnit.SECONDS.toMillis(2));
		try { 
   
			target.setAddress(new UdpAddress(InetAddress.getLocalHost(), targetPort));
		} catch (UnknownHostException e1) { 
   
			e1.printStackTrace();
		}
		return this.snmp.send(pdu, target);
	}

	private static PDU createTrip() { 
   
		PDU pdu = new PDU();
		pdu.setType(PDU.TRAP);
		VariableBinding vb = new VariableBinding(new OID("1.3.6.1.2.1.3.4.1.1"));
		vb.setVariable(new Integer32(34));
		pdu.add(vb);
		return pdu;
	}

	public static void main(String[] args) throws InterruptedException { 
   
		SnmpAgent agent = new SnmpAgent();
		System.out.println("snmp-agent started...");

		Thread.sleep(5000);

		 try { 
   
		 // trip测试
		 agent.send(createTrip());
		
		 System.out.println("send trip success-------");
		 } catch (IOException e) { 
   
		 e.printStackTrace();
		 }

		/** * 让主线程一直处理等待中，因为DefaultUdpTransportMapping中监听本地(161端口)的线程为守护线程， * 一旦主线程运行完毕，守护线程也随即销毁 如果采用多线程方式启动(如调用 init2方法)，则不需要主线程等待，但会额外开启新的线程。所以，视情况而定 */
		synchronized (agent) { 
   
			agent.wait();
		}
		
	}

}

测试端代码：

/** * snmp测试 * @author Fan * */
public class SnmpTest { 
   

	public static void main(String[] args) { 
   

		SnmpCollector collector = new SnmpCollector("192.168.0.102");
		PDU pdu = new PDU();
		pdu.setType(PDU.GET);
		pdu.add(new VariableBinding(new OID("1.3.6.1.2.1.3.4.1")));
		
		try { 
   
			ResponseEvent event = collector.send(pdu);
			PDU response = event.getResponse();
			if (response == null) { 
   
				System.out.println("未能采集到设备信息");
				return;
			} 
			Vector<? extends VariableBinding> bindings = response.getVariableBindings();
			for (VariableBinding var : bindings) { 
   
				System.out.println("发送的OID是：" + var.getOid().toString());
				System.out.println("采集的值是：" + var.getVariable().toInt());
			}
			
			
		} catch (IOException e) { 
   
			e.printStackTrace();
		}
// while(true) { 
   
// try { 
   
// Thread.sleep(TimeUnit.SECONDS.toMillis(3));
// } catch (InterruptedException e) { 
   
// e.printStackTrace();
// }
// System.out.println("over!");
// }
		
	}

}

先启动SnmpAgent，再启动SnmpTest，测试结果如下：

今天的文章Java实现Snmp分享到此就结束了，感谢您的阅读，如果确实帮到您，您可以动动手指转发给其他人。