aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/dhcp
diff options
context:
space:
mode:
authorDave Wallace <dwallacelf@gmail.com>2021-05-12 21:43:59 -0400
committerDamjan Marion <dmarion@me.com>2021-05-13 09:33:06 +0000
commiteddd8e3588561039985b27edf059db6033bfdfab (patch)
tree44896887d6070853ea77a18cae218f5d4ef4d93a /src/plugins/dhcp
parentfd77f8c00c8e9d528d91a9cefae1878e383582ed (diff)
tests: move test source to vpp/test
- Generate copyright year and version instead of using hard-coded data Type: refactor Signed-off-by: Dave Wallace <dwallacelf@gmail.com> Change-Id: I6058f5025323b3aa483f5df4a2c4371e27b5914e
Diffstat (limited to 'src/plugins/dhcp')
-rw-r--r--src/plugins/dhcp/test/test_dhcp.py1686
-rw-r--r--src/plugins/dhcp/test/test_dhcp6.py805
-rw-r--r--src/plugins/dhcp/test/vpp_dhcp.py131
3 files changed, 0 insertions, 2622 deletions
diff --git a/src/plugins/dhcp/test/test_dhcp.py b/src/plugins/dhcp/test/test_dhcp.py
deleted file mode 100644
index e17b0049df7..00000000000
--- a/src/plugins/dhcp/test/test_dhcp.py
+++ /dev/null
@@ -1,1686 +0,0 @@
-#!/usr/bin/env python3
-
-import unittest
-import socket
-import struct
-import six
-
-from framework import VppTestCase, VppTestRunner, running_extended_tests
-from framework import tag_run_solo
-from vpp_neighbor import VppNeighbor
-from vpp_ip_route import find_route, VppIpTable
-from util import mk_ll_addr
-import scapy.compat
-from scapy.layers.l2 import Ether, getmacbyip, ARP, Dot1Q
-from scapy.layers.inet import IP, UDP, ICMP
-from scapy.layers.inet6 import IPv6, in6_getnsmac
-from scapy.utils6 import in6_mactoifaceid
-from scapy.layers.dhcp import DHCP, BOOTP, DHCPTypes
-from scapy.layers.dhcp6 import DHCP6, DHCP6_Solicit, DHCP6_RelayForward, \
- DHCP6_RelayReply, DHCP6_Advertise, DHCP6OptRelayMsg, DHCP6OptIfaceId, \
- DHCP6OptStatusCode, DHCP6OptVSS, DHCP6OptClientLinkLayerAddr, DHCP6_Request
-from socket import AF_INET, AF_INET6, inet_pton, inet_ntop
-from scapy.utils6 import in6_ptop
-from vpp_papi import mac_pton, VppEnum
-from vpp_sub_interface import VppDot1QSubint
-from vpp_qos import VppQosEgressMap, VppQosMark
-from vpp_dhcp import VppDHCPClient, VppDHCPProxy
-
-
-DHCP4_CLIENT_PORT = 68
-DHCP4_SERVER_PORT = 67
-DHCP6_CLIENT_PORT = 547
-DHCP6_SERVER_PORT = 546
-
-
-@tag_run_solo
-class TestDHCP(VppTestCase):
- """ DHCP Test Case """
-
- @classmethod
- def setUpClass(cls):
- super(TestDHCP, cls).setUpClass()
-
- @classmethod
- def tearDownClass(cls):
- super(TestDHCP, cls).tearDownClass()
-
- def setUp(self):
- super(TestDHCP, self).setUp()
-
- # create 6 pg interfaces for pg0 to pg5
- self.create_pg_interfaces(range(6))
- self.tables = []
-
- # pg0 to 2 are IP configured in VRF 0, 1 and 2.
- # pg3 to 5 are non IP-configured in VRF 0, 1 and 2.
- table_id = 0
- for table_id in range(1, 4):
- tbl4 = VppIpTable(self, table_id)
- tbl4.add_vpp_config()
- self.tables.append(tbl4)
- tbl6 = VppIpTable(self, table_id, is_ip6=1)
- tbl6.add_vpp_config()
- self.tables.append(tbl6)
-
- table_id = 0
- for i in self.pg_interfaces[:3]:
- i.admin_up()
- i.set_table_ip4(table_id)
- i.set_table_ip6(table_id)
- i.config_ip4()
- i.resolve_arp()
- i.config_ip6()
- i.resolve_ndp()
- table_id += 1
-
- table_id = 0
- for i in self.pg_interfaces[3:]:
- i.admin_up()
- i.set_table_ip4(table_id)
- i.set_table_ip6(table_id)
- table_id += 1
-
- def tearDown(self):
- for i in self.pg_interfaces[:3]:
- i.unconfig_ip4()
- i.unconfig_ip6()
-
- for i in self.pg_interfaces:
- i.set_table_ip4(0)
- i.set_table_ip6(0)
- i.admin_down()
- super(TestDHCP, self).tearDown()
-
- def verify_dhcp_has_option(self, pkt, option, value):
- dhcp = pkt[DHCP]
- found = False
-
- for i in dhcp.options:
- if isinstance(i, tuple):
- if i[0] == option:
- self.assertEqual(i[1], value)
- found = True
-
- self.assertTrue(found)
-
- def validate_relay_options(self, pkt, intf, ip_addr, vpn_id, fib_id, oui):
- dhcp = pkt[DHCP]
- found = 0
- data = []
- id_len = len(vpn_id)
-
- for i in dhcp.options:
- if isinstance(i, tuple):
- if i[0] == "relay_agent_Information":
- #
- # There are two sb-options present - each of length 6.
- #
- data = i[1]
- if oui != 0:
- self.assertEqual(len(data), 24)
- elif len(vpn_id) > 0:
- self.assertEqual(len(data), len(vpn_id) + 17)
- else:
- self.assertEqual(len(data), 12)
-
- #
- # First sub-option is ID 1, len 4, then encoded
- # sw_if_index. This test uses low valued indicies
- # so [2:4] are 0.
- # The ID space is VPP internal - so no matching value
- # scapy
- #
- self.assertEqual(six.byte2int(data[0:1]), 1)
- self.assertEqual(six.byte2int(data[1:2]), 4)
- self.assertEqual(six.byte2int(data[2:3]), 0)
- self.assertEqual(six.byte2int(data[3:4]), 0)
- self.assertEqual(six.byte2int(data[4:5]), 0)
- self.assertEqual(six.byte2int(data[5:6]),
- intf._sw_if_index)
-
- #
- # next sub-option is the IP address of the client side
- # interface.
- # sub-option ID=5, length (of a v4 address)=4
- #
- claddr = socket.inet_pton(AF_INET, ip_addr)
-
- self.assertEqual(six.byte2int(data[6:7]), 5)
- self.assertEqual(six.byte2int(data[7:8]), 4)
- self.assertEqual(data[8], claddr[0])
- self.assertEqual(data[9], claddr[1])
- self.assertEqual(data[10], claddr[2])
- self.assertEqual(data[11], claddr[3])
-
- if oui != 0:
- # sub-option 151 encodes vss_type 1,
- # the 3 byte oui and the 4 byte fib_id
- self.assertEqual(id_len, 0)
- self.assertEqual(six.byte2int(data[12:13]), 151)
- self.assertEqual(six.byte2int(data[13:14]), 8)
- self.assertEqual(six.byte2int(data[14:15]), 1)
- self.assertEqual(six.byte2int(data[15:16]), 0)
- self.assertEqual(six.byte2int(data[16:17]), 0)
- self.assertEqual(six.byte2int(data[17:18]), oui)
- self.assertEqual(six.byte2int(data[18:19]), 0)
- self.assertEqual(six.byte2int(data[19:20]), 0)
- self.assertEqual(six.byte2int(data[20:21]), 0)
- self.assertEqual(six.byte2int(data[21:22]), fib_id)
-
- # VSS control sub-option
- self.assertEqual(six.byte2int(data[22:23]), 152)
- self.assertEqual(six.byte2int(data[23:24]), 0)
-
- if id_len > 0:
- # sub-option 151 encode vss_type of 0
- # followerd by vpn_id in ascii
- self.assertEqual(oui, 0)
- self.assertEqual(six.byte2int(data[12:13]), 151)
- self.assertEqual(six.byte2int(data[13:14]), id_len + 1)
- self.assertEqual(six.byte2int(data[14:15]), 0)
- self.assertEqual(data[15:15 + id_len].decode('ascii'),
- vpn_id)
-
- # VSS control sub-option
- self.assertEqual(six.byte2int(data[15 + len(vpn_id):
- 16 + len(vpn_id)]),
- 152)
- self.assertEqual(six.byte2int(data[16 + len(vpn_id):
- 17 + len(vpn_id)]),
- 0)
-
- found = 1
- self.assertTrue(found)
-
- return data
-
- def verify_dhcp_msg_type(self, pkt, name):
- dhcp = pkt[DHCP]
- found = False
- for o in dhcp.options:
- if isinstance(o, tuple):
- if o[0] == "message-type" \
- and DHCPTypes[o[1]] == name:
- found = True
- self.assertTrue(found)
-
- def verify_dhcp_offer(self, pkt, intf, vpn_id="", fib_id=0, oui=0):
- ether = pkt[Ether]
- self.assertEqual(ether.dst, "ff:ff:ff:ff:ff:ff")
- self.assertEqual(ether.src, intf.local_mac)
-
- ip = pkt[IP]
- self.assertEqual(ip.dst, "255.255.255.255")
- self.assertEqual(ip.src, intf.local_ip4)
-
- udp = pkt[UDP]
- self.assertEqual(udp.dport, DHCP4_CLIENT_PORT)
- self.assertEqual(udp.sport, DHCP4_SERVER_PORT)
-
- self.verify_dhcp_msg_type(pkt, "offer")
- data = self.validate_relay_options(pkt, intf, intf.local_ip4,
- vpn_id, fib_id, oui)
-
- def verify_orig_dhcp_pkt(self, pkt, intf, dscp, l2_bc=True):
- ether = pkt[Ether]
- if l2_bc:
- self.assertEqual(ether.dst, "ff:ff:ff:ff:ff:ff")
- else:
- self.assertEqual(ether.dst, intf.remote_mac)
- self.assertEqual(ether.src, intf.local_mac)
-
- ip = pkt[IP]
-
- if (l2_bc):
- self.assertEqual(ip.dst, "255.255.255.255")
- self.assertEqual(ip.src, "0.0.0.0")
- else:
- self.assertEqual(ip.dst, intf.remote_ip4)
- self.assertEqual(ip.src, intf.local_ip4)
- self.assertEqual(ip.tos, dscp)
-
- udp = pkt[UDP]
- self.assertEqual(udp.dport, DHCP4_SERVER_PORT)
- self.assertEqual(udp.sport, DHCP4_CLIENT_PORT)
-
- def verify_orig_dhcp_discover(self, pkt, intf, hostname, client_id=None,
- broadcast=True, dscp=0):
- self.verify_orig_dhcp_pkt(pkt, intf, dscp)
-
- self.verify_dhcp_msg_type(pkt, "discover")
- self.verify_dhcp_has_option(pkt, "hostname",
- hostname.encode('ascii'))
- if client_id:
- client_id = '\x00' + client_id
- self.verify_dhcp_has_option(pkt, "client_id",
- client_id.encode('ascii'))
- bootp = pkt[BOOTP]
- self.assertEqual(bootp.ciaddr, "0.0.0.0")
- self.assertEqual(bootp.giaddr, "0.0.0.0")
- if broadcast:
- self.assertEqual(bootp.flags, 0x8000)
- else:
- self.assertEqual(bootp.flags, 0x0000)
-
- def verify_orig_dhcp_request(self, pkt, intf, hostname, ip,
- broadcast=True,
- l2_bc=True,
- dscp=0):
- self.verify_orig_dhcp_pkt(pkt, intf, dscp, l2_bc=l2_bc)
-
- self.verify_dhcp_msg_type(pkt, "request")
- self.verify_dhcp_has_option(pkt, "hostname",
- hostname.encode('ascii'))
- self.verify_dhcp_has_option(pkt, "requested_addr", ip)
- bootp = pkt[BOOTP]
-
- if l2_bc:
- self.assertEqual(bootp.ciaddr, "0.0.0.0")
- else:
- self.assertEqual(bootp.ciaddr, intf.local_ip4)
- self.assertEqual(bootp.giaddr, "0.0.0.0")
-
- if broadcast:
- self.assertEqual(bootp.flags, 0x8000)
- else:
- self.assertEqual(bootp.flags, 0x0000)
-
- def verify_relayed_dhcp_discover(self, pkt, intf, src_intf=None,
- fib_id=0, oui=0,
- vpn_id="",
- dst_mac=None, dst_ip=None):
- if not dst_mac:
- dst_mac = intf.remote_mac
- if not dst_ip:
- dst_ip = intf.remote_ip4
-
- ether = pkt[Ether]
- self.assertEqual(ether.dst, dst_mac)
- self.assertEqual(ether.src, intf.local_mac)
-
- ip = pkt[IP]
- self.assertEqual(ip.dst, dst_ip)
- self.assertEqual(ip.src, intf.local_ip4)
-
- udp = pkt[UDP]
- self.assertEqual(udp.dport, DHCP4_SERVER_PORT)
- self.assertEqual(udp.sport, DHCP4_CLIENT_PORT)
-
- dhcp = pkt[DHCP]
-
- is_discover = False
- for o in dhcp.options:
- if isinstance(o, tuple):
- if o[0] == "message-type" \
- and DHCPTypes[o[1]] == "discover":
- is_discover = True
- self.assertTrue(is_discover)
-
- data = self.validate_relay_options(pkt, src_intf,
- src_intf.local_ip4,
- vpn_id,
- fib_id, oui)
- return data
-
- def verify_dhcp6_solicit(self, pkt, intf,
- peer_ip, peer_mac,
- vpn_id="",
- fib_id=0,
- oui=0,
- dst_mac=None,
- dst_ip=None):
- if not dst_mac:
- dst_mac = intf.remote_mac
- if not dst_ip:
- dst_ip = in6_ptop(intf.remote_ip6)
-
- ether = pkt[Ether]
- self.assertEqual(ether.dst, dst_mac)
- self.assertEqual(ether.src, intf.local_mac)
-
- ip = pkt[IPv6]
- self.assertEqual(in6_ptop(ip.dst), dst_ip)
- self.assertEqual(in6_ptop(ip.src), in6_ptop(intf.local_ip6))
-
- udp = pkt[UDP]
- self.assertEqual(udp.dport, DHCP6_CLIENT_PORT)
- self.assertEqual(udp.sport, DHCP6_SERVER_PORT)
-
- relay = pkt[DHCP6_RelayForward]
- self.assertEqual(in6_ptop(relay.peeraddr), in6_ptop(peer_ip))
- oid = pkt[DHCP6OptIfaceId]
- cll = pkt[DHCP6OptClientLinkLayerAddr]
- self.assertEqual(cll.optlen, 8)
- self.assertEqual(cll.lltype, 1)
- self.assertEqual(cll.clladdr, peer_mac)
-
- id_len = len(vpn_id)
-
- if fib_id != 0:
- self.assertEqual(id_len, 0)
- vss = pkt[DHCP6OptVSS]
- self.assertEqual(vss.optlen, 8)
- self.assertEqual(vss.type, 1)
- # the OUI and FIB-id are really 3 and 4 bytes resp.
- # but the tested range is small
- self.assertEqual(six.byte2int(vss.data[0:1]), 0)
- self.assertEqual(six.byte2int(vss.data[1:2]), 0)
- self.assertEqual(six.byte2int(vss.data[2:3]), oui)
- self.assertEqual(six.byte2int(vss.data[3:4]), 0)
- self.assertEqual(six.byte2int(vss.data[4:5]), 0)
- self.assertEqual(six.byte2int(vss.data[5:6]), 0)
- self.assertEqual(six.byte2int(vss.data[6:7]), fib_id)
-
- if id_len > 0:
- self.assertEqual(oui, 0)
- vss = pkt[DHCP6OptVSS]
- self.assertEqual(vss.optlen, id_len + 1)
- self.assertEqual(vss.type, 0)
- self.assertEqual(vss.data[0:id_len].decode('ascii'),
- vpn_id)
-
- # the relay message should be an encoded Solicit
- msg = pkt[DHCP6OptRelayMsg]
- sol = DHCP6_Solicit()
- self.assertEqual(msg.optlen, len(sol))
- self.assertEqual(sol, msg[1])
-
- def verify_dhcp6_advert(self, pkt, intf, peer):
- ether = pkt[Ether]
- self.assertEqual(ether.dst, "ff:ff:ff:ff:ff:ff")
- self.assertEqual(ether.src, intf.local_mac)
-
- ip = pkt[IPv6]
- self.assertEqual(in6_ptop(ip.dst), in6_ptop(peer))
- self.assertEqual(in6_ptop(ip.src), in6_ptop(intf.local_ip6))
-
- udp = pkt[UDP]
- self.assertEqual(udp.dport, DHCP6_SERVER_PORT)
- self.assertEqual(udp.sport, DHCP6_CLIENT_PORT)
-
- # not sure why this is not decoding
- # adv = pkt[DHCP6_Advertise]
-
- def wait_for_no_route(self, address, length,
- n_tries=50, s_time=1):
- while (n_tries):
- if not find_route(self, address, length):
- return True
- n_tries = n_tries - 1
- self.sleep(s_time)
-
- return False
-
- def test_dhcp_proxy(self):
- """ DHCPv4 Proxy """
-
- #
- # Verify no response to DHCP request without DHCP config
- #
- p_disc_vrf0 = (Ether(dst="ff:ff:ff:ff:ff:ff",
- src=self.pg3.remote_mac) /
- IP(src="0.0.0.0", dst="255.255.255.255") /
- UDP(sport=DHCP4_CLIENT_PORT,
- dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'discover'), ('end')]))
- pkts_disc_vrf0 = [p_disc_vrf0]
- p_disc_vrf1 = (Ether(dst="ff:ff:ff:ff:ff:ff",
- src=self.pg4.remote_mac) /
- IP(src="0.0.0.0", dst="255.255.255.255") /
- UDP(sport=DHCP4_CLIENT_PORT,
- dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'discover'), ('end')]))
- pkts_disc_vrf1 = [p_disc_vrf1]
- p_disc_vrf2 = (Ether(dst="ff:ff:ff:ff:ff:ff",
- src=self.pg5.remote_mac) /
- IP(src="0.0.0.0", dst="255.255.255.255") /
- UDP(sport=DHCP4_CLIENT_PORT,
- dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'discover'), ('end')]))
- pkts_disc_vrf2 = [p_disc_vrf2]
-
- self.send_and_assert_no_replies(self.pg3, pkts_disc_vrf0,
- "DHCP with no configuration")
- self.send_and_assert_no_replies(self.pg4, pkts_disc_vrf1,
- "DHCP with no configuration")
- self.send_and_assert_no_replies(self.pg5, pkts_disc_vrf2,
- "DHCP with no configuration")
-
- #
- # Enable DHCP proxy in VRF 0
- #
- server_addr = self.pg0.remote_ip4
- src_addr = self.pg0.local_ip4
-
- Proxy = VppDHCPProxy(self, server_addr, src_addr, rx_vrf_id=0)
- Proxy.add_vpp_config()
-
- #
- # Discover packets from the client are dropped because there is no
- # IP address configured on the client facing interface
- #
- self.send_and_assert_no_replies(self.pg3, pkts_disc_vrf0,
- "Discover DHCP no relay address")
-
- #
- # Inject a response from the server
- # dropped, because there is no IP addrees on the
- # client interfce to fill in the option.
- #
- p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IP(src=self.pg0.remote_ip4, dst=self.pg0.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'), ('end')]))
- pkts = [p]
-
- self.send_and_assert_no_replies(self.pg3, pkts,
- "Offer DHCP no relay address")
-
- #
- # configure an IP address on the client facing interface
- #
- self.pg3.config_ip4()
-
- #
- # Try again with a discover packet
- # Rx'd packet should be to the server address and from the configured
- # source address
- # UDP source ports are unchanged
- # we've no option 82 config so that should be absent
- #
- self.pg3.add_stream(pkts_disc_vrf0)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg0.get_capture(1)
- rx = rx[0]
-
- option_82 = self.verify_relayed_dhcp_discover(rx, self.pg0,
- src_intf=self.pg3)
-
- #
- # Create an DHCP offer reply from the server with a correctly formatted
- # option 82. i.e. send back what we just captured
- # The offer, sent mcast to the client, still has option 82.
- #
- p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IP(src=self.pg0.remote_ip4, dst=self.pg0.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', option_82),
- ('end')]))
- pkts = [p]
-
- self.pg0.add_stream(pkts)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
- rx = rx[0]
-
- self.verify_dhcp_offer(rx, self.pg3)
-
- #
- # Bogus Option 82:
- #
- # 1. not our IP address = not checked by VPP? so offer is replayed
- # to client
- bad_ip = option_82[0:8] + scapy.compat.chb(33) + option_82[9:]
-
- p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IP(src=self.pg0.remote_ip4, dst=self.pg0.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', bad_ip),
- ('end')]))
- pkts = [p]
- self.send_and_assert_no_replies(self.pg0, pkts,
- "DHCP offer option 82 bad address")
-
- # 2. Not a sw_if_index VPP knows
- bad_if_index = option_82[0:2] + scapy.compat.chb(33) + option_82[3:]
-
- p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IP(src=self.pg0.remote_ip4, dst=self.pg0.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', bad_if_index),
- ('end')]))
- pkts = [p]
- self.send_and_assert_no_replies(self.pg0, pkts,
- "DHCP offer option 82 bad if index")
-
- #
- # Send a DHCP request in VRF 1. should be dropped.
- #
- self.send_and_assert_no_replies(self.pg4, pkts_disc_vrf1,
- "DHCP with no configuration VRF 1")
-
- #
- # Delete the DHCP config in VRF 0
- # Should now drop requests.
- #
- Proxy.remove_vpp_config()
-
- self.send_and_assert_no_replies(self.pg3, pkts_disc_vrf0,
- "DHCP config removed VRF 0")
- self.send_and_assert_no_replies(self.pg4, pkts_disc_vrf1,
- "DHCP config removed VRF 1")
-
- #
- # Add DHCP config for VRF 1 & 2
- #
- server_addr1 = self.pg1.remote_ip4
- src_addr1 = self.pg1.local_ip4
- Proxy1 = VppDHCPProxy(
- self,
- server_addr1,
- src_addr1,
- rx_vrf_id=1,
- server_vrf_id=1)
- Proxy1.add_vpp_config()
-
- server_addr2 = self.pg2.remote_ip4
- src_addr2 = self.pg2.local_ip4
- Proxy2 = VppDHCPProxy(
- self,
- server_addr2,
- src_addr2,
- rx_vrf_id=2,
- server_vrf_id=2)
- Proxy2.add_vpp_config()
-
- #
- # Confim DHCP requests ok in VRF 1 & 2.
- # - dropped on IP config on client interface
- #
- self.send_and_assert_no_replies(self.pg4, pkts_disc_vrf1,
- "DHCP config removed VRF 1")
- self.send_and_assert_no_replies(self.pg5, pkts_disc_vrf2,
- "DHCP config removed VRF 2")
-
- #
- # configure an IP address on the client facing interface
- #
- self.pg4.config_ip4()
- self.pg4.add_stream(pkts_disc_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
- rx = self.pg1.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg1, src_intf=self.pg4)
-
- self.pg5.config_ip4()
- self.pg5.add_stream(pkts_disc_vrf2)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
- rx = self.pg2.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg2, src_intf=self.pg5)
-
- #
- # Add VSS config
- # table=1, vss_type=1, vpn_index=1, oui=4
- # table=2, vss_type=0, vpn_id = "ip4-table-2"
- self.vapi.dhcp_proxy_set_vss(tbl_id=1, vss_type=1,
- vpn_index=1, oui=4, is_add=1)
- self.vapi.dhcp_proxy_set_vss(tbl_id=2, vss_type=0,
- vpn_ascii_id="ip4-table-2", is_add=1)
-
- self.pg4.add_stream(pkts_disc_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg1,
- src_intf=self.pg4,
- fib_id=1, oui=4)
-
- self.pg5.add_stream(pkts_disc_vrf2)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg2.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg2,
- src_intf=self.pg5,
- vpn_id="ip4-table-2")
-
- #
- # Add a second DHCP server in VRF 1
- # expect clients messages to be relay to both configured servers
- #
- self.pg1.generate_remote_hosts(2)
- server_addr12 = self.pg1.remote_hosts[1].ip4
-
- Proxy12 = VppDHCPProxy(
- self,
- server_addr12,
- src_addr,
- rx_vrf_id=1,
- server_vrf_id=1)
- Proxy12.add_vpp_config()
-
- #
- # We'll need an ARP entry for the server to send it packets
- #
- arp_entry = VppNeighbor(self,
- self.pg1.sw_if_index,
- self.pg1.remote_hosts[1].mac,
- self.pg1.remote_hosts[1].ip4)
- arp_entry.add_vpp_config()
-
- #
- # Send a discover from the client. expect two relayed messages
- # The frist packet is sent to the second server
- # We're not enforcing that here, it's just the way it is.
- #
- self.pg4.add_stream(pkts_disc_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(2)
-
- option_82 = self.verify_relayed_dhcp_discover(
- rx[0], self.pg1,
- src_intf=self.pg4,
- dst_mac=self.pg1.remote_hosts[1].mac,
- dst_ip=self.pg1.remote_hosts[1].ip4,
- fib_id=1, oui=4)
- self.verify_relayed_dhcp_discover(rx[1], self.pg1,
- src_intf=self.pg4,
- fib_id=1, oui=4)
-
- #
- # Send both packets back. Client gets both.
- #
- p1 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
- IP(src=self.pg1.remote_ip4, dst=self.pg1.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', option_82),
- ('end')]))
- p2 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
- IP(src=self.pg1.remote_hosts[1].ip4, dst=self.pg1.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', option_82),
- ('end')]))
- pkts = [p1, p2]
-
- self.pg1.add_stream(pkts)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg4.get_capture(2)
-
- self.verify_dhcp_offer(rx[0], self.pg4, fib_id=1, oui=4)
- self.verify_dhcp_offer(rx[1], self.pg4, fib_id=1, oui=4)
-
- #
- # Ensure offers from non-servers are dropeed
- #
- p2 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
- IP(src="8.8.8.8", dst=self.pg1.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'offer'),
- ('relay_agent_Information', option_82),
- ('end')]))
- self.send_and_assert_no_replies(self.pg1, p2,
- "DHCP offer from non-server")
-
- #
- # Ensure only the discover is sent to multiple servers
- #
- p_req_vrf1 = (Ether(dst="ff:ff:ff:ff:ff:ff",
- src=self.pg4.remote_mac) /
- IP(src="0.0.0.0", dst="255.255.255.255") /
- UDP(sport=DHCP4_CLIENT_PORT,
- dport=DHCP4_SERVER_PORT) /
- BOOTP(op=1) /
- DHCP(options=[('message-type', 'request'),
- ('end')]))
-
- self.pg4.add_stream(p_req_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- #
- # Remove the second DHCP server
- #
- Proxy12.remove_vpp_config()
-
- #
- # Test we can still relay with the first
- #
- self.pg4.add_stream(pkts_disc_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg1,
- src_intf=self.pg4,
- fib_id=1, oui=4)
-
- #
- # Remove the VSS config
- # relayed DHCP has default vlaues in the option.
- #
- self.vapi.dhcp_proxy_set_vss(tbl_id=1, is_add=0)
- self.vapi.dhcp_proxy_set_vss(tbl_id=2, is_add=0)
-
- self.pg4.add_stream(pkts_disc_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
- rx = rx[0]
- self.verify_relayed_dhcp_discover(rx, self.pg1, src_intf=self.pg4)
-
- #
- # remove DHCP config to cleanup
- #
- Proxy1.remove_vpp_config()
- Proxy2.remove_vpp_config()
-
- self.send_and_assert_no_replies(self.pg3, pkts_disc_vrf0,
- "DHCP cleanup VRF 0")
- self.send_and_assert_no_replies(self.pg4, pkts_disc_vrf1,
- "DHCP cleanup VRF 1")
- self.send_and_assert_no_replies(self.pg5, pkts_disc_vrf2,
- "DHCP cleanup VRF 2")
-
- self.pg3.unconfig_ip4()
- self.pg4.unconfig_ip4()
- self.pg5.unconfig_ip4()
-
- def test_dhcp6_proxy(self):
- """ DHCPv6 Proxy"""
- #
- # Verify no response to DHCP request without DHCP config
- #
- dhcp_solicit_dst = "ff02::1:2"
- dhcp_solicit_src_vrf0 = mk_ll_addr(self.pg3.remote_mac)
- dhcp_solicit_src_vrf1 = mk_ll_addr(self.pg4.remote_mac)
- dhcp_solicit_src_vrf2 = mk_ll_addr(self.pg5.remote_mac)
- server_addr_vrf0 = self.pg0.remote_ip6
- src_addr_vrf0 = self.pg0.local_ip6
- server_addr_vrf1 = self.pg1.remote_ip6
- src_addr_vrf1 = self.pg1.local_ip6
- server_addr_vrf2 = self.pg2.remote_ip6
- src_addr_vrf2 = self.pg2.local_ip6
-
- dmac = in6_getnsmac(inet_pton(socket.AF_INET6, dhcp_solicit_dst))
- p_solicit_vrf0 = (Ether(dst=dmac, src=self.pg3.remote_mac) /
- IPv6(src=dhcp_solicit_src_vrf0,
- dst=dhcp_solicit_dst) /
- UDP(sport=DHCP6_SERVER_PORT,
- dport=DHCP6_CLIENT_PORT) /
- DHCP6_Solicit())
- p_solicit_vrf1 = (Ether(dst=dmac, src=self.pg4.remote_mac) /
- IPv6(src=dhcp_solicit_src_vrf1,
- dst=dhcp_solicit_dst) /
- UDP(sport=DHCP6_SERVER_PORT,
- dport=DHCP6_CLIENT_PORT) /
- DHCP6_Solicit())
- p_solicit_vrf2 = (Ether(dst=dmac, src=self.pg5.remote_mac) /
- IPv6(src=dhcp_solicit_src_vrf2,
- dst=dhcp_solicit_dst) /
- UDP(sport=DHCP6_SERVER_PORT,
- dport=DHCP6_CLIENT_PORT) /
- DHCP6_Solicit())
-
- self.send_and_assert_no_replies(self.pg3, p_solicit_vrf0,
- "DHCP with no configuration")
- self.send_and_assert_no_replies(self.pg4, p_solicit_vrf1,
- "DHCP with no configuration")
- self.send_and_assert_no_replies(self.pg5, p_solicit_vrf2,
- "DHCP with no configuration")
-
- #
- # DHCPv6 config in VRF 0.
- # Packets still dropped because the client facing interface has no
- # IPv6 config
- #
- Proxy = VppDHCPProxy(
- self,
- server_addr_vrf0,
- src_addr_vrf0,
- rx_vrf_id=0,
- server_vrf_id=0)
- Proxy.add_vpp_config()
-
- self.send_and_assert_no_replies(self.pg3, p_solicit_vrf0,
- "DHCP with no configuration")
- self.send_and_assert_no_replies(self.pg4, p_solicit_vrf1,
- "DHCP with no configuration")
-
- #
- # configure an IP address on the client facing interface
- #
- self.pg3.config_ip6()
-
- #
- # Now the DHCP requests are relayed to the server
- #
- self.pg3.add_stream(p_solicit_vrf0)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg0.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg0,
- dhcp_solicit_src_vrf0,
- self.pg3.remote_mac)
-
- #
- # Exception cases for rejected relay responses
- #
-
- # 1 - not a relay reply
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_Advertise())
- self.send_and_assert_no_replies(self.pg3, p_adv_vrf0,
- "DHCP6 not a relay reply")
-
- # 2 - no relay message option
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply() /
- DHCP6_Advertise())
- self.send_and_assert_no_replies(self.pg3, p_adv_vrf0,
- "DHCP not a relay message")
-
- # 3 - no circuit ID
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply() /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise())
- self.send_and_assert_no_replies(self.pg3, p_adv_vrf0,
- "DHCP6 no circuit ID")
- # 4 - wrong circuit ID
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply() /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x05') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise())
- self.send_and_assert_no_replies(self.pg3, p_adv_vrf0,
- "DHCP6 wrong circuit ID")
-
- #
- # Send the relay response (the advertisement)
- # - no peer address
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply() /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x04') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
- pkts_adv_vrf0 = [p_adv_vrf0]
-
- self.pg0.add_stream(pkts_adv_vrf0)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
-
- self.verify_dhcp6_advert(rx[0], self.pg3, "::")
-
- #
- # Send the relay response (the advertisement)
- # - with peer address
- p_adv_vrf0 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
- IPv6(dst=self.pg0.local_ip6, src=self.pg0.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply(peeraddr=dhcp_solicit_src_vrf0) /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x04') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
- pkts_adv_vrf0 = [p_adv_vrf0]
-
- self.pg0.add_stream(pkts_adv_vrf0)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
-
- self.verify_dhcp6_advert(rx[0], self.pg3, dhcp_solicit_src_vrf0)
-
- #
- # Add all the config for VRF 1 & 2
- #
- Proxy1 = VppDHCPProxy(
- self,
- server_addr_vrf1,
- src_addr_vrf1,
- rx_vrf_id=1,
- server_vrf_id=1)
- Proxy1.add_vpp_config()
- self.pg4.config_ip6()
-
- Proxy2 = VppDHCPProxy(
- self,
- server_addr_vrf2,
- src_addr_vrf2,
- rx_vrf_id=2,
- server_vrf_id=2)
- Proxy2.add_vpp_config()
- self.pg5.config_ip6()
-
- #
- # VRF 1 solicit
- #
- self.pg4.add_stream(p_solicit_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac)
-
- #
- # VRF 2 solicit
- #
- self.pg5.add_stream(p_solicit_vrf2)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg2.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg2,
- dhcp_solicit_src_vrf2,
- self.pg5.remote_mac)
-
- #
- # VRF 1 Advert
- #
- p_adv_vrf1 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
- IPv6(dst=self.pg1.local_ip6, src=self.pg1.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply(peeraddr=dhcp_solicit_src_vrf1) /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x05') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
- pkts_adv_vrf1 = [p_adv_vrf1]
-
- self.pg1.add_stream(pkts_adv_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg4.get_capture(1)
-
- self.verify_dhcp6_advert(rx[0], self.pg4, dhcp_solicit_src_vrf1)
-
- #
- # Add VSS config
- #
- self.vapi.dhcp_proxy_set_vss(
- tbl_id=1, vss_type=1, oui=4, vpn_index=1, is_ipv6=1, is_add=1)
- self.vapi.dhcp_proxy_set_vss(
- tbl_id=2,
- vss_type=0,
- vpn_ascii_id="IPv6-table-2",
- is_ipv6=1,
- is_add=1)
-
- self.pg4.add_stream(p_solicit_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac,
- fib_id=1,
- oui=4)
-
- self.pg5.add_stream(p_solicit_vrf2)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg2.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg2,
- dhcp_solicit_src_vrf2,
- self.pg5.remote_mac,
- vpn_id="IPv6-table-2")
-
- #
- # Remove the VSS config
- # relayed DHCP has default vlaues in the option.
- #
- self.vapi.dhcp_proxy_set_vss(tbl_id=1, is_ipv6=1, is_add=0)
-
- self.pg4.add_stream(p_solicit_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac)
-
- #
- # Add a second DHCP server in VRF 1
- # expect clients messages to be relay to both configured servers
- #
- self.pg1.generate_remote_hosts(2)
- server_addr12 = self.pg1.remote_hosts[1].ip6
-
- Proxy12 = VppDHCPProxy(
- self,
- server_addr12,
- src_addr_vrf1,
- rx_vrf_id=1,
- server_vrf_id=1)
- Proxy12.add_vpp_config()
-
- #
- # We'll need an ND entry for the server to send it packets
- #
- nd_entry = VppNeighbor(self,
- self.pg1.sw_if_index,
- self.pg1.remote_hosts[1].mac,
- self.pg1.remote_hosts[1].ip6)
- nd_entry.add_vpp_config()
-
- #
- # Send a discover from the client. expect two relayed messages
- # The frist packet is sent to the second server
- # We're not enforcing that here, it's just the way it is.
- #
- self.pg4.add_stream(p_solicit_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(2)
-
- self.verify_dhcp6_solicit(rx[0], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac)
- self.verify_dhcp6_solicit(rx[1], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac,
- dst_mac=self.pg1.remote_hosts[1].mac,
- dst_ip=self.pg1.remote_hosts[1].ip6)
-
- #
- # Send both packets back. Client gets both.
- #
- p1 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_mac) /
- IPv6(dst=self.pg1.local_ip6, src=self.pg1.remote_ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply(peeraddr=dhcp_solicit_src_vrf1) /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x05') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
- p2 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_hosts[1].mac) /
- IPv6(dst=self.pg1.local_ip6, src=self.pg1._remote_hosts[1].ip6) /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply(peeraddr=dhcp_solicit_src_vrf1) /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x05') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
-
- pkts = [p1, p2]
-
- self.pg1.add_stream(pkts)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg4.get_capture(2)
-
- self.verify_dhcp6_advert(rx[0], self.pg4, dhcp_solicit_src_vrf1)
- self.verify_dhcp6_advert(rx[1], self.pg4, dhcp_solicit_src_vrf1)
-
- #
- # Ensure only solicit messages are duplicated
- #
- p_request_vrf1 = (Ether(dst=dmac, src=self.pg4.remote_mac) /
- IPv6(src=dhcp_solicit_src_vrf1,
- dst=dhcp_solicit_dst) /
- UDP(sport=DHCP6_SERVER_PORT,
- dport=DHCP6_CLIENT_PORT) /
- DHCP6_Request())
-
- self.pg4.add_stream(p_request_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- #
- # Test we drop DHCP packets from addresses that are not configured as
- # DHCP servers
- #
- p2 = (Ether(dst=self.pg1.local_mac, src=self.pg1.remote_hosts[1].mac) /
- IPv6(dst=self.pg1.local_ip6, src="3001::1") /
- UDP(sport=DHCP6_SERVER_PORT, dport=DHCP6_SERVER_PORT) /
- DHCP6_RelayReply(peeraddr=dhcp_solicit_src_vrf1) /
- DHCP6OptIfaceId(optlen=4, ifaceid='\x00\x00\x00\x05') /
- DHCP6OptRelayMsg(optlen=0) /
- DHCP6_Advertise(trid=1) /
- DHCP6OptStatusCode(statuscode=0))
- self.send_and_assert_no_replies(self.pg1, p2,
- "DHCP6 not from server")
-
- #
- # Remove the second DHCP server
- #
- Proxy12.remove_vpp_config()
-
- #
- # Test we can still relay with the first
- #
- self.pg4.add_stream(p_solicit_vrf1)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg1.get_capture(1)
-
- self.verify_dhcp6_solicit(rx[0], self.pg1,
- dhcp_solicit_src_vrf1,
- self.pg4.remote_mac)
-
- #
- # Cleanup
- #
- Proxy.remove_vpp_config()
- Proxy1.remove_vpp_config()
- Proxy2.remove_vpp_config()
-
- self.pg3.unconfig_ip6()
- self.pg4.unconfig_ip6()
- self.pg5.unconfig_ip6()
-
- def test_dhcp_client(self):
- """ DHCP Client"""
-
- vdscp = VppEnum.vl_api_ip_dscp_t
- hostname = 'universal-dp'
-
- self.pg_enable_capture(self.pg_interfaces)
-
- #
- # Configure DHCP client on PG3 and capture the discover sent
- #
- Client = VppDHCPClient(self, self.pg3.sw_if_index, hostname)
- Client.add_vpp_config()
- self.assertTrue(Client.query_vpp_config())
-
- rx = self.pg3.get_capture(1)
-
- self.verify_orig_dhcp_discover(rx[0], self.pg3, hostname)
-
- #
- # Send back on offer, expect the request
- #
- p_offer = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst="255.255.255.255") /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1,
- yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'offer'),
- ('server_id', self.pg3.remote_ip4),
- 'end']))
-
- self.pg3.add_stream(p_offer)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
- self.verify_orig_dhcp_request(rx[0], self.pg3, hostname,
- self.pg3.local_ip4)
-
- #
- # Send an acknowledgment
- #
- p_ack = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst="255.255.255.255") /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'ack'),
- ('subnet_mask', "255.255.255.0"),
- ('router', self.pg3.remote_ip4),
- ('server_id', self.pg3.remote_ip4),
- ('lease_time', 43200),
- 'end']))
-
- self.pg3.add_stream(p_ack)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- #
- # We'll get an ARP request for the router address
- #
- rx = self.pg3.get_capture(1)
-
- self.assertEqual(rx[0][ARP].pdst, self.pg3.remote_ip4)
- self.pg_enable_capture(self.pg_interfaces)
-
- #
- # At the end of this procedure there should be a connected route
- # in the FIB
- #
- self.assertTrue(find_route(self, self.pg3.local_ip4, 24))
- self.assertTrue(find_route(self, self.pg3.local_ip4, 32))
-
- #
- # remove the DHCP config
- #
- Client.remove_vpp_config()
-
- #
- # and now the route should be gone
- #
- self.assertFalse(find_route(self, self.pg3.local_ip4, 32))
- self.assertFalse(find_route(self, self.pg3.local_ip4, 24))
-
- #
- # Start the procedure again. this time have VPP send the client-ID
- # and set the DSCP value
- #
- self.pg3.admin_down()
- self.sleep(1)
- self.pg3.admin_up()
- Client.set_client(self.pg3.sw_if_index, hostname,
- id=self.pg3.local_mac,
- dscp=vdscp.IP_API_DSCP_EF)
- Client.add_vpp_config()
-
- rx = self.pg3.get_capture(1)
-
- self.verify_orig_dhcp_discover(rx[0], self.pg3, hostname,
- self.pg3.local_mac,
- dscp=vdscp.IP_API_DSCP_EF)
-
- # TODO: VPP DHCP client should not accept DHCP OFFER message with
- # the XID (Transaction ID) not matching the XID of the most recent
- # DHCP DISCOVERY message.
- # Such DHCP OFFER message must be silently discarded - RFC2131.
- # Reported in Jira ticket: VPP-99
- self.pg3.add_stream(p_offer)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
- self.verify_orig_dhcp_request(rx[0], self.pg3, hostname,
- self.pg3.local_ip4,
- dscp=vdscp.IP_API_DSCP_EF)
-
- #
- # unicast the ack to the offered address
- #
- p_ack = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst=self.pg3.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'ack'),
- ('subnet_mask', "255.255.255.0"),
- ('router', self.pg3.remote_ip4),
- ('server_id', self.pg3.remote_ip4),
- ('lease_time', 43200),
- 'end']))
-
- self.pg3.add_stream(p_ack)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- #
- # We'll get an ARP request for the router address
- #
- rx = self.pg3.get_capture(1)
-
- self.assertEqual(rx[0][ARP].pdst, self.pg3.remote_ip4)
- self.pg_enable_capture(self.pg_interfaces)
-
- #
- # At the end of this procedure there should be a connected route
- # in the FIB
- #
- self.assertTrue(find_route(self, self.pg3.local_ip4, 32))
- self.assertTrue(find_route(self, self.pg3.local_ip4, 24))
-
- #
- # remove the DHCP config
- #
- Client.remove_vpp_config()
-
- self.assertFalse(find_route(self, self.pg3.local_ip4, 32))
- self.assertFalse(find_route(self, self.pg3.local_ip4, 24))
-
- #
- # Rince and repeat, this time with VPP configured not to set
- # the braodcast flag in the discover and request messages,
- # and for the server to unicast the responses.
- #
- # Configure DHCP client on PG3 and capture the discover sent
- #
- Client.set_client(
- self.pg3.sw_if_index,
- hostname,
- set_broadcast_flag=False)
- Client.add_vpp_config()
-
- rx = self.pg3.get_capture(1)
-
- self.verify_orig_dhcp_discover(rx[0], self.pg3, hostname,
- broadcast=False)
-
- #
- # Send back on offer, unicasted to the offered address.
- # Expect the request.
- #
- p_offer = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst=self.pg3.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'offer'),
- ('server_id', self.pg3.remote_ip4),
- 'end']))
-
- self.pg3.add_stream(p_offer)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
- self.verify_orig_dhcp_request(rx[0], self.pg3, hostname,
- self.pg3.local_ip4,
- broadcast=False)
-
- #
- # Send an acknowledgment, the lease renewal time is 2 seconds
- # so we should expect the renew straight after
- #
- p_ack = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst=self.pg3.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'ack'),
- ('subnet_mask', "255.255.255.0"),
- ('router', self.pg3.remote_ip4),
- ('server_id', self.pg3.remote_ip4),
- ('lease_time', 43200),
- ('renewal_time', 2),
- 'end']))
-
- self.pg3.add_stream(p_ack)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- #
- # We'll get an ARP request for the router address
- #
- rx = self.pg3.get_capture(1)
-
- self.assertEqual(rx[0][ARP].pdst, self.pg3.remote_ip4)
- self.pg_enable_capture(self.pg_interfaces)
-
- #
- # At the end of this procedure there should be a connected route
- # in the FIB
- #
- self.assertTrue(find_route(self, self.pg3.local_ip4, 24))
- self.assertTrue(find_route(self, self.pg3.local_ip4, 32))
-
- #
- # read the DHCP client details from a dump
- #
- clients = self.vapi.dhcp_client_dump()
-
- self.assertEqual(clients[0].client.sw_if_index,
- self.pg3.sw_if_index)
- self.assertEqual(clients[0].lease.sw_if_index,
- self.pg3.sw_if_index)
- self.assertEqual(clients[0].client.hostname, hostname)
- self.assertEqual(clients[0].lease.hostname, hostname)
- # 0 = DISCOVER, 1 = REQUEST, 2 = BOUND
- self.assertEqual(clients[0].lease.state, 2)
- self.assertEqual(clients[0].lease.mask_width, 24)
- self.assertEqual(str(clients[0].lease.router_address),
- self.pg3.remote_ip4)
- self.assertEqual(str(clients[0].lease.host_address),
- self.pg3.local_ip4)
-
- #
- # wait for the unicasted renewal
- # the first attempt will be an ARP packet, since we have not yet
- # responded to VPP's request
- #
- self.logger.info(self.vapi.cli("sh dhcp client intfc pg3 verbose"))
- rx = self.pg3.get_capture(1, timeout=10)
-
- self.assertEqual(rx[0][ARP].pdst, self.pg3.remote_ip4)
-
- # respond to the arp
- p_arp = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- ARP(op="is-at",
- hwdst=self.pg3.local_mac,
- hwsrc=self.pg3.remote_mac,
- pdst=self.pg3.local_ip4,
- psrc=self.pg3.remote_ip4))
- self.pg3.add_stream(p_arp)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- # the next packet is the unicasted renewal
- rx = self.pg3.get_capture(1, timeout=10)
- self.verify_orig_dhcp_request(rx[0], self.pg3, hostname,
- self.pg3.local_ip4,
- l2_bc=False,
- broadcast=False)
-
- # send an ACK with different data from the original offer *
- self.pg3.generate_remote_hosts(4)
- p_ack = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst=self.pg3.local_ip4) /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.remote_hosts[3].ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'ack'),
- ('subnet_mask', "255.255.255.0"),
- ('router', self.pg3.remote_hosts[1].ip4),
- ('server_id', self.pg3.remote_hosts[2].ip4),
- ('lease_time', 43200),
- ('renewal_time', 2),
- 'end']))
-
- self.pg3.add_stream(p_ack)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- #
- # read the DHCP client details from a dump
- #
- clients = self.vapi.dhcp_client_dump()
-
- self.assertEqual(clients[0].client.sw_if_index,
- self.pg3.sw_if_index)
- self.assertEqual(clients[0].lease.sw_if_index,
- self.pg3.sw_if_index)
- self.assertEqual(clients[0].client.hostname, hostname)
- self.assertEqual(clients[0].lease.hostname, hostname)
- # 0 = DISCOVER, 1 = REQUEST, 2 = BOUND
- self.assertEqual(clients[0].lease.state, 2)
- self.assertEqual(clients[0].lease.mask_width, 24)
- self.assertEqual(str(clients[0].lease.router_address),
- self.pg3.remote_hosts[1].ip4)
- self.assertEqual(str(clients[0].lease.host_address),
- self.pg3.remote_hosts[3].ip4)
-
- #
- # remove the DHCP config
- #
- Client.remove_vpp_config()
-
- #
- # and now the route should be gone
- #
- self.assertFalse(find_route(self, self.pg3.local_ip4, 32))
- self.assertFalse(find_route(self, self.pg3.local_ip4, 24))
-
- #
- # Start the procedure again. Use requested lease time option.
- # this time wait for the lease to expire and the client to
- # self-destruct
- #
- hostname += "-2"
- self.pg3.admin_down()
- self.sleep(1)
- self.pg3.admin_up()
- self.pg_enable_capture(self.pg_interfaces)
- Client.set_client(self.pg3.sw_if_index, hostname)
- Client.add_vpp_config()
-
- rx = self.pg3.get_capture(1)
-
- self.verify_orig_dhcp_discover(rx[0], self.pg3, hostname)
-
- #
- # Send back on offer with requested lease time, expect the request
- #
- lease_time = 1
- p_offer = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst='255.255.255.255') /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1,
- yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'offer'),
- ('server_id', self.pg3.remote_ip4),
- ('lease_time', lease_time),
- 'end']))
-
- self.pg3.add_stream(p_offer)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- rx = self.pg3.get_capture(1)
- self.verify_orig_dhcp_request(rx[0], self.pg3, hostname,
- self.pg3.local_ip4)
-
- #
- # Send an acknowledgment
- #
- p_ack = (Ether(dst=self.pg3.local_mac, src=self.pg3.remote_mac) /
- IP(src=self.pg3.remote_ip4, dst='255.255.255.255') /
- UDP(sport=DHCP4_SERVER_PORT, dport=DHCP4_CLIENT_PORT) /
- BOOTP(op=1, yiaddr=self.pg3.local_ip4,
- chaddr=mac_pton(self.pg3.local_mac)) /
- DHCP(options=[('message-type', 'ack'),
- ('subnet_mask', '255.255.255.0'),
- ('router', self.pg3.remote_ip4),
- ('server_id', self.pg3.remote_ip4),
- ('lease_time', lease_time),
- 'end']))
-
- self.pg3.add_stream(p_ack)
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- #
- # We'll get an ARP request for the router address
- #
- rx = self.pg3.get_capture(1)
-
- self.assertEqual(rx[0][ARP].pdst, self.pg3.remote_ip4)
-
- #
- # At the end of this procedure there should be a connected route
- # in the FIB
- #
- self.assertTrue(find_route(self, self.pg3.local_ip4, 32))
- self.assertTrue(find_route(self, self.pg3.local_ip4, 24))
-
- #
- # the route should be gone after the lease expires
- #
- self.assertTrue(self.wait_for_no_route(self.pg3.local_ip4, 32))
- self.assertTrue(self.wait_for_no_route(self.pg3.local_ip4, 24))
-
- #
- # remove the DHCP config
- #
- Client.remove_vpp_config()
-
- def test_dhcp_client_vlan(self):
- """ DHCP Client w/ VLAN"""
-
- vdscp = VppEnum.vl_api_ip_dscp_t
- vqos = VppEnum.vl_api_qos_source_t
- hostname = 'universal-dp'
-
- self.pg_enable_capture(self.pg_interfaces)
-
- vlan_100 = VppDot1QSubint(self, self.pg3, 100)
- vlan_100.admin_up()
-
- output = [scapy.compat.chb(4)] * 256
- os = b''.join(output)
- rows = [{'outputs': os},
- {'outputs': os},
- {'outputs': os},
- {'outputs': os}]
-
- qem1 = VppQosEgressMap(self, 1, rows).add_vpp_config()
- qm1 = VppQosMark(self, vlan_100, qem1,
- vqos.QOS_API_SOURCE_VLAN).add_vpp_config()
-
- #
- # Configure DHCP client on PG3 and capture the discover sent
- #
- Client = VppDHCPClient(
- self,
- vlan_100.sw_if_index,
- hostname,
- dscp=vdscp.IP_API_DSCP_EF)
- Client.add_vpp_config()
-
- rx = self.pg3.get_capture(1)
-
- self.assertEqual(rx[0][Dot1Q].vlan, 100)
- self.assertEqual(rx[0][Dot1Q].prio, 2)
-
- self.verify_orig_dhcp_discover(rx[0], self.pg3, hostname,
- dscp=vdscp.IP_API_DSCP_EF)
-
-
-if __name__ == '__main__':
- unittest.main(testRunner=VppTestRunner)
diff --git a/src/plugins/dhcp/test/test_dhcp6.py b/src/plugins/dhcp/test/test_dhcp6.py
deleted file mode 100644
index 57eb113fb13..00000000000
--- a/src/plugins/dhcp/test/test_dhcp6.py
+++ /dev/null
@@ -1,805 +0,0 @@
-from socket import AF_INET6, inet_ntop, inet_pton
-
-from scapy.layers.dhcp6 import DHCP6_Advertise, DHCP6OptClientId, \
- DHCP6OptStatusCode, DHCP6OptPref, DHCP6OptIA_PD, DHCP6OptIAPrefix, \
- DHCP6OptServerId, DHCP6_Solicit, DHCP6_Reply, DHCP6_Request, DHCP6_Renew, \
- DHCP6_Rebind, DUID_LL, DHCP6_Release, DHCP6OptElapsedTime, DHCP6OptIA_NA, \
- DHCP6OptIAAddress
-from scapy.layers.inet6 import IPv6, Ether, UDP
-from scapy.utils6 import in6_mactoifaceid
-
-from framework import tag_fixme_vpp_workers
-from framework import VppTestCase
-from framework import tag_run_solo
-from vpp_papi import VppEnum
-import util
-import os
-
-
-def ip6_normalize(ip6):
- return inet_ntop(AF_INET6, inet_pton(AF_INET6, ip6))
-
-
-class TestDHCPv6DataPlane(VppTestCase):
- """ DHCPv6 Data Plane Test Case """
-
- @classmethod
- def setUpClass(cls):
- super(TestDHCPv6DataPlane, cls).setUpClass()
-
- @classmethod
- def tearDownClass(cls):
- super(TestDHCPv6DataPlane, cls).tearDownClass()
-
- def setUp(self):
- super(TestDHCPv6DataPlane, self).setUp()
-
- self.create_pg_interfaces(range(1))
- self.interfaces = list(self.pg_interfaces)
- for i in self.interfaces:
- i.admin_up()
- i.config_ip6()
-
- self.server_duid = DUID_LL(lladdr=self.pg0.remote_mac)
-
- def tearDown(self):
- for i in self.interfaces:
- i.unconfig_ip6()
- i.admin_down()
- super(TestDHCPv6DataPlane, self).tearDown()
-
- def test_dhcp_ia_na_send_solicit_receive_advertise(self):
- """ Verify DHCPv6 IA NA Solicit packet and Advertise event """
-
- self.vapi.dhcp6_clients_enable_disable(enable=1)
-
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
- address = {'address': '1:2:3::5',
- 'preferred_time': 60,
- 'valid_time': 120}
- self.vapi.dhcp6_send_client_message(
- server_index=0xffffffff,
- mrc=1,
- msg_type=VppEnum.vl_api_dhcpv6_msg_type_t.DHCPV6_MSG_API_SOLICIT,
- sw_if_index=self.pg0.sw_if_index,
- T1=20,
- T2=40,
- addresses=[address],
- n_addresses=len(
- [address]))
- rx_list = self.pg0.get_capture(1)
- self.assertEqual(len(rx_list), 1)
- packet = rx_list[0]
-
- self.assertEqual(packet.haslayer(IPv6), 1)
- self.assertEqual(packet[IPv6].haslayer(DHCP6_Solicit), 1)
-
- client_duid = packet[DHCP6OptClientId].duid
- trid = packet[DHCP6_Solicit].trid
-
- dst = ip6_normalize(packet[IPv6].dst)
- dst2 = ip6_normalize("ff02::1:2")
- self.assert_equal(dst, dst2)
- src = ip6_normalize(packet[IPv6].src)
- src2 = ip6_normalize(self.pg0.local_ip6_ll)
- self.assert_equal(src, src2)
- ia_na = packet[DHCP6OptIA_NA]
- self.assert_equal(ia_na.T1, 20)
- self.assert_equal(ia_na.T2, 40)
- self.assert_equal(len(ia_na.ianaopts), 1)
- address = ia_na.ianaopts[0]
- self.assert_equal(address.addr, '1:2:3::5')
- self.assert_equal(address.preflft, 60)
- self.assert_equal(address.validlft, 120)
-
- self.vapi.want_dhcp6_reply_events(enable_disable=1,
- pid=os.getpid())
-
- try:
- ia_na_opts = DHCP6OptIAAddress(addr='7:8::2', preflft=60,
- validlft=120)
- p = (Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) /
- IPv6(src=util.mk_ll_addr(self.pg0.remote_mac),
- dst=self.pg0.local_ip6_ll) /
- UDP(sport=547, dport=546) /
- DHCP6_Advertise(trid=trid) /
- DHCP6OptServerId(duid=self.server_duid) /
- DHCP6OptClientId(duid=client_duid) /
- DHCP6OptPref(prefval=7) /
- DHCP6OptStatusCode(statuscode=1) /
- DHCP6OptIA_NA(iaid=1, T1=20, T2=40, ianaopts=ia_na_opts)
- )
- self.pg0.add_stream([p])
- self.pg_start()
-
- ev = self.vapi.wait_for_event(1, "dhcp6_reply_event")
-
- self.assert_equal(ev.preference, 7)
- self.assert_equal(ev.status_code, 1)
- self.assert_equal(ev.T1, 20)
- self.assert_equal(ev.T2, 40)
-
- reported_address = ev.addresses[0]
- address = ia_na_opts.getfieldval("addr")
- self.assert_equal(str(reported_address.address), address)
- self.assert_equal(reported_address.preferred_time,
- ia_na_opts.getfieldval("preflft"))
- self.assert_equal(reported_address.valid_time,
- ia_na_opts.getfieldval("validlft"))
-
- finally:
- self.vapi.want_dhcp6_reply_events(enable_disable=0)
- self.vapi.dhcp6_clients_enable_disable(enable=0)
-
- def test_dhcp_pd_send_solicit_receive_advertise(self):
- """ Verify DHCPv6 PD Solicit packet and Advertise event """
-
- self.vapi.dhcp6_clients_enable_disable(enable=1)
-
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- prefix = {'prefix': {'address': '1:2:3::', 'len': 50},
- 'preferred_time': 60,
- 'valid_time': 120}
- prefixes = [prefix]
- self.vapi.dhcp6_pd_send_client_message(
- server_index=0xffffffff,
- mrc=1,
- msg_type=VppEnum.vl_api_dhcpv6_msg_type_t.DHCPV6_MSG_API_SOLICIT,
- sw_if_index=self.pg0.sw_if_index,
- T1=20,
- T2=40,
- prefixes=prefixes,
- n_prefixes=len(prefixes))
- rx_list = self.pg0.get_capture(1)
- self.assertEqual(len(rx_list), 1)
- packet = rx_list[0]
-
- self.assertEqual(packet.haslayer(IPv6), 1)
- self.assertEqual(packet[IPv6].haslayer(DHCP6_Solicit), 1)
-
- client_duid = packet[DHCP6OptClientId].duid
- trid = packet[DHCP6_Solicit].trid
-
- dst = ip6_normalize(packet[IPv6].dst)
- dst2 = ip6_normalize("ff02::1:2")
- self.assert_equal(dst, dst2)
- src = ip6_normalize(packet[IPv6].src)
- src2 = ip6_normalize(self.pg0.local_ip6_ll)
- self.assert_equal(src, src2)
- ia_pd = packet[DHCP6OptIA_PD]
- self.assert_equal(ia_pd.T1, 20)
- self.assert_equal(ia_pd.T2, 40)
- self.assert_equal(len(ia_pd.iapdopt), 1)
- prefix = ia_pd.iapdopt[0]
- self.assert_equal(prefix.prefix, '1:2:3::')
- self.assert_equal(prefix.plen, 50)
- self.assert_equal(prefix.preflft, 60)
- self.assert_equal(prefix.validlft, 120)
-
- self.vapi.want_dhcp6_pd_reply_events(enable_disable=1,
- pid=os.getpid())
-
- try:
- ia_pd_opts = DHCP6OptIAPrefix(prefix='7:8::', plen=56, preflft=60,
- validlft=120)
- p = (Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) /
- IPv6(src=util.mk_ll_addr(self.pg0.remote_mac),
- dst=self.pg0.local_ip6_ll) /
- UDP(sport=547, dport=546) /
- DHCP6_Advertise(trid=trid) /
- DHCP6OptServerId(duid=self.server_duid) /
- DHCP6OptClientId(duid=client_duid) /
- DHCP6OptPref(prefval=7) /
- DHCP6OptStatusCode(statuscode=1) /
- DHCP6OptIA_PD(iaid=1, T1=20, T2=40, iapdopt=ia_pd_opts)
- )
- self.pg0.add_stream([p])
- self.pg_start()
-
- ev = self.vapi.wait_for_event(1, "dhcp6_pd_reply_event")
-
- self.assert_equal(ev.preference, 7)
- self.assert_equal(ev.status_code, 1)
- self.assert_equal(ev.T1, 20)
- self.assert_equal(ev.T2, 40)
-
- reported_prefix = ev.prefixes[0]
- prefix = ia_pd_opts.getfieldval("prefix")
- self.assert_equal(
- str(reported_prefix.prefix).split('/')[0], prefix)
- self.assert_equal(int(str(reported_prefix.prefix).split('/')[1]),
- ia_pd_opts.getfieldval("plen"))
- self.assert_equal(reported_prefix.preferred_time,
- ia_pd_opts.getfieldval("preflft"))
- self.assert_equal(reported_prefix.valid_time,
- ia_pd_opts.getfieldval("validlft"))
-
- finally:
- self.vapi.want_dhcp6_pd_reply_events(enable_disable=0)
- self.vapi.dhcp6_clients_enable_disable(enable=0)
-
-
-@tag_run_solo
-class TestDHCPv6IANAControlPlane(VppTestCase):
- """ DHCPv6 IA NA Control Plane Test Case """
-
- @classmethod
- def setUpClass(cls):
- super(TestDHCPv6IANAControlPlane, cls).setUpClass()
-
- @classmethod
- def tearDownClass(cls):
- super(TestDHCPv6IANAControlPlane, cls).tearDownClass()
-
- def setUp(self):
- super(TestDHCPv6IANAControlPlane, self).setUp()
-
- self.create_pg_interfaces(range(1))
- self.interfaces = list(self.pg_interfaces)
- for i in self.interfaces:
- i.admin_up()
-
- self.server_duid = DUID_LL(lladdr=self.pg0.remote_mac)
- self.client_duid = None
- self.T1 = 1
- self.T2 = 2
-
- fib = self.vapi.ip_route_dump(0, True)
- self.initial_addresses = set(self.get_interface_addresses(fib,
- self.pg0))
-
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- self.vapi.dhcp6_client_enable_disable(sw_if_index=self.pg0.sw_if_index,
- enable=1)
-
- def tearDown(self):
- self.vapi.dhcp6_client_enable_disable(sw_if_index=self.pg0.sw_if_index,
- enable=0)
-
- for i in self.interfaces:
- i.admin_down()
-
- super(TestDHCPv6IANAControlPlane, self).tearDown()
-
- @staticmethod
- def get_interface_addresses(fib, pg):
- lst = []
- for entry in fib:
- if entry.route.prefix.prefixlen == 128:
- path = entry.route.paths[0]
- if path.sw_if_index == pg.sw_if_index:
- lst.append(str(entry.route.prefix.network_address))
- return lst
-
- def get_addresses(self):
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg0))
- return addresses.difference(self.initial_addresses)
-
- def validate_duid_ll(self, duid):
- DUID_LL(duid)
-
- def validate_packet(self, packet, msg_type, is_resend=False):
- try:
- self.assertEqual(packet.haslayer(msg_type), 1)
- client_duid = packet[DHCP6OptClientId].duid
- if self.client_duid is None:
- self.client_duid = client_duid
- self.validate_duid_ll(client_duid)
- else:
- self.assertEqual(self.client_duid, client_duid)
- if msg_type != DHCP6_Solicit and msg_type != DHCP6_Rebind:
- server_duid = packet[DHCP6OptServerId].duid
- self.assertEqual(server_duid, self.server_duid)
- if is_resend:
- self.assertEqual(self.trid, packet[msg_type].trid)
- else:
- self.trid = packet[msg_type].trid
- ip = packet[IPv6]
- udp = packet[UDP]
- self.assertEqual(ip.dst, 'ff02::1:2')
- self.assertEqual(udp.sport, 546)
- self.assertEqual(udp.dport, 547)
- dhcpv6 = packet[msg_type]
- elapsed_time = dhcpv6[DHCP6OptElapsedTime]
- if (is_resend):
- self.assertNotEqual(elapsed_time.elapsedtime, 0)
- else:
- self.assertEqual(elapsed_time.elapsedtime, 0)
- except BaseException:
- packet.show()
- raise
-
- def wait_for_packet(self, msg_type, timeout=None, is_resend=False):
- if timeout is None:
- timeout = 3
- rx_list = self.pg0.get_capture(1, timeout=timeout)
- packet = rx_list[0]
- self.validate_packet(packet, msg_type, is_resend=is_resend)
-
- def wait_for_solicit(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Solicit, timeout, is_resend=is_resend)
-
- def wait_for_request(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Request, timeout, is_resend=is_resend)
-
- def wait_for_renew(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Renew, timeout, is_resend=is_resend)
-
- def wait_for_rebind(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Rebind, timeout, is_resend=is_resend)
-
- def wait_for_release(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Release, timeout, is_resend=is_resend)
-
- def send_packet(self, msg_type, t1=None, t2=None, ianaopts=None):
- if t1 is None:
- t1 = self.T1
- if t2 is None:
- t2 = self.T2
- if ianaopts is None:
- opt_ia_na = DHCP6OptIA_NA(iaid=1, T1=t1, T2=t2)
- else:
- opt_ia_na = DHCP6OptIA_NA(iaid=1, T1=t1, T2=t2, ianaopts=ianaopts)
- p = (Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) /
- IPv6(src=util.mk_ll_addr(self.pg0.remote_mac),
- dst=self.pg0.local_ip6_ll) /
- UDP(sport=547, dport=546) /
- msg_type(trid=self.trid) /
- DHCP6OptServerId(duid=self.server_duid) /
- DHCP6OptClientId(duid=self.client_duid) /
- opt_ia_na
- )
- self.pg0.add_stream([p])
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- def send_advertise(self, t1=None, t2=None, ianaopts=None):
- self.send_packet(DHCP6_Advertise, t1, t2, ianaopts)
-
- def send_reply(self, t1=None, t2=None, ianaopts=None):
- self.send_packet(DHCP6_Reply, t1, t2, ianaopts)
-
- def test_T1_and_T2_timeouts(self):
- """ Test T1 and T2 timeouts """
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- self.send_reply()
-
- self.sleep(1)
-
- self.wait_for_renew()
-
- self.pg_enable_capture(self.pg_interfaces)
-
- self.sleep(1)
-
- self.wait_for_rebind()
-
- def test_addresses(self):
- """ Test handling of addresses """
-
- ia_na_opts = DHCP6OptIAAddress(addr='7:8::2', preflft=1,
- validlft=2)
-
- self.wait_for_solicit()
- self.send_advertise(t1=20, t2=40, ianaopts=ia_na_opts)
- self.wait_for_request()
- self.send_reply(t1=20, t2=40, ianaopts=ia_na_opts)
- self.sleep(0.1)
-
- # check FIB for new address
- new_addresses = self.get_addresses()
- self.assertEqual(len(new_addresses), 1)
- addr = list(new_addresses)[0]
- self.assertEqual(addr, '7:8::2')
-
- self.sleep(2)
-
- # check that the address is deleted
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg0))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
- def test_sending_client_messages_solicit(self):
- """ VPP receives messages from DHCPv6 client """
-
- self.wait_for_solicit()
- self.send_packet(DHCP6_Solicit)
- self.send_packet(DHCP6_Request)
- self.send_packet(DHCP6_Renew)
- self.send_packet(DHCP6_Rebind)
- self.sleep(1)
- self.wait_for_solicit(is_resend=True)
-
- def test_sending_inappropriate_packets(self):
- """ Server sends messages with inappropriate message types """
-
- self.wait_for_solicit()
- self.send_reply()
- self.wait_for_solicit(is_resend=True)
- self.send_advertise()
- self.wait_for_request()
- self.send_advertise()
- self.wait_for_request(is_resend=True)
- self.send_reply()
- self.wait_for_renew()
-
- def test_no_address_available_in_advertise(self):
- """ Advertise message contains NoAddrsAvail status code """
-
- self.wait_for_solicit()
- noavail = DHCP6OptStatusCode(statuscode=2) # NoAddrsAvail
- self.send_advertise(ianaopts=noavail)
- self.wait_for_solicit(is_resend=True)
-
- def test_preferred_greater_than_valid_lifetime(self):
- """ Preferred lifetime is greater than valid lifetime """
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- ia_na_opts = DHCP6OptIAAddress(addr='7:8::2', preflft=4, validlft=3)
- self.send_reply(ianaopts=ia_na_opts)
-
- self.sleep(0.5)
-
- # check FIB contains no addresses
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg0))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
- def test_T1_greater_than_T2(self):
- """ T1 is greater than T2 """
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- ia_na_opts = DHCP6OptIAAddress(addr='7:8::2', preflft=4, validlft=8)
- self.send_reply(t1=80, t2=40, ianaopts=ia_na_opts)
-
- self.sleep(0.5)
-
- # check FIB contains no addresses
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg0))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
-
-@tag_fixme_vpp_workers
-class TestDHCPv6PDControlPlane(VppTestCase):
- """ DHCPv6 PD Control Plane Test Case """
-
- @classmethod
- def setUpClass(cls):
- super(TestDHCPv6PDControlPlane, cls).setUpClass()
-
- @classmethod
- def tearDownClass(cls):
- super(TestDHCPv6PDControlPlane, cls).tearDownClass()
-
- def setUp(self):
- super(TestDHCPv6PDControlPlane, self).setUp()
-
- self.create_pg_interfaces(range(2))
- self.interfaces = list(self.pg_interfaces)
- for i in self.interfaces:
- i.admin_up()
-
- self.server_duid = DUID_LL(lladdr=self.pg0.remote_mac)
- self.client_duid = None
- self.T1 = 1
- self.T2 = 2
-
- fib = self.vapi.ip_route_dump(0, True)
- self.initial_addresses = set(self.get_interface_addresses(fib,
- self.pg1))
-
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- self.prefix_group = 'my-pd-prefix-group'
-
- self.vapi.dhcp6_pd_client_enable_disable(
- enable=1,
- sw_if_index=self.pg0.sw_if_index,
- prefix_group=self.prefix_group)
-
- def tearDown(self):
- self.vapi.dhcp6_pd_client_enable_disable(self.pg0.sw_if_index,
- enable=0)
-
- for i in self.interfaces:
- i.admin_down()
-
- super(TestDHCPv6PDControlPlane, self).tearDown()
-
- @staticmethod
- def get_interface_addresses(fib, pg):
- lst = []
- for entry in fib:
- if entry.route.prefix.prefixlen == 128:
- path = entry.route.paths[0]
- if path.sw_if_index == pg.sw_if_index:
- lst.append(str(entry.route.prefix.network_address))
- return lst
-
- def get_addresses(self):
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg1))
- return addresses.difference(self.initial_addresses)
-
- def validate_duid_ll(self, duid):
- DUID_LL(duid)
-
- def validate_packet(self, packet, msg_type, is_resend=False):
- try:
- self.assertEqual(packet.haslayer(msg_type), 1)
- client_duid = packet[DHCP6OptClientId].duid
- if self.client_duid is None:
- self.client_duid = client_duid
- self.validate_duid_ll(client_duid)
- else:
- self.assertEqual(self.client_duid, client_duid)
- if msg_type != DHCP6_Solicit and msg_type != DHCP6_Rebind:
- server_duid = packet[DHCP6OptServerId].duid
- self.assertEqual(server_duid, self.server_duid)
- if is_resend:
- self.assertEqual(self.trid, packet[msg_type].trid)
- else:
- self.trid = packet[msg_type].trid
- ip = packet[IPv6]
- udp = packet[UDP]
- self.assertEqual(ip.dst, 'ff02::1:2')
- self.assertEqual(udp.sport, 546)
- self.assertEqual(udp.dport, 547)
- dhcpv6 = packet[msg_type]
- elapsed_time = dhcpv6[DHCP6OptElapsedTime]
- if (is_resend):
- self.assertNotEqual(elapsed_time.elapsedtime, 0)
- else:
- self.assertEqual(elapsed_time.elapsedtime, 0)
- except BaseException:
- packet.show()
- raise
-
- def wait_for_packet(self, msg_type, timeout=None, is_resend=False):
- if timeout is None:
- timeout = 3
- rx_list = self.pg0.get_capture(1, timeout=timeout)
- packet = rx_list[0]
- self.validate_packet(packet, msg_type, is_resend=is_resend)
-
- def wait_for_solicit(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Solicit, timeout, is_resend=is_resend)
-
- def wait_for_request(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Request, timeout, is_resend=is_resend)
-
- def wait_for_renew(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Renew, timeout, is_resend=is_resend)
-
- def wait_for_rebind(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Rebind, timeout, is_resend=is_resend)
-
- def wait_for_release(self, timeout=None, is_resend=False):
- self.wait_for_packet(DHCP6_Release, timeout, is_resend=is_resend)
-
- def send_packet(self, msg_type, t1=None, t2=None, iapdopt=None):
- if t1 is None:
- t1 = self.T1
- if t2 is None:
- t2 = self.T2
- if iapdopt is None:
- opt_ia_pd = DHCP6OptIA_PD(iaid=1, T1=t1, T2=t2)
- else:
- opt_ia_pd = DHCP6OptIA_PD(iaid=1, T1=t1, T2=t2, iapdopt=iapdopt)
- p = (Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) /
- IPv6(src=util.mk_ll_addr(self.pg0.remote_mac),
- dst=self.pg0.local_ip6_ll) /
- UDP(sport=547, dport=546) /
- msg_type(trid=self.trid) /
- DHCP6OptServerId(duid=self.server_duid) /
- DHCP6OptClientId(duid=self.client_duid) /
- opt_ia_pd
- )
- self.pg0.add_stream([p])
- self.pg_enable_capture(self.pg_interfaces)
- self.pg_start()
-
- def send_advertise(self, t1=None, t2=None, iapdopt=None):
- self.send_packet(DHCP6_Advertise, t1, t2, iapdopt)
-
- def send_reply(self, t1=None, t2=None, iapdopt=None):
- self.send_packet(DHCP6_Reply, t1, t2, iapdopt)
-
- def test_T1_and_T2_timeouts(self):
- """ Test T1 and T2 timeouts """
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- self.send_reply()
-
- self.sleep(1)
-
- self.wait_for_renew()
-
- self.pg_enable_capture(self.pg_interfaces)
-
- self.sleep(1)
-
- self.wait_for_rebind()
-
- def test_prefixes(self):
- """ Test handling of prefixes """
-
- address1 = '::2:0:0:0:405/60'
- address2 = '::76:0:0:0:406/62'
- try:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address1,
- prefix_group=self.prefix_group)
-
- ia_pd_opts = DHCP6OptIAPrefix(prefix='7:8::', plen=56, preflft=2,
- validlft=3)
-
- self.wait_for_solicit()
- self.send_advertise(t1=20, t2=40, iapdopt=ia_pd_opts)
- self.wait_for_request()
- self.send_reply(t1=20, t2=40, iapdopt=ia_pd_opts)
- self.sleep(0.1)
-
- # check FIB for new address
- new_addresses = self.get_addresses()
- self.assertEqual(len(new_addresses), 1)
- addr = list(new_addresses)[0]
- self.assertEqual(addr, '7:8:0:2::405')
-
- self.sleep(1)
-
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address2,
- prefix_group=self.prefix_group)
-
- self.sleep(1)
-
- # check FIB contains 2 addresses
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg1))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 2)
- addr1 = list(new_addresses)[0]
- addr2 = list(new_addresses)[1]
- if addr1 == '7:8:0:76::406':
- addr1, addr2 = addr2, addr1
- self.assertEqual(addr1, '7:8:0:2::405')
- self.assertEqual(addr2, '7:8:0:76::406')
-
- self.sleep(1)
-
- # check that the addresses are deleted
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg1))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
- finally:
- if address1 is not None:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address1,
- prefix_group=self.prefix_group, is_add=0)
- if address2 is not None:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address2,
- prefix_group=self.prefix_group, is_add=0)
-
- def test_sending_client_messages_solicit(self):
- """ VPP receives messages from DHCPv6 client """
-
- self.wait_for_solicit()
- self.send_packet(DHCP6_Solicit)
- self.send_packet(DHCP6_Request)
- self.send_packet(DHCP6_Renew)
- self.send_packet(DHCP6_Rebind)
- self.sleep(1)
- self.wait_for_solicit(is_resend=True)
-
- def test_sending_inappropriate_packets(self):
- """ Server sends messages with inappropriate message types """
-
- self.wait_for_solicit()
- self.send_reply()
- self.wait_for_solicit(is_resend=True)
- self.send_advertise()
- self.wait_for_request()
- self.send_advertise()
- self.wait_for_request(is_resend=True)
- self.send_reply()
- self.wait_for_renew()
-
- def test_no_prefix_available_in_advertise(self):
- """ Advertise message contains NoPrefixAvail status code """
-
- self.wait_for_solicit()
- noavail = DHCP6OptStatusCode(statuscode=6) # NoPrefixAvail
- self.send_advertise(iapdopt=noavail)
- self.wait_for_solicit(is_resend=True)
-
- def test_preferred_greater_than_valid_lifetime(self):
- """ Preferred lifetime is greater than valid lifetime """
-
- address1 = '::2:0:0:0:405/60'
- try:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address1,
- prefix_group=self.prefix_group)
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- ia_pd_opts = DHCP6OptIAPrefix(prefix='7:8::', plen=56, preflft=4,
- validlft=3)
- self.send_reply(iapdopt=ia_pd_opts)
-
- self.sleep(0.5)
-
- # check FIB contains no addresses
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg1))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
- finally:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address1,
- prefix_group=self.prefix_group,
- is_add=0)
-
- def test_T1_greater_than_T2(self):
- """ T1 is greater than T2 """
-
- address1 = '::2:0:0:0:405/60'
- try:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- address_with_prefix=address1,
- prefix_group=self.prefix_group)
-
- self.wait_for_solicit()
- self.send_advertise()
- self.wait_for_request()
- ia_pd_opts = DHCP6OptIAPrefix(prefix='7:8::', plen=56, preflft=4,
- validlft=8)
- self.send_reply(t1=80, t2=40, iapdopt=ia_pd_opts)
-
- self.sleep(0.5)
-
- # check FIB contains no addresses
- fib = self.vapi.ip_route_dump(0, True)
- addresses = set(self.get_interface_addresses(fib, self.pg1))
- new_addresses = addresses.difference(self.initial_addresses)
- self.assertEqual(len(new_addresses), 0)
-
- finally:
- self.vapi.ip6_add_del_address_using_prefix(
- sw_if_index=self.pg1.sw_if_index,
- prefix_group=self.prefix_group,
- address_with_prefix=address1,
- is_add=False)
diff --git a/src/plugins/dhcp/test/vpp_dhcp.py b/src/plugins/dhcp/test/vpp_dhcp.py
deleted file mode 100644
index f8265a26252..00000000000
--- a/src/plugins/dhcp/test/vpp_dhcp.py
+++ /dev/null
@@ -1,131 +0,0 @@
-from vpp_object import VppObject
-
-
-class VppDHCPProxy(VppObject):
-
- def __init__(
- self,
- test,
- dhcp_server,
- dhcp_src_address,
- rx_vrf_id=0,
- server_vrf_id=0,
- ):
- self._test = test
- self._rx_vrf_id = rx_vrf_id
- self._server_vrf_id = server_vrf_id
- self._dhcp_server = dhcp_server
- self._dhcp_src_address = dhcp_src_address
-
- def set_proxy(
- self,
- dhcp_server,
- dhcp_src_address,
- rx_vrf_id=0,
- server_vrf_id=0):
- if self.query_vpp_config():
- raise Exception('Vpp config present')
- self._rx_vrf_id = rx_vrf_id
- self._server_vrf_id = server_vrf_id
- self._dhcp_server = dhcp_server
- self._dhcp_src_address = dhcp_src_address
-
- def add_vpp_config(self):
- self._test.vapi.dhcp_proxy_config(
- is_add=1,
- rx_vrf_id=self._rx_vrf_id,
- server_vrf_id=self._server_vrf_id,
- dhcp_server=self._dhcp_server,
- dhcp_src_address=self._dhcp_src_address)
- self._test.registry.register(self, self._test.logger)
-
- def remove_vpp_config(self):
- self._test.vapi.dhcp_proxy_config(
- rx_vrf_id=self._rx_vrf_id,
- server_vrf_id=self._server_vrf_id,
- dhcp_server=self._dhcp_server,
- dhcp_src_address=self._dhcp_src_address,
- is_add=0)
-
- def get_vpp_dump(self):
- dump = self._test.vapi.dhcp_proxy_dump()
- for entry in dump:
- if entry.rx_vrf_id == self._rx_vrf_id:
- return entry
-
- def query_vpp_config(self):
- dump = self.get_vpp_dump()
- return True if dump else False
-
- def object_id(self):
- return "dhcp-proxy-%d" % self._rx_vrf_id
-
-
-class VppDHCPClient(VppObject):
-
- def __init__(
- self,
- test,
- sw_if_index,
- hostname,
- id=None,
- want_dhcp_event=False,
- set_broadcast_flag=True,
- dscp=None,
- pid=None):
- self._test = test
- self._sw_if_index = sw_if_index
- self._hostname = hostname
- self._id = id
- self._want_dhcp_event = want_dhcp_event
- self._set_broadcast_flag = set_broadcast_flag
- self._dscp = dscp
- self._pid = pid
-
- def set_client(
- self,
- sw_if_index,
- hostname,
- id=None,
- want_dhcp_event=False,
- set_broadcast_flag=True,
- dscp=None,
- pid=None):
- if self.query_vpp_config():
- raise Exception('Vpp config present')
- self._sw_if_index = sw_if_index
- self._hostname = hostname
- self._id = id
- self._want_dhcp_event = want_dhcp_event
- self._set_broadcast_flag = set_broadcast_flag
- self._dscp = dscp
- self._pid = pid
-
- def add_vpp_config(self):
- id = self._id.encode('ascii') if self._id else None
- client = {'sw_if_index': self._sw_if_index, 'hostname': self._hostname,
- 'id': id,
- 'want_dhcp_event': self._want_dhcp_event,
- 'set_broadcast_flag': self._set_broadcast_flag,
- 'dscp': self._dscp, 'pid': self._pid}
- self._test.vapi.dhcp_client_config(is_add=1, client=client)
- self._test.registry.register(self, self._test.logger)
-
- def remove_vpp_config(self):
- client = client = {
- 'sw_if_index': self._sw_if_index,
- 'hostname': self._hostname}
- self._test.vapi.dhcp_client_config(client=client, is_add=0)
-
- def get_vpp_dump(self):
- dump = self._test.vapi.dhcp_client_dump()
- for entry in dump:
- if entry.client.sw_if_index == self._sw_if_index:
- return entry
-
- def query_vpp_config(self):
- dump = self.get_vpp_dump()
- return True if dump else False
-
- def object_id(self):
- return "dhcp-client-%s/%d" % (self._hostname, self._sw_if_index)