package com.sitech.ismp.coll.centercoll.windows;

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 IpAddrEntryWalker {
	// 1.3.6.1.2.1.4.20.1.1
	public static final String ROOT_OID = "1.3.6.1.2.1.4.20.1.1";

	public static final int ATTRIBCOUNT = 5;

	private String IP;

	Vector ipVector = new Vector();

	private void addEntry(IpAddrEntry ip) {
		ipVector.add(ip);
	}

	public Vector getEntry() {
		return ipVector;
	}

	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 + " 采集RunEntry失败]");
		}

		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
			IpAddrEntry ipEntry = new IpAddrEntry();
			ipEntry.setRowIndex(key);
			ipEntry.setIpAddr(value);
			this.addEntry(ipEntry);
		}

		// try {
		for (int i = 0; i < ipVector.size(); i++) {
			IpAddrEntry ipEntry = (IpAddrEntry) ipVector.elementAt(i);
			String rowIndex = ipEntry.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.20.1.2.")) {
						ipEntry.setIfIndex(var);
					} else if (oid.startsWith("1.3.6.1.2.1.4.20.1.3.")) {
						ipEntry.setNetMask(var);
					} else if (oid.startsWith("1.3.6.1.2.1.4.20.1.4.")) {
						ipEntry.setBcastAddr(var);
					} else if (oid.startsWith("1.3.6.1.2.1.4.20.1.5.")) {
						ipEntry.setReasmMaxSize(var);
					}
				}
			}

		}

		snmpGet.release();
		System.out.println("[IP: " + IP + " RunEntry采集完毕]");

	}

	private Vector getOtherOids(String rowIndex, int attribCount) {
		Vector vector = new Vector();
		for (int i = 2; i <= attribCount; i++) {
			String oid = releaseRootOidLast(ROOT_OID, rowIndex, i);
			// System.out.println("rowIndex change oid: " + oid);
			vector.add(oid);
		}
		return vector;
	}

	private 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;
	}
}