memstat.go

package sw

import (
	"errors"
	"log"
	"strconv"
	"time"

	"github.com/ulricqin/gosnmp"
)

func MemUtilization(ip, community string, timeout, retry int) (int, error) {
	defer func() {
		if r := recover(); r != nil {
			log.Println(ip+" Recovered in MemUtilization", r)
		}
	}()
	vendor, err := SysVendor(ip, community, retry, timeout)
	if err != nil {
		return 0, err
	}
	method := "get"
	var oid string

	switch vendor {
	case "Cisco_NX":
		oid = "1.3.6.1.4.1.9.9.305.1.1.2.0"
	case "Cisco", "Cisco_IOS_XE", "Cisco_old":
		memUsedOid := "1.3.6.1.4.1.9.9.48.1.1.1.5.1"
		snmpMemUsed, _ := RunSnmp(ip, community, memUsedOid, method, timeout)

		memFreeOid := "1.3.6.1.4.1.9.9.48.1.1.1.6.1"
		snmpMemFree, _ := RunSnmp(ip, community, memFreeOid, method, timeout)

		if len(snmpMemFree) == 0 || len(snmpMemUsed) == 0 {
			err := errors.New(ip + " No Such Object available on this agent at this OID")
			return 0, err
		} else {
			if snmpMemUsed[0].Value == nil || snmpMemFree[0].Value == nil {
				err := errors.New(ip + " mem value return nil")
				return 0, err
			}
			memUsed := snmpMemUsed[0].Value.(int)
			memFree := snmpMemFree[0].Value.(int)
			if memUsed+memFree != 0 {
				memUtili := float64(memUsed) / float64(memUsed+memFree)
				return int(memUtili * 100), nil
			}
		}
	case "Cisco_IOS_XR":
		return getCisco_IOS_XR_Mem(ip, community, timeout, retry)
	case "Cisco_ASA", "Cisco_ASA_OLD":
		return getCisco_ASA_Mem(ip, community, timeout, retry)
	case "Huawei", "Huawei_V5.70", "Huawei_V5.130":
		oid = "1.3.6.1.4.1.2011.5.25.31.1.1.1.1.7"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Huawei_V3.10", "H3c_V3.10":
		return getOldHuawei_Mem(ip, community, timeout, retry)
	case "Huawei_ME60":
		return getHuawei_Me60_Mem(ip, community, timeout, retry)
	case "H3C", "H3C_V5", "H3C_V7":
		oid = "1.3.6.1.4.1.25506.2.6.1.1.1.1.8"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "H3C_S9500":
		oid = "1.3.6.1.4.1.2011.10.2.6.1.1.1.1.8"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Juniper":
		oid = "1.3.6.1.4.1.2636.3.1.13.1.11"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Ruijie":
		oid = "1.3.6.1.4.1.4881.1.1.10.2.35.1.1.1.3"
		return getRuijiecpumem(ip, community, oid, timeout, retry)
	case "Dell":
		return GetDellMem(ip, community, timeout, retry)
	case "Linux":
		return GetLinuxMem(ip, community, timeout, retry)
	case "A10":
		return GetA10Mem(ip, community, timeout, retry)
	case "Aruba":
		return GetArubaMem(ip, community, timeout, retry)
	case "Cisco_Controller":
		return GetCiscoControllerMem(ip, community, timeout, retry)
	case "Hillstone":
		return GetHillSoneMem(ip, community, timeout, retry)
	default:
		err = errors.New(ip + " Switch Vendor is not defined")
		return 0, err
	}

	var snmpPDUs []gosnmp.SnmpPDU
	for i := 0; i < retry; i++ {
		snmpPDUs, err = RunSnmp(ip, community, oid, method, timeout)
		if len(snmpPDUs) > 0 {
			break
		}
		time.Sleep(100 * time.Millisecond)
	}

	if err == nil {
		for _, pdu := range snmpPDUs {
			return pdu.Value.(int), err
		}
	}

	return 0, err
}
func getCisco_IOS_XR_Mem(ip, community string, timeout, retry int) (int, error) {
	defer func() {
		if r := recover(); r != nil {
			log.Println(ip+" Recovered in MemUtilization", r)
		}
	}()
	cpuindex := "1.3.6.1.4.1.9.9.109.1.1.1.1.2"
	method := "getnext"
	var snmpPDUs []gosnmp.SnmpPDU
	var err error
	var index string
	for i := 0; i < retry; i++ {
		snmpPDUs, err = RunSnmp(ip, community, cpuindex, method, timeout)
		if len(snmpPDUs) > 0 {
			break
		}
		time.Sleep(100 * time.Millisecond)
	}
	index = strconv.Itoa(snmpPDUs[0].Value.(int))
	method = "get"
	memUsedOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.18." + index + ".1"
	snmpMemUsed, _ := RunSnmp(ip, community, memUsedOid, method, timeout)
	memFreeOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.20." + index + ".1"
	snmpMemFree, _ := RunSnmp(ip, community, memFreeOid, method, timeout)
	if len(snmpMemFree) == 0 || len(snmpMemUsed) == 0 {
		err := errors.New(ip + " No Such Object available on this agent at this OID")
		return 0, err
	} else {
		if snmpMemUsed[0].Value == nil || snmpMemFree[0].Value == nil {
			err := errors.New(ip + " mem value return nil")
			return 0, err
		}
		memUsed := snmpMemUsed[0].Value.(uint64)
		memFree := snmpMemFree[0].Value.(uint64)
		if memUsed+memFree != 0 {
			memUtili := float64(memUsed) / float64(memUsed+memFree)
			return int(memUtili * 100), nil
		}
	}
	return 0, err
}

