CSIT-353: Add basic map-t tests

Change-Id: I78075e445cdd61e0e3cf9353bddeaf6e6bc978fa
Signed-off-by: Matej Klotton <mklotton@cisco.com>

2 files changed, 120 insertions, 2 deletions

diff --git a/resources/libraries/python/Map.py b/resources/libraries/python/Map.py
index b98d488e73..a1db3a8461 100644
--- a/resources/libraries/python/Map.py
+++ b/resources/libraries/python/Map.py
@@ -24,7 +24,7 @@ class Map(object):
 
     @staticmethod
     def map_add_domain(vpp_node, ip4_pfx, ip6_pfx, ip6_src, ea_bits_len,
-                       psid_offset, psid_len):
+                       psid_offset, psid_len, map_t=False):
         """Add map domain on node.
 
         :param vpp_node: VPP node to add map domain on.
@@ -34,6 +34,8 @@ class Map(object):
         :param ea_bits_len: Embedded Address bits length.
         :param psid_offset: Port Set Identifier (PSID) offset.
         :param psid_len: Port Set Identifier (PSID) length.
+        :param map_t: Mapping using translation instead of encapsulation.
+        Default False.
         :type vpp_node: dict
         :type ip4_pfx: str
         :type ip6_pfx: str
@@ -41,17 +43,21 @@ class Map(object):
         :type ea_bits_len: int
         :type psid_offset: int
         :type psid_len: int
+        :type map_t: bool
         :return: Index of created map domain.
         :rtype: int
         :raises RuntimeError: If unable to add map domain.
         """
+        translate = 'map-t' if map_t else ''
+
         output = VatExecutor.cmd_from_template(vpp_node, "map_add_domain.vat",
                                                ip4_pfx=ip4_pfx,
                                                ip6_pfx=ip6_pfx,
                                                ip6_src=ip6_src,
                                                ea_bits_len=ea_bits_len,
                                                psid_offset=psid_offset,
-                                               psid_len=psid_len)
+                                               psid_len=psid_len,
+                                               map_t=translate)
         if output[0]["retval"] == 0:
             return output[0]["index"]
         else:
@@ -261,3 +267,22 @@ class Map(object):
         address |= interface_id  # add Interface ID bits
 
         return str(ipaddress.ip_address(address))
+
+    @staticmethod
+    def compute_ipv6_map_source_address(ipv6_pfx, ipv4_src):
+        """Compute IPv6 source address from IPv4 address for MAP-T algorithm.
+
+        :param ipv6_pfx: 96 bit long IPv6 prefix.
+        :param ipv4_src: IPv4 source address
+        :type ipv6_pfx: str
+        :type ipv4_src: str
+        :return: IPv6 address, combination of IPv6 prefix and IPv4 address.
+        :rtype: str
+        """
+        ipv6_net = ipaddress.ip_network(unicode(ipv6_pfx))
+        ipv4_host = ipaddress.ip_address(unicode(ipv4_src))
+
+        address = ipv6_net.network_address._ip
+        address |= ipv4_host._ip
+
+        return str(ipaddress.ip_address(address))
diff --git a/resources/libraries/robot/map.robot b/resources/libraries/robot/map.robot
index 7462befa41..fc7744e361 100644
--- a/resources/libraries/robot/map.robot
+++ b/resources/libraries/robot/map.robot
@@ -216,3 +216,96 @@
 | | ...
 | | Run Traffic Script On Node
 | | ... | send_lw_4o6_check_hairpinning_udp.py | ${tg_node} | ${args}
