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 <rte_config.h>
  47
  48 #include <rte_common.h>
  49 #include <rte_log.h>
  50 #include <rte_memory.h>
  51 #include <rte_memzone.h>
  52 #include <rte_tailq.h>
  53 #include <rte_eal.h>
  54 #include <rte_per_lcore.h>
  55 #include <rte_launch.h>
  56 #include <rte_atomic.h>
  57 #include <rte_cycles.h>
  58 #include <rte_prefetch.h>
  59 #include <rte_lcore.h>
  60 #include <rte_per_lcore.h>
  61 #include <rte_branch_prediction.h>
  62 #include <rte_interrupts.h>
  63 #include <rte_pci.h>
  64 #include <rte_random.h>
  65 #include <rte_debug.h>
  66 #include <rte_ether.h>
  67 #include <rte_ethdev.h>
  68 #include <rte_ring.h>
  69 #include <rte_mempool.h>
  70 #include <rte_mbuf.h>
  71 #include <rte_version.h>
  72
  73 #include <vlib/vlib.h>
  74 #include <vnet/vnet.h>
  75 #include <dpdk/device/dpdk.h>
  76 #include <dpdk/device/dpdk_priv.h>
  77
  78
  79 STATIC_ASSERT (VLIB_BUFFER_PRE_DATA_SIZE == RTE_PKTMBUF_HEADROOM,
  80                "VLIB_BUFFER_PRE_DATA_SIZE must be equal to RTE_PKTMBUF_HEADROOM");
  81
  82 typedef struct
  83 {
  84   CLIB_CACHE_LINE_ALIGN_MARK (cacheline0);
  85   struct rte_mbuf ***mbuf_pending_free_list;
  86
  87   /* cached last pool */
  88   struct rte_mempool *last_pool;
  89   u8 last_buffer_pool_index;
  90 } dpdk_buffer_per_thread_data;
  91
  92 typedef struct
  93 {
  94   dpdk_buffer_per_thread_data *ptd;
  95 } dpdk_buffer_main_t;
  96
  97 dpdk_buffer_main_t dpdk_buffer_main;
  98
  99 static_always_inline void
 100 dpdk_rte_pktmbuf_free (vlib_main_t * vm, u32 thread_index, vlib_buffer_t * b)
 101 {
 102   vlib_buffer_t *hb = b;
 103   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
 104   dpdk_buffer_per_thread_data *d = vec_elt_at_index (dbm->ptd, thread_index);
 105   struct rte_mbuf *mb;
 106   u32 next, flags;
 107   mb = rte_mbuf_from_vlib_buffer (hb);
 108
 109 next:
 110   flags = b->flags;
 111   next = b->next_buffer;
 112   mb = rte_mbuf_from_vlib_buffer (b);
 113
 114   if (PREDICT_FALSE (b->n_add_refs))
 115     {
 116       rte_mbuf_refcnt_update (mb, b->n_add_refs);
 117       b->n_add_refs = 0;
 118     }
 119
 120   mb = rte_pktmbuf_prefree_seg (mb);
 121   if (mb)
 122     {
 123       if (mb->pool != d->last_pool)
 124         {
 125           d->last_pool = mb->pool;
 126           dpdk_mempool_private_t *privp = rte_mempool_get_priv (d->last_pool);
 127           d->last_buffer_pool_index = privp->buffer_pool_index;
 128           vec_validate_aligned (d->mbuf_pending_free_list,
 129                                 d->last_buffer_pool_index,
 130                                 CLIB_CACHE_LINE_BYTES);
 131         }
 132       vec_add1 (d->mbuf_pending_free_list[d->last_buffer_pool_index], mb);
 133     }
 134
 135   if (flags & VLIB_BUFFER_NEXT_PRESENT)
 136     {
 137       b = vlib_get_buffer (vm, next);
 138       goto next;
 139     }
 140 }
 141
 142 #ifndef CLIB_MULTIARCH_VARIANT
 143 static void
 144 del_free_list (vlib_main_t * vm, vlib_buffer_free_list_t * f)
 145 {
 146   u32 i;
 147   vlib_buffer_t *b;
 148   u32 thread_index = vlib_get_thread_index ();
 149
 150   for (i = 0; i < vec_len (f->buffers); i++)
 151     {
 152       b = vlib_get_buffer (vm, f->buffers[i]);
 153       dpdk_rte_pktmbuf_free (vm, thread_index, b);
 154     }
 155
 156   vec_free (f->name);
 157   vec_free (f->buffers);
 158 }
 159
 160 /* Add buffer free list. */
 161 static void
 162 dpdk_buffer_delete_free_list (vlib_main_t * vm,
 163                               vlib_buffer_free_list_index_t free_list_index)
 164 {
 165   vlib_buffer_main_t *bm = vm->buffer_main;
 166   vlib_buffer_free_list_t *f;
 167   vlib_buffer_free_list_index_t merge_index;
 168   int i;
 169
 170   ASSERT (vlib_get_thread_index () == 0);
 171
 172   f = vlib_buffer_get_free_list (vm, free_list_index);
 173
 174   merge_index = vlib_buffer_get_free_list_with_size (vm, f->n_data_bytes);
 175   if (merge_index != (vlib_buffer_free_list_index_t) ~ 0 &&
 176       merge_index != free_list_index)
 177     {
 178       vlib_buffer_merge_free_lists (pool_elt_at_index
 179                                     (bm->buffer_free_list_pool, merge_index),
 180                                     f);
 181     }
 182
 183   del_free_list (vm, f);
 184
 185   /* Poison it. */
 186   memset (f, 0xab, sizeof (f[0]));
 187
 188   pool_put (bm->buffer_free_list_pool, f);
 189
 190   for (i = 1; i < vec_len (vlib_mains); i++)
 191     {
 192       bm = vlib_mains[i]->buffer_main;
 193       f = vlib_buffer_get_free_list (vlib_mains[i], free_list_index);;
 194       memset (f, 0xab, sizeof (f[0]));
 195       pool_put (bm->buffer_free_list_pool, f);
 196     }
 197 }
 198 #endif
 199
 200 /* Make sure free list has at least given number of free buffers. */
 201 uword
 202 CLIB_MULTIARCH_FN (dpdk_buffer_fill_free_list) (vlib_main_t * vm,
 203                                                 vlib_buffer_free_list_t * fl,
 204                                                 uword min_free_buffers)
 205 {
 206   dpdk_main_t *dm = &dpdk_main;
 207   vlib_buffer_t *b0, *b1, *b2, *b3;
 208   int n, i;
 209   u32 bi0, bi1, bi2, bi3;
 210   unsigned socket_id = rte_socket_id ();
 211   struct rte_mempool *rmp = dm->pktmbuf_pools[socket_id];
 212   dpdk_mempool_private_t *privp = rte_mempool_get_priv (rmp);
 213   struct rte_mbuf *mb0, *mb1, *mb2, *mb3;
 214   vlib_buffer_t bt;
 215
 216   /* Too early? */
 217   if (PREDICT_FALSE (rmp == 0))
 218     return 0;
 219
 220   /* Already have enough free buffers on free list? */
 221   n = min_free_buffers - vec_len (fl->buffers);
 222   if (n <= 0)
 223     return min_free_buffers;
 224
 225   /* Always allocate round number of buffers. */
 226   n = round_pow2 (n, CLIB_CACHE_LINE_BYTES / sizeof (u32));
 227
 228   /* Always allocate new buffers in reasonably large sized chunks. */
 229   n = clib_max (n, fl->min_n_buffers_each_physmem_alloc);
 230
 231   vec_validate_aligned (vm->mbuf_alloc_list, n - 1, CLIB_CACHE_LINE_BYTES);
 232
 233   if (rte_mempool_get_bulk (rmp, vm->mbuf_alloc_list, n) < 0)
 234     return 0;
 235
 236   memset (&bt, 0, sizeof (vlib_buffer_t));
 237   vlib_buffer_init_for_free_list (&bt, fl);
 238   bt.buffer_pool_index = privp->buffer_pool_index;
 239
 240   _vec_len (vm->mbuf_alloc_list) = n;
 241
 242   i = 0;
 243   int f = vec_len (fl->buffers);
 244   vec_resize_aligned (fl->buffers, n, CLIB_CACHE_LINE_BYTES);
 245
 246   while (i < (n - 7))
 247     {
 248       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf
 249                                    (vm->mbuf_alloc_list[i + 4]), STORE);
 250       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf
 251                                    (vm->mbuf_alloc_list[i + 5]), STORE);
 252       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf
 253                                    (vm->mbuf_alloc_list[i + 6]), STORE);
 254       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf
 255                                    (vm->mbuf_alloc_list[i + 7]), STORE);
 256
 257       mb0 = vm->mbuf_alloc_list[i];
 258       mb1 = vm->mbuf_alloc_list[i + 1];
 259       mb2 = vm->mbuf_alloc_list[i + 2];
 260       mb3 = vm->mbuf_alloc_list[i + 3];
 261
 262       b0 = vlib_buffer_from_rte_mbuf (mb0);
 263       b1 = vlib_buffer_from_rte_mbuf (mb1);
 264       b2 = vlib_buffer_from_rte_mbuf (mb2);
 265       b3 = vlib_buffer_from_rte_mbuf (mb3);
 266
 267       bi0 = vlib_get_buffer_index (vm, b0);
 268       bi1 = vlib_get_buffer_index (vm, b1);
 269       bi2 = vlib_get_buffer_index (vm, b2);
 270       bi3 = vlib_get_buffer_index (vm, b3);
 271
 272       fl->buffers[f++] = bi0;
 273       fl->buffers[f++] = bi1;
 274       fl->buffers[f++] = bi2;
 275       fl->buffers[f++] = bi3;
 276
 277       clib_memcpy64_x4 (b0, b1, b2, b3, &bt);
 278
 279       if (fl->buffer_init_function)
 280         {
 281           fl->buffer_init_function (vm, fl, &bi0, 1);
 282           fl->buffer_init_function (vm, fl, &bi1, 1);
 283           fl->buffer_init_function (vm, fl, &bi2, 1);
 284           fl->buffer_init_function (vm, fl, &bi3, 1);
 285         }
 286       i += 4;
 287     }
 288
 289   while (i < n)
 290     {
 291       mb0 = vm->mbuf_alloc_list[i];
 292
 293       b0 = vlib_buffer_from_rte_mbuf (mb0);
 294       bi0 = vlib_get_buffer_index (vm, b0);
 295
 296       fl->buffers[f++] = bi0;
 297       clib_memcpy (b0, &bt, sizeof (vlib_buffer_t));
 298
 299       if (fl->buffer_init_function)
 300         fl->buffer_init_function (vm, fl, &bi0, 1);
 301       i++;
 302     }
 303
 304   fl->n_alloc += n;
 305
 306   return n;
 307 }
 308
 309 static_always_inline void
 310 dpdk_prefetch_buffer_by_index (vlib_main_t * vm, u32 bi)
 311 {
 312   vlib_buffer_t *b;
 313   struct rte_mbuf *mb;
 314   b = vlib_get_buffer (vm, bi);
 315   mb = rte_mbuf_from_vlib_buffer (b);
 316   CLIB_PREFETCH (mb, 2 * CLIB_CACHE_LINE_BYTES, STORE);
 317   CLIB_PREFETCH (b, CLIB_CACHE_LINE_BYTES, LOAD);
 318 }
 319
 320 static_always_inline void
 321 recycle_or_free (vlib_main_t * vm, vlib_buffer_main_t * bm, u32 bi,
 322                  vlib_buffer_t * b)
 323 {
 324   vlib_buffer_free_list_t *fl;
 325   u32 thread_index = vlib_get_thread_index ();
 326   vlib_buffer_free_list_index_t fi;
 327   fl = vlib_buffer_get_buffer_free_list (vm, b, &fi);
 328
 329   /* The only current use of this callback: multicast recycle */
 330   if (PREDICT_FALSE (fl->buffers_added_to_freelist_function != 0))
 331     {
 332       int j;
 333
 334       vlib_buffer_add_to_free_list (vm, fl, bi,
 335                                     (b->flags & VLIB_BUFFER_RECYCLE) == 0);
 336
 337       for (j = 0; j < vec_len (bm->announce_list); j++)
 338         {
 339           if (fl == bm->announce_list[j])
 340             goto already_announced;
 341         }
 342       vec_add1 (bm->announce_list, fl);
 343     already_announced:
 344       ;
 345     }
 346   else
 347     {
 348       if (PREDICT_TRUE ((b->flags & VLIB_BUFFER_RECYCLE) == 0))
 349         dpdk_rte_pktmbuf_free (vm, thread_index, b);
 350     }
 351 }
 352
 353 static_always_inline void
 354 vlib_buffer_free_inline (vlib_main_t * vm,
 355                          u32 * buffers, u32 n_buffers, u32 follow_buffer_next)
 356 {
 357   vlib_buffer_main_t *bm = vm->buffer_main;
 358   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
 359   vlib_buffer_t *b0, *b1, *b2, *b3;
 360   u32 thread_index = vlib_get_thread_index ();
 361   dpdk_buffer_per_thread_data *d = vec_elt_at_index (dbm->ptd, thread_index);
 362   int i = 0;
 363   u32 (*cb) (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
 364              u32 follow_buffer_next);
 365
 366   cb = bm->buffer_free_callback;
 367
 368   if (PREDICT_FALSE (cb != 0))
 369     n_buffers = (*cb) (vm, buffers, n_buffers, follow_buffer_next);
 370
 371   if (!n_buffers)
 372     return;
 373
 374   while (i + 7 < n_buffers)
 375     {
 376       dpdk_prefetch_buffer_by_index (vm, buffers[i + 4]);
 377       dpdk_prefetch_buffer_by_index (vm, buffers[i + 5]);
 378       dpdk_prefetch_buffer_by_index (vm, buffers[i + 6]);
 379       dpdk_prefetch_buffer_by_index (vm, buffers[i + 7]);
 380
 381       b0 = vlib_get_buffer (vm, buffers[i]);
 382       b1 = vlib_get_buffer (vm, buffers[i + 1]);
 383       b2 = vlib_get_buffer (vm, buffers[i + 2]);
 384       b3 = vlib_get_buffer (vm, buffers[i + 3]);
 385
 386       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0);
 387       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b1);
 388       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b2);
 389       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b3);
 390
 391       recycle_or_free (vm, bm, buffers[i], b0);
 392       recycle_or_free (vm, bm, buffers[i + 1], b1);
 393       recycle_or_free (vm, bm, buffers[i + 2], b2);
 394       recycle_or_free (vm, bm, buffers[i + 3], b3);
 395
 396       i += 4;
 397     }
 398   while (i < n_buffers)
 399     {
 400       b0 = vlib_get_buffer (vm, buffers[i]);
 401       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0);
 402       recycle_or_free (vm, bm, buffers[i], b0);
 403       i++;
 404     }
 405   if (vec_len (bm->announce_list))
 406     {
 407       vlib_buffer_free_list_t *fl;
 408       for (i = 0; i < vec_len (bm->announce_list); i++)
 409         {
 410           fl = bm->announce_list[i];
 411           fl->buffers_added_to_freelist_function (vm, fl);
 412         }
 413       _vec_len (bm->announce_list) = 0;
 414     }
 415
 416   vec_foreach_index (i, d->mbuf_pending_free_list)
 417   {
 418     int len = vec_len (d->mbuf_pending_free_list[i]);
 419     if (len)
 420       {
 421         rte_mempool_put_bulk (d->mbuf_pending_free_list[i][len - 1]->pool,
 422                               (void *) d->mbuf_pending_free_list[i], len);
 423         vec_reset_length (d->mbuf_pending_free_list[i]);
 424       }
 425   }
 426 }
 427
 428 void
 429 CLIB_MULTIARCH_FN (dpdk_buffer_free) (vlib_main_t * vm, u32 * buffers,
 430                                       u32 n_buffers)
 431 {
 432   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
 433                            1);
 434 }
 435
 436 void
 437 CLIB_MULTIARCH_FN (dpdk_buffer_free_no_next) (vlib_main_t * vm, u32 * buffers,
 438                                               u32 n_buffers)
 439 {
 440   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
 441                            0);
 442 }
 443
 444 #ifndef CLIB_MULTIARCH_VARIANT
 445 static void
 446 dpdk_packet_template_init (vlib_main_t * vm,
 447                            void *vt,
 448                            void *packet_data,
 449                            uword n_packet_data_bytes,
 450                            uword min_n_buffers_each_physmem_alloc, u8 * name)
 451 {
 452   vlib_packet_template_t *t = (vlib_packet_template_t *) vt;
 453
 454   vlib_worker_thread_barrier_sync (vm);
 455   memset (t, 0, sizeof (t[0]));
 456
 457   vec_add (t->packet_data, packet_data, n_packet_data_bytes);
 458
 459   vlib_worker_thread_barrier_release (vm);
 460 }
 461
 462 clib_error_t *
 463 dpdk_pool_create (vlib_main_t * vm, u8 * pool_name, u32 elt_size,
 464                   u32 num_elts, u32 pool_priv_size, u16 cache_size, u8 numa,
 465                   struct rte_mempool **_mp, vlib_physmem_region_index_t * pri)
 466 {
 467   struct rte_mempool *mp;
 468   vlib_physmem_region_t *pr;
 469   clib_error_t *error = 0;
 470   u32 size, obj_size;
 471   i32 ret;
 472
 473   obj_size = rte_mempool_calc_obj_size (elt_size, 0, 0);
 474 #if RTE_VERSION < RTE_VERSION_NUM(17, 11, 0, 0)
 475   size = rte_mempool_xmem_size (num_elts, obj_size, 21);
 476 #else
 477   size = rte_mempool_xmem_size (num_elts, obj_size, 21, 0);
 478 #endif
 479
 480   error =
 481     vlib_physmem_region_alloc (vm, (i8 *) pool_name, size, numa, 0, pri);
 482   if (error)
 483     return error;
 484
 485   pr = vlib_physmem_get_region (vm, pri[0]);
 486
 487   mp =
 488     rte_mempool_create_empty ((i8 *) pool_name, num_elts, elt_size,
 489                               512, pool_priv_size, numa, 0);
 490   if (!mp)
 491     return clib_error_return (0, "failed to create %s", pool_name);
 492
 493   rte_mempool_set_ops_byname (mp, RTE_MBUF_DEFAULT_MEMPOOL_OPS, NULL);
 494
 495 #if RTE_VERSION < RTE_VERSION_NUM(17, 11, 0, 0)
 496   ret =
 497     rte_mempool_populate_phys_tab (mp, pr->mem, pr->page_table, pr->n_pages,
 498                                    pr->log2_page_size, NULL, NULL);
 499 #else
 500   ret =
 501     rte_mempool_populate_iova_tab (mp, pr->mem, pr->page_table, pr->n_pages,
 502                                    pr->log2_page_size, NULL, NULL);
 503 #endif
 504   if (ret != (i32) mp->size)
 505     {
 506       rte_mempool_free (mp);
 507       return clib_error_return (0, "failed to populate %s", pool_name);
 508     }
 509
 510   _mp[0] = mp;
 511
 512   return 0;
 513 }
 514
 515 clib_error_t *
 516 dpdk_buffer_pool_create (vlib_main_t * vm, unsigned num_mbufs,
 517                          unsigned socket_id)
 518 {
 519   dpdk_main_t *dm = &dpdk_main;
 520   struct rte_mempool *rmp;
 521   dpdk_mempool_private_t priv;
 522   vlib_physmem_region_index_t pri;
 523   clib_error_t *error = 0;
 524   u8 *pool_name;
 525   u32 elt_size, i;
 526
 527   vec_validate_aligned (dm->pktmbuf_pools, socket_id, CLIB_CACHE_LINE_BYTES);
 528
 529   /* pool already exists, nothing to do */
 530   if (dm->pktmbuf_pools[socket_id])
 531     return 0;
 532
 533   pool_name = format (0, "dpdk_mbuf_pool_socket%u%c", socket_id, 0);
 534
 535   elt_size = sizeof (struct rte_mbuf) +
 536     VLIB_BUFFER_HDR_SIZE /* priv size */  +
 537     VLIB_BUFFER_PRE_DATA_SIZE + VLIB_BUFFER_DATA_SIZE;  /*data room size */
 538
 539   error =
 540     dpdk_pool_create (vm, pool_name, elt_size, num_mbufs,
 541                       sizeof (dpdk_mempool_private_t), 512, socket_id,
 542                       &rmp, &pri);
 543
 544   vec_free (pool_name);
 545
 546   if (!error)
 547     {
 548       priv.mbp_priv.mbuf_data_room_size = VLIB_BUFFER_PRE_DATA_SIZE +
 549         VLIB_BUFFER_DATA_SIZE;
 550       priv.mbp_priv.mbuf_priv_size = VLIB_BUFFER_HDR_SIZE;
 551
 552       /* call the mempool priv initializer */
 553       rte_pktmbuf_pool_init (rmp, &priv);
 554
 555       /* call the object initializers */
 556       rte_mempool_obj_iter (rmp, rte_pktmbuf_init, 0);
 557
 558       dpdk_mempool_private_t *privp = rte_mempool_get_priv (rmp);
 559       privp->buffer_pool_index = vlib_buffer_add_physmem_region (vm, pri);
 560
 561       dm->pktmbuf_pools[socket_id] = rmp;
 562
 563       return 0;
 564     }
 565
 566   clib_error_report (error);
 567
 568   /* no usable pool for this socket, try to use pool from another one */
 569   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 570     {
 571       if (dm->pktmbuf_pools[i])
 572         {
 573           clib_warning ("WARNING: Failed to allocate mempool for CPU socket "
 574                         "%u. Threads running on socket %u will use socket %u "
 575                         "mempool.", socket_id, socket_id, i);
 576           dm->pktmbuf_pools[socket_id] = dm->pktmbuf_pools[i];
 577           return 0;
 578         }
 579     }
 580
 581   return clib_error_return (0, "failed to allocate mempool on socket %u",
 582                             socket_id);
 583 }
 584
 585 #if CLIB_DEBUG > 0
 586
 587 u32 *vlib_buffer_state_validation_lock;
 588 uword *vlib_buffer_state_validation_hash;
 589 void *vlib_buffer_state_heap;
 590
 591 static clib_error_t *
 592 buffer_state_validation_init (vlib_main_t * vm)
 593 {
 594   void *oldheap;
 595
 596   vlib_buffer_state_heap = mheap_alloc (0, 10 << 20);
 597
 598   oldheap = clib_mem_set_heap (vlib_buffer_state_heap);
 599
 600   vlib_buffer_state_validation_hash = hash_create (0, sizeof (uword));
 601   vec_validate_aligned (vlib_buffer_state_validation_lock, 0,
 602                         CLIB_CACHE_LINE_BYTES);
 603   clib_mem_set_heap (oldheap);
 604   return 0;
 605 }
 606
 607 VLIB_INIT_FUNCTION (buffer_state_validation_init);
 608 #endif
 609
 610 #if CLI_DEBUG
 611 struct dpdk_validate_buf_result
 612 {
 613   u32 invalid;
 614   u32 uninitialized;
 615 };
 616
 617 #define DPDK_TRAJECTORY_POISON 31
 618
 619 static void
 620 dpdk_buffer_validate_trajectory (struct rte_mempool *mp, void *opaque,
 621                                  void *obj, unsigned obj_idx)
 622 {
 623   vlib_buffer_t *b;
 624   struct dpdk_validate_buf_result *counter = opaque;
 625   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
 626   if (b->pre_data[0] != 0)
 627     {
 628       if (b->pre_data[0] == DPDK_TRAJECTORY_POISON)
 629         counter->uninitialized++;
 630       else
 631         counter->invalid++;
 632     }
 633 }
 634
 635 int
 636 dpdk_buffer_validate_trajectory_all (u32 * uninitialized)
 637 {
 638   dpdk_main_t *dm = &dpdk_main;
 639   struct dpdk_validate_buf_result counter = { 0 };
 640   int i;
 641
 642   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 643     rte_mempool_obj_iter (dm->pktmbuf_pools[i],
 644                           dpdk_buffer_validate_trajectory, &counter);
 645   if (uninitialized)
 646     *uninitialized = counter.uninitialized;
 647   return counter.invalid;
 648 }
 649
 650 static void
 651 dpdk_buffer_poison_trajectory (struct rte_mempool *mp, void *opaque,
 652                                void *obj, unsigned obj_idx)
 653 {
 654   vlib_buffer_t *b;
 655   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
 656   b->pre_data[0] = DPDK_TRAJECTORY_POISON;
 657 }
 658
 659 void
 660 dpdk_buffer_poison_trajectory_all (void)
 661 {
 662   dpdk_main_t *dm = &dpdk_main;
 663   int i;
 664
 665   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
 666     rte_mempool_obj_iter (dm->pktmbuf_pools[i], dpdk_buffer_poison_trajectory,
 667                           0);
 668 }
 669 #endif
 670
 671 static clib_error_t *
 672 dpdk_buffer_init (vlib_main_t * vm)
 673 {
 674   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
 675   vlib_thread_main_t *tm = vlib_get_thread_main ();
 676   vec_validate_aligned (dbm->ptd, tm->n_vlib_mains - 1,
 677                         CLIB_CACHE_LINE_BYTES);
 678   return 0;
 679 }
 680
 681 VLIB_INIT_FUNCTION (dpdk_buffer_init);
 682
 683 /* *INDENT-OFF* */
 684 VLIB_BUFFER_REGISTER_CALLBACKS (dpdk, static) = {
 685   .vlib_buffer_fill_free_list_cb = &dpdk_buffer_fill_free_list,
 686   .vlib_buffer_free_cb = &dpdk_buffer_free,
 687   .vlib_buffer_free_no_next_cb = &dpdk_buffer_free_no_next,
 688   .vlib_packet_template_init_cb = &dpdk_packet_template_init,
 689   .vlib_buffer_delete_free_list_cb = &dpdk_buffer_delete_free_list,
 690 };
 691 /* *INDENT-ON* */
 692
 693 #if __x86_64__
 694 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx512;
 695 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx2;
 696 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx512;
 697 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx2;
 698 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx512;
 699 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx2;
 700
 701 static void __clib_constructor
 702 dpdk_input_multiarch_select (void)
 703 {
 704   vlib_buffer_callbacks_t *cb = &__dpdk_buffer_callbacks;
 705   if (dpdk_buffer_fill_free_list_avx512 && clib_cpu_supports_avx512f ())
 706     {
 707       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx512;
 708       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx512;
 709       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx512;
 710     }
 711   else if (dpdk_buffer_fill_free_list_avx2 && clib_cpu_supports_avx2 ())
 712     {
 713       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx2;
 714       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx2;
 715       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx2;
 716     }
 717 }
 718 #endif
 719 #endif
 720
 721 /** @endcond */
 722 /*
 723  * fd.io coding-style-patch-verification: ON
 724  *
 725  * Local Variables:
 726  * eval: (c-set-style "gnu")
 727  * End:
 728  */