Add support for MAP-T CE (VPP-1058)
[vpp.git] / src / vnet / map / ip4_map_t.c
index b89840c..0a1dc8f 100644 (file)
@@ -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;
            }