package com.sitech.ismp.coll;

import java.io.IOException;
import java.net.UnknownHostException;
import java.util.Date;
import java.util.Vector;

import org.apache.log4j.Logger;
import org.snmp4j.CommunityTarget;
import org.snmp4j.PDU;
import org.snmp4j.Snmp;
import org.snmp4j.event.ResponseEvent;
import org.snmp4j.smi.Address;
import org.snmp4j.smi.GenericAddress;
import org.snmp4j.smi.OID;
import org.snmp4j.smi.OctetString;
import org.snmp4j.smi.UdpAddress;
import org.snmp4j.smi.VariableBinding;

import com.sitech.ismp.coll.basic.TblATO_KPIDETAIL;
import com.sitech.ismp.coll.net.ifOperStatusHelper;
import com.sitech.ismp.coll.net.ifTypeHelper;
import com.sitech.ismp.coll.net.cisco.PhysicalEntry;
import com.sitech.ismp.coll.net.cisco.PhysicalEntryWalker;
import com.sitech.ismp.coll.net.config.IpMacEntry;
import com.sitech.ismp.coll.net.config.IpMacEntryWalker;
import com.sitech.ismp.coll.net.table.IfTable;
import com.sitech.ismp.coll.net.table.IfTableRow;
import com.sitech.util.Formater;

public class CollCiscoRouterWithSNMP {
	private Logger logger = Logger.getLogger("COLL");
	private String PRE_UNITID = "10-17-10"; // Cisco Router
	private String HOST_IP = "";
	private int PORT = 161;
	private String COMMUNITY = "public";
	private String HOST_NAME = "";	
	private String VERSION = "";
	private String NEAT_HOST_NAME = "";
	private Snmp mSnmp = null;
	private CommunityTarget mCommunityTarget = null;

	public CollCiscoRouterWithSNMP(String ip, int port, String community,
			String host_name, String version) {
		this.HOST_IP = ip;
		this.PORT = port;
		this.COMMUNITY = community;
		this.HOST_NAME = host_name;
		this.NEAT_HOST_NAME = com.sitech.util.Formater.neatenunitid(host_name);
		this.VERSION = version;
	}

	/**
	 * 初始化SNMP target
	 * 
	 */
	public void init() {
		try {
			String addr = "udp:" + this.HOST_IP + "/"
					+ String.valueOf(this.PORT);
			OctetString community = new OctetString(this.COMMUNITY);
			Address address = GenericAddress.parse(addr);
			mCommunityTarget = new CommunityTarget(address, community);
			org.snmp4j.TransportMapping vTransport = new org.snmp4j.transport.DefaultUdpTransportMapping();
			mSnmp = new org.snmp4j.Snmp(vTransport);
			vTransport.listen();
			mCommunityTarget.setCommunity(community);
			mCommunityTarget.setAddress(new UdpAddress(java.net.InetAddress
					.getByName(this.HOST_IP), this.PORT));
			mCommunityTarget.setRetries(2);
			mCommunityTarget.setTimeout(200);
			// mCommunityTarget.setVersion(org.snmp4j.mp.SnmpConstants.version2c);
		} catch (UnknownHostException e) {
			logger.error("UnknownHostError", e);
			this.release();
		} catch (IOException e) {
			logger.error("IOError", e);
			this.release();
		}
	}// end init

	/**
	 * 释放SNMP连接
	 * 
	 */
	public void release() {
		try {
			if (mSnmp != null) {
				mSnmp.close();
			}
		} catch (IOException e) {
			mSnmp = null;
		}
	}// end release

	/**
	 * 采集主机名称
	 * 
	 * @return
	 */
	public TblATO_KPIDETAIL getHostName() {
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		return this.createKPI(UNIT_ID, "CM-00-01-001-01", this.HOST_NAME);
	}

	/**
	 * 采集IpAddr
	 * 
	 * @return
	 */
	public TblATO_KPIDETAIL getIpAddr() {
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		return this.createKPI(UNIT_ID, "CM-00-02-001-05", this.HOST_IP);
	}

	/**
	 * 采集主机状态
	 * 
	 * @return
	 */
	public TblATO_KPIDETAIL getDeviceState() {
		String oid = ".1.3.6.1.2.1.1.2.0";
		String temp = this.getKPIInfo(oid);
		String status = "";
		if (temp == null || temp.equals("")) {
			status = "DOWN";
		} else {
			status = "UP";
		}

		// FM-01-01-01-01
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		return this.createKPI(UNIT_ID, "FM-00-02-001-01", status);
	}

