diff options
author | Kent Overstreet <kent.overstreet@linux.dev> | 2024-02-27 01:38:17 -0500 |
---|---|---|
committer | Kent Overstreet <kent.overstreet@linux.dev> | 2024-02-27 02:23:15 -0500 |
commit | 5abb640aa3372a191d33fd8d968b4dd34f6f5167 (patch) | |
tree | 742386afda8ef2d70c8657c28ccd1634a3a430e3 | |
parent | d206a76d7d2726f3b096037f2079ce0bd3ba329b (diff) |
mm: vmalloc_bytes()
Add a __ksize() equivalent for vmalloc.
This is being added so we can track amount of memory stranded waiting
for an RCU grace period.
Cc: linux-mm@kvack.org
Signed-off-by: Kent Overstreet <kent.overstreet@linux.dev>
-rw-r--r-- | include/linux/vmalloc.h | 1 | ||||
-rw-r--r-- | mm/vmalloc.c | 20 |
2 files changed, 21 insertions, 0 deletions
diff --git a/include/linux/vmalloc.h b/include/linux/vmalloc.h index c720be70c8dd..3a6e29de8818 100644 --- a/include/linux/vmalloc.h +++ b/include/linux/vmalloc.h @@ -160,6 +160,7 @@ extern void *vcalloc(size_t n, size_t size) __alloc_size(1, 2); extern void vfree(const void *addr); extern void vfree_atomic(const void *addr); +extern size_t vmalloc_bytes(const void *addr); extern void *vmap(struct page **pages, unsigned int count, unsigned long flags, pgprot_t prot); diff --git a/mm/vmalloc.c b/mm/vmalloc.c index d12a17fc0c17..3642fca84c34 100644 --- a/mm/vmalloc.c +++ b/mm/vmalloc.c @@ -2849,6 +2849,26 @@ void vfree(const void *addr) EXPORT_SYMBOL(vfree); /** + * vmalloc_bytes - Return size of a vmalloc() allocation + * @addr: Memory base address + * + * Returns the size of the allocation as passed to vmalloc() rounded up to + * PAGE_SIZE; does not include extra internal allocations. + */ +size_t vmalloc_bytes(const void *addr) +{ + struct vm_struct *vm = find_vm_area(addr); + if (unlikely(!vm)) { + WARN(1, KERN_ERR "vmalloc_bytes() called on nonexistent vm area (%p)\n", + addr); + return 0; + } + + return vm->nr_pages * PAGE_SIZE; +} +EXPORT_SYMBOL(vmalloc_bytes); + +/** * vunmap - release virtual mapping obtained by vmap() * @addr: memory base address * |