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 #include <vppinfra/format.h>
17 #include <vppinfra/dlmalloc.h>
18 #include <vppinfra/os.h>
19 #include <vppinfra/lock.h>
20 #include <vppinfra/hash.h>
21 #include <vppinfra/elf_clib.h>
22 #include <vppinfra/sanitizer.h>
24 void *clib_per_cpu_mheaps[CLIB_MAX_MHEAPS];
25 void *clib_per_numa_mheaps[CLIB_MAX_NUMAS];
29 /* Address of callers: outer first, inner last. */
32 /* Count of allocations with this traceback. */
35 /* Count of bytes allocated with this traceback. */
38 /* Offset of this item */
47 mheap_trace_t *traces;
49 /* Indices of free traces. */
52 /* Hash table mapping callers to trace index. */
53 uword *trace_by_callers;
55 /* Hash table mapping mheap offset to trace index. */
56 uword *trace_index_by_offset;
58 /* So we can easily shut off current segment trace, if any */
59 void *current_traced_mheap;
63 mheap_trace_main_t mheap_trace_main;
66 mheap_get_trace (uword offset, uword size)
68 mheap_trace_main_t *tm = &mheap_trace_main;
70 uword i, n_callers, trace_index, *p;
74 if (tm->enabled == 0 || (clib_mem_get_heap () != tm->current_traced_mheap))
77 /* Spurious Coverity warnings be gone. */
78 clib_memset (&trace, 0, sizeof (trace));
80 /* Skip our frame and mspace_get_aligned's frame */
81 n_callers = clib_backtrace (trace.callers, ARRAY_LEN (trace.callers), 2);
85 clib_spinlock_lock (&tm->lock);
87 /* Turn off tracing to avoid embarrassment... */
88 save_enabled = tm->enabled;
91 if (!tm->trace_by_callers)
92 tm->trace_by_callers =
93 hash_create_shmem (0, sizeof (trace.callers), sizeof (uword));
95 p = hash_get_mem (tm->trace_by_callers, &trace.callers);
99 t = tm->traces + trace_index;
103 i = vec_len (tm->trace_free_list);
106 trace_index = tm->trace_free_list[i - 1];
107 _vec_len (tm->trace_free_list) = i - 1;
111 mheap_trace_t *old_start = tm->traces;
112 mheap_trace_t *old_end = vec_end (tm->traces);
114 vec_add2 (tm->traces, t, 1);
116 if (tm->traces != old_start)
121 hash_foreach_pair (p, tm->trace_by_callers,
123 q = uword_to_pointer (p->key, mheap_trace_t *);
124 ASSERT (q >= old_start && q < old_end);
125 p->key = pointer_to_uword (tm->traces + (q - old_start));
129 trace_index = t - tm->traces;
132 t = tm->traces + trace_index;
134 t->n_allocations = 0;
136 hash_set_mem (tm->trace_by_callers, t->callers, trace_index);
139 t->n_allocations += 1;
141 t->offset = offset; /* keep a sample to autopsy */
142 hash_set (tm->trace_index_by_offset, offset, t - tm->traces);
143 tm->enabled = save_enabled;
144 clib_spinlock_unlock (&tm->lock);
148 mheap_put_trace (uword offset, uword size)
151 uword trace_index, *p;
152 mheap_trace_main_t *tm = &mheap_trace_main;
155 if (tm->enabled == 0)
158 clib_spinlock_lock (&tm->lock);
160 /* Turn off tracing for a moment */
161 save_enabled = tm->enabled;
164 p = hash_get (tm->trace_index_by_offset, offset);
167 tm->enabled = save_enabled;
168 clib_spinlock_unlock (&tm->lock);
173 hash_unset (tm->trace_index_by_offset, offset);
174 ASSERT (trace_index < vec_len (tm->traces));
176 t = tm->traces + trace_index;
177 ASSERT (t->n_allocations > 0);
178 ASSERT (t->n_bytes >= size);
179 t->n_allocations -= 1;
181 if (t->n_allocations == 0)
183 hash_unset_mem (tm->trace_by_callers, t->callers);
184 vec_add1 (tm->trace_free_list, trace_index);
185 clib_memset (t, 0, sizeof (t[0]));
187 tm->enabled = save_enabled;
188 clib_spinlock_unlock (&tm->lock);
192 mheap_trace_main_free (mheap_trace_main_t * tm)
194 vec_free (tm->traces);
195 vec_free (tm->trace_free_list);
196 hash_free (tm->trace_by_callers);
197 hash_free (tm->trace_index_by_offset);
200 /* Initialize CLIB heap based on memory/size given by user.
201 Set memory to 0 and CLIB will try to allocate its own heap. */
203 clib_mem_init_internal (void *memory, uword memory_size, int set_heap)
209 heap = create_mspace_with_base (memory, memory_size, 1 /* locked */ );
210 mspace_disable_expand (heap);
213 heap = create_mspace (memory_size, 1 /* locked */ );
215 CLIB_MEM_POISON (mspace_least_addr (heap), mspace_footprint (heap));
218 clib_mem_set_heap (heap);
220 if (mheap_trace_main.lock == 0)
221 clib_spinlock_init (&mheap_trace_main.lock);
227 clib_mem_init (void *memory, uword memory_size)
229 return clib_mem_init_internal (memory, memory_size,
230 1 /* do clib_mem_set_heap */ );
234 clib_mem_init_thread_safe (void *memory, uword memory_size)
236 return clib_mem_init_internal (memory, memory_size,
237 1 /* do clib_mem_set_heap */ );
241 clib_mem_destroy_mspace (void *mspace)
243 mheap_trace_main_t *tm = &mheap_trace_main;
245 if (tm->enabled && mspace == tm->current_traced_mheap)
248 destroy_mspace (mspace);
252 clib_mem_destroy (void)
254 clib_mem_destroy_mspace (clib_mem_get_heap ());
258 clib_mem_init_thread_safe_numa (void *memory, uword memory_size, u8 numa)
260 clib_mem_vm_alloc_t alloc = { 0 };
264 alloc.size = memory_size;
265 alloc.flags = CLIB_MEM_VM_F_NUMA_FORCE;
266 alloc.numa_node = numa;
267 if ((err = clib_mem_vm_ext_alloc (&alloc)))
269 clib_error_report (err);
273 heap = clib_mem_init_internal (memory, memory_size,
274 0 /* do NOT clib_mem_set_heap */ );
282 format_clib_mem_usage (u8 * s, va_list * va)
284 int verbose = va_arg (*va, int);
285 return format (s, "$$$$ heap at %llx verbose %d", clib_mem_get_heap (),
290 * Magic decoder ring for mallinfo stats (ala dlmalloc):
292 * size_t arena; / * Non-mmapped space allocated (bytes) * /
293 * size_t ordblks; / * Number of free chunks * /
294 * size_t smblks; / * Number of free fastbin blocks * /
295 * size_t hblks; / * Number of mmapped regions * /
296 * size_t hblkhd; / * Space allocated in mmapped regions (bytes) * /
297 * size_t usmblks; / * Maximum total allocated space (bytes) * /
298 * size_t fsmblks; / * Space in freed fastbin blocks (bytes) * /
299 * size_t uordblks; / * Total allocated space (bytes) * /
300 * size_t fordblks; / * Total free space (bytes) * /
301 * size_t keepcost; / * Top-most, releasable space (bytes) * /
306 format_msize (u8 * s, va_list * va)
308 uword a = va_arg (*va, uword);
311 s = format (s, "%.2fG", (((f64) a) / ((f64) (1ULL << 30))));
312 else if (a >= 1ULL << 20)
313 s = format (s, "%.2fM", (((f64) a) / ((f64) (1ULL << 20))));
314 else if (a >= 1ULL << 10)
315 s = format (s, "%.2fK", (((f64) a) / ((f64) (1ULL << 10))));
317 s = format (s, "%lld", a);
322 mheap_trace_sort (const void *_t1, const void *_t2)
324 const mheap_trace_t *t1 = _t1;
325 const mheap_trace_t *t2 = _t2;
328 cmp = (word) t2->n_bytes - (word) t1->n_bytes;
330 cmp = (word) t2->n_allocations - (word) t1->n_allocations;
335 format_mheap_trace (u8 * s, va_list * va)
337 mheap_trace_main_t *tm = va_arg (*va, mheap_trace_main_t *);
338 int verbose = va_arg (*va, int);
342 clib_spinlock_lock (&tm->lock);
343 if (vec_len (tm->traces) > 0 &&
344 clib_mem_get_heap () == tm->current_traced_mheap)
348 /* Make a copy of traces since we'll be sorting them. */
349 mheap_trace_t *t, *traces_copy;
350 u32 indent, total_objects_traced;
352 traces_copy = vec_dup (tm->traces);
354 qsort (traces_copy, vec_len (traces_copy), sizeof (traces_copy[0]),
357 total_objects_traced = 0;
358 s = format (s, "\n");
359 vec_foreach (t, traces_copy)
361 /* Skip over free elements. */
362 if (t->n_allocations == 0)
365 total_objects_traced += t->n_allocations;
367 /* When not verbose only report allocations of more than 1k. */
368 if (!verbose && t->n_bytes < 1024)
371 if (t == traces_copy)
372 s = format (s, "%=9s%=9s %=10s Traceback\n", "Bytes", "Count",
374 s = format (s, "%9d%9d %p", t->n_bytes, t->n_allocations, t->offset);
375 indent = format_get_indent (s);
376 for (i = 0; i < ARRAY_LEN (t->callers) && t->callers[i]; i++)
379 s = format (s, "%U", format_white_space, indent);
380 #if defined(CLIB_UNIX) && !defined(__APPLE__)
381 /* $$$$ does this actually work? */
383 format (s, " %U\n", format_clib_elf_symbol_with_address,
386 s = format (s, " %p\n", t->callers[i]);
391 s = format (s, "%d total traced objects\n", total_objects_traced);
393 vec_free (traces_copy);
395 clib_spinlock_unlock (&tm->lock);
396 if (have_traces == 0)
397 s = format (s, "no traced allocations\n");
404 format_mheap (u8 * s, va_list * va)
406 void *heap = va_arg (*va, u8 *);
407 int verbose = va_arg (*va, int);
408 struct dlmallinfo mi;
409 mheap_trace_main_t *tm = &mheap_trace_main;
411 mi = mspace_mallinfo (heap);
413 s = format (s, "total: %U, used: %U, free: %U, trimmable: %U",
414 format_msize, mi.arena,
415 format_msize, mi.uordblks,
416 format_msize, mi.fordblks, format_msize, mi.keepcost);
419 s = format (s, "\n free chunks %llu free fastbin blks %llu",
420 mi.ordblks, mi.smblks);
422 format (s, "\n max total allocated %U", format_msize, mi.usmblks);
425 if (mspace_is_traced (heap))
426 s = format (s, "\n%U", format_mheap_trace, tm, verbose);
431 clib_mem_usage (clib_mem_usage_t * u)
433 clib_warning ("unimp");
437 mheap_usage (void *heap, clib_mem_usage_t * usage)
439 struct dlmallinfo mi = mspace_mallinfo (heap);
441 /* TODO: Fill in some more values */
442 usage->object_count = 0;
443 usage->bytes_total = mi.arena;
444 usage->bytes_overhead = 0;
445 usage->bytes_max = 0;
446 usage->bytes_used = mi.uordblks;
447 usage->bytes_free = mi.fordblks;
448 usage->bytes_free_reclaimed = 0;
451 /* Call serial number for debugger breakpoints. */
452 uword clib_mem_validate_serial = 0;
455 clib_mem_validate (void)
457 clib_warning ("unimp");
461 mheap_trace (void *v, int enable)
463 (void) mspace_enable_disable_trace (v, enable);
466 mheap_trace_main_free (&mheap_trace_main);
470 clib_mem_trace (int enable)
472 mheap_trace_main_t *tm = &mheap_trace_main;
473 void *current_heap = clib_mem_get_heap ();
475 tm->enabled = enable;
476 mheap_trace (current_heap, enable);
479 tm->current_traced_mheap = current_heap;
481 tm->current_traced_mheap = 0;
485 clib_mem_is_traced (void)
487 return mspace_is_traced (clib_mem_get_heap ());
491 clib_mem_trace_enable_disable (uword enable)
494 mheap_trace_main_t *tm = &mheap_trace_main;
497 tm->enabled = enable;
502 * These API functions seem like layering violations, but
503 * by introducing them we greatly reduce the number
504 * of code changes required to use dlmalloc spaces
507 mheap_alloc_with_lock (void *memory, uword size, int locked)
511 return create_mspace (size, locked);
514 rv = create_mspace_with_base (memory, size, locked);
516 mspace_disable_expand (rv);
522 * fd.io coding-style-patch-verification: ON
525 * eval: (c-set-style "gnu")