func getOldHuawei_Mem(ip, community string, timeout, retry int) (int, error) {
	defer func() {
		if r := recover(); r != nil {
			log.Println(ip+" Recovered in MemUtilization", r)
		}
	}()
	method := "walk"
	memTotalOid := "1.3.6.1.4.1.2011.6.1.2.1.1.2"
	snmpMemTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)

	memFreeOid := "1.3.6.1.4.1.2011.6.1.2.1.1.3"
	snmpMemFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if len(snmpMemFree) == 0 || len(snmpMemTotal) == 0 {
		err := errors.New(ip + " No Such Object available on this agent at this OID")
		return 0, err
	} else {
		memTotal := snmpMemTotal[0].Value.(int)
		memFree := snmpMemFree[0].Value.(int)
		if memTotal != 0 {
			memUtili := float64(memTotal-memFree) / float64(memTotal)
			return int(memUtili * 100), err
		}
	}
	return 0, err
}

func getCisco_ASA_Mem(ip, community string, timeout, retry int) (int, error) {
	defer func() {
		if r := recover(); r != nil {
			log.Println(ip+" Recovered in MemUtilization", r)
		}
	}()
	method := "getnext"
	memUsedOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.18"
	snmpMemUsed, err := RunSnmp(ip, community, memUsedOid, method, timeout)
	time.Sleep(100 * time.Millisecond)
	memFreeOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.20"
	snmpMemFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if len(snmpMemFree) == 0 || len(snmpMemUsed) == 0 {
		err := errors.New(ip + " No Such Object available on this agent at this OID")
		return 0, err
	} else {
		if snmpMemUsed[0].Value == nil || snmpMemFree[0].Value == nil {
			err := errors.New(ip + " mem value return nil")
			return 0, err
		}
		memUsed := snmpMemUsed[0].Value.(uint64)
		memFree := snmpMemFree[0].Value.(uint64)
		if memUsed+memFree != 0 {
			memUtili := float64(memUsed) / float64(memUsed+memFree)
			return int(memUtili * 100), nil
		}
	}
	return 0, err
}

