论坛首页 → Java企业应用论坛 →

关于用snmpv3开发网管系统

浏览 6032 次
锁定老帖子 主题：关于用snmpv3开发网管系统
精华帖 (0) :: 良好帖 (0) :: 新手帖 (0) :: 隐藏帖 (1)
作者 正文
  • 279135628
  • 等级: 初级会员
  • 性别:
  • 文章: 22
  • 积分: 40
  • 来自: 杭州
   发表时间：2010-10-12  
以下是我用JAVA实现的利用snmpv3写的一段测试代码：
1：JAVA实现的SNMP代理部分 
package com.medou.agent;
import org.snmp4j.*;
import org.snmp4j.mp.*;
import org.snmp4j.security.AuthMD5;
import org.snmp4j.security.AuthSHA;
import org.snmp4j.security.PrivAES128;
import org.snmp4j.security.PrivAES192;
import org.snmp4j.security.PrivAES256;
import org.snmp4j.security.PrivDES;
import org.snmp4j.security.SecurityModels;
import org.snmp4j.security.SecurityProtocols;
import org.snmp4j.security.USM;
import org.snmp4j.security.UsmUser;
import org.snmp4j.smi.*;
import org.snmp4j.transport.DefaultTcpTransportMapping;
import org.snmp4j.transport.DefaultUdpTransportMapping;
import org.snmp4j.util.MultiThreadedMessageDispatcher;
import org.snmp4j.util.ThreadPool;

public class SnmpProcessor implements CommandResponder {
	private MultiThreadedMessageDispatcher dispatcher;
	private Address listenAddress;
	private ThreadPool threadPool;
	private OID authProtocol;
	private OID privProtocol;
	private OctetString privPassphrase;
	private OctetString authPassphrase;
	private OctetString securityName = new OctetString();
	private int numDispatcherThreads = 2;
	private int version;
	private String host;
    
	public SnmpProcessor(String host, String user, 
			  String authProtocol, String authPasshrase, String privProtocol, 
	          String privPassphrase, int version) {
		  this.authPassphrase = new OctetString(authPasshrase);
		  this.securityName = new OctetString(user);
		  this.privPassphrase = new OctetString(privPassphrase);
		  this.version = version;
		  this.host = host;
		  if (authProtocol.equals("MD5")) {
			  this.authProtocol = AuthMD5.ID;
	      } else if (authProtocol.equals("SHA")) {
	    	  this.authProtocol = AuthSHA.ID;
	      }
	      if (privProtocol.equals("DES")) {
	    	  this.privProtocol = PrivDES.ID;
	      } else if ((privProtocol.equals("AES128"))
	              || (privProtocol.equals("AES"))) {
	    	  this.privProtocol = PrivAES128.ID;
	      } else if (privProtocol.equals("AES192")) {
	    	  this.privProtocol = PrivAES192.ID;
	      } else if (privProtocol.equals("AES256")) {
	    	  this.privProtocol = PrivAES256.ID;
	      }
	}
	
	public synchronized void run() {
		try {
            init();
        } catch (Exception ex) {
        	ex.printStackTrace();
        } 
	}

