1from IfInfo import IfInfo 2from Slaacctl import ShowInterface 3import subprocess 4import sys 5from scapy.all import * 6import unittest 7 8 9rtadv_if = IfInfo(sys.argv[1]) 10slaac_if = IfInfo(sys.argv[2]) 11sock = sys.argv[3] 12 13eth = Ether(src=rtadv_if.mac) 14ip = IPv6(dst="ff02::1", src=rtadv_if.ll) 15ra = ICMPv6ND_RA(prf='Medium (default)', routerlifetime=1800) 16pref = ICMPv6NDOptPrefixInfo(prefixlen=64, prefix='2001:db8:1::', 17 validlifetime=2592000, preferredlifetime=604800, L=1, A=1) 18mtu = ICMPv6NDOptMTU(mtu=1500) 19rdnss = ICMPv6NDOptRDNSS(lifetime=86400, dns=['2001:db8:53::a', 20 '2001:db8:53::b']) 21dnssl = ICMPv6NDOptDNSSL(lifetime=86400, searchlist=['invalid', 'home.invalid']) 22 23p = eth/ip/ra/pref/mtu/rdnss/dnssl 24 25sendp(p, iface=rtadv_if.ifname, verbose=0) 26 27slaac_show_interface = ShowInterface(slaac_if.ifname, sock, debug=0) 28 29 30class TestRouterAdvertisementParsing(unittest.TestCase): 31 def test_number_ras(self): 32 self.assertEqual(len(slaac_show_interface.RAs), 1) 33 34 def test_number_addr_proposals(self): 35 self.assertEqual(len(slaac_show_interface.addr_proposals), 2) 36 37 def test_number_def_router_proposals(self): 38 self.assertEqual(len( 39 slaac_show_interface.def_router_proposals), 1) 40 41 def test_number_rdns_proposals(self): 42 self.assertEqual(len( 43 slaac_show_interface.rdns_proposals), 1) 44 45 def test_number_rdns_proposals_rnds(self): 46 self.assertEqual(len( 47 slaac_show_interface.rdns_proposals[0]['rdns']), 2) 48 49if __name__ == '__main__': 50 suite = unittest.TestLoader().loadTestsFromTestCase( 51 TestRouterAdvertisementParsing) 52 if not unittest.TextTestRunner(verbosity=2).run(suite).wasSuccessful(): 53 print slaac_show_interface 54 sys.exit(1) 55