当前位置：首页>Linux>RISC-V Linux 内存管理相关arch目录解析

RISC-V Linux 内存管理相关arch目录解析

2026-06-29 15:15:00

arch/riscv/mm/是RISC‑V架构内存管理子系统的核心实现目录。它与对应的头文件，以及arch/riscv/kernel下的个别文件共同构成了连接MMU与内核通用内存管理模块（mm/）的桥梁。本文尝试理解其中主要文件的作用，以及关键接口的定义位置，以便为深入掌握RISC‑V虚拟内存机制打下基础。使用的内核版本是7.0.10。

一、内存初始化

init.c 包含了系统启动时最核心的两个内存初始化函数：

setup_bootmem()

– 负责解析设备树中的物理内存信息，填充 memblock 数据结构；根据内核启动参数“mem=”获得可用内存地址范围；预留内核镜像、DTB、initrd 等关键区域；并初始化 DMA/CMA 连续内存池。

paging_init()

– 调用 setup_vm_final() 建立最终的内核页表，包括线性映射区（将物理内存 1:1 映射到内核虚拟空间）和固定映射区（用于早期 I/O 映射）。最后写入 satp 寄存器，正式启用 MMU。

二、页表操作

页表操作的函数定义分布在多个头文件和通用代码中。下面按功能类别逐一说明。

2.1 页表项内容操作（读、写、测试、构造）

这些函数全部以 static inline 形式定义在 arch/riscv/include/asm/pgtable.h 中。常见的包括：

pte_none(pte) – 检查页表项是否为空
pte_present(pte) – 判断页表项对应的物理页是否在内存中
mk_pte(page, prot) – 根据物理页结构体和权限生成页表项
pfn_pte(pfn, prot) – 将物理页框号与权限组合成页表项
pte_read(pte), pte_write(pte), pte_exec(pte) – 分别获取读、写、执行权限位
pte_set_read, pte_set_write 等修改操作也在该文件中

2.2 页表项、页分配/释放（非最底层）

负责分配或释放一整张页表页（如 PGD 页、PTE 页）。它们的定义分两种情况。

PGD 相关的定义在 arch/riscv/include/asm/pgalloc.h 中：

pgd_alloc(mm) – 分配一个页全局目录（PGD）页。它通常是一个内联函数，内部调用 __pgd_alloc()。
pgd_free(mm, pgd) – 释放一个 PGD 页，内部调用 __pgd_free()。

PTE 及其他页表页的操作直接使用了 include/asm-generic/pgalloc.h 中的通用版本，因此以下函数的实际定义位于该通用头文件中：

pte_alloc_one(mm) – 分配一个页表项（PTE）页（用于用户空间）
pte_alloc_one_kernel(mm) – 为内核分配一个 PTE 页
pte_free(mm, pte) – 释放一个 PTE 页
pte_free_kernel(mm, pte) – 释放内核使用的 PTE 页

类似地，pmd_alloc_one、pud_alloc_one 等也在通用代码中提供

2.3 底层页表项、页内存分配

真正执行物理内存页分配的底层函数，在最近的 Linux 内核中已被移入通用代码。具体来说，__pgd_alloc(mm, vm_flags) 和 __pgd_free(pgd)定义在 include/asm-generic/pgalloc.h 中。

所以，查找页表操作函数时，请先判断它属于哪一类：

页表项内容 → arch/riscv/include/asm/pgtable.h
PGD 高层封装 → arch/riscv/include/asm/pgalloc.h
PTE / PMD / PUD 高层封装 → include/asm-generic/pgalloc.h
底层物理分配（__pgd_alloc 等） → include/asm-generic/pgalloc.h

这种分层设计兼顾了性能（通过内联）和代码复用（通用实现）。

三、缺页异常处理：traps.c 与 fault.c

RISC‑V 的缺页异常处理由两个文件共同完成：arch/riscv/kernel/traps.c 用作入口调度，arch/riscv/mm/fault.c 执行具体操作。

