src/vnet/qos/qos_record.c

   1 /*
   2  * Copyright (c) 2018 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/qos/qos_record.h>
  17 #include <vnet/ip/ip.h>
  18 #include <vnet/ip/ip6_to_ip4.h>
  19 #include <vnet/feature/feature.h>
  20 #include <vnet/qos/qos_types.h>
  21 #include <vnet/l2/l2_input.h>
  22 #include <vnet/l2/feat_bitmap.h>
  23
  24 /**
  25  * Per-interface, per-protocol vector of feature on/off configurations
  26  */
  27 u8 *qos_record_configs[QOS_N_SOURCES];
  28 u32 l2_qos_input_next[QOS_N_SOURCES][32];
  29
  30 static void
  31 qos_record_feature_config (u32 sw_if_index,
  32                            qos_source_t input_source, u8 enable)
  33 {
  34   switch (input_source)
  35     {
  36     case QOS_SOURCE_IP:
  37       ip_feature_enable_disable (AF_IP6, N_SAFI, IP_FEATURE_INPUT,
  38                                  "ip6-qos-record",
  39                                  sw_if_index, enable, NULL, 0);
  40       ip_feature_enable_disable (AF_IP4, N_SAFI, IP_FEATURE_INPUT,
  41                                  "ip4-qos-record",
  42                                  sw_if_index, enable, NULL, 0);
  43       l2input_intf_bitmap_enable (sw_if_index, L2INPUT_FEAT_L2_IP_QOS_RECORD,
  44                                   enable);
  45       break;
  46     case QOS_SOURCE_MPLS:
  47       vnet_feature_enable_disable ("mpls-input", "mpls-qos-record",
  48                                    sw_if_index, enable, NULL, 0);
  49       break;
  50     case QOS_SOURCE_VLAN:
  51       ip_feature_enable_disable (AF_IP6, N_SAFI, IP_FEATURE_INPUT,
  52                                  "vlan-ip6-qos-record",
  53                                  sw_if_index, enable, NULL, 0);
  54       ip_feature_enable_disable (AF_IP4, N_SAFI, IP_FEATURE_INPUT,
  55                                  "vlan-ip4-qos-record",
  56                                  sw_if_index, enable, NULL, 0);
  57       vnet_feature_enable_disable ("mpls-input", "vlan-mpls-qos-record",
  58                                    sw_if_index, enable, NULL, 0);
  59       break;
  60     case QOS_SOURCE_EXT:
  61       /* not a valid option for recording */
  62       break;
  63     }
  64 }
  65
  66 int
  67 qos_record_enable (u32 sw_if_index, qos_source_t input_source)
  68 {
  69   vec_validate (qos_record_configs[input_source], sw_if_index);
  70
  71   if (0 == qos_record_configs[input_source][sw_if_index])
  72     {
  73       qos_record_feature_config (sw_if_index, input_source, 1);
  74     }
  75
  76   qos_record_configs[input_source][sw_if_index]++;
  77   return (0);
  78 }
  79
  80 int
  81 qos_record_disable (u32 sw_if_index, qos_source_t input_source)
  82 {
  83   if (vec_len (qos_record_configs[input_source]) <= sw_if_index)
  84     return VNET_API_ERROR_NO_MATCHING_INTERFACE;
  85
  86   if (0 == qos_record_configs[input_source][sw_if_index])
  87     return VNET_API_ERROR_VALUE_EXIST;
  88
  89   qos_record_configs[input_source][sw_if_index]--;
  90
  91   if (0 == qos_record_configs[input_source][sw_if_index])
  92     {
  93       qos_record_feature_config (sw_if_index, input_source, 0);
  94     }
  95
  96   return (0);
  97 }
  98
  99 void
 100 qos_record_walk (qos_record_walk_cb_t fn, void *c)
 101 {
 102   qos_source_t qs;
 103
 104   FOR_EACH_QOS_SOURCE (qs)
 105   {
 106     u32 sw_if_index;
 107
 108     vec_foreach_index (sw_if_index, qos_record_configs[qs])
 109     {
 110       if (0 != qos_record_configs[qs][sw_if_index])
 111         fn (sw_if_index, qs, c);
 112     }
 113   }
 114 }
 115
 116 /*
 117  * Disable recording feature for all protocols when the interface
 118  * is deleted
 119  */
 120 static clib_error_t *
 121 qos_record_ip_interface_add_del (vnet_main_t * vnm,
 122                                  u32 sw_if_index, u32 is_add)
 123 {
 124   if (!is_add)
 125     {
 126       qos_source_t qs;
 127
 128       FOR_EACH_QOS_SOURCE (qs)
 129       {
 130         while (qos_record_disable (sw_if_index, qs) == 0);
 131       }
 132     }
 133
 134   return (NULL);
 135 }
 136
 137 VNET_SW_INTERFACE_ADD_DEL_FUNCTION (qos_record_ip_interface_add_del);
 138
 139 clib_error_t *
 140 qos_record_init (vlib_main_t * vm)
 141 {
 142   qos_source_t qs;
 143   vlib_node_t *node = vlib_get_node_by_name (vm, (u8 *) "l2-ip-qos-record");
 144
 145   /* Initialize the feature next-node indexes */
 146   FOR_EACH_QOS_SOURCE (qs)
 147     feat_bitmap_init_next_nodes (vm,
 148                                  node->index,
 149                                  L2INPUT_N_FEAT,
 150                                  l2input_get_feat_names (),
 151                                  l2_qos_input_next[qs]);
 152   return 0;
 153 }
 154
 155 VLIB_INIT_FUNCTION (qos_record_init);
 156
 157 static clib_error_t *
 158 qos_record_cli (vlib_main_t * vm,
 159                 unformat_input_t * input, vlib_cli_command_t * cmd)
 160 {
 161   vnet_main_t *vnm = vnet_get_main ();
 162   u32 sw_if_index, qs;
 163   u8 enable;
 164
 165   qs = 0xff;
 166   enable = 1;
 167   sw_if_index = ~0;
 168
 169   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
 170     {
 171       if (unformat (input, "%U", unformat_vnet_sw_interface,
 172                     vnm, &sw_if_index))
 173         ;
 174       else if (unformat (input, "%U", unformat_qos_source, &qs))
 175         ;
 176       else if (unformat (input, "enable"))
 177         enable = 1;
 178       else if (unformat (input, "disable"))
 179         enable = 0;
 180       else
 181         break;
 182     }
 183
 184   if (~0 == sw_if_index)
 185     return clib_error_return (0, "interface must be specified");
 186   if (0xff == qs)
 187     return clib_error_return (0, "input location must be specified");
 188
 189   if (enable)
 190     qos_record_enable (sw_if_index, qs);
 191   else
 192     qos_record_disable (sw_if_index, qs);
 193
 194   return (NULL);
 195 }
 196
 197 /*?
 198  * Enable QoS bit recording on an interface using the packet's input DSCP bits
 199  * Which input QoS bits to use are either; IP, MPLS or VLAN. If more than
 200  * one protocol is chosen (which is foolish) the higher layers override the
 201  * lower.
 202  *
 203  * @cliexpar
 204  * @cliexcmd{qos record ip GigEthernet0/1/0}
 205  ?*/
 206 VLIB_CLI_COMMAND (qos_record_command, static) = {
 207   .path = "qos record",
 208   .short_help = "qos record <record-source> <INTERFACE> [disable]",
 209   .function = qos_record_cli,
 210   .is_mp_safe = 1,
 211 };
 212
 213 static void
 214 qos_record_show_one_interface (vlib_main_t * vm, u32 sw_if_index)
 215 {
 216   u8 n_cfgs[QOS_N_SOURCES] = { };
 217   qos_source_t qs;
 218   bool set;
 219
 220   set = false;
 221
 222   FOR_EACH_QOS_SOURCE (qs)
 223   {
 224     if (vec_len (qos_record_configs[qs]) <= sw_if_index)
 225       continue;
 226     if (0 != (n_cfgs[qs] = qos_record_configs[qs][sw_if_index]))
 227       set = true;
 228   }
 229
 230   if (set)
 231     {
 232       vlib_cli_output (vm, " %U:", format_vnet_sw_if_index_name,
 233                        vnet_get_main (), sw_if_index);
 234
 235       FOR_EACH_QOS_SOURCE (qs)
 236       {
 237         if (n_cfgs[qs] != 0)
 238           vlib_cli_output (vm, "  %U", format_qos_source, qs);
 239       }
 240     }
 241 }
 242
 243 static clib_error_t *
 244 qos_record_show (vlib_main_t * vm,
 245                  unformat_input_t * input, vlib_cli_command_t * cmd)
 246 {
 247   vnet_main_t *vnm = vnet_get_main ();
 248   qos_source_t qs;
 249   u32 sw_if_index;
 250
 251   sw_if_index = ~0;
 252
 253   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
 254     {
 255       if (unformat (input, "%U", unformat_vnet_sw_interface,
 256                     vnm, &sw_if_index))
 257         ;
 258     }
 259
 260   if (~0 == sw_if_index)
 261     {
 262       u32 ii, n_ints = 0;
 263
 264       FOR_EACH_QOS_SOURCE (qs)
 265       {
 266         n_ints = clib_max (n_ints, vec_len (qos_record_configs[qs]));
 267       }
 268
 269       for (ii = 0; ii < n_ints; ii++)
 270         {
 271           qos_record_show_one_interface (vm, ii);
 272         }
 273     }
 274   else
 275     qos_record_show_one_interface (vm, sw_if_index);
 276
 277   return (NULL);
 278 }
 279
 280 /*?
 281  * Show Egress Qos Maps
 282  *
 283  * @cliexpar
 284  * @cliexcmd{show qos egress map}
 285  ?*/
 286 VLIB_CLI_COMMAND (qos_record_show_command, static) = {
 287   .path = "show qos record",
 288   .short_help = "show qos record [interface]",
 289   .function = qos_record_show,
 290   .is_mp_safe = 1,
 291 };
 292
 293 /*
 294  * fd.io coding-style-patch-verification: ON
 295  *
 296  * Local Variables:
 297  * eval: (c-set-style "gnu")
 298  * End:
 299  */