Python 스위치에서 트래픽 정보 수집

#!/usr/bin/python
 
import os
import re
import sys
import getopt
import time
class snmpwalk:
    def __init__(self, switchIP):
        self.snmpwalk = "/usr/bin/snmpwalk"
        self.appOption = "-v1"
        self.comOption = "-c public"
        self.switchIP = switchIP
        self.snmpwalkRetry = 0
    def getMacToOID(self, mac):
        return reduce(lambda x,y:x+'.'+str(int(y,16)), mac.replace(' ', ':').replace('-', ':').split(':'), '')
    def getSNMPInfo(self, imb):
        cmd = "%s %s %s %s %s" % (self.snmpwalk, self.appOption, self.comOption, self.switchIP, imb)
        #print cmd
        result = os.popen(cmd).read().strip()
        if result.strip() == "":
            for i in range(self.snmpwalkRetry):
                if result.strip() == "":
                    time.sleep(1)
                    result = os.popen(cmd).read().strip()
        #print result
        return result
    def getPort(self, mac):
        port = None
        mac_oid = self.getMacToOID(mac)
        #print "mac_oid=", mac_oid
        imb = "1.3.6.1.2.1.17.4.3.1.2%s" % (mac_oid) # dot1dTpFdbPort
        info = self.getSNMPInfo(imb)
        if info != '' and info.count(" = INTEGER: ") == 1:
            port = info.split(" = INTEGER: ")[-1]
        #print "port=", port
        return port
    def getIfindexFromIP(self, ip):
        ifindex = None
        imb = "1.3.6.1.2.1.4.20.1.2.%s" % (ip) # ipAdEntIfIndex
        info = self.getSNMPInfo(imb)
        if info != '' and info.count(" = INTEGER: ") == 1:
            ifindex = info.split(" = INTEGER: ")[-1]
        if ifindex == None:
            imb = "1.3.6.1.2.1.3.1.1.1" # atIfIndex
            info = self.getSNMPInfo(imb)
            if info:
                regex = re.compile(".*:(?:atIfindex|1\.3\.6\.1\.2\.1\.3\.1\.1\.1)\.(?:[0-9\.]+)\.%s\s=\s(?:INTEGER):(\d+)" % (ip))
                ifindexs = regex.findall(info)
                if len(ifindexs) == 1:
                    ifindex = ifindexs[0]
                elif len(ifindexs) > 1:
                    raise Exception, info
        if ifindex == None:
            imb = "1.3.6.1.2.1.4.35.1.4"  # ipNetToPhysicalPhysAddress
            info = self.getSNMPInfo(imb)
            if info:
                regex = re.compile(".*:(?:ipNetToPhysicalPhysAddress|1\.3\.6\.1\.2\.1\.4\.35\.1\.4)\.([0-9]+)\.ipv4\.%s\s=\s(?:STRING):\s(?:[0-9a-f:]+)?" % (ip))
                ifindexs = regex.findall(info)
                if len(ifindexs) == 1:
                    ifindex = ifindexs[0]
                elif len(ifindexs) > 1:
                    raise Exception, info
        if ifindex == None:
            imb = "1.3.6.1.2.1.4.22.1.2" # ipNetToMediaPhysAddress
            info = self.getSNMPInfo(imb)
            if info:
                regex = re.compile(".*:(?:ipNetToMediaPhysAddress|1\.3\.6\.1\.2\.1\.4\.22\.1\.2)\.([0-9]+)\.ipv4\.%s\s=\s(?:STRING):\s(?:[0-9a-f:]+)?" % (ip))
                ifindexs = regex.findall(info)
                if len(ifindexs) == 1:
                    ifindex = ifindexs[0]
                elif len(ifindexs) >1:
                    raise Exception, info
        return ifindex
    def getIfindexFromMac(self, mac, vendor):
        ifindex = None
        port = self.getPort(mac)
        if port != None:
            imb = "1.3.6.1.2.1.17.1.4.1.2.%s" % (port)  # dot1dBasePortIfIndex
            info = self.getSNMPInfo(imb)
            if info != '' and info.count(" = INTEGER: ") == 1:
                ifindex = info.split(" = INTEGER: ")[-1]
                if vendor == "JUNIFER":
                    imb = "1.3.6.1.2.1.2.2.1.2.%s"  % (ifindex)  # ifDescr
                    info = self.getSNMPInfo(imb)
                    if info.count(" = STRING: ") == 1:
                        logicalPort = info.split(" = STRING: ")[-1]
                        physicalPort = logicalPort.rsplit(".0", 1)[0]
                        imb = "1.3.6.1.2.1.2.2.1.2"
                        info = self.getSNMPInfo(imb)
                        for value in info.split("\n"):
                            temp = value.split(" = STRING: ")
                            port = temp[-1]
                            if port == physicalPort:
                                ifindex = temp[0].split(".")[-1]
                                break
                    else:
                        raise Exception, info
        if ifindex == None:
            mac_oid = self.getMacToOID(mac)
            imb = "mib-2.17.7.1.2.2.1.2.1%s" % (mac_oid)
            info = self.getSNMPInfo(imb)
            if info != '' and info.count(" = INTEGER: ") == 1:
                ifindex = info.split(" = INTEGER: ")[-1]
        if ifindex == None:
            imb = "1.3.6.1.2.1.3.1.1.2" # atPhysAddress
            info = self.getSNMPInfo(imb)
            if info:
                regex_str = "[:\-\s]".join(mac.replace(' ', ':').replace('-', ':').split(':'))
                regex = re.compile(".*:(?:atPhysAddress|1\.3\.6\.1\.2\.1\.3\.1\.1\.2)\.([0-9\.]+)\s=\s(?:Hex-STRING):\s%s" % (regex_str), re.I)
                oids = regex.findall(info)
                if len(oids) == 1:
                    oid = oids[0]
                    imb = "1.3.6.1.2.1.3.1.1.1.%s" % (oid) # atIfIndex
                    info = self.getSNMPInfo(imb)
                    if info != '' and info.count(" = INTEGER: ") == 1:
                        ifindex = info.split(" = INTEGER: ")[-1]
                    else:
                        raise Exception, info
                elif len(oids) > 1:
                    raise Exception, info
        if ifindex == None:
            imb = "1.3.6.1.2.1.4.35.1.4"  # ipNetToPhysicalPhysAddress
            info = self.getSNMPInfo(imb)
            if info:
                regex_str = "[:\-\s]".join(map(lambda x: str("%x" % int(x,16)), mac.replace(' ', ':').replace('-', ':').split(':')))
                regex = re.compile(".*:(?:ipNetToPhysicalPhysAddress|1\.3\.6\.1\.2\.1\.4\.35\.1\.4)\.([0-9]+)\..*\s=\s(?:STRING):\s%s" % (regex_str))
                ifindexs = regex.findall(info)
                if len(ifindexs) == 1:
                    ifindex = ifindexs[0]
                elif len(ifindexs) > 1:
                    raise Exception, info
        if ifindex == None:
            imb = "1.3.6.1.2.1.4.22.1.2" # ipNetToMediaPhysAddress
            info = self.getSNMPInfo(imb)
            if info:
                regex_str = "[:\-\s]".join(map(lambda x: str("%x" % int(x,16)), mac.replace(' ', ':').replace('-', ':').split(':')))
                regex = re.compile(".*:(?:ipNetToMediaPhysAddress|1\.3\.6\.1\.2\.1\.4\.22\.1\.2)\.([0-9]+)\..*\s=\s(?:STRING):\s%s" % (regex_str))
                ifindexs = regex.findall(info)
                if len(ifindexs) == 1:
                    ifindex = ifindexs[0]
                elif len(ifindexs) > 1:
                    raise Exception, info
        return ifindex
    def getMacFromIfindex(self, ifindex):
        mac = None
        imb = "1.3.6.1.2.1.2.2.1.6.%s" % (ifindex)  # ifPhysAddress
        info = self.getSNMPInfo(imb)
        if info != '' and info.count(" = STRING: ") == 1:
            mac = info.split(" = STRING: ")[-1]
        return mac
    def getIfHCInOctets(self, ifindex):
        octets = None
        imb = "1.3.6.1.2.1.31.1.1.1.6.%s" % (ifindex) # ifHCInOctets
        info = self.getSNMPInfo(imb)
        if info != '' and info.cout(" = Counter") == 1:
            octets = info.split(": ")[-1]
        if octets != None:
            octets = int(octets)
        return octets
    def getIfInOctets(self, ifindex):
        octets = None
        imb = "1.3.6.1.2.1.2.2.1.10.%s" % (ifindex) # ifInOctets
        info = self.getSNMPInfo(imb)
        if info != '' and info.cout(" = Counter") == 1:
            octets = info.split(": ")[-1]
        if octets != None:
            octets = int(octets)
        return octets
    def getIfHCOutOctets(self, ifindex):
        octets = None
        imb = "1.3.6.1.2.1.31.1.1.1.10.%s" % (ifindex) # ifHCOutOctets
        info = self.getSNMPInfo(imb)
        if info != '' and info.cout(" = Counter") == 1:
            octets = info.split(": ")[-1]
        if octets != None:
            octets = int(octets)
        return octets
    def getIfOutOctets(self, ifindex):
        octets = None
        imb = "1.3.6.1.2.1.2.2.1.16.%s" % (ifindex) # ifOutOctets
        info = self.getSNMPInfo(imb)
        if info != '' and info.cout(" = Counter") == 1:
            octets = info.split(": ")[-1]
        if octets != None:
            octets = int(octets)
        return octets
    def getIfIndex(self, ifindex):
        imb = "1.3.6.1.2.1.2.2.1.1.%s" % (ifindex) # ifIndex
        info = self.getSNMPInfo(imb)
        if info != '' and info.count(" = INTEGER: ") == 1:
            ifindex = info.split(" = INTEGER: ")[-1]
        return ifindex
    def getSysVendor(self):
        imb = "sysDescr"
        info = str(self.getSNMPInfo(imb)).upper()
        #print "desc=", info
        return self.getModelByVendor(info)
    def getModelByVendor(self, model):
        info = model.upper()
        if info.count("JUNIFER") or info.count("EX2200") or info.count("EX3200") or info.count("EX3300") or info.count("EX4200"):
             return "JUNIFER"
        elif info.count("HITACHI") or info.count("AX3630"):
             return "HITACHI"
        elif info.count("FORCE10") or info.count("S25P"):
             return "FORCE10"
        elif info.count("DASAN") or info.count("V2324G") or info.count("V2624G") or info.count("V2524G") \
             or info.count("V5424G") or info.count("V5524XG") or info.count("V6524G"):
             return "DASAN"
        elif info.count("NETGEAR") or info.count("7212") or info.count("7224") or info.count("7248") or info.count("750T"):
             return "NETGEAR"
        elif info.count("EXTREME"):
             return "EXTREME"
        elif info.count("HUAWEI"):
             return "HUAWEI"
        else:
             return "UNKNOWN"
       
    def __del__(self):
        pass
def main():
    usage = "Usage : %s -s SWITCH_IP [-m SERVER_MAC_ADDRESS |-i SERVER_IP_ADDRESS]" % (sys.argv[0])
    switch_ip = ""
    mac_address = ""
    ip_address = ""
    options, args = getopt.getopt(sys.argv[1:], 'hs:m:i:')
    for op, p in options:
        if op == '-s':
            switch_ip = p
        if op == '-m':
            mac_address = p
        if op == '-i':
            ip_address = p
        if op == '-h':
            print usage
            os._exit(1)
    if switch_ip == "" :
        print usage
        os._exit(1)
    if mac_address == "" and ip_address == "":
        print usage
        os._exit(1)
    sw = snmpwalk(switch_ip)
    if mac_address :
        vendor = sw.getSysVendor()
        ifindex = sw.getIfindexFromMac(mac_address, vendor)
    elif ip_address:
        ifindex = sw.getIfindexFromIP(ip_address)
    #print "Vendor     : %s" % (vendor)
    #print "Server MAC : %s" % (mac_address.upper())
    print "Ifindex    : %s" % (ifindex)
if __name__=="__main__" :
    main()