Add support for MAP-T CE (VPP-1058)

Change-Id: Ib99058323c50f20fcdb44185aee027c5c6ef8c91
Signed-off-by: Juraj Sloboda <jsloboda@cisco.com>

1 files changed, 135 insertions, 49 deletions

diff --git a/src/vnet/map/ip4_map_t.c b/src/vnet/map/ip4_map_t.c
index b89840ccfaf..0a1dc8f3212 100644
--- a/src/vnet/map/ip4_map_t.c
+++ b/src/vnet/map/ip4_map_t.c
@@ -102,19 +102,33 @@ ip4_map_fragment_get_port (ip4_header_t * ip4)
 typedef struct
 {
   map_domain_t *d;
-  u16 recv_port;
+  u16 id;
 } icmp_to_icmp6_ctx_t;
 
 static int
 ip4_to_ip6_set_icmp_cb (ip4_header_t * ip4, ip6_header_t * ip6, void *arg)
 {
   icmp_to_icmp6_ctx_t *ctx = arg;
+  map_main_t *mm = &map_main;
 
-  ip4_map_t_embedded_address (ctx->d, &ip6->src_address, &ip4->src_address);
-  ip6->dst_address.as_u64[0] =
-    map_get_pfx_net (ctx->d, ip4->dst_address.as_u32, ctx->recv_port);
-  ip6->dst_address.as_u64[1] =
-    map_get_sfx_net (ctx->d, ip4->dst_address.as_u32, ctx->recv_port);
+  if (mm->is_ce)
+    {
+      ip6->src_address.as_u64[0] =
+	map_get_pfx_net (ctx->d, ip4->src_address.as_u32, ctx->id);
+      ip6->src_address.as_u64[1] =
+	map_get_sfx_net (ctx->d, ip4->src_address.as_u32, ctx->id);
+      ip4_map_t_embedded_address (ctx->d, &ip6->dst_address,
+				  &ip4->dst_address);
+    }
+  else
+    {
+      ip4_map_t_embedded_address (ctx->d, &ip6->src_address,
+				  &ip4->src_address);
+      ip6->dst_address.as_u64[0] =
+	map_get_pfx_net (ctx->d, ip4->dst_address.as_u32, ctx->id);
+      ip6->dst_address.as_u64[1] =
+	map_get_sfx_net (ctx->d, ip4->dst_address.as_u32, ctx->id);
+    }
 
   return 0;
 }
@@ -124,14 +138,30 @@ ip4_to_ip6_set_inner_icmp_cb (ip4_header_t * ip4, ip6_header_t * ip6,
 			      void *arg)
 {
   icmp_to_icmp6_ctx_t *ctx = arg;
+  map_main_t *mm = &map_main;
 
-  //Note that the source address is within the domain
-  //while the destination address is the one outside the domain
-  ip4_map_t_embedded_address (ctx->d, &ip6->dst_address, &ip4->dst_address);
-  ip6->src_address.as_u64[0] =
-    map_get_pfx_net (ctx->d, ip4->src_address.as_u32, ctx->recv_port);
-  ip6->src_address.as_u64[1] =
-    map_get_sfx_net (ctx->d, ip4->src_address.as_u32, ctx->recv_port);
+  if (mm->is_ce)
+    {
+      //Note that the destination address is within the domain
+      //while the source address is the one outside the domain
+      ip4_map_t_embedded_address (ctx->d, &ip6->src_address,
+				  &ip4->src_address);
+      ip6->dst_address.as_u64[0] =
+	map_get_pfx_net (ctx->d, ip4->dst_address.as_u32, ctx->id);
+      ip6->dst_address.as_u64[1] =
+	map_get_sfx_net (ctx->d, ip4->dst_address.as_u32, ctx->id);
+    }
+  else
+    {
+      //Note that the source address is within the domain
+      //while the destination address is the one outside the domain
+      ip4_map_t_embedded_address (ctx->d, &ip6->dst_address,
+				  &ip4->dst_address);
+      ip6->src_address.as_u64[0] =
+	map_get_pfx_net (ctx->d, ip4->src_address.as_u32, ctx->id);
+      ip6->src_address.as_u64[1] =
+	map_get_sfx_net (ctx->d, ip4->src_address.as_u32, ctx->id);
+    }
 
   return 0;
 }
