2 * Copyright (c) 2017 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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
15 #include <nat/dslite.h>
16 #include <nat/nat_inlines.h>
18 vlib_node_registration_t dslite_in2out_node;
19 vlib_node_registration_t dslite_in2out_slowpath_node;
23 DSLITE_IN2OUT_NEXT_IP4_LOOKUP,
24 DSLITE_IN2OUT_NEXT_IP6_ICMP,
25 DSLITE_IN2OUT_NEXT_DROP,
26 DSLITE_IN2OUT_NEXT_SLOWPATH,
28 } dslite_in2out_next_t;
30 static char *dslite_in2out_error_strings[] = {
31 #define _(sym,string) string,
37 slow_path (dslite_main_t * dm, dslite_session_key_t * in2out_key,
38 dslite_session_t ** sp, u32 next, u8 * error, u32 thread_index)
41 clib_bihash_kv_16_8_t b4_kv, b4_value;
42 clib_bihash_kv_24_8_t in2out_kv;
43 clib_bihash_kv_8_8_t out2in_kv;
44 dlist_elt_t *head_elt, *oldest_elt, *elt;
47 snat_session_key_t out2in_key;
50 out2in_key.protocol = in2out_key->proto;
51 out2in_key.fib_index = 0;
53 b4_kv.key[0] = in2out_key->softwire_id.as_u64[0];
54 b4_kv.key[1] = in2out_key->softwire_id.as_u64[1];
56 if (clib_bihash_search_16_8
57 (&dm->per_thread_data[thread_index].b4_hash, &b4_kv, &b4_value))
59 pool_get (dm->per_thread_data[thread_index].b4s, b4);
60 memset (b4, 0, sizeof (*b4));
61 b4->addr.as_u64[0] = in2out_key->softwire_id.as_u64[0];
62 b4->addr.as_u64[1] = in2out_key->softwire_id.as_u64[1];
64 pool_get (dm->per_thread_data[thread_index].list_pool, head_elt);
65 b4->sessions_per_b4_list_head_index =
66 head_elt - dm->per_thread_data[thread_index].list_pool;
67 clib_dlist_init (dm->per_thread_data[thread_index].list_pool,
68 b4->sessions_per_b4_list_head_index);
70 b4_kv.value = b4 - dm->per_thread_data[thread_index].b4s;
71 clib_bihash_add_del_16_8 (&dm->per_thread_data[thread_index].b4_hash,
77 pool_elt_at_index (dm->per_thread_data[thread_index].b4s,
81 //TODO configurable quota
82 if (b4->nsessions >= 1000)
85 clib_dlist_remove_head (dm->per_thread_data[thread_index].list_pool,
86 b4->sessions_per_b4_list_head_index);
87 ASSERT (oldest_index != ~0);
88 clib_dlist_addtail (dm->per_thread_data[thread_index].list_pool,
89 b4->sessions_per_b4_list_head_index, oldest_index);
91 pool_elt_at_index (dm->per_thread_data[thread_index].list_pool,
94 pool_elt_at_index (dm->per_thread_data[thread_index].sessions,
97 in2out_kv.key[0] = s->in2out.as_u64[0];
98 in2out_kv.key[1] = s->in2out.as_u64[1];
99 in2out_kv.key[2] = s->in2out.as_u64[2];
100 clib_bihash_add_del_24_8 (&dm->per_thread_data[thread_index].in2out,
102 out2in_kv.key = s->out2in.as_u64;
103 clib_bihash_add_del_8_8 (&dm->per_thread_data[thread_index].out2in,
105 snat_free_outside_address_and_port (dm->addr_pool, thread_index,
107 s->outside_address_index = ~0;
109 if (snat_alloc_outside_address_and_port
110 (dm->addr_pool, 0, thread_index, &out2in_key,
111 &s->outside_address_index, dm->port_per_thread, thread_index))
116 if (snat_alloc_outside_address_and_port
117 (dm->addr_pool, 0, thread_index, &out2in_key, &address_index,
118 dm->port_per_thread, thread_index))
120 *error = DSLITE_ERROR_OUT_OF_PORTS;
121 return DSLITE_IN2OUT_NEXT_DROP;
123 pool_get (dm->per_thread_data[thread_index].sessions, s);
124 memset (s, 0, sizeof (*s));
125 s->outside_address_index = address_index;
128 pool_get (dm->per_thread_data[thread_index].list_pool, elt);
129 clib_dlist_init (dm->per_thread_data[thread_index].list_pool,
130 elt - dm->per_thread_data[thread_index].list_pool);
131 elt->value = s - dm->per_thread_data[thread_index].sessions;
132 s->per_b4_index = elt - dm->per_thread_data[thread_index].list_pool;
133 s->per_b4_list_head_index = b4->sessions_per_b4_list_head_index;
134 clib_dlist_addtail (dm->per_thread_data[thread_index].list_pool,
135 s->per_b4_list_head_index,
136 elt - dm->per_thread_data[thread_index].list_pool);
139 s->in2out = *in2out_key;
140 s->out2in = out2in_key;
142 in2out_kv.key[0] = s->in2out.as_u64[0];
143 in2out_kv.key[1] = s->in2out.as_u64[1];
144 in2out_kv.key[2] = s->in2out.as_u64[2];
145 in2out_kv.value = s - dm->per_thread_data[thread_index].sessions;
146 clib_bihash_add_del_24_8 (&dm->per_thread_data[thread_index].in2out,
148 out2in_kv.key = s->out2in.as_u64;
149 out2in_kv.value = s - dm->per_thread_data[thread_index].sessions;
150 clib_bihash_add_del_8_8 (&dm->per_thread_data[thread_index].out2in,
157 dslite_icmp_in2out (dslite_main_t * dm, ip6_header_t * ip6,
158 ip4_header_t * ip4, dslite_session_t ** sp, u32 next,
159 u8 * error, u32 thread_index)
161 dslite_session_t *s = 0;
162 icmp46_header_t *icmp = ip4_next_header (ip4);
163 clib_bihash_kv_24_8_t kv, value;
164 dslite_session_key_t key;
166 icmp_echo_header_t *echo;
167 u32 new_addr, old_addr;
171 if (icmp_is_error_message (icmp))
173 n = DSLITE_IN2OUT_NEXT_DROP;
174 *error = DSLITE_ERROR_BAD_ICMP_TYPE;
178 echo = (icmp_echo_header_t *) (icmp + 1);
180 key.addr = ip4->src_address;
181 key.port = echo->identifier;
182 key.proto = SNAT_PROTOCOL_ICMP;
183 key.softwire_id.as_u64[0] = ip6->src_address.as_u64[0];
184 key.softwire_id.as_u64[1] = ip6->src_address.as_u64[1];
186 kv.key[0] = key.as_u64[0];
187 kv.key[1] = key.as_u64[1];
188 kv.key[2] = key.as_u64[2];
190 if (clib_bihash_search_24_8
191 (&dm->per_thread_data[thread_index].in2out, &kv, &value))
193 n = slow_path (dm, &key, &s, next, error, thread_index);
194 if (PREDICT_FALSE (next == DSLITE_IN2OUT_NEXT_DROP))
200 pool_elt_at_index (dm->per_thread_data[thread_index].sessions,
204 old_addr = ip4->src_address.as_u32;
205 ip4->src_address = s->out2in.addr;
206 new_addr = ip4->src_address.as_u32;
208 sum = ip_csum_update (sum, old_addr, new_addr, ip4_header_t, src_address);
209 ip4->checksum = ip_csum_fold (sum);
211 old_id = echo->identifier;
212 echo->identifier = new_id = s->out2in.port;
213 sum = icmp->checksum;
214 sum = ip_csum_update (sum, old_id, new_id, icmp_echo_header_t, identifier);
215 icmp->checksum = ip_csum_fold (sum);
223 dslite_in2out_node_fn_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
224 vlib_frame_t * frame, u8 is_slow_path)
226 u32 n_left_from, *from, *to_next;
227 dslite_in2out_next_t next_index;
229 vlib_node_runtime_t *error_node;
230 u32 thread_index = vm->thread_index;
231 f64 now = vlib_time_now (vm);
232 dslite_main_t *dm = &dslite_main;
235 is_slow_path ? dslite_in2out_slowpath_node.
236 index : dslite_in2out_node.index;
238 error_node = vlib_node_get_runtime (vm, node_index);
240 from = vlib_frame_vector_args (frame);
241 n_left_from = frame->n_vectors;
242 next_index = node->cached_next_index;
244 while (n_left_from > 0)
248 vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
250 while (n_left_from > 0 && n_left_to_next > 0)
254 u32 next0 = DSLITE_IN2OUT_NEXT_IP4_LOOKUP;
257 u8 error0 = DSLITE_ERROR_IN2OUT;
259 dslite_session_t *s0 = 0;
260 clib_bihash_kv_24_8_t kv0, value0;
261 dslite_session_key_t key0;
265 u32 new_addr0, old_addr0;
266 u16 old_port0, new_port0;
268 /* speculatively enqueue b0 to the current next frame */
276 b0 = vlib_get_buffer (vm, bi0);
277 ip60 = vlib_buffer_get_current (b0);
279 if (PREDICT_FALSE (ip60->protocol != IP_PROTOCOL_IP_IN_IP))
281 if (ip60->protocol == IP_PROTOCOL_ICMP6)
283 next0 = DSLITE_IN2OUT_NEXT_IP6_ICMP;
286 error0 = DSLITE_ERROR_BAD_IP6_PROTOCOL;
287 next0 = DSLITE_IN2OUT_NEXT_DROP;
291 ip40 = vlib_buffer_get_current (b0) + sizeof (ip6_header_t);
292 proto0 = ip_proto_to_snat_proto (ip40->protocol);
294 if (PREDICT_FALSE (proto0 == ~0))
296 error0 = DSLITE_ERROR_UNSUPPORTED_PROTOCOL;
297 next0 = DSLITE_IN2OUT_NEXT_DROP;
301 udp0 = ip4_next_header (ip40);
302 tcp0 = (tcp_header_t *) udp0;
306 if (PREDICT_FALSE (proto0 == SNAT_PROTOCOL_ICMP))
309 dslite_icmp_in2out (dm, ip60, ip40, &s0, next0, &error0,
311 if (PREDICT_FALSE (next0 == DSLITE_IN2OUT_NEXT_DROP))
319 if (PREDICT_FALSE (proto0 == SNAT_PROTOCOL_ICMP))
321 next0 = DSLITE_IN2OUT_NEXT_SLOWPATH;
326 key0.addr = ip40->src_address;
327 key0.port = udp0->src_port;
329 key0.softwire_id.as_u64[0] = ip60->src_address.as_u64[0];
330 key0.softwire_id.as_u64[1] = ip60->src_address.as_u64[1];
332 kv0.key[0] = key0.as_u64[0];
333 kv0.key[1] = key0.as_u64[1];
334 kv0.key[2] = key0.as_u64[2];
336 if (clib_bihash_search_24_8
337 (&dm->per_thread_data[thread_index].in2out, &kv0, &value0))
342 slow_path (dm, &key0, &s0, next0, &error0, thread_index);
343 if (PREDICT_FALSE (next0 == DSLITE_IN2OUT_NEXT_DROP))
348 next0 = DSLITE_IN2OUT_NEXT_SLOWPATH;
355 pool_elt_at_index (dm->per_thread_data[thread_index].sessions,
359 old_addr0 = ip40->src_address.as_u32;
360 ip40->src_address = s0->out2in.addr;
361 new_addr0 = ip40->src_address.as_u32;
362 sum0 = ip40->checksum;
364 ip_csum_update (sum0, old_addr0, new_addr0, ip4_header_t,
366 ip40->checksum = ip_csum_fold (sum0);
367 if (PREDICT_TRUE (proto0 == SNAT_PROTOCOL_TCP))
369 old_port0 = tcp0->src_port;
370 tcp0->src_port = s0->out2in.port;
371 new_port0 = tcp0->src_port;
373 sum0 = tcp0->checksum;
375 ip_csum_update (sum0, old_addr0, new_addr0, ip4_header_t,
378 ip_csum_update (sum0, old_port0, new_port0, ip4_header_t,
380 tcp0->checksum = ip_csum_fold (sum0);
384 old_port0 = udp0->src_port;
385 udp0->src_port = s0->out2in.port;
391 s0->last_heard = now;
393 s0->total_bytes += vlib_buffer_length_in_chain (vm, b0);
394 /* Per-B4 LRU list maintenance */
395 clib_dlist_remove (dm->per_thread_data[thread_index].list_pool,
397 clib_dlist_addtail (dm->per_thread_data[thread_index].list_pool,
398 s0->per_b4_list_head_index, s0->per_b4_index);
401 (clib_net_to_host_u32
402 (ip60->ip_version_traffic_class_and_flow_label) & 0x0ff00000) >>
404 vlib_buffer_advance (b0, sizeof (ip6_header_t));
407 if (PREDICT_FALSE ((node->flags & VLIB_NODE_FLAG_TRACE)
408 && (b0->flags & VLIB_BUFFER_IS_TRACED)))
410 dslite_trace_t *t = vlib_add_trace (vm, node, b0, sizeof (*t));
411 t->next_index = next0;
412 t->session_index = ~0;
415 s0 - dm->per_thread_data[thread_index].sessions;
418 b0->error = error_node->errors[error0];
420 /* verify speculative enqueue, maybe switch current next frame */
421 vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
422 n_left_to_next, bi0, next0);
424 vlib_put_next_frame (vm, node, next_index, n_left_to_next);
427 return frame->n_vectors;
431 dslite_in2out_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node,
432 vlib_frame_t * frame)
434 return dslite_in2out_node_fn_inline (vm, node, frame, 0);
438 VLIB_REGISTER_NODE (dslite_in2out_node) = {
439 .function = dslite_in2out_node_fn,
440 .name = "dslite-in2out",
441 .vector_size = sizeof (u32),
442 .format_trace = format_dslite_trace,
443 .type = VLIB_NODE_TYPE_INTERNAL,
444 .n_errors = ARRAY_LEN (dslite_in2out_error_strings),
445 .error_strings = dslite_in2out_error_strings,
446 .n_next_nodes = DSLITE_IN2OUT_N_NEXT,
447 /* edit / add dispositions here */
449 [DSLITE_IN2OUT_NEXT_DROP] = "error-drop",
450 [DSLITE_IN2OUT_NEXT_IP4_LOOKUP] = "ip4-lookup",
451 [DSLITE_IN2OUT_NEXT_IP6_ICMP] = "ip6-icmp-input",
452 [DSLITE_IN2OUT_NEXT_SLOWPATH] = "dslite-in2out-slowpath",
457 VLIB_NODE_FUNCTION_MULTIARCH (dslite_in2out_node, dslite_in2out_node_fn);
460 dslite_in2out_slowpath_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node,
461 vlib_frame_t * frame)
463 return dslite_in2out_node_fn_inline (vm, node, frame, 1);
467 VLIB_REGISTER_NODE (dslite_in2out_slowpath_node) = {
468 .function = dslite_in2out_slowpath_node_fn,
469 .name = "dslite-in2out-slowpath",
470 .vector_size = sizeof (u32),
471 .format_trace = format_dslite_trace,
472 .type = VLIB_NODE_TYPE_INTERNAL,
473 .n_errors = ARRAY_LEN (dslite_in2out_error_strings),
474 .error_strings = dslite_in2out_error_strings,
475 .n_next_nodes = DSLITE_IN2OUT_N_NEXT,
476 /* edit / add dispositions here */
478 [DSLITE_IN2OUT_NEXT_DROP] = "error-drop",
479 [DSLITE_IN2OUT_NEXT_IP4_LOOKUP] = "ip4-lookup",
480 [DSLITE_IN2OUT_NEXT_IP6_ICMP] = "ip6-lookup",
481 [DSLITE_IN2OUT_NEXT_SLOWPATH] = "dslite-in2out-slowpath",
486 VLIB_NODE_FUNCTION_MULTIARCH (dslite_in2out_slowpath_node,
487 dslite_in2out_slowpath_node_fn);
490 * fd.io coding-style-patch-verification: ON
493 * eval: (c-set-style "gnu")