	/**
	 * 采集端口数量
	 * 
	 * @return
	 */
	public TblATO_KPIDETAIL getPortNum() {
		// 1.3.6.1.2.1.2.1.0
		String oid = "1.3.6.1.2.1.2.1.0";
		String temp = "unknown";
		try {
			temp = this.getKPIInfo(oid);
		} catch (RuntimeException e) {
			temp = "unknown";
		}
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		return this.createKPI(UNIT_ID, "CM-00-02-002-06", temp);
	}

	/**
	 * 采集主机软件版本
	 * 
	 * @return
	 */
	public TblATO_KPIDETAIL getSoftVersion() {
		String temp = this.getSV();
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		return this.createKPI(UNIT_ID, "CM-00-02-001-03", temp);
	}

	/**
	 * 采集主机软件版本
	 * 
	 * @return
	 */
	private String getSV() {
		String oid = "1.3.6.1.2.1.1.1.0";
		String temp = "--";
		try {
			temp = this.getKPIInfo(oid);
			temp = temp.substring(temp.indexOf("IOS"));
			String[] ss = temp.split(",");
			temp = ss[0].trim() + "," + ss[1];
		} catch (RuntimeException e) {
			return temp;
		}
		return temp;
	}

	/**
	 * 端口指标采集
	 */
	public Vector getPortKpi() {
		Vector kpiVector = new Vector();
		IfTable ifTable = new IfTable();
		ifTable.FillTable(HOST_IP, PORT, COMMUNITY);
		Vector ifTableRows = ifTable.getRrows();
		for (int i = 0; i < ifTableRows.size(); i++) {
			IfTableRow row = (IfTableRow) ifTableRows.get(i);
			String ifIndex = row.getIfindex();
			String r_ifDesc = "--";
			String ifDesc = row.getIfdesc();
			if (ifDesc != null && (!(ifDesc.equals("")))
					&& (!(ifDesc.equalsIgnoreCase("null")))) {
				r_ifDesc = ifDesc;
			}
			String ifType = ifTypeHelper.getTypeDesc(row.getIftype());
			String ifOperStatus = ifOperStatusHelper.getOperStatusDesc(row
					.getIfoperstatus());
			String r_ifPhyAddr = "--";
			String ifPhyAddr = row.getIfphyaddr();
			if (ifPhyAddr != null && (!(ifPhyAddr.equals("")))) {
				r_ifPhyAddr = ifPhyAddr;
			}

			String UNIT_ID = PRE_UNITID + "-11" + ":" + NEAT_HOST_NAME + "-"
					+ ifIndex + "-interface";
			// 端口索引号 CM-01-02-02-01
			kpiVector.add(createKPI(UNIT_ID, "CM-01-02-02-01", ifIndex));
			// 端口描述 CM-01-02-02-05
			kpiVector.add(createKPI(UNIT_ID, "CM-01-02-02-05", r_ifDesc));
			// 端口类型 CM-01-02-02-02
			kpiVector.add(createKPI(UNIT_ID, "CM-01-02-02-02", ifType));
			// 端口Mac地址 CM-01-02-02-04
			kpiVector.add(createKPI(UNIT_ID, "CM-01-02-02-04", r_ifPhyAddr));
			// 端口状态 FM-01-02-01-02
			kpiVector.add(createKPI(UNIT_ID, "FM-01-02-01-02", ifOperStatus));

			String ifspeed = row.getIfspeed();
			// 端口设置速率 CM-00-02-002-03
			kpiVector.add(createKPI(UNIT_ID, "CM-00-02-002-03", ifspeed));
		}
		return kpiVector;
	}
	
     /**
      * CM-00-13-001-14	10-13	实体详细分类
        CM-00-13-001-15	10-13	实体索引
		CM-00-13-001-16	10-13	实体描述
		CM-00-13-001-17	10-13	类型oid
		CM-00-13-001-18	10-13	实体类别
		CM-00-13-001-19	10-13	实体名
	  */
	
	public Vector getPhysical(){
		Vector kpiVector = new Vector();
		PhysicalEntryWalker walker = new PhysicalEntryWalker();
		walker.walker(HOST_IP, PORT, COMMUNITY);
		Vector physicalVector = walker.getEntry();
		for (int i = 0; i < physicalVector.size(); i++) {
			PhysicalEntry row = (PhysicalEntry) physicalVector.get(i);
            String UNIT_ID = PRE_UNITID + "-19" + ":" + NEAT_HOST_NAME+"-"+Formater.neatenunitid(row.getRowIndex());
            kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-15", row.getRowIndex()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-16", row.getPhysicalDescr()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-17", row.getPhysicalVendorType()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-18", row.getPhysicalClass()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-19", row.getPhysicalName()));
		}
		return kpiVector;
	}
	
