c11 safe string handling support
[vpp.git] / 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   vlib_buffer_init_for_free_list (&bt, fl);
214   bt.buffer_pool_index = privp->buffer_pool_index;
215
216   _vec_len (d->mbuf_alloc_list) = n_alloc;
217
218   first = vec_len (fl->buffers);
219   vec_resize_aligned (fl->buffers, n_alloc, CLIB_CACHE_LINE_BYTES);
220
221   n_left = n_alloc;
222   mb = d->mbuf_alloc_list;
223   bi = fl->buffers + first;
224
225   ASSERT (n_left % 8 == 0);
226
227   while (n_left >= 8)
228     {
229       if (PREDICT_FALSE (n_left < 24))
230         goto no_prefetch;
231
232       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[16]), STORE);
233       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[17]), STORE);
234       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[18]), STORE);
235       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[19]), STORE);
236       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[20]), STORE);
237       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[21]), STORE);
238       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[22]), STORE);
239       vlib_prefetch_buffer_header (vlib_buffer_from_rte_mbuf (mb[23]), STORE);
240
241     no_prefetch:
242       vlib_get_buffer_indices_with_offset (vm, (void **) mb, bi, 8,
243                                            sizeof (struct rte_mbuf));
244       clib_memcpy64_x4 (vlib_buffer_from_rte_mbuf (mb[0]),
245                         vlib_buffer_from_rte_mbuf (mb[1]),
246                         vlib_buffer_from_rte_mbuf (mb[2]),
247                         vlib_buffer_from_rte_mbuf (mb[3]), &bt);
248       clib_memcpy64_x4 (vlib_buffer_from_rte_mbuf (mb[4]),
249                         vlib_buffer_from_rte_mbuf (mb[5]),
250                         vlib_buffer_from_rte_mbuf (mb[6]),
251                         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 = &buffer_main;
289   vlib_buffer_t *bufp[n_buffers], **b = bufp;
290   u32 thread_index = vlib_get_thread_index ();
291   int i = 0;
292   u32 simple_mask = (VLIB_BUFFER_NON_DEFAULT_FREELIST |
293                      VLIB_BUFFER_NEXT_PRESENT);
294   u32 n_left, *bi;
295   u32 (*cb) (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
296              u32 follow_buffer_next);
297
298   cb = bm->buffer_free_callback;
299
300   if (PREDICT_FALSE (cb != 0))
301     n_buffers = (*cb) (vm, buffers, n_buffers, follow_buffer_next);
302
303   if (!n_buffers)
304     return;
305
306   n_left = n_buffers;
307   bi = buffers;
308   b = bufp;
309   vlib_get_buffers (vm, bi, b, n_buffers);
310
311   while (n_left >= 4)
312     {
313       u32 or_flags;
314       vlib_buffer_t **p;
315
316       if (n_left < 16)
317         goto no_prefetch;
318
319       p = b + 12;
320       dpdk_prefetch_buffer (p[0]);
321       dpdk_prefetch_buffer (p[1]);
322       dpdk_prefetch_buffer (p[2]);
323       dpdk_prefetch_buffer (p[3]);
324     no_prefetch:
325
326       for (i = 0; i < 4; i++)
327         VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[i]);
328
329       or_flags = b[0]->flags | b[1]->flags | b[2]->flags | b[3]->flags;
330
331       if (or_flags & simple_mask)
332         {
333           recycle_or_free (vm, bm, bi[0], b[0]);
334           recycle_or_free (vm, bm, bi[1], b[1]);
335           recycle_or_free (vm, bm, bi[2], b[2]);
336           recycle_or_free (vm, bm, bi[3], b[3]);
337         }
338       else
339         {
340           dpdk_rte_pktmbuf_free (vm, thread_index, b[0], 0);
341           dpdk_rte_pktmbuf_free (vm, thread_index, b[1], 0);
342           dpdk_rte_pktmbuf_free (vm, thread_index, b[2], 0);
343           dpdk_rte_pktmbuf_free (vm, thread_index, b[3], 0);
344         }
345       bi += 4;
346       b += 4;
347       n_left -= 4;
348     }
349   while (n_left)
350     {
351       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[0]);
352       recycle_or_free (vm, bm, bi[0], b[0]);
353       bi += 1;
354       b += 1;
355       n_left -= 1;
356     }
357 }
358
359 void
360 CLIB_MULTIARCH_FN (dpdk_buffer_free) (vlib_main_t * vm, u32 * buffers,
361                                       u32 n_buffers)
362 {
363   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
364                            1);
365 }
366
367 void
368 CLIB_MULTIARCH_FN (dpdk_buffer_free_no_next) (vlib_main_t * vm, u32 * buffers,
369                                               u32 n_buffers)
370 {
371   vlib_buffer_free_inline (vm, buffers, n_buffers,      /* follow_buffer_next */
372                            0);
373 }
374
375 #ifndef CLIB_MARCH_VARIANT
376 static void
377 dpdk_packet_template_init (vlib_main_t * vm,
378                            void *vt,
379                            void *packet_data,
380                            uword n_packet_data_bytes,
381                            uword min_n_buffers_each_alloc, u8 * name)
382 {
383   vlib_packet_template_t *t = (vlib_packet_template_t *) vt;
384
385   vlib_worker_thread_barrier_sync (vm);
386   clib_memset (t, 0, sizeof (t[0]));
387
388   vec_add (t->packet_data, packet_data, n_packet_data_bytes);
389
390   vlib_worker_thread_barrier_release (vm);
391 }
392
393 clib_error_t *
394 dpdk_pool_create (vlib_main_t * vm, u8 * pool_name, u32 elt_size,
395                   u32 num_elts, u32 pool_priv_size, u16 cache_size, u8 numa,
396                   struct rte_mempool **_mp, vlib_physmem_region_index_t * pri)
397 {
398   struct rte_mempool *mp;
399   enum rte_iova_mode iova_mode;
400   vlib_physmem_region_t *pr;
401   dpdk_mempool_private_t priv;
402   clib_error_t *error = 0;
403   size_t min_chunk_size, align;
404   int map_dma = 1;
405   u32 size;
406   i32 ret;
407   uword i;
408
409
410   mp = rte_mempool_create_empty ((char *) pool_name, num_elts, elt_size,
411                                  512, pool_priv_size, numa, 0);
412   if (!mp)
413     return clib_error_return (0, "failed to create %s", pool_name);
414
415   rte_mempool_set_ops_byname (mp, RTE_MBUF_DEFAULT_MEMPOOL_OPS, NULL);
416
417   size = rte_mempool_op_calc_mem_size_default (mp, num_elts, 21,
418                                                &min_chunk_size, &align);
419
420   error = vlib_physmem_region_alloc (vm, (char *) pool_name, size, numa,
421                                      VLIB_PHYSMEM_F_HUGETLB |
422                                      VLIB_PHYSMEM_F_SHARED, pri);
423   if (error)
424     {
425       rte_mempool_free (mp);
426       return error;
427     }
428
429   pr = vlib_physmem_get_region (vm, pri[0]);
430
431   /* Call the mempool priv initializer */
432   priv.mbp_priv.mbuf_data_room_size = VLIB_BUFFER_PRE_DATA_SIZE +
433     VLIB_BUFFER_DATA_SIZE;
434   priv.mbp_priv.mbuf_priv_size = VLIB_BUFFER_HDR_SIZE;
435   rte_pktmbuf_pool_init (mp, &priv);
436
437   if (rte_eth_dev_count_avail () == 0)
438     map_dma = 0;
439
440   iova_mode = rte_eal_iova_mode ();
441   for (i = 0; i < pr->n_pages; i++)
442     {
443       size_t page_sz = 1ull << pr->log2_page_size;
444       char *va = ((char *) pr->mem) + i * page_sz;
445       uword pa = iova_mode == RTE_IOVA_VA ?
446         pointer_to_uword (va) : pr->page_table[i];
447       ret = rte_mempool_populate_iova (mp, va, pa, page_sz, 0, 0);
448       if (ret < 0)
449         {
450           rte_mempool_free (mp);
451           return clib_error_return (0, "failed to populate %s", pool_name);
452         }
453       /* -1 likely means there is no PCI devices assigned to vfio
454          container or noiommu mode is used  so we stop trying */
455       if (map_dma && rte_vfio_dma_map (pointer_to_uword (va), pa, page_sz))
456         map_dma = 0;
457     }
458
459   _mp[0] = mp;
460
461   return 0;
462 }
463
464 clib_error_t *
465 dpdk_buffer_pool_create (vlib_main_t * vm, unsigned num_mbufs,
466                          unsigned socket_id)
467 {
468   dpdk_main_t *dm = &dpdk_main;
469   struct rte_mempool *rmp;
470   vlib_physmem_region_index_t pri;
471   clib_error_t *error = 0;
472   u8 *pool_name;
473   u32 elt_size, i;
474
475   vec_validate_aligned (dm->pktmbuf_pools, socket_id, CLIB_CACHE_LINE_BYTES);
476
477   /* pool already exists, nothing to do */
478   if (dm->pktmbuf_pools[socket_id])
479     return 0;
480
481   pool_name = format (0, "dpdk_mbuf_pool_socket%u%c", socket_id, 0);
482
483   elt_size = sizeof (struct rte_mbuf) +
484     VLIB_BUFFER_HDR_SIZE /* priv size */  +
485     VLIB_BUFFER_PRE_DATA_SIZE + VLIB_BUFFER_DATA_SIZE;  /*data room size */
486
487   error =
488     dpdk_pool_create (vm, pool_name, elt_size, num_mbufs,
489                       sizeof (dpdk_mempool_private_t), 512, socket_id,
490                       &rmp, &pri);
491
492   vec_free (pool_name);
493
494   if (!error)
495     {
496       /* call the object initializers */
497       rte_mempool_obj_iter (rmp, rte_pktmbuf_init, 0);
498
499       dpdk_mempool_private_t *privp = rte_mempool_get_priv (rmp);
500       privp->buffer_pool_index = vlib_buffer_pool_create (vm, pri, 0);
501
502       dm->pktmbuf_pools[socket_id] = rmp;
503
504       return 0;
505     }
506
507   clib_error_report (error);
508
509   /* no usable pool for this socket, try to use pool from another one */
510   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
511     {
512       if (dm->pktmbuf_pools[i])
513         {
514           clib_warning ("WARNING: Failed to allocate mempool for CPU socket "
515                         "%u. Threads running on socket %u will use socket %u "
516                         "mempool.", socket_id, socket_id, i);
517           dm->pktmbuf_pools[socket_id] = dm->pktmbuf_pools[i];
518           return 0;
519         }
520     }
521
522   return clib_error_return (0, "failed to allocate mempool on socket %u",
523                             socket_id);
524 }
525
526 #if CLIB_DEBUG > 0
527
528 u32 *vlib_buffer_state_validation_lock;
529 uword *vlib_buffer_state_validation_hash;
530 void *vlib_buffer_state_heap;
531
532 static clib_error_t *
533 buffer_state_validation_init (vlib_main_t * vm)
534 {
535   void *oldheap;
536
537   vlib_buffer_state_heap =
538     mheap_alloc_with_lock (0, 10 << 20, 0 /* locked */ );
539   oldheap = clib_mem_set_heap (vlib_buffer_state_heap);
540
541   vlib_buffer_state_validation_hash = hash_create (0, sizeof (uword));
542   vec_validate_aligned (vlib_buffer_state_validation_lock, 0,
543                         CLIB_CACHE_LINE_BYTES);
544   clib_mem_set_heap (oldheap);
545   return 0;
546 }
547
548 VLIB_INIT_FUNCTION (buffer_state_validation_init);
549 #endif
550
551 #if CLI_DEBUG
552 struct dpdk_validate_buf_result
553 {
554   u32 invalid;
555   u32 uninitialized;
556 };
557
558 #define DPDK_TRAJECTORY_POISON 31
559
560 static void
561 dpdk_buffer_validate_trajectory (struct rte_mempool *mp, void *opaque,
562                                  void *obj, unsigned obj_idx)
563 {
564   vlib_buffer_t *b;
565   struct dpdk_validate_buf_result *counter = opaque;
566   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
567   if (b->pre_data[0] != 0)
568     {
569       if (b->pre_data[0] == DPDK_TRAJECTORY_POISON)
570         counter->uninitialized++;
571       else
572         counter->invalid++;
573     }
574 }
575
576 int
577 dpdk_buffer_validate_trajectory_all (u32 * uninitialized)
578 {
579   dpdk_main_t *dm = &dpdk_main;
580   struct dpdk_validate_buf_result counter = { 0 };
581   int i;
582
583   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
584     rte_mempool_obj_iter (dm->pktmbuf_pools[i],
585                           dpdk_buffer_validate_trajectory, &counter);
586   if (uninitialized)
587     *uninitialized = counter.uninitialized;
588   return counter.invalid;
589 }
590
591 static void
592 dpdk_buffer_poison_trajectory (struct rte_mempool *mp, void *opaque,
593                                void *obj, unsigned obj_idx)
594 {
595   vlib_buffer_t *b;
596   b = vlib_buffer_from_rte_mbuf ((struct rte_mbuf *) obj);
597   b->pre_data[0] = DPDK_TRAJECTORY_POISON;
598 }
599
600 void
601 dpdk_buffer_poison_trajectory_all (void)
602 {
603   dpdk_main_t *dm = &dpdk_main;
604   int i;
605
606   for (i = 0; i < vec_len (dm->pktmbuf_pools); i++)
607     rte_mempool_obj_iter (dm->pktmbuf_pools[i], dpdk_buffer_poison_trajectory,
608                           0);
609 }
610 #endif
611
612 static clib_error_t *
613 dpdk_buffer_init (vlib_main_t * vm)
614 {
615   dpdk_buffer_main_t *dbm = &dpdk_buffer_main;
616   vlib_thread_main_t *tm = vlib_get_thread_main ();
617
618   vec_validate_aligned (dbm->ptd, tm->n_vlib_mains - 1,
619                         CLIB_CACHE_LINE_BYTES);
620
621   dbm->vfio_container_fd = -1;
622
623   return 0;
624 }
625
626 VLIB_INIT_FUNCTION (dpdk_buffer_init);
627
628 /* *INDENT-OFF* */
629 VLIB_BUFFER_REGISTER_CALLBACKS (dpdk, static) = {
630   .vlib_buffer_fill_free_list_cb = &dpdk_buffer_fill_free_list,
631   .vlib_buffer_free_cb = &dpdk_buffer_free,
632   .vlib_buffer_free_no_next_cb = &dpdk_buffer_free_no_next,
633   .vlib_packet_template_init_cb = &dpdk_packet_template_init,
634   .vlib_buffer_delete_free_list_cb = &dpdk_buffer_delete_free_list,
635 };
636 /* *INDENT-ON* */
637
638 #if __x86_64__
639 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx512;
640 vlib_buffer_fill_free_list_cb_t __clib_weak dpdk_buffer_fill_free_list_avx2;
641 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx512;
642 vlib_buffer_free_cb_t __clib_weak dpdk_buffer_free_avx2;
643 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx512;
644 vlib_buffer_free_no_next_cb_t __clib_weak dpdk_buffer_free_no_next_avx2;
645
646 static void __clib_constructor
647 dpdk_input_multiarch_select (void)
648 {
649   vlib_buffer_callbacks_t *cb = &__dpdk_buffer_callbacks;
650   if (dpdk_buffer_fill_free_list_avx512 && clib_cpu_supports_avx512f ())
651     {
652       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx512;
653       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx512;
654       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx512;
655     }
656   else if (dpdk_buffer_fill_free_list_avx2 && clib_cpu_supports_avx2 ())
657     {
658       cb->vlib_buffer_fill_free_list_cb = dpdk_buffer_fill_free_list_avx2;
659       cb->vlib_buffer_free_cb = dpdk_buffer_free_avx2;
660       cb->vlib_buffer_free_no_next_cb = dpdk_buffer_free_no_next_avx2;
661     }
662 }
663 #endif
664 #endif
665
666 /** @endcond */
667 /*
668  * fd.io coding-style-patch-verification: ON
669  *
670  * Local Variables:
671  * eval: (c-set-style "gnu")
672  * End:
673  */