src/vnet/session/application_namespace.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
  16 #include <vnet/session/application_namespace.h>
  17 #include <vnet/session/session_table.h>
  18 #include <vnet/session/session.h>
  19 #include <vnet/fib/fib_table.h>
  20
  21 /**
  22  * Hash table of application namespaces by app ns ids
  23  */
  24 uword *app_namespace_lookup_table;
  25
  26 /**
  27  * Pool of application namespaces
  28  */
  29 static app_namespace_t *app_namespace_pool;
  30
  31 app_namespace_t *
  32 app_namespace_get (u32 index)
  33 {
  34   return pool_elt_at_index (app_namespace_pool, index);
  35 }
  36
  37 app_namespace_t *
  38 app_namespace_get_from_id (const u8 * ns_id)
  39 {
  40   u32 index = app_namespace_index_from_id (ns_id);
  41   if (index == APP_NAMESPACE_INVALID_INDEX)
  42     return 0;
  43   return app_namespace_get (index);
  44 }
  45
  46 u32
  47 app_namespace_index (app_namespace_t * app_ns)
  48 {
  49   return (app_ns - app_namespace_pool);
  50 }
  51
  52 app_namespace_t *
  53 app_namespace_alloc (u8 * ns_id)
  54 {
  55   app_namespace_t *app_ns;
  56   pool_get (app_namespace_pool, app_ns);
  57   memset (app_ns, 0, sizeof (*app_ns));
  58   app_ns->ns_id = vec_dup (ns_id);
  59   hash_set_mem (app_namespace_lookup_table, app_ns->ns_id,
  60                 app_ns - app_namespace_pool);
  61   return app_ns;
  62 }
  63
  64 clib_error_t *
  65 vnet_app_namespace_add_del (vnet_app_namespace_add_del_args_t * a)
  66 {
  67   app_namespace_t *app_ns;
  68   session_table_t *st;
  69
  70   if (a->is_add)
  71     {
  72       if (a->sw_if_index != APP_NAMESPACE_INVALID_INDEX
  73           && !vnet_get_sw_interface_safe (vnet_get_main (), a->sw_if_index))
  74         return clib_error_return_code (0, VNET_API_ERROR_INVALID_SW_IF_INDEX,
  75                                        0, "sw_if_index %u doesn't exist",
  76                                        a->sw_if_index);
  77
  78       if (a->sw_if_index != APP_NAMESPACE_INVALID_INDEX)
  79         {
  80           a->ip4_fib_id =
  81             fib_table_get_table_id_for_sw_if_index (FIB_PROTOCOL_IP4,
  82                                                     a->sw_if_index);
  83           a->ip6_fib_id =
  84             fib_table_get_table_id_for_sw_if_index (FIB_PROTOCOL_IP4,
  85                                                     a->sw_if_index);
  86         }
  87       if (a->sw_if_index == APP_NAMESPACE_INVALID_INDEX
  88           && a->ip4_fib_id == APP_NAMESPACE_INVALID_INDEX)
  89         return clib_error_return_code (0, VNET_API_ERROR_INVALID_VALUE, 0,
  90                                        "sw_if_index or fib_id must be "
  91                                        "configured");
  92       st = session_table_alloc ();
  93       session_table_init (st);
  94
  95       app_ns = app_namespace_get_from_id (a->ns_id);
  96       if (!app_ns)
  97         app_ns = app_namespace_alloc (a->ns_id);
  98       app_ns->ns_secret = a->secret;
  99       app_ns->sw_if_index = a->sw_if_index;
 100       app_ns->local_table_index = session_table_index (st);
 101       app_ns->ip4_fib_index =
 102         fib_table_find (FIB_PROTOCOL_IP4, a->ip4_fib_id);
 103       app_ns->ip6_fib_index =
 104         fib_table_find (FIB_PROTOCOL_IP6, a->ip6_fib_id);
 105     }
 106   else
 107     {
 108       return clib_error_return_code (0, VNET_API_ERROR_UNIMPLEMENTED, 0,
 109                                      "namespace deletion not supported");
 110     }
 111   return 0;
 112 }
 113
 114 const u8 *
 115 app_namespace_id (app_namespace_t * app_ns)
 116 {
 117   return app_ns->ns_id;
 118 }
 119
 120 u32
 121 app_namespace_index_from_id (const u8 * ns_id)
 122 {
 123   uword *indexp;
 124   indexp = hash_get_mem (app_namespace_lookup_table, ns_id);
 125   if (!indexp)
 126     return APP_NAMESPACE_INVALID_INDEX;
 127   return *indexp;
 128 }
 129
 130 const u8 *
 131 app_namespace_id_from_index (u32 index)
 132 {
 133   app_namespace_t *app_ns;
 134
 135   app_ns = app_namespace_get (index);
 136   return app_namespace_id (app_ns);
 137 }
 138
 139 void
 140 app_namespaces_init (void)
 141 {
 142   u8 *ns_id = format (0, "default");
 143   app_namespace_lookup_table =
 144     hash_create_vec (0, sizeof (u8), sizeof (uword));
 145
 146   /*
 147    * Allocate default namespace
 148    */
 149   vnet_app_namespace_add_del_args_t a = {
 150     .ns_id = ns_id,
 151     .secret = 0,
 152     .sw_if_index = APP_NAMESPACE_INVALID_INDEX,
 153     .is_add = 1
 154   };
 155   vnet_app_namespace_add_del (&a);
 156   vec_free (ns_id);
 157 }
 158
 159 static clib_error_t *
 160 app_ns_fn (vlib_main_t * vm, unformat_input_t * input,
 161            vlib_cli_command_t * cmd)
 162 {
 163   unformat_input_t _line_input, *line_input = &_line_input;
 164   u8 is_add = 0, *ns_id = 0, secret_set = 0, sw_if_index_set = 0;
 165   u32 sw_if_index, fib_id = APP_NAMESPACE_INVALID_INDEX;
 166   u64 secret;
 167   clib_error_t *error = 0;
 168
 169   session_cli_return_if_not_enabled ();
 170
 171   if (!unformat_user (input, unformat_line_input, line_input))
 172     return 0;
 173
 174   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
 175     {
 176       if (unformat (line_input, "add"))
 177         is_add = 1;
 178       else if (unformat (line_input, "id %_%v%_", &ns_id))
 179         ;
 180       else if (unformat (line_input, "secret %lu", &secret))
 181         secret_set = 1;
 182       else if (unformat (line_input, "sw_if_index %u", &sw_if_index))
 183         sw_if_index_set = 1;
 184       else if (unformat (line_input, "fib_id", &fib_id))
 185         ;
 186       else
 187         {
 188           error = clib_error_return (0, "unknown input `%U'",
 189                                      format_unformat_error, line_input);
 190           unformat_free (line_input);
 191           return error;
 192         }
 193     }
 194   unformat_free (line_input);
 195
 196   if (!ns_id || !secret_set || !sw_if_index_set)
 197     {
 198       vlib_cli_output (vm, "namespace-id, secret and sw_if_index must be "
 199                        "provided");
 200       return 0;
 201     }
 202
 203   if (is_add)
 204     {
 205       vnet_app_namespace_add_del_args_t args = {
 206         .ns_id = ns_id,
 207         .secret = secret,
 208         .sw_if_index = sw_if_index,
 209         .ip4_fib_id = fib_id,
 210         .is_add = 1
 211       };
 212       error = vnet_app_namespace_add_del (&args);
 213     }
 214
 215   return error;
 216 }
 217
 218 /* *INDENT-OFF* */
 219 VLIB_CLI_COMMAND (app_ns_command, static) =
 220 {
 221   .path = "app ns",
 222   .short_help = "app ns [add] id <namespace-id> secret <secret> "
 223       "sw_if_index <sw_if_index>",
 224   .function = app_ns_fn,
 225 };
 226 /* *INDENT-ON* */
 227
 228 u8 *
 229 format_app_namespace (u8 * s, va_list * args)
 230 {
 231   app_namespace_t *app_ns = va_arg (*args, app_namespace_t *);
 232   s = format (s, "%-20v%-20lu%-20u", app_ns->ns_id, app_ns->ns_secret,
 233               app_ns->sw_if_index);
 234   return s;
 235 }
 236
 237 static clib_error_t *
 238 show_app_ns_fn (vlib_main_t * vm, unformat_input_t * input,
 239                 vlib_cli_command_t * cmd)
 240 {
 241   app_namespace_t *app_ns;
 242   session_table_t *st;
 243   u8 *ns_id, do_table = 0;
 244
 245   session_cli_return_if_not_enabled ();
 246
 247   if (unformat_peek_input (input) != UNFORMAT_END_OF_INPUT)
 248     {
 249       if (unformat (input, "table %_%v%_", &ns_id))
 250         do_table = 1;
 251       else
 252         {
 253           vlib_cli_output (vm, "unknown input [%U]",
 254                            format_unformat_error, input);
 255           return clib_error_return (0, "unknown input `%U'",
 256                                     format_unformat_error, input);
 257         }
 258     }
 259
 260   if (do_table)
 261     {
 262       app_ns = app_namespace_get_from_id (ns_id);
 263       if (!app_ns)
 264         {
 265           vlib_cli_output (vm, "ns %v not found", ns_id);
 266           return 0;
 267         }
 268       st = session_table_get (app_ns->local_table_index);
 269       if (!st)
 270         {
 271           vlib_cli_output (vm, "table for ns %v could not be found", ns_id);
 272           return 0;
 273         }
 274       session_lookup_show_table_entries (vm, st, 0, 1);
 275       vec_free (ns_id);
 276       return 0;
 277     }
 278
 279   vlib_cli_output (vm, "%-20s%-20s%-20s", "Namespace", "Secret",
 280                    "sw_if_index");
 281
 282   /* *INDENT-OFF* */
 283   pool_foreach (app_ns, app_namespace_pool, ({
 284     vlib_cli_output (vm, "%U", format_app_namespace, app_ns);
 285   }));
 286   /* *INDENT-ON* */
 287
 288   return 0;
 289 }
 290
 291 /* *INDENT-OFF* */
 292 VLIB_CLI_COMMAND (show_app_ns_command, static) =
 293 {
 294   .path = "show app ns",
 295   .short_help = "show app ns",
 296   .function = show_app_ns_fn,
 297 };
 298 /* *INDENT-ON* */
 299
 300 /*
 301  * fd.io coding-style-patch-verification: ON
 302  *
 303  * Local Variables:
 304  * eval: (c-set-style "gnu")
 305  * End:
 306  */