snmp4jtrapd.java.svn-base

snmp hibernate 源码, 类似hibernate的映射.

package org.opengoss.snmphibernate.impl.snmp4j;

import java.io.IOException;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;


import org.opengoss.snmphibernate.api.ISnmpTrapD;
import org.opengoss.snmphibernate.api.ISnmpTrapHandler;
import org.opengoss.snmphibernate.api.ISnmpTrapMsg;
import org.snmp4j.CommandResponder;
import org.snmp4j.CommandResponderEvent;
import org.snmp4j.MessageDispatcherImpl;
import org.snmp4j.PDU;
import org.snmp4j.Snmp;
import org.snmp4j.TransportMapping;
import org.snmp4j.mp.MPv1;
import org.snmp4j.mp.MPv2c;
import org.snmp4j.mp.MPv3;
import org.snmp4j.security.SecurityModels;
import org.snmp4j.security.SecurityProtocols;
import org.snmp4j.security.USM;
import org.snmp4j.smi.Address;
import org.snmp4j.smi.GenericAddress;
import org.snmp4j.smi.OctetString;
import org.snmp4j.smi.TcpAddress;
import org.snmp4j.smi.UdpAddress;
import org.snmp4j.transport.DefaultTcpTransportMapping;
import org.snmp4j.transport.DefaultUdpTransportMapping;
import org.snmp4j.util.MultiThreadedMessageDispatcher;
import org.snmp4j.util.ThreadPool;

public class Snmp4JTrapD implements ISnmpTrapD, CommandResponder {
	
	static Logger logger = Logger.getLogger(Snmp4JTrapD.class.getName());
	
	private String listenAddress = "udp:0.0.0.0/162";
	
	private int poolSize = 3;

	private MultiThreadedMessageDispatcher dispatcher;

	private Snmp snmp = null;

	private ThreadPool threadPool;
	
	private List<ISnmpTrapHandler> handlerList = Collections
		.synchronizedList(new ArrayList<ISnmpTrapHandler>(5));
	
	/**
	 * Address Format: udp:0.0.0.0/162
	 * 
	 * @param address
	 */
	public void setListenAddress(String address) {
		this.listenAddress = address;
	}
	
	public void setPoolSize(int size) {
		this.poolSize = size;
	}

	public void start() throws Exception {
		init();
		snmp.addCommandResponder(this);
	}

	private void init() throws UnknownHostException, IOException {
		//create thread pool
		threadPool = ThreadPool.create("Trap Receiver ThreadPool", poolSize);
		//create dispatcher
		dispatcher = new MultiThreadedMessageDispatcher(threadPool,
				new MessageDispatcherImpl());
		//create transport
		Address address = GenericAddress.parse(this.listenAddress);
		TransportMapping transport = null;
		if (address instanceof UdpAddress) {
			transport = new DefaultUdpTransportMapping((UdpAddress) address);
		} else {
			transport = new DefaultTcpTransportMapping((TcpAddress) address);
		}
		//create snmp
		snmp = new Snmp(dispatcher, transport);
		snmp.getMessageDispatcher().addMessageProcessingModel(new MPv1());
		snmp.getMessageDispatcher().addMessageProcessingModel(new MPv2c());
		snmp.getMessageDispatcher().addMessageProcessingModel(new MPv3());
		USM usm = new USM(
				SecurityProtocols.getInstance(),
				new OctetString(MPv3.createLocalEngineID()), 0);
		SecurityModels.getInstance().addSecurityModel(usm);
		snmp.listen();
	}

	public void stop() throws Exception {
		snmp.close();
		threadPool.cancel();
	}

	public void processPdu(CommandResponderEvent event) {
		try {
			PDU pdu = event.getPDU();
			if( !(pdu.getType() == PDU.TRAP || pdu.getType() == PDU.V1TRAP) ) {
				logger.log(Level.WARNING, "Not a trap.");
				return;
			}
			logger.log(Level.INFO, "A Trap Received!", pdu);
			ISnmpTrapMsg trapMsg = new Snmp4JTrapMsg(event.getPeerAddress(), pdu);
			logger.log(Level.INFO, "TrapMessage Created:" + trapMsg.toMap().toString());
			ISnmpTrapHandler[] handlers = getHandlers();
			for (ISnmpTrapHandler handler : handlers) {
				if(handler.onTrapMsg(trapMsg)) {
					event.setProcessed(true);
					return;
				}
			}
		} catch (Exception e) {
			logger.log(Level.SEVERE, "Error happened when handling trap!", e);
		}
	}

	public void addTrapHandler(ISnmpTrapHandler handler) {
		handlerList.add(handler);
	}

	public void removeTrapHandler(ISnmpTrapHandler handler) {
		handlerList.remove(handler);
	}
	
	ISnmpTrapHandler[] getHandlers() {
		return handlerList.toArray(new ISnmpTrapHandler[handlerList.size()]);
	}

}