src/vlib/node.c

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