+
+| Send IPv4 UDP and check IPv6 headers for MAP-T
+| | [Documentation]
+| | ... | Send a UDP in IPv4 and check if IPv4 source and destination \
+| | ... | addresses are correctly translated into IPv6 addresses.
+| | ...
+| | ... | *Arguments:*
+| | ... | - tg_node - Node where to run traffic script. Type: string
+| | ... | - tx_if - Interface from where to send IPv4 UDP packet. Type: string
+| | ... | - rx_if - Interface where to receive IPv6 UDP packet. Type: string
+| | ... | - tx_dst_mac - Destination MAC address of IPv4 packet. Type: string
+| | ... | - tx_dst_ipv4 - Destination IPv4 address. Type: string
+| | ... | - tx_src_ipv4 - Source IPv4 address. Type: string
+| | ... | - tx_dst_udp_port - Destination UDP port. Type: integer
+| | ... | - rx_dst_mac - Expected destination MAC address. Type: string
+| | ... | - rx_src_mac - Expected source MAC address. Type: string
+| | ... | - dst_ipv6 - Expected destination IPv6 address. Type: string
+| | ... | - src_ipv6 - Expected source IPv6 address. Type: string
+| | ...
+| | ... | *Return:*
+| | ... | - No value returned
+| | ...
+| | ... | *Example:*
+| | ...
+| | ... | \| Send IPv4 UDP and check IPv6 headers for MAP-T \
+| | ... | \| ${tg_node} \| port3 \| port3 \| 08:00:27:66:b8:57 \
+| | ... | \| 20.169.201.219 \| 100.0.0.1 \| ${1232} \| 08:00:27:46:2b:4c \
+| | ... | \| 08:00:27:f3:be:f0 \| 2001:db8::14a9:c9db:0 \
+| | ... | \| 2001:db8:ffff::6400:1 \|
+| | ...
+| | [Arguments]
+| | ... | ${tg_node} | ${tx_if} | ${rx_if}
+| | ... | ${tx_dst_mac} | ${tx_dst_ipv4} | ${tx_src_ipv4} | ${tx_dst_udp_port}
+| | ... | ${rx_dst_mac} | ${rx_src_mac} | ${dst_ipv6} | ${src_ipv6}
+| | ...
+| | ${tx_name}= | Get interface name | ${tg_node} | ${tx_if}
+| | ${rx_name}= | Get interface name | ${tg_node} | ${rx_if}
+| | ${args}= | Catenate
+| | ... | --tx_if | ${tx_name} | --rx_if | ${rx_name}
+| | ... | --tx_dst_mac | ${tx_dst_mac}
+| | ... | --tx_src_ipv4 | ${tx_src_ipv4} | --tx_dst_ipv4 | ${tx_dst_ipv4}
+| | ... | --tx_dst_udp_port | ${tx_dst_udp_port}
+| | ... | --rx_dst_mac | ${rx_dst_mac} | --rx_src_mac | ${rx_src_mac}
+| | ... | --rx_src_ipv6 | ${src_ipv6} | --rx_dst_ipv6 | ${dst_ipv6}
+| | ...
+| | Run Traffic Script On Node
+| | ... | send_ipv4_udp_check_map_t.py | ${tg_node} | ${args}
+
+| Send IPv6 UDP and check IPv4 headers for MAP-T
+| | [Documentation]
+| | ... | Send a UDP in IPv6 and check if IPv6 source and destination \
+| | ... | addresses are correctly translated into IPv4 addresses.
+| | ...
+| | ... | *Arguments:*
+| | ... | - tg_node - Node where to run traffic script. Type: string
+| | ... | - tx_if - Interface from where to send IPv4 UDP packet. Type: string
+| | ... | - rx_if - Interface where to receive IPv6 UDP packet. Type: string
+| | ... | - tx_dst_mac - Destination MAC address of IPv4 packet. Type: string
+| | ... | - tx_dst_ipv6 - Destination IPv6 address. Type: string
+| | ... | - tx_src_ipv6 - Source IPv6 address. Type: string
+| | ... | - tx_src_udp_port - Source UDP port. Type: integer
+| | ... | - rx_dst_mac - Expected destination MAC address. Type: string
+| | ... | - rx_src_mac - Expected source MAC address. Type: string
+| | ... | - dst_ipv4 - Expected destination IPv4 address. Type: string
+| | ... | - src_ipv4 - Expected source IPv4 address. Type: string
+| | ...
+| | ... | *Return:*
+| | ... | - No value returned
+| | ...
+| | ... | *Example:*
+| | ...
+| | ... | \| Send IPv6 UDP and check IPv4 headers for MAP-T \
+| | ... | \| port3 \| port4 \| 08:00:27:f3:be:f0 \| 2001:db8:ffff::6400:1 \
+| | ... | \| 2001:db8::14a9:c9db:0 \| ${1232} \| 08:00:27:58:71:eb \
+| | ... | \| 08:00:27:66:b8:57 \| 100.0.0.1 \| 20.169.201.219 \|
+| | ...
+| | [Arguments]
+| | ... | ${tg_node} | ${tx_if} | ${rx_if}
+| | ... | ${tx_dst_mac} | ${tx_dst_ipv6} | ${tx_src_ipv6} | ${tx_src_udp_port}
+| | ... | ${rx_dst_mac} | ${rx_src_mac} | ${dst_ipv4} | ${src_ipv4}
+| | ...
+| | ${tx_name}= | Get interface name | ${tg_node} | ${tx_if}
+| | ${rx_name}= | Get interface name | ${tg_node} | ${rx_if}
+| | ${args}= | Catenate
+| | ... | --tx_if | ${tx_name} | --rx_if | ${rx_name}
+| | ... | --tx_dst_mac | ${tx_dst_mac}
+| | ... | --tx_src_ipv6 | ${tx_src_ipv6} | --tx_dst_ipv6 | ${tx_dst_ipv6}
+| | ... | --tx_src_udp_port | ${tx_src_udp_port}
+| | ... | --rx_dst_mac | ${rx_dst_mac} | --rx_src_mac | ${rx_src_mac}
+| | ... | --rx_src_ipv4 | ${src_ipv4} | --rx_dst_ipv4 | ${dst_ipv4}
+| | ...
+| | Run Traffic Script On Node
+| | ... | send_ipv6_udp_check_map_t.py | ${tg_node} | ${args}