package com.wgcloud.util;
|
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.snmp4j.*;
|
import org.snmp4j.event.ResponseEvent;
|
import org.snmp4j.mp.SnmpConstants;
|
import org.snmp4j.security.SecurityModel;
|
import org.snmp4j.smi.*;
|
import org.snmp4j.transport.DefaultUdpTransportMapping;
|
|
import java.io.IOException;
|
import java.util.*;
|
|
public class SnmpTestUtil {
|
private static final Logger logger = LoggerFactory.getLogger(SnmpTestUtil.class);
|
private static Snmp snmp = null;
|
private CommunityTarget target;
|
|
//传入要查询的主机ip
|
private String hostIp;
|
//传入要查询的端口号
|
private String hostPort;
|
//传入要查询的团体名
|
private String community ;
|
//接收要查询的oids
|
private String oid;
|
|
public String getHostIp() {
|
return hostIp;
|
}
|
|
public void setHostIp(String hostIp) {
|
this.hostIp = hostIp;
|
}
|
|
public String getHostPort() {
|
return hostPort;
|
}
|
|
public void setHostPort(String hostPort) {
|
this.hostPort = hostPort;
|
}
|
|
public String getCommunity() {
|
return community;
|
}
|
|
public void setCommunity(String community) {
|
this.community = community;
|
}
|
|
private String addressGet ;
|
|
//获取设备描述信息
|
private String healthOID1 = "1.3.6.1.4.1";
|
//设备电源状态
|
String healthOID2 = "1.3.6.1.";
|
//获取系统温度
|
private String healthOID3 = "1.3.6.";
|
|
public SnmpTestUtil(String hostIp, String hostPort, String snmpCommunity, String oid){
|
this.setHostIp(hostIp);
|
this.setHostPort(hostPort);
|
this.setCommunity(snmpCommunity);
|
this.oid = oid;
|
}
|
|
public void initSnmp() throws IOException {
|
snmp = new Snmp(new DefaultUdpTransportMapping());
|
//开启监听
|
snmp.listen();
|
}
|
|
private Target createTarget(String oid) {
|
//先得到连接的ip和端口号
|
this.addressGet = "udp:" + this.getHostIp()+"/"+this.getHostPort();
|
|
Target target = null;
|
int version = 1;
|
if (!(version == SnmpConstants.version3 || version == SnmpConstants.version2c || version == SnmpConstants.version1)) {
|
//log.error("参数version异常");
|
return target;
|
}
|
// if (version == SnmpConstants.version3) {
|
// target = new UserTarget();
|
// //snmpV3需要设置安全级别和安全名称，其中安全名称是创建snmp指定user设置的new OctetString("SNMPV3")
|
// target.setSecurityLevel(SecurityLevel.AUTH_PRIV);
|
// target.setSecurityName(new OctetString(this.username));
|
// } else
|
{
|
//snmpV1和snmpV2需要指定团体名名称
|
target = new CommunityTarget();
|
((CommunityTarget) target).setCommunity(new OctetString(this.community));
|
if (version == SnmpConstants.version2c) {
|
target.setSecurityModel(SecurityModel.SECURITY_MODEL_SNMPv2c);
|
}
|
}
|
target.setVersion(version);
|
//必须指定，没有设置就会报错。
|
target.setAddress(GenericAddress.parse(this.addressGet));
|
target.setRetries(3);
|
target.setTimeout(2000);
|
return target;
|
}
|
|
private static PDU createPDU(int version, int type, String oid) {
|
PDU pdu = null;
|
if (version == SnmpConstants.version3) {
|
pdu = new ScopedPDU();
|
} else {
|
pdu = new PDUv1();
|
}
|
pdu.setType(type);
|
//可以添加多个变量oid
|
/*for(String oid:oids){
|
pdu.add(new VariableBinding(new OID(oid)));
|
}*/
|
pdu.add(new VariableBinding(new OID(oid)));
|
return pdu;
|
}
|
|
|
/**
|
* 这个方法有大用处，有时候我们查出的数据是35:32:30:30:30:30:30:30:30:31:31:37:31:30:32类似于这种格式，然后我们需要把它转为用户可识别的格式，就用它
|
* @param octestString
|
* @return
|
*/
|
public static String getChinese(String octestString){
|
try{
|
|
String[] temps = octestString.split(":");
|
byte[] bs = new byte[temps.length];
|
for (int j=0;j< temps.length;j++){
|
bs[j] = (byte)Integer.parseInt(temps[j],16);
|
}
|
return new String(bs,"GB2312");
|
}catch (Exception e){
|
return octestString;
|
}
|
}
|
/**
|
* WALK方式请求
|
*功能简介：这个方法是getnext，就是遍历该oid下的所有子节点
|
*/
|
public List<Map<String,Object>> snmpWalk() {
|
try {
|
|
List<Map<String,Object>> list = new ArrayList<>();
|
|
//1、初始化snmp,并开启监听
|
initSnmp();
|
//2、创建目标对象
|
Target target = createTarget(this.oid);
|
//3、创建报文
|
PDU pdu = createPDU(2, PDU.GETNEXT, this.oid);
|
//4、发送报文，并获取返回结果
|
boolean matched = true;
|
while (matched) {
|
ResponseEvent responseEvent = snmp.send(pdu, target);
|
if (responseEvent == null || responseEvent.getResponse() == null) {
|
//LogUtil.info("snmp TimeOut...");
|
break;
|
}
|
PDU response = responseEvent.getResponse();
|
String nextOid = null;
|
Vector<? extends VariableBinding> variableBindings = response.getVariableBindings();
|
for (int i = 0; i < variableBindings.size(); i++) {
|
|
VariableBinding variableBinding = variableBindings.elementAt(i);
|
Variable variable = variableBinding.getVariable();
|
nextOid = variableBinding.getOid().toDottedString();
|
//如果不是这个节点下的oid则终止遍历，否则会输出很多，直到整个遍历完。
|
if (!nextOid.startsWith(oid)) {
|
matched = false;
|
break;
|
}
|
Map<String,Object> map = new HashMap();
|
//放入oid以及oid对应获得的值
|
map.put(nextOid,variable);
|
list.add(map);
|
}
|
if (!matched) {
|
break;
|
}
|
pdu.clear();
|
pdu.add(new VariableBinding(new OID(nextOid)));
|
}
|
|
System.out.println("list的大小 = " +list.size());
|
return list;
|
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
return null;
|
}
|
|
public static List<Map<String,Object>> snmpWalk2(SnmpTestUtil snmpTestUtil) {
|
try {
|
|
List<Map<String,Object>> list = new ArrayList<>();
|
|
//1、初始化snmp,并开启监听
|
snmp = new Snmp(new DefaultUdpTransportMapping());
|
//开启监听
|
snmp.listen();
|
//2、创建目标对象
|
//先得到连接的ip和端口号
|
String addressGet = "udp:" + snmpTestUtil.getHostIp()+"/"+snmpTestUtil.getHostPort();
|
|
Target target = null;
|
int version = 1;
|
{
|
//snmpV1和snmpV2需要指定团体名名称
|
target = new CommunityTarget();
|
((CommunityTarget) target).setCommunity(new OctetString(snmpTestUtil.getCommunity()));
|
if (version == SnmpConstants.version2c) {
|
target.setSecurityModel(SecurityModel.SECURITY_MODEL_SNMPv2c);
|
}
|
}
|
target.setVersion(version);
|
//必须指定，没有设置就会报错。
|
target.setAddress(GenericAddress.parse(addressGet));
|
target.setRetries(3);
|
target.setTimeout(2000);
|
//3、创建报文
|
PDU pdu = createPDU(2, PDU.GETNEXT,"1.3.6.1.2.1.31.1.1.1.1");
|
//4、发送报文，并获取返回结果
|
boolean matched = true;
|
while (matched) {
|
ResponseEvent responseEvent = snmp.send(pdu, target);
|
if (responseEvent == null || responseEvent.getResponse() == null) {
|
//LogUtil.info("snmp TimeOut...");
|
break;
|
}
|
PDU response = responseEvent.getResponse();
|
String nextOid = null;
|
Vector<? extends VariableBinding> variableBindings = response.getVariableBindings();
|
for (int i = 0; i < variableBindings.size(); i++) {
|
|
VariableBinding variableBinding = variableBindings.elementAt(i);
|
Variable variable = variableBinding.getVariable();
|
nextOid = variableBinding.getOid().toDottedString();
|
//如果不是这个节点下的oid则终止遍历，否则会输出很多，直到整个遍历完。
|
if (!nextOid.startsWith("1.3.6.1.2.1.31.1.1.1.1")) {
|
matched = false;
|
break;
|
}
|
Map<String,Object> map = new HashMap();
|
//放入oid以及oid对应获得的值
|
map.put(nextOid,variable);
|
list.add(map);
|
}
|
if (!matched) {
|
break;
|
}
|
pdu.clear();
|
pdu.add(new VariableBinding(new OID(nextOid)));
|
}
|
|
System.out.println("list的大小 = " +list.size());
|
return list;
|
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
return null;
|
}
|
|
|
/**
|
* 功能简介：获取单个节点oid对应的值
|
* @throws IOException
|
*/
|
public void snmpGet( ) throws IOException {
|
|
|
try {
|
|
//1、初始化snmp,并开启监听
|
initSnmp();
|
//2、创建目标对象
|
Target target = createTarget(this.oid);
|
//3、创建报文
|
PDU pdu = createPDU(2, PDU.GET, this.oid);
|
//4、发送报文，并获取返回结果
|
|
|
ResponseEvent responseEvent = snmp.send(pdu, target);
|
PDU response = responseEvent.getResponse();
|
|
if (response == null) {
|
System.out.println("response is null, request time out");
|
} else {
|
|
System.out.println("response pdu size is " + response.size());
|
for (int i = 0; i < response.size(); i++) {
|
VariableBinding vb = response.get(i);
|
System.out.println(vb.getOid() + " = " + vb.getVariable());
|
}
|
|
}
|
System.out.println("SNMP GET one OID value finished !");
|
} catch (Exception e) {
|
e.printStackTrace();
|
System.out.println("SNMP Get Exception:" + e);
|
} finally {
|
if (snmp != null) {
|
try {
|
snmp.close();
|
} catch (IOException ex1) {
|
snmp = null;
|
}
|
}
|
|
}
|
}
|
|
public static void main(String[] args) {
|
SnmpTestUtil snmpTestUtil = new SnmpTestUtil("192.168.8.150","161","public","1.3.6.1.2.1.31.1.1.1.1");
|
List<Map<String, Object>> mapList = SnmpTestUtil.snmpWalk2(snmpTestUtil);
|
Map<String,Integer> result = new HashMap<String,Integer>();
|
int flat= 0;
|
List<String> resultNeed = new ArrayList<>();
|
for (Map<String, Object> map: mapList){
|
for (String key:map.keySet()){
|
System.out.println("key= "+key +" value= "+ map.get(key));
|
//华为
|
if (String.valueOf(map.get(key)).equals("9")){
|
String substring = key.substring(25, key.length());
|
resultNeed.add(substring);
|
System.out.println("找到我需要的数据了"+substring+"9");
|
break;
|
}
|
|
}
|
}
|
System.out.println("这是我需要的数据"+resultNeed.get(0));
|
|
}
|
}
|