	public synchronized void init() throws Exception {
        try {
        	threadPool = ThreadPool.create("Trap", numDispatcherThreads);
            dispatcher = new MultiThreadedMessageDispatcher(threadPool, new MessageDispatcherImpl());
            listenAddress = GenericAddress.parse(host);
            TransportMapping transport;
            if (listenAddress instanceof UdpAddress) {
              transport = new DefaultUdpTransportMapping((UdpAddress)listenAddress);
            } else {
              transport = new DefaultTcpTransportMapping((TcpAddress)listenAddress);
            }
            dispatcher.addMessageProcessingModel(new MPv1());
            dispatcher.addMessageProcessingModel(new MPv2c());
            dispatcher.addMessageProcessingModel(new MPv3());
            SecurityProtocols.getInstance().addDefaultProtocols();
            Snmp snmp = new Snmp(dispatcher, transport);
            if (version == SnmpConstants.version3) {
        	    USM usm = new USM(SecurityProtocols.getInstance(), new OctetString(MPv3.createLocalEngineID()), 0);
        	    SecurityModels.getInstance().addSecurityModel(usm);
        	    // Add the configured user to the USM
        	    addUsmUser(snmp);
            }
            snmp.addCommandResponder(this);
            transport.listen();
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
	
	private void addUsmUser(Snmp snmp) {
	    snmp.getUSM().addUser(securityName, new UsmUser(securityName,
	                                                    authProtocol,
	                                                    authPassphrase,
	                                                    privProtocol,
	                                                    privPassphrase));
    }
	
	public String getResponse (String cmdType, String oid, String value) {
		return "lkj";
	}
	
@SuppressWarnings("unchecked")
public synchronized void processPdu(CommandResponderEvent aEvent) {
      PDU pdu = aEvent.getPDU();
      if (pdu == null) {
          return;
      } 
      String cmdType = "";
      OID oid = pdu.get(0).getOid();
      String oidStr = oid.toString();
      System.out.println(oidStr);
      String value = null==pdu.get(0).getVariable().toString()?"":pdu.get(0).getVariable().toString();
      System.out.println(value);
      switch (pdu.getType()) {
          case PDU.GET:
              cmdType = "0";
              break;
          case PDU.GETNEXT:
              cmdType = "1";
              break;
          case PDU.SET:
              cmdType = "3";
              break;
          default:
              break;
      }
      StatusInformation statusInformation = new StatusInformation();
      StateReference stateRef = aEvent.getStateReference();
      try {
          pdu.setType(PDU.RESPONSE);
          String response = getResponse(cmdType, oidStr, value);
          Variable rariable = null;
          rariable = new OctetString(response);
          pdu.set(0, new VariableBinding(oid, rariable));
          aEvent.getMessageDispatcher().returnResponsePdu(
              aEvent.getMessageProcessingModel(),
              aEvent.getSecurityModel(), aEvent.getSecurityName(),
              aEvent.getSecurityLevel(), pdu,
              aEvent.getMaxSizeResponsePDU(), stateRef,
              statusInformation);
      } catch (MessageException ex) {
    	  ex.printStackTrace();
      }
  }
	
	public static void main(String[] args) {
		SnmpProcessor a = new SnmpProcessor("udp:192.168.0.118/161", "lkj", "MD5", "mysnmplkj", "DES", "mysnmplkj", SnmpConstants.version3);
		a.run();
	}

}


2：以下是我写的网管与代理交互部分： 
package com.medou.agent;

import java.io.IOException;
import java.util.Vector;
import org.snmp4j.CommunityTarget;
import org.snmp4j.PDU;
import org.snmp4j.ScopedPDU;
import org.snmp4j.Snmp;
import org.snmp4j.Target;
import org.snmp4j.TransportMapping;
import org.snmp4j.UserTarget;
import org.snmp4j.event.ResponseEvent;
import org.snmp4j.mp.MPv3;
import org.snmp4j.mp.SnmpConstants;
import org.snmp4j.security.AuthMD5;
import org.snmp4j.security.AuthSHA;
import org.snmp4j.security.PrivAES128;
import org.snmp4j.security.PrivAES192;
import org.snmp4j.security.PrivAES256;
import org.snmp4j.security.PrivDES;
import org.snmp4j.security.SecurityLevel;
import org.snmp4j.security.SecurityModels;
import org.snmp4j.security.SecurityProtocols;
import org.snmp4j.security.USM;
import org.snmp4j.security.UsmUser;
import org.snmp4j.smi.Address;
import org.snmp4j.smi.GenericAddress;
import org.snmp4j.smi.OID;
import org.snmp4j.smi.OctetString;
import org.snmp4j.smi.TcpAddress;
import org.snmp4j.smi.VariableBinding;
import org.snmp4j.transport.DefaultTcpTransportMapping;
import org.snmp4j.transport.DefaultUdpTransportMapping;

/**
 * 本类用于向管理进程发送GET信息
 *
 * @author 
 *
 */
public class SnmpUtilSendGet {
    private Address targetAddress = null;
    private OID authProtocol;
    private OID privProtocol;
    private OctetString privPassphrase;
    private OctetString authPassphrase;
    private OctetString securityName = new OctetString();
    private int version;
    private String host;
    private Target target;
    private OctetString community = new OctetString("public");
    private TransportMapping transport;
    
    /**
     * 初始化Snmp信息
     * @param host
     * @param user
     * @param authProtocol
     * @param authPasshrase
     * @param privProtocol
     * @param privPassphrase
     * @param version
     */
    public SnmpUtilSendGet(String host, String user, 
  		  String authProtocol, String authPasshrase, String privProtocol, 
            String privPassphrase, int version) {
          this.authPassphrase = new OctetString(authPasshrase);
    	  this.securityName = new OctetString(user);
    	  this.privPassphrase = new OctetString(privPassphrase);
    	  this.version = version;
    	  this.host = host;
    	  if (authProtocol.equals("MD5")) {
    		  this.authProtocol = AuthMD5.ID;
          } else if (authProtocol.equals("SHA")) {
        	  this.authProtocol = AuthSHA.ID;
          }
          if (privProtocol.equals("DES")) {
        	  this.privProtocol = PrivDES.ID;
          } else if ((privProtocol.equals("AES128"))
                  || (privProtocol.equals("AES"))) {
        	  this.privProtocol = PrivAES128.ID;
          } else if (privProtocol.equals("AES192")) {
        	  this.privProtocol = PrivAES192.ID;
          } else if (privProtocol.equals("AES256")) {
        	  this.privProtocol = PrivAES256.ID;
          }
      }
    
    /**
     * 创建snmp
     * @throws IOException
     */
    public Snmp createSnmpSession() throws IOException {
        // 设置管理进程的IP和端口
        targetAddress = GenericAddress.parse(host);
        if (targetAddress instanceof TcpAddress) {
            transport = new DefaultTcpTransportMapping();
        } else {
            transport = new DefaultUdpTransportMapping();
        }
        Snmp snmp = new Snmp(transport);
        if (version == SnmpConstants.version3) {
    	    USM usm = new USM(SecurityProtocols.getInstance(), new OctetString(MPv3.createLocalEngineID()), 0);
    	    SecurityModels.getInstance().addSecurityModel(usm);
    	    // Add the configured user to the USM
    	    addUsmUser(snmp);
        }
        return snmp;
    }
    
    /**
     * 加入user信息（snmp3）
     * @param snmp
     */
    private void addUsmUser(Snmp snmp) {
	    snmp.getUSM().addUser(securityName, new UsmUser(securityName,
	                                                    authProtocol,
	                                                    authPassphrase,
	                                                    privProtocol,
	                                                    privPassphrase));
    }
    
    /**
     * 创建PUD
     * @param target
     * @return
     */
    public PDU createPDU(Target target) {
        PDU request;
        if (target.getVersion() == SnmpConstants.version3) {
            request = new ScopedPDU();
        } else {
            request = new PDU();
        }
        return request;
    }
    
    /**
     * 创建Target
     * @return
     */
    private Target createTarget() {
        if (version == SnmpConstants.version3) {
          UserTarget target = new UserTarget();
          if (authPassphrase != null) {
            if (privPassphrase != null) {
              target.setSecurityLevel(SecurityLevel.AUTH_PRIV);
            } else {
              target.setSecurityLevel(SecurityLevel.AUTH_NOPRIV);
            }
          } else {
            target.setSecurityLevel(SecurityLevel.NOAUTH_NOPRIV);
          }
          target.setSecurityName(securityName);
          return target;
       } else {
          CommunityTarget target = new CommunityTarget();
          target.setCommunity(community);
          return target;
       }
    }
    
    /**
     * 向管理进程发送Trap报文
     *
     * @throws IOException
     */
    @SuppressWarnings("unchecked")
    public void sendPDU() throws IOException {
        // 设置 target
    	Snmp snmp = createSnmpSession();
    	target = createTarget();
        target.setAddress(targetAddress);
        // 通信不成功时的重试次数
        target.setRetries(2);
        // 超时时间
        target.setTimeout(1500);
        // snmp版本
        target.setVersion(version);
        
        transport.listen();
        // 创建 PDU
        PDU pdu = createPDU(target);
//        pdu.add(new VariableBinding(new OID(".1.3.6.1.2.3377.10.1.1.1.2"),
//                new OctetString("JavaEE")));
        pdu.add(new VariableBinding(new OID(".1.3.4")));
        pdu.setType(PDU.GET);
        System.out.println(pdu);
        ResponseEvent respEvnt = snmp.send(pdu, target);
        System.out.println(respEvnt.getResponse());
        // 解析Response
        if (respEvnt != null && respEvnt.getResponse() != null) {
            Vector<VariableBinding> recVBs = respEvnt.getResponse()
            .getVariableBindings();
            for (int i = 0; i < recVBs.size(); i++) {
                VariableBinding recVB = recVBs.elementAt(i);
                System.out.println(recVB.getOid() + " : " + recVB.getVariable());
            }
        }
        snmp.close();
    }

    public static void main(String[] args) {
        try {
            SnmpUtilSendGet util = new SnmpUtilSendGet("udp:192.168.0.118/161", "lkj", "MD5", "mysnmplkj", "DES", "mysnmplkj", SnmpConstants.version3);
            util.createSnmpSession();
            util.sendPDU();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}



为什么我运行了代理后，执行交互部分的代码，代理不会跳入processPdu方法运行？ 我试过用SNMP2V的版本是可以的，有谁知道的请解答下   谢谢
返回顶楼
         
  • allskyloveok
  • 等级: 初级会员
  • 性别:
  • 文章: 16
  • 积分: 0
  • 来自: 深圳
   发表时间：2010-10-12  
这个其实就是一个协议的东西，不过要再整理下发出来，讨论！
返回顶楼
          回帖地址
0 请登录后投票
  • super_zhao0303
  • 等级: 初级会员
  • 文章: 1
  • 积分: 30
   发表时间：2011-09-22   最后修改：2011-09-22
在main方法里最后添上：
synchronized (SnmpProcessor.class)
{
    try
    {
        SnmpProcessor.class.wait();
    }
    catch(InterruptedException ie)
    {
       System.out.println("waiting Request interrupted..");
       System.exit(-1);
    }
}

就可以了.自己调试通过了.

另外,SnmpUtilSendGet 类的sendPdu方法的注释有点小错误，应该是发送Get消息，不是Trap

谢谢搂主分享，最近正在愁这个。
返回顶楼
          回帖地址
0 请登录后投票
论坛首页 → Java企业应用版

跳转论坛:
Global site tag (gtag.js) - Google Analytics