vnet/vnet/map/ip4_map.c

   1 /*
   2  * Copyright (c) 2015 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 /*
  16  * Defines used for testing various optimisation schemes
  17  */
  18 #define MAP_ENCAP_DUAL 0
  19
  20 #include "map.h"
  21 #include "../ip/ip_frag.h"
  22
  23 vlib_node_registration_t ip4_map_reass_node;
  24
  25 enum ip4_map_next_e {
  26   IP4_MAP_NEXT_IP6_LOOKUP,
  27 #ifdef MAP_SKIP_IP6_LOOKUP
  28   IP4_MAP_NEXT_IP6_REWRITE,
  29 #endif
  30   IP4_MAP_NEXT_IP4_FRAGMENT,
  31   IP4_MAP_NEXT_IP6_FRAGMENT,
  32   IP4_MAP_NEXT_REASS,
  33   IP4_MAP_NEXT_ICMP_ERROR,
  34   IP4_MAP_NEXT_DROP,
  35   IP4_MAP_N_NEXT,
  36 };
  37
  38 enum ip4_map_reass_next_t {
  39   IP4_MAP_REASS_NEXT_IP6_LOOKUP,
  40   IP4_MAP_REASS_NEXT_IP4_FRAGMENT,
  41   IP4_MAP_REASS_NEXT_DROP,
  42   IP4_MAP_REASS_N_NEXT,
  43 };
  44
  45 typedef struct {
  46   u32 map_domain_index;
  47   u16 port;
  48   u8 cached;
  49 } map_ip4_map_reass_trace_t;
  50
  51 u8 *
  52 format_ip4_map_reass_trace (u8 *s, va_list *args)
  53 {
  54   CLIB_UNUSED(vlib_main_t *vm) = va_arg (*args, vlib_main_t *);
  55   CLIB_UNUSED(vlib_node_t *node) = va_arg (*args, vlib_node_t *);
  56   map_ip4_map_reass_trace_t *t = va_arg (*args, map_ip4_map_reass_trace_t *);
  57   return format(s, "MAP domain index: %d L4 port: %u Status: %s", t->map_domain_index,
  58                 t->port, t->cached?"cached":"forwarded");
  59 }
  60
  61 /*
  62  * ip4_map_get_port
  63  */
  64 u16
  65 ip4_map_get_port (ip4_header_t *ip, map_dir_e dir)
  66 {
  67   /* Find port information */
  68   if (PREDICT_TRUE((ip->protocol == IP_PROTOCOL_TCP) ||
  69                    (ip->protocol == IP_PROTOCOL_UDP))) {
  70     udp_header_t *udp = (void *)(ip + 1);
  71     return (dir == MAP_SENDER ? udp->src_port : udp->dst_port);
  72   } else if (ip->protocol == IP_PROTOCOL_ICMP) {
  73     /*
  74      * 1) ICMP Echo request or Echo reply
  75      * 2) ICMP Error with inner packet being UDP or TCP
  76      * 3) ICMP Error with inner packet being ICMP Echo request or Echo reply
  77      */
  78     icmp46_header_t *icmp = (void *)(ip + 1);
  79     if (icmp->type == ICMP4_echo_request || icmp->type == ICMP4_echo_reply) {
  80       return *((u16 *)(icmp + 1));
  81     } else if (clib_net_to_host_u16(ip->length) >= 56) { // IP + ICMP + IP + L4 header
  82       ip4_header_t *icmp_ip = (ip4_header_t *)(icmp + 2);
  83       if (PREDICT_TRUE((icmp_ip->protocol == IP_PROTOCOL_TCP) ||
  84                        (icmp_ip->protocol == IP_PROTOCOL_UDP))) {
  85         udp_header_t *udp = (void *)(icmp_ip + 1);
  86         return (dir == MAP_SENDER ? udp->dst_port : udp->src_port);
  87       } else if (icmp_ip->protocol == IP_PROTOCOL_ICMP) {
  88         icmp46_header_t *inner_icmp = (void *)(icmp_ip + 1);
  89         if (inner_icmp->type == ICMP4_echo_request || inner_icmp->type == ICMP4_echo_reply)
  90           return (*((u16 *)(inner_icmp + 1)));
  91       }
  92     }
  93   }
  94   return (0);
  95 }
  96
  97 static_always_inline u16
  98 ip4_map_port_and_security_check (map_domain_t *d, ip4_header_t *ip, u32 *next, u8 *error)
  99 {
 100   u16 port = 0;
 101
 102   if (d->psid_length > 0) {
 103     if (!ip4_is_fragment(ip)) {
 104       if (PREDICT_FALSE((ip->ip_version_and_header_length != 0x45) || clib_host_to_net_u16(ip->length) < 28)) {
 105         return 0;
 106       }
 107       port = ip4_map_get_port(ip, MAP_RECEIVER);
 108       if (port) {
 109         /* Verify that port is not among the well-known ports */
 110         if ((d->psid_offset > 0) && (clib_net_to_host_u16(port) < (0x1 << (16 - d->psid_offset)))) {
 111           *error = MAP_ERROR_ENCAP_SEC_CHECK;
 112         } else {
 113           return (port);
 114         }
 115       } else {
 116         *error = MAP_ERROR_BAD_PROTOCOL;
 117       }
 118     } else {
 119       *next = IP4_MAP_NEXT_REASS;
 120     }
 121   }
 122   return (0);
 123 }
 124
 125 /*
 126  * ip4_map_vtcfl
 127  */
 128 static_always_inline u32
 129 ip4_map_vtcfl (ip4_header_t *ip4, vlib_buffer_t *p)
 130 {
 131   map_main_t *mm = &map_main;
 132   u8 tc = mm->tc_copy ? ip4->tos : mm->tc;
 133   u32 vtcfl = 0x6 << 28;
 134   vtcfl |= tc << 20;
 135   vtcfl |= vnet_buffer(p)->ip.flow_hash & 0x000fffff;
 136
 137   return (clib_host_to_net_u32(vtcfl));
 138 }
 139
 140 static_always_inline bool
 141 ip4_map_ip6_lookup_bypass (vlib_buffer_t *p0, ip4_header_t *ip)
 142 {
 143 #ifdef MAP_SKIP_IP6_LOOKUP
 144   map_main_t *mm = &map_main;
 145   u32 adj_index0 = mm->adj6_index;
 146   if (adj_index0 > 0) {
 147     ip_lookup_main_t *lm6 = &ip6_main.lookup_main;
 148     ip_adjacency_t *adj = ip_get_adjacency(lm6, mm->adj6_index);
 149     if (adj->n_adj > 1) {
 150       u32 hash_c0 = ip4_compute_flow_hash(ip, IP_FLOW_HASH_DEFAULT);
 151       adj_index0 += (hash_c0 & (adj->n_adj - 1));
 152     }
 153     vnet_buffer(p0)->ip.adj_index[VLIB_TX] = adj_index0;
 154     return (true);
 155   }
 156 #endif
 157   return (false);
 158 }
 159
 160 /*
 161  * ip4_map_ttl
 162  */
 163 static inline void
 164 ip4_map_decrement_ttl (ip4_header_t *ip, u8 *error)
 165 {
 166   i32 ttl = ip->ttl;
 167
 168   /* Input node should have reject packets with ttl 0. */
 169   ASSERT (ip->ttl > 0);
 170
 171   u32 checksum = ip->checksum + clib_host_to_net_u16(0x0100);
 172   checksum += checksum >= 0xffff;
 173   ip->checksum = checksum;
 174   ttl -= 1;
 175   ip->ttl = ttl;
 176   *error = ttl <= 0 ? IP4_ERROR_TIME_EXPIRED : *error;
 177
 178   /* Verify checksum. */
 179   ASSERT (ip->checksum == ip4_header_checksum(ip));
 180 }
 181
 182 static u32
 183 ip4_map_fragment (vlib_buffer_t *b, u16 mtu, bool df, u8 *error)
 184 {
 185   map_main_t *mm = &map_main;
 186
 187   if (mm->frag_inner) {
 188     ip_frag_set_vnet_buffer(b, sizeof(ip6_header_t), mtu, IP4_FRAG_NEXT_IP6_LOOKUP, IP_FRAG_FLAG_IP6_HEADER);
 189     return (IP4_MAP_NEXT_IP4_FRAGMENT);
 190   } else {
 191     if (df && !mm->frag_ignore_df) {
 192       icmp4_error_set_vnet_buffer(b, ICMP4_destination_unreachable,
 193                                   ICMP4_destination_unreachable_fragmentation_needed_and_dont_fragment_set, mtu);
 194       vlib_buffer_advance(b, sizeof(ip6_header_t));
 195       *error = MAP_ERROR_DF_SET;
 196       return (IP4_MAP_NEXT_ICMP_ERROR);
 197     }
 198     ip_frag_set_vnet_buffer(b, 0, mtu, IP6_FRAG_NEXT_IP6_LOOKUP, IP_FRAG_FLAG_IP6_HEADER);
 199     return (IP4_MAP_NEXT_IP6_FRAGMENT);
 200   }
 201 }
 202
 203 /*
 204  * ip4_map
 205  */
 206 static uword
 207 ip4_map (vlib_main_t *vm,
 208          vlib_node_runtime_t *node,
 209          vlib_frame_t *frame)
 210 {
 211   u32 n_left_from, *from, next_index, *to_next, n_left_to_next;
 212   vlib_node_runtime_t *error_node = vlib_node_get_runtime(vm, ip4_map_node.index);
 213   from = vlib_frame_vector_args(frame);
 214   n_left_from = frame->n_vectors;
 215   next_index = node->cached_next_index;
 216   map_main_t *mm = &map_main;
 217   vlib_combined_counter_main_t *cm = mm->domain_counters;
 218   u32 cpu_index = os_get_cpu_number();
 219
 220   while (n_left_from > 0) {
 221     vlib_get_next_frame(vm, node, next_index, to_next, n_left_to_next);
 222
 223     /* Dual loop */
 224     while (n_left_from >= 4 && n_left_to_next >= 2) {
 225       u32 pi0, pi1;
 226       vlib_buffer_t *p0, *p1;
 227       map_domain_t *d0, *d1;
 228       u8 error0 = MAP_ERROR_NONE, error1 = MAP_ERROR_NONE;
 229       ip4_header_t *ip40, *ip41;
 230       u16 port0 = 0, port1 = 0;
 231       ip6_header_t *ip6h0, *ip6h1;
 232       u32 map_domain_index0 = ~0, map_domain_index1 = ~0;
 233       u32 next0 = IP4_MAP_NEXT_IP6_LOOKUP, next1 = IP4_MAP_NEXT_IP6_LOOKUP;
 234
 235       /* Prefetch next iteration. */
 236       {
 237         vlib_buffer_t *p2, *p3;
 238
 239         p2 = vlib_get_buffer(vm, from[2]);
 240         p3 = vlib_get_buffer(vm, from[3]);
 241
 242         vlib_prefetch_buffer_header(p2, STORE);
 243         vlib_prefetch_buffer_header(p3, STORE);
 244         /* IPv4 + 8 = 28. possibly plus -40 */
 245         CLIB_PREFETCH (p2->data-40, 68, STORE);
 246         CLIB_PREFETCH (p3->data-40, 68, STORE);
 247       }
 248
 249       pi0 = to_next[0] = from[0];
 250       pi1 = to_next[1] = from[1];
 251       from += 2;
 252       n_left_from -= 2;
 253       to_next +=2;
 254       n_left_to_next -= 2;
 255
 256       p0 = vlib_get_buffer(vm, pi0);
 257       p1 = vlib_get_buffer(vm, pi1);
 258       ip40 = vlib_buffer_get_current(p0);
 259       ip41 = vlib_buffer_get_current(p1);
 260       p0->current_length = clib_net_to_host_u16(ip40->length);
 261       p1->current_length = clib_net_to_host_u16(ip41->length);
 262       d0 = ip4_map_get_domain(vnet_buffer(p0)->ip.adj_index[VLIB_TX], &map_domain_index0);
 263       d1 = ip4_map_get_domain(vnet_buffer(p1)->ip.adj_index[VLIB_TX], &map_domain_index1);
 264       ASSERT(d0);
 265       ASSERT(d1);
 266
 267       /*
 268        * Shared IPv4 address
 269        */
 270       port0 = ip4_map_port_and_security_check(d0, ip40, &next0, &error0);
 271       port1 = ip4_map_port_and_security_check(d1, ip41, &next1, &error1);
 272
 273       /* Decrement IPv4 TTL */
 274       ip4_map_decrement_ttl(ip40, &error0);
 275       ip4_map_decrement_ttl(ip41, &error1);
 276       bool df0 = ip40->flags_and_fragment_offset & clib_host_to_net_u16(IP4_HEADER_FLAG_DONT_FRAGMENT);
 277       bool df1 = ip41->flags_and_fragment_offset & clib_host_to_net_u16(IP4_HEADER_FLAG_DONT_FRAGMENT);
 278
 279       /* MAP calc */
 280       u32 da40 = clib_net_to_host_u32(ip40->dst_address.as_u32);
 281       u32 da41 = clib_net_to_host_u32(ip41->dst_address.as_u32);
 282       u16 dp40 = clib_net_to_host_u16(port0);
 283       u16 dp41 = clib_net_to_host_u16(port1);
 284       u64 dal60 = map_get_pfx(d0, da40, dp40);
 285       u64 dal61 = map_get_pfx(d1, da41, dp41);
 286       u64 dar60 = map_get_sfx(d0, da40, dp40);
 287       u64 dar61 = map_get_sfx(d1, da41, dp41);
 288       if (dal60 == 0 && dar60 == 0) error0 = MAP_ERROR_NO_BINDING;
 289       if (dal61 == 0 && dar61 == 0) error1 = MAP_ERROR_NO_BINDING;
 290
 291       /* construct ipv6 header */
 292       vlib_buffer_advance(p0, - sizeof(ip6_header_t));
 293       vlib_buffer_advance(p1, - sizeof(ip6_header_t));
 294       ip6h0 = vlib_buffer_get_current(p0);
 295       ip6h1 = vlib_buffer_get_current(p1);
 296       vnet_buffer(p0)->sw_if_index[VLIB_TX] = (u32)~0;
 297       vnet_buffer(p1)->sw_if_index[VLIB_TX] = (u32)~0;
 298
 299       ip6h0->ip_version_traffic_class_and_flow_label = ip4_map_vtcfl(ip40, p0);
 300       ip6h1->ip_version_traffic_class_and_flow_label = ip4_map_vtcfl(ip41, p1);
 301       ip6h0->payload_length = ip40->length;
 302       ip6h1->payload_length = ip41->length;
 303       ip6h0->protocol = IP_PROTOCOL_IP_IN_IP;
 304       ip6h1->protocol = IP_PROTOCOL_IP_IN_IP;
 305       ip6h0->hop_limit = 0x40;
 306       ip6h1->hop_limit = 0x40;
 307       ip6h0->src_address = d0->ip6_src;
 308       ip6h1->src_address = d1->ip6_src;
 309       ip6h0->dst_address.as_u64[0] = clib_host_to_net_u64(dal60);
 310       ip6h0->dst_address.as_u64[1] = clib_host_to_net_u64(dar60);
 311       ip6h1->dst_address.as_u64[0] = clib_host_to_net_u64(dal61);
 312       ip6h1->dst_address.as_u64[1] = clib_host_to_net_u64(dar61);
 313
 314       /*
 315        * Determine next node. Can be one of:
 316        * ip6-lookup, ip6-rewrite, ip4-fragment, ip4-virtreass, error-drop
 317        */
 318       if (PREDICT_TRUE(error0 == MAP_ERROR_NONE)) {
 319         if (PREDICT_FALSE(d0->mtu && (clib_net_to_host_u16(ip6h0->payload_length) + sizeof(*ip6h0) > d0->mtu))) {
 320           next0 = ip4_map_fragment(p0, d0->mtu, df0, &error0);
 321         } else {
 322           next0 = ip4_map_ip6_lookup_bypass(p0, ip40) ? IP4_MAP_NEXT_IP6_REWRITE : next0;
 323           vlib_increment_combined_counter(cm + MAP_DOMAIN_COUNTER_TX, cpu_index, map_domain_index0, 1,
 324                                           clib_net_to_host_u16(ip6h0->payload_length) + 40);
 325         }
 326       } else {
 327         next0 = IP4_MAP_NEXT_DROP;
 328       }
 329
 330       /*
 331        * Determine next node. Can be one of:
 332        * ip6-lookup, ip6-rewrite, ip4-fragment, ip4-virtreass, error-drop
 333        */
 334       if (PREDICT_TRUE(error1 == MAP_ERROR_NONE)) {
 335         if (PREDICT_FALSE(d1->mtu && (clib_net_to_host_u16(ip6h1->payload_length) + sizeof(*ip6h1) > d1->mtu))) {
 336           next1 = ip4_map_fragment(p1, d1->mtu, df1, &error1);
 337         } else {
 338           next1 = ip4_map_ip6_lookup_bypass(p1, ip41) ? IP4_MAP_NEXT_IP6_REWRITE : next1;
 339           vlib_increment_combined_counter(cm + MAP_DOMAIN_COUNTER_TX, cpu_index, map_domain_index1, 1,
 340                                           clib_net_to_host_u16(ip6h1->payload_length) + 40);
 341         }
 342       } else {
 343         next1 = IP4_MAP_NEXT_DROP;
 344       }
 345
 346       if (PREDICT_FALSE(p0->flags & VLIB_BUFFER_IS_TRACED)) {
 347         map_trace_t *tr = vlib_add_trace(vm, node, p0, sizeof(*tr));
 348         tr->map_domain_index = map_domain_index0;
 349         tr->port = port0;
 350       }
 351       if (PREDICT_FALSE(p1->flags & VLIB_BUFFER_IS_TRACED)) {
 352         map_trace_t *tr = vlib_add_trace(vm, node, p1, sizeof(*tr));
 353         tr->map_domain_index = map_domain_index1;
 354         tr->port = port1;
 355       }
 356
 357       p0->error = error_node->errors[error0];
 358       p1->error = error_node->errors[error1];
 359
 360       vlib_validate_buffer_enqueue_x2(vm, node, next_index, to_next, n_left_to_next, pi0, pi1, next0, next1);
 361     }
 362
 363     while (n_left_from > 0 && n_left_to_next > 0) {
 364       u32 pi0;
 365       vlib_buffer_t *p0;
 366       map_domain_t *d0;
 367       u8 error0 = MAP_ERROR_NONE;
 368       ip4_header_t *ip40;
 369       u16 port0 = 0;
 370       ip6_header_t *ip6h0;
 371       u32 next0 = IP4_MAP_NEXT_IP6_LOOKUP;
 372       u32 map_domain_index0 = ~0;
 373
 374       pi0 = to_next[0] = from[0];
 375       from += 1;
 376       n_left_from -= 1;
 377       to_next +=1;
 378       n_left_to_next -= 1;
 379
 380       p0 = vlib_get_buffer(vm, pi0);
 381       ip40 = vlib_buffer_get_current(p0);
 382       p0->current_length = clib_net_to_host_u16(ip40->length);
 383       d0 = ip4_map_get_domain(vnet_buffer(p0)->ip.adj_index[VLIB_TX], &map_domain_index0);
 384       ASSERT(d0);
 385
 386       /*
 387        * Shared IPv4 address
 388        */
 389       port0 = ip4_map_port_and_security_check(d0, ip40, &next0, &error0);
 390
 391       /* Decrement IPv4 TTL */
 392       ip4_map_decrement_ttl(ip40, &error0);
 393       bool df0 = ip40->flags_and_fragment_offset & clib_host_to_net_u16(IP4_HEADER_FLAG_DONT_FRAGMENT);
 394
 395       /* MAP calc */
 396       u32 da40 = clib_net_to_host_u32(ip40->dst_address.as_u32);
 397       u16 dp40 = clib_net_to_host_u16(port0);
 398       u64 dal60 = map_get_pfx(d0, da40, dp40);
 399       u64 dar60 = map_get_sfx(d0, da40, dp40);
 400       if (dal60 == 0 && dar60 == 0 && error0 == MAP_ERROR_NONE) error0 = MAP_ERROR_NO_BINDING;
 401
 402       /* construct ipv6 header */
 403       vlib_buffer_advance(p0, - (sizeof(ip6_header_t)));
 404       ip6h0 = vlib_buffer_get_current(p0);
 405       vnet_buffer(p0)->sw_if_index[VLIB_TX] = (u32)~0;
 406
 407       ip6h0->ip_version_traffic_class_and_flow_label = ip4_map_vtcfl(ip40, p0);
 408       ip6h0->payload_length = ip40->length;
 409       ip6h0->protocol = IP_PROTOCOL_IP_IN_IP;
 410       ip6h0->hop_limit = 0x40;
 411       ip6h0->src_address = d0->ip6_src;
 412       ip6h0->dst_address.as_u64[0] = clib_host_to_net_u64(dal60);
 413       ip6h0->dst_address.as_u64[1] = clib_host_to_net_u64(dar60);
 414
 415       /*
 416        * Determine next node. Can be one of:
 417        * ip6-lookup, ip6-rewrite, ip4-fragment, ip4-virtreass, error-drop
 418        */
 419       if (PREDICT_TRUE(error0 == MAP_ERROR_NONE)) {
 420         if (PREDICT_FALSE(d0->mtu && (clib_net_to_host_u16(ip6h0->payload_length) + sizeof(*ip6h0) > d0->mtu))) {
 421           next0 = ip4_map_fragment(p0, d0->mtu, df0, &error0);
 422         } else {
 423           next0 = ip4_map_ip6_lookup_bypass(p0, ip40) ? IP4_MAP_NEXT_IP6_REWRITE : next0;
 424           vlib_increment_combined_counter(cm + MAP_DOMAIN_COUNTER_TX, cpu_index, map_domain_index0, 1,
 425                                           clib_net_to_host_u16(ip6h0->payload_length) + 40);
 426         }
 427       } else {
 428         next0 = IP4_MAP_NEXT_DROP;
 429       }
 430
 431       if (PREDICT_FALSE(p0->flags & VLIB_BUFFER_IS_TRACED)) {
 432         map_trace_t *tr = vlib_add_trace(vm, node, p0, sizeof(*tr));
 433         tr->map_domain_index = map_domain_index0;
 434         tr->port = port0;
 435       }
 436
 437       p0->error = error_node->errors[error0];
 438       vlib_validate_buffer_enqueue_x1(vm, node, next_index, to_next, n_left_to_next, pi0, next0);
 439     }
 440     vlib_put_next_frame(vm, node, next_index, n_left_to_next);
 441   }
 442
 443   return frame->n_vectors;
 444 }
 445
 446 /*
 447  * ip4_map_reass
 448  */
 449 static uword
 450 ip4_map_reass (vlib_main_t *vm,
 451                vlib_node_runtime_t *node,
 452                vlib_frame_t *frame)
 453 {
 454   u32 n_left_from, *from, next_index, *to_next, n_left_to_next;
 455   vlib_node_runtime_t *error_node = vlib_node_get_runtime(vm, ip4_map_reass_node.index);
 456   from = vlib_frame_vector_args(frame);
 457   n_left_from = frame->n_vectors;
 458   next_index = node->cached_next_index;
 459   map_main_t *mm = &map_main;
 460   vlib_combined_counter_main_t *cm = mm->domain_counters;
 461   u32 cpu_index = os_get_cpu_number();
 462   u32 *fragments_to_drop = NULL;
 463   u32 *fragments_to_loopback = NULL;
 464
 465   while (n_left_from > 0) {
 466     vlib_get_next_frame(vm, node, next_index, to_next, n_left_to_next);
 467
 468     while (n_left_from > 0 && n_left_to_next > 0) {
 469       u32 pi0;
 470       vlib_buffer_t *p0;
 471       map_domain_t *d0;
 472       u8 error0 = MAP_ERROR_NONE;
 473       ip4_header_t *ip40;
 474       i32 port0 = 0;
 475       ip6_header_t *ip60;
 476       u32 next0 = IP4_MAP_REASS_NEXT_IP6_LOOKUP;
 477       u32 map_domain_index0;
 478       u8 cached = 0;
 479
 480       pi0 = to_next[0] = from[0];
 481       from += 1;
 482       n_left_from -= 1;
 483       to_next +=1;
 484       n_left_to_next -= 1;
 485
 486       p0 = vlib_get_buffer(vm, pi0);
 487       ip60 = vlib_buffer_get_current(p0);
 488       ip40 = (ip4_header_t *)(ip60 + 1);
 489       d0 = ip4_map_get_domain(vnet_buffer(p0)->ip.adj_index[VLIB_TX], &map_domain_index0);
 490
 491       map_ip4_reass_lock();
 492       map_ip4_reass_t *r = map_ip4_reass_get(ip40->src_address.as_u32, ip40->dst_address.as_u32,
 493                                              ip40->fragment_id, ip40->protocol, &fragments_to_drop);
 494       if (PREDICT_FALSE(!r)) {
 495         // Could not create a caching entry
 496         error0 = MAP_ERROR_FRAGMENT_MEMORY;
 497       } else if (PREDICT_TRUE(ip4_get_fragment_offset(ip40))) {
 498         if (r->port >= 0) {
 499           // We know the port already
 500           port0 = r->port;
 501         } else if (map_ip4_reass_add_fragment(r, pi0)) {
 502           // Not enough space for caching
 503           error0 = MAP_ERROR_FRAGMENT_MEMORY;
 504           map_ip4_reass_free(r, &fragments_to_drop);
 505         } else {
 506           cached = 1;
 507         }
 508       } else if ((port0 = ip4_get_port(ip40, MAP_RECEIVER, p0->current_length)) < 0) {
 509         // Could not find port. We'll free the reassembly.
 510         error0 = MAP_ERROR_BAD_PROTOCOL;
 511         port0 = 0;
 512         map_ip4_reass_free(r, &fragments_to_drop);
 513       } else {
 514         r->port = port0;
 515         map_ip4_reass_get_fragments(r, &fragments_to_loopback);
 516       }
 517
 518 #ifdef MAP_IP4_REASS_COUNT_BYTES
 519       if (!cached && r) {
 520         r->forwarded += clib_host_to_net_u16(ip40->length) - 20;
 521         if (!ip4_get_fragment_more(ip40))
 522           r->expected_total = ip4_get_fragment_offset(ip40) * 8 + clib_host_to_net_u16(ip40->length) - 20;
 523         if(r->forwarded >= r->expected_total)
 524           map_ip4_reass_free(r, &fragments_to_drop);
 525       }
 526 #endif
 527
 528       map_ip4_reass_unlock();
 529
 530       // NOTE: Most operations have already been performed by ip4_map
 531       // All we need is the right destination address
 532       ip60->dst_address.as_u64[0] = map_get_pfx_net(d0, ip40->dst_address.as_u32, port0);
 533       ip60->dst_address.as_u64[1] = map_get_sfx_net(d0, ip40->dst_address.as_u32, port0);
 534
 535       if (PREDICT_FALSE(d0->mtu && (clib_net_to_host_u16(ip60->payload_length) + sizeof(*ip60) > d0->mtu))) {
 536         vnet_buffer(p0)->ip_frag.header_offset = sizeof(*ip60);
 537         vnet_buffer(p0)->ip_frag.next_index = IP4_FRAG_NEXT_IP6_LOOKUP;
 538         vnet_buffer(p0)->ip_frag.mtu = d0->mtu;
 539         vnet_buffer(p0)->ip_frag.flags = IP_FRAG_FLAG_IP6_HEADER;
 540         next0 = IP4_MAP_REASS_NEXT_IP4_FRAGMENT;
 541       }
 542
 543       if (PREDICT_FALSE(p0->flags & VLIB_BUFFER_IS_TRACED)) {
 544         map_ip4_map_reass_trace_t *tr = vlib_add_trace(vm, node, p0, sizeof(*tr));
 545         tr->map_domain_index = map_domain_index0;
 546         tr->port = port0;
 547         tr->cached = cached;
 548       }
 549
 550       if(cached) {
 551         //Dequeue the packet
 552         n_left_to_next++;
 553         to_next--;
 554       } else {
 555         if (error0 == MAP_ERROR_NONE)
 556           vlib_increment_combined_counter(cm + MAP_DOMAIN_COUNTER_TX, cpu_index, map_domain_index0, 1,
 557                                           clib_net_to_host_u16(ip60->payload_length) + 40);
 558         next0 = (error0 == MAP_ERROR_NONE) ? next0 : IP4_MAP_REASS_NEXT_DROP;
 559         p0->error = error_node->errors[error0];
 560         vlib_validate_buffer_enqueue_x1(vm, node, next_index, to_next, n_left_to_next, pi0, next0);
 561       }
 562
 563       //Loopback when we reach the end of the inpu vector
 564       if(n_left_from == 0 && vec_len(fragments_to_loopback)) {
 565         from = vlib_frame_vector_args(frame);
 566         u32 len = vec_len(fragments_to_loopback);
 567         if(len <= VLIB_FRAME_SIZE) {
 568           memcpy(from, fragments_to_loopback, sizeof(u32)*len);
 569           n_left_from = len;
 570           vec_reset_length(fragments_to_loopback);
 571         } else {
 572           memcpy(from, fragments_to_loopback + (len - VLIB_FRAME_SIZE), sizeof(u32)*VLIB_FRAME_SIZE);
 573           n_left_from = VLIB_FRAME_SIZE;
 574           _vec_len(fragments_to_loopback) = len - VLIB_FRAME_SIZE;
 575         }
 576       }
 577     }
 578     vlib_put_next_frame(vm, node, next_index, n_left_to_next);
 579   }
 580
 581   map_send_all_to_node(vm, fragments_to_drop, node,
 582                        &error_node->errors[MAP_ERROR_FRAGMENT_DROPPED],
 583                        IP4_MAP_REASS_NEXT_DROP);
 584
 585   vec_free(fragments_to_drop);
 586   vec_free(fragments_to_loopback);
 587   return frame->n_vectors;
 588 }
 589
 590 static char *map_error_strings[] = {
 591 #define _(sym,string) string,
 592   foreach_map_error
 593 #undef _
 594 };
 595
 596 VLIB_REGISTER_NODE(ip4_map_node) = {
 597   .function = ip4_map,
 598   .name = "ip4-map",
 599   .vector_size = sizeof(u32),
 600   .format_trace = format_map_trace,
 601   .type = VLIB_NODE_TYPE_INTERNAL,
 602
 603   .n_errors = MAP_N_ERROR,
 604   .error_strings = map_error_strings,
 605
 606   .n_next_nodes = IP4_MAP_N_NEXT,
 607   .next_nodes = {
 608     [IP4_MAP_NEXT_IP6_LOOKUP] = "ip6-lookup",
 609 #ifdef MAP_SKIP_IP6_LOOKUP
 610     [IP4_MAP_NEXT_IP6_REWRITE] = "ip6-rewrite",
 611 #endif
 612     [IP4_MAP_NEXT_IP4_FRAGMENT] = "ip4-frag",
 613     [IP4_MAP_NEXT_IP6_FRAGMENT] = "ip6-frag",
 614     [IP4_MAP_NEXT_REASS] = "ip4-map-reass",
 615     [IP4_MAP_NEXT_ICMP_ERROR] = "ip4-icmp-error",
 616     [IP4_MAP_NEXT_DROP] = "error-drop",
 617   },
 618 };
 619
 620 VLIB_REGISTER_NODE(ip4_map_reass_node) = {
 621   .function = ip4_map_reass,
 622   .name = "ip4-map-reass",
 623   .vector_size = sizeof(u32),
 624   .format_trace = format_ip4_map_reass_trace,
 625   .type = VLIB_NODE_TYPE_INTERNAL,
 626
 627   .n_errors = MAP_N_ERROR,
 628   .error_strings = map_error_strings,
 629
 630   .n_next_nodes = IP4_MAP_REASS_N_NEXT,
 631   .next_nodes = {
 632     [IP4_MAP_REASS_NEXT_IP6_LOOKUP] = "ip6-lookup",
 633     [IP4_MAP_REASS_NEXT_IP4_FRAGMENT] = "ip4-frag",
 634     [IP4_MAP_REASS_NEXT_DROP] = "error-drop",
 635   },
 636 };