blob: e9d0704c83711cdfd039a61cf503161f2a454109 [file] [log] [blame]
Ed Warnickecb9cada2015-12-08 15:45:58 -07001/*
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 * buffer_funcs.h: VLIB buffer related functions/inlines
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#ifndef included_vlib_buffer_funcs_h
41#define included_vlib_buffer_funcs_h
42
43#include <vppinfra/hash.h>
Dave Barachc3a06552018-10-01 09:25:32 -040044#include <vppinfra/fifo.h>
Klement Sekeraf883f6a2019-02-13 11:01:32 +010045#include <vlib/buffer.h>
46#include <vlib/physmem_funcs.h>
47#include <vlib/main.h>
48#include <vlib/node.h>
Ed Warnickecb9cada2015-12-08 15:45:58 -070049
50/** \file
51 vlib buffer access methods.
52*/
53
Damjan Marion24dcbe42019-01-31 12:29:39 +010054always_inline void
55vlib_buffer_validate (vlib_main_t * vm, vlib_buffer_t * b)
56{
57 vlib_buffer_main_t *bm = vm->buffer_main;
58 vlib_buffer_pool_t *bp;
59
60 /* reference count in allocated buffer always must be 1 or higher */
61 ASSERT (b->ref_count > 0);
62
Damjan Marion24dcbe42019-01-31 12:29:39 +010063 /* verify that buffer pool index is valid */
64 bp = vec_elt_at_index (bm->buffer_pools, b->buffer_pool_index);
65 ASSERT (pointer_to_uword (b) >= bp->start);
66 ASSERT (pointer_to_uword (b) < bp->start + bp->size -
Damjan Marion5de3fec2019-02-06 14:22:32 +010067 (bp->data_size + sizeof (vlib_buffer_t)));
Damjan Marion24dcbe42019-01-31 12:29:39 +010068}
69
70always_inline void *
71vlib_buffer_ptr_from_index (uword buffer_mem_start, u32 buffer_index,
72 uword offset)
73{
74 offset += ((uword) buffer_index) << CLIB_LOG2_CACHE_LINE_BYTES;
75 return uword_to_pointer (buffer_mem_start + offset, vlib_buffer_t *);
76}
Ed Warnickecb9cada2015-12-08 15:45:58 -070077
78/** \brief Translate buffer index into buffer pointer
79
80 @param vm - (vlib_main_t *) vlib main data structure pointer
81 @param buffer_index - (u32) buffer index
82 @return - (vlib_buffer_t *) buffer pointer
Dave Barach9b8ffd92016-07-08 08:13:45 -040083*/
Ed Warnickecb9cada2015-12-08 15:45:58 -070084always_inline vlib_buffer_t *
85vlib_get_buffer (vlib_main_t * vm, u32 buffer_index)
86{
Damjan Mariond50e3472019-01-20 00:03:56 +010087 vlib_buffer_main_t *bm = vm->buffer_main;
Damjan Marion24dcbe42019-01-31 12:29:39 +010088 vlib_buffer_t *b;
Damjan Marion04a7f052017-07-10 15:06:17 +020089
Damjan Marion24dcbe42019-01-31 12:29:39 +010090 b = vlib_buffer_ptr_from_index (bm->buffer_mem_start, buffer_index, 0);
91 vlib_buffer_validate (vm, b);
92 return b;
Ed Warnickecb9cada2015-12-08 15:45:58 -070093}
94
Damjan Marion5de3fec2019-02-06 14:22:32 +010095static_always_inline u32
Damjan Marion8934a042019-02-09 23:29:26 +010096vlib_buffer_get_default_data_size (vlib_main_t * vm)
Damjan Marion5de3fec2019-02-06 14:22:32 +010097{
98 return vm->buffer_main->default_data_size;
99}
100
Damjan Marione58041f2019-01-18 19:56:09 +0100101static_always_inline void
Damjan Marion64d557c2019-01-18 20:03:41 +0100102vlib_buffer_copy_indices (u32 * dst, u32 * src, u32 n_indices)
103{
Damjan Marion3e94c412019-03-13 12:41:54 +0100104#if defined(CLIB_HAVE_VEC512)
105 while (n_indices >= 16)
106 {
107 u32x16_store_unaligned (u32x16_load_unaligned (src), dst);
108 dst += 16;
109 src += 16;
110 n_indices -= 16;
111 }
112#endif
113
114#if defined(CLIB_HAVE_VEC256)
115 while (n_indices >= 8)
116 {
117 u32x8_store_unaligned (u32x8_load_unaligned (src), dst);
118 dst += 8;
119 src += 8;
120 n_indices -= 8;
121 }
122#endif
123
124#if defined(CLIB_HAVE_VEC128)
125 while (n_indices >= 4)
126 {
127 u32x4_store_unaligned (u32x4_load_unaligned (src), dst);
128 dst += 4;
129 src += 4;
130 n_indices -= 4;
131 }
132#endif
133
134 while (n_indices)
135 {
136 dst[0] = src[0];
137 dst += 1;
138 src += 1;
139 n_indices -= 1;
140 }
Damjan Marion64d557c2019-01-18 20:03:41 +0100141}
142
Damjan Marion9a8a12a2019-01-23 16:52:10 +0100143STATIC_ASSERT_OFFSET_OF (vlib_buffer_t, template_end, 64);
Damjan Marion64d557c2019-01-18 20:03:41 +0100144static_always_inline void
Damjan Marione58041f2019-01-18 19:56:09 +0100145vlib_buffer_copy_template (vlib_buffer_t * b, vlib_buffer_t * bt)
146{
Damjan Marion22f23ae2019-01-24 15:36:57 +0100147#if defined CLIB_HAVE_VEC512
148 b->as_u8x64[0] = bt->as_u8x64[0];
149#elif defined (CLIB_HAVE_VEC256)
Damjan Marion9a8a12a2019-01-23 16:52:10 +0100150 b->as_u8x32[0] = bt->as_u8x32[0];
151 b->as_u8x32[1] = bt->as_u8x32[1];
152#elif defined (CLIB_HAVE_VEC128)
153 b->as_u8x16[0] = bt->as_u8x16[0];
154 b->as_u8x16[1] = bt->as_u8x16[1];
155 b->as_u8x16[2] = bt->as_u8x16[2];
156 b->as_u8x16[3] = bt->as_u8x16[3];
157#else
158 clib_memcpy_fast (b, bt, 64);
159#endif
Damjan Marione58041f2019-01-18 19:56:09 +0100160}
161
Damjan Marion910d3692019-01-21 11:48:34 +0100162always_inline u8
163vlib_buffer_pool_get_default_for_numa (vlib_main_t * vm, u32 numa_node)
164{
Damjan Marionb592d1b2019-02-28 23:16:11 +0100165 ASSERT (numa_node < VLIB_BUFFER_MAX_NUMA_NODES);
166 return vm->buffer_main->default_buffer_pool_index_for_numa[numa_node];
Damjan Marion910d3692019-01-21 11:48:34 +0100167}
168
Damjan Marionafe56de2018-05-17 12:44:00 +0200169/** \brief Translate array of buffer indices into buffer pointers with offset
170
171 @param vm - (vlib_main_t *) vlib main data structure pointer
172 @param bi - (u32 *) array of buffer indices
173 @param b - (void **) array to store buffer pointers
174 @param count - (uword) number of elements
175 @param offset - (i32) offset applied to each pointer
176*/
177static_always_inline void
178vlib_get_buffers_with_offset (vlib_main_t * vm, u32 * bi, void **b, int count,
179 i32 offset)
180{
Damjan Mariond50e3472019-01-20 00:03:56 +0100181 uword buffer_mem_start = vm->buffer_main->buffer_mem_start;
Damjan Marionafe56de2018-05-17 12:44:00 +0200182#ifdef CLIB_HAVE_VEC256
Damjan Mariond50e3472019-01-20 00:03:56 +0100183 u64x4 off = u64x4_splat (buffer_mem_start + offset);
Damjan Marionafe56de2018-05-17 12:44:00 +0200184 /* if count is not const, compiler will not unroll while loop
185 se we maintain two-in-parallel variant */
186 while (count >= 8)
187 {
188 u64x4 b0 = u32x4_extend_to_u64x4 (u32x4_load_unaligned (bi));
189 u64x4 b1 = u32x4_extend_to_u64x4 (u32x4_load_unaligned (bi + 4));
190 /* shift and add to get vlib_buffer_t pointer */
191 u64x4_store_unaligned ((b0 << CLIB_LOG2_CACHE_LINE_BYTES) + off, b);
192 u64x4_store_unaligned ((b1 << CLIB_LOG2_CACHE_LINE_BYTES) + off, b + 4);
193 b += 8;
194 bi += 8;
195 count -= 8;
196 }
197#endif
198 while (count >= 4)
199 {
200#ifdef CLIB_HAVE_VEC256
201 u64x4 b0 = u32x4_extend_to_u64x4 (u32x4_load_unaligned (bi));
202 /* shift and add to get vlib_buffer_t pointer */
203 u64x4_store_unaligned ((b0 << CLIB_LOG2_CACHE_LINE_BYTES) + off, b);
Sirshak Das759226e2018-08-22 08:46:52 +0800204#elif defined (CLIB_HAVE_VEC128)
Damjan Mariond50e3472019-01-20 00:03:56 +0100205 u64x2 off = u64x2_splat (buffer_mem_start + offset);
Damjan Marion5df580e2018-07-27 01:47:57 +0200206 u32x4 bi4 = u32x4_load_unaligned (bi);
207 u64x2 b0 = u32x4_extend_to_u64x2 ((u32x4) bi4);
Sirshak Das759226e2018-08-22 08:46:52 +0800208#if defined (__aarch64__)
209 u64x2 b1 = u32x4_extend_to_u64x2_high ((u32x4) bi4);
210#else
Damjan Marion5df580e2018-07-27 01:47:57 +0200211 bi4 = u32x4_shuffle (bi4, 2, 3, 0, 1);
212 u64x2 b1 = u32x4_extend_to_u64x2 ((u32x4) bi4);
Sirshak Das759226e2018-08-22 08:46:52 +0800213#endif
Damjan Marion5df580e2018-07-27 01:47:57 +0200214 u64x2_store_unaligned ((b0 << CLIB_LOG2_CACHE_LINE_BYTES) + off, b);
215 u64x2_store_unaligned ((b1 << CLIB_LOG2_CACHE_LINE_BYTES) + off, b + 2);
Damjan Marionafe56de2018-05-17 12:44:00 +0200216#else
Damjan Marion24dcbe42019-01-31 12:29:39 +0100217 b[0] = vlib_buffer_ptr_from_index (buffer_mem_start, bi[0], offset);
218 b[1] = vlib_buffer_ptr_from_index (buffer_mem_start, bi[1], offset);
219 b[2] = vlib_buffer_ptr_from_index (buffer_mem_start, bi[2], offset);
220 b[3] = vlib_buffer_ptr_from_index (buffer_mem_start, bi[3], offset);
Damjan Marionafe56de2018-05-17 12:44:00 +0200221#endif
222 b += 4;
223 bi += 4;
224 count -= 4;
225 }
226 while (count)
227 {
Damjan Marion24dcbe42019-01-31 12:29:39 +0100228 b[0] = vlib_buffer_ptr_from_index (buffer_mem_start, bi[0], offset);
Damjan Marionafe56de2018-05-17 12:44:00 +0200229 b += 1;
230 bi += 1;
231 count -= 1;
232 }
233}
234
235/** \brief Translate array of buffer indices into buffer pointers
236
237 @param vm - (vlib_main_t *) vlib main data structure pointer
238 @param bi - (u32 *) array of buffer indices
239 @param b - (vlib_buffer_t **) array to store buffer pointers
240 @param count - (uword) number of elements
241*/
242
243static_always_inline void
244vlib_get_buffers (vlib_main_t * vm, u32 * bi, vlib_buffer_t ** b, int count)
245{
246 vlib_get_buffers_with_offset (vm, bi, (void **) b, count, 0);
247}
248
Ed Warnickecb9cada2015-12-08 15:45:58 -0700249/** \brief Translate buffer pointer into buffer index
250
251 @param vm - (vlib_main_t *) vlib main data structure pointer
Chris Luked4024f52016-09-06 09:32:36 -0400252 @param p - (void *) buffer pointer
Ed Warnickecb9cada2015-12-08 15:45:58 -0700253 @return - (u32) buffer index
Dave Barach9b8ffd92016-07-08 08:13:45 -0400254*/
Damjan Marion04a7f052017-07-10 15:06:17 +0200255
Ed Warnickecb9cada2015-12-08 15:45:58 -0700256always_inline u32
Dave Barach9b8ffd92016-07-08 08:13:45 -0400257vlib_get_buffer_index (vlib_main_t * vm, void *p)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700258{
Damjan Mariond50e3472019-01-20 00:03:56 +0100259 vlib_buffer_main_t *bm = vm->buffer_main;
Damjan Marion04a7f052017-07-10 15:06:17 +0200260 uword offset = pointer_to_uword (p) - bm->buffer_mem_start;
261 ASSERT (pointer_to_uword (p) >= bm->buffer_mem_start);
262 ASSERT (offset < bm->buffer_mem_size);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400263 ASSERT ((offset % (1 << CLIB_LOG2_CACHE_LINE_BYTES)) == 0);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700264 return offset >> CLIB_LOG2_CACHE_LINE_BYTES;
265}
266
Damjan Marionafe56de2018-05-17 12:44:00 +0200267/** \brief Translate array of buffer pointers into buffer indices with offset
268
269 @param vm - (vlib_main_t *) vlib main data structure pointer
270 @param b - (void **) array of buffer pointers
271 @param bi - (u32 *) array to store buffer indices
272 @param count - (uword) number of elements
273 @param offset - (i32) offset applied to each pointer
274*/
275static_always_inline void
276vlib_get_buffer_indices_with_offset (vlib_main_t * vm, void **b, u32 * bi,
277 uword count, i32 offset)
278{
279#ifdef CLIB_HAVE_VEC256
280 u32x8 mask = { 0, 2, 4, 6, 1, 3, 5, 7 };
Damjan Mariond50e3472019-01-20 00:03:56 +0100281 u64x4 off4 = u64x4_splat (vm->buffer_main->buffer_mem_start - offset);
Damjan Marionafe56de2018-05-17 12:44:00 +0200282
283 while (count >= 8)
284 {
285 /* load 4 pointers into 256-bit register */
286 u64x4 v0 = u64x4_load_unaligned (b);
287 u64x4 v1 = u64x4_load_unaligned (b + 4);
288 u32x8 v2, v3;
289
290 v0 -= off4;
291 v1 -= off4;
292
293 v0 >>= CLIB_LOG2_CACHE_LINE_BYTES;
294 v1 >>= CLIB_LOG2_CACHE_LINE_BYTES;
295
296 /* permute 256-bit register so lower u32s of each buffer index are
297 * placed into lower 128-bits */
298 v2 = u32x8_permute ((u32x8) v0, mask);
299 v3 = u32x8_permute ((u32x8) v1, mask);
300
301 /* extract lower 128-bits and save them to the array of buffer indices */
302 u32x4_store_unaligned (u32x8_extract_lo (v2), bi);
303 u32x4_store_unaligned (u32x8_extract_lo (v3), bi + 4);
304 bi += 8;
305 b += 8;
306 count -= 8;
307 }
308#endif
309 while (count >= 4)
310 {
311 /* equivalent non-nector implementation */
312 bi[0] = vlib_get_buffer_index (vm, ((u8 *) b[0]) + offset);
313 bi[1] = vlib_get_buffer_index (vm, ((u8 *) b[1]) + offset);
314 bi[2] = vlib_get_buffer_index (vm, ((u8 *) b[2]) + offset);
315 bi[3] = vlib_get_buffer_index (vm, ((u8 *) b[3]) + offset);
316 bi += 4;
317 b += 4;
318 count -= 4;
319 }
320 while (count)
321 {
Zhiyong Yang462072a2018-05-30 22:12:57 -0400322 bi[0] = vlib_get_buffer_index (vm, ((u8 *) b[0]) + offset);
Damjan Marionafe56de2018-05-17 12:44:00 +0200323 bi += 1;
324 b += 1;
325 count -= 1;
326 }
327}
328
329/** \brief Translate array of buffer pointers into buffer indices
330
331 @param vm - (vlib_main_t *) vlib main data structure pointer
332 @param b - (vlib_buffer_t **) array of buffer pointers
333 @param bi - (u32 *) array to store buffer indices
334 @param count - (uword) number of elements
335*/
336static_always_inline void
337vlib_get_buffer_indices (vlib_main_t * vm, vlib_buffer_t ** b, u32 * bi,
338 uword count)
339{
340 vlib_get_buffer_indices_with_offset (vm, (void **) b, bi, count, 0);
341}
342
Ed Warnickecb9cada2015-12-08 15:45:58 -0700343/** \brief Get next buffer in buffer linklist, or zero for end of list.
344
345 @param vm - (vlib_main_t *) vlib main data structure pointer
346 @param b - (void *) buffer pointer
347 @return - (vlib_buffer_t *) next buffer, or NULL
Dave Barach9b8ffd92016-07-08 08:13:45 -0400348*/
Ed Warnickecb9cada2015-12-08 15:45:58 -0700349always_inline vlib_buffer_t *
350vlib_get_next_buffer (vlib_main_t * vm, vlib_buffer_t * b)
351{
352 return (b->flags & VLIB_BUFFER_NEXT_PRESENT
Dave Barach9b8ffd92016-07-08 08:13:45 -0400353 ? vlib_get_buffer (vm, b->next_buffer) : 0);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700354}
355
Dave Barach9b8ffd92016-07-08 08:13:45 -0400356uword vlib_buffer_length_in_chain_slow_path (vlib_main_t * vm,
357 vlib_buffer_t * b_first);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700358
359/** \brief Get length in bytes of the buffer chain
360
361 @param vm - (vlib_main_t *) vlib main data structure pointer
362 @param b - (void *) buffer pointer
363 @return - (uword) length of buffer chain
Dave Barach9b8ffd92016-07-08 08:13:45 -0400364*/
Ed Warnickecb9cada2015-12-08 15:45:58 -0700365always_inline uword
366vlib_buffer_length_in_chain (vlib_main_t * vm, vlib_buffer_t * b)
367{
Damjan Marion072401e2017-07-13 18:53:27 +0200368 uword len = b->current_length;
369
370 if (PREDICT_TRUE ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0))
371 return len;
372
373 if (PREDICT_TRUE (b->flags & VLIB_BUFFER_TOTAL_LENGTH_VALID))
374 return len + b->total_length_not_including_first_buffer;
375
376 return vlib_buffer_length_in_chain_slow_path (vm, b);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700377}
378
379/** \brief Get length in bytes of the buffer index buffer chain
380
381 @param vm - (vlib_main_t *) vlib main data structure pointer
382 @param bi - (u32) buffer index
383 @return - (uword) length of buffer chain
Dave Barach9b8ffd92016-07-08 08:13:45 -0400384*/
Ed Warnickecb9cada2015-12-08 15:45:58 -0700385always_inline uword
386vlib_buffer_index_length_in_chain (vlib_main_t * vm, u32 bi)
387{
Dave Barach9b8ffd92016-07-08 08:13:45 -0400388 vlib_buffer_t *b = vlib_get_buffer (vm, bi);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700389 return vlib_buffer_length_in_chain (vm, b);
390}
391
392/** \brief Copy buffer contents to memory
393
394 @param vm - (vlib_main_t *) vlib main data structure pointer
Chris Luked4024f52016-09-06 09:32:36 -0400395 @param buffer_index - (u32) buffer index
Ed Warnickecb9cada2015-12-08 15:45:58 -0700396 @param contents - (u8 *) memory, <strong>must be large enough</strong>
397 @return - (uword) length of buffer chain
Dave Barach9b8ffd92016-07-08 08:13:45 -0400398*/
Ed Warnickecb9cada2015-12-08 15:45:58 -0700399always_inline uword
400vlib_buffer_contents (vlib_main_t * vm, u32 buffer_index, u8 * contents)
401{
402 uword content_len = 0;
403 uword l;
Dave Barach9b8ffd92016-07-08 08:13:45 -0400404 vlib_buffer_t *b;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700405
406 while (1)
407 {
408 b = vlib_get_buffer (vm, buffer_index);
409 l = b->current_length;
Dave Barach178cf492018-11-13 16:34:13 -0500410 clib_memcpy_fast (contents + content_len, b->data + b->current_data, l);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700411 content_len += l;
Dave Barach9b8ffd92016-07-08 08:13:45 -0400412 if (!(b->flags & VLIB_BUFFER_NEXT_PRESENT))
Ed Warnickecb9cada2015-12-08 15:45:58 -0700413 break;
414 buffer_index = b->next_buffer;
415 }
416
417 return content_len;
418}
419
Damjan Marion8f499362018-10-22 13:07:02 +0200420always_inline uword
421vlib_buffer_get_pa (vlib_main_t * vm, vlib_buffer_t * b)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700422{
Damjan Marion68b4da62018-09-30 18:26:20 +0200423 return vlib_physmem_get_pa (vm, b->data);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700424}
425
Damjan Marion8f499362018-10-22 13:07:02 +0200426always_inline uword
427vlib_buffer_get_current_pa (vlib_main_t * vm, vlib_buffer_t * b)
428{
429 return vlib_buffer_get_pa (vm, b) + b->current_data;
430}
431
Ed Warnickecb9cada2015-12-08 15:45:58 -0700432/** \brief Prefetch buffer metadata by buffer index
433 The first 64 bytes of buffer contains most header information
434
435 @param vm - (vlib_main_t *) vlib main data structure pointer
436 @param bi - (u32) buffer index
437 @param type - LOAD, STORE. In most cases, STORE is the right answer
438*/
439/* Prefetch buffer header given index. */
440#define vlib_prefetch_buffer_with_index(vm,bi,type) \
441 do { \
442 vlib_buffer_t * _b = vlib_get_buffer (vm, bi); \
443 vlib_prefetch_buffer_header (_b, type); \
444 } while (0)
445
Dave Barach9b8ffd92016-07-08 08:13:45 -0400446typedef enum
447{
Ed Warnickecb9cada2015-12-08 15:45:58 -0700448 /* Index is unknown. */
449 VLIB_BUFFER_UNKNOWN,
450
451 /* Index is known and free/allocated. */
452 VLIB_BUFFER_KNOWN_FREE,
453 VLIB_BUFFER_KNOWN_ALLOCATED,
454} vlib_buffer_known_state_t;
455
Damjan Marionc8a26c62017-11-24 20:15:23 +0100456void vlib_buffer_validate_alloc_free (vlib_main_t * vm, u32 * buffers,
457 uword n_buffers,
458 vlib_buffer_known_state_t
459 expected_state);
460
Ed Warnickecb9cada2015-12-08 15:45:58 -0700461always_inline vlib_buffer_known_state_t
Damjan Mariond50e3472019-01-20 00:03:56 +0100462vlib_buffer_is_known (vlib_main_t * vm, u32 buffer_index)
Ed Warnickecb9cada2015-12-08 15:45:58 -0700463{
Damjan Mariond50e3472019-01-20 00:03:56 +0100464 vlib_buffer_main_t *bm = vm->buffer_main;
Ed Warnickecb9cada2015-12-08 15:45:58 -0700465
Damjan Marion6b0f5892017-07-27 04:01:24 -0400466 clib_spinlock_lock (&bm->buffer_known_hash_lockp);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400467 uword *p = hash_get (bm->buffer_known_hash, buffer_index);
Damjan Marion6b0f5892017-07-27 04:01:24 -0400468 clib_spinlock_unlock (&bm->buffer_known_hash_lockp);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700469 return p ? p[0] : VLIB_BUFFER_UNKNOWN;
470}
471
Ed Warnickecb9cada2015-12-08 15:45:58 -0700472/* Validates sanity of a single buffer.
473 Returns format'ed vector with error message if any. */
Dave Barach9b8ffd92016-07-08 08:13:45 -0400474u8 *vlib_validate_buffer (vlib_main_t * vm, u32 buffer_index,
475 uword follow_chain);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700476
Damjan Marion910d3692019-01-21 11:48:34 +0100477static_always_inline vlib_buffer_pool_t *
478vlib_get_buffer_pool (vlib_main_t * vm, u8 buffer_pool_index)
Dave Barach9b8ffd92016-07-08 08:13:45 -0400479{
Damjan Marion910d3692019-01-21 11:48:34 +0100480 vlib_buffer_main_t *bm = vm->buffer_main;
481 return vec_elt_at_index (bm->buffer_pools, buffer_pool_index);
Dave Barach9b8ffd92016-07-08 08:13:45 -0400482}
Ed Warnickecb9cada2015-12-08 15:45:58 -0700483
Damjan Marion910d3692019-01-21 11:48:34 +0100484static_always_inline uword
485vlib_buffer_pool_get (vlib_main_t * vm, u8 buffer_pool_index, u32 * buffers,
486 u32 n_buffers)
487{
488 vlib_buffer_pool_t *bp = vlib_get_buffer_pool (vm, buffer_pool_index);
489 u32 len;
490
491 ASSERT (bp->buffers);
492
493 clib_spinlock_lock (&bp->lock);
494 len = vec_len (bp->buffers);
495 if (PREDICT_TRUE (n_buffers < len))
496 {
497 len -= n_buffers;
498 vlib_buffer_copy_indices (buffers, bp->buffers + len, n_buffers);
499 _vec_len (bp->buffers) = len;
500 clib_spinlock_unlock (&bp->lock);
501 return n_buffers;
502 }
503 else
504 {
505 vlib_buffer_copy_indices (buffers, bp->buffers, len);
506 _vec_len (bp->buffers) = 0;
507 clib_spinlock_unlock (&bp->lock);
508 return len;
509 }
510}
511
512
513/** \brief Allocate buffers from specific pool into supplied array
Ed Warnickecb9cada2015-12-08 15:45:58 -0700514
515 @param vm - (vlib_main_t *) vlib main data structure pointer
516 @param buffers - (u32 * ) buffer index array
517 @param n_buffers - (u32) number of buffers requested
Dave Barach9b8ffd92016-07-08 08:13:45 -0400518 @return - (u32) number of buffers actually allocated, may be
Ed Warnickecb9cada2015-12-08 15:45:58 -0700519 less than the number requested or zero
520*/
Damjan Marion910d3692019-01-21 11:48:34 +0100521
Damjan Marion878c6092017-01-04 13:19:27 +0100522always_inline u32
Damjan Marion910d3692019-01-21 11:48:34 +0100523vlib_buffer_alloc_from_pool (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
524 u8 buffer_pool_index)
Damjan Marion878c6092017-01-04 13:19:27 +0100525{
Damjan Mariond50e3472019-01-20 00:03:56 +0100526 vlib_buffer_main_t *bm = vm->buffer_main;
Damjan Marion910d3692019-01-21 11:48:34 +0100527 vlib_buffer_pool_t *bp;
528 vlib_buffer_pool_thread_t *bpt;
529 u32 *src, *dst, len, n_left;
Damjan Marion878c6092017-01-04 13:19:27 +0100530
Damjan Marion910d3692019-01-21 11:48:34 +0100531 bp = vec_elt_at_index (bm->buffer_pools, buffer_pool_index);
532 bpt = vec_elt_at_index (bp->threads, vm->thread_index);
Damjan Marion878c6092017-01-04 13:19:27 +0100533
Damjan Marion910d3692019-01-21 11:48:34 +0100534 dst = buffers;
535 n_left = n_buffers;
536 len = vec_len (bpt->cached_buffers);
Damjan Marionc8a26c62017-11-24 20:15:23 +0100537
Damjan Marion910d3692019-01-21 11:48:34 +0100538 /* per-thread cache contains enough buffers */
539 if (len >= n_buffers)
Damjan Marionc8a26c62017-11-24 20:15:23 +0100540 {
Damjan Marion910d3692019-01-21 11:48:34 +0100541 src = bpt->cached_buffers + len - n_buffers;
542 vlib_buffer_copy_indices (dst, src, n_buffers);
543 _vec_len (bpt->cached_buffers) -= n_buffers;
Damjan Marionc8a26c62017-11-24 20:15:23 +0100544
Damjan Marion910d3692019-01-21 11:48:34 +0100545 if (CLIB_DEBUG > 0)
546 vlib_buffer_validate_alloc_free (vm, buffers, n_buffers,
547 VLIB_BUFFER_KNOWN_FREE);
Damjan Marionc8a26c62017-11-24 20:15:23 +0100548 return n_buffers;
549 }
550
Damjan Marion910d3692019-01-21 11:48:34 +0100551 /* take everything available in the cache */
552 if (len)
553 {
554 vlib_buffer_copy_indices (dst, bpt->cached_buffers, len);
555 _vec_len (bpt->cached_buffers) = 0;
556 dst += len;
557 n_left -= len;
558 }
559
560 len = round_pow2 (n_left, 32);
561 vec_validate_aligned (bpt->cached_buffers, len - 1, CLIB_CACHE_LINE_BYTES);
562 len = vlib_buffer_pool_get (vm, buffer_pool_index, bpt->cached_buffers,
563 len);
564 _vec_len (bpt->cached_buffers) = len;
565
566 if (len)
567 {
568 u32 n_copy = clib_min (len, n_left);
569 src = bpt->cached_buffers + len - n_copy;
570 vlib_buffer_copy_indices (dst, src, n_copy);
571 _vec_len (bpt->cached_buffers) -= n_copy;
572 n_left -= n_copy;
573 }
574
575 n_buffers -= n_left;
Damjan Marionc8a26c62017-11-24 20:15:23 +0100576
577 /* Verify that buffers are known free. */
Damjan Marion910d3692019-01-21 11:48:34 +0100578 if (CLIB_DEBUG > 0)
579 vlib_buffer_validate_alloc_free (vm, buffers, n_buffers,
580 VLIB_BUFFER_KNOWN_FREE);
Damjan Marionc8a26c62017-11-24 20:15:23 +0100581
582 return n_buffers;
583}
584
Damjan Marion910d3692019-01-21 11:48:34 +0100585/** \brief Allocate buffers from specific numa node into supplied array
586
587 @param vm - (vlib_main_t *) vlib main data structure pointer
588 @param buffers - (u32 * ) buffer index array
589 @param n_buffers - (u32) number of buffers requested
590 @param numa_node - (u32) numa node
591 @return - (u32) number of buffers actually allocated, may be
592 less than the number requested or zero
593*/
594always_inline u32
595vlib_buffer_alloc_on_numa (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
596 u32 numa_node)
597{
598 u8 index = vlib_buffer_pool_get_default_for_numa (vm, numa_node);
599 return vlib_buffer_alloc_from_pool (vm, buffers, n_buffers, index);
600}
601
Damjan Marionc8a26c62017-11-24 20:15:23 +0100602/** \brief Allocate buffers into supplied array
603
604 @param vm - (vlib_main_t *) vlib main data structure pointer
605 @param buffers - (u32 * ) buffer index array
606 @param n_buffers - (u32) number of buffers requested
607 @return - (u32) number of buffers actually allocated, may be
608 less than the number requested or zero
609*/
Damjan Marion910d3692019-01-21 11:48:34 +0100610
Damjan Marionc8a26c62017-11-24 20:15:23 +0100611always_inline u32
612vlib_buffer_alloc (vlib_main_t * vm, u32 * buffers, u32 n_buffers)
613{
Damjan Marion910d3692019-01-21 11:48:34 +0100614 return vlib_buffer_alloc_on_numa (vm, buffers, n_buffers, vm->numa_node);
Damjan Marion878c6092017-01-04 13:19:27 +0100615}
Ed Warnickecb9cada2015-12-08 15:45:58 -0700616
Damjan Marionc58408c2018-01-18 14:54:04 +0100617/** \brief Allocate buffers into ring
618
619 @param vm - (vlib_main_t *) vlib main data structure pointer
620 @param buffers - (u32 * ) buffer index ring
621 @param start - (u32) first slot in the ring
622 @param ring_size - (u32) ring size
623 @param n_buffers - (u32) number of buffers requested
624 @return - (u32) number of buffers actually allocated, may be
625 less than the number requested or zero
626*/
627always_inline u32
628vlib_buffer_alloc_to_ring (vlib_main_t * vm, u32 * ring, u32 start,
629 u32 ring_size, u32 n_buffers)
630{
631 u32 n_alloc;
632
633 ASSERT (n_buffers <= ring_size);
634
635 if (PREDICT_TRUE (start + n_buffers <= ring_size))
636 return vlib_buffer_alloc (vm, ring + start, n_buffers);
637
638 n_alloc = vlib_buffer_alloc (vm, ring + start, ring_size - start);
639
640 if (PREDICT_TRUE (n_alloc == ring_size - start))
641 n_alloc += vlib_buffer_alloc (vm, ring, n_buffers - n_alloc);
642
643 return n_alloc;
644}
645
Damjan Marion910d3692019-01-21 11:48:34 +0100646/** \brief Allocate buffers into ring from specific buffer pool
647
648 @param vm - (vlib_main_t *) vlib main data structure pointer
649 @param buffers - (u32 * ) buffer index ring
650 @param start - (u32) first slot in the ring
651 @param ring_size - (u32) ring size
652 @param n_buffers - (u32) number of buffers requested
653 @return - (u32) number of buffers actually allocated, may be
654 less than the number requested or zero
655*/
656always_inline u32
657vlib_buffer_alloc_to_ring_from_pool (vlib_main_t * vm, u32 * ring, u32 start,
658 u32 ring_size, u32 n_buffers,
659 u8 buffer_pool_index)
660{
661 u32 n_alloc;
662
663 ASSERT (n_buffers <= ring_size);
664
665 if (PREDICT_TRUE (start + n_buffers <= ring_size))
666 return vlib_buffer_alloc_from_pool (vm, ring + start, n_buffers,
667 buffer_pool_index);
668
669 n_alloc = vlib_buffer_alloc_from_pool (vm, ring + start, ring_size - start,
670 buffer_pool_index);
671
672 if (PREDICT_TRUE (n_alloc == ring_size - start))
673 n_alloc += vlib_buffer_alloc_from_pool (vm, ring, n_buffers - n_alloc,
674 buffer_pool_index);
675
676 return n_alloc;
677}
678
Damjan Marionf646d742019-01-31 18:50:04 +0100679static_always_inline void
Damjan Marion910d3692019-01-21 11:48:34 +0100680vlib_buffer_pool_put (vlib_main_t * vm, u8 buffer_pool_index,
681 u32 * buffers, u32 n_buffers)
682{
683 vlib_buffer_pool_t *bp = vlib_get_buffer_pool (vm, buffer_pool_index);
684 vlib_buffer_pool_thread_t *bpt =
685 vec_elt_at_index (bp->threads, vm->thread_index);
686
Damjan Marionf646d742019-01-31 18:50:04 +0100687 if (CLIB_DEBUG > 0)
688 vlib_buffer_validate_alloc_free (vm, buffers, n_buffers,
689 VLIB_BUFFER_KNOWN_ALLOCATED);
690
Damjan Marion910d3692019-01-21 11:48:34 +0100691 vec_add_aligned (bpt->cached_buffers, buffers, n_buffers,
692 CLIB_CACHE_LINE_BYTES);
693
694 if (vec_len (bpt->cached_buffers) > 4 * VLIB_FRAME_SIZE)
695 {
696 clib_spinlock_lock (&bp->lock);
697 /* keep last stored buffers, as they are more likely hot in the cache */
698 vec_add_aligned (bp->buffers, bpt->cached_buffers, VLIB_FRAME_SIZE,
699 CLIB_CACHE_LINE_BYTES);
700 vec_delete (bpt->cached_buffers, VLIB_FRAME_SIZE, 0);
701 bpt->n_alloc -= VLIB_FRAME_SIZE;
702 clib_spinlock_unlock (&bp->lock);
703 }
704}
705
706static_always_inline void
707vlib_buffer_free_inline (vlib_main_t * vm, u32 * buffers, u32 n_buffers,
708 int maybe_next)
709{
710 const int queue_size = 128;
711 vlib_buffer_pool_t *bp = 0;
712 u8 buffer_pool_index = ~0;
713 u32 n_queue = 0, queue[queue_size + 4];
714 vlib_buffer_t bt = { };
Lijian.Zhange6a47cf2019-03-12 18:32:39 +0800715#if defined(CLIB_HAVE_VEC128)
Damjan Marion910d3692019-01-21 11:48:34 +0100716 vlib_buffer_t bpi_mask = {.buffer_pool_index = ~0 };
717 vlib_buffer_t bpi_vec = {.buffer_pool_index = ~0 };
718 vlib_buffer_t flags_refs_mask = {
719 .flags = VLIB_BUFFER_NEXT_PRESENT,
Damjan Marion0e209242019-03-13 12:04:10 +0100720 .ref_count = ~1
Damjan Marion910d3692019-01-21 11:48:34 +0100721 };
722#endif
723
724 while (n_buffers)
725 {
726 vlib_buffer_t *b[8];
727 u32 bi, sum = 0, flags, next;
728
729 if (n_buffers < 12)
730 goto one_by_one;
731
732 vlib_get_buffers (vm, buffers, b, 4);
733 vlib_get_buffers (vm, buffers + 8, b + 4, 4);
734
735 vlib_prefetch_buffer_header (b[4], LOAD);
736 vlib_prefetch_buffer_header (b[5], LOAD);
737 vlib_prefetch_buffer_header (b[6], LOAD);
738 vlib_prefetch_buffer_header (b[7], LOAD);
739
Lijian.Zhange6a47cf2019-03-12 18:32:39 +0800740#if defined(CLIB_HAVE_VEC128)
Damjan Marion910d3692019-01-21 11:48:34 +0100741 u8x16 p0, p1, p2, p3, r;
742 p0 = u8x16_load_unaligned (b[0]);
743 p1 = u8x16_load_unaligned (b[1]);
744 p2 = u8x16_load_unaligned (b[2]);
745 p3 = u8x16_load_unaligned (b[3]);
746
747 r = p0 ^ bpi_vec.as_u8x16[0];
748 r |= p1 ^ bpi_vec.as_u8x16[0];
749 r |= p2 ^ bpi_vec.as_u8x16[0];
750 r |= p3 ^ bpi_vec.as_u8x16[0];
751 r &= bpi_mask.as_u8x16[0];
752 r |= (p0 | p1 | p2 | p3) & flags_refs_mask.as_u8x16[0];
753
754 sum = !u8x16_is_all_zero (r);
755#else
756 sum |= b[0]->flags;
757 sum |= b[1]->flags;
758 sum |= b[2]->flags;
759 sum |= b[3]->flags;
760 sum &= VLIB_BUFFER_NEXT_PRESENT;
761 sum += b[0]->ref_count - 1;
762 sum += b[1]->ref_count - 1;
763 sum += b[2]->ref_count - 1;
764 sum += b[3]->ref_count - 1;
765 sum |= b[0]->buffer_pool_index ^ buffer_pool_index;
766 sum |= b[1]->buffer_pool_index ^ buffer_pool_index;
767 sum |= b[2]->buffer_pool_index ^ buffer_pool_index;
768 sum |= b[3]->buffer_pool_index ^ buffer_pool_index;
769#endif
770
771 if (sum)
772 goto one_by_one;
773
774 vlib_buffer_copy_indices (queue + n_queue, buffers, 4);
775 vlib_buffer_copy_template (b[0], &bt);
776 vlib_buffer_copy_template (b[1], &bt);
777 vlib_buffer_copy_template (b[2], &bt);
778 vlib_buffer_copy_template (b[3], &bt);
779 n_queue += 4;
780
Damjan Marion24dcbe42019-01-31 12:29:39 +0100781 vlib_buffer_validate (vm, b[0]);
782 vlib_buffer_validate (vm, b[1]);
783 vlib_buffer_validate (vm, b[2]);
784 vlib_buffer_validate (vm, b[3]);
785
Damjan Marion910d3692019-01-21 11:48:34 +0100786 VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[0]);
787 VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[1]);
788 VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[2]);
789 VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[3]);
790
791 if (n_queue >= queue_size)
792 {
793 vlib_buffer_pool_put (vm, buffer_pool_index, queue, n_queue);
794 n_queue = 0;
795 }
796 buffers += 4;
797 n_buffers -= 4;
798 continue;
799
800 one_by_one:
801 bi = buffers[0];
802
803 next_in_chain:
804 b[0] = vlib_get_buffer (vm, bi);
805 flags = b[0]->flags;
806 next = b[0]->next_buffer;
807
808 if (PREDICT_FALSE (buffer_pool_index != b[0]->buffer_pool_index))
809 {
Damjan Marion910d3692019-01-21 11:48:34 +0100810
811 if (n_queue)
812 {
813 vlib_buffer_pool_put (vm, buffer_pool_index, queue, n_queue);
814 n_queue = 0;
815 }
Lollita Liu8f6c1dd2019-02-14 05:02:44 -0500816
817 buffer_pool_index = b[0]->buffer_pool_index;
Lijian.Zhange6a47cf2019-03-12 18:32:39 +0800818#if defined(CLIB_HAVE_VEC128)
Lollita Liu8f6c1dd2019-02-14 05:02:44 -0500819 bpi_vec.buffer_pool_index = buffer_pool_index;
820#endif
821 bp = vlib_get_buffer_pool (vm, buffer_pool_index);
822 vlib_buffer_copy_template (&bt, &bp->buffer_template);
Damjan Marion910d3692019-01-21 11:48:34 +0100823 }
824
Damjan Marion24dcbe42019-01-31 12:29:39 +0100825 vlib_buffer_validate (vm, b[0]);
Damjan Marion910d3692019-01-21 11:48:34 +0100826
827 VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b[0]);
828
829 if (clib_atomic_sub_fetch (&b[0]->ref_count, 1) == 0)
830 {
Damjan Marion910d3692019-01-21 11:48:34 +0100831 vlib_buffer_copy_template (b[0], &bt);
832 queue[n_queue++] = bi;
833 }
834
835 if (n_queue == queue_size)
836 {
837 vlib_buffer_pool_put (vm, buffer_pool_index, queue, queue_size);
838 n_queue = 0;
839 }
840
841 if (flags & VLIB_BUFFER_NEXT_PRESENT)
842 {
843 bi = next;
844 goto next_in_chain;
845 }
846
847 buffers++;
848 n_buffers--;
849 }
850
851 if (n_queue)
852 vlib_buffer_pool_put (vm, buffer_pool_index, queue, n_queue);
853}
854
855
Ed Warnickecb9cada2015-12-08 15:45:58 -0700856/** \brief Free buffers
857 Frees the entire buffer chain for each buffer
858
859 @param vm - (vlib_main_t *) vlib main data structure pointer
860 @param buffers - (u32 * ) buffer index array
861 @param n_buffers - (u32) number of buffers to free
862
863*/
Damjan Marion878c6092017-01-04 13:19:27 +0100864always_inline void
865vlib_buffer_free (vlib_main_t * vm,
866 /* pointer to first buffer */
867 u32 * buffers,
868 /* number of buffers to free */
869 u32 n_buffers)
870{
Damjan Marion910d3692019-01-21 11:48:34 +0100871 vlib_buffer_free_inline (vm, buffers, n_buffers, /* maybe next */ 1);
Damjan Marion878c6092017-01-04 13:19:27 +0100872}
Ed Warnickecb9cada2015-12-08 15:45:58 -0700873
874/** \brief Free buffers, does not free the buffer chain for each buffer
875
876 @param vm - (vlib_main_t *) vlib main data structure pointer
877 @param buffers - (u32 * ) buffer index array
878 @param n_buffers - (u32) number of buffers to free
879
880*/
Damjan Marion878c6092017-01-04 13:19:27 +0100881always_inline void
882vlib_buffer_free_no_next (vlib_main_t * vm,
883 /* pointer to first buffer */
884 u32 * buffers,
885 /* number of buffers to free */
886 u32 n_buffers)
887{
Damjan Marion910d3692019-01-21 11:48:34 +0100888 vlib_buffer_free_inline (vm, buffers, n_buffers, /* maybe next */ 0);
Damjan Marion878c6092017-01-04 13:19:27 +0100889}
Ed Warnickecb9cada2015-12-08 15:45:58 -0700890
891/** \brief Free one buffer
Dave Barach9b8ffd92016-07-08 08:13:45 -0400892 Shorthand to free a single buffer chain.
Ed Warnickecb9cada2015-12-08 15:45:58 -0700893
894 @param vm - (vlib_main_t *) vlib main data structure pointer
895 @param buffer_index - (u32) buffer index to free
896*/
897always_inline void
898vlib_buffer_free_one (vlib_main_t * vm, u32 buffer_index)
899{
Damjan Marion910d3692019-01-21 11:48:34 +0100900 vlib_buffer_free_inline (vm, &buffer_index, 1, /* maybe next */ 1);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700901}
902
Damjan Mariona3731492018-02-25 22:50:39 +0100903/** \brief Free buffers from ring
904
905 @param vm - (vlib_main_t *) vlib main data structure pointer
906 @param buffers - (u32 * ) buffer index ring
907 @param start - (u32) first slot in the ring
908 @param ring_size - (u32) ring size
909 @param n_buffers - (u32) number of buffers
910*/
911always_inline void
912vlib_buffer_free_from_ring (vlib_main_t * vm, u32 * ring, u32 start,
913 u32 ring_size, u32 n_buffers)
914{
915 ASSERT (n_buffers <= ring_size);
916
917 if (PREDICT_TRUE (start + n_buffers <= ring_size))
918 {
919 vlib_buffer_free (vm, ring + start, n_buffers);
920 }
921 else
922 {
923 vlib_buffer_free (vm, ring + start, ring_size - start);
924 vlib_buffer_free (vm, ring, n_buffers - (ring_size - start));
925 }
926}
927
Damjan Marioncef1db92018-03-28 18:27:38 +0200928/** \brief Free buffers from ring without freeing tail buffers
929
930 @param vm - (vlib_main_t *) vlib main data structure pointer
931 @param buffers - (u32 * ) buffer index ring
932 @param start - (u32) first slot in the ring
933 @param ring_size - (u32) ring size
934 @param n_buffers - (u32) number of buffers
935*/
936always_inline void
937vlib_buffer_free_from_ring_no_next (vlib_main_t * vm, u32 * ring, u32 start,
938 u32 ring_size, u32 n_buffers)
939{
940 ASSERT (n_buffers <= ring_size);
941
942 if (PREDICT_TRUE (start + n_buffers <= ring_size))
943 {
Damjan Marion4a973932018-06-09 19:29:16 +0200944 vlib_buffer_free_no_next (vm, ring + start, n_buffers);
Damjan Marioncef1db92018-03-28 18:27:38 +0200945 }
946 else
947 {
948 vlib_buffer_free_no_next (vm, ring + start, ring_size - start);
949 vlib_buffer_free_no_next (vm, ring, n_buffers - (ring_size - start));
950 }
951}
Damjan Mariona3731492018-02-25 22:50:39 +0100952
Ed Warnickecb9cada2015-12-08 15:45:58 -0700953/* Append given data to end of buffer, possibly allocating new buffers. */
Damjan Marionab9b7ec2019-01-18 20:24:44 +0100954int vlib_buffer_add_data (vlib_main_t * vm, u32 * buffer_index, void *data,
955 u32 n_data_bytes);
Ed Warnickecb9cada2015-12-08 15:45:58 -0700956
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100957/* duplicate all buffers in chain */
958always_inline vlib_buffer_t *
959vlib_buffer_copy (vlib_main_t * vm, vlib_buffer_t * b)
960{
961 vlib_buffer_t *s, *d, *fd;
962 uword n_alloc, n_buffers = 1;
Damjan Marion67655492016-11-15 12:50:28 +0100963 u32 flag_mask = VLIB_BUFFER_NEXT_PRESENT | VLIB_BUFFER_TOTAL_LENGTH_VALID;
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100964 int i;
965
966 s = b;
967 while (s->flags & VLIB_BUFFER_NEXT_PRESENT)
968 {
969 n_buffers++;
970 s = vlib_get_buffer (vm, s->next_buffer);
971 }
Neale Ranns9d676af2017-03-15 01:28:31 -0700972 u32 new_buffers[n_buffers];
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100973
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100974 n_alloc = vlib_buffer_alloc (vm, new_buffers, n_buffers);
Dave Barach26cd8c12017-02-23 17:11:26 -0500975
976 /* No guarantee that we'll get all the buffers we asked for */
977 if (PREDICT_FALSE (n_alloc < n_buffers))
978 {
979 if (n_alloc > 0)
980 vlib_buffer_free (vm, new_buffers, n_alloc);
Dave Barach26cd8c12017-02-23 17:11:26 -0500981 return 0;
982 }
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100983
984 /* 1st segment */
985 s = b;
986 fd = d = vlib_get_buffer (vm, new_buffers[0]);
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100987 d->current_data = s->current_data;
988 d->current_length = s->current_length;
Damjan Marion67655492016-11-15 12:50:28 +0100989 d->flags = s->flags & flag_mask;
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100990 d->total_length_not_including_first_buffer =
991 s->total_length_not_including_first_buffer;
Dave Barach178cf492018-11-13 16:34:13 -0500992 clib_memcpy_fast (d->opaque, s->opaque, sizeof (s->opaque));
993 clib_memcpy_fast (d->opaque2, s->opaque2, sizeof (s->opaque2));
994 clib_memcpy_fast (vlib_buffer_get_current (d),
995 vlib_buffer_get_current (s), s->current_length);
Damjan Marion05ab8cb2016-11-03 20:16:04 +0100996
997 /* next segments */
998 for (i = 1; i < n_buffers; i++)
999 {
1000 /* previous */
1001 d->next_buffer = new_buffers[i];
1002 /* current */
1003 s = vlib_get_buffer (vm, s->next_buffer);
1004 d = vlib_get_buffer (vm, new_buffers[i]);
1005 d->current_data = s->current_data;
1006 d->current_length = s->current_length;
Dave Barach178cf492018-11-13 16:34:13 -05001007 clib_memcpy_fast (vlib_buffer_get_current (d),
1008 vlib_buffer_get_current (s), s->current_length);
Damjan Marion67655492016-11-15 12:50:28 +01001009 d->flags = s->flags & flag_mask;
Damjan Marion05ab8cb2016-11-03 20:16:04 +01001010 }
1011
1012 return fd;
1013}
1014
Ole Troanda7f7b62019-03-11 13:15:54 +01001015/* duplicate first buffer in chain */
1016always_inline vlib_buffer_t *
1017vlib_buffer_copy_no_chain (vlib_main_t * vm, vlib_buffer_t * b, u32 * di)
1018{
1019 vlib_buffer_t *d;
1020
1021 if ((vlib_buffer_alloc (vm, di, 1)) != 1)
1022 return 0;
1023
1024 d = vlib_get_buffer (vm, *di);
1025 /* 1st segment */
1026 d->current_data = b->current_data;
1027 d->current_length = b->current_length;
1028 clib_memcpy_fast (d->opaque, b->opaque, sizeof (b->opaque));
1029 clib_memcpy_fast (d->opaque2, b->opaque2, sizeof (b->opaque2));
1030 clib_memcpy_fast (vlib_buffer_get_current (d),
1031 vlib_buffer_get_current (b), b->current_length);
1032
1033 return d;
1034}
1035
John Lof545caa2019-04-01 11:30:07 -04001036/* \brief Move packet from current position to offset position in buffer.
1037 Only work for small packet using one buffer with room to fit the move
1038 @param vm - (vlib_main_t *) vlib main data structure pointer
1039 @param b - (vlib_buffer_t *) pointer to buffer
1040 @param offset - (i16) position to move the packet in buffer
1041 */
1042always_inline void
1043vlib_buffer_move (vlib_main_t * vm, vlib_buffer_t * b, i16 offset)
1044{
1045 ASSERT ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0);
1046 ASSERT (offset + VLIB_BUFFER_PRE_DATA_SIZE >= 0);
1047 ASSERT (offset + b->current_length <
1048 vlib_buffer_get_default_data_size (vm));
1049
1050 u8 *source = vlib_buffer_get_current (b);
1051 b->current_data = offset;
1052 u8 *destination = vlib_buffer_get_current (b);
1053 u16 length = b->current_length;
1054
1055 if (source + length <= destination) /* no overlap */
1056 clib_memcpy_fast (destination, source, length);
1057 else
1058 memmove (destination, source, length);
1059}
1060
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001061/** \brief Create a maximum of 256 clones of buffer and store them
1062 in the supplied array
Damjan Marionc47ed032017-01-25 14:18:03 +01001063
1064 @param vm - (vlib_main_t *) vlib main data structure pointer
1065 @param src_buffer - (u32) source buffer index
1066 @param buffers - (u32 * ) buffer index array
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001067 @param n_buffers - (u16) number of buffer clones requested (<=256)
Damjan Marionc47ed032017-01-25 14:18:03 +01001068 @param head_end_offset - (u16) offset relative to current position
1069 where packet head ends
John Lof545caa2019-04-01 11:30:07 -04001070 @param offset - (i16) copy packet head at current position if 0,
1071 else at offset position to change headroom space as specified
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001072 @return - (u16) number of buffers actually cloned, may be
Damjan Marionc47ed032017-01-25 14:18:03 +01001073 less than the number requested or zero
1074*/
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001075always_inline u16
1076vlib_buffer_clone_256 (vlib_main_t * vm, u32 src_buffer, u32 * buffers,
John Lof545caa2019-04-01 11:30:07 -04001077 u16 n_buffers, u16 head_end_offset, i16 offset)
Damjan Marionc47ed032017-01-25 14:18:03 +01001078{
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001079 u16 i;
Damjan Marionc47ed032017-01-25 14:18:03 +01001080 vlib_buffer_t *s = vlib_get_buffer (vm, src_buffer);
1081
Damjan Marion910d3692019-01-21 11:48:34 +01001082 ASSERT (s->ref_count == 1);
Damjan Marionc47ed032017-01-25 14:18:03 +01001083 ASSERT (n_buffers);
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001084 ASSERT (n_buffers <= 256);
John Lof545caa2019-04-01 11:30:07 -04001085 ASSERT (offset + VLIB_BUFFER_PRE_DATA_SIZE >= 0);
1086 ASSERT ((offset + head_end_offset) <
1087 vlib_buffer_get_default_data_size (vm));
Damjan Marionc47ed032017-01-25 14:18:03 +01001088
1089 if (s->current_length <= head_end_offset + CLIB_CACHE_LINE_BYTES * 2)
1090 {
1091 buffers[0] = src_buffer;
John Lof545caa2019-04-01 11:30:07 -04001092 if (offset)
1093 vlib_buffer_move (vm, s, offset);
1094
Damjan Marionc47ed032017-01-25 14:18:03 +01001095 for (i = 1; i < n_buffers; i++)
1096 {
1097 vlib_buffer_t *d;
1098 d = vlib_buffer_copy (vm, s);
1099 if (d == 0)
1100 return i;
1101 buffers[i] = vlib_get_buffer_index (vm, d);
1102
1103 }
1104 return n_buffers;
1105 }
1106
John Lof545caa2019-04-01 11:30:07 -04001107 if (PREDICT_FALSE ((n_buffers == 1) && (offset == 0)))
Damjan Marionc47ed032017-01-25 14:18:03 +01001108 {
1109 buffers[0] = src_buffer;
1110 return 1;
1111 }
1112
Damjan Marion910d3692019-01-21 11:48:34 +01001113 n_buffers = vlib_buffer_alloc_from_pool (vm, buffers, n_buffers,
1114 s->buffer_pool_index);
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001115
Damjan Marionc47ed032017-01-25 14:18:03 +01001116 for (i = 0; i < n_buffers; i++)
1117 {
1118 vlib_buffer_t *d = vlib_get_buffer (vm, buffers[i]);
John Lof545caa2019-04-01 11:30:07 -04001119 if (offset)
1120 d->current_data = offset;
1121 else
1122 d->current_data = s->current_data;
1123
Damjan Marionc47ed032017-01-25 14:18:03 +01001124 d->current_length = head_end_offset;
Damjan Marion910d3692019-01-21 11:48:34 +01001125 ASSERT (d->buffer_pool_index == s->buffer_pool_index);
1126
Yoann Desmouceaux1977a342018-05-29 13:38:44 +02001127 d->total_length_not_including_first_buffer = s->current_length -
Damjan Marionc47ed032017-01-25 14:18:03 +01001128 head_end_offset;
Yoann Desmouceaux1977a342018-05-29 13:38:44 +02001129 if (PREDICT_FALSE (s->flags & VLIB_BUFFER_NEXT_PRESENT))
1130 {
1131 d->total_length_not_including_first_buffer +=
1132 s->total_length_not_including_first_buffer;
1133 }
Damjan Marionc47ed032017-01-25 14:18:03 +01001134 d->flags = s->flags | VLIB_BUFFER_NEXT_PRESENT;
1135 d->flags &= ~VLIB_BUFFER_EXT_HDR_VALID;
Dave Barach178cf492018-11-13 16:34:13 -05001136 clib_memcpy_fast (d->opaque, s->opaque, sizeof (s->opaque));
1137 clib_memcpy_fast (d->opaque2, s->opaque2, sizeof (s->opaque2));
1138 clib_memcpy_fast (vlib_buffer_get_current (d),
1139 vlib_buffer_get_current (s), head_end_offset);
Damjan Marionc47ed032017-01-25 14:18:03 +01001140 d->next_buffer = src_buffer;
1141 }
1142 vlib_buffer_advance (s, head_end_offset);
Damjan Marion910d3692019-01-21 11:48:34 +01001143 s->ref_count = n_buffers;
Damjan Marionc47ed032017-01-25 14:18:03 +01001144 while (s->flags & VLIB_BUFFER_NEXT_PRESENT)
1145 {
1146 s = vlib_get_buffer (vm, s->next_buffer);
Damjan Marion910d3692019-01-21 11:48:34 +01001147 s->ref_count = n_buffers;
Damjan Marionc47ed032017-01-25 14:18:03 +01001148 }
1149
1150 return n_buffers;
1151}
1152
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001153/** \brief Create multiple clones of buffer and store them
1154 in the supplied array
1155
1156 @param vm - (vlib_main_t *) vlib main data structure pointer
1157 @param src_buffer - (u32) source buffer index
1158 @param buffers - (u32 * ) buffer index array
1159 @param n_buffers - (u16) number of buffer clones requested (<=256)
1160 @param head_end_offset - (u16) offset relative to current position
1161 where packet head ends
John Lof545caa2019-04-01 11:30:07 -04001162 @param offset - (i16) copy packet head at current position if 0,
1163 else at offset position to change headroom space as specified
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001164 @return - (u16) number of buffers actually cloned, may be
1165 less than the number requested or zero
1166*/
1167always_inline u16
John Lof545caa2019-04-01 11:30:07 -04001168vlib_buffer_clone_at_offset (vlib_main_t * vm, u32 src_buffer, u32 * buffers,
1169 u16 n_buffers, u16 head_end_offset, i16 offset)
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001170{
1171 vlib_buffer_t *s = vlib_get_buffer (vm, src_buffer);
1172 u16 n_cloned = 0;
1173
1174 while (n_buffers > 256)
1175 {
1176 vlib_buffer_t *copy;
1177 copy = vlib_buffer_copy (vm, s);
1178 n_cloned += vlib_buffer_clone_256 (vm,
1179 vlib_get_buffer_index (vm, copy),
1180 (buffers + n_cloned),
John Lof545caa2019-04-01 11:30:07 -04001181 256, head_end_offset, offset);
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001182 n_buffers -= 256;
1183 }
1184 n_cloned += vlib_buffer_clone_256 (vm, src_buffer,
1185 buffers + n_cloned,
John Lof545caa2019-04-01 11:30:07 -04001186 n_buffers, head_end_offset, offset);
Neale Ranns8f36e4a2018-01-11 09:02:01 -08001187
1188 return n_cloned;
1189}
1190
John Lof545caa2019-04-01 11:30:07 -04001191/** \brief Create multiple clones of buffer and store them
1192 in the supplied array
1193
1194 @param vm - (vlib_main_t *) vlib main data structure pointer
1195 @param src_buffer - (u32) source buffer index
1196 @param buffers - (u32 * ) buffer index array
1197 @param n_buffers - (u16) number of buffer clones requested (<=256)
1198 @param head_end_offset - (u16) offset relative to current position
1199 where packet head ends
1200 @return - (u16) number of buffers actually cloned, may be
1201 less than the number requested or zero
1202*/
1203always_inline u16
1204vlib_buffer_clone (vlib_main_t * vm, u32 src_buffer, u32 * buffers,
1205 u16 n_buffers, u16 head_end_offset)
1206{
1207 return vlib_buffer_clone_at_offset (vm, src_buffer, buffers, n_buffers,
1208 head_end_offset, 0);
1209}
1210
Damjan Marionc47ed032017-01-25 14:18:03 +01001211/** \brief Attach cloned tail to the buffer
1212
1213 @param vm - (vlib_main_t *) vlib main data structure pointer
1214 @param head - (vlib_buffer_t *) head buffer
1215 @param tail - (Vlib buffer_t *) tail buffer to clone and attach to head
1216*/
1217
1218always_inline void
1219vlib_buffer_attach_clone (vlib_main_t * vm, vlib_buffer_t * head,
1220 vlib_buffer_t * tail)
1221{
1222 ASSERT ((head->flags & VLIB_BUFFER_NEXT_PRESENT) == 0);
Damjan Marion910d3692019-01-21 11:48:34 +01001223 ASSERT (head->buffer_pool_index == tail->buffer_pool_index);
Damjan Marionc47ed032017-01-25 14:18:03 +01001224
1225 head->flags |= VLIB_BUFFER_NEXT_PRESENT;
1226 head->flags &= ~VLIB_BUFFER_TOTAL_LENGTH_VALID;
1227 head->flags &= ~VLIB_BUFFER_EXT_HDR_VALID;
1228 head->flags |= (tail->flags & VLIB_BUFFER_TOTAL_LENGTH_VALID);
1229 head->next_buffer = vlib_get_buffer_index (vm, tail);
1230 head->total_length_not_including_first_buffer = tail->current_length +
1231 tail->total_length_not_including_first_buffer;
1232
1233next_segment:
Damjan Marion910d3692019-01-21 11:48:34 +01001234 clib_atomic_add_fetch (&tail->ref_count, 1);
Damjan Marionc47ed032017-01-25 14:18:03 +01001235
1236 if (tail->flags & VLIB_BUFFER_NEXT_PRESENT)
1237 {
1238 tail = vlib_get_buffer (vm, tail->next_buffer);
1239 goto next_segment;
1240 }
1241}
1242
Pierre Pfister328e99b2016-02-12 13:18:42 +00001243/* Initializes the buffer as an empty packet with no chained buffers. */
1244always_inline void
Dave Barach9b8ffd92016-07-08 08:13:45 -04001245vlib_buffer_chain_init (vlib_buffer_t * first)
Pierre Pfister328e99b2016-02-12 13:18:42 +00001246{
1247 first->total_length_not_including_first_buffer = 0;
1248 first->current_length = 0;
1249 first->flags &= ~VLIB_BUFFER_NEXT_PRESENT;
1250 first->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
Pierre Pfister328e99b2016-02-12 13:18:42 +00001251}
1252
1253/* The provided next_bi buffer index is appended to the end of the packet. */
1254always_inline vlib_buffer_t *
Eyal Barib688fb12018-11-12 16:13:49 +02001255vlib_buffer_chain_buffer (vlib_main_t * vm, vlib_buffer_t * last, u32 next_bi)
Pierre Pfister328e99b2016-02-12 13:18:42 +00001256{
Dave Barach9b8ffd92016-07-08 08:13:45 -04001257 vlib_buffer_t *next_buffer = vlib_get_buffer (vm, next_bi);
Pierre Pfister328e99b2016-02-12 13:18:42 +00001258 last->next_buffer = next_bi;
1259 last->flags |= VLIB_BUFFER_NEXT_PRESENT;
1260 next_buffer->current_length = 0;
1261 next_buffer->flags &= ~VLIB_BUFFER_NEXT_PRESENT;
Pierre Pfister328e99b2016-02-12 13:18:42 +00001262 return next_buffer;
1263}
1264
1265/* Increases or decreases the packet length.
1266 * It does not allocate or deallocate new buffers.
1267 * Therefore, the added length must be compatible
1268 * with the last buffer. */
1269always_inline void
Dave Barach9b8ffd92016-07-08 08:13:45 -04001270vlib_buffer_chain_increase_length (vlib_buffer_t * first,
1271 vlib_buffer_t * last, i32 len)
Pierre Pfister328e99b2016-02-12 13:18:42 +00001272{
1273 last->current_length += len;
1274 if (first != last)
1275 first->total_length_not_including_first_buffer += len;
Pierre Pfister328e99b2016-02-12 13:18:42 +00001276}
1277
1278/* Copy data to the end of the packet and increases its length.
1279 * It does not allocate new buffers.
1280 * Returns the number of copied bytes. */
1281always_inline u16
Dave Barach9b8ffd92016-07-08 08:13:45 -04001282vlib_buffer_chain_append_data (vlib_main_t * vm,
Dave Barach9b8ffd92016-07-08 08:13:45 -04001283 vlib_buffer_t * first,
1284 vlib_buffer_t * last, void *data, u16 data_len)
Pierre Pfister328e99b2016-02-12 13:18:42 +00001285{
Damjan Marion8934a042019-02-09 23:29:26 +01001286 u32 n_buffer_bytes = vlib_buffer_get_default_data_size (vm);
Dave Barach9b8ffd92016-07-08 08:13:45 -04001287 ASSERT (n_buffer_bytes >= last->current_length + last->current_data);
1288 u16 len = clib_min (data_len,
1289 n_buffer_bytes - last->current_length -
1290 last->current_data);
Dave Barach178cf492018-11-13 16:34:13 -05001291 clib_memcpy_fast (vlib_buffer_get_current (last) + last->current_length,
1292 data, len);
Dave Barach9b8ffd92016-07-08 08:13:45 -04001293 vlib_buffer_chain_increase_length (first, last, len);
Pierre Pfister328e99b2016-02-12 13:18:42 +00001294 return len;
1295}
1296
1297/* Copy data to the end of the packet and increases its length.
1298 * Allocates additional buffers from the free list if necessary.
1299 * Returns the number of copied bytes.
1300 * 'last' value is modified whenever new buffers are allocated and
1301 * chained and points to the last buffer in the chain. */
1302u16
Dave Barach9b8ffd92016-07-08 08:13:45 -04001303vlib_buffer_chain_append_data_with_alloc (vlib_main_t * vm,
Dave Barach9b8ffd92016-07-08 08:13:45 -04001304 vlib_buffer_t * first,
Damjan Mariondac03522018-02-01 15:30:13 +01001305 vlib_buffer_t ** last, void *data,
1306 u16 data_len);
Dave Barach9b8ffd92016-07-08 08:13:45 -04001307void vlib_buffer_chain_validate (vlib_main_t * vm, vlib_buffer_t * first);
Pierre Pfister328e99b2016-02-12 13:18:42 +00001308
Dave Barach9b8ffd92016-07-08 08:13:45 -04001309format_function_t format_vlib_buffer, format_vlib_buffer_and_data,
1310 format_vlib_buffer_contents;
Ed Warnickecb9cada2015-12-08 15:45:58 -07001311
Dave Barach9b8ffd92016-07-08 08:13:45 -04001312typedef struct
1313{
Ed Warnickecb9cada2015-12-08 15:45:58 -07001314 /* Vector of packet data. */
Dave Barach9b8ffd92016-07-08 08:13:45 -04001315 u8 *packet_data;
Ed Warnickecb9cada2015-12-08 15:45:58 -07001316
Damjan Mariond1274cb2018-03-13 21:32:17 +01001317 /* Number of buffers to allocate in each call to allocator. */
1318 u32 min_n_buffers_each_alloc;
Ed Warnickecb9cada2015-12-08 15:45:58 -07001319
Damjan Marion671e60e2018-12-30 18:09:59 +01001320 u8 *name;
Ed Warnickecb9cada2015-12-08 15:45:58 -07001321} vlib_packet_template_t;
1322
Ed Warnickecb9cada2015-12-08 15:45:58 -07001323void vlib_packet_template_init (vlib_main_t * vm,
1324 vlib_packet_template_t * t,
Dave Barach9b8ffd92016-07-08 08:13:45 -04001325 void *packet_data,
Ed Warnickecb9cada2015-12-08 15:45:58 -07001326 uword n_packet_data_bytes,
Damjan Mariond1274cb2018-03-13 21:32:17 +01001327 uword min_n_buffers_each_alloc,
Dave Barach9b8ffd92016-07-08 08:13:45 -04001328 char *fmt, ...);
Ed Warnickecb9cada2015-12-08 15:45:58 -07001329
Dave Barach9b8ffd92016-07-08 08:13:45 -04001330void *vlib_packet_template_get_packet (vlib_main_t * vm,
1331 vlib_packet_template_t * t,
1332 u32 * bi_result);
Ed Warnickecb9cada2015-12-08 15:45:58 -07001333
1334always_inline void
1335vlib_packet_template_free (vlib_main_t * vm, vlib_packet_template_t * t)
1336{
1337 vec_free (t->packet_data);
1338}
1339
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001340always_inline u32
1341vlib_buffer_space_left_at_end (vlib_main_t * vm, vlib_buffer_t * b)
Klement Sekera75e7d132017-09-20 08:26:30 +02001342{
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001343 return b->data + vlib_buffer_get_default_data_size (vm) -
1344 ((u8 *) vlib_buffer_get_current (b) + b->current_length);
Klement Sekera75e7d132017-09-20 08:26:30 +02001345}
1346
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001347always_inline u32
1348vlib_buffer_chain_linearize (vlib_main_t * vm, vlib_buffer_t * b)
Eyal Barid3d42412018-11-05 13:29:25 +02001349{
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001350 vlib_buffer_t *db = b, *sb, *first = b;
1351 int is_cloned = 0;
1352 u32 bytes_left = 0, data_size;
1353 u16 src_left, dst_left, n_buffers = 1;
1354 u8 *dp, *sp;
1355 u32 to_free = 0;
Eyal Barid3d42412018-11-05 13:29:25 +02001356
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001357 if (PREDICT_TRUE ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0))
1358 return 1;
Eyal Barid3d42412018-11-05 13:29:25 +02001359
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001360 data_size = vlib_buffer_get_default_data_size (vm);
1361
1362 dst_left = vlib_buffer_space_left_at_end (vm, b);
1363
1364 while (b->flags & VLIB_BUFFER_NEXT_PRESENT)
Eyal Barid3d42412018-11-05 13:29:25 +02001365 {
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001366 b = vlib_get_buffer (vm, b->next_buffer);
1367 if (b->ref_count > 1)
1368 is_cloned = 1;
1369 bytes_left += b->current_length;
1370 n_buffers++;
Eyal Barid3d42412018-11-05 13:29:25 +02001371 }
1372
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001373 /* if buffer is cloned, create completely new chain - unless everything fits
1374 * into one buffer */
1375 if (is_cloned && bytes_left >= dst_left)
Eyal Barid3d42412018-11-05 13:29:25 +02001376 {
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001377 u32 len = 0;
1378 u32 space_needed = bytes_left - dst_left;
1379 u32 tail;
1380
1381 if (vlib_buffer_alloc (vm, &tail, 1) == 0)
1382 return 0;
1383
1384 ++n_buffers;
1385 len += data_size;
1386 b = vlib_get_buffer (vm, tail);
1387
1388 while (len < space_needed)
Eyal Barid3d42412018-11-05 13:29:25 +02001389 {
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001390 u32 bi;
1391 if (vlib_buffer_alloc (vm, &bi, 1) == 0)
1392 {
1393 vlib_buffer_free_one (vm, tail);
1394 return 0;
1395 }
1396 b->flags = VLIB_BUFFER_NEXT_PRESENT;
1397 b->next_buffer = bi;
1398 b = vlib_get_buffer (vm, bi);
1399 len += data_size;
1400 n_buffers++;
1401 }
1402 sb = vlib_get_buffer (vm, first->next_buffer);
1403 to_free = first->next_buffer;
1404 first->next_buffer = tail;
1405 }
1406 else
1407 sb = vlib_get_buffer (vm, first->next_buffer);
1408
1409 src_left = sb->current_length;
1410 sp = vlib_buffer_get_current (sb);
1411 dp = vlib_buffer_get_tail (db);
1412
1413 while (bytes_left)
1414 {
1415 u16 bytes_to_copy;
1416
1417 if (dst_left == 0)
1418 {
1419 if (db != first)
1420 db->current_data = 0;
1421 db->current_length = dp - (u8 *) vlib_buffer_get_current (db);
1422 ASSERT (db->flags & VLIB_BUFFER_NEXT_PRESENT);
1423 db = vlib_get_buffer (vm, db->next_buffer);
1424 dst_left = data_size;
1425 dp = db->data;
1426 }
1427
1428 while (src_left == 0)
1429 {
1430 ASSERT (sb->flags & VLIB_BUFFER_NEXT_PRESENT);
1431 sb = vlib_get_buffer (vm, sb->next_buffer);
1432 src_left = sb->current_length;
1433 sp = vlib_buffer_get_current (sb);
1434 }
1435
1436 bytes_to_copy = clib_min (dst_left, src_left);
1437
1438 if (dp != sp)
1439 {
1440 if (sb == db)
1441 bytes_to_copy = clib_min (bytes_to_copy, sp - dp);
1442
1443 clib_memcpy_fast (dp, sp, bytes_to_copy);
1444 }
1445
1446 src_left -= bytes_to_copy;
1447 dst_left -= bytes_to_copy;
1448 dp += bytes_to_copy;
1449 sp += bytes_to_copy;
1450 bytes_left -= bytes_to_copy;
1451 }
1452 if (db != first)
1453 db->current_data = 0;
1454 db->current_length = dp - (u8 *) vlib_buffer_get_current (db);
1455
1456 if (is_cloned && to_free)
1457 vlib_buffer_free_one (vm, to_free);
1458 else
1459 {
1460 if (db->flags & VLIB_BUFFER_NEXT_PRESENT)
1461 vlib_buffer_free_one (vm, db->next_buffer);
1462 db->flags &= ~VLIB_BUFFER_NEXT_PRESENT;
1463 b = first;
1464 n_buffers = 1;
1465 while (b->flags & VLIB_BUFFER_NEXT_PRESENT)
1466 {
1467 b = vlib_get_buffer (vm, b->next_buffer);
1468 ++n_buffers;
Eyal Barid3d42412018-11-05 13:29:25 +02001469 }
1470 }
Klement Sekeraf883f6a2019-02-13 11:01:32 +01001471
1472 first->flags &= ~VLIB_BUFFER_TOTAL_LENGTH_VALID;
1473
1474 return n_buffers;
Eyal Barid3d42412018-11-05 13:29:25 +02001475}
1476
Ed Warnickecb9cada2015-12-08 15:45:58 -07001477#endif /* included_vlib_buffer_funcs_h */
Dave Barach9b8ffd92016-07-08 08:13:45 -04001478
1479/*
1480 * fd.io coding-style-patch-verification: ON
1481 *
1482 * Local Variables:
1483 * eval: (c-set-style "gnu")
1484 * End:
1485 */