src/plugins/dpdk/buffer.c

   1 /*
   2  * Copyright (c) 2017 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  * buffer.c: allocate/free network buffers.
  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 /**
  41  * @file
  42  *
  43  * Allocate/free network buffers.
  44  */
  45
  46 #include <unistd.h>
  47
  48 #include <rte_config.h>
  49
  50 #include <rte_common.h>
  51 #include <rte_log.h>
  52 #include <rte_memory.h>
  53 #include <rte_memzone.h>
  54 #include <rte_tailq.h>
  55 #include <rte_eal.h>
  56 #include <rte_per_lcore.h>
  57 #include <rte_launch.h>
  58 #include <rte_atomic.h>
  59 #include <rte_cycles.h>
  60 #include <rte_prefetch.h>
  61 #include <rte_lcore.h>
  62 #include <rte_per_lcore.h>
  63 #include <rte_branch_prediction.h>
  64 #include <rte_interrupts.h>
  65 #include <rte_vfio.h>
  66 #include <rte_random.h>
  67 #include <rte_debug.h>
  68 #include <rte_ether.h>
  69 #include <rte_ethdev.h>
  70 #include <rte_ring.h>
  71 #include <rte_mempool.h>
  72 #include <rte_mbuf.h>
  73 #include <rte_version.h>
  74
  75 #include <vlib/vlib.h>
  76 #include <vlib/unix/unix.h>
  77 #include <vnet/vnet.h>
  78 #include <dpdk/device/dpdk.h>
  79 #include <dpdk/device/dpdk_priv.h>
  80
  81 STATIC_ASSERT (VLIB_BUFFER_PRE_DATA_SIZE == RTE_PKTMBUF_HEADROOM,
  82                "VLIB_BUFFER_PRE_DATA_SIZE must be equal to RTE_PKTMBUF_HEADROOM");
  83
  84 typedef struct
  85 {
  86   CLIB_CACHE_LINE_ALIGN_MARK (cacheline0);
  87   struct rte_mbuf **mbuf_alloc_list;
  88 } dpdk_buffer_per_thread_data;
  89
  90 typedef struct
  91 {
  92   int vfio_container_fd;
  93   dpdk_buffer_per_thread_data *ptd;
  94 } dpdk_buffer_main_t;
  95
  96 dpdk_buffer_main_t dpdk_buffer_main;
  97
  98 static_always_inline void
  99 dpdk_rte_pktmbuf_free (vlib_main_t * vm, u32 thread_index, vlib_buffer_t * b,
 100                        int maybe_next)
 101 {
 102   struct rte_mbuf *mb;
 103   u32 next, flags;
 104
 105 next:
 106   flags = b->flags;
 107   next = b->next_buffer;
 108   mb = rte_mbuf_from_vlib_buffer (b);
 109
 110   if (PREDICT_FALSE (b->n_add_refs))
 111     {
 112       rte_mbuf_refcnt_update (mb, b->n_add_refs);
 113       b->n_add_refs = 0;
 114     }
 115
 116   if ((mb = rte_pktmbuf_prefree_seg (mb)))
 117     rte_mempool_put (mb->pool, mb);
 118
 119   if (maybe_next && (flags & VLIB_BUFFER_NEXT_PRESENT))
 120     {
 121       b = vlib_get_buffer (vm, next);
 122       goto next;
 123     }
 124 }
 125
 126 #ifndef CLIB_MARCH_VARIANT
 127 static void
 128 del_free_list (vlib_main_t * vm, vlib_buffer_free_list_t * f)
 129 {
 130   u32 i;
 131   vlib_buffer_t *b;
 132   u32 thread_index = vlib_get_thread_index ();
 133
 134   for (i = 0; i < vec_len (f->buffers); i++)
 135     {
 136       b = vlib_get_buffer (vm, f->buffers[i]);
 137       dpdk_rte_pktmbuf_free (vm, thread_index, b, 1);
 138     }
 139
 140   vec_free (f->name);
 141   vec_free (f->buffers);
 142   /* Poison it. */
 143   clib_memset (f, 0xab, sizeof (f[0]));
 144 }
 145
 146 /* Add buffer free list. */
 147 static void
 148 dpdk_buffer_delete_free_list (vlib_main_t * vm,
 149                               vlib_buffer_free_list_index_t free_list_index)
 150 {
 151   vlib_buffer_free_list_t *f;
 152   int i;
 153
 154   ASSERT (vlib_get_thread_index () == 0);
 155
 156   f = vlib_buffer_get_free_list (vm, free_list_index);
 157
 158   del_free_list (vm, f);
 159
 160   pool_put (vm->buffer_free_list_pool, f);
 161
 162   for (i = 1; i < vec_len (vlib_mains); i++)
 163     {
 164       vlib_main_t *wvm = vlib_mains[i];
 165       f = vlib_buffer_get_free_list (vlib_mains[i], free_list_index);
 166       del_free_list (wvm, f);
 167       pool_put (wvm->buffer_free_list_pool, f);
 168     }
 169 }
 170 #endif
 171
 172 /* Make sure free list has at least given number of free buffers. */
 173 uword
 174 CLIB_MULTIARCH_FN (dpdk_buffer_fill_free_list) (vlib_main_t * vm,
 175                                                 vlib_buffer_free_list_t * fl,
 176                                                 uword min_free_buffers)
 177 {
 178   dpdk_main_t *dm = &dpdk_main;
 179   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
 180   struct rte_mbuf **mb;
 181   uword n_left, first;
 182   word n_alloc;
 183   unsigned socket_id = rte_socket_id ();
 184   u32 thread_index = vlib_get_thread_index ();
 185   dpdk_buffer_per_thread_data *d = vec_elt_at_index (dbm->ptd, thread_index);
 186   struct rte_mempool *rmp = dm->pktmbuf_pools[socket_id];
 187   dpdk_mempool_private_t *privp = rte_mempool_get_priv (rmp);
 188   vlib_buffer_t bt;
 189   u32 *bi;
 190
 191   /* Too early? */
 192   if (PREDICT_FALSE (rmp == 0))
 193     return 0;
 194
 195   /* Already have enough free buffers on free list? */
 196   n_alloc = min_free_buffers - vec_len (fl->buffers);
 197   if (n_alloc <= 0)
 198     return min_free_buffers;
 199
 200   /* Always allocate round number of buffers. */
 201   n_alloc = round_pow2 (n_alloc, CLIB_CACHE_LINE_BYTES / sizeof (u32));
 202
 203   /* Always allocate new buffers in reasonably large sized chunks. */
 204   n_alloc = clib_max (n_alloc, fl->min_n_buffers_each_alloc);
 205
 206   vec_validate_aligned (d->mbuf_alloc_list, n_alloc - 1,
 207                         CLIB_CACHE_LINE_BYTES);
 208
 209   if (rte_mempool_get_bulk (rmp, (void *) d->mbuf_alloc_list, n_alloc) < 0)
 210     return 0;
 211
 212   clib_memset (&bt, 0, sizeof (vlib_buffer_t));
 213   bt.buffer_pool_index = privp->buffer_pool_index;
 214
 215   _vec_len (d->mbuf_alloc_list) = n_alloc;
 216
 217   first = vec_len (fl->buffers);
 218   vec_resize_aligned (fl->buffers, n_alloc, CLIB_CACHE_LINE_BYTES);
 219
 220   n_left = n_alloc;
 221   mb = d->mbuf_alloc_list;
 222   bi = fl->buffers + first;
 223
 224   ASSERT (n_left % 8 == 0);
 225
 226   while (n_left >= 8)
 227     {
 228       if (PREDICT_FALSE (n_left < 24))
 229         goto no_prefetch;
 230
 231       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[16]), STORE);
 232       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[17]), STORE);
 233       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[18]), STORE);
 234       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[19]), STORE);
 235       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[20]), STORE);
 236       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[21]), STORE);
 237       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[22]), STORE);
 238       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[23]), STORE);
 239
 240     no_prefetch:
 241       vlib_get_buffer_indices_with_offset (vm, (void **) mb, bi, 8,
 242                                            sizeof (struct rte_mbuf));
 243
 244       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[0]), &bt);
 245       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[1]), &bt);
 246       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[2]), &bt);
 247       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[3]), &bt);
 248       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[4]), &bt);
 249       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[5]), &bt);
 250       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[6]), &bt);
 251       vlib_buffer_copy_template (vlib_buffer_from_rte_mbuf (mb[7]), &bt);
 252
 253       n_left -= 8;
 254       mb += 8;
 255       bi += 8;
 256     }
 257
 258   if (fl->buffer_init_function)
 259     fl->buffer_init_function (vm, fl, fl->buffers + first, n_alloc);
 260
 261   fl->n_alloc += n_alloc;
 262
 263   return n_alloc;
 264 }
 265
 266 static_always_inline void
 267 dpdk_prefetch_buffer (vlib_buffer_t * b)
 268 {
 269   struct rte_mbuf *mb;
 270   mb = rte_mbuf_from_vlib_buffer (b);
 271   CLIB_PREFETCH (mb, 2 * CLIB_CACHE_LINE_BYTES, LOAD);
 272   CLIB_PREFETCH (b, CLIB_CACHE_LINE_BYTES, LOAD);
 273 }
 274
 275 static_always_inline void
 276 recycle_or_free (vlib_main_t * vm, vlib_buffer_main_t * bm, u32 bi,
 277                  vlib_buffer_t * b)
 278 {
 279   u32 thread_index = vlib_get_thread_index ();
 280
 281   dpdk_rte_pktmbuf_free (vm, thread_index, b, 1);
 282 }
 283
 284 static_always_inline void
 285 vlib_buffer_free_inline (vlib_main_t * vm,
 286                          u32 * buffers, u32 n_buffers, u32 follow_buffer_next)
 287 {
 288   vlib_buffer_main_t *bm = vm->buffer_main;
 289   vlib_buffer_t *bufp[n_buffers], **b = bufp;
 290   u32 thread_index = vm->thread_index;
 291   int i = 0;
 292   u32 simple_mask = VLIB_BUFFER_NEXT_PRESENT;
 293   u32 n_left, *bi;
 294   u32 (*cb) (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
 295              u32 follow_buffer_next);
 296
 297   cb = bm->buffer_free_callback;
 298
 299   if (PREDICT_FALSE (cb != 0))
 300     n_buffers = (*cb) (vm, buffers, n_buffers, follow_buffer_next);
 301
 302   if (!n_buffers)
 303     return;
 304
 305   n_left = n_buffers;
 306   bi = buffers;
 307   b = bufp;
 308   vlib_get_buffers (vm, bi, b, n_buffers);
 309
 310   while (n_left >= 4)
 311     {
 312       u32 or_flags;
 313       vlib_buffer_t **p;
 314
 315       if (n_left < 16)
 316         goto no_prefetch;
 317
 318       p = b + 12;
 319       dpdk_prefetch_buffer (p[0]);
 320       dpdk_prefetch_buffer (p[1]);
 321       dpdk_prefetch_buffer (p[2]);
 322       dpdk_prefetch_buffer (p[3]);
 323     no_prefetch:
 324
 325       for (i = 0; i < 4; i++)
 326         VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[i]);
 327
 328       or_flags = b[0]->flags | b[1]->flags | b[2]->flags | b[3]->flags;
 329
 330       if (or_flags & simple_mask)
 331         {
 332           recycle_or_free (vm, bm, bi[0], b[0]);
 333           recycle_or_free (vm, bm, bi[1], b[1]);
 334           recycle_or_free (vm, bm, bi[2], b[2]);
 335           recycle_or_free (vm, bm, bi[3], b[3]);
 336         }
 337       else
 338         {
 339           dpdk_rte_pktmbuf_free (vm, thread_index, b[0], 0);
 340           dpdk_rte_pktmbuf_free (vm, thread_index, b[1], 0);
 341           dpdk_rte_pktmbuf_free (vm, thread_index, b[2], 0);
 342           dpdk_rte_pktmbuf_free (vm, thread_index, b[3], 0);
 343         }
 344       bi += 4;
 345       b += 4;
 346       n_left -= 4;
 347     }
 348   while (n_left)
 349     {
 350       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[0]);
 351       recycle_or_free (vm, bm, bi[0], b[0]);
 352       bi += 1;
 353       b += 1;
 354       n_left -= 1;
 355     }
 356 }
 357
 358 void
 359 CLIB_MULTIARCH_FN (dpdk_buffer_free) (vlib_main_t * vm, u32 * buffers,
 360                                       u32 n_buffers)
 361 {
 362   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
 363                            1);
 364 }
 365
 366 void
 367 CLIB_MULTIARCH_FN (dpdk_buffer_free_no_next) (vlib_main_t * vm, u32 * buffers,
 368                                               u32 n_buffers)
 369 {
 370   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
 371                            0);
 372 }
 373
 374 #ifndef CLIB_MARCH_VARIANT
 375 clib_error_t *
 376 dpdk_pool_create (vlib_main_t * vm, u8 * pool_name, u32 elt_size,
 377                   u32 num_elts, u32 pool_priv_size, u16 cache_size, u8 numa,
 378                   struct rte_mempool **_mp, u32 * map_index)
 379 {
 380   struct rte_mempool *mp;
 381   enum rte_iova_mode iova_mode;
 382   dpdk_mempool_private_t priv;
 383   vlib_physmem_map_t *pm;
 384   clib_error_t *error = 0;
 385   size_t min_chunk_size, align;
 386   int map_dma = 1;
 387   u32 size;
 388   i32 ret;
 389   uword i;
 390
 391   mp = rte_mempool_create_empty ((char *) pool_name, num_elts, elt_size,
 392                                  512, pool_priv_size, numa, 0);
 393   if (!mp)
 394     return clib_error_return (0, "failed to create %s", pool_name);
 395
 396   rte_mempool_set_ops_byname (mp, RTE_MBUF_DEFAULT_MEMPOOL_OPS, NULL);
 397
 398   size = rte_mempool_op_calc_mem_size_default (mp, num_elts, 21,
 399                                                &min_chunk_size, &align);
 400
 401   if ((error = vlib_physmem_shared_map_create (vm, (char *) pool_name, size,
 402                                                0, numa, map_index)))
 403     {
 404       rte_mempool_free (mp);
 405       return error;
 406     }
 407   pm = vlib_physmem_get_map (vm, *map_index);
 408
 409   /* Call the mempool priv initializer */
 410   priv.mbp_priv.mbuf_data_room_size = VLIB_BUFFER_PRE_DATA_SIZE +
 411     VLIB_BUFFER_DATA_SIZE;
 412   priv.mbp_priv.mbuf_priv_size = VLIB_BUFFER_HDR_SIZE;
 413   rte_pktmbuf_pool_init (mp, &priv);
 414
 415   if (rte_eth_dev_count_avail () == 0)
 416     map_dma = 0;
 417
 418   iova_mode = rte_eal_iova_mode ();
 419   for (i = 0; i < pm->n_pages; i++)
 420     {
 421       size_t page_sz = 1ULL << pm->log2_page_size;
 422       char *va = ((char *) pm->base) + i * page_sz;
 423       uword pa = iova_mode == RTE_IOVA_VA ?
 424         pointer_to_uword (va) : pm->page_table[i];
 425       ret = rte_mempool_populate_iova (mp, va, pa, page_sz, 0, 0);
 426       if (ret < 0)
 427         {
 428           rte_mempool_free (mp);
 429           return clib_error_return (0, "failed to populate %s", pool_name);
 430         }
 431       /* -1 likely means there is no PCI devices assigned to vfio
 432          container or noiommu mode is used  so we stop trying */
 433       if (map_dma && rte_vfio_dma_map (pointer_to_uword (va), pa, page_sz))
 434         map_dma = 0;
 435     }
 436
 437   _mp[0] = mp;
 438
 439   return 0;
 440 }
 441
 442 clib_error_t *
 443 dpdk_buffer_pool_create (vlib_main_t * vm, unsigned num_mbufs,
 444                          unsigned socket_id)
 445 {
 446   dpdk_main_t *dm = &dpdk_main;
 447   struct rte_mempool *rmp;
 448   clib_error_t *error = 0;
 449   u8 *pool_name;
 450   u32 elt_size, i;
 451   u32 map_index;
 452
 453   vec_validate_aligned (dm->pktmbuf_pools, socket_id, CLIB_CACHE_LINE_BYTES);
 454
 455   /* pool already exists, nothing to do */
 456   if (dm->pktmbuf_pools[socket_id])
 457     return 0;
 458
 459   pool_name = format (0, "dpdk_mbuf_pool_socket%u%c", socket_id, 0);
 460
 461   elt_size = sizeof (struct rte_mbuf) +
 462     VLIB_BUFFER_HDR_SIZE /* priv size */  +
 463     VLIB_BUFFER_PRE_DATA_SIZE + VLIB_BUFFER_DATA_SIZE;  /*data room size */
 464
 465   error = dpdk_pool_create (vm, pool_name, elt_size, num_mbufs,
 466                             sizeof (dpdk_mempool_private_t), 512, socket_id,
 467                             &rmp, &map_index);
 468
 469   vec_free (pool_name);
 470
 471   if (!error)
 472     {
 473       /* call the object initializers */
 474       rte_mempool_obj_iter (rmp, rte_pktmbuf_init, 0);
 475
 476       dpdk_mempool_private_t *privp = rte_mempool_get_priv (rmp);
 477       privp->buffer_pool_index =
 478         vlib_buffer_register_physmem_map (vm, map_index);
 479
 480       dm->pktmbuf_pools[socket_id] = rmp;
 481
 482       return 0;
 483     }
 484
 485   clib_error_report (error);
 486
 487   /* no usable pool for this socket, try to use pool from another one */
 488   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 489     {
 490       if (dm->pktmbuf_pools[i])
 491         {
 492           clib_warning ("WARNING: Failed to allocate mempool for CPU socket "
 493                         "%u. Threads running on socket %u will use socket %u "
 494                         "mempool.", socket_id, socket_id, i);
 495           dm->pktmbuf_pools[socket_id] = dm->pktmbuf_pools[i];
 496           return 0;
 497         }
 498     }
 499
 500   return clib_error_return (0, "failed to allocate mempool on socket %u",
 501                             socket_id);
 502 }
 503
 504 #if CLIB_DEBUG > 0
 505
 506 u32 *vlib_buffer_state_validation_lock;
 507 uword *vlib_buffer_state_validation_hash;
 508 void *vlib_buffer_state_heap;
 509
 510 static clib_error_t *
 511 buffer_state_validation_init (vlib_main_t * vm)
 512 {
 513   void *oldheap;
 514
 515   vlib_buffer_state_heap =
 516     mheap_alloc_with_lock (0, 10 << 20, 0 /* locked */ );
 517   oldheap = clib_mem_set_heap (vlib_buffer_state_heap);
 518
 519   vlib_buffer_state_validation_hash = hash_create (0, sizeof (uword));
 520   vec_validate_aligned (vlib_buffer_state_validation_lock, 0,
 521                         CLIB_CACHE_LINE_BYTES);
 522   clib_mem_set_heap (oldheap);
 523   return 0;
 524 }
 525
 526 VLIB_INIT_FUNCTION (buffer_state_validation_init);
 527 #endif
 528
 529 #if CLI_DEBUG
 530 struct dpdk_validate_buf_result
 531 {
 532   u32 invalid;
 533   u32 uninitialized;
 534 };
 535
 536 #define DPDK_TRAJECTORY_POISON 31
 537
 538 static void
 539 dpdk_buffer_validate_trajectory (struct rte_mempool *mp, void *opaque,
 540                                  void *obj, unsigned obj_idx)
 541 {
 542   vlib_buffer_t *b;
 543   struct dpdk_validate_buf_result *counter = opaque;
 544   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
 545   if (b->pre_data[0] != 0)
 546     {
 547       if (b->pre_data[0] == DPDK_TRAJECTORY_POISON)
 548         counter->uninitialized++;
 549       else
 550         counter->invalid++;
 551     }
 552 }
 553
 554 int
 555 dpdk_buffer_validate_trajectory_all (u32 * uninitialized)
 556 {
 557   dpdk_main_t *dm = &dpdk_main;
 558   struct dpdk_validate_buf_result counter = { 0 };
 559   int i;
 560
 561   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 562     rte_mempool_obj_iter (dm->pktmbuf_pools[i],
 563                           dpdk_buffer_validate_trajectory, &counter);
 564   if (uninitialized)
 565     *uninitialized = counter.uninitialized;
 566   return counter.invalid;
 567 }
 568
 569 static void
 570 dpdk_buffer_poison_trajectory (struct rte_mempool *mp, void *opaque,
 571                                void *obj, unsigned obj_idx)
 572 {
 573   vlib_buffer_t *b;
 574   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
 575   b->pre_data[0] = DPDK_TRAJECTORY_POISON;
 576 }
 577
 578 void
 579 dpdk_buffer_poison_trajectory_all (void)
 580 {
 581   dpdk_main_t *dm = &dpdk_main;
 582   int i;
 583
 584   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 585     rte_mempool_obj_iter (dm->pktmbuf_pools[i], dpdk_buffer_poison_trajectory,
 586                           0);
 587 }
 588 #endif
 589
 590 static clib_error_t *
 591 dpdk_buffer_init (vlib_main_t * vm)
 592 {
 593   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
 594   vlib_thread_main_t *tm = vlib_get_thread_main ();
 595
 596   vec_validate_aligned (dbm->ptd, tm->n_vlib_mains - 1,
 597                         CLIB_CACHE_LINE_BYTES);
 598
 599   dbm->vfio_container_fd = -1;
 600
 601   return 0;
 602 }
 603
 604 VLIB_INIT_FUNCTION (dpdk_buffer_init);
 605
 606 /* *INDENT-OFF* */
 607 VLIB_BUFFER_REGISTER_CALLBACKS (dpdk, static) = {
 608   .vlib_buffer_fill_free_list_cb = &dpdk_buffer_fill_free_list,
 609   .vlib_buffer_free_cb = &dpdk_buffer_free,
 610   .vlib_buffer_free_no_next_cb = &dpdk_buffer_free_no_next,
 611   .vlib_buffer_delete_free_list_cb = &dpdk_buffer_delete_free_list,
 612 };
 613 /* *INDENT-ON* */
 614
 615 #if __x86_64__
 616 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx512;
 617 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx2;
 618 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx512;
 619 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx2;
 620 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx512;
 621 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx2;
 622
 623 static void __clib_constructor
 624 dpdk_input_multiarch_select (void)
 625 {
 626   vlib_buffer_callbacks_t *cb = &__dpdk_buffer_callbacks;
 627   if (dpdk_buffer_fill_free_list_avx512 && clib_cpu_supports_avx512f ())
 628     {
 629       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx512;
 630       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx512;
 631       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx512;
 632     }
 633   else if (dpdk_buffer_fill_free_list_avx2 && clib_cpu_supports_avx2 ())
 634     {
 635       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx2;
 636       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx2;
 637       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx2;
 638     }
 639 }
 640 #endif
 641 #endif
 642
 643 /** @endcond */
 644 /*
 645  * fd.io coding-style-patch-verification: ON
 646  *
 647  * Local Variables:
 648  * eval: (c-set-style "gnu")
 649  * End:
 650  */