3. Linux PCIe 子系统架构：核心数据结构与驱动生命周期

在 Linux 内核中，PCIe 子系统为驱动开发者提供了一套标准化的框架。理解 pci_driver 结构体、pci_device_id 匹配表以及 probe/remove 回调函数的生命周期，是编写稳定、高效 PCIe 驱动的基础。本章将深入剖析这三个核心要素，并结合工程案例说明其在实际链路故障排查中的应用。

3.1 pci_driver 结构体：驱动的“身份证”

pci_driver 是 PCIe 驱动与内核之间的契约。每个 PCIe 驱动都必须实例化并注册一个 pci_driver 结构体，内核通过它来识别、匹配和管理驱动。

// 一个典型的 pci_driver 定义示例
static struct pci_driver my_pcie_driver = {
    .name       = "my_pcie_device",
    .id_table   = my_pcie_ids,       // 匹配表
    .probe      = my_pcie_probe,     // 探测函数
    .remove     = my_pcie_remove,    // 移除函数
    .shutdown   = my_pcie_shutdown,  // 关机处理
    .driver     = {
        .pm     = &my_pcie_pm_ops,   // 电源管理
    },
};

3.2 pci_device_id 匹配表：驱动与设备的“红娘”

pci_device_id 表定义了驱动所支持的 PCIe 设备集合。内核在枚举 PCIe 总线时，会遍历所有已注册驱动的 id_table，寻找匹配的设备。

// 匹配表示例：支持多个设备
#define PCI_VENDOR_ID_MY_COMPANY   0x1234
#define PCI_DEVICE_ID_ACCEL_V1     0x5678
#define PCI_DEVICE_ID_ACCEL_V2     0x5679

static const struct pci_device_id my_pcie_ids[] = {
    { PCI_DEVICE(PCI_VENDOR_ID_MY_COMPANY, PCI_DEVICE_ID_ACCEL_V1) },
    { PCI_DEVICE(PCI_VENDOR_ID_MY_COMPANY, PCI_DEVICE_ID_ACCEL_V2) },
    { 0, }  // 必须以空条目结束
};
MODULE_DEVICE_TABLE(pci, my_pcie_ids);

3.3 probe/remove 回调函数生命周期

probe 和 remove 是 PCIe 驱动生命周期中最重要的两个回调函数。它们分别在设备绑定和解除绑定时被内核调用，负责资源的初始化和清理。

3.3.1 probe 回调：驱动的“构造函数”

当 pci_device_id 匹配成功后，内核调用驱动的 probe 函数。典型的 probe 流程包括：

1. 使能设备：调用 pci_enable_device() 激活 PCIe 设备。
2. 请求 MMIO/BAR 资源：使用 pci_request_regions() 或 pcim_iomap_regions()。
3. 设置 DMA 掩码：调用 dma_set_mask_and_coherent()。
4. 初始化硬件：读写配置空间、复位设备等。
5. 注册中断：使用 request_irq() 或 pci_alloc_irq_vectors()。
6. 创建内核接口：如注册字符设备、net_device 等。

static int my_pcie_probe(struct pci_dev *pdev, const struct pci_device_id *id)
{
    struct my_dev *dev;
    int ret;

    // 1. 分配私有数据结构
    dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
    if (!dev)
        return -ENOMEM;

    // 2. 使能 PCIe 设备
    ret = pci_enable_device(pdev);
    if (ret)
        return ret;

    // 3. 请求 BAR0 资源并 ioremap
    ret = pcim_iomap_regions(pdev, BIT(0), "my_pcie");
    if (ret)
        goto err_disable;

    dev->regs = pcim_iomap_table(pdev)[0];

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

    // 5. 设置驱动私有数据
    pci_set_drvdata(pdev, dev);

    // 6. 注册中断（MSI-X 优先）
    ret = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_MSIX | PCI_IRQ_MSI);
    if (ret < 0)
        goto err_disable;

    dev->irq = pci_irq_vector(pdev, 0);
    ret = request_irq(dev->irq, my_pcie_isr, 0, "my_pcie", dev);
    if (ret)
        goto err_free_vectors;

    dev_info(&pdev->dev, "Probe successful\n");
    return 0;

err_free_vectors:
    pci_free_irq_vectors(pdev);
err_disable:
    pci_disable_device(pdev);
    return ret;
}

3.3.2 remove 回调：驱动的“析构函数”

remove 函数负责释放 probe 中分配的所有资源，顺序通常与 probe 相反：

• 释放中断 → 释放 MSI/MSI-X 向量 → 释放 IOMMU/DMA 映射 → 释放 MMIO 映射 → 禁用设备。

static void my_pcie_remove(struct pci_dev *pdev)
{
    struct my_dev *dev = pci_get_drvdata(pdev);

    // 1. 释放中断
    free_irq(dev->irq, dev);

    // 2. 释放 MSI/MSI-X 向量
    pci_free_irq_vectors(pdev);

    // 3. 禁用设备（自动释放 pcim_iomap_regions 的资源）
    pci_disable_device(pdev);

    // 4. 私有数据由 devm_ 自动释放
    dev_info(&pdev->dev, "Remove complete\n");
}

3.4 工程案例：probe 失败链路排查

排查步骤：