	/**
	 * 
	 * 
	 * CM-00-13-001-11 10-13 接口对应IP
	   CM-00-13-001-12 10-13 物理地址
	   CM-00-13-001-13 10-13 转换类型
	*/
	public Vector getIpNetToMedia(){
		Vector kpiVector = new Vector();
		IpMacEntryWalker walker = new IpMacEntryWalker();
		walker.walker(HOST_IP, PORT, COMMUNITY);
		Vector ipVector = walker.getEntry();
		for (int i = 0; i < ipVector.size(); i++) {
			IpMacEntry row = (IpMacEntry) ipVector.get(i);
            String UNIT_ID = PRE_UNITID + "-18" + ":" + NEAT_HOST_NAME+"-"+row.getIpNetToMediaifIndex();
            kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-11", row.getIpNetToMediaNetAddress()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-12", row.getIpNetToMediaPhyAddress()));
			kpiVector.add(createKPI(UNIT_ID, "CM-00-13-001-13", row.getIpNetToMediaType()));
		}
		return kpiVector;
	}

	/**
	 * 生成TblATO_KPIDETAIL
	 * 
	 * @param KBP_ID
	 * @param KPI_ID
	 * @param KPI_VALUE
	 * @return
	 */
	private TblATO_KPIDETAIL createKPI(String KBP_ID, String KPI_ID,
			String KPI_VALUE) {
		TblATO_KPIDETAIL tblato_kpidetail = new TblATO_KPIDETAIL();
		tblato_kpidetail.UNIT_ID = KBP_ID;
		tblato_kpidetail.KPI_ID = KPI_ID;
		tblato_kpidetail.KPI_VALUE = KPI_VALUE;
		tblato_kpidetail.CLL_TIME = new Date();
		return tblato_kpidetail;
	}

	/**
	 * 得到单个oid对应值
	 * 
	 * @param oid
	 * @return
	 */
	public String getKPIInfo(String oid) {
		String result = "";
		this.init();
		PDU pdu = new PDU();
		VariableBinding var = new VariableBinding(new OID(oid),
				new org.snmp4j.smi.OctetString());
		pdu.add(var);
		pdu.setType(PDU.GET);
		try {
			ResponseEvent res = mSnmp.send(pdu, mCommunityTarget);
			PDU result_pdu = res.getResponse();
			if (result_pdu != null) {
				VariableBinding vb = result_pdu.get(0);
				result = vb.getVariable().toString();
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
		this.release();
		return result;
	}
	/**
	 * PM-00-02-001-07 设备内存利用率 
	 * PM-00-02-001-06 设备CPU利用率
	 */
	public Vector getSystemPerformance() {
		Vector kpiVector = new Vector();
		String UNIT_ID = PRE_UNITID + "-10" + ":" + NEAT_HOST_NAME + "-total";
		String cup_use_rate = this.getKPIInfo(".1.3.6.1.4.1.9.2.1.57.0");

		kpiVector.add(createKPI(UNIT_ID, "PM-00-02-001-06", cup_use_rate));
		SNMPTarget snmptarget = null;
		try {
			snmptarget = new SNMPTarget();
			snmptarget.init(HOST_IP, this.PORT, this.COMMUNITY);
			Vector memusedpool = snmptarget.walkKPIInfo(".1.3.6.1.4.1.9.9.48.1.1.1.5");
			Vector memfreepool = snmptarget.walkKPIInfo(".1.3.6.1.4.1.9.9.48.1.1.1.6");
			int usedmem = snmptarget.sumKPI(memusedpool);
			int freemem = snmptarget.sumKPI(memfreepool);
			int mem = usedmem + freemem;
			float memper = usedmem * 100F / mem;
			kpiVector.add(createKPI(UNIT_ID, "PM-00-02-001-07", String.valueOf((int) memper)));
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			snmptarget.release();
		}
		return kpiVector;
	}
	
	public static void main(String[] args) {
		CollCiscoRouterWithSNMP cisco = new CollCiscoRouterWithSNMP("10.208.192.67", 161, "sx_boss41", "AAA", "2");
		cisco.init();
		System.out.println(cisco.getSV());
		cisco.release();
	}
}