第23章 PCI/PCIe设备驱动

课程目录第23章 PCI/PCIe设备驱动

PCI/PCIe设备驱动：配置空间、BAR空间、中断处理(MSI/MSI-X)、DMA传输、MMIO访问

说实话，PCI/PCIe设备驱动这块，是很多内核开发者绕不过去的坎。我早年刚接触时也觉得头大，一堆寄存器、地址空间、中断模式，稍不留神就蓝屏。但等你真正搞懂了，会发现它其实有一套非常清晰的逻辑。

今天咱们就把PCI/PCIe设备驱动的几个核心模块拆开揉碎，一个个讲清楚。配置空间怎么读？BAR空间映射到哪？MSI中断和传统中断有啥区别？DMA传输怎么避免踩坑？嗯，咱们一个一个来。

核心要点：PCI/PCIe驱动开发，本质上就是操作系统与硬件设备之间的“翻译官”。你既要懂硬件的寄存器布局，又要会操作系统的内存管理、中断分配、DMA引擎。

1. 配置空间：设备的“身份证”

每个PCI/PCIe设备都有一个256字节的配置空间（PCIe扩展到了4KB）。这里面存着设备的Vendor ID、Device ID、Class Code、BAR地址等信息。说白了，这就是设备的“身份证”。

我习惯在驱动入口处先读取配置空间，确认设备是否存在、版本对不对。你想想看，如果连设备ID都对不上，后面做再多工作也是白搭。

// 读取配置空间中的 Vendor ID 和 Device ID
USHORT vendorId = READ_CONFIG_USHORT(pdo, PCI_VENDOR_ID_REGISTER);
USHORT deviceId = READ_CONFIG_USHORT(pdo, PCI_DEVICE_ID_REGISTER);

if (vendorId != MY_VENDOR_ID || deviceId != MY_DEVICE_ID) {
    // 设备不匹配，直接返回失败
    return STATUS_DEVICE_DOES_NOT_EXIST;
}

这里有个坑——配置空间的访问方式。传统PCI用I/O端口0xCF8/0xCFC来访问，而PCIe则推荐用MMIO方式（通过Memory Mapped Configuration Space）。我在一个老项目中遇到过，用传统方式访问PCIe配置空间，结果某些寄存器读出来全是0xFF。后来改用MMIO方式，一切正常。

注意：PCIe的配置空间扩展到了4KB，传统PCI只有256字节。如果你的驱动要兼容两者，记得先判断设备类型，再决定读取范围。

3. BAR空间：设备与CPU的“通信窗口”

BAR（Base Address Register）是配置空间里最关键的寄存器之一。它告诉操作系统：我这个设备需要多少内存空间或I/O空间，以及这些空间映射到哪。

我个人习惯在驱动初始化时，先遍历所有BAR，看看哪些是有效的，哪些是空的。然后调用HalAssignSlotResources或IoAssignResources来分配资源。

// 遍历6个BAR（PCI设备最多6个）
for (int i = 0; i < 6; i++) {
    ULONG barValue = READ_CONFIG_ULONG(pdo, PCI_BASE_ADDRESS_REGISTER_0 + i * 4);
    if (barValue == 0) continue;  // 未使用的BAR

    if (barValue & 0x01) {
        // I/O空间
        ULONG ioBase = barValue & ~0x03;
        DbgPrint("BAR%d: I/O base = 0x%X\n", i, ioBase);
    } else {
        // 内存空间
        ULONG memBase = barValue & ~0x0F;
        DbgPrint("BAR%d: Memory base = 0x%X\n", i, memBase);
    }
}

你可能会问：为什么有的BAR是I/O空间，有的是内存空间？这取决于硬件设计。我见过一个网卡芯片，它的控制寄存器放在I/O空间，数据缓冲区放在内存空间。嗯，设计者这么安排，自然有他的道理。

小技巧：读取BAR后，先写全1进去，再读回来，就能知道这个BAR需要多大空间。这是PCI规范规定的“大小探测”方法。

4. 中断处理：MSI/MSI-X vs 传统INTx

中断处理是驱动开发中最容易出问题的环节。传统PCI使用INTx中断线（共享中断），而PCIe引入了MSI（Message Signaled Interrupt）和MSI-X。

MSI/MSI-X的好处很明显：不需要共享中断线，每个中断都有自己的向量。我在项目中遇到过，用传统INTx时，多个设备共享一条中断线，结果一个设备的中断处理函数里，还得判断是不是自己的中断，麻烦得很。

换成MSI后，每个中断独立，处理函数直接对应，代码清爽多了。

