Linux内核模块：kzalloc+GFP_DMA分配mmap缓冲区的正确性与对齐疑问

问题背景 #

在Linux内核模块开发中，需使用kzalloc与GFP_DMA分配供mmap调用的缓冲区。已知GFP_DMA可确保内存在物理与虚拟空间连续，而mmap要求映射区域为PAGE_SIZE倍数且物理起始地址对齐PAGE_SIZE。现有需求为映射对应如下结构体的缓冲区：

struct s {
  uint16_t a;
  uint16_t b;
};

疑问1：直接分配结构体大小是否可行？是否必须按整页分配？ #

直接执行mmap_buffer = kzalloc(sizeof(struct s), GFP_DMA);完全错误，原因如下：

• mmap的映射机制基于页实现，内核无法向用户空间映射非PAGE_SIZE整数倍的内存区域，用户空间的mmap调用也强制要求映射大小为页大小的倍数，非整页分配无法满足mmap的基本要求。
• 即使强行尝试映射，用户空间访问时会按页粒度操作，超出结构体大小的部分会访问到内核其他内存，引发越界风险或系统崩溃。

正确的做法是按向上取整的整页数分配内存：

size_t page_count = (sizeof(struct s) + PAGE_SIZE - 1) / PAGE_SIZE;
void *mmap_buffer = kzalloc(page_count * PAGE_SIZE, GFP_DMA);

对于上述4字节的结构体，page_count会计算为1，即分配一整页内存。

疑问2：如何确保缓冲区物理起始地址按PAGE_SIZE对齐？kzalloc是否自动保证？ #

kzalloc不自动保证物理地址按PAGE_SIZE对齐：

• kzalloc的对齐规则由SLAB分配器决定，默认仅保证与分配对象大小匹配的对齐（如8字节、16字节对齐），对于小于PAGE_SIZE的分配，无法确保物理页对齐；即使分配大小等于PAGE_SIZE，也依赖内核SLAB的实现，并非100%可靠。

要确保物理地址页对齐，推荐以下两种可靠方案：

方案1：使用页分配器直接分配

内核提供的页分配器可直接返回物理页对齐的内存，搭配__GFP_ZERO实现清零（等效于kzalloc的清零效果）：

struct page *page = alloc_pages(GFP_DMA | __GFP_ZERO, 0);
if (!page) {
    // 分配失败处理
}
void *mmap_buffer = page_address(page);

• alloc_pages的第二个参数为order，order=0表示分配1页（2^0 * PAGE_SIZE），若需要多页可调整order值。
• page_address将struct page转换为内核虚拟地址，该地址对应的物理地址必然是PAGE_SIZE对齐的。

方案2：使用kzalloc的对齐扩展参数

部分内核版本支持在kzalloc中指定对齐要求，通过__GFP_COMP标记或直接分配整页大小：

// 借助__GFP_COMP标记强制页对齐
void *mmap_buffer = kzalloc(sizeof(struct s), GFP_DMA | __GFP_COMP);
// 或直接分配整页大小
void *mmap_buffer = kzalloc(PAGE_SIZE, GFP_DMA);

但此方案兼容性不如页分配器，且仅当分配大小为PAGE_SIZE时才能稳定保证对齐，因此更推荐方案1。

参考资料 #

• kzalloc（内核5.0版本）文档说明：kzalloc用于分配已清零的内核内存，其对齐行为由SLAB缓存配置决定，默认不保证物理地址按PAGE_SIZE对齐。
• 《Linux设备驱动程序》中mmap限制说明：mmap要求映射的内核内存必须物理连续且按PAGE_SIZE对齐，映射区域大小也必须是PAGE_SIZE的整数倍，驱动需严格满足这些条件才能完成合法映射。

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