2 * Copyright (c) 2018 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.
18 #include <sys/types.h>
21 #include <linux/mempolicy.h>
22 #include <linux/memfd.h>
24 #include <vppinfra/format.h>
25 #include <vppinfra/linux/syscall.h>
26 #include <vppinfra/linux/sysfs.h>
27 #include <vppinfra/mem.h>
28 #include <vppinfra/hash.h>
29 #include <vppinfra/pmalloc.h>
31 #if __SIZEOF_POINTER__ >= 8
32 #define DEFAULT_RESERVED_MB 16384
34 #define DEFAULT_RESERVED_MB 256
37 static inline clib_pmalloc_chunk_t *
38 get_chunk (clib_pmalloc_page_t * pp, u32 index)
40 return pool_elt_at_index (pp->chunks, index);
44 pmalloc_size2pages (uword size, u32 log2_page_sz)
46 return round_pow2 (size, 1ULL << log2_page_sz) >> log2_page_sz;
50 pmalloc_validate_numa_node (u32 * numa_node)
52 if (*numa_node == CLIB_PMALLOC_NUMA_LOCAL)
55 if (getcpu (&cpu, numa_node, 0) != 0)
62 clib_pmalloc_init (clib_pmalloc_main_t * pm, uword size)
67 ASSERT (pm->error == 0);
69 pagesize = clib_mem_get_default_hugepage_size ();
70 pm->def_log2_page_sz = min_log2 (pagesize);
71 pm->sys_log2_page_sz = min_log2 (sysconf (_SC_PAGESIZE));
72 pm->lookup_log2_page_sz = pm->def_log2_page_sz;
74 /* check if pagemap is accessible */
75 pt = clib_mem_vm_get_paddr (&pt, pm->sys_log2_page_sz, 1);
76 if (pt == 0 || pt[0] == 0)
77 pm->flags |= CLIB_PMALLOC_F_NO_PAGEMAP;
79 size = size ? size : ((u64) DEFAULT_RESERVED_MB) << 20;
80 size = round_pow2 (size, pagesize);
82 pm->max_pages = size >> pm->def_log2_page_sz;
84 /* reserve VA space for future growth */
85 pm->base = mmap (0, size + pagesize, PROT_NONE,
86 MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
88 if (pm->base == MAP_FAILED)
90 pm->error = clib_error_return_unix (0, "failed to reserve %u pages");
94 off = round_pow2 (pointer_to_uword (pm->base), pagesize) -
95 pointer_to_uword (pm->base);
97 /* trim start and end of reservation to be page aligned */
100 munmap (pm->base, off);
104 munmap (pm->base + (pm->max_pages * pagesize), pagesize - off);
109 alloc_chunk_from_page (clib_pmalloc_main_t * pm, clib_pmalloc_page_t * pp,
110 u32 n_blocks, u32 block_align, u32 numa_node)
112 clib_pmalloc_chunk_t *c = 0;
113 clib_pmalloc_arena_t *a;
116 u32 alloc_chunk_index;
118 a = pool_elt_at_index (pm->arenas, pp->arena_index);
122 u32 i, start = 0, prev = ~0;
124 for (i = 0; i < a->subpages_per_page; i++)
126 pool_get (pp->chunks, c);
129 c->size = pp->n_free_blocks / a->subpages_per_page;
132 pp->first_chunk_index = c - pp->chunks;
134 pp->chunks[prev].next = c - pp->chunks;
135 prev = c - pp->chunks;
138 pp->n_free_chunks = a->subpages_per_page;
141 alloc_chunk_index = pp->first_chunk_index;
144 c = pool_elt_at_index (pp->chunks, alloc_chunk_index);
145 off = (block_align - (c->start & (block_align - 1))) & (block_align - 1);
147 if (c->used || n_blocks + off > c->size)
151 alloc_chunk_index = c->next;
155 /* if alignment is needed create new empty chunk */
158 u32 offset_chunk_index;
159 clib_pmalloc_chunk_t *co;
160 pool_get (pp->chunks, c);
162 offset_chunk_index = alloc_chunk_index;
163 alloc_chunk_index = c - pp->chunks;
165 co = pool_elt_at_index (pp->chunks, offset_chunk_index);
166 c->size = co->size - off;
168 c->start = co->start + off;
169 c->prev = offset_chunk_index;
171 co->next = alloc_chunk_index;
175 if (c->size > n_blocks)
177 u32 tail_chunk_index;
178 clib_pmalloc_chunk_t *ct;
179 pool_get (pp->chunks, ct);
181 tail_chunk_index = ct - pp->chunks;
182 c = pool_elt_at_index (pp->chunks, alloc_chunk_index);
183 ct->size = c->size - n_blocks;
185 ct->prev = alloc_chunk_index;
186 ct->start = c->start + n_blocks;
189 c->next = tail_chunk_index;
191 pool_elt_at_index (pp->chunks, ct->next)->prev = tail_chunk_index;
193 else if (c->next != ~0)
194 pool_elt_at_index (pp->chunks, c->next)->prev = alloc_chunk_index;
196 c = get_chunk (pp, alloc_chunk_index);
197 va = pm->base + ((pp - pm->pages) << pm->def_log2_page_sz) +
198 (c->start << PMALLOC_LOG2_BLOCK_SZ);
199 hash_set (pm->chunk_index_by_va, pointer_to_uword (va), alloc_chunk_index);
200 pp->n_free_blocks -= n_blocks;
206 pmalloc_update_lookup_table (clib_pmalloc_main_t * pm, u32 first, u32 count)
208 uword seek, va, pa, p;
210 u32 elts_per_page = 1U << (pm->def_log2_page_sz - pm->lookup_log2_page_sz);
212 vec_validate_aligned (pm->lookup_table, vec_len (pm->pages) *
213 elts_per_page - 1, CLIB_CACHE_LINE_BYTES);
215 p = first * elts_per_page;
216 if (pm->flags & CLIB_PMALLOC_F_NO_PAGEMAP)
218 while (p < elts_per_page * count)
220 pm->lookup_table[p] = pointer_to_uword (pm->base) +
221 (p << pm->lookup_log2_page_sz);
227 fd = open ((char *) "/proc/self/pagemap", O_RDONLY);
228 while (p < elts_per_page * count)
230 va = pointer_to_uword (pm->base) + (p << pm->lookup_log2_page_sz);
231 seek = (va >> pm->sys_log2_page_sz) * sizeof (pa);
232 if (fd != -1 && lseek (fd, seek, SEEK_SET) == seek &&
233 read (fd, &pa, sizeof (pa)) == (sizeof (pa)) &&
234 pa & (1ULL << 63) /* page present bit */ )
236 pa = (pa & pow2_mask (55)) << pm->sys_log2_page_sz;
238 pm->lookup_table[p] = va - pa;
246 static inline clib_pmalloc_page_t *
247 pmalloc_map_pages (clib_pmalloc_main_t * pm, clib_pmalloc_arena_t * a,
248 u32 numa_node, u32 n_pages)
250 clib_pmalloc_page_t *pp = 0;
251 int status, rv, i, mmap_flags;
254 long unsigned int mask[16] = { 0 };
255 long unsigned int old_mask[16] = { 0 };
256 uword size = (uword) n_pages << pm->def_log2_page_sz;
258 clib_error_free (pm->error);
260 if (pm->max_pages <= vec_len (pm->pages))
262 pm->error = clib_error_return (0, "maximum number of pages reached");
266 if (a->log2_subpage_sz != pm->sys_log2_page_sz)
268 pm->error = clib_sysfs_prealloc_hugepages (numa_node,
269 a->log2_subpage_sz, n_pages);
275 rv = get_mempolicy (&old_mpol, old_mask, sizeof (old_mask) * 8 + 1, 0, 0);
276 /* failure to get mempolicy means we can only proceed with numa 0 maps */
277 if (rv == -1 && numa_node != 0)
279 pm->error = clib_error_return_unix (0, "failed to get mempolicy");
283 mask[0] = 1 << numa_node;
284 rv = set_mempolicy (MPOL_BIND, mask, sizeof (mask) * 8 + 1);
285 if (rv == -1 && numa_node != 0)
287 pm->error = clib_error_return_unix (0, "failed to set mempolicy for "
288 "numa node %u", numa_node);
292 mmap_flags = MAP_FIXED | MAP_ANONYMOUS;
294 if ((pm->flags & CLIB_PMALLOC_F_NO_PAGEMAP) == 0)
295 mmap_flags |= MAP_LOCKED;
297 if (a->log2_subpage_sz != pm->sys_log2_page_sz)
298 mmap_flags |= MAP_HUGETLB | MAP_LOCKED;
300 if (a->flags & CLIB_PMALLOC_ARENA_F_SHARED_MEM)
302 mmap_flags |= MAP_SHARED;
303 if (mmap_flags & MAP_HUGETLB)
304 pm->error = clib_mem_create_hugetlb_fd ((char *) a->name, &a->fd);
306 pm->error = clib_mem_create_fd ((char *) a->name, &a->fd);
312 mmap_flags |= MAP_PRIVATE;
316 va = pm->base + (((uword) vec_len (pm->pages)) << pm->def_log2_page_sz);
317 if (mmap (va, size, PROT_READ | PROT_WRITE, mmap_flags, a->fd, 0) ==
320 pm->error = clib_error_return_unix (0, "failed to mmap %u pages at %p "
321 "fd %d numa %d flags 0x%x", n_pages,
322 va, a->fd, numa_node, mmap_flags);
326 clib_memset (va, 0, size);
328 rv = set_mempolicy (old_mpol, old_mask, sizeof (old_mask) * 8 + 1);
329 if (rv == -1 && numa_node != 0)
331 pm->error = clib_error_return_unix (0, "failed to restore mempolicy");
335 /* we tolerate move_pages failure only if request os for numa node 0
336 to support non-numa kernels */
337 rv = move_pages (0, 1, &va, 0, &status, 0);
338 if ((rv == 0 && status != numa_node) || (rv != 0 && numa_node != 0))
340 pm->error = rv == -1 ?
341 clib_error_return_unix (0, "page allocated on wrong node, numa node "
342 "%u status %d", numa_node, status) :
343 clib_error_return (0, "page allocated on wrong node, numa node "
344 "%u status %d", numa_node, status);
346 /* unmap & reesrve */
348 mmap (va, size, PROT_NONE, MAP_FIXED | MAP_PRIVATE | MAP_ANONYMOUS,
353 for (i = 0; i < n_pages; i++)
355 vec_add2 (pm->pages, pp, 1);
356 pp->n_free_blocks = 1 << (pm->def_log2_page_sz - PMALLOC_LOG2_BLOCK_SZ);
357 pp->index = pp - pm->pages;
358 pp->arena_index = a->index;
359 vec_add1 (a->page_indices, pp->index);
364 /* if new arena is using smaller page size, we need to rebuild whole
366 if (a->log2_subpage_sz < pm->lookup_log2_page_sz)
368 pm->lookup_log2_page_sz = a->log2_subpage_sz;
369 pmalloc_update_lookup_table (pm, vec_len (pm->pages) - n_pages,
373 pmalloc_update_lookup_table (pm, 0, vec_len (pm->pages));
375 /* return pointer to 1st page */
376 return pp - (n_pages - 1);
385 clib_pmalloc_create_shared_arena (clib_pmalloc_main_t * pm, char *name,
386 uword size, u32 log2_page_sz, u32 numa_node)
388 clib_pmalloc_arena_t *a;
389 clib_pmalloc_page_t *pp;
392 clib_error_free (pm->error);
394 if (log2_page_sz == 0)
395 log2_page_sz = pm->def_log2_page_sz;
396 else if (log2_page_sz != pm->def_log2_page_sz &&
397 log2_page_sz != pm->sys_log2_page_sz)
399 pm->error = clib_error_create ("unsupported page size (%uKB)",
400 1 << (log2_page_sz - 10));
404 n_pages = pmalloc_size2pages (size, pm->def_log2_page_sz);
406 if (n_pages + vec_len (pm->pages) > pm->max_pages)
409 if (pmalloc_validate_numa_node (&numa_node))
412 pool_get (pm->arenas, a);
413 a->index = a - pm->arenas;
414 a->name = format (0, "%s%c", name, 0);
415 a->numa_node = numa_node;
416 a->flags = CLIB_PMALLOC_ARENA_F_SHARED_MEM;
417 a->log2_subpage_sz = log2_page_sz;
418 a->subpages_per_page = 1U << (pm->def_log2_page_sz - log2_page_sz);
420 if ((pp = pmalloc_map_pages (pm, a, numa_node, n_pages)) == 0)
423 memset (a, 0, sizeof (*a));
424 pool_put (pm->arenas, a);
428 return pm->base + (pp->index << pm->def_log2_page_sz);
432 clib_pmalloc_alloc_inline (clib_pmalloc_main_t * pm, clib_pmalloc_arena_t * a,
433 uword size, uword align, u32 numa_node)
435 clib_pmalloc_page_t *pp;
436 u32 n_blocks, block_align, *page_index;
438 ASSERT (is_pow2 (align));
440 if (pmalloc_validate_numa_node (&numa_node))
445 if (size > 1ULL << pm->def_log2_page_sz)
448 vec_validate_init_empty (pm->default_arena_for_numa_node,
450 if (pm->default_arena_for_numa_node[numa_node] == ~0)
452 pool_get (pm->arenas, a);
453 pm->default_arena_for_numa_node[numa_node] = a - pm->arenas;
454 a->name = format (0, "default-numa-%u%c", numa_node, 0);
455 a->numa_node = numa_node;
456 a->log2_subpage_sz = pm->def_log2_page_sz;
457 a->subpages_per_page = 1;
460 a = pool_elt_at_index (pm->arenas,
461 pm->default_arena_for_numa_node[numa_node]);
463 else if (size > 1ULL << a->log2_subpage_sz)
466 n_blocks = round_pow2 (size, PMALLOC_BLOCK_SZ) / PMALLOC_BLOCK_SZ;
467 block_align = align >> PMALLOC_LOG2_BLOCK_SZ;
469 vec_foreach (page_index, a->page_indices)
471 pp = vec_elt_at_index (pm->pages, *page_index);
472 void *rv = alloc_chunk_from_page (pm, pp, n_blocks, block_align,
479 if ((a->flags & CLIB_PMALLOC_ARENA_F_SHARED_MEM) == 0 &&
480 (pp = pmalloc_map_pages (pm, a, numa_node, 1)))
481 return alloc_chunk_from_page (pm, pp, n_blocks, block_align, numa_node);
487 clib_pmalloc_alloc_aligned_on_numa (clib_pmalloc_main_t * pm, uword size,
488 uword align, u32 numa_node)
490 return clib_pmalloc_alloc_inline (pm, 0, size, align, numa_node);
494 clib_pmalloc_alloc_aligned (clib_pmalloc_main_t * pm, uword size, uword align)
496 return clib_pmalloc_alloc_inline (pm, 0, size, align,
497 CLIB_PMALLOC_NUMA_LOCAL);
501 clib_pmalloc_alloc_from_arena (clib_pmalloc_main_t * pm, void *arena_va,
502 uword size, uword align)
504 clib_pmalloc_arena_t *a = clib_pmalloc_get_arena (pm, arena_va);
505 return clib_pmalloc_alloc_inline (pm, a, size, align, 0);
509 pmalloc_chunks_mergeable (clib_pmalloc_arena_t * a, clib_pmalloc_page_t * pp,
512 clib_pmalloc_chunk_t *c1, *c2;
514 if (ci1 == ~0 || ci2 == ~0)
517 c1 = get_chunk (pp, ci1);
518 c2 = get_chunk (pp, ci2);
520 if (c1->used || c2->used)
523 if (c1->start >> (a->log2_subpage_sz - PMALLOC_LOG2_BLOCK_SZ) !=
524 c2->start >> (a->log2_subpage_sz - PMALLOC_LOG2_BLOCK_SZ))
531 clib_pmalloc_free (clib_pmalloc_main_t * pm, void *va)
533 clib_pmalloc_page_t *pp;
534 clib_pmalloc_chunk_t *c;
535 clib_pmalloc_arena_t *a;
537 u32 chunk_index, page_index;
539 p = hash_get (pm->chunk_index_by_va, pointer_to_uword (va));
545 page_index = clib_pmalloc_get_page_index (pm, va);
546 hash_unset (pm->chunk_index_by_va, pointer_to_uword (va));
548 pp = vec_elt_at_index (pm->pages, page_index);
549 c = pool_elt_at_index (pp->chunks, chunk_index);
550 a = pool_elt_at_index (pm->arenas, pp->arena_index);
552 pp->n_free_blocks += c->size;
555 /* merge with next if free */
556 if (pmalloc_chunks_mergeable (a, pp, chunk_index, c->next))
558 clib_pmalloc_chunk_t *next = get_chunk (pp, c->next);
559 c->size += next->size;
560 c->next = next->next;
561 if (next->next != ~0)
562 get_chunk (pp, next->next)->prev = chunk_index;
563 memset (next, 0, sizeof (*next));
564 pool_put (pp->chunks, next);
568 /* merge with prev if free */
569 if (pmalloc_chunks_mergeable (a, pp, c->prev, chunk_index))
571 clib_pmalloc_chunk_t *prev = get_chunk (pp, c->prev);
572 prev->size += c->size;
573 prev->next = c->next;
575 get_chunk (pp, c->next)->prev = c->prev;
576 memset (c, 0, sizeof (*c));
577 pool_put (pp->chunks, c);
583 format_log2_page_size (u8 * s, va_list * va)
585 u32 log2_page_sz = va_arg (*va, u32);
587 if (log2_page_sz >= 30)
588 return format (s, "%uGB", 1 << (log2_page_sz - 30));
590 if (log2_page_sz >= 20)
591 return format (s, "%uMB", 1 << (log2_page_sz - 20));
593 if (log2_page_sz >= 10)
594 return format (s, "%uKB", 1 << (log2_page_sz - 10));
596 return format (s, "%uB", 1 << log2_page_sz);
601 format_pmalloc_page (u8 * s, va_list * va)
603 clib_pmalloc_page_t *pp = va_arg (*va, clib_pmalloc_page_t *);
604 int verbose = va_arg (*va, int);
605 u32 indent = format_get_indent (s);
607 s = format (s, "page %u: phys-addr %p ", pp->index, pp->pa);
612 s = format (s, "free %u chunks %u free-chunks %d ",
613 (pp->n_free_blocks) << PMALLOC_LOG2_BLOCK_SZ,
614 pool_elts (pp->chunks), pp->n_free_chunks);
618 clib_pmalloc_chunk_t *c;
619 c = pool_elt_at_index (pp->chunks, pp->first_chunk_index);
620 s = format (s, "\n%U%12s%12s%8s%8s%8s%8s",
621 format_white_space, indent + 2,
622 "chunk offset", "size", "used", "index", "prev", "next");
625 s = format (s, "\n%U%12u%12u%8s%8d%8d%8d",
626 format_white_space, indent + 2,
627 c->start << PMALLOC_LOG2_BLOCK_SZ,
628 c->size << PMALLOC_LOG2_BLOCK_SZ,
629 c->used ? "yes" : "no",
630 c - pp->chunks, c->prev, c->next);
633 c = pool_elt_at_index (pp->chunks, c->next);
640 format_pmalloc (u8 * s, va_list * va)
642 clib_pmalloc_main_t *pm = va_arg (*va, clib_pmalloc_main_t *);
643 int verbose = va_arg (*va, int);
644 u32 indent = format_get_indent (s);
646 clib_pmalloc_page_t *pp;
647 clib_pmalloc_arena_t *a;
649 s = format (s, "used-pages %u reserved-pages %u default-page-size %U "
650 "lookup-page-size %U%s", vec_len (pm->pages), pm->max_pages,
651 format_log2_page_size, pm->def_log2_page_sz,
652 format_log2_page_size, pm->lookup_log2_page_sz,
653 pm->flags & CLIB_PMALLOC_F_NO_PAGEMAP ? " no-pagemap" : "");
657 s = format (s, " va-start %p", pm->base);
660 s = format (s, "\n%Ulast-error: %U", format_white_space, indent + 2,
661 format_clib_error, pm->error);
665 pool_foreach (a, pm->arenas,
668 s = format (s, "\n%Uarena '%s' pages %u subpage-size %U numa-node %u",
669 format_white_space, indent + 2, a->name,
670 vec_len (a->page_indices), format_log2_page_size,
671 a->log2_subpage_sz, a->numa_node);
673 s = format (s, " shared fd %d", a->fd);
675 vec_foreach (page_index, a->page_indices)
677 pp = vec_elt_at_index (pm->pages, *page_index);
678 s = format (s, "\n%U%U", format_white_space, indent + 4,
679 format_pmalloc_page, pp, verbose);
688 * fd.io coding-style-patch-verification: ON
691 * eval: (c-set-style "gnu")