| /* |
| * Copyright (c) 2015 Cisco and/or its affiliates. |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at: |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include <stdint.h> |
| #include <vnet/policer/policer.h> |
| #include <vnet/classify/vnet_classify.h> |
| |
| vnet_policer_main_t vnet_policer_main; |
| |
| clib_error_t * |
| policer_add_del (vlib_main_t * vm, |
| u8 * name, |
| sse2_qos_pol_cfg_params_st * cfg, |
| u32 * policer_index, u8 is_add) |
| { |
| vnet_policer_main_t *pm = &vnet_policer_main; |
| policer_read_response_type_st test_policer; |
| policer_read_response_type_st *policer; |
| uword *p; |
| u32 pi; |
| int rv; |
| |
| p = hash_get_mem (pm->policer_config_by_name, name); |
| |
| if (is_add == 0) |
| { |
| /* free policer config and template */ |
| if (p == 0) |
| { |
| vec_free (name); |
| return clib_error_return (0, "No such policer configuration"); |
| } |
| pool_put_index (pm->configs, p[0]); |
| pool_put_index (pm->policer_templates, p[0]); |
| hash_unset_mem (pm->policer_config_by_name, name); |
| |
| /* free policer */ |
| p = hash_get_mem (pm->policer_index_by_name, name); |
| if (p == 0) |
| { |
| vec_free (name); |
| return clib_error_return (0, "No such policer"); |
| } |
| pool_put_index (pm->policers, p[0]); |
| hash_unset_mem (pm->policer_index_by_name, name); |
| |
| vec_free (name); |
| return 0; |
| } |
| |
| if (p != 0) |
| { |
| vec_free (name); |
| return clib_error_return (0, "Policer already exists"); |
| } |
| |
| /* Vet the configuration before adding it to the table */ |
| rv = sse2_pol_logical_2_physical (cfg, &test_policer); |
| |
| if (rv == 0) |
| { |
| policer_read_response_type_st *pp; |
| sse2_qos_pol_cfg_params_st *cp; |
| |
| pool_get (pm->configs, cp); |
| pool_get (pm->policer_templates, pp); |
| |
| ASSERT (cp - pm->configs == pp - pm->policer_templates); |
| |
| clib_memcpy (cp, cfg, sizeof (*cp)); |
| clib_memcpy (pp, &test_policer, sizeof (*pp)); |
| |
| hash_set_mem (pm->policer_config_by_name, name, cp - pm->configs); |
| pool_get_aligned (pm->policers, policer, CLIB_CACHE_LINE_BYTES); |
| policer[0] = pp[0]; |
| pi = policer - pm->policers; |
| hash_set_mem (pm->policer_index_by_name, name, pi); |
| *policer_index = pi; |
| } |
| else |
| { |
| vec_free (name); |
| return clib_error_return (0, "Config failed sanity check"); |
| } |
| |
| return 0; |
| } |
| |
| u8 * |
| format_policer_instance (u8 * s, va_list * va) |
| { |
| policer_read_response_type_st *i |
| = va_arg (*va, policer_read_response_type_st *); |
| |
| s = format (s, "policer at %llx: %s rate, %s color-aware\n", |
| i, i->single_rate ? "single" : "dual", |
| i->color_aware ? "is" : "not"); |
| s = format (s, "cir %u tok/period, pir %u tok/period, scale %u\n", |
| i->cir_tokens_per_period, i->pir_tokens_per_period, i->scale); |
| s = format (s, "cur lim %u, cur bkt %u, ext lim %u, ext bkt %u\n", |
| i->current_limit, |
| i->current_bucket, i->extended_limit, i->extended_bucket); |
| s = format (s, "last update %llu\n", i->last_update_time); |
| return s; |
| } |
| |
| static u8 * |
| format_policer_round_type (u8 * s, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (c->rnd_type == SSE2_QOS_ROUND_TO_CLOSEST) |
| s = format (s, "closest"); |
| else if (c->rnd_type == SSE2_QOS_ROUND_TO_UP) |
| s = format (s, "up"); |
| else if (c->rnd_type == SSE2_QOS_ROUND_TO_DOWN) |
| s = format (s, "down"); |
| else |
| s = format (s, "ILLEGAL"); |
| return s; |
| } |
| |
| |
| static u8 * |
| format_policer_rate_type (u8 * s, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (c->rate_type == SSE2_QOS_RATE_KBPS) |
| s = format (s, "kbps"); |
| else if (c->rate_type == SSE2_QOS_RATE_PPS) |
| s = format (s, "pps"); |
| else |
| s = format (s, "ILLEGAL"); |
| return s; |
| } |
| |
| static u8 * |
| format_policer_type (u8 * s, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (c->rfc == SSE2_QOS_POLICER_TYPE_1R2C) |
| s = format (s, "1r2c"); |
| |
| else if (c->rfc == SSE2_QOS_POLICER_TYPE_1R3C_RFC_2697) |
| s = format (s, "1r3c"); |
| |
| else if (c->rfc == SSE2_QOS_POLICER_TYPE_2R3C_RFC_2698) |
| s = format (s, "2r3c-2698"); |
| |
| else if (c->rfc == SSE2_QOS_POLICER_TYPE_2R3C_RFC_4115) |
| s = format (s, "2r3c-4115"); |
| |
| else if (c->rfc == SSE2_QOS_POLICER_TYPE_2R3C_RFC_MEF5CF1) |
| s = format (s, "2r3c-mef5cf1"); |
| else |
| s = format (s, "ILLEGAL"); |
| return s; |
| } |
| |
| static u8 * |
| format_dscp (u8 * s, va_list * va) |
| { |
| u32 i = va_arg (*va, u32); |
| char *t = 0; |
| |
| switch (i) |
| { |
| #define _(v,f,str) case VNET_DSCP_##f: t = str; break; |
| foreach_vnet_dscp |
| #undef _ |
| default: |
| return format (s, "ILLEGAL"); |
| } |
| s = format (s, "%s", t); |
| return s; |
| } |
| |
| static u8 * |
| format_policer_action_type (u8 * s, va_list * va) |
| { |
| sse2_qos_pol_action_params_st *a |
| = va_arg (*va, sse2_qos_pol_action_params_st *); |
| |
| if (a->action_type == SSE2_QOS_ACTION_DROP) |
| s = format (s, "drop"); |
| else if (a->action_type == SSE2_QOS_ACTION_TRANSMIT) |
| s = format (s, "transmit"); |
| else if (a->action_type == SSE2_QOS_ACTION_MARK_AND_TRANSMIT) |
| s = format (s, "mark-and-transmit %U", format_dscp, a->dscp); |
| else |
| s = format (s, "ILLEGAL"); |
| return s; |
| } |
| |
| u8 * |
| format_policer_config (u8 * s, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| s = format (s, "type %U cir %u eir %u cb %u eb %u\n", |
| format_policer_type, c, |
| c->rb.kbps.cir_kbps, |
| c->rb.kbps.eir_kbps, c->rb.kbps.cb_bytes, c->rb.kbps.eb_bytes); |
| s = format (s, "rate type %U, round type %U\n", |
| format_policer_rate_type, c, format_policer_round_type, c); |
| s = format (s, "conform action %U, exceed action %U, violate action %U\n", |
| format_policer_action_type, &c->conform_action, |
| format_policer_action_type, &c->exceed_action, |
| format_policer_action_type, &c->violate_action); |
| return s; |
| } |
| |
| static uword |
| unformat_policer_type (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (!unformat (input, "type")) |
| return 0; |
| |
| if (unformat (input, "1r2c")) |
| c->rfc = SSE2_QOS_POLICER_TYPE_1R2C; |
| else if (unformat (input, "1r3c")) |
| c->rfc = SSE2_QOS_POLICER_TYPE_1R3C_RFC_2697; |
| else if (unformat (input, "2r3c-2698")) |
| c->rfc = SSE2_QOS_POLICER_TYPE_2R3C_RFC_2698; |
| else if (unformat (input, "2r3c-4115")) |
| c->rfc = SSE2_QOS_POLICER_TYPE_2R3C_RFC_4115; |
| else if (unformat (input, "2r3c-mef5cf1")) |
| c->rfc = SSE2_QOS_POLICER_TYPE_2R3C_RFC_MEF5CF1; |
| else |
| return 0; |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_round_type (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (!unformat (input, "round")) |
| return 0; |
| |
| if (unformat (input, "closest")) |
| c->rnd_type = SSE2_QOS_ROUND_TO_CLOSEST; |
| else if (unformat (input, "up")) |
| c->rnd_type = SSE2_QOS_ROUND_TO_UP; |
| else if (unformat (input, "down")) |
| c->rnd_type = SSE2_QOS_ROUND_TO_DOWN; |
| else |
| return 0; |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_rate_type (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (!unformat (input, "rate")) |
| return 0; |
| |
| if (unformat (input, "kbps")) |
| c->rate_type = SSE2_QOS_RATE_KBPS; |
| else if (unformat (input, "pps")) |
| c->rate_type = SSE2_QOS_RATE_PPS; |
| else |
| return 0; |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_cir (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (unformat (input, "cir %u", &c->rb.kbps.cir_kbps)) |
| return 1; |
| return 0; |
| } |
| |
| static uword |
| unformat_policer_eir (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (unformat (input, "eir %u", &c->rb.kbps.eir_kbps)) |
| return 1; |
| return 0; |
| } |
| |
| static uword |
| unformat_policer_cb (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (unformat (input, "cb %u", &c->rb.kbps.cb_bytes)) |
| return 1; |
| return 0; |
| } |
| |
| static uword |
| unformat_policer_eb (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (unformat (input, "eb %u", &c->rb.kbps.eb_bytes)) |
| return 1; |
| return 0; |
| } |
| |
| static uword |
| unformat_dscp (unformat_input_t * input, va_list * va) |
| { |
| u8 *r = va_arg (*va, u8 *); |
| |
| if (0); |
| #define _(v,f,str) else if (unformat (input, str)) *r = VNET_DSCP_##f; |
| foreach_vnet_dscp |
| #undef _ |
| else |
| return 0; |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_action_type (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_action_params_st *a |
| = va_arg (*va, sse2_qos_pol_action_params_st *); |
| |
| if (unformat (input, "drop")) |
| a->action_type = SSE2_QOS_ACTION_DROP; |
| else if (unformat (input, "transmit")) |
| a->action_type = SSE2_QOS_ACTION_TRANSMIT; |
| else if (unformat (input, "mark-and-transmit %U", unformat_dscp, &a->dscp)) |
| a->action_type = SSE2_QOS_ACTION_MARK_AND_TRANSMIT; |
| else |
| return 0; |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_action (unformat_input_t * input, va_list * va) |
| { |
| sse2_qos_pol_cfg_params_st *c = va_arg (*va, sse2_qos_pol_cfg_params_st *); |
| |
| if (unformat (input, "conform-action %U", unformat_policer_action_type, |
| &c->conform_action)) |
| return 1; |
| else if (unformat (input, "exceed-action %U", unformat_policer_action_type, |
| &c->exceed_action)) |
| return 1; |
| else if (unformat (input, "violate-action %U", unformat_policer_action_type, |
| &c->violate_action)) |
| return 1; |
| return 0; |
| } |
| |
| static uword |
| unformat_policer_classify_next_index (unformat_input_t * input, va_list * va) |
| { |
| u32 *r = va_arg (*va, u32 *); |
| vnet_policer_main_t *pm = &vnet_policer_main; |
| uword *p; |
| u8 *match_name = 0; |
| |
| if (unformat (input, "%s", &match_name)) |
| ; |
| else |
| return 0; |
| |
| p = hash_get_mem (pm->policer_index_by_name, match_name); |
| |
| if (p == 0) |
| return 0; |
| |
| *r = p[0]; |
| |
| return 1; |
| } |
| |
| static uword |
| unformat_policer_classify_precolor (unformat_input_t * input, va_list * va) |
| { |
| u32 *r = va_arg (*va, u32 *); |
| |
| if (unformat (input, "conform-color")) |
| *r = POLICE_CONFORM; |
| else if (unformat (input, "exceed-color")) |
| *r = POLICE_EXCEED; |
| else |
| return 0; |
| |
| return 1; |
| } |
| |
| #define foreach_config_param \ |
| _(eb) \ |
| _(cb) \ |
| _(eir) \ |
| _(cir) \ |
| _(rate_type) \ |
| _(round_type) \ |
| _(type) \ |
| _(action) |
| |
| static clib_error_t * |
| configure_policer_command_fn (vlib_main_t * vm, |
| unformat_input_t * input, |
| vlib_cli_command_t * cmd) |
| { |
| sse2_qos_pol_cfg_params_st c; |
| unformat_input_t _line_input, *line_input = &_line_input; |
| u8 is_add = 1; |
| u8 *name = 0; |
| u32 pi; |
| clib_error_t *error = NULL; |
| |
| /* Get a line of input. */ |
| if (!unformat_user (input, unformat_line_input, line_input)) |
| return 0; |
| |
| memset (&c, 0, sizeof (c)); |
| |
| while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT) |
| { |
| if (unformat (line_input, "del")) |
| is_add = 0; |
| else if (unformat (line_input, "name %s", &name)) |
| ; |
| else if (unformat (line_input, "color-aware")) |
| c.color_aware = 1; |
| |
| #define _(a) else if (unformat (line_input, "%U", unformat_policer_##a, &c)) ; |
| foreach_config_param |
| #undef _ |
| else |
| { |
| error = clib_error_return (0, "unknown input `%U'", |
| format_unformat_error, line_input); |
| goto done; |
| } |
| } |
| |
| error = policer_add_del (vm, name, &c, &pi, is_add); |
| |
| done: |
| unformat_free (line_input); |
| |
| return error; |
| } |
| |
| /* *INDENT-OFF* */ |
| VLIB_CLI_COMMAND (configure_policer_command, static) = { |
| .path = "configure policer", |
| .short_help = "configure policer name <name> <params> ", |
| .function = configure_policer_command_fn, |
| }; |
| /* *INDENT-ON* */ |
| |
| static clib_error_t * |
| show_policer_command_fn (vlib_main_t * vm, |
| unformat_input_t * input, vlib_cli_command_t * cmd) |
| { |
| vnet_policer_main_t *pm = &vnet_policer_main; |
| hash_pair_t *p; |
| u32 pool_index; |
| u8 *match_name = 0; |
| u8 *name; |
| sse2_qos_pol_cfg_params_st *config; |
| policer_read_response_type_st *templ; |
| |
| (void) unformat (input, "name %s", &match_name); |
| |
| /* *INDENT-OFF* */ |
| hash_foreach_pair (p, pm->policer_config_by_name, |
| ({ |
| name = (u8 *) p->key; |
| if (match_name == 0 || !strcmp((char *) name, (char *) match_name)) |
| { |
| pool_index = p->value[0]; |
| config = pool_elt_at_index (pm->configs, pool_index); |
| templ = pool_elt_at_index (pm->policer_templates, pool_index); |
| vlib_cli_output (vm, "Name \"%s\" %U ", |
| name, format_policer_config, config); |
| vlib_cli_output (vm, "Template %U", |
| format_policer_instance, templ); |
| vlib_cli_output (vm, "-----------"); |
| } |
| })); |
| /* *INDENT-ON* */ |
| return 0; |
| } |
| |
| |
| /* *INDENT-OFF* */ |
| VLIB_CLI_COMMAND (show_policer_command, static) = { |
| .path = "show policer", |
| .short_help = "show policer [name]", |
| .function = show_policer_command_fn, |
| }; |
| /* *INDENT-ON* */ |
| |
| static clib_error_t * |
| show_policer_pools_command_fn (vlib_main_t * vm, |
| unformat_input_t * input, |
| vlib_cli_command_t * cmd) |
| { |
| vnet_policer_main_t *pm = &vnet_policer_main; |
| |
| vlib_cli_output (vm, "pool sizes: configs=%d templates=%d policers=%d", |
| pool_elts (pm->configs), |
| pool_elts (pm->policer_templates), |
| pool_elts (pm->policers)); |
| return 0; |
| } |
| /* *INDENT-OFF* */ |
| VLIB_CLI_COMMAND (show_policer_pools_command, static) = { |
| .path = "show policer pools", |
| .short_help = "show policer pools", |
| .function = show_policer_pools_command_fn, |
| }; |
| /* *INDENT-ON* */ |
| |
| clib_error_t * |
| policer_init (vlib_main_t * vm) |
| { |
| vnet_policer_main_t *pm = &vnet_policer_main; |
| void vnet_policer_node_funcs_reference (void); |
| |
| vnet_policer_node_funcs_reference (); |
| |
| pm->vlib_main = vm; |
| pm->vnet_main = vnet_get_main (); |
| |
| pm->policer_config_by_name = hash_create_string (0, sizeof (uword)); |
| pm->policer_index_by_name = hash_create_string (0, sizeof (uword)); |
| |
| vnet_classify_register_unformat_policer_next_index_fn |
| (unformat_policer_classify_next_index); |
| vnet_classify_register_unformat_opaque_index_fn |
| (unformat_policer_classify_precolor); |
| |
| return 0; |
| } |
| |
| VLIB_INIT_FUNCTION (policer_init); |
| |
| |
| |
| /* |
| * fd.io coding-style-patch-verification: ON |
| * |
| * Local Variables: |
| * eval: (c-set-style "gnu") |
| * End: |
| */ |