Line data Source code
1 : /*
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. */
44 : vlib_node_t *
45 3090000 : vlib_get_node_by_name (vlib_main_t * vm, u8 * name)
46 : {
47 3090000 : vlib_node_main_t *nm = &vm->node_main;
48 : uword *p;
49 3090000 : u8 *key = name;
50 3090000 : key = format (0, "%s", key);
51 3090000 : p = hash_get (nm->node_by_name, key);
52 3090000 : if (key != name)
53 3090000 : vec_free (key);
54 3090000 : return p ? vec_elt (nm->nodes, p[0]) : 0;
55 : }
56 :
57 : static void
58 417396 : node_set_elog_name (vlib_main_t * vm, uword node_index)
59 : {
60 417396 : vlib_node_t *n = vlib_get_node (vm, node_index);
61 : elog_event_type_t *t;
62 :
63 417396 : t = vec_elt_at_index (vm->node_call_elog_event_types, node_index);
64 417396 : vec_free (t->format);
65 417396 : t->format = (char *) format (0, "%v-call: %%d%c", n->name, 0);
66 :
67 417396 : t = vec_elt_at_index (vm->node_return_elog_event_types, node_index);
68 417396 : vec_free (t->format);
69 417396 : t->format = (char *) format (0, "%v-return: %%d%c", n->name, 0);
70 :
71 417396 : n->name_elog_string =
72 417396 : elog_string (&vlib_global_main.elog_main, "%v%c", n->name, 0);
73 417396 : }
74 :
75 : void
76 2910 : vlib_node_rename (vlib_main_t * vm, u32 node_index, char *fmt, ...)
77 : {
78 : va_list va;
79 2910 : vlib_node_main_t *nm = &vm->node_main;
80 2910 : vlib_node_t *n = vlib_get_node (vm, node_index);
81 :
82 2910 : va_start (va, fmt);
83 2910 : hash_unset (nm->node_by_name, n->name);
84 2910 : vec_free (n->name);
85 2910 : n->name = va_format (0, fmt, &va);
86 2910 : va_end (va);
87 2910 : hash_set (nm->node_by_name, n->name, n->index);
88 :
89 2910 : node_set_elog_name (vm, node_index);
90 :
91 : /* Propagate the change to all worker threads */
92 2910 : vlib_worker_thread_node_runtime_update ();
93 2910 : }
94 :
95 : static void
96 1346350 : vlib_node_runtime_update (vlib_main_t * vm, u32 node_index, u32 next_index)
97 : {
98 1346350 : 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;
103 : i32 i, j, n_insert;
104 :
105 1346350 : node = vec_elt (nm->nodes, node_index);
106 1346350 : r = vlib_node_get_runtime (vm, node_index);
107 :
108 1346350 : n_insert = vec_len (node->next_nodes) - r->n_next_nodes;
109 1346350 : if (n_insert > 0)
110 : {
111 680465 : i = r->next_frame_index + r->n_next_nodes;
112 680465 : vec_insert (nm->next_frames, n_insert, i);
113 :
114 : /* Initialize newly inserted next frames. */
115 1360930 : for (j = 0; j < n_insert; j++)
116 680465 : vlib_next_frame_init (nm->next_frames + i + j);
117 :
118 : /* Relocate other next frames at higher indices. */
119 486768000 : for (j = 0; j < vec_len (nm->nodes); j++)
120 : {
121 486088000 : s = vlib_node_get_runtime (vm, j);
122 486088000 : if (j != node_index && s->next_frame_index >= i)
123 168045000 : s->next_frame_index += n_insert;
124 : }
125 :
126 : /* Pending frames may need to be relocated also. */
127 716781 : vec_foreach (pf, nm->pending_frames)
128 : {
129 36316 : if (pf->next_frame_index != VLIB_PENDING_FRAME_NO_NEXT_FRAME
130 35114 : && pf->next_frame_index >= i)
131 15702 : pf->next_frame_index += n_insert;
132 : }
133 : /* *INDENT-OFF* */
134 3187110 : pool_foreach (pf, nm->suspended_process_frames) {
135 2506640 : if (pf->next_frame_index != ~0 && pf->next_frame_index >= i)
136 0 : pf->next_frame_index += n_insert;
137 : }
138 : /* *INDENT-ON* */
139 :
140 680465 : r->n_next_nodes = vec_len (node->next_nodes);
141 : }
142 :
143 : /* Set frame's node runtime index. */
144 1346350 : next_node = vlib_get_node (vm, node->next_nodes[next_index]);
145 1346350 : nf = nm->next_frames + r->next_frame_index + next_index;
146 1346350 : nf->node_runtime_index = next_node->runtime_index;
147 :
148 1346350 : vlib_worker_thread_node_runtime_update ();
149 1346350 : }
150 :
151 : uword
152 1057510 : vlib_node_get_next (vlib_main_t * vm, uword node_index, uword next_node_index)
153 : {
154 1057510 : vlib_node_main_t *nm = &vm->node_main;
155 : vlib_node_t *node;
156 : uword *p;
157 :
158 1057510 : node = vec_elt (nm->nodes, node_index);
159 :
160 : /* Runtime has to be initialized. */
161 1057510 : ASSERT (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED);
162 :
163 1057510 : if ((p = hash_get (node->next_slot_by_node, next_node_index)))
164 : {
165 1056910 : return p[0];
166 : }
167 :
168 599 : return (~0);
169 : }
170 :
171 : /* Add next node to given node in given slot. */
172 : uword
173 5264030 : vlib_node_add_next_with_slot (vlib_main_t * vm,
174 : uword node_index,
175 : uword next_node_index, uword slot)
176 : {
177 5264030 : vlib_node_main_t *nm = &vm->node_main;
178 : vlib_node_t *node, *next, *old_next;
179 : u32 old_next_index;
180 : uword *p;
181 :
182 5264030 : ASSERT (vlib_get_thread_index () == 0);
183 :
184 5264030 : node = vec_elt (nm->nodes, node_index);
185 5264030 : next = vec_elt (nm->nodes, next_node_index);
186 :
187 : /* Runtime has to be initialized. */
188 5264030 : ASSERT (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED);
189 :
190 5264030 : if ((p = hash_get (node->next_slot_by_node, next_node_index)))
191 : {
192 : /* Next already exists: slot must match. */
193 3917680 : if (slot != ~0)
194 3298380 : ASSERT (slot == p[0]);
195 3917680 : return p[0];
196 : }
197 :
198 1346350 : vlib_worker_thread_barrier_sync (vm);
199 :
200 1346350 : if (slot == ~0)
201 228807 : slot = vec_len (node->next_nodes);
202 :
203 2026810 : vec_validate_init_empty (node->next_nodes, slot, ~0);
204 1346350 : vec_validate (node->n_vectors_by_next_node, slot);
205 :
206 1346350 : if ((old_next_index = node->next_nodes[slot]) != ~0u)
207 : {
208 4 : hash_unset (node->next_slot_by_node, old_next_index);
209 4 : old_next = vlib_get_node (vm, old_next_index);
210 4 : old_next->prev_node_bitmap =
211 4 : clib_bitmap_andnoti (old_next->prev_node_bitmap, node_index);
212 : }
213 :
214 1346350 : node->next_nodes[slot] = next_node_index;
215 1346350 : hash_set (node->next_slot_by_node, next_node_index, slot);
216 :
217 1346350 : vlib_node_runtime_update (vm, node_index, slot);
218 :
219 1346350 : 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;
225 : vlib_node_t *sib_node;
226 : /* *INDENT-OFF* */
227 5088190 : clib_bitmap_foreach (sib_node_index, node->sibling_bitmap) {
228 3741840 : sib_node = vec_elt (nm->nodes, sib_node_index);
229 3741840 : if (sib_node != node)
230 : {
231 3741840 : sib_slot = vlib_node_add_next_with_slot (vm, sib_node_index, next_node_index, slot);
232 3741840 : ASSERT (sib_slot == slot);
233 : }
234 : }
235 : /* *INDENT-ON* */
236 : }
237 :
238 1346350 : vlib_worker_thread_barrier_release (vm);
239 1346350 : return slot;
240 : }
241 :
242 : /* Add named next node to given node in given slot. */
243 : uword
244 1083540 : vlib_node_add_named_next_with_slot (vlib_main_t * vm,
245 : uword node, char *name, uword slot)
246 : {
247 : vlib_node_main_t *nm;
248 : vlib_node_t *n, *n_next;
249 :
250 1083540 : nm = &vm->node_main;
251 1083540 : n = vlib_get_node (vm, node);
252 :
253 1083540 : n_next = vlib_get_node_by_name (vm, (u8 *) name);
254 1083540 : if (!n_next)
255 : {
256 0 : if (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED)
257 0 : return ~0;
258 :
259 0 : if (slot == ~0)
260 0 : slot = clib_max (vec_len (n->next_node_names),
261 : vec_len (n->next_nodes));
262 0 : vec_validate (n->next_node_names, slot);
263 0 : n->next_node_names[slot] = name;
264 0 : return slot;
265 : }
266 :
267 1083540 : return vlib_node_add_next_with_slot (vm, node, n_next->index, slot);
268 : }
269 :
270 : static void
271 414486 : node_elog_init (vlib_main_t * vm, uword ni)
272 : {
273 : elog_event_type_t t;
274 :
275 414486 : clib_memset (&t, 0, sizeof (t));
276 :
277 : /* 2 event types for this node: one when node function is called.
278 : One when it returns. */
279 414486 : vec_validate (vm->node_call_elog_event_types, ni);
280 414486 : vm->node_call_elog_event_types[ni] = t;
281 :
282 414486 : vec_validate (vm->node_return_elog_event_types, ni);
283 414486 : vm->node_return_elog_event_types[ni] = t;
284 :
285 414486 : node_set_elog_name (vm, ni);
286 414486 : }
287 :
288 : #ifdef CLIB_UNIX
289 : #define STACK_ALIGN (clib_mem_get_page_size())
290 : #else
291 : #define STACK_ALIGN CLIB_CACHE_LINE_BYTES
292 : #endif
293 :
294 : vlib_node_function_t *
295 272585 : vlib_node_get_preferred_node_fn_variant (vlib_main_t *vm,
296 : vlib_node_fn_registration_t *regs)
297 : {
298 272585 : vlib_node_main_t *nm = &vm->node_main;
299 : vlib_node_fn_registration_t *r;
300 : vlib_node_fn_variant_t *v;
301 272585 : vlib_node_function_t *fn = 0;
302 272585 : int priority = -1;
303 :
304 272585 : if (nm->node_fn_default_march_variant != ~0)
305 : {
306 0 : r = regs;
307 0 : while (r)
308 : {
309 0 : if (r->march_variant == nm->node_fn_default_march_variant)
310 0 : return r->function;
311 0 : r = r->next_registration;
312 : }
313 : }
314 :
315 272585 : r = regs;
316 1226020 : while (r)
317 : {
318 953436 : v = vec_elt_at_index (nm->variants, r->march_variant);
319 953436 : if (v->priority > priority)
320 : {
321 341865 : priority = v->priority;
322 341865 : fn = r->function;
323 : }
324 953436 : r = r->next_registration;
325 : }
326 :
327 272585 : ASSERT (fn);
328 272585 : return fn;
329 : }
330 :
331 : static void
332 52330 : vlib_node_add_to_sibling_bitmap (vlib_main_t *vm, vlib_node_t *n,
333 : vlib_node_t *sib)
334 : {
335 52330 : vlib_node_main_t *nm = &vm->node_main;
336 : u32 si;
337 :
338 181150 : clib_bitmap_foreach (si, sib->sibling_bitmap)
339 : {
340 128820 : vlib_node_t *m = vec_elt (nm->nodes, si);
341 :
342 : /* Connect all of sibling's siblings to us. */
343 128820 : m->sibling_bitmap = clib_bitmap_ori (m->sibling_bitmap, n->index);
344 :
345 : /* Connect us to all of sibling's siblings. */
346 128820 : n->sibling_bitmap = clib_bitmap_ori (n->sibling_bitmap, si);
347 : }
348 :
349 : /* Connect sibling to us. */
350 52330 : sib->sibling_bitmap = clib_bitmap_ori (sib->sibling_bitmap, n->index);
351 :
352 : /* Connect us to sibling. */
353 52330 : n->sibling_bitmap = clib_bitmap_ori (n->sibling_bitmap, sib->index);
354 52330 : }
355 :
356 : u32
357 414486 : vlib_register_node (vlib_main_t *vm, vlib_node_registration_t *r, char *fmt,
358 : ...)
359 : {
360 414486 : vlib_node_main_t *nm = &vm->node_main;
361 414486 : vlib_node_t *n, *sib = 0;
362 : va_list va;
363 : u32 size;
364 : int i;
365 :
366 414486 : if (r->sibling_of)
367 : {
368 52330 : if (r->n_next_nodes > 0)
369 0 : clib_error ("sibling node should not have any next nodes `%v'",
370 : r->name);
371 52330 : if (nm->flags & VLIB_NODE_MAIN_RUNTIME_STARTED)
372 : {
373 0 : sib = vlib_get_node_by_name (vm, (u8 *) r->sibling_of);
374 :
375 0 : if (sib == 0)
376 0 : clib_error ("unknown sibling node '%s'", r->sibling_of);
377 : }
378 : }
379 :
380 : if (CLIB_DEBUG > 0)
381 : {
382 : /* Default (0) type should match INTERNAL. */
383 414486 : vlib_node_t zero = { 0 };
384 414486 : ASSERT (VLIB_NODE_TYPE_INTERNAL == zero.type);
385 : }
386 :
387 414486 : if (r->node_fn_registrations)
388 : {
389 : /* to avoid confusion, please remove ".function " statiement from
390 : CLIB_NODE_REGISTRATION() if using function function candidates */
391 266769 : ASSERT (r->function == 0);
392 :
393 266769 : r->function =
394 266769 : vlib_node_get_preferred_node_fn_variant (vm, r->node_fn_registrations);
395 : }
396 :
397 414486 : ASSERT (r->function != 0);
398 :
399 414486 : n = clib_mem_alloc_no_fail (sizeof (n[0]));
400 414486 : clib_memset (n, 0, sizeof (n[0]));
401 414486 : n->index = vec_len (nm->nodes);
402 414486 : n->node_fn_registrations = r->node_fn_registrations;
403 414486 : n->protocol_hint = r->protocol_hint;
404 :
405 414486 : vec_add1 (nm->nodes, n);
406 :
407 414486 : va_start (va, fmt);
408 414486 : n->name = va_format (0, fmt, &va);
409 414486 : va_end (va);
410 :
411 414486 : if (!nm->node_by_name)
412 575 : nm->node_by_name = hash_create_vec ( /* size */ 32,
413 : sizeof (n->name[0]), sizeof (uword));
414 :
415 : /* Node names must be unique. */
416 : {
417 : /* vlib_get_node_by_name() expects NULL-terminated strings */
418 414486 : u8 *name = format (0, "%v%c", n->name, 0);
419 414486 : vlib_node_t *o = vlib_get_node_by_name (vm, name);
420 414486 : vec_free (name);
421 414486 : if (o)
422 0 : clib_error ("more than one node named `%v'", n->name);
423 : }
424 :
425 414486 : hash_set (nm->node_by_name, n->name, n->index);
426 :
427 414486 : r->index = n->index; /* save index in registration */
428 414486 : n->function = r->function;
429 :
430 414486 : if (r->type == VLIB_NODE_TYPE_INTERNAL)
431 369049 : ASSERT (r->vector_size > 0);
432 :
433 : #define _(f) n->f = r->f
434 :
435 414486 : _(type);
436 414486 : _(flags);
437 414486 : _(state);
438 414486 : _(format_buffer);
439 414486 : _(unformat_buffer);
440 414486 : _(format_trace);
441 414486 : _(validate_frame);
442 :
443 414486 : size = round_pow2 (sizeof (vlib_frame_t), VLIB_FRAME_DATA_ALIGN);
444 :
445 : /* scalar data size */
446 414486 : if (r->scalar_size)
447 : {
448 5029 : n->scalar_offset = size;
449 5029 : size += round_pow2 (r->scalar_size, VLIB_FRAME_DATA_ALIGN);
450 : }
451 : else
452 409457 : n->scalar_offset = 0;
453 :
454 : /* Vecor data size */
455 414486 : n->vector_offset = size;
456 414486 : size += r->vector_size * VLIB_FRAME_SIZE;
457 :
458 : /* Allocate a few extra slots of vector data to support
459 : speculative vector enqueues which overflow vector data in next frame. */
460 414486 : size += r->vector_size * VLIB_FRAME_SIZE_EXTRA;
461 :
462 : /* space for VLIB_FRAME_MAGIC */
463 414486 : n->magic_offset = size;
464 414486 : size += sizeof (u32);
465 :
466 : /* round size to VLIB_FRAME_DATA_ALIGN */
467 414486 : size = round_pow2 (size, VLIB_FRAME_DATA_ALIGN);
468 :
469 414486 : if (r->aux_size)
470 : {
471 2300 : n->aux_offset = size;
472 2300 : size += r->aux_size * VLIB_FRAME_SIZE;
473 : }
474 : else
475 412186 : n->aux_offset = 0;
476 :
477 : /* final size */
478 414486 : n->frame_size = size = round_pow2 (size, CLIB_CACHE_LINE_BYTES);
479 414486 : ASSERT (size <= __UINT16_MAX__);
480 :
481 414486 : vlib_frame_size_t *fs = 0;
482 :
483 414486 : n->frame_size_index = (u16) ~0;
484 465673 : vec_foreach (fs, nm->frame_sizes)
485 463373 : if (fs->frame_size == size)
486 : {
487 412186 : n->frame_size_index = fs - nm->frame_sizes;
488 412186 : break;
489 : }
490 :
491 414486 : if (n->frame_size_index == (u16) ~0)
492 : {
493 2300 : vec_add2 (nm->frame_sizes, fs, 1);
494 2300 : fs->frame_size = size;
495 2300 : n->frame_size_index = fs - nm->frame_sizes;
496 : }
497 :
498 : /* Register error counters. */
499 414486 : vlib_register_errors (vm, n->index, r->n_errors, r->error_strings,
500 : r->error_counters);
501 414486 : node_elog_init (vm, n->index);
502 :
503 414486 : _(runtime_data_bytes);
504 414486 : if (r->runtime_data_bytes > 0)
505 : {
506 18254 : vec_resize (n->runtime_data, r->runtime_data_bytes);
507 18254 : if (r->runtime_data)
508 4454 : clib_memcpy (n->runtime_data, r->runtime_data, r->runtime_data_bytes);
509 : }
510 :
511 414486 : vec_resize (n->next_node_names, r->n_next_nodes);
512 1080360 : for (i = 0; i < r->n_next_nodes; i++)
513 665877 : n->next_node_names[i] = r->next_nodes[i];
514 :
515 1080360 : vec_validate_init_empty (n->next_nodes, r->n_next_nodes - 1, ~0);
516 414486 : vec_validate (n->n_vectors_by_next_node, r->n_next_nodes - 1);
517 :
518 414486 : n->owner_node_index = n->owner_next_index = ~0;
519 :
520 : /* Initialize node runtime. */
521 : {
522 : vlib_node_runtime_t *rt;
523 : u32 i;
524 :
525 414486 : if (n->type == VLIB_NODE_TYPE_PROCESS)
526 : {
527 : vlib_process_t *p;
528 : uword log2_n_stack_bytes;
529 :
530 27612 : log2_n_stack_bytes = clib_max (r->process_log2_n_stack_bytes,
531 : VLIB_PROCESS_LOG2_STACK_SIZE);
532 27612 : log2_n_stack_bytes = clib_max (log2_n_stack_bytes,
533 : clib_mem_get_log2_page_size ());
534 :
535 27612 : p = clib_mem_alloc_aligned (sizeof (p[0]), CLIB_CACHE_LINE_BYTES);
536 27612 : clib_memset (p, 0, sizeof (p[0]));
537 27612 : p->log2_n_stack_bytes = log2_n_stack_bytes;
538 :
539 27612 : p->stack = clib_mem_vm_map_stack (1ULL << log2_n_stack_bytes,
540 : CLIB_MEM_PAGE_SZ_DEFAULT,
541 : "process stack: %U",
542 : format_vlib_node_name, vm,
543 : n->index);
544 :
545 27612 : if (p->stack == CLIB_MEM_VM_MAP_FAILED)
546 0 : clib_panic ("failed to allocate process stack (%d bytes)",
547 : 1ULL << log2_n_stack_bytes);
548 :
549 : /* Process node's runtime index is really index into process
550 : pointer vector. */
551 27612 : n->runtime_index = vec_len (nm->processes);
552 :
553 27612 : vec_add1 (nm->processes, p);
554 :
555 : /* Paint first stack word with magic number so we can at least
556 : detect process stack overruns. */
557 27612 : p->stack[0] = VLIB_PROCESS_STACK_MAGIC;
558 :
559 : /* Node runtime is stored inside of process. */
560 27612 : rt = &p->node_runtime;
561 : }
562 : else
563 : {
564 386874 : vec_add2_aligned (nm->nodes_by_type[n->type], rt, 1,
565 : /* align */ CLIB_CACHE_LINE_BYTES);
566 386874 : if (n->type == VLIB_NODE_TYPE_INPUT)
567 16100 : clib_interrupt_resize (&nm->input_node_interrupts,
568 16100 : vec_len (nm->nodes_by_type[n->type]));
569 370774 : else if (n->type == VLIB_NODE_TYPE_PRE_INPUT)
570 1725 : clib_interrupt_resize (&nm->pre_input_node_interrupts,
571 1725 : vec_len (nm->nodes_by_type[n->type]));
572 386874 : n->runtime_index = rt - nm->nodes_by_type[n->type];
573 : }
574 :
575 414486 : if (n->type == VLIB_NODE_TYPE_INPUT)
576 16100 : nm->input_node_counts_by_state[n->state] += 1;
577 :
578 414486 : rt->function = n->function;
579 414486 : rt->flags = n->flags;
580 414486 : rt->state = n->state;
581 414486 : rt->node_index = n->index;
582 :
583 414486 : rt->n_next_nodes = r->n_next_nodes;
584 414486 : rt->next_frame_index = vec_len (nm->next_frames);
585 :
586 414486 : vec_resize (nm->next_frames, rt->n_next_nodes);
587 1080360 : for (i = 0; i < rt->n_next_nodes; i++)
588 665877 : vlib_next_frame_init (nm->next_frames + rt->next_frame_index + i);
589 :
590 414486 : vec_resize (rt->errors, r->n_errors);
591 2689230 : for (i = 0; i < vec_len (rt->errors); i++)
592 2274740 : rt->errors[i] = n->error_heap_index + i;
593 :
594 : STATIC_ASSERT_SIZEOF (vlib_node_runtime_t, 128);
595 414486 : ASSERT (vec_len (n->runtime_data) <= VLIB_NODE_RUNTIME_DATA_SIZE);
596 :
597 414486 : if (vec_len (n->runtime_data) > 0)
598 18254 : clib_memcpy (rt->runtime_data, n->runtime_data,
599 : vec_len (n->runtime_data));
600 : else
601 396232 : clib_memset (rt->runtime_data, 0, VLIB_NODE_RUNTIME_DATA_SIZE);
602 :
603 414486 : vec_free (n->runtime_data);
604 : }
605 : #undef _
606 :
607 414486 : if (sib)
608 : {
609 : u32 slot, i;
610 :
611 0 : vec_foreach_index (i, sib->next_nodes)
612 : {
613 0 : slot =
614 0 : vlib_node_add_next_with_slot (vm, n->index, sib->next_nodes[i], i);
615 0 : ASSERT (slot == i);
616 : }
617 :
618 0 : vlib_node_add_to_sibling_bitmap (vm, n, sib);
619 :
620 0 : r->n_next_nodes = vec_len (n->next_nodes);
621 : }
622 414486 : n->sibling_of = r->sibling_of;
623 :
624 414486 : return r->index;
625 : }
626 :
627 : static uword
628 0 : null_node_fn (vlib_main_t * vm,
629 : vlib_node_runtime_t * node, vlib_frame_t * frame)
630 : {
631 0 : u16 n_vectors = frame->n_vectors;
632 :
633 0 : vlib_node_increment_counter (vm, node->node_index, 0, n_vectors);
634 0 : vlib_buffer_free (vm, vlib_frame_vector_args (frame), n_vectors);
635 0 : vlib_frame_free (vm, frame);
636 :
637 0 : return n_vectors;
638 : }
639 :
640 : void
641 575 : vlib_register_all_node_march_variants (vlib_main_t *vm)
642 : {
643 575 : vlib_node_main_t *nm = &vm->node_main;
644 : vlib_node_fn_variant_t *v;
645 575 : int prio = -1;
646 :
647 575 : nm->node_fn_default_march_variant = ~0;
648 575 : ASSERT (nm->variants == 0);
649 575 : vec_add2 (nm->variants, v, 1);
650 575 : v->desc = v->suffix = "default";
651 575 : v->index = CLIB_MARCH_VARIANT_TYPE;
652 :
653 : #define _(s, n) \
654 : vec_add2 (nm->variants, v, 1); \
655 : v->suffix = #s; \
656 : v->index = CLIB_MARCH_VARIANT_TYPE_##s; \
657 : v->priority = clib_cpu_march_priority_##s (); \
658 : v->desc = n;
659 :
660 575 : foreach_march_variant;
661 : #undef _
662 :
663 575 : nm->node_fn_march_variant_by_suffix = hash_create_string (0, sizeof (u32));
664 :
665 6325 : vec_foreach (v, nm->variants)
666 : {
667 5750 : ASSERT (v->index == v - nm->variants);
668 5750 : hash_set (nm->node_fn_march_variant_by_suffix, v->suffix, v->index);
669 5750 : if (v->priority > prio)
670 2300 : prio = v->priority;
671 : }
672 575 : }
673 :
674 : void
675 575 : vlib_register_all_static_nodes (vlib_main_t * vm)
676 : {
677 575 : vlib_global_main_t *vgm = vlib_get_global_main ();
678 : vlib_node_registration_t *r;
679 :
680 : static char *null_node_error_strings[] = {
681 : "blackholed packets",
682 : };
683 :
684 : static vlib_node_registration_t null_node_reg = {
685 : .function = null_node_fn,
686 : .vector_size = sizeof (u32),
687 : .n_errors = 1,
688 : .error_strings = null_node_error_strings,
689 : };
690 :
691 : /* make sure that node index 0 is not used by
692 : real node */
693 575 : vlib_register_node (vm, &null_node_reg, "null-node");
694 :
695 575 : r = vgm->node_registrations;
696 410020 : while (r)
697 : {
698 409445 : vlib_register_node (vm, r, "%s", r->name);
699 409445 : r = r->next_registration;
700 : }
701 575 : }
702 :
703 : void
704 149 : vlib_node_get_nodes (vlib_main_t * vm, u32 max_threads, int include_stats,
705 : int barrier_sync, vlib_node_t **** node_dupsp,
706 : vlib_main_t *** stat_vmsp)
707 : {
708 149 : vlib_node_main_t *nm = &vm->node_main;
709 : vlib_node_t *n;
710 149 : vlib_node_t ***node_dups = *node_dupsp;
711 : vlib_node_t **nodes;
712 149 : vlib_main_t **stat_vms = *stat_vmsp;
713 : vlib_main_t *stat_vm;
714 : uword i, j;
715 : u32 threads_to_serialize;
716 :
717 149 : if (vec_len (stat_vms) == 0)
718 : {
719 298 : for (i = 0; i < vlib_get_n_threads (); i++)
720 : {
721 149 : stat_vm = vlib_get_main_by_index (i);
722 149 : if (stat_vm)
723 149 : vec_add1 (stat_vms, stat_vm);
724 : }
725 : }
726 :
727 149 : threads_to_serialize = clib_min (max_threads, vec_len (stat_vms));
728 :
729 149 : vec_validate (node_dups, threads_to_serialize - 1);
730 :
731 : /*
732 : * Barrier sync across stats scraping.
733 : * Otherwise, the counts will be grossly inaccurate.
734 : */
735 149 : if (barrier_sync)
736 0 : vlib_worker_thread_barrier_sync (vm);
737 :
738 298 : for (j = 0; j < threads_to_serialize; j++)
739 : {
740 149 : stat_vm = stat_vms[j];
741 149 : nm = &stat_vm->node_main;
742 :
743 149 : if (include_stats)
744 : {
745 109004 : for (i = 0; i < vec_len (nm->nodes); i++)
746 : {
747 108855 : n = nm->nodes[i];
748 108855 : vlib_node_sync_stats (stat_vm, n);
749 : }
750 : }
751 :
752 149 : nodes = node_dups[j];
753 149 : vec_validate (nodes, vec_len (nm->nodes) - 1);
754 149 : clib_memcpy (nodes, nm->nodes, vec_len (nm->nodes) * sizeof (nodes[0]));
755 149 : node_dups[j] = nodes;
756 : }
757 :
758 149 : if (barrier_sync)
759 0 : vlib_worker_thread_barrier_release (vm);
760 :
761 149 : *node_dupsp = node_dups;
762 149 : *stat_vmsp = stat_vms;
763 149 : }
764 :
765 : clib_error_t *
766 575 : vlib_node_main_init (vlib_main_t * vm)
767 : {
768 575 : vlib_node_main_t *nm = &vm->node_main;
769 575 : clib_error_t *error = 0;
770 : vlib_node_t *n;
771 : uword ni;
772 :
773 575 : nm->flags |= VLIB_NODE_MAIN_RUNTIME_STARTED;
774 :
775 : /* Generate sibling relationships */
776 : {
777 : vlib_node_t *n, *sib;
778 :
779 410595 : for (ni = 0; ni < vec_len (nm->nodes); ni++)
780 : {
781 410020 : n = vec_elt (nm->nodes, ni);
782 :
783 410020 : if (!n->sibling_of)
784 357690 : continue;
785 :
786 52330 : sib = vlib_get_node_by_name (vm, (u8 *) n->sibling_of);
787 52330 : if (!sib)
788 : {
789 0 : error = clib_error_create ("sibling `%s' not found for node `%v'",
790 : n->sibling_of, n->name);
791 0 : goto done;
792 : }
793 :
794 52330 : vlib_node_add_to_sibling_bitmap (vm, n, sib);
795 : }
796 : }
797 :
798 : /* Resolve next names into next indices. */
799 410595 : for (ni = 0; ni < vec_len (nm->nodes); ni++)
800 : {
801 : uword i;
802 :
803 410020 : n = vec_elt (nm->nodes, ni);
804 :
805 1075900 : for (i = 0; i < vec_len (n->next_node_names); i++)
806 : {
807 665877 : char *a = n->next_node_names[i];
808 :
809 665877 : if (!a)
810 0 : continue;
811 :
812 665877 : if (~0 == vlib_node_add_named_next_with_slot (vm, n->index, a, i))
813 : {
814 0 : error = clib_error_create
815 : ("node `%v' refers to unknown node `%s'", n->name, a);
816 0 : goto done;
817 : }
818 : }
819 :
820 410020 : vec_free (n->next_node_names);
821 : }
822 :
823 : /* Set previous node pointers. */
824 410595 : for (ni = 0; ni < vec_len (nm->nodes); ni++)
825 : {
826 : vlib_node_t *n_next;
827 : uword i;
828 :
829 410020 : n = vec_elt (nm->nodes, ni);
830 :
831 1418040 : for (i = 0; i < vec_len (n->next_nodes); i++)
832 : {
833 1008020 : if (n->next_nodes[i] >= vec_len (nm->nodes))
834 0 : continue;
835 :
836 1008020 : n_next = vec_elt (nm->nodes, n->next_nodes[i]);
837 1008020 : n_next->prev_node_bitmap =
838 1008020 : clib_bitmap_ori (n_next->prev_node_bitmap, n->index);
839 : }
840 : }
841 :
842 : {
843 : vlib_next_frame_t *nf;
844 : vlib_node_runtime_t *r;
845 : vlib_node_t *next;
846 : uword i;
847 :
848 365170 : vec_foreach (r, nm->nodes_by_type[VLIB_NODE_TYPE_INTERNAL])
849 : {
850 364595 : if (r->n_next_nodes == 0)
851 31075 : continue;
852 :
853 333520 : n = vlib_get_node (vm, r->node_index);
854 333520 : nf = vec_elt_at_index (nm->next_frames, r->next_frame_index);
855 :
856 1282880 : for (i = 0; i < vec_len (n->next_nodes); i++)
857 : {
858 949365 : next = vlib_get_node (vm, n->next_nodes[i]);
859 :
860 : /* Validate node runtime indices are correctly initialized. */
861 949365 : ASSERT (nf[i].node_runtime_index == next->runtime_index);
862 :
863 949365 : nf[i].flags = 0;
864 949365 : if (next->flags & VLIB_NODE_FLAG_FRAME_NO_FREE_AFTER_DISPATCH)
865 575 : nf[i].flags |= VLIB_FRAME_NO_FREE_AFTER_DISPATCH;
866 : }
867 : }
868 : }
869 :
870 575 : done:
871 575 : return error;
872 : }
873 :
874 : u32
875 12 : vlib_process_create (vlib_main_t * vm, char *name,
876 : vlib_node_function_t * f, u32 log2_n_stack_bytes)
877 : {
878 : vlib_node_registration_t r;
879 : vlib_node_t *n;
880 :
881 12 : memset (&r, 0, sizeof (r));
882 :
883 12 : r.function = f;
884 12 : r.process_log2_n_stack_bytes = log2_n_stack_bytes;
885 12 : r.type = VLIB_NODE_TYPE_PROCESS;
886 :
887 12 : vlib_worker_thread_barrier_sync (vm);
888 :
889 12 : vlib_register_node (vm, &r, "%s", name);
890 12 : vec_free (r.name);
891 :
892 12 : vlib_worker_thread_node_runtime_update ();
893 12 : vlib_worker_thread_barrier_release (vm);
894 :
895 12 : n = vlib_get_node (vm, r.index);
896 12 : vlib_start_process (vm, n->runtime_index);
897 :
898 12 : return (r.index);
899 : }
900 :
901 : int
902 2146 : vlib_node_set_march_variant (vlib_main_t *vm, u32 node_index,
903 : clib_march_variant_type_t march_variant)
904 : {
905 : vlib_node_fn_registration_t *fnr;
906 : vlib_node_fn_variant_t *v;
907 2146 : vlib_node_t *n = vlib_get_node (vm, node_index);
908 :
909 2146 : if (n->node_fn_registrations == 0)
910 762 : return -1;
911 :
912 1384 : fnr = n->node_fn_registrations;
913 1384 : v = vec_elt_at_index (vm->node_main.variants, march_variant);
914 :
915 3784 : while (fnr)
916 : {
917 3784 : if (fnr->march_variant == v->index)
918 : {
919 1384 : n->function = fnr->function;
920 :
921 2769 : for (int i = 0; i < vlib_get_n_threads (); i++)
922 : {
923 : vlib_node_runtime_t *nrt;
924 : nrt =
925 1385 : vlib_node_get_runtime (vlib_get_main_by_index (i), n->index);
926 1385 : nrt->function = fnr->function;
927 : }
928 1384 : return 0;
929 : }
930 2400 : fnr = fnr->next_registration;
931 : }
932 0 : return -1;
933 : }
934 : /*
935 : * fd.io coding-style-patch-verification: ON
936 : *
937 : * Local Variables:
938 : * eval: (c-set-style "gnu")
939 : * End:
940 : */
|
