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