Add DHCP IP2ME copp test (#5111)

Author: kellyyeh

ansible/roles/test/files/ptftests/copp_tests.py

@@ -7,6 +7,8 @@
 # ARPTest
 # DHCPTest
 # DHCPTopoT1Test
+# DHCPIP2METest
+# DHCPOTHERIPTest
 # DHCP6Test
 # LLDPTest
 # BGPTest
@@ -55,6 +57,7 @@ def __init__(self):
 
         self.myip = test_params.get('myip', None)
         self.peerip = test_params.get('peerip', None)
+        self.randip = test_params.get('randip', None)
         self.default_server_send_rate_limit_pps = test_params.get('send_rate_limit', 2000)
 
         self.needPreSend = None
@@ -353,6 +356,76 @@ def contruct_packet(self, port_number):
 
         return packet
 
+class DHCPIP2METest(NoPolicyTest):
+    def __init__(self):
+        NoPolicyTest.__init__(self)
+
+    def runTest(self):
+        self.log("DHCPIP2METest")
+        self.run_suite()
+
+    def contruct_packet(self, port_number):
+        src_mac = self.my_mac[port_number]
+        dst_mac = self.peer_mac[port_number]
+        dst_ip = self.peerip
+
+        packet = testutils.simple_udp_packet(
+            pktlen=100,
+            eth_src=src_mac,
+            eth_dst=dst_mac,
+            dl_vlan_enable=False,
+            vlan_vid=0,
+            vlan_pcp=0,
+            dl_vlan_cfi=0,
+            ip_src='0.0.0.0',
+            ip_dst=dst_ip,
+            ip_tos=0,
+            ip_ttl=64,
+            udp_sport=68,
+            udp_dport=67,
+            ip_ihl=None,
+            ip_options=False,
+            with_udp_chksum=True
+        )
+
+        return packet
+
+class DHCPOTHERIPTest(PolicyTest):
+    def __init__(self):
+        PolicyTest.__init__(self)
+        # Police rate is expected to be 0
+        self.PPS_LIMIT_MIN = 0
+        self.PPS_LIMIT_MAX = 0
+
+    def runTest(self):
+        self.log("DHCPOTHERIPTest")
+        self.run_suite()
+
+    def contruct_packet(self, port_number):
+        src_mac = self.my_mac[port_number]
+        dst_mac = self.peer_mac[port_number]
+
+        packet = testutils.simple_udp_packet(
+            pktlen=100,
+            eth_dst=dst_mac,
+            eth_src=src_mac,
+            dl_vlan_enable=False,
+            vlan_vid=0,
+            vlan_pcp=0,
+            dl_vlan_cfi=0,
+            ip_src='0.0.0.0',
+            ip_dst=self.randip,
+            ip_tos=0,
+            ip_ttl=64,
+            udp_sport=68,
+            udp_dport=67,
+            ip_ihl=None,
+            ip_options=False,
+            with_udp_chksum=True
+        )
+
+        return packet
+
 
 # SONIC configuration has no policer limiting for DHCPv6
 class DHCP6Test(NoPolicyTest):

tests/copp/test_copp.py

@@ -30,6 +30,7 @@
 from tests.ptf_runner import ptf_runner
 from tests.common import config_reload, constants
 from tests.common.system_utils import docker
+from tests.common.helpers.generators import generate_ip_through_default_route
 from tests.common.reboot import reboot
 from tests.common.utilities import skip_release
 from tests.common.utilities import wait_until
@@ -73,7 +74,8 @@ class TestCOPP(object):
     @pytest.mark.parametrize("protocol", ["ARP",
                                           "IP2ME",
                                           "SNMP",
-                                          "SSH"])
+                                          "SSH",
+                                          "DHCPOTHERIP"])
     def test_policer(self, protocol, duthosts, enum_rand_one_per_hwsku_frontend_hostname, ptfhost, copp_testbed, dut_type):
         """
             Validates that rate-limited COPP groups work as expected.
@@ -90,6 +92,7 @@ def test_policer(self, protocol, duthosts, enum_rand_one_per_hwsku_frontend_host
 
     @pytest.mark.parametrize("protocol", ["BGP",
                                           "DHCP",
+                                          "DHCPIP2ME",
                                           "DHCP6",
                                           "LACP",
                                           "LLDP",
@@ -264,6 +267,7 @@ def _copp_runner(dut, ptf, protocol, test_params, dut_type, has_trap=True):
               "target_port": test_params.nn_target_port,
               "myip": test_params.myip,
               "peerip": test_params.peerip,
+              "randip": generate_ip_through_default_route(duthost, [test_params.peerip]),
               "send_rate_limit": test_params.send_rate_limit,
               "has_trap": has_trap}