// 注册MSI中断处理
NTSTATUS RegisterMsiInterrupt(PDEVICE_OBJECT deviceObject) {
    // 分配MSI中断资源
    // 注意：MSI中断数量必须是2的幂（1, 2, 4, 8...）
    ULONG messageCount = 1;
    NTSTATUS status = IoConnectInterruptEx(
        &intParams,
        deviceObject,
        MyInterruptServiceRoutine,
        NULL,
        &messageCount,
        MSI
    );

    if (!NT_SUCCESS(status)) {
        // 回退到传统INTx
        DbgPrint("MSI注册失败，回退到INTx\n");
        return RegisterLegacyInterrupt(deviceObject);
    }
    return STATUS_SUCCESS;
}

经验之谈：MSI-X比MSI更灵活，它支持每个中断独立配置，数量也不受2的幂限制。如果你的设备支持MSI-X，优先用MSI-X。我曾在某款NVMe SSD驱动中，用MSI-X分配了16个中断队列，性能提升非常明显。

5. DMA传输：让设备自己搬数据

DMA（Direct Memory Access）是PCIe设备的核心能力。没有DMA，CPU就得亲自搬数据，那效率就太低了。

DMA传输的关键是物理地址连续。用户态分配的内存通常是虚拟地址连续的，但物理地址可能不连续。所以驱动里要用AllocateCommonBuffer或MmAllocateContiguousMemory来分配DMA缓冲区。

// 分配DMA缓冲区
PHYSICAL_ADDRESS highestAcceptable;
highestAcceptable.QuadPart = 0xFFFFFFFF;  // 32位设备

PVOID dmaBuffer = MmAllocateContiguousMemory(
    DMA_BUFFER_SIZE,
    highestAcceptable
);

if (dmaBuffer == NULL) {
    DbgPrint("DMA缓冲区分配失败\n");
    return STATUS_INSUFFICIENT_RESOURCES;
}

// 获取物理地址
PHYSICAL_ADDRESS physAddr = MmGetPhysicalAddress(dmaBuffer);
DbgPrint("DMA缓冲区虚拟地址: 0x%p, 物理地址: 0x%I64X\n", 
         dmaBuffer, physAddr.QuadPart);

这里有个容易忽略的点：DMA缓冲区的一致性。如果CPU和DMA设备同时访问同一块内存，需要处理好缓存一致性问题。我建议用MmAllocateContiguousMemorySpecifyCache指定非缓存或写合并属性。

警告：千万不要在DMA传输过程中释放缓冲区！我曾经犯过这个错，结果设备还在写内存，缓冲区已经被回收了，系统直接蓝屏。血的教训。

6. MMIO访问：直接操作设备寄存器

MMIO（Memory Mapped I/O）就是把设备的寄存器映射到CPU的内存地址空间。这样CPU就可以用普通的读写指令来操作设备寄存器，非常方便。

在Windows驱动中，MMIO映射通常通过MmMapIoSpace来实现。映射后，你就可以像操作普通内存一样读写设备寄存器了。

// 映射MMIO空间
PHYSICAL_ADDRESS physBase;
physBase.QuadPart = barPhysicalAddress;

ULONG numberOfBytes = barLength;
PVOID mappedBase = MmMapIoSpace(physBase, numberOfBytes, MmNonCached);

if (mappedBase == NULL) {
    DbgPrint("MMIO映射失败\n");
    return STATUS_UNSUCCESSFUL;
}

// 读写设备寄存器
volatile ULONG *reg = (volatile ULONG *)mappedBase;
*reg = 0x12345678;  // 写寄存器
ULONG value = *reg;  // 读寄存器

// 使用完毕后取消映射
MmUnmapIoSpace(mappedBase, numberOfBytes);

我个人习惯在MMIO映射时指定MmNonCached缓存类型。为什么？因为设备寄存器的值随时可能变化，如果用缓存，读到的可能是旧值。你想想看，读状态寄存器读到缓存里的旧值，那判断就全错了。

建议：MMIO访问时，一定要用volatile关键字。这告诉编译器：每次读写都直接访问内存，不要优化到寄存器里。否则编译器可能把多次读操作合并成一次，导致逻辑错误。

7. 知识体系总览

下面这张图总结了PCI/PCIe设备驱动的核心模块和它们之间的关系。你可以把它当作开发时的“地图”。

这张图把PCI/PCIe驱动开发的四个核心模块串起来了。配置空间是第一步，用来识别设备；BAR空间建立CPU与设备的通信窗口；中断处理让设备能主动通知CPU；DMA传输则实现高效的数据搬运。这四个模块环环相扣，缺一不可。

好了，关于PCI/PCIe设备驱动的核心内容就讲到这里。这些知识点都是我在实际项目中反复验证过的，希望能帮你少走弯路。如果你在开发中遇到具体问题，欢迎随时交流。

📚 课程目录

PCI/PCIe设备驱动：配置空间、BAR空间、中断处理(MSI/MSI-X)、DMA传输、MMIO访问

1. 配置空间：设备的“身份证”

3. BAR空间：设备与CPU的“通信窗口”

4. 中断处理：MSI/MSI-X vs 传统INTx

5. DMA传输：让设备自己搬数据

6. MMIO访问：直接操作设备寄存器

7. 知识体系总览