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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 Copyright (c) 2001, 2002, 2003 Eliot Dresselhaus
18 Permission is hereby granted, free of charge, to any person obtaining
19 a copy of this software and associated documentation files (the
20 "Software"), to deal in the Software without restriction, including
21 without limitation the rights to use, copy, modify, merge, publish,
22 distribute, sublicense, and/or sell copies of the Software, and to
23 permit persons to whom the Software is furnished to do so, subject to
24 the following conditions:
26 The above copyright notice and this permission notice shall be
27 included in all copies or substantial portions of the Software.
29 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
30 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
31 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
32 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
33 LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
34 OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
35 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
38 #include <vppinfra/vec.h>
39 #include <vppinfra/mem.h>
41 #ifndef CLIB_VECTOR_GROW_BY_ONE
42 #define CLIB_VECTOR_GROW_BY_ONE 0
45 /* Vector resize operator. Called as needed by various macros such as
46 vec_add1() when we need to allocate memory. */
48 vec_resize_allocate_memory (void *v,
49 word length_increment,
51 uword header_bytes, uword data_align,
54 vec_header_t *vh = _vec_find (v);
55 uword old_alloc_bytes, new_alloc_bytes;
59 header_bytes = vec_header_bytes (header_bytes);
61 data_bytes += header_bytes;
63 if (PREDICT_FALSE (numa_id != VEC_NUMA_UNSPECIFIED))
65 oldheap = clib_mem_get_per_cpu_heap ();
66 clib_mem_set_per_cpu_heap (clib_mem_get_per_numa_heap (numa_id));
71 new = clib_mem_alloc_aligned_at_offset (data_bytes, data_align, header_bytes, 1 /* yes, call os_out_of_memory */
73 new_alloc_bytes = clib_mem_size (new);
74 CLIB_MEM_UNPOISON (new + data_bytes, new_alloc_bytes - data_bytes);
75 clib_memset (new, 0, new_alloc_bytes);
76 CLIB_MEM_POISON (new + data_bytes, new_alloc_bytes - data_bytes);
77 v = new + header_bytes;
78 _vec_len (v) = length_increment;
79 _vec_numa (v) = numa_id;
80 if (PREDICT_FALSE (numa_id != VEC_NUMA_UNSPECIFIED))
81 clib_mem_set_per_cpu_heap (oldheap);
85 vh->len += length_increment;
86 old = v - header_bytes;
88 /* Vector header must start heap object. */
89 ASSERT (clib_mem_is_heap_object (old));
91 old_alloc_bytes = clib_mem_size (old);
94 if (data_bytes <= old_alloc_bytes)
96 CLIB_MEM_UNPOISON (v, data_bytes);
97 if (PREDICT_FALSE (numa_id != VEC_NUMA_UNSPECIFIED))
98 clib_mem_set_per_cpu_heap (oldheap);
102 #if CLIB_VECTOR_GROW_BY_ONE > 0
103 new_alloc_bytes = data_bytes;
105 new_alloc_bytes = (old_alloc_bytes * 3) / 2;
106 if (new_alloc_bytes < data_bytes)
107 new_alloc_bytes = data_bytes;
111 clib_mem_alloc_aligned_at_offset (new_alloc_bytes, data_align,
113 1 /* yes, call os_out_of_memory */ );
115 /* FIXME fail gracefully. */
118 ("vec_resize fails, length increment %d, data bytes %d, alignment %d",
119 length_increment, data_bytes, data_align);
121 CLIB_MEM_UNPOISON (old, old_alloc_bytes);
122 clib_memcpy_fast (new, old, old_alloc_bytes);
125 /* Allocator may give a bit of extra room. */
126 new_alloc_bytes = clib_mem_size (new);
129 /* Zero new memory. */
130 CLIB_MEM_UNPOISON (new + data_bytes, new_alloc_bytes - data_bytes);
131 memset (v + old_alloc_bytes, 0, new_alloc_bytes - old_alloc_bytes);
132 CLIB_MEM_POISON (new + data_bytes, new_alloc_bytes - data_bytes);
134 _vec_numa ((v + header_bytes)) = numa_id;
135 if (PREDICT_FALSE (numa_id != VEC_NUMA_UNSPECIFIED))
136 clib_mem_set_per_cpu_heap (oldheap);
138 return v + header_bytes;
142 clib_mem_is_vec_h (void *v, uword header_bytes)
144 return clib_mem_is_heap_object (vec_header (v, header_bytes));
148 vec_len_not_inline (void *v)
154 vec_free_not_inline (void *v)
166 main (int argc, char *argv[])
168 word n = atoi (argv[1]);
176 FOO *foos = vec_init (FOO, 10), *f;
178 vec_validate (foos, 100);
182 for (i = 0; i < n; i++)
185 vec_add2 (foos, f, 1);
194 vec_delete (foos, n, m);
200 vec_insert (foos, n, m);
211 * fd.io coding-style-patch-verification: ON
214 * eval: (c-set-style "gnu")