如何在内核态环境下使用Hugepage大页内存？

前置准备 #

在使用内核态大页前，需确保系统已配置好Hugepage：

• 临时配置：执行 echo 10 > /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages（数值按需调整，示例为预留10个2MB大页）
• 永久配置：在系统启动参数中添加 hugepages=10，重启后生效

方式一：直接分配物理大页并映射到内核虚拟地址 #

内核提供专用API分配Hugepage物理页，分配后可通过vmap映射到内核虚拟地址空间，直接进行内存操作。

示例代码 #

#include <linux/hugetlb.h>
#include <linux/vmalloc.h>
#include <linux/module.h>

static struct page *huge_page;
static void *kernel_vaddr;

static int __init hugepage_demo_init(void) {
    // 分配一个2MB规格的Hugepage（需提前预留足够大页）
    huge_page = alloc_hugepage(HPAGE_SIZE);
    if (!huge_page) {
        pr_err("Failed to allocate hugepage\n");
        return -ENOMEM;
    }

    // 将物理页映射到内核虚拟地址
    kernel_vaddr = vmap(&huge_page, 1, VM_MAP, PAGE_KERNEL);
    if (!kernel_vaddr) {
        pr_err("Failed to vmap hugepage\n");
        free_hugepage(huge_page);
        return -ENOMEM;
    }

    // 测试大页内存写入
    memset(kernel_vaddr, 0xAA, HPAGE_SIZE);
    pr_info("Hugepage allocated and mapped at virtual address: %p\n", kernel_vaddr);

    return 0;
}

static void __exit hugepage_demo_exit(void) {
    if (kernel_vaddr) {
        vunmap(kernel_vaddr);
    }
    if (huge_page) {
        free_hugepage(huge_page);
    }
    pr_info("Hugepage released\n");
}

module_init(hugepage_demo_init);
module_exit(hugepage_demo_exit);
MODULE_LICENSE("GPL");

方式二：通过hugetlbfs文件映射 #

这种方式类似用户态的mmap逻辑，内核态通过操作hugetlbfs中的文件，将其映射到内核地址空间。

步骤与示例代码 #

1. 先在用户态挂载hugetlbfs：mount -t hugetlbfs hugetlbfs /mnt/huge
2. 内核态实现代码：

#include <linux/fs.h>
#include <linux/mm.h>
#include <linux/hugetlb.h>
#include <linux/module.h>
#include <linux/slab.h>

static struct file *huge_file;
static void *kernel_vaddr;

static int __init hugetlbfs_demo_init(void) {
    loff_t size = HPAGE_SIZE;
    struct inode *inode;

    // 打开hugetlbfs目录下的文件
    huge_file = filp_open("/mnt/huge/huge_demo", O_RDWR | O_CREAT, 0644);
    if (IS_ERR(huge_file)) {
        pr_err("Failed to open hugetlbfs file\n");
        return PTR_ERR(huge_file);
    }

    inode = huge_file->f_inode;
    // 确保文件大小匹配一个大页规格
    if (inode->i_size != size) {
        if (vmtruncate(inode, size)) {
            pr_err("Failed to truncate hugetlbfs file\n");
            filp_close(huge_file, NULL);
            return -EINVAL;
        }
    }

    // 将hugetlb文件映射到内核虚拟地址
    kernel_vaddr = vmap(&inode->i_pages[0], 1, VM_MAP, PAGE_KERNEL);
    if (!kernel_vaddr) {
        pr_err("Failed to vmap hugetlbfs file\n");
        filp_close(huge_file, NULL);
        return -ENOMEM;
    }

    pr_info("Hugetlbfs file mapped at virtual address: %p\n", kernel_vaddr);
    return 0;
}

static void __exit hugetlbfs_demo_exit(void) {
    if (kernel_vaddr) {
        vunmap(kernel_vaddr);
    }
    if (huge_file) {
        filp_close(huge_file, NULL);
    }
    pr_info("Hugetlbfs resources released\n");
}

module_init(hugetlbfs_demo_init);
module_exit(hugetlbfs_demo_exit);
MODULE_LICENSE("GPL");

注意事项 #

• 分配大页前必须确保系统预留了足够的空闲Hugepage，否则分配会直接失败
• 内核态使用大页后必须严格执行释放操作，避免内存泄漏
• 不同内核版本的Hugepage API可能存在细节差异，需根据实际内核版本调整代码

内容的提问来源于stack exchange，提问作者liuye