📚 课程目录第2章 PCIe驱动框架

2. PCIe驱动框架：Linux PCI子系统、pci_driver结构体、probe/remove回调机制

在Linux内核中，PCIe设备驱动并非直接操作硬件寄存器，而是通过内核提供的PCI子系统进行统一管理。理解PCI子系统的分层架构、驱动核心结构体pci_driver以及生命周期回调probe/remove，是编写高效PCIe驱动的基础。本节将拆解这三部分，并配合SVG架构图与表格，帮助您建立清晰的驱动框架认知。

2.1 Linux PCI子系统分层架构

Linux PCI子系统采用分层设计，从上至下依次为：设备驱动层、PCI核心层、总线配置层和硬件访问层。每一层职责明确，驱动开发者主要与设备驱动层和PCI核心层交互。

  关键理解：驱动开发者只需实现pci_driver结构体并注册到PCI核心层，内核会自动完成设备枚举、资源分配、电源管理等底层操作。

从上图可以看出，驱动开发者主要工作在第1层（设备驱动层），通过实现pci_driver结构体中的回调函数，与第2层（PCI核心层）交互。核心层负责管理设备生命周期，驱动只需关注业务逻辑。

2.2 pci_driver结构体深度解析

pci_driver是PCI设备驱动的核心结构体，定义在include/linux/pci.h中。它包含了驱动名称、设备ID表、probe/remove回调、电源管理回调等关键成员。

成员名称	类型	说明	是否必须
`name`	`const char *`	驱动名称，用于sysfs和调试	是
`id_table`	`const struct pci_device_id *`	支持的设备Vendor/Device ID列表	是
`probe`	`int (probe)(struct pci_dev , const struct pci_device_id *)`	设备匹配成功时调用，初始化硬件	是
`remove`	`void (remove)(struct pci_dev )`	设备移除或驱动卸载时调用，释放资源	是
`suspend`	`int (suspend)(struct pci_dev , pm_message_t)`	系统挂起时调用，保存设备状态	否
`resume`	`int (resume)(struct pci_dev )`	系统恢复时调用，恢复设备状态	否
`driver`	`struct device_driver`	内嵌通用设备驱动结构体	自动

实战提示：id_table是驱动与设备匹配的关键。通常使用PCI_VDEVICE()宏来填充，例如：PCI_VDEVICE(VENDOR_ID, DEVICE_ID)。如果驱动支持多个设备，可以定义数组并以{0, }结尾。

下面是一个典型的pci_driver定义示例：

/* 设备ID表：支持两个不同型号的PCIe设备 */
static const struct pci_device_id my_pci_ids[] = {
    { PCI_VDEVICE(0x10EE, 0x7011), 0 },  /* Xilinx DMA设备 */
    { PCI_VDEVICE(0x10EE, 0x7012), 0 },  /* Xilinx DMA设备v2 */
    { 0, }
};
MODULE_DEVICE_TABLE(pci, my_pci_ids);

/* probe回调：设备匹配成功时调用 */
static int my_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
{
    int ret;
    /* 启用PCIe设备 */
    ret = pcim_enable_device(pdev);
    if (ret)
        return ret;

    /* 请求MMIO/BAR资源 */
    ret = pcim_iomap_regions(pdev, BIT(0), "my_pci_driver");
    if (ret)
        return ret;

    /* 设置DMA掩码（64位） */
    ret = dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(64));
    if (ret)
        return ret;

    dev_info(&pdev->dev, "PCIe设备初始化成功\n");
    return 0;
}

/* remove回调：设备移除或驱动卸载时调用 */
static void my_pci_remove(struct pci_dev *pdev)
{
    /* 释放中断、DMA缓冲、取消映射等 */
    dev_info(&pdev->dev, "PCIe设备已移除\n");
}

/* 定义pci_driver结构体 */
static struct pci_driver my_pci_driver = {
    .name     = "my_pci_driver",
    .id_table = my_pci_ids,
    .probe    = my_pci_probe,
    .remove   = my_pci_remove,
};

/* 模块入口/出口 */
module_pci_driver(my_pci_driver);

2.3 probe/remove回调机制与生命周期

probe和remove是PCIe驱动中最重要的两个回调函数，它们决定了设备从发现到销毁的完整生命周期。

2.3.1 probe回调执行流程

当PCI核心层检测到新设备，且设备的Vendor/Device ID与驱动的id_table匹配时，内核会调用驱动的probe回调。标准probe流程如下：

启用设备：调用pcim_enable_device()，使能PCIe设备并分配电源。
请求资源：调用pcim_iomap_regions()或pci_request_regions()，映射BAR空间到CPU虚拟地址。
设置DMA掩码：调用dma_set_mask_and_coherent()，告知设备支持的DMA地址宽度（32位或64位）。
分配中断：调用pci_alloc_irq_vectors()，申请MSI-X或MSI中断向量。
初始化硬件：配置设备寄存器、分配DMA缓冲、注册字符设备或网络设备等。
注册设备：将设备注册到内核框架（如miscdevice、net_device等）。

2.3.2 remove回调执行流程

remove回调在设备热移除、驱动卸载或系统关闭时调用。其职责是probe的逆操作，确保所有资源被正确释放：

停止数据传输：通知硬件停止DMA操作，等待未完成的事务完成。
注销设备：从内核框架中注销字符设备、网络设备等。
释放中断：调用pci_free_irq_vectors()释放MSI/MSI-X中断。
释放DMA缓冲：调用dma_free_coherent()释放一致性DMA内存。
取消映射BAR：调用pcim_iounmap_regions()（如果使用非pcim_*函数则需要手动释放）。

常见陷阱：如果probe中使用了pcim_enable_device()和pcim_iomap_regions()（带pcim_前缀），内核会在remove时自动释放资源，无需手动调用pci_disable_device()。但DMA缓冲和中断必须手动释放，否则会导致内存泄漏或中断风暴。

2.4 设备匹配与注册流程

从驱动加载到设备匹配成功，内核内部经历以下步骤：

步骤	执行者	动作
1	驱动模块	调用`module_pci_driver()`或`pci_register_driver()`注册驱动
2	PCI核心层	将驱动添加到`pci_bus_type`的驱动链表
3	PCI核心层	遍历所有已枚举的PCIe设备，匹配`id_table`
4	PCI核心层	匹配成功，调用驱动的`probe`回调
5	驱动	`probe`执行初始化，返回0表示成功
6	PCI核心层	将设备标记为`已绑定`，创建sysfs关联

2.5 关键避坑与最佳实践

  工程经验总结：
  优先使用pcim_*系列函数：如pcim_enable_device()、pcim_iomap_regions()，它们会在remove时自动释放资源，减少内存泄漏风险。
DMA掩码设置要尽早：在probe中申请任何DMA内存之前，必须先调用dma_set_mask_and_coherent()，否则DMA分配可能失败。
中断申请使用MSI-X优先：MSI-X支持更多中断向量且性能优于传统INTx，调用pci_alloc_irq_vectors()时优先指定PCI_IRQ_MSIX标志。
避免在probe中做耗时操作：probe运行在进程上下文但可能持有锁，长时间阻塞会影响系统启动速度。大块内存分配或固件加载应使用工作队列。
设备ID表要精确：使用PCI_VDEVICE()宏时，确保Vendor ID和Device ID与硬件规格书一致。可以使用lspci -n命令查看实际设备ID。