src/vnet/ip/ip_container_proxy.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  * ip/ip_lookup.c: ip4/6 adjacency and lookup table management
  17  *
  18  * Copyright (c) 2008 Eliot Dresselhaus
  19  *
  20  * Permission is hereby granted, free of charge, to any person obtaining
  21  * a copy of this software and associated documentation files (the
  22  * "Software"), to deal in the Software without restriction, including
  23  * without limitation the rights to use, copy, modify, merge, publish,
  24  * distribute, sublicense, and/or sell copies of the Software, and to
  25  * permit persons to whom the Software is furnished to do so, subject to
  26  * the following conditions:
  27  *
  28  * The above copyright notice and this permission notice shall be
  29  * included in all copies or substantial portions of the Software.
  30  *
  31  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  32  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  33  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  34  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  35  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  36  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  37  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  38  */
  39
  40 #include <vnet/ip/ip_container_proxy.h>
  41 #include <vnet/ip/format.h>
  42 #include <vnet/fib/fib_table.h>
  43 #include <vnet/dpo/l3_proxy_dpo.h>
  44 #include <vnet/dpo/load_balance.h>
  45
  46 clib_error_t *
  47 vnet_ip_container_proxy_add_del (vnet_ip_container_proxy_args_t * args)
  48 {
  49   u32 fib_index;
  50
  51   if (!vnet_sw_interface_is_api_valid (vnet_get_main (), args->sw_if_index))
  52     return clib_error_return_code (0, VNET_API_ERROR_INVALID_INTERFACE, 0,
  53                                    "invalid sw_if_index");
  54
  55   fib_index = fib_table_get_table_id_for_sw_if_index (args->prefix.fp_proto,
  56                                                       args->sw_if_index);
  57   if (args->is_add)
  58     {
  59       dpo_id_t proxy_dpo = DPO_INVALID;
  60       l3_proxy_dpo_add_or_lock (fib_proto_to_dpo (args->prefix.fp_proto),
  61                                 args->sw_if_index, &proxy_dpo);
  62       fib_table_entry_special_dpo_add (fib_index,
  63                                        &args->prefix,
  64                                        FIB_SOURCE_PROXY,
  65                                        FIB_ENTRY_FLAG_EXCLUSIVE, &proxy_dpo);
  66       dpo_reset (&proxy_dpo);
  67     }
  68   else
  69     {
  70       fib_table_entry_special_remove (fib_index, &args->prefix,
  71                                       FIB_SOURCE_PROXY);
  72     }
  73   return 0;
  74 }
  75
  76 u8
  77 ip_container_proxy_is_set (fib_prefix_t * pfx, u32 sw_if_index)
  78 {
  79   u32 fib_index;
  80   fib_node_index_t fei;
  81   const dpo_id_t *dpo;
  82   l3_proxy_dpo_t *l3p;
  83   load_balance_t *lb0;
  84
  85   fib_index = fib_table_get_table_id_for_sw_if_index (pfx->fp_proto,
  86                                                       sw_if_index);
  87   if (fib_index == ~0)
  88     return 0;
  89
  90   fei = fib_table_lookup_exact_match (fib_index, pfx);
  91   if (fei == FIB_NODE_INDEX_INVALID)
  92     return 0;
  93
  94   dpo = fib_entry_contribute_ip_forwarding (fei);
  95   lb0 = load_balance_get (dpo->dpoi_index);
  96   dpo = load_balance_get_bucket_i (lb0, 0);
  97   if (dpo->dpoi_type != DPO_L3_PROXY)
  98     return 0;
  99
 100   l3p = l3_proxy_dpo_get (dpo->dpoi_index);
 101   return (l3p->l3p_sw_if_index == sw_if_index);
 102 }
 103
 104 typedef struct ip_container_proxy_walk_ctx_t_
 105 {
 106   ip_container_proxy_cb_t cb;
 107   void *ctx;
 108 } ip_container_proxy_walk_ctx_t;
 109
 110 static fib_table_walk_rc_t
 111 ip_container_proxy_fib_table_walk (fib_node_index_t fei, void *arg)
 112 {
 113   ip_container_proxy_walk_ctx_t *ctx = arg;
 114   const fib_prefix_t *pfx;
 115   const dpo_id_t *dpo;
 116   load_balance_t *lb;
 117   l3_proxy_dpo_t *l3p;
 118
 119   pfx = fib_entry_get_prefix (fei);
 120   if (fib_entry_is_sourced (fei, FIB_SOURCE_PROXY))
 121     {
 122       dpo = fib_entry_contribute_ip_forwarding (fei);
 123       lb = load_balance_get (dpo->dpoi_index);
 124       dpo = load_balance_get_bucket_i (lb, 0);
 125       l3p = l3_proxy_dpo_get (dpo->dpoi_index);
 126       ctx->cb (pfx, l3p->l3p_sw_if_index, ctx->ctx);
 127     }
 128
 129   return FIB_TABLE_WALK_CONTINUE;
 130 }
 131
 132 void
 133 ip_container_proxy_walk (ip_container_proxy_cb_t cb, void *ctx)
 134 {
 135   ip_container_proxy_walk_ctx_t wctx = {
 136     .cb = cb,
 137     .ctx = ctx,
 138   };
 139   u32 fib_index;
 140
 141   pool_foreach_index (fib_index, ip4_main.fibs)
 142     {
 143       fib_table_walk (fib_index, FIB_PROTOCOL_IP4,
 144                       ip_container_proxy_fib_table_walk, &wctx);
 145     }
 146   pool_foreach_index (fib_index, ip6_main.fibs)
 147     {
 148       fib_table_walk (fib_index, FIB_PROTOCOL_IP6,
 149                       ip_container_proxy_fib_table_walk, &wctx);
 150     }
 151 }
 152
 153 clib_error_t *
 154 ip_container_cmd (vlib_main_t * vm,
 155                   unformat_input_t * main_input, vlib_cli_command_t * cmd)
 156 {
 157   unformat_input_t _line_input, *line_input = &_line_input;
 158   fib_prefix_t pfx;
 159   u32 is_del, addr_set = 0;
 160   vnet_main_t *vnm;
 161   u32 sw_if_index;
 162
 163   vnm = vnet_get_main ();
 164   is_del = 0;
 165   sw_if_index = ~0;
 166   clib_memset (&pfx, 0, sizeof (pfx));
 167
 168   /* Get a line of input. */
 169   if (!unformat_user (main_input, unformat_line_input, line_input))
 170     return 0;
 171
 172   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
 173     {
 174       if (unformat (line_input, "%U", unformat_ip4_address, &pfx.fp_addr.ip4))
 175         {
 176           pfx.fp_proto = FIB_PROTOCOL_IP4;
 177           pfx.fp_len = 32;
 178           addr_set = 1;
 179         }
 180       else if (unformat (line_input, "%U",
 181                          unformat_ip6_address, &pfx.fp_addr.ip6))
 182         {
 183           pfx.fp_proto = FIB_PROTOCOL_IP6;
 184           pfx.fp_len = 128;
 185           addr_set = 1;
 186         }
 187       else if (unformat (line_input, "%U",
 188                          unformat_vnet_sw_interface, vnm, &sw_if_index))
 189         ;
 190       else if (unformat (line_input, "del"))
 191         is_del = 1;
 192       else
 193         {
 194           unformat_free (line_input);
 195           return (clib_error_return (0, "unknown input '%U'",
 196                                      format_unformat_error, line_input));
 197         }
 198     }
 199
 200   if (~0 == sw_if_index || !addr_set)
 201     {
 202       unformat_free (line_input);
 203       vlib_cli_output (vm, "interface and address must be set");
 204       return 0;
 205     }
 206
 207   vnet_ip_container_proxy_args_t args = {
 208     .prefix = pfx,
 209     .sw_if_index = sw_if_index,
 210     .is_add = !is_del,
 211   };
 212   vnet_ip_container_proxy_add_del (&args);
 213   unformat_free (line_input);
 214   return (NULL);
 215 }
 216
 217 VLIB_CLI_COMMAND (ip_container_command_node, static) = {
 218   .path = "ip container",
 219   .function = ip_container_cmd,
 220   .short_help = "ip container <address> <interface>",
 221   .is_mp_safe = 1,
 222 };
 223
 224 clib_error_t *
 225 show_ip_container_cmd_fn (vlib_main_t * vm, unformat_input_t * main_input,
 226                           vlib_cli_command_t * cmd)
 227 {
 228   unformat_input_t _line_input, *line_input = &_line_input;
 229   vnet_main_t *vnm = vnet_get_main ();
 230   fib_prefix_t pfx;
 231   u32 sw_if_index = ~0;
 232   u8 has_proxy;
 233
 234   if (!unformat_user (main_input, unformat_line_input, line_input))
 235     return 0;
 236   while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
 237     {
 238       if (unformat (line_input, "%U", unformat_ip4_address, &pfx.fp_addr.ip4))
 239         {
 240           pfx.fp_proto = FIB_PROTOCOL_IP4;
 241           pfx.fp_len = 32;
 242         }
 243       else if (unformat (line_input, "%U",
 244                          unformat_ip6_address, &pfx.fp_addr.ip6))
 245         {
 246           pfx.fp_proto = FIB_PROTOCOL_IP6;
 247           pfx.fp_len = 128;
 248         }
 249       else if (unformat (line_input, "%U",
 250                          unformat_vnet_sw_interface, vnm, &sw_if_index))
 251         ;
 252       else
 253         {
 254           unformat_free (line_input);
 255           return (clib_error_return (0, "unknown input '%U'",
 256                                      format_unformat_error, line_input));
 257         }
 258     }
 259
 260   if (~0 == sw_if_index)
 261     {
 262       unformat_free (line_input);
 263       vlib_cli_output (vm, "no interface");
 264       return (clib_error_return (0, "no interface"));
 265     }
 266
 267   has_proxy = ip_container_proxy_is_set (&pfx, sw_if_index);
 268   vlib_cli_output (vm, "ip container proxy is: %s", has_proxy ? "on" : "off");
 269
 270   unformat_free (line_input);
 271   return 0;
 272 }
 273
 274 VLIB_CLI_COMMAND (show_ip_container_command, static) = {
 275   .path = "show ip container",
 276   .function = show_ip_container_cmd_fn,
 277   .short_help = "show ip container <address> <interface>",
 278   .is_mp_safe = 1,
 279 };
 280
 281 /*
 282  * fd.io coding-style-patch-verification: ON
 283  *
 284  * Local Variables:
 285  * eval: (c-set-style "gnu")
 286  * End:
 287  */