func getHuawei_Me60_Mem(ip, community string, timeout, retry int) (int, error) {
	memTotalOid := "1.3.6.1.4.1.2011.6.3.5.1.1.2"

	memTotal, _, err := snmp_walk_sum(ip, community, memTotalOid, timeout, retry)

	memFreeOid := "1.3.6.1.4.1.2011.6.3.5.1.1.3"
	memFree, _, err := snmp_walk_sum(ip, community, memFreeOid, timeout, retry)
	if memTotal != 0 && memFree != 0 {
		memUtili := float64(memTotal-memFree) / float64(memTotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}

func GetDellMem(ip, community string, timeout, retry int) (int, error) {
	method := "getnext"
	memTotalOid := "1.3.6.1.4.1.674.10895.5000.2.6132.1.1.1.1.4.2"
	memTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)
	memFreeOid := "1.3.6.1.4.1.674.10895.5000.2.6132.1.1.1.1.4.1"
	memFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &memTotal[0] != nil && &memFree[0] != nil {
		memfree := memFree[0].Value.(int)
		memtotal := memTotal[0].Value.(int)
		memUtili := float64(memtotal-memfree) / float64(memtotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}
func GetLinuxMem(ip, community string, timeout, retry int) (int, error) {
	method := "get"
	memTotalOid := "1.3.6.1.4.1.2021.4.5.0"
	memTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)
	memFreeOid := "1.3.6.1.4.1.2021.4.11.0"
	memFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &memTotal[0] != nil && &memFree[0] != nil {
		memfree := memFree[0].Value.(int)
		memtotal := memTotal[0].Value.(int)
		memUtili := float64(memtotal-memfree) / float64(memtotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}

func GetA10Mem(ip, community string, timeout, retry int) (int, error) {
	method := "getnext"
	memTotalOid := "1.3.6.1.4.1.22610.2.4.1.2.1"
	memTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)
	memUsedOid := "1.3.6.1.4.1.22610.2.4.1.2.2"
	memUsed, err := RunSnmp(ip, community, memUsedOid, method, timeout)
	if &memTotal[0] != nil && &memUsed[0] != nil {
		memused := memUsed[0].Value.(int)
		memtotal := memTotal[0].Value.(int)
		memUtili := float64(memused) / float64(memtotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}

func GetArubaMem(ip, community string, timeout, retry int) (int, error) {
	method := "getnext"
	memFreeOid := "1.3.6.1.4.1.14823.2.2.1.2.1.15.1.4"
	memFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	memUsedOid := "1.3.6.1.4.1.14823.2.2.1.2.1.15.1.3"
	memUsed, err := RunSnmp(ip, community, memUsedOid, method, timeout)
	if &memFree[0] != nil && &memUsed[0] != nil {
		memused := memUsed[0].Value.(int)
		memfree := memFree[0].Value.(int)
		if memused+memfree != 0 {
			memUtili := float64(memused) / float64(memused+memfree)
			return int(memUtili * 100), nil
		}
	}
	return 0, err
}

func GetCiscoControllerMem(ip, community string, timeout, retry int) (int, error) {
	method := "getnext"
	memTotalOid := "1.3.6.1.4.1.14179.1.1.5.2"
	memTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)
	memFreeOid := "1.3.6.1.4.1.14179.1.1.5.3"
	memFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &memTotal[0] != nil && &memFree[0] != nil {
		memfree := memFree[0].Value.(int)
		memtotal := memTotal[0].Value.(int)
		memUtili := float64(memtotal-memfree) / float64(memtotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}

func GetHillSoneMem(ip, community string, timeout, retry int) (int, error) {

	method := "getnext"
	memTotalOid := ".1.3.6.1.4.1.28557.2.2.1.4"
	memTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)
	memFreeOid := ".1.3.6.1.4.1.28557.2.2.1.5"
	memFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &memTotal[0] != nil && &memFree[0] != nil {
		memfree := memFree[0].Value.(int)
		memtotal := memTotal[0].Value.(int)
		memUtili := float64(memtotal-memfree) / float64(memtotal)
		return int(memUtili * 100), nil
	}
	return 0, err

}