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/dslite_dpo.h>
17 #include <vnet/fib/fib_table.h>
19 dslite_main_t dslite_main;
22 dslite_init (vlib_main_t * vm)
24 dslite_main_t *dm = &dslite_main;
25 vlib_thread_registration_t *tr;
26 vlib_thread_main_t *tm = vlib_get_thread_main ();
28 dslite_per_thread_data_t *td;
29 u32 translation_buckets = 1024;
30 u32 translation_memory_size = 128 << 20;
32 u32 b4_memory_size = 64 << 20;
34 dm->first_worker_index = 0;
37 p = hash_get_mem (tm->thread_registrations_by_name, "workers");
40 tr = (vlib_thread_registration_t *) p[0];
43 dm->num_workers = tr->count;
44 dm->first_worker_index = tr->first_index;
49 dm->port_per_thread = (0xffff - 1024) / dm->num_workers;
51 dm->port_per_thread = 0xffff - 1024;
53 vec_validate (dm->per_thread_data, tm->n_vlib_mains - 1);
56 vec_foreach (td, dm->per_thread_data)
58 clib_bihash_init_24_8 (&td->in2out, "in2out", translation_buckets,
59 translation_memory_size);
61 clib_bihash_init_8_8 (&td->out2in, "out2in", translation_buckets,
62 translation_memory_size);
64 clib_bihash_init_16_8 (&td->b4_hash, "b4s", b4_buckets, b4_memory_size);
70 dslite_dpo_module_init ();
74 dslite_set_ce (dslite_main_t * dm, u8 set)
76 dm->is_ce = (set != 0);
80 dslite_set_aftr_ip6_addr (dslite_main_t * dm, ip6_address_t * addr)
82 dpo_id_t dpo = DPO_INVALID;
86 dslite_ce_dpo_create (DPO_PROTO_IP4, 0, &dpo);
88 .fp_proto = FIB_PROTOCOL_IP4,
90 .fp_addr.ip4.as_u32 = 0,
92 fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
93 FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
97 dslite_dpo_create (DPO_PROTO_IP6, 0, &dpo);
99 .fp_proto = FIB_PROTOCOL_IP6,
101 .fp_addr.ip6.as_u64[0] = addr->as_u64[0],
102 .fp_addr.ip6.as_u64[1] = addr->as_u64[1],
104 fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
105 FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
110 dm->aftr_ip6_addr.as_u64[0] = addr->as_u64[0];
111 dm->aftr_ip6_addr.as_u64[1] = addr->as_u64[1];
116 dslite_set_aftr_ip4_addr (dslite_main_t * dm, ip4_address_t * addr)
118 dm->aftr_ip4_addr.as_u32 = addr->as_u32;
123 dslite_set_b4_ip6_addr (dslite_main_t * dm, ip6_address_t * addr)
127 dpo_id_t dpo = DPO_INVALID;
129 dslite_ce_dpo_create (DPO_PROTO_IP6, 0, &dpo);
131 .fp_proto = FIB_PROTOCOL_IP6,
133 .fp_addr.ip6.as_u64[0] = addr->as_u64[0],
134 .fp_addr.ip6.as_u64[1] = addr->as_u64[1],
136 fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
137 FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
141 dm->b4_ip6_addr.as_u64[0] = addr->as_u64[0];
142 dm->b4_ip6_addr.as_u64[1] = addr->as_u64[1];
146 return VNET_API_ERROR_FEATURE_DISABLED;
153 dslite_set_b4_ip4_addr (dslite_main_t * dm, ip4_address_t * addr)
157 dm->b4_ip4_addr.as_u32 = addr->as_u32;
161 return VNET_API_ERROR_FEATURE_DISABLED;
168 dslite_add_del_pool_addr (dslite_main_t * dm, ip4_address_t * addr, u8 is_add)
170 vlib_thread_main_t *tm = vlib_get_thread_main ();
171 snat_address_t *a = 0;
173 dpo_id_t dpo_v4 = DPO_INVALID;
175 .fp_proto = FIB_PROTOCOL_IP4,
177 .fp_addr.ip4.as_u32 = addr->as_u32,
180 for (i = 0; i < vec_len (dm->addr_pool); i++)
182 if (dm->addr_pool[i].addr.as_u32 == addr->as_u32)
184 a = dm->addr_pool + i;
191 return VNET_API_ERROR_VALUE_EXIST;
192 vec_add2 (dm->addr_pool, a, 1);
194 #define _(N, i, n, s) \
195 clib_bitmap_alloc (a->busy_##n##_port_bitmap, 65535); \
196 a->busy_##n##_ports = 0; \
197 vec_validate_init_empty (a->busy_##n##_ports_per_thread, tm->n_vlib_mains - 1, 0);
198 foreach_snat_protocol
200 dslite_dpo_create (DPO_PROTO_IP4, 0, &dpo_v4);
201 fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
202 FIB_ENTRY_FLAG_EXCLUSIVE, &dpo_v4);
208 return VNET_API_ERROR_NO_SUCH_ENTRY;
209 #define _(N, id, n, s) \
210 clib_bitmap_free (a->busy_##n##_port_bitmap); \
211 vec_free (a->busy_##n##_ports_per_thread);
212 foreach_snat_protocol
214 fib_table_entry_special_remove (0, &pfx, FIB_SOURCE_PLUGIN_HI);
215 vec_del1 (dm->addr_pool, i);
221 format_dslite_trace (u8 * s, va_list * args)
223 CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
224 CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
225 dslite_trace_t *t = va_arg (*args, dslite_trace_t *);
228 format (s, "next index %d, session %d", t->next_index, t->session_index);
234 format_dslite_ce_trace (u8 * s, va_list * args)
236 CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
237 CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
238 dslite_ce_trace_t *t = va_arg (*args, dslite_ce_trace_t *);
240 s = format (s, "next index %d", t->next_index);
246 * fd.io coding-style-patch-verification: ON
249 * eval: (c-set-style "gnu")