@@ -163,6 +193,7 @@ ip4_map_t_icmp (vlib_main_t * vm,
 	  u16 len0;
 	  icmp_to_icmp6_ctx_t ctx0;
 	  ip4_header_t *ip40;
+	  icmp46_header_t *icmp0;
 
 	  next0 = IP4_MAPT_ICMP_NEXT_IP6_LOOKUP;
 	  pi0 = to_next[0] = from[0];
@@ -182,9 +213,11 @@ ip4_map_t_icmp (vlib_main_t * vm,
 			       vnet_buffer (p0)->map_t.map_domain_index);
 
 	  ip40 = vlib_buffer_get_current (p0);
-	  ctx0.recv_port = ip4_get_port (ip40, 1);
+	  icmp0 = (icmp46_header_t *) (ip40 + 1);
+
+	  ctx0.id = ip4_get_port (ip40, icmp0->type == ICMP6_echo_request);
 	  ctx0.d = d0;
-	  if (ctx0.recv_port == 0)
+	  if (ctx0.id == 0)
 	    {
 	      // In case of 1:1 mapping, we don't care about the port
 	      if (!(d0->ea_bits_len == 0 && d0->rules))
@@ -442,6 +475,14 @@ ip4_map_t_classify (vlib_buffer_t * p0, map_domain_t * d0,
 		    ip4_header_t * ip40, u16 ip4_len0, i32 * dst_port0,
 		    u8 * error0, ip4_mapt_next_t * next0)
 {
+  map_main_t *mm = &map_main;
+  u32 port_offset;
+
+  if (mm->is_ce)
+    port_offset = 0;
+  else
+    port_offset = 2;
+
   if (PREDICT_FALSE (ip4_get_fragment_offset (ip40)))
     {
       *next0 = IP4_MAPT_NEXT_MAPT_FRAGMENTED;
@@ -460,14 +501,16 @@ ip4_map_t_classify (vlib_buffer_t * p0, map_domain_t * d0,
       vnet_buffer (p0)->map_t.checksum_offset = 36;
       *next0 = IP4_MAPT_NEXT_MAPT_TCP_UDP;
       *error0 = ip4_len0 < 40 ? MAP_ERROR_MALFORMED : *error0;
-      *dst_port0 = (i32) * ((u16 *) u8_ptr_add (ip40, sizeof (*ip40) + 2));
+      *dst_port0 =
+	(i32) * ((u16 *) u8_ptr_add (ip40, sizeof (*ip40) + port_offset));
     }
   else if (PREDICT_TRUE (ip40->protocol == IP_PROTOCOL_UDP))
     {
       vnet_buffer (p0)->map_t.checksum_offset = 26;
       *next0 = IP4_MAPT_NEXT_MAPT_TCP_UDP;
       *error0 = ip4_len0 < 28 ? MAP_ERROR_MALFORMED : *error0;
-      *dst_port0 = (i32) * ((u16 *) u8_ptr_add (ip40, sizeof (*ip40) + 2));
+      *dst_port0 =
+	(i32) * ((u16 *) u8_ptr_add (ip40, sizeof (*ip40) + port_offset));
     }
   else if (ip40->protocol == IP_PROTOCOL_ICMP)
     {
@@ -496,6 +539,7 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
   from = vlib_frame_vector_args (frame);
   n_left_from = frame->n_vectors;
   next_index = node->cached_next_index;
+  map_main_t *mm = &map_main;
   vlib_combined_counter_main_t *cm = map_main.domain_counters;
   u32 thread_index = vlib_get_thread_index ();
 
@@ -513,7 +557,7 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 	  ip4_mapt_next_t next0 = 0, next1 = 0;
 	  u16 ip4_len0, ip4_len1;
 	  u8 error0, error1;
-	  i32 dst_port0, dst_port1;
+	  i32 map_port0, map_port1;
 	  ip4_mapt_pseudo_header_t *pheader0, *pheader1;
 
 	  pi0 = to_next[0] = from[0];
@@ -556,12 +600,12 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 	  vnet_buffer (p0)->map_t.mtu = d0->mtu ? d0->mtu : ~0;
 	  vnet_buffer (p1)->map_t.mtu = d1->mtu ? d1->mtu : ~0;
 
-	  dst_port0 = -1;
-	  dst_port1 = -1;
+	  map_port0 = -1;
+	  map_port1 = -1;
 
-	  ip4_map_t_classify (p0, d0, ip40, ip4_len0, &dst_port0, &error0,
+	  ip4_map_t_classify (p0, d0, ip40, ip4_len0, &map_port0, &error0,
 			      &next0);
-	  ip4_map_t_classify (p1, d1, ip41, ip4_len1, &dst_port1, &error1,
+	  ip4_map_t_classify (p1, d1, ip41, ip4_len1, &map_port1, &error1,
 			      &next1);
 
 	  //Add MAP-T pseudo header in front of the packet
@@ -571,31 +615,57 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 	  pheader1 = vlib_buffer_get_current (p1);
 
 	  //Save addresses within the packet
-	  ip4_map_t_embedded_address (d0, &pheader0->saddr,
-				      &ip40->src_address);
-	  ip4_map_t_embedded_address (d1, &pheader1->saddr,
-				      &ip41->src_address);
-	  pheader0->daddr.as_u64[0] =
-	    map_get_pfx_net (d0, ip40->dst_address.as_u32, (u16) dst_port0);
-	  pheader0->daddr.as_u64[1] =
-	    map_get_sfx_net (d0, ip40->dst_address.as_u32, (u16) dst_port0);
-	  pheader1->daddr.as_u64[0] =
-	    map_get_pfx_net (d1, ip41->dst_address.as_u32, (u16) dst_port1);
-	  pheader1->daddr.as_u64[1] =
-	    map_get_sfx_net (d1, ip41->dst_address.as_u32, (u16) dst_port1);
+	  if (mm->is_ce)
+	    {
+	      ip4_map_t_embedded_address (d0, &pheader0->daddr,
+					  &ip40->dst_address);
+	      ip4_map_t_embedded_address (d1, &pheader1->daddr,
+					  &ip41->dst_address);
+	      pheader0->saddr.as_u64[0] =
+		map_get_pfx_net (d0, ip40->src_address.as_u32,
+				 (u16) map_port0);
+	      pheader0->saddr.as_u64[1] =
+		map_get_sfx_net (d0, ip40->src_address.as_u32,
+				 (u16) map_port0);
+	      pheader1->saddr.as_u64[0] =
+		map_get_pfx_net (d1, ip41->src_address.as_u32,
+				 (u16) map_port1);
+	      pheader1->saddr.as_u64[1] =
+		map_get_sfx_net (d1, ip41->src_address.as_u32,
+				 (u16) map_port1);
+	    }
+	  else
+	    {
+	      ip4_map_t_embedded_address (d0, &pheader0->saddr,
+					  &ip40->src_address);
+	      ip4_map_t_embedded_address (d1, &pheader1->saddr,
+					  &ip41->src_address);
+	      pheader0->daddr.as_u64[0] =
+		map_get_pfx_net (d0, ip40->dst_address.as_u32,
+				 (u16) map_port0);
+	      pheader0->daddr.as_u64[1] =
+		map_get_sfx_net (d0, ip40->dst_address.as_u32,
+				 (u16) map_port0);
+	      pheader1->daddr.as_u64[0] =
+		map_get_pfx_net (d1, ip41->dst_address.as_u32,
+				 (u16) map_port1);
+	      pheader1->daddr.as_u64[1] =
+		map_get_sfx_net (d1, ip41->dst_address.as_u32,
+				 (u16) map_port1);
+	    }
 
 	  if (PREDICT_FALSE
-	      (ip4_is_first_fragment (ip40) && (dst_port0 != -1)
+	      (ip4_is_first_fragment (ip40) && (map_port0 != -1)
 	       && (d0->ea_bits_len != 0 || !d0->rules)
-	       && ip4_map_fragment_cache (ip40, dst_port0)))
+	       && ip4_map_fragment_cache (ip40, map_port0)))
 	    {
 	      error0 = MAP_ERROR_FRAGMENT_MEMORY;
 	    }
 
 	  if (PREDICT_FALSE
-	      (ip4_is_first_fragment (ip41) && (dst_port1 != -1)
+	      (ip4_is_first_fragment (ip41) && (map_port1 != -1)
 	       && (d1->ea_bits_len != 0 || !d1->rules)
-	       && ip4_map_fragment_cache (ip41, dst_port1)))
+	       && ip4_map_fragment_cache (ip41, map_port1)))
 	    {
 	      error1 = MAP_ERROR_FRAGMENT_MEMORY;
 	    }
@@ -641,7 +711,7 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 	  ip4_mapt_next_t next0;
 	  u16 ip4_len0;
 	  u8 error0;
-	  i32 dst_port0;
+	  i32 map_port0;
 	  ip4_mapt_pseudo_header_t *pheader0;
 
 	  pi0 = to_next[0] = from[0];
@@ -667,8 +737,8 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 
 	  vnet_buffer (p0)->map_t.mtu = d0->mtu ? d0->mtu : ~0;
 
-	  dst_port0 = -1;
-	  ip4_map_t_classify (p0, d0, ip40, ip4_len0, &dst_port0, &error0,
+	  map_port0 = -1;
+	  ip4_map_t_classify (p0, d0, ip40, ip4_len0, &map_port0, &error0,
 			      &next0);
 
 	  //Add MAP-T pseudo header in front of the packet
@@ -676,20 +746,36 @@ ip4_map_t (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame)
 	  pheader0 = vlib_buffer_get_current (p0);
 
 	  //Save addresses within the packet
-	  ip4_map_t_embedded_address (d0, &pheader0->saddr,
-				      &ip40->src_address);
-	  pheader0->daddr.as_u64[0] =
-	    map_get_pfx_net (d0, ip40->dst_address.as_u32, (u16) dst_port0);
-	  pheader0->daddr.as_u64[1] =
-	    map_get_sfx_net (d0, ip40->dst_address.as_u32, (u16) dst_port0);
+	  if (mm->is_ce)
+	    {
+	      ip4_map_t_embedded_address (d0, &pheader0->daddr,
+					  &ip40->dst_address);
+	      pheader0->saddr.as_u64[0] =
+		map_get_pfx_net (d0, ip40->src_address.as_u32,
+				 (u16) map_port0);
+	      pheader0->saddr.as_u64[1] =
+		map_get_sfx_net (d0, ip40->src_address.as_u32,
+				 (u16) map_port0);
+	    }
+	  else
+	    {
+	      ip4_map_t_embedded_address (d0, &pheader0->saddr,
+					  &ip40->src_address);
+	      pheader0->daddr.as_u64[0] =
+		map_get_pfx_net (d0, ip40->dst_address.as_u32,
+				 (u16) map_port0);
+	      pheader0->daddr.as_u64[1] =
+		map_get_sfx_net (d0, ip40->dst_address.as_u32,
+				 (u16) map_port0);
+	    }
 
 	  //It is important to cache at this stage because the result might be necessary
 	  //for packets within the same vector.
 	  //Actually, this approach even provides some limited out-of-order fragments support
 	  if (PREDICT_FALSE
-	      (ip4_is_first_fragment (ip40) && (dst_port0 != -1)
+	      (ip4_is_first_fragment (ip40) && (map_port0 != -1)
 	       && (d0->ea_bits_len != 0 || !d0->rules)
-	       && ip4_map_fragment_cache (ip40, dst_port0)))
+	       && ip4_map_fragment_cache (ip40, map_port0)))
 	    {
 	      error0 = MAP_ERROR_UNKNOWN;
 	    }