src/plugins/nat/dslite.c

   1 /*
   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:
   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 #include <nat/dslite.h>
  16 #include <nat/dslite_dpo.h>
  17 #include <vnet/fib/fib_table.h>
  18
  19 dslite_main_t dslite_main;
  20
  21 void
  22 dslite_init (vlib_main_t * vm)
  23 {
  24   dslite_main_t *dm = &dslite_main;
  25   vlib_thread_registration_t *tr;
  26   vlib_thread_main_t *tm = vlib_get_thread_main ();
  27   uword *p;
  28   dslite_per_thread_data_t *td;
  29   u32 translation_buckets = 1024;
  30   u32 translation_memory_size = 128 << 20;
  31   u32 b4_buckets = 128;
  32   u32 b4_memory_size = 64 << 20;
  33
  34   dm->first_worker_index = 0;
  35   dm->num_workers = 0;
  36
  37   p = hash_get_mem (tm->thread_registrations_by_name, "workers");
  38   if (p)
  39     {
  40       tr = (vlib_thread_registration_t *) p[0];
  41       if (tr)
  42         {
  43           dm->num_workers = tr->count;
  44           dm->first_worker_index = tr->first_index;
  45         }
  46     }
  47
  48   if (dm->num_workers)
  49     dm->port_per_thread = (0xffff - 1024) / dm->num_workers;
  50   else
  51     dm->port_per_thread = 0xffff - 1024;
  52
  53   vec_validate (dm->per_thread_data, tm->n_vlib_mains - 1);
  54
  55   /* *INDENT-OFF* */
  56   vec_foreach (td, dm->per_thread_data)
  57     {
  58       clib_bihash_init_24_8 (&td->in2out, "in2out", translation_buckets,
  59                              translation_memory_size);
  60
  61       clib_bihash_init_8_8 (&td->out2in, "out2in", translation_buckets,
  62                             translation_memory_size);
  63
  64       clib_bihash_init_16_8 (&td->b4_hash, "b4s", b4_buckets, b4_memory_size);
  65     }
  66   /* *INDENT-ON* */
  67
  68   dm->is_ce = 0;
  69
  70   dslite_dpo_module_init ();
  71 }
  72
  73 void
  74 dslite_set_ce (dslite_main_t * dm, u8 set)
  75 {
  76   dm->is_ce = (set != 0);
  77 }
  78
  79 int
  80 dslite_set_aftr_ip6_addr (dslite_main_t * dm, ip6_address_t * addr)
  81 {
  82   dpo_id_t dpo = DPO_INVALID;
  83
  84   if (dm->is_ce)
  85     {
  86       dslite_ce_dpo_create (DPO_PROTO_IP4, 0, &dpo);
  87       fib_prefix_t pfx = {
  88         .fp_proto = FIB_PROTOCOL_IP4,
  89         .fp_len = 0,
  90         .fp_addr.ip4.as_u32 = 0,
  91       };
  92       fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
  93                                        FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
  94     }
  95   else
  96     {
  97       dslite_dpo_create (DPO_PROTO_IP6, 0, &dpo);
  98       fib_prefix_t pfx = {
  99         .fp_proto = FIB_PROTOCOL_IP6,
 100         .fp_len = 128,
 101         .fp_addr.ip6.as_u64[0] = addr->as_u64[0],
 102         .fp_addr.ip6.as_u64[1] = addr->as_u64[1],
 103       };
 104       fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
 105                                        FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
 106     }
 107
 108   dpo_reset (&dpo);
 109
 110   dm->aftr_ip6_addr.as_u64[0] = addr->as_u64[0];
 111   dm->aftr_ip6_addr.as_u64[1] = addr->as_u64[1];
 112   return 0;
 113 }
 114
 115 int
 116 dslite_set_aftr_ip4_addr (dslite_main_t * dm, ip4_address_t * addr)
 117 {
 118   dm->aftr_ip4_addr.as_u32 = addr->as_u32;
 119   return 0;
 120 }
 121
 122 int
 123 dslite_set_b4_ip6_addr (dslite_main_t * dm, ip6_address_t * addr)
 124 {
 125   if (dm->is_ce)
 126     {
 127       dpo_id_t dpo = DPO_INVALID;
 128
 129       dslite_ce_dpo_create (DPO_PROTO_IP6, 0, &dpo);
 130       fib_prefix_t pfx = {
 131         .fp_proto = FIB_PROTOCOL_IP6,
 132         .fp_len = 128,
 133         .fp_addr.ip6.as_u64[0] = addr->as_u64[0],
 134         .fp_addr.ip6.as_u64[1] = addr->as_u64[1],
 135       };
 136       fib_table_entry_special_dpo_add (0, &pfx, FIB_SOURCE_PLUGIN_HI,
 137                                        FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
 138
 139       dpo_reset (&dpo);
 140
 141       dm->b4_ip6_addr.as_u64[0] = addr->as_u64[0];
 142       dm->b4_ip6_addr.as_u64[1] = addr->as_u64[1];
 143     }
 144   else
 145     {
 146       return VNET_API_ERROR_FEATURE_DISABLED;
 147     }
 148
 149   return 0;
 150 }
 151
 152 int
 153 dslite_set_b4_ip4_addr (dslite_main_t * dm, ip4_address_t * addr)
 154 {
 155   if (dm->is_ce)
 156     {
 157       dm->b4_ip4_addr.as_u32 = addr->as_u32;
 158     }
 159   else
 160     {
 161       return VNET_API_ERROR_FEATURE_DISABLED;
 162     }
 163
 164   return 0;
 165 }
 166
 167 int
 168 dslite_add_del_pool_addr (dslite_main_t * dm, ip4_address_t * addr, u8 is_add)
 169 {
 170   vlib_thread_main_t *tm = vlib_get_thread_main ();
 171   snat_address_t *a = 0;
 172   int i = 0;
 173   dpo_id_t dpo_v4 = DPO_INVALID;
 174   fib_prefix_t pfx = {
 175     .fp_proto = FIB_PROTOCOL_IP4,
 176     .fp_len = 32,
 177     .fp_addr.ip4.as_u32 = addr->as_u32,
 178   };
 179
 180   for (i = 0; i < vec_len (dm->addr_pool); i++)
 181     {
 182       if (dm->addr_pool[i].addr.as_u32 == addr->as_u32)
 183         {
 184           a = dm->addr_pool + i;
 185           break;
 186         }
 187     }
 188   if (is_add)
 189     {
 190       if (a)
 191         return VNET_API_ERROR_VALUE_EXIST;
 192       vec_add2 (dm->addr_pool, a, 1);
 193       a->addr = *addr;
 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
 199 #undef _
 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);
 203       dpo_reset (&dpo_v4);
 204     }
 205   else
 206     {
 207       if (!a)
 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
 213 #undef _
 214         fib_table_entry_special_remove (0, &pfx, FIB_SOURCE_PLUGIN_HI);
 215       vec_del1 (dm->addr_pool, i);
 216     }
 217   return 0;
 218 }
 219
 220 u8 *
 221 format_dslite_trace (u8 * s, va_list * args)
 222 {
 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 *);
 226
 227   s =
 228     format (s, "next index %d, session %d", t->next_index, t->session_index);
 229
 230   return s;
 231 }
 232
 233 u8 *
 234 format_dslite_ce_trace (u8 * s, va_list * args)
 235 {
 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 *);
 239
 240   s = format (s, "next index %d", t->next_index);
 241
 242   return s;
 243 }
 244
 245 /*
 246  * fd.io coding-style-patch-verification: ON
 247  *
 248  * Local Variables:
 249  * eval: (c-set-style "gnu")
 250  * End:
 251  */