def set_deafult_gw(self, args):
"""
Set GW for given uplink network
Args:
args: IP address of GW.
Returns:
"""
gw_ip = ip_address(args.ip)
gw_info = UplinkGatewayInfo()
gw_info.update_ip(str(gw_ip))
print("set Default gw IP to %s" % gw_info.get_gw_ip())
def list_all_record(self, args):
"""
Print list DHCP and GW records.
Args:
args: None
Returns: None
"""
for k, v in self.dhcp_client_state.items():
print("mac: %s DHCP state: %s" % (k.replace('_', ':'), str(v)))
gw_info = UplinkGatewayInfo(store.GatewayInfoMap())
print("Default GW: %s" % gw_info.get_gw_ip())
print("GW Mac address: %s" % gw_info.get_gw_mac())
class DefGwTest(unittest.TestCase):
"""
Validate default router setting.
"""
def setUp(self):
self.gw_store = defaultdict(str)
self.dhcp_gw_info = UplinkGatewayInfo(self.gw_store)
def test_gw_ip_for_DHCP(self):
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), None)
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), None)
def test_gw_ip_for_Ip_pool(self):
self.dhcp_gw_info.read_default_gw()
def_gw_cmd = "ip route show |grep default| awk '{print $3}'"
p = subprocess.Popen(
[def_gw_cmd],
stdout=subprocess.PIPE,
shell=True,
)
def_ip = p.stdout.read().decode("utf-8").strip()
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), '')
mac1 = "11:22:33:44:55:66"
self.dhcp_gw_info.update_mac(def_ip, mac1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), mac1)
# updating IP with same address shld keep mac
self.dhcp_gw_info.update_ip(def_ip)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), mac1)
ip1 = "1.2.3.4"
self.dhcp_gw_info.update_ip(ip1)
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), '')
def test_vlan_gw_info(self):
ip1 = "1.2.3.4"
vlan1 = "1"
self.dhcp_gw_info.update_ip(ip1, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
mac1 = "11:22:33:44:55:66"
self.dhcp_gw_info.update_mac(ip1, mac1, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
# updating IP with same address shld keep mac
self.dhcp_gw_info.update_mac(ip1, mac1, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
# any IP update shld invalidate MAC address.
ip2 = "2.2.3.4"
self.dhcp_gw_info.update_ip(ip2, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip2))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), '')
def test_vlan_gw_info_none(self):
ip1 = "1.2.3.4"
vlan1 = "1"
mac1 = "11:22:33:44:55:66"
self.dhcp_gw_info.update_mac(ip1, mac1, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
# check None IP or mac updates
self.dhcp_gw_info.update_ip(None, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
self.dhcp_gw_info.update_mac(ip1, None, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
self.dhcp_gw_info.update_mac(None, mac1, vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
self.dhcp_gw_info.update_mac(ip1, '', vlan1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan1), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan1), mac1)
vlan2 = None
mac2 = "22:22:33:44:55:66"
self.dhcp_gw_info.update_mac(ip1, mac2, vlan2)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(vlan2), str(ip1))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(vlan2), mac2)
def test_vlan_gw_info_list(self):
ip1 = "1.2.3.4"
vlan1 = "1"
gw1 = _get_gw_info(ip1, '', vlan1)
self.dhcp_gw_info.update_ip(ip1, vlan1)
ip2 = "2.2.3.4"
vlan2 = "2"
gw2 = _get_gw_info(ip2, '', vlan2)
self.dhcp_gw_info.update_ip(ip2, vlan2)
ip3 = "3.2.3.4"
gw3 = _get_gw_info(ip3)
self.dhcp_gw_info.update_ip(ip3)
ip4 = "4.2.3.4"
vlan4 = "4"
mac4 = "11:22:33:44:55:66"
gw4 = _get_gw_info(ip4, mac4, vlan4)
self.dhcp_gw_info.update_mac(ip4, mac4, vlan4)
ip5 = "2020::1"
vlan5 = "4"
mac5 = "11:22:33:44:55:66"
gw5 = _get_gw_info(ip5, mac5, vlan5)
self.dhcp_gw_info.update_mac(ip5, mac5, vlan5, IPAddress.IPV6)
gw_list = self.dhcp_gw_info.get_all_router_ips()
expected = gw_list_to_set([gw1, gw2, gw3, gw4, gw5])
self.assertEqual(gw_list_to_set(gw_list), expected)
class DefGwTestIpv6(unittest.TestCase):
"""
Validate v6 router setting.
"""
@classmethod
def setUpClass(cls, *_):
"""
Starts the thread which launches ryu apps
Create a testing bridge, add a port, setup the port interfaces. Then
launch the ryu apps for testing pipelined. Gets the references
to apps launched by using futures.
"""
super(DefGwTestIpv6, cls).setUpClass()
uplink_gw.netifaces = MockedNetInterface()
def setUp(self):
self.gw_store = defaultdict(str)
self.dhcp_gw_info = UplinkGatewayInfo(self.gw_store)
def test_gw_ip_for_DHCP(self):
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), None)
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), None)
def test_gw_ip_for_Ip_pool(self):
self.dhcp_gw_info.read_default_gw()
def_ip = '10.0.2.3'
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), '')
mac1 = "11:22:33:44:55:66"
self.dhcp_gw_info.update_mac(def_ip, mac1)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), mac1)
# updating IP with same address shld keep mac
self.dhcp_gw_info.update_ip(def_ip)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), mac1)
ip1 = "1.2.3.4"
self.dhcp_gw_info.update_ip(ip1)
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), '')
def test_gw_ip_for_Ip_pool_v6(self):
self.dhcp_gw_info.read_default_gw_v6()
def_ip_v6 = '2002::1'
self.assertEqual(self.dhcp_gw_info.get_gw_ip(version=IPAddress.IPV6), str(def_ip_v6))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(version=IPAddress.IPV6), '')
mac6 = "11:22:33:44:55:66"
self.dhcp_gw_info.update_mac(def_ip_v6, mac6, version=IPAddress.IPV6)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(version=IPAddress.IPV6), str(def_ip_v6))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(version=IPAddress.IPV6), mac6)
self.dhcp_gw_info.read_default_gw()
def_ip = '10.0.2.3'
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), '')
mac4 = "11:22:33:44:55:44"
self.dhcp_gw_info.update_mac(def_ip, mac4)
self.assertEqual(self.dhcp_gw_info.get_gw_ip(), str(def_ip))
self.assertEqual(self.dhcp_gw_info.get_gw_mac(), mac4)
ip4 = def_ip
mac4 = mac4
gw4 = _get_gw_info(ip4, mac4)
ip6 = def_ip_v6
mac6 = mac6
gw6 = _get_gw_info(ip6, mac6)
gw_list = self.dhcp_gw_info.get_all_router_ips()
expected = gw_list_to_set([gw4, gw6])
self.assertEqual(gw_list_to_set(gw_list), expected)