3.1 入口调度：traps.c

主要函数是 do_page_fault()，该函数是缺页异常的 C 语言入口，由汇编级别的异常向量表调用。它的主要职责是：

1. 从 CSR 寄存器中读取异常原因（cause）和错误地址（badaddr）。

2. 判断异常发生的上下文（用户态还是内核态）。如果内核态访问非法地址，直接触发 Oops。

3. 针对特定地址区间（如 vmalloc 区）做早期过滤。

4. 调用 arch/riscv/mm/fault.c 中的 handle_mm_fault完成真正的缺页修复。

3.2 业务执行：fault.c

主要函数是handle_page_fault，该函数是架构相关的核心缺页处理逻辑。它的执行流程如下：

1. 获取异常码，主要是读取 RISC-V 的 cause 寄存器。

2. 通过 RCU 读取 VMA，调用 access_error 检查权限，合法则进入修复。

3. 调用通用函数 handle_mm_fault 分配物理页或处理写时复制，并根据结果返回或发送信号。

其它一些函数：

vmalloc_fault()

专门处理内核 vmalloc 区域的缺页异常。当进程访问尚未建立映射的 vmalloc 地址时，该函数会从全局内核页表 init_mm 中复制所需的页表项到当前进程的页表中。这是实现“每个进程页表都包含内核映射副本”的关键技术，避免了进程切换时频繁切换页表基址寄存器。

access_error()

该函数接收异常原因（cause）和 VMA 权限标志，判断本次访问是否合法。例如，写操作需要 VMA 具有 VM_WRITE 标志，执行操作需要 VM_EXEC 标志。如果合法返回 false，否则返回 true 表示发生权限错误。

四、TLB、Cache 与 DMA 一致性

4.1 TLB 刷新：tlbflush.c

TLB（Translation Lookaside Buffer）是 CPU 内部的地址转换缓存。当页表被修改后，必须刷新 TLB。tlbflush.c 提供以下接口：

flush_tlb_all() – 全局刷新所有 TLB 条目（开销大，仅用于系统启动等极少数情况）。
flush_tlb_range(vma, start, end) – 按虚拟地址范围刷新 TLB，精度更高，性能更好。
flush_tlb_page(vma, addr) – 刷新单个页面的 TLB 条目。

RISC‑V 使用 sfence.vma 指令执行实际的 TLB 失效操作。该指令支持按 ASID（地址空间标识符）和虚拟地址粒度刷新。在多核系统中，上述函数会通过 IPI（核间中断）通知其他核心执行本地 TLB 刷新。

4.2 Cache 刷新：cacheflush.c

cacheflush.c主要负责刷新指令cache，提供：

flush_icache_all() – 刷新所有 CPU 核心的指令 Cache，用于模块加载、BPF JIT 等场景。
flush_icache_range(start, end) – 按地址范围刷新指令 Cache。

对于数据缓存一致性，底层依赖 RISC‑V 的 Zicbom 扩展指令集：cbo.clean（写回）、cbo.flush（写回并失效）、cbo.inval（失效）等；而对于指令缓存刷新，则使用 fence.i 指令。

4.3 非一致 DMA 支持：dma-noncoherent.c

对于不支持硬件 Cache 一致性的 DMA 设备，软件必须负责同步。dma-noncoherent.c 实现了通用的 DMA 回调：

arch_sync_dma_for_device(dev, paddr, size, dir) – 在 CPU 将缓冲区交给设备前调用。它会根据 DMA 方向执行必要的 Cache 写回（writeback），确保设备从内存中读到最新数据。
arch_sync_dma_for_cpu(dev, paddr, size, dir) – 在设备完成 DMA、CPU 重新获得所有权时调用。它会执行 Cache 无效化（invalidate），防止 CPU 读到 Cache 中的陈旧数据。

这些函数被通用的 DMA API（如 dma_map_single 和 dma_unmap_single）调用，对驱动开发者透明。