Linux内核空间下基于kmalloc实现aligned_alloc功能的方法咨询

内核中没有直接对应用户态<stdlib.h>里aligned_alloc的函数，但可以通过原生接口或自定义封装来实现相同行为：

一、使用内核原生对齐分配接口 #

1. 页对齐内存分配
   如果对齐要求是PAGE_SIZE（或其整数倍），直接使用__get_free_pages(GFP_KERNEL, order)，其中order代表页阶数（分配大小为2^order * PAGE_SIZE）。分配的内存天然按页对齐，释放时调用free_pages(ptr, order)即可。

2. 通用小内存对齐分配
   内核kmalloc支持通过__GFP_ALIGN_MASK指定自定义对齐规则，示例代码：

   void *aligned_ptr = kmalloc(size, GFP_KERNEL | __GFP_ALIGN_MASK(alignment));
   

   注意：alignment必须是2的幂，且不能超过KMALLOC_MAX_ALIGN（不同架构有差异，通常为页大小）。释放时直接调用kfree(aligned_ptr)。

二、自定义封装模拟aligned_alloc #

如果需要完全复现用户态aligned_alloc的参数校验和任意2的幂对齐行为，可以基于kmalloc封装：

#include <linux/slab.h>
#include <linux/types.h>

void *aligned_alloc(size_t alignment, size_t size)
{
    // 校验对齐参数：必须是2的幂，且不小于指针宽度
    if ((alignment & (alignment - 1)) != 0 || alignment < sizeof(void*))
        return NULL;
    // 校验size必须是alignment的整数倍（遵循用户态接口规则）
    if (size % alignment != 0)
        return NULL;

    // 分配额外内存：用于存储原始指针 + 预留对齐偏移空间
    size_t alloc_size = size + alignment - 1 + sizeof(void*);
    void *raw_ptr = kmalloc(alloc_size, GFP_KERNEL);
    if (!raw_ptr)
        return NULL;

    // 计算对齐后的目标地址
    uintptr_t aligned_addr = ((uintptr_t)raw_ptr + sizeof(void*) + alignment - 1) & ~(alignment - 1);
    // 保存原始指针到对齐地址的前一个指针位，用于后续释放
    void **ptr_store = (void**)(aligned_addr - sizeof(void*));
    *ptr_store = raw_ptr;

    return (void*)aligned_addr;
}

void aligned_free(void *ptr)
{
    if (!ptr)
        return;
    void **ptr_store = (void**)((uintptr_t)ptr - sizeof(void*));
    kfree(*ptr_store);
}

注意事项： #

• 封装的aligned_alloc严格遵循用户态接口规则，参数不合法时返回NULL
• 释放内存必须使用自定义的aligned_free，不能直接调用kfree，否则会引发内存泄漏或内核崩溃

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