blob: c6a13fb15e00bc55b1fad8ab175ce5666b37bb2d [file] [log] [blame]
Ed Warnickecb9cada2015-12-08 15:45:58 -07001/*
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 * node.c: VLIB processing nodes
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 <vlib/vlib.h>
41#include <vlib/threads.h>
42
43/* Query node given name. */
Dave Barach9b8ffd92016-07-08 08:13:45 -040044vlib_node_t *
45vlib_get_node_by_name (vlib_main_t * vm, u8 * name)
Ed Warnickecb9cada2015-12-08 15:45:58 -070046{
Dave Barach9b8ffd92016-07-08 08:13:45 -040047 vlib_node_main_t *nm = &vm->node_main;
48 uword *p;
49 u8 *key = name;
Nathan Skrzypczak4b2946a2020-09-03 11:37:56 +020050 key = format (0, "%s", key);
Ed Warnickecb9cada2015-12-08 15:45:58 -070051 p = hash_get (nm->node_by_name, key);
52 if (key != name)
53 vec_free (key);
54 return p ? vec_elt (nm->nodes, p[0]) : 0;
55}
56
Dave Barach9b8ffd92016-07-08 08:13:45 -040057static void
58node_set_elog_name (vlib_main_t * vm, uword node_index)
Ed Warnickecb9cada2015-12-08 15:45:58 -070059{
Dave Barach9b8ffd92016-07-08 08:13:45 -040060 vlib_node_t *n = vlib_get_node (vm, node_index);
61 elog_event_type_t *t;
Ed Warnickecb9cada2015-12-08 15:45:58 -070062
63 t = vec_elt_at_index (vm->node_call_elog_event_types, node_index);
64 vec_free (t->format);
Dave Barachfb6e59d2016-03-26 18:45:42 -040065 t->format = (char *) format (0, "%v-call: %%d%c", n->name, 0);
Ed Warnickecb9cada2015-12-08 15:45:58 -070066
67 t = vec_elt_at_index (vm->node_return_elog_event_types, node_index);
68 vec_free (t->format);
Dave Barachfb6e59d2016-03-26 18:45:42 -040069 t->format = (char *) format (0, "%v-return: %%d%c", n->name, 0);
Ed Warnickecb9cada2015-12-08 15:45:58 -070070
Damjan Marionf553a2c2021-03-26 13:45:37 +010071 n->name_elog_string =
72 elog_string (&vlib_global_main.elog_main, "%v%c", n->name, 0);
Ed Warnickecb9cada2015-12-08 15:45:58 -070073}
74
Dave Barach9b8ffd92016-07-08 08:13:45 -040075void
76vlib_node_rename (vlib_main_t * vm, u32 node_index, char *fmt, ...)
Ed Warnickecb9cada2015-12-08 15:45:58 -070077{
78 va_list va;
Dave Barach9b8ffd92016-07-08 08:13:45 -040079 vlib_node_main_t *nm = &vm->node_main;
80 vlib_node_t *n = vlib_get_node (vm, node_index);
Ed Warnickecb9cada2015-12-08 15:45:58 -070081
82 va_start (va, fmt);
83 hash_unset (nm->node_by_name, n->name);
84 vec_free (n->name);
85 n->name = va_format (0, fmt, &va);
86 va_end (va);
87 hash_set (nm->node_by_name, n->name, n->index);
88
89 node_set_elog_name (vm, node_index);
Igor Mikhailov (imichail)19e9d952017-07-03 17:01:50 -070090
91 /* Propagate the change to all worker threads */
Steven Luong1eae8ec2020-06-10 23:38:41 -070092 vlib_worker_thread_node_runtime_update ();
Ed Warnickecb9cada2015-12-08 15:45:58 -070093}
94
95static void
Dave Barach9b8ffd92016-07-08 08:13:45 -040096vlib_node_runtime_update (vlib_main_t * vm, u32 node_index, u32 next_index)
Ed Warnickecb9cada2015-12-08 15:45:58 -070097{
Dave Barach9b8ffd92016-07-08 08:13:45 -040098 vlib_node_main_t *nm = &vm->node_main;
99 vlib_node_runtime_t *r, *s;
100 vlib_node_t *node, *next_node;
101 vlib_next_frame_t *nf;
102 vlib_pending_frame_t *pf;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700103 i32 i, j, n_insert;
104
Ed Warnickecb9cada2015-12-08 15:45:58 -0700105 node = vec_elt (nm->nodes, node_index);
106 r = vlib_node_get_runtime (vm, node_index);
107
108 n_insert = vec_len (node->next_nodes) - r->n_next_nodes;
109 if (n_insert > 0)
110 {
111 i = r->next_frame_index + r->n_next_nodes;
112 vec_insert (nm->next_frames, n_insert, i);
113
114 /* Initialize newly inserted next frames. */
115 for (j = 0; j < n_insert; j++)
116 vlib_next_frame_init (nm->next_frames + i + j);
117
118 /* Relocate other next frames at higher indices. */
119 for (j = 0; j < vec_len (nm->nodes); j++)
120 {
121 s = vlib_node_get_runtime (vm, j);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400122 if (j != node_index && s->next_frame_index >= i)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700123 s->next_frame_index += n_insert;
124 }
125
126 /* Pending frames may need to be relocated also. */
127 vec_foreach (pf, nm->pending_frames)
Dave Barach9b8ffd92016-07-08 08:13:45 -0400128 {
129 if (pf->next_frame_index != VLIB_PENDING_FRAME_NO_NEXT_FRAME
130 && pf->next_frame_index >= i)
131 pf->next_frame_index += n_insert;
132 }
133 /* *INDENT-OFF* */
Damjan Marionb2c31b62020-12-13 21:47:40 +0100134 pool_foreach (pf, nm->suspended_process_frames) {
Ed Warnickecb9cada2015-12-08 15:45:58 -0700135 if (pf->next_frame_index != ~0 && pf->next_frame_index >= i)
136 pf->next_frame_index += n_insert;
Damjan Marionb2c31b62020-12-13 21:47:40 +0100137 }
Dave Barach9b8ffd92016-07-08 08:13:45 -0400138 /* *INDENT-ON* */
Ed Warnickecb9cada2015-12-08 15:45:58 -0700139
140 r->n_next_nodes = vec_len (node->next_nodes);
141 }
142
143 /* Set frame's node runtime index. */
144 next_node = vlib_get_node (vm, node->next_nodes[next_index]);
145 nf = nm->next_frames + r->next_frame_index + next_index;
146 nf->node_runtime_index = next_node->runtime_index;
147
Dave Barach9b8ffd92016-07-08 08:13:45 -0400148 vlib_worker_thread_node_runtime_update ();
Ed Warnickecb9cada2015-12-08 15:45:58 -0700149}
150
Neale Rannsbb620d72017-06-29 00:19:08 -0700151uword
152vlib_node_get_next (vlib_main_t * vm, uword node_index, uword next_node_index)
153{
154 vlib_node_main_t *nm = &vm->node_main;
155 vlib_node_t *node;
156 uword *p;
157
158 node = vec_elt (nm->nodes, node_index);
159
160 /* Runtime has to be initialized. */
161 ASSERT (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED);
162
163 if ((p = hash_get (node->next_slot_by_node, next_node_index)))
164 {
165 return p[0];
166 }
167
168 return (~0);
169}
170
Ed Warnickecb9cada2015-12-08 15:45:58 -0700171/* Add next node to given node in given slot. */
172uword
173vlib_node_add_next_with_slot (vlib_main_t * vm,
174 uword node_index,
Dave Barach9b8ffd92016-07-08 08:13:45 -0400175 uword next_node_index, uword slot)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700176{
Dave Barach9b8ffd92016-07-08 08:13:45 -0400177 vlib_node_main_t *nm = &vm->node_main;
Christian Hopps2e8b0612019-11-03 00:59:49 -0400178 vlib_node_t *node, *next, *old_next;
179 u32 old_next_index;
Dave Barach9b8ffd92016-07-08 08:13:45 -0400180 uword *p;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700181
Christian E. Hoppsd3122ef2019-09-28 21:36:36 -0400182 ASSERT (vlib_get_thread_index () == 0);
183
Ed Warnickecb9cada2015-12-08 15:45:58 -0700184 node = vec_elt (nm->nodes, node_index);
185 next = vec_elt (nm->nodes, next_node_index);
186
Ole Troan964f93e2016-06-10 13:22:36 +0200187 /* Runtime has to be initialized. */
Dave Barach9b8ffd92016-07-08 08:13:45 -0400188 ASSERT (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700189
190 if ((p = hash_get (node->next_slot_by_node, next_node_index)))
191 {
John Lo405e41b2016-04-23 15:14:12 -0400192 /* Next already exists: slot must match. */
193 if (slot != ~0)
194 ASSERT (slot == p[0]);
195 return p[0];
Ed Warnickecb9cada2015-12-08 15:45:58 -0700196 }
197
Christian E. Hoppsd3122ef2019-09-28 21:36:36 -0400198 vlib_worker_thread_barrier_sync (vm);
199
Ed Warnickecb9cada2015-12-08 15:45:58 -0700200 if (slot == ~0)
201 slot = vec_len (node->next_nodes);
202
203 vec_validate_init_empty (node->next_nodes, slot, ~0);
204 vec_validate (node->n_vectors_by_next_node, slot);
205
Christian Hopps2e8b0612019-11-03 00:59:49 -0400206 if ((old_next_index = node->next_nodes[slot]) != ~0u)
207 {
208 hash_unset (node->next_slot_by_node, old_next_index);
209 old_next = vlib_get_node (vm, old_next_index);
210 old_next->prev_node_bitmap =
211 clib_bitmap_andnoti (old_next->prev_node_bitmap, node_index);
212 }
213
Ed Warnickecb9cada2015-12-08 15:45:58 -0700214 node->next_nodes[slot] = next_node_index;
John Lo405e41b2016-04-23 15:14:12 -0400215 hash_set (node->next_slot_by_node, next_node_index, slot);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700216
217 vlib_node_runtime_update (vm, node_index, slot);
218
219 next->prev_node_bitmap = clib_bitmap_ori (next->prev_node_bitmap,
220 node_index);
221
222 /* Siblings all get same node structure. */
223 {
224 uword sib_node_index, sib_slot;
Dave Barach9b8ffd92016-07-08 08:13:45 -0400225 vlib_node_t *sib_node;
226 /* *INDENT-OFF* */
Damjan Marionf0ca1e82020-12-13 23:26:56 +0100227 clib_bitmap_foreach (sib_node_index, node->sibling_bitmap) {
Ed Warnickecb9cada2015-12-08 15:45:58 -0700228 sib_node = vec_elt (nm->nodes, sib_node_index);
229 if (sib_node != node)
230 {
231 sib_slot = vlib_node_add_next_with_slot (vm, sib_node_index, next_node_index, slot);
232 ASSERT (sib_slot == slot);
233 }
Damjan Marionf0ca1e82020-12-13 23:26:56 +0100234 }
Dave Barach9b8ffd92016-07-08 08:13:45 -0400235 /* *INDENT-ON* */
Ed Warnickecb9cada2015-12-08 15:45:58 -0700236 }
237
Christian E. Hoppsd3122ef2019-09-28 21:36:36 -0400238 vlib_worker_thread_barrier_release (vm);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700239 return slot;
240}
241
242/* Add named next node to given node in given slot. */
243uword
244vlib_node_add_named_next_with_slot (vlib_main_t * vm,
Dave Barach9b8ffd92016-07-08 08:13:45 -0400245 uword node, char *name, uword slot)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700246{
Dave Barach9b8ffd92016-07-08 08:13:45 -0400247 vlib_node_main_t *nm;
248 vlib_node_t *n, *n_next;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700249
250 nm = &vm->node_main;
251 n = vlib_get_node (vm, node);
252
253 n_next = vlib_get_node_by_name (vm, (u8 *) name);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400254 if (!n_next)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700255 {
256 if (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED)
257 return ~0;
258
259 if (slot == ~0)
260 slot = clib_max (vec_len (n->next_node_names),
261 vec_len (n->next_nodes));
262 vec_validate (n->next_node_names, slot);
263 n->next_node_names[slot] = name;
264 return slot;
265 }
266
267 return vlib_node_add_next_with_slot (vm, node, n_next->index, slot);
268}
269
Dave Barach9b8ffd92016-07-08 08:13:45 -0400270static void
271node_elog_init (vlib_main_t * vm, uword ni)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700272{
273 elog_event_type_t t;
274
Dave Barachb7b92992018-10-17 10:38:51 -0400275 clib_memset (&t, 0, sizeof (t));
Ed Warnickecb9cada2015-12-08 15:45:58 -0700276
277 /* 2 event types for this node: one when node function is called.
278 One when it returns. */
279 vec_validate (vm->node_call_elog_event_types, ni);
280 vm->node_call_elog_event_types[ni] = t;
281
282 vec_validate (vm->node_return_elog_event_types, ni);
283 vm->node_return_elog_event_types[ni] = t;
284
285 node_set_elog_name (vm, ni);
286}
287
288#ifdef CLIB_UNIX
Dave Barachbfdedbd2016-01-20 09:11:55 -0500289#define STACK_ALIGN (clib_mem_get_page_size())
Ed Warnickecb9cada2015-12-08 15:45:58 -0700290#else
291#define STACK_ALIGN CLIB_CACHE_LINE_BYTES
292#endif
293
Damjan Mariona31698b2021-03-10 14:35:28 +0100294vlib_node_function_t *
295vlib_node_get_preferred_node_fn_variant (vlib_main_t *vm,
296 vlib_node_fn_registration_t *regs)
297{
298 vlib_node_main_t *nm = &vm->node_main;
299 vlib_node_fn_registration_t *r;
300 vlib_node_fn_variant_t *v;
301 vlib_node_function_t *fn = 0;
302 int priority = -1;
303
304 if (nm->node_fn_default_march_variant != ~0)
305 {
306 r = regs;
307 while (r)
308 {
309 if (r->march_variant == nm->node_fn_default_march_variant)
310 return r->function;
311 r = r->next_registration;
312 }
313 }
314
315 r = regs;
316 while (r)
317 {
318 v = vec_elt_at_index (nm->variants, r->march_variant);
319 if (v->priority > priority)
320 {
321 priority = v->priority;
322 fn = r->function;
323 }
324 r = r->next_registration;
325 }
326
327 ASSERT (fn);
328 return fn;
329}
330
Damjan Marionf87acfa2022-03-23 17:36:56 +0100331u32
332vlib_register_node (vlib_main_t *vm, vlib_node_registration_t *r, char *fmt,
333 ...)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700334{
Dave Barach9b8ffd92016-07-08 08:13:45 -0400335 vlib_node_main_t *nm = &vm->node_main;
336 vlib_node_t *n;
Damjan Marionf87acfa2022-03-23 17:36:56 +0100337 va_list va;
Damjan Marionb32bd702021-12-23 17:05:02 +0100338 u32 size;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700339 int i;
340
341 if (CLIB_DEBUG > 0)
342 {
343 /* Default (0) type should match INTERNAL. */
Dave Barach9b8ffd92016-07-08 08:13:45 -0400344 vlib_node_t zero = { 0 };
Ed Warnickecb9cada2015-12-08 15:45:58 -0700345 ASSERT (VLIB_NODE_TYPE_INTERNAL == zero.type);
346 }
347
Damjan Marion812b32d2018-05-28 21:26:47 +0200348 if (r->node_fn_registrations)
349 {
Damjan Marion812b32d2018-05-28 21:26:47 +0200350 /* to avoid confusion, please remove ".function " statiement from
351 CLIB_NODE_REGISTRATION() if using function function candidates */
352 ASSERT (r->function == 0);
353
Damjan Mariona31698b2021-03-10 14:35:28 +0100354 r->function =
355 vlib_node_get_preferred_node_fn_variant (vm, r->node_fn_registrations);
Damjan Marion812b32d2018-05-28 21:26:47 +0200356 }
357
Ed Warnickecb9cada2015-12-08 15:45:58 -0700358 ASSERT (r->function != 0);
359
360 n = clib_mem_alloc_no_fail (sizeof (n[0]));
Dave Barachb7b92992018-10-17 10:38:51 -0400361 clib_memset (n, 0, sizeof (n[0]));
Ed Warnickecb9cada2015-12-08 15:45:58 -0700362 n->index = vec_len (nm->nodes);
Damjan Marion69abe442018-08-27 22:43:23 +0200363 n->node_fn_registrations = r->node_fn_registrations;
Dave Barach7fff3d22018-11-27 16:52:59 -0500364 n->protocol_hint = r->protocol_hint;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700365
366 vec_add1 (nm->nodes, n);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400367
Damjan Marionf87acfa2022-03-23 17:36:56 +0100368 va_start (va, fmt);
369 n->name = va_format (0, fmt, &va);
370 va_end (va);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700371
Dave Barach9b8ffd92016-07-08 08:13:45 -0400372 if (!nm->node_by_name)
373 nm->node_by_name = hash_create_vec ( /* size */ 32,
374 sizeof (n->name[0]), sizeof (uword));
Ed Warnickecb9cada2015-12-08 15:45:58 -0700375
376 /* Node names must be unique. */
377 {
Benoît Ganne268e3b62020-09-10 14:12:06 +0200378 /* vlib_get_node_by_name() expects NULL-terminated strings */
379 u8 *name = format (0, "%v%c", n->name, 0);
380 vlib_node_t *o = vlib_get_node_by_name (vm, name);
381 vec_free (name);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700382 if (o)
383 clib_error ("more than one node named `%v'", n->name);
384 }
385
386 hash_set (nm->node_by_name, n->name, n->index);
387
388 r->index = n->index; /* save index in registration */
389 n->function = r->function;
390
391 /* Node index of next sibling will be filled in by vlib_node_main_init. */
392 n->sibling_of = r->sibling_of;
Ole Troan964f93e2016-06-10 13:22:36 +0200393 if (r->sibling_of && r->n_next_nodes > 0)
394 clib_error ("sibling node should not have any next nodes `%v'", n->name);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700395
396 if (r->type == VLIB_NODE_TYPE_INTERNAL)
397 ASSERT (r->vector_size > 0);
398
399#define _(f) n->f = r->f
400
Dave Barach9b8ffd92016-07-08 08:13:45 -0400401 _(type);
402 _(flags);
403 _(state);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400404 _(format_buffer);
405 _(unformat_buffer);
406 _(format_trace);
407 _(validate_frame);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700408
Damjan Marionb32bd702021-12-23 17:05:02 +0100409 size = round_pow2 (sizeof (vlib_frame_t), VLIB_FRAME_DATA_ALIGN);
410
411 /* scalar data size */
412 if (r->scalar_size)
413 {
414 n->scalar_offset = size;
415 size += round_pow2 (r->scalar_size, VLIB_FRAME_DATA_ALIGN);
416 }
417 else
418 n->scalar_offset = 0;
419
420 /* Vecor data size */
421 n->vector_offset = size;
422 size += r->vector_size * VLIB_FRAME_SIZE;
423
424 /* Allocate a few extra slots of vector data to support
425 speculative vector enqueues which overflow vector data in next frame. */
426 size += r->vector_size * VLIB_FRAME_SIZE_EXTRA;
427
428 /* space for VLIB_FRAME_MAGIC */
429 n->magic_offset = size;
430 size += sizeof (u32);
431
432 /* round size to VLIB_FRAME_DATA_ALIGN */
433 size = round_pow2 (size, VLIB_FRAME_DATA_ALIGN);
434
435 if (r->aux_size)
436 {
437 n->aux_offset = size;
438 size += r->aux_size * VLIB_FRAME_SIZE;
439 }
440 else
441 n->aux_offset = 0;
442
443 /* final size */
444 n->frame_size = size = round_pow2 (size, CLIB_CACHE_LINE_BYTES);
445 ASSERT (size <= __UINT16_MAX__);
446
447 vlib_frame_size_t *fs = 0;
448
449 n->frame_size_index = (u16) ~0;
450 vec_foreach (fs, nm->frame_sizes)
451 if (fs->frame_size == size)
452 {
453 n->frame_size_index = fs - nm->frame_sizes;
454 break;
455 }
456
457 if (n->frame_size_index == (u16) ~0)
458 {
459 vec_add2 (nm->frame_sizes, fs, 1);
460 fs->frame_size = size;
461 n->frame_size_index = fs - nm->frame_sizes;
462 }
463
Ed Warnickecb9cada2015-12-08 15:45:58 -0700464 /* Register error counters. */
Ole Troan148c7b72020-10-07 18:05:37 +0200465 vlib_register_errors (vm, n->index, r->n_errors, r->error_strings,
466 r->error_counters);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700467 node_elog_init (vm, n->index);
468
Dave Barach9b8ffd92016-07-08 08:13:45 -0400469 _(runtime_data_bytes);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700470 if (r->runtime_data_bytes > 0)
471 {
472 vec_resize (n->runtime_data, r->runtime_data_bytes);
473 if (r->runtime_data)
Damjan Marionf1213b82016-03-13 02:22:06 +0100474 clib_memcpy (n->runtime_data, r->runtime_data, r->runtime_data_bytes);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700475 }
476
477 vec_resize (n->next_node_names, r->n_next_nodes);
478 for (i = 0; i < r->n_next_nodes; i++)
479 n->next_node_names[i] = r->next_nodes[i];
480
481 vec_validate_init_empty (n->next_nodes, r->n_next_nodes - 1, ~0);
482 vec_validate (n->n_vectors_by_next_node, r->n_next_nodes - 1);
483
484 n->owner_node_index = n->owner_next_index = ~0;
485
486 /* Initialize node runtime. */
487 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400488 vlib_node_runtime_t *rt;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700489 u32 i;
490
491 if (n->type == VLIB_NODE_TYPE_PROCESS)
492 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400493 vlib_process_t *p;
Damjan Marionfc639ff2020-09-11 22:25:34 +0200494 uword log2_n_stack_bytes;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700495
Damjan Marionef587582020-05-20 22:01:44 +0200496 log2_n_stack_bytes = clib_max (r->process_log2_n_stack_bytes,
497 VLIB_PROCESS_LOG2_STACK_SIZE);
498 log2_n_stack_bytes = clib_max (log2_n_stack_bytes,
Damjan Marionfc639ff2020-09-11 22:25:34 +0200499 clib_mem_get_log2_page_size ());
Ed Warnickecb9cada2015-12-08 15:45:58 -0700500
Damjan Marionef587582020-05-20 22:01:44 +0200501 p = clib_mem_alloc_aligned (sizeof (p[0]), CLIB_CACHE_LINE_BYTES);
Dave Barachb7b92992018-10-17 10:38:51 -0400502 clib_memset (p, 0, sizeof (p[0]));
Ed Warnickecb9cada2015-12-08 15:45:58 -0700503 p->log2_n_stack_bytes = log2_n_stack_bytes;
504
Damjan Marionfc639ff2020-09-11 22:25:34 +0200505 p->stack = clib_mem_vm_map_stack (1ULL << log2_n_stack_bytes,
506 CLIB_MEM_PAGE_SZ_DEFAULT,
507 "process stack: %U",
508 format_vlib_node_name, vm,
509 n->index);
Damjan Marionef587582020-05-20 22:01:44 +0200510
Damjan Marionfc639ff2020-09-11 22:25:34 +0200511 if (p->stack == CLIB_MEM_VM_MAP_FAILED)
Damjan Marionef587582020-05-20 22:01:44 +0200512 clib_panic ("failed to allocate process stack (%d bytes)",
Damjan Marionfc639ff2020-09-11 22:25:34 +0200513 1ULL << log2_n_stack_bytes);
Damjan Marionef587582020-05-20 22:01:44 +0200514
Ed Warnickecb9cada2015-12-08 15:45:58 -0700515 /* Process node's runtime index is really index into process
516 pointer vector. */
517 n->runtime_index = vec_len (nm->processes);
518
519 vec_add1 (nm->processes, p);
520
521 /* Paint first stack word with magic number so we can at least
522 detect process stack overruns. */
523 p->stack[0] = VLIB_PROCESS_STACK_MAGIC;
524
525 /* Node runtime is stored inside of process. */
526 rt = &p->node_runtime;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700527 }
528 else
529 {
530 vec_add2_aligned (nm->nodes_by_type[n->type], rt, 1,
531 /* align */ CLIB_CACHE_LINE_BYTES);
Damjan Marion94100532020-11-06 23:25:57 +0100532 if (n->type == VLIB_NODE_TYPE_INPUT)
533 clib_interrupt_resize (&nm->interrupts,
534 vec_len (nm->nodes_by_type[n->type]));
Ed Warnickecb9cada2015-12-08 15:45:58 -0700535 n->runtime_index = rt - nm->nodes_by_type[n->type];
536 }
537
538 if (n->type == VLIB_NODE_TYPE_INPUT)
539 nm->input_node_counts_by_state[n->state] += 1;
540
541 rt->function = n->function;
542 rt->flags = n->flags;
543 rt->state = n->state;
544 rt->node_index = n->index;
545
546 rt->n_next_nodes = r->n_next_nodes;
547 rt->next_frame_index = vec_len (nm->next_frames);
548
549 vec_resize (nm->next_frames, rt->n_next_nodes);
550 for (i = 0; i < rt->n_next_nodes; i++)
551 vlib_next_frame_init (nm->next_frames + rt->next_frame_index + i);
552
553 vec_resize (rt->errors, r->n_errors);
554 for (i = 0; i < vec_len (rt->errors); i++)
Dave Barach687c9022019-07-23 10:22:31 -0400555 rt->errors[i] = n->error_heap_index + i;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700556
Christophe Fontaine33e81952016-12-19 14:41:52 +0100557 STATIC_ASSERT_SIZEOF (vlib_node_runtime_t, 128);
Damjan Marione9f929b2017-03-16 11:32:09 +0100558 ASSERT (vec_len (n->runtime_data) <= VLIB_NODE_RUNTIME_DATA_SIZE);
Damjan Marioneb90b7f2016-11-01 01:26:01 +0100559
Ed Warnickecb9cada2015-12-08 15:45:58 -0700560 if (vec_len (n->runtime_data) > 0)
Dave Barach9b8ffd92016-07-08 08:13:45 -0400561 clib_memcpy (rt->runtime_data, n->runtime_data,
562 vec_len (n->runtime_data));
Mohammed Hawarie848c8f2020-12-21 18:19:46 +0100563 else
564 clib_memset (rt->runtime_data, 0, VLIB_NODE_RUNTIME_DATA_SIZE);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700565
566 vec_free (n->runtime_data);
567 }
Damjan Mariona31698b2021-03-10 14:35:28 +0100568#undef _
Ed Warnickecb9cada2015-12-08 15:45:58 -0700569 return r->index;
570}
571
Damjan Mariond4798a32016-09-06 22:26:03 +0200572static uword
573null_node_fn (vlib_main_t * vm,
574 vlib_node_runtime_t * node, vlib_frame_t * frame)
575{
576 u16 n_vectors = frame->n_vectors;
577
578 vlib_node_increment_counter (vm, node->node_index, 0, n_vectors);
Damjan Mariona3d59862018-11-10 10:23:00 +0100579 vlib_buffer_free (vm, vlib_frame_vector_args (frame), n_vectors);
Damjan Mariond4798a32016-09-06 22:26:03 +0200580 vlib_frame_free (vm, node, frame);
581
582 return n_vectors;
583}
584
Dave Barach9b8ffd92016-07-08 08:13:45 -0400585void
Damjan Mariona31698b2021-03-10 14:35:28 +0100586vlib_register_all_node_march_variants (vlib_main_t *vm)
587{
588 vlib_node_main_t *nm = &vm->node_main;
589 vlib_node_fn_variant_t *v;
590 int prio = -1;
591
592 nm->node_fn_default_march_variant = ~0;
593 ASSERT (nm->variants == 0);
594 vec_add2 (nm->variants, v, 1);
595 v->desc = v->suffix = "default";
596 v->index = CLIB_MARCH_VARIANT_TYPE;
597
598#define _(s, n) \
599 vec_add2 (nm->variants, v, 1); \
600 v->suffix = #s; \
601 v->index = CLIB_MARCH_VARIANT_TYPE_##s; \
602 v->priority = clib_cpu_march_priority_##s (); \
603 v->desc = n;
604
605 foreach_march_variant;
606#undef _
607
608 nm->node_fn_march_variant_by_suffix = hash_create_string (0, sizeof (u32));
609
610 vec_foreach (v, nm->variants)
611 {
612 ASSERT (v->index == v - nm->variants);
613 hash_set (nm->node_fn_march_variant_by_suffix, v->suffix, v->index);
614 if (v->priority > prio)
615 prio = v->priority;
616 }
617}
618
619void
Dave Barach9b8ffd92016-07-08 08:13:45 -0400620vlib_register_all_static_nodes (vlib_main_t * vm)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700621{
Damjan Marionfd8deb42021-03-06 12:26:28 +0100622 vlib_global_main_t *vgm = vlib_get_global_main ();
Dave Barach9b8ffd92016-07-08 08:13:45 -0400623 vlib_node_registration_t *r;
624
Damjan Mariond4798a32016-09-06 22:26:03 +0200625 static char *null_node_error_strings[] = {
626 "blackholed packets",
627 };
628
629 static vlib_node_registration_t null_node_reg = {
630 .function = null_node_fn,
631 .vector_size = sizeof (u32),
Damjan Mariond4798a32016-09-06 22:26:03 +0200632 .n_errors = 1,
633 .error_strings = null_node_error_strings,
634 };
635
636 /* make sure that node index 0 is not used by
637 real node */
Damjan Marionf87acfa2022-03-23 17:36:56 +0100638 vlib_register_node (vm, &null_node_reg, "null-node");
Damjan Mariond4798a32016-09-06 22:26:03 +0200639
Damjan Marionfd8deb42021-03-06 12:26:28 +0100640 r = vgm->node_registrations;
Dave Barach9b8ffd92016-07-08 08:13:45 -0400641 while (r)
642 {
Damjan Marionf87acfa2022-03-23 17:36:56 +0100643 vlib_register_node (vm, r, "%s", r->name);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400644 r = r->next_registration;
645 }
Ed Warnickecb9cada2015-12-08 15:45:58 -0700646}
647
Dave Barach1ddbc012018-06-13 09:26:05 -0400648void
649vlib_node_get_nodes (vlib_main_t * vm, u32 max_threads, int include_stats,
650 int barrier_sync, vlib_node_t **** node_dupsp,
651 vlib_main_t *** stat_vmsp)
Florin Corase86a8ed2018-01-05 03:20:25 -0800652{
653 vlib_node_main_t *nm = &vm->node_main;
654 vlib_node_t *n;
Dave Barach1ddbc012018-06-13 09:26:05 -0400655 vlib_node_t ***node_dups = *node_dupsp;
Florin Corase86a8ed2018-01-05 03:20:25 -0800656 vlib_node_t **nodes;
Dave Barach1ddbc012018-06-13 09:26:05 -0400657 vlib_main_t **stat_vms = *stat_vmsp;
Florin Corase86a8ed2018-01-05 03:20:25 -0800658 vlib_main_t *stat_vm;
659 uword i, j;
660 u32 threads_to_serialize;
661
Florin Corase86a8ed2018-01-05 03:20:25 -0800662 if (vec_len (stat_vms) == 0)
663 {
Damjan Marion6ffb7c62021-03-26 13:06:13 +0100664 for (i = 0; i < vlib_get_n_threads (); i++)
Florin Corase86a8ed2018-01-05 03:20:25 -0800665 {
Damjan Marion6ffb7c62021-03-26 13:06:13 +0100666 stat_vm = vlib_get_main_by_index (i);
Florin Corase86a8ed2018-01-05 03:20:25 -0800667 if (stat_vm)
668 vec_add1 (stat_vms, stat_vm);
669 }
670 }
671
672 threads_to_serialize = clib_min (max_threads, vec_len (stat_vms));
673
Dave Barach1ddbc012018-06-13 09:26:05 -0400674 vec_validate (node_dups, threads_to_serialize - 1);
675
Florin Corase86a8ed2018-01-05 03:20:25 -0800676 /*
677 * Barrier sync across stats scraping.
678 * Otherwise, the counts will be grossly inaccurate.
679 */
Dave Barach1ddbc012018-06-13 09:26:05 -0400680 if (barrier_sync)
681 vlib_worker_thread_barrier_sync (vm);
Florin Corase86a8ed2018-01-05 03:20:25 -0800682
683 for (j = 0; j < threads_to_serialize; j++)
684 {
685 stat_vm = stat_vms[j];
686 nm = &stat_vm->node_main;
687
688 if (include_stats)
689 {
690 for (i = 0; i < vec_len (nm->nodes); i++)
691 {
692 n = nm->nodes[i];
693 vlib_node_sync_stats (stat_vm, n);
694 }
695 }
696
Dave Barach1ddbc012018-06-13 09:26:05 -0400697 nodes = node_dups[j];
698 vec_validate (nodes, vec_len (nm->nodes) - 1);
699 clib_memcpy (nodes, nm->nodes, vec_len (nm->nodes) * sizeof (nodes[0]));
700 node_dups[j] = nodes;
Florin Corase86a8ed2018-01-05 03:20:25 -0800701 }
Florin Corase86a8ed2018-01-05 03:20:25 -0800702
Dave Barach1ddbc012018-06-13 09:26:05 -0400703 if (barrier_sync)
704 vlib_worker_thread_barrier_release (vm);
705
706 *node_dupsp = node_dups;
707 *stat_vmsp = stat_vms;
Florin Corase86a8ed2018-01-05 03:20:25 -0800708}
709
Ed Warnickecb9cada2015-12-08 15:45:58 -0700710clib_error_t *
711vlib_node_main_init (vlib_main_t * vm)
712{
Dave Barach9b8ffd92016-07-08 08:13:45 -0400713 vlib_node_main_t *nm = &vm->node_main;
714 clib_error_t *error = 0;
715 vlib_node_t *n;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700716 uword ni;
717
718 nm->flags |= VLIB_NODE_MAIN_RUNTIME_STARTED;
719
Ole Troan964f93e2016-06-10 13:22:36 +0200720 /* Generate sibling relationships */
721 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400722 vlib_node_t *n, *sib;
Ole Troan964f93e2016-06-10 13:22:36 +0200723 uword si;
724
725 for (ni = 0; ni < vec_len (nm->nodes); ni++)
726 {
727 n = vec_elt (nm->nodes, ni);
728
Dave Barach9b8ffd92016-07-08 08:13:45 -0400729 if (!n->sibling_of)
Ole Troan964f93e2016-06-10 13:22:36 +0200730 continue;
731
732 sib = vlib_get_node_by_name (vm, (u8 *) n->sibling_of);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400733 if (!sib)
Ed Warnicke853e7202016-08-12 11:42:26 -0700734 {
735 error = clib_error_create ("sibling `%s' not found for node `%v'",
736 n->sibling_of, n->name);
737 goto done;
738 }
Ole Troan964f93e2016-06-10 13:22:36 +0200739
Dave Barach9b8ffd92016-07-08 08:13:45 -0400740 /* *INDENT-OFF* */
Damjan Marionf0ca1e82020-12-13 23:26:56 +0100741 clib_bitmap_foreach (si, sib->sibling_bitmap) {
Ole Troan964f93e2016-06-10 13:22:36 +0200742 vlib_node_t * m = vec_elt (nm->nodes, si);
743
744 /* Connect all of sibling's siblings to us. */
745 m->sibling_bitmap = clib_bitmap_ori (m->sibling_bitmap, n->index);
746
747 /* Connect us to all of sibling's siblings. */
748 n->sibling_bitmap = clib_bitmap_ori (n->sibling_bitmap, si);
Damjan Marionf0ca1e82020-12-13 23:26:56 +0100749 }
Dave Barach9b8ffd92016-07-08 08:13:45 -0400750 /* *INDENT-ON* */
Ole Troan964f93e2016-06-10 13:22:36 +0200751
752 /* Connect sibling to us. */
753 sib->sibling_bitmap = clib_bitmap_ori (sib->sibling_bitmap, n->index);
754
755 /* Connect us to sibling. */
756 n->sibling_bitmap = clib_bitmap_ori (n->sibling_bitmap, sib->index);
757 }
758 }
759
Ed Warnickecb9cada2015-12-08 15:45:58 -0700760 /* Resolve next names into next indices. */
761 for (ni = 0; ni < vec_len (nm->nodes); ni++)
762 {
763 uword i;
764
765 n = vec_elt (nm->nodes, ni);
766
767 for (i = 0; i < vec_len (n->next_node_names); i++)
768 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400769 char *a = n->next_node_names[i];
Ed Warnickecb9cada2015-12-08 15:45:58 -0700770
Dave Barach9b8ffd92016-07-08 08:13:45 -0400771 if (!a)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700772 continue;
773
774 if (~0 == vlib_node_add_named_next_with_slot (vm, n->index, a, i))
775 {
776 error = clib_error_create
777 ("node `%v' refers to unknown node `%s'", n->name, a);
778 goto done;
779 }
780 }
781
782 vec_free (n->next_node_names);
783 }
784
785 /* Set previous node pointers. */
786 for (ni = 0; ni < vec_len (nm->nodes); ni++)
787 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400788 vlib_node_t *n_next;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700789 uword i;
790
791 n = vec_elt (nm->nodes, ni);
792
793 for (i = 0; i < vec_len (n->next_nodes); i++)
794 {
795 if (n->next_nodes[i] >= vec_len (nm->nodes))
796 continue;
797
798 n_next = vec_elt (nm->nodes, n->next_nodes[i]);
799 n_next->prev_node_bitmap =
800 clib_bitmap_ori (n_next->prev_node_bitmap, n->index);
801 }
802 }
803
804 {
Dave Barach9b8ffd92016-07-08 08:13:45 -0400805 vlib_next_frame_t *nf;
806 vlib_node_runtime_t *r;
807 vlib_node_t *next;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700808 uword i;
809
810 vec_foreach (r, nm->nodes_by_type[VLIB_NODE_TYPE_INTERNAL])
Dave Barach9b8ffd92016-07-08 08:13:45 -0400811 {
812 if (r->n_next_nodes == 0)
813 continue;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700814
Dave Barach9b8ffd92016-07-08 08:13:45 -0400815 n = vlib_get_node (vm, r->node_index);
816 nf = vec_elt_at_index (nm->next_frames, r->next_frame_index);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700817
Dave Barach9b8ffd92016-07-08 08:13:45 -0400818 for (i = 0; i < vec_len (n->next_nodes); i++)
819 {
820 next = vlib_get_node (vm, n->next_nodes[i]);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700821
Dave Barach9b8ffd92016-07-08 08:13:45 -0400822 /* Validate node runtime indices are correctly initialized. */
823 ASSERT (nf[i].node_runtime_index == next->runtime_index);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700824
Dave Barach9b8ffd92016-07-08 08:13:45 -0400825 nf[i].flags = 0;
826 if (next->flags & VLIB_NODE_FLAG_FRAME_NO_FREE_AFTER_DISPATCH)
827 nf[i].flags |= VLIB_FRAME_NO_FREE_AFTER_DISPATCH;
828 }
829 }
Ed Warnickecb9cada2015-12-08 15:45:58 -0700830 }
831
Dave Barach9b8ffd92016-07-08 08:13:45 -0400832done:
Ed Warnickecb9cada2015-12-08 15:45:58 -0700833 return error;
834}
Dave Barach9b8ffd92016-07-08 08:13:45 -0400835
Dave Barach11965c72019-05-28 16:31:05 -0400836u32
837vlib_process_create (vlib_main_t * vm, char *name,
838 vlib_node_function_t * f, u32 log2_n_stack_bytes)
839{
840 vlib_node_registration_t r;
841 vlib_node_t *n;
842
843 memset (&r, 0, sizeof (r));
844
Dave Barach11965c72019-05-28 16:31:05 -0400845 r.function = f;
846 r.process_log2_n_stack_bytes = log2_n_stack_bytes;
847 r.type = VLIB_NODE_TYPE_PROCESS;
848
849 vlib_worker_thread_barrier_sync (vm);
850
Damjan Marionf87acfa2022-03-23 17:36:56 +0100851 vlib_register_node (vm, &r, "%s", name);
Dave Barach11965c72019-05-28 16:31:05 -0400852 vec_free (r.name);
853
854 vlib_worker_thread_node_runtime_update ();
855 vlib_worker_thread_barrier_release (vm);
856
857 n = vlib_get_node (vm, r.index);
858 vlib_start_process (vm, n->runtime_index);
859
860 return (r.index);
861}
862
Damjan Mariona31698b2021-03-10 14:35:28 +0100863int
864vlib_node_set_march_variant (vlib_main_t *vm, u32 node_index,
865 clib_march_variant_type_t march_variant)
866{
867 vlib_node_fn_registration_t *fnr;
868 vlib_node_fn_variant_t *v;
869 vlib_node_t *n = vlib_get_node (vm, node_index);
870
871 if (n->node_fn_registrations == 0)
872 return -1;
873
874 fnr = n->node_fn_registrations;
875 v = vec_elt_at_index (vm->node_main.variants, march_variant);
876
877 while (fnr)
878 {
879 if (fnr->march_variant == v->index)
880 {
881 n->function = fnr->function;
882
Damjan Marion6ffb7c62021-03-26 13:06:13 +0100883 for (int i = 0; i < vlib_get_n_threads (); i++)
Damjan Mariona31698b2021-03-10 14:35:28 +0100884 {
885 vlib_node_runtime_t *nrt;
Damjan Marion6ffb7c62021-03-26 13:06:13 +0100886 nrt =
887 vlib_node_get_runtime (vlib_get_main_by_index (i), n->index);
Damjan Mariona31698b2021-03-10 14:35:28 +0100888 nrt->function = fnr->function;
889 }
890 return 0;
891 }
892 fnr = fnr->next_registration;
893 }
894 return -1;
895}
Dave Barach9b8ffd92016-07-08 08:13:45 -0400896/*
897 * fd.io coding-style-patch-verification: ON
898 *
899 * Local Variables:
900 * eval: (c-set-style "gnu")
901 * End:
902 */