从零编写 GPU 驱动：Asahi Linux AGX 架构与 Apple M 系列图形栈逆向工程

\n\n

2022 年，Ahern 社区启动了 Asahi Linux 项目，目标是在 Apple M 系列芯片上运行原生 Linux。最大的挑战之一是 GPU 驱动——Apple 的 GPU 完全没有公开文档。三年过去，Asahi Linux 不仅跑通了 M1/M2/M3 的 GPU，还向 Mesa 上游提交了完整的 AGX（Apple Graphics）驱动栈，成为 Linux 内核中第一个用 Rust 编写 GPU 驱动的项目。这背后的工程方法论值得每一个系统程序员学习。

\n\n

一、Apple Silicon GPU 架构概览

\n\n

Apple M 系列芯片中的 GPU 被称为 AGX，与传统 GPU 的架构理念截然不同。它采用基于 tile 的延迟渲染（Tile-Based Deferred Rendering, TBDR）架构，统一内存访问（UMA），以及与 CPU 共享内存一致性总线。

\n\n

核心架构组件：

\n\n

\n
• GX6450 家族：虽然是 PowerVR 授权衍生，但 Apple 做了大量定制，特别是统一着色器架构和硬件同步原语
\n
• AGX 固件：GPU 运行专有固件，负责内存管理和上下文调度，主机驱动必须与固件通信
\n
• UAT（Unified Address Translation）：GPU 页表与 CPU 共享 IOMMU 域，实现真正的零拷贝统一内存
\n
• AIC（Apple Interrupt Controller）：定制中断控制器，与传统 GIC 完全不同
\n

\n\n

最棘手的部分是逆向工程——没有数据手册、没有寄存器文档、没有任何公开信息。Asahi 团队从 iBoot 固件、macOS 内核扩展、以及硬件 TRACE 三个方向入手。

\n\n

二、逆向工程方法论

\n\n

2.1 固件提取与解析

\n\n

Apple 的 GPU 固件是 IMG 格式的专有 blob。逆向的第一步是提取并解析它：

\n\n

# 从 macOS IPSW 固件包中提取 GPU 固件\n$ python3 extract_firmware.py --ipsw iPhone14,3_16.0_20A362_Restore.ipsw \\n    --output agx_firmware/\n\n# 反汇编固件入口点\n$ llvm-objdump -d agx_firmware/agx_g13p.bin | head -200

\n\n

固件解析揭示了一个微内核架构：固件负责 GPU 内存分配、上下文切换、命令缓冲区验证。主机驱动通过共享内存中环形缓冲区与固件通信。

\n\n

2.2 寄存器追踪

\n\n

没有数据手册意味着你不知道每个寄存器的功能。Asahi 团队使用了三种追踪方法：

\n\n

// 通过 macOS 内核扩展 hook 捕获寄存器写操作\nstruct RegAccessTrace {\n    offset: u64,\n    value: u64,\n    timestamp: u64,\n    callstack: [u64; 8],\n}\n\n// 使用 DTrace 在 macOS 层追踪 IOGPU 调用\n$ sudo dtrace -n 'ioservice:::start / arg0 == \"IOAccelerator\" / {\n    printf(\"offset=0x%x value=0x%x\", arg1, arg2);\n}'

\n\n

2.3 硬件行为推断

\n\n

最难的部分是推断寄存器语义。Asahi 团队采用了灰盒测试方法：写入不同值，观察 GPU 行为变化，逐步构建寄存器语义模型。这通常需要数周时间才能理解一个子系统的用途。

\n\n

三、Rust for Linux 驱动架构

\n\n

Asahi GPU 驱动选择 Rust 编写，原因有二：一是 Rust 的所有权模型天然适合 GPU 内存管理中的所有权语义；二是驱动代码的复杂度要求比 C 更强的安全保障。

\n\n

3.1 驱动入口与内核注册

\n\n

//! drivers/gpu/drm/asahi/mod.rs\n\nuse kernel::prelude::*;\nuse kernel::drm::{self, Device};\nuse kernel::pci;\n\nmodule! {\n    type: AsahiDriver,\n    name: \"asahi\",\n    author: \"Asahi Linux Contributors\",\n    description: \"Apple AGX GPU driver\",\n    license: \"GPL-2.0-only\",\n}\n\nstruct AsahiDriver {\n    drm: Box