4. kmalloc 标志位：GFP_KERNEL、GFP_ATOMIC、GFP_DMA、GFP_HIGHUSER 等标志详解

在 Linux 内核中，kmalloc 是驱动开发者最常用的内存分配函数之一。与用户空间的 malloc 不同，kmalloc 的第一个参数是请求的字节数，第二个参数则是 GFP（Get Free Page）标志。GFP 标志决定了内核在分配内存时的行为模式，包括是否允许睡眠、从哪个内存区域分配、以及分配失败时的处理策略。理解这些标志是编写健壮驱动程序的基石。

4.1 GFP 标志的分类与作用域

GFP 标志并非单一值，而是一个位掩码。内核通过组合不同的标志位来定义标准分配模式。下表列出了最常用的标准 GFP 标志及其核心特性：

4.2 核心标志深度解析

4.2.1 GFP_KERNEL：标准进程上下文分配器

GFP_KERNEL 是内核中最常用的分配标志。当调用 kmalloc(size, GFP_KERNEL) 时，内核会执行以下操作：

• 允许睡眠： 如果当前内存不足，调用进程会进入睡眠状态，等待内核回收页面或执行 OOM（Out-Of-Memory）处理。
• 来源区域： 优先从 ZONE_NORMAL 分配。如果设备需要 ZONE_DMA 区域的内存，可以组合使用 GFP_KERNEL | GFP_DMA。
• 重试机制： 内核会多次尝试回收内存，直到成功或触发 OOM killer。
• 使用限制： 只能在进程上下文（可以睡眠的上下文）中使用。在中断、软中断、tasklet 或持有自旋锁时禁止使用。

// 典型用法：在进程上下文中分配 1024 字节
void *buf = kmalloc(1024, GFP_KERNEL);
if (!buf) {
    pr_err("内存分配失败\n");
    return -ENOMEM;
}
// ... 使用 buf ...
kfree(buf);

4.2.2 GFP_ATOMIC：原子上下文救星

GFP_ATOMIC 用于那些不能睡眠的上下文。它的行为与 GFP_KERNEL 截然不同：

• 禁止睡眠： 如果内存不足，分配器不会阻塞当前执行流，而是立即返回 NULL。
• 紧急储备： 内核为原子分配预留了少量的紧急内存池（emergency pools），确保在极端情况下也能满足小规模分配。
• 分配成功率： 由于不能回收页面，GFP_ATOMIC 的分配成功率低于 GFP_KERNEL，且通常只能分配较小的内存块（通常 < 1 页）。
• 使用场景： 中断处理程序、软中断（softirq）、tasklet、持有 spinlock 或 rwlock 的代码路径。

// 中断处理程序中的分配
static irqreturn_t my_interrupt(int irq, void *dev_id) {
    void *data;
    // 必须使用 GFP_ATOMIC，因为我们在中断上下文
    data = kmalloc(64, GFP_ATOMIC);
    if (!data) {
        pr_warn("中断中分配内存失败\n");
        return IRQ_NONE;
    }
    // ... 处理中断数据 ...
    kfree(data);
    return IRQ_HANDLED;
}

4.2.3 GFP_DMA：物理地址受限的分配

某些硬件（尤其是老旧的 ISA 设备或特定嵌入式外设）要求 DMA 缓冲区位于物理内存的前 16MB（ZONE_DMA）。GFP_DMA 标志强制分配器从 ZONE_DMA 区域分配内存：

• 物理连续性： kmalloc 本身保证物理连续，GFP_DMA 进一步限制了分配区域。
• 稀缺性： ZONE_DMA 区域通常很小（x86 上为 16MB），频繁使用 GFP_DMA 可能导致该区域耗尽。
• 组合使用： 通常与 GFP_KERNEL 或 GFP_ATOMIC 组合：kmalloc(size, GFP_KERNEL | GFP_DMA)。
• 现代趋势： 对于支持 64 位寻址的现代设备（如 PCIe），应使用 dma_alloc_coherent 或 dma_map_single 等 DMA API，而非直接使用 GFP_DMA。

// 为 ISA DMA 控制器分配缓冲区
void *dma_buf = kmalloc(512, GFP_KERNEL | GFP_DMA);
if (!dma_buf) {
    pr_err("无法从 ZONE_DMA 分配内存\n");
    return -ENOMEM;
}
// 获取物理地址（用于设置 DMA 控制器寄存器）
dma_addr_t phys_addr = virt_to_phys(dma_buf);

4.2.4 GFP_HIGHUSER：高端内存优先策略

在 32 位系统上，内核直接映射区域（ZONE_NORMAL）通常只有 896MB。为了支持更大的内存总量，内核引入了 ZONE_HIGHMEM（高端内存）。GFP_HIGHUSER 标志指示分配器优先从高端内存分配：

• 节省低端内存： 将宝贵的 ZONE_NORMAL 留给内核核心使用，用户空间的大块数据（如页缓存）尽量使用高端内存。
• 非直接映射： 高端内存没有直接的虚拟地址映射，需要通过 kmap / kmap_atomic 临时映射才能访问。
• 64 位系统： 在 64 位系统上，高端内存概念通常不存在（虚拟地址空间足够大），GFP_HIGHUSER 的行为退化为类似 GFP_KERNEL。
• 驱动使用： 驱动程序通常不需要直接使用 GFP_HIGHUSER，除非在实现文件系统或页缓存相关功能时。

4.3 标志选择决策流程

下面的 SVG 流程图展示了驱动开发者在选择 GFP 标志时应遵循的决策路径：

4.4 其他重要标志与组合技巧

除了上述四个核心标志，内核还提供了多个辅助标志用于精细控制内存分配行为：

4.5 常见错误与调试方法

GFP 标志使用不当是内核驱动开发中最常见的错误来源之一。以下是几个典型问题及其排查方法：

• 原子上下文中睡眠： 在中断或持有自旋锁时使用 GFP_KERNEL。内核会输出 BUG: sleeping function called from invalid context 和调用栈。使用 in_interrupt() 或 in_atomic() 宏在运行时检查上下文。
• ZONE_DMA 耗尽： 频繁使用 GFP_DMA 分配大块内存。使用 cat /proc/zoneinfo 查看各区域剩余页面数。
• 内存泄漏： 分配后忘记释放。使用 kmemleak 内核调试工具扫描未释放的内存。
• 分配过大： kmalloc 最大分配通常限制在 4MB 左右（取决于内存碎片）。对于大块分配，应使用 __get_free_pages 或 vmalloc。

// 运行时上下文检查示例
#include <linux/preempt.h>

void safe_alloc_example(void) {
    void *buf;

    if (in_interrupt() || irqs_disabled() || in_atomic()) {
        // 原子上下文：使用 GFP_ATOMIC
        buf = kmalloc(128, GFP_ATOMIC);
        pr_debug("原子上下文分配\n");
    } else {
        // 进程上下文：使用 GFP_KERNEL
        buf = kmalloc(128, GFP_KERNEL);
        pr_debug("进程上下文分配\n");
    }

    if (!buf) {
        pr_err("分配失败\n");
        return;
    }
    kfree(buf);
}