LCOV - code coverage report
Current view: top level - vnet/classify - policer_classify.c (source / functions) Hit Total Coverage
Test: coverage-filtered.info Lines: 2 97 2.1 %
Date: 2023-07-05 22:20:52 Functions: 4 9 44.4 %

          Line data    Source code
       1             : /*
       2             :  * Copyright (c) 2016 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 <vnet/classify/policer_classify.h>
      16             : #include <vnet/l2/l2_input.h>
      17             : 
      18             : policer_classify_main_t policer_classify_main;
      19             : 
      20             : static void
      21           0 : vnet_policer_classify_feature_enable (vlib_main_t * vnm,
      22             :                                       policer_classify_main_t * pcm,
      23             :                                       u32 sw_if_index,
      24             :                                       policer_classify_table_id_t tid,
      25             :                                       int feature_enable)
      26             : {
      27           0 :   if (tid == POLICER_CLASSIFY_TABLE_L2)
      28             :     {
      29           0 :       l2input_intf_bitmap_enable (sw_if_index, L2INPUT_FEAT_POLICER_CLAS,
      30             :                                   feature_enable);
      31             :     }
      32             :   else
      33             :     {
      34             :       vnet_feature_config_main_t *fcm;
      35             :       u8 arc;
      36             : 
      37           0 :       if (tid == POLICER_CLASSIFY_TABLE_IP4)
      38             :         {
      39           0 :           vnet_feature_enable_disable ("ip4-unicast", "ip4-policer-classify",
      40             :                                        sw_if_index, feature_enable, 0, 0);
      41           0 :           arc = vnet_get_feature_arc_index ("ip4-unicast");
      42             :         }
      43             : 
      44             :       else
      45             :         {
      46           0 :           vnet_feature_enable_disable ("ip6-unicast", "ip6-policer-classify",
      47             :                                        sw_if_index, feature_enable, 0, 0);
      48           0 :           arc = vnet_get_feature_arc_index ("ip6-unicast");
      49             :         }
      50             : 
      51           0 :       fcm = vnet_get_feature_arc_config_main (arc);
      52           0 :       pcm->vnet_config_main[tid] = &fcm->config_main;
      53             :     }
      54           0 : }
      55             : 
      56             : int
      57           0 : vnet_set_policer_classify_intfc (vlib_main_t * vm, u32 sw_if_index,
      58             :                                  u32 ip4_table_index, u32 ip6_table_index,
      59             :                                  u32 l2_table_index, u32 is_add)
      60             : {
      61           0 :   policer_classify_main_t *pcm = &policer_classify_main;
      62           0 :   vnet_classify_main_t *vcm = pcm->vnet_classify_main;
      63           0 :   u32 pct[POLICER_CLASSIFY_N_TABLES] = { ip4_table_index, ip6_table_index,
      64             :     l2_table_index
      65             :   };
      66             :   u32 ti;
      67             : 
      68             :   /* Assume that we've validated sw_if_index in the API layer */
      69             : 
      70           0 :   for (ti = 0; ti < POLICER_CLASSIFY_N_TABLES; ti++)
      71             :     {
      72           0 :       if (pct[ti] == ~0)
      73           0 :         continue;
      74             : 
      75           0 :       if (pool_is_free_index (vcm->tables, pct[ti]))
      76           0 :         return VNET_API_ERROR_NO_SUCH_TABLE;
      77             : 
      78           0 :       vec_validate_init_empty
      79             :         (pcm->classify_table_index_by_sw_if_index[ti], sw_if_index, ~0);
      80             : 
      81             :       /* Reject any DEL operation with wrong sw_if_index */
      82           0 :       if (!is_add &&
      83           0 :           (pct[ti] !=
      84           0 :            pcm->classify_table_index_by_sw_if_index[ti][sw_if_index]))
      85             :         {
      86           0 :           clib_warning
      87             :             ("Non-existent intf_idx=%d with table_index=%d for delete",
      88             :              sw_if_index, pct[ti]);
      89           0 :           return VNET_API_ERROR_NO_SUCH_TABLE;
      90             :         }
      91             : 
      92             :       /* Return ok on ADD operaton if feature is already enabled */
      93           0 :       if (is_add &&
      94           0 :           pcm->classify_table_index_by_sw_if_index[ti][sw_if_index] != ~0)
      95           0 :         return 0;
      96             : 
      97           0 :       vnet_policer_classify_feature_enable (vm, pcm, sw_if_index, ti, is_add);
      98             : 
      99           0 :       if (is_add)
     100           0 :         pcm->classify_table_index_by_sw_if_index[ti][sw_if_index] = pct[ti];
     101             :       else
     102           0 :         pcm->classify_table_index_by_sw_if_index[ti][sw_if_index] = ~0;
     103             :     }
     104             : 
     105             : 
     106           0 :   return 0;
     107             : }
     108             : 
     109             : static clib_error_t *
     110           0 : set_policer_classify_command_fn (vlib_main_t * vm,
     111             :                                  unformat_input_t * input,
     112             :                                  vlib_cli_command_t * cmd)
     113             : {
     114           0 :   vnet_main_t *vnm = vnet_get_main ();
     115           0 :   u32 sw_if_index = ~0;
     116           0 :   u32 ip4_table_index = ~0;
     117           0 :   u32 ip6_table_index = ~0;
     118           0 :   u32 l2_table_index = ~0;
     119           0 :   u32 is_add = 1;
     120           0 :   u32 idx_cnt = 0;
     121             :   int rv;
     122             : 
     123           0 :   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
     124             :     {
     125           0 :       if (unformat (input, "interface %U", unformat_vnet_sw_interface,
     126             :                     vnm, &sw_if_index))
     127             :         ;
     128           0 :       else if (unformat (input, "ip4-table %d", &ip4_table_index))
     129           0 :         idx_cnt++;
     130           0 :       else if (unformat (input, "ip6-table %d", &ip6_table_index))
     131           0 :         idx_cnt++;
     132           0 :       else if (unformat (input, "l2-table %d", &l2_table_index))
     133           0 :         idx_cnt++;
     134           0 :       else if (unformat (input, "del"))
     135           0 :         is_add = 0;
     136             :       else
     137           0 :         break;
     138             :     }
     139             : 
     140           0 :   if (sw_if_index == ~0)
     141           0 :     return clib_error_return (0, "Interface must be specified.");
     142             : 
     143           0 :   if (!idx_cnt)
     144           0 :     return clib_error_return (0, "Table index should be specified.");
     145             : 
     146           0 :   if (idx_cnt > 1)
     147           0 :     return clib_error_return (0, "Only one table index per API is allowed.");
     148             : 
     149           0 :   rv = vnet_set_policer_classify_intfc (vm, sw_if_index, ip4_table_index,
     150             :                                         ip6_table_index, l2_table_index,
     151             :                                         is_add);
     152             : 
     153           0 :   switch (rv)
     154             :     {
     155           0 :     case 0:
     156           0 :       break;
     157             : 
     158           0 :     case VNET_API_ERROR_NO_MATCHING_INTERFACE:
     159           0 :       return clib_error_return (0, "No such interface");
     160             : 
     161           0 :     case VNET_API_ERROR_NO_SUCH_ENTRY:
     162           0 :       return clib_error_return (0, "No such classifier table");
     163             :     }
     164           0 :   return 0;
     165             : }
     166             : 
     167             : /* *INDENT-OFF* */
     168      272887 : VLIB_CLI_COMMAND (set_policer_classify_command, static) = {
     169             :     .path = "set policer classify",
     170             :     .short_help =
     171             :     "set policer classify interface <int> [ip4-table <index>]\n"
     172             :     "  [ip6-table <index>] [l2-table <index>] [del]",
     173             :     .function = set_policer_classify_command_fn,
     174             : };
     175             : /* *INDENT-ON* */
     176             : 
     177             : static uword
     178           0 : unformat_table_type (unformat_input_t * input, va_list * va)
     179             : {
     180           0 :   u32 *r = va_arg (*va, u32 *);
     181             :   u32 tid;
     182             : 
     183           0 :   if (unformat (input, "ip4"))
     184           0 :     tid = POLICER_CLASSIFY_TABLE_IP4;
     185           0 :   else if (unformat (input, "ip6"))
     186           0 :     tid = POLICER_CLASSIFY_TABLE_IP6;
     187           0 :   else if (unformat (input, "l2"))
     188           0 :     tid = POLICER_CLASSIFY_TABLE_L2;
     189             :   else
     190           0 :     return 0;
     191             : 
     192           0 :   *r = tid;
     193           0 :   return 1;
     194             : }
     195             : 
     196             : static clib_error_t *
     197           0 : show_policer_classify_command_fn (vlib_main_t * vm,
     198             :                                   unformat_input_t * input,
     199             :                                   vlib_cli_command_t * cmd)
     200             : {
     201           0 :   policer_classify_main_t *pcm = &policer_classify_main;
     202           0 :   u32 type = POLICER_CLASSIFY_N_TABLES;
     203             :   u32 *vec_tbl;
     204             :   int i;
     205             : 
     206           0 :   if (unformat (input, "type %U", unformat_table_type, &type))
     207             :     ;
     208             :   else
     209           0 :     return clib_error_return (0, "Type must be specified.");;
     210             : 
     211           0 :   if (type == POLICER_CLASSIFY_N_TABLES)
     212           0 :     return clib_error_return (0, "Invalid table type.");
     213             : 
     214           0 :   vec_tbl = pcm->classify_table_index_by_sw_if_index[type];
     215             : 
     216           0 :   if (vec_len (vec_tbl))
     217           0 :     vlib_cli_output (vm, "%10s%20s\t\t%s", "Intfc idx", "Classify table",
     218             :                      "Interface name");
     219             :   else
     220           0 :     vlib_cli_output (vm, "No tables configured.");
     221             : 
     222           0 :   for (i = 0; i < vec_len (vec_tbl); i++)
     223             :     {
     224           0 :       if (vec_elt (vec_tbl, i) == ~0)
     225           0 :         continue;
     226             : 
     227           0 :       vlib_cli_output (vm, "%10d%20d\t\t%U", i, vec_elt (vec_tbl, i),
     228             :                        format_vnet_sw_if_index_name, pcm->vnet_main, i);
     229             :     }
     230             : 
     231           0 :   return 0;
     232             : }
     233             : 
     234             : /* *INDENT-OFF* */
     235      272887 : VLIB_CLI_COMMAND (show_policer_classify_command, static) = {
     236             :     .path = "show classify policer",
     237             :     .short_help = "show classify policer type [ip4|ip6|l2]",
     238             :     .function = show_policer_classify_command_fn,
     239             : };
     240             : /* *INDENT-ON* */
     241             : 
     242             : /*
     243             :  * fd.io coding-style-patch-verification: ON
     244             :  *
     245             :  * Local Variables:
     246             :  * eval: (c-set-style "gnu")
     247             :  * End:
     248             :  */

Generated by: LCOV version 1.14