package com.sitech.ismp.coll.net.config;

import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;
import java.util.Vector;

import org.snmp4j.smi.VariableBinding;

import com.sitech.ismp.snmp.CommSnmpGet;

public class IpMacEntryWalker {
	
	public static final String ROOT_OID = "1.3.6.1.2.1.4.22.1.1";
	public static final int ATTRIBCOUNT = 4;
	private String IP;

	Vector ipAddrVecotor = new Vector();

	private void addEntry(IpMacEntry ipAddr) {
		ipAddrVecotor.add(ipAddr);
	}
	public Vector getEntry(){
		return ipAddrVecotor;
	}
	public void walker(String ip, int port, String community){
		this.IP = ip;
		CommSnmpGet snmpGet = new CommSnmpGet(ip, port, community);
		snmpGet.init();
		snmpGet.setRootOid(ROOT_OID);
		try {
			snmpGet.workTable(ROOT_OID);
		} catch (IOException e) {
			// TODO Auto-generated catch block
			System.out.println("[IP: " + IP + " 采集IpNetToMedia失败]");
		}

		HashMap hm = snmpGet.getWalkHashMap();
		Set set = hm.keySet();
		Iterator it = set.iterator();
		while (it.hasNext()) {
			String key = (String) it.next();
			String value = (String) hm.get(key);
			// 填充Index
			IpMacEntry ipAddr = new IpMacEntry();
			ipAddr.setRowIndex(key);
			ipAddr.setIpNetToMediaifIndex(value);
			this.ipAddrVecotor.add(ipAddr);
		}

		// try {
		for (int i = 0; i < ipAddrVecotor.size(); i++) {
			IpMacEntry ipAddr = (IpMacEntry) ipAddrVecotor.elementAt(i);
			String rowIndex = ipAddr.getRowIndex();
			//System.out.println("rowIndex: " + rowIndex);
			Vector oidsVector = getOtherOids(rowIndex, ATTRIBCOUNT);
			Vector resultVector = null;
			
			try {
				resultVector = snmpGet.getPDU(oidsVector);
			} catch (Exception e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}

			if (resultVector != null) {
				for (int j = 0; j < resultVector.size(); j++) {
					VariableBinding vb = (VariableBinding) resultVector.get(j);
					String oid = vb.getOid().toString();
					String var = vb.getVariable().toString();
					 System.out.println(oid + " = " + var);
					if (oid.startsWith("1.3.6.1.2.1.4.22.1.1")) {
						ipAddr.setRowIndex(rowIndex);
					} else if(oid.startsWith("1.3.6.1.2.1.4.22.1.2")){
						ipAddr.setIpNetToMediaPhyAddress(var);
					}else if(oid.startsWith("1.3.6.1.2.1.4.22.1.3")){
						ipAddr.setIpNetToMediaNetAddress(var);
					}else if(oid.startsWith("1.3.6.1.2.1.4.22.1.4")){
						ipAddr.setIpNetToMediaType(var);
					}
				}
			}
		}
		snmpGet.release();
	} 
	
	public Vector getOtherOids(String rowIndex, int attribCount) {
		Vector vector = new Vector();
		for (int i = 2; i <= attribCount; i++) {
			String oid = releaseRootOidLast(ROOT_OID, rowIndex, i);
			vector.add(oid);
		}
		return vector;
	}
	public String releaseRootOidLast(String rootOid, String rowIndex, int int_i) {
		String result = "";
		String rowIndexLast = rowIndex.substring(rootOid.length(), rowIndex
				.length());
		String rootOidStart = rootOid.substring(0, rootOid.length() - 1);
		result = rootOidStart + int_i + rowIndexLast;
		return result;
	}
}