1 /* SPDX-License-Identifier: Apache-2.0
2 * Copyright(c) 2022 Cisco Systems, Inc.
7 * @brief SR Path Tracing (PT)
13 #include <vlib/vlib.h>
14 #include <vnet/vnet.h>
15 #include <vnet/srv6/sr.h>
16 #include <vnet/ip/ip.h>
17 #include <vnet/srv6/sr_packet.h>
18 #include <vnet/ip/ip6_packet.h>
19 #include <vnet/fib/ip6_fib.h>
20 #include <vnet/dpo/dpo.h>
21 #include <vnet/adj/adj.h>
22 #include <vnet/srv6/sr_pt.h>
24 #include <vppinfra/error.h>
25 #include <vppinfra/elog.h>
27 sr_pt_main_t sr_pt_main;
30 sr_pt_find_iface (u32 iface)
32 sr_pt_main_t *sr_pt = &sr_pt_main;
35 /* Search for the item */
36 p = mhash_get (&sr_pt->sr_pt_iface_index_hash, &iface);
39 /* Retrieve sr_pt_iface */
40 return pool_elt_at_index (sr_pt->sr_pt_iface, p[0]);
46 sr_pt_add_iface (u32 iface, u16 id, u8 ingress_load, u8 egress_load,
49 sr_pt_main_t *sr_pt = &sr_pt_main;
52 sr_pt_iface_t *ls = 0;
54 if (iface == (u32) ~0)
55 return SR_PT_ERR_IFACE_INVALID;
57 /* Search for the item */
58 p = mhash_get (&sr_pt->sr_pt_iface_index_hash, &iface);
61 return SR_PT_ERR_EXIST;
63 if (id > SR_PT_ID_MAX)
64 return SR_PT_ERR_ID_INVALID;
66 if (ingress_load > SR_PT_LOAD_MAX || egress_load > SR_PT_LOAD_MAX)
67 return SR_PT_ERR_LOAD_INVALID;
69 if (tts_template > SR_PT_TTS_TEMPLATE_MAX)
70 return SR_PT_ERR_TTS_TEMPLATE_INVALID;
72 /* Create a new sr_pt_iface */
73 pool_get_zero (sr_pt->sr_pt_iface, ls);
76 ls->ingress_load = ingress_load;
77 ls->egress_load = egress_load;
78 ls->tts_template = tts_template;
80 /* Set hash key for searching sr_pt_iface by iface */
81 mhash_set (&sr_pt->sr_pt_iface_index_hash, &iface, ls - sr_pt->sr_pt_iface,
87 sr_pt_del_iface (u32 iface)
89 sr_pt_main_t *sr_pt = &sr_pt_main;
92 sr_pt_iface_t *ls = 0;
94 if (iface == (u32) ~0)
95 return SR_PT_ERR_IFACE_INVALID;
97 /* Search for the item */
98 p = mhash_get (&sr_pt->sr_pt_iface_index_hash, &iface);
102 /* Retrieve sr_pt_iface */
103 ls = pool_elt_at_index (sr_pt->sr_pt_iface, p[0]);
104 /* Delete sr_pt_iface */
105 pool_put (sr_pt->sr_pt_iface, ls);
106 mhash_unset (&sr_pt->sr_pt_iface_index_hash, &iface, NULL);
110 return SR_PT_ERR_NOENT;
116 * @brief "sr pt add iface" CLI function.
118 * @see sr_pt_add_iface
120 static clib_error_t *
121 sr_pt_add_iface_command_fn (vlib_main_t *vm, unformat_input_t *input,
122 vlib_cli_command_t *cmd)
124 vnet_main_t *vnm = vnet_get_main ();
125 u32 iface = (u32) ~0;
127 u32 ingress_load = 0;
129 u32 tts_template = SR_PT_TTS_TEMPLATE_DEFAULT;
133 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
135 if (unformat (input, "%U", unformat_vnet_sw_interface, vnm, &iface))
137 else if (unformat (input, "id %u", &id))
139 else if (unformat (input, "ingress-load %u", &ingress_load))
141 else if (unformat (input, "egress-load %u", &egress_load))
143 else if (unformat (input, "tts-template %u", &tts_template))
149 rv = sr_pt_add_iface (iface, id, ingress_load, egress_load, tts_template);
155 case SR_PT_ERR_EXIST:
156 return clib_error_return (0, "Error: Identical iface already exists.");
157 case SR_PT_ERR_IFACE_INVALID:
158 return clib_error_return (0, "Error: The iface name invalid.");
159 case SR_PT_ERR_ID_INVALID:
160 return clib_error_return (0, "Error: The iface id value invalid.");
161 case SR_PT_ERR_LOAD_INVALID:
162 return clib_error_return (
163 0, "Error: The iface ingress or egress load value invalid.");
164 case SR_PT_ERR_TTS_TEMPLATE_INVALID:
165 return clib_error_return (
166 0, "Error: The iface TTS Template value invalid.");
168 return clib_error_return (0, "Error: unknown error.");
174 * @brief "sr pt del iface" CLI function.
176 * @see sr_pt_del_iface
178 static clib_error_t *
179 sr_pt_del_iface_command_fn (vlib_main_t *vm, unformat_input_t *input,
180 vlib_cli_command_t *cmd)
182 vnet_main_t *vnm = vnet_get_main ();
183 u32 iface = (u32) ~0;
187 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
189 if (unformat (input, "%U", unformat_vnet_sw_interface, vnm, &iface))
195 rv = sr_pt_del_iface (iface);
201 case SR_PT_ERR_NOENT:
202 return clib_error_return (0, "Error: No such iface.");
203 case SR_PT_ERR_IFACE_INVALID:
204 return clib_error_return (0, "Error: The iface name is not valid.");
206 return clib_error_return (0, "Error: unknown error.");
212 * @brief CLI function to show all SR PT interfcaes
214 static clib_error_t *
215 sr_pt_show_iface_command_fn (vlib_main_t *vm, unformat_input_t *input,
216 vlib_cli_command_t *cmd)
218 vnet_main_t *vnm = vnet_get_main ();
219 sr_pt_main_t *sr_pt = &sr_pt_main;
220 sr_pt_iface_t **sr_pt_iface_list = 0;
224 vlib_cli_output (vm, "SR PT Interfaces");
225 vlib_cli_output (vm, "==================================");
227 pool_foreach (ls, sr_pt->sr_pt_iface)
229 vec_add1 (sr_pt_iface_list, ls);
232 for (i = 0; i < vec_len (sr_pt_iface_list); i++)
234 ls = sr_pt_iface_list[i];
237 "\tiface : \t%U\n\tid : \t%d\n\tingress-load: "
238 "\t%d\n\tegress-load : \t%d\n\ttts-template: \t%d ",
239 format_vnet_sw_if_index_name, vnm, ls->iface, ls->id, ls->ingress_load,
240 ls->egress_load, ls->tts_template);
241 vlib_cli_output (vm, "--------------------------------");
247 VLIB_CLI_COMMAND (sr_pt_add_iface_command, static) = {
248 .path = "sr pt add iface",
249 .short_help = "sr pt add iface <iface-name> id <pt-iface-id> ingress-load "
250 "<ingress-load-value> egress-load <egress-load-value> "
251 "tts-template <tts-template-value>",
252 .function = sr_pt_add_iface_command_fn,
255 VLIB_CLI_COMMAND (sr_pt_del_iface_command, static) = {
256 .path = "sr pt del iface",
257 .short_help = "sr pt del iface <iface-name>",
258 .function = sr_pt_del_iface_command_fn,
261 VLIB_CLI_COMMAND (sr_pt_show_iface_command, static) = {
262 .path = "sr pt show iface",
263 .short_help = "sr pt show iface",
264 .function = sr_pt_show_iface_command_fn,
268 * * @brief SR PT initialization
271 sr_pt_init (vlib_main_t *vm)
273 sr_pt_main_t *pt = &sr_pt_main;
274 mhash_init (&pt->sr_pt_iface_index_hash, sizeof (uword), sizeof (u32));
278 VLIB_INIT_FUNCTION (sr_pt_init);