4. PCI设备探测与初始化：probe函数详解、资源申请与Master使能

PCI设备驱动的核心入口是probe回调函数。当PCI核心枚举总线、发现设备并与驱动ID表匹配成功后，便会调用驱动的probe函数。本节将深入剖析probe函数的典型实现模式，涵盖资源申请（IO端口、内存映射、中断）、pci_enable_device与pci_set_master的调用时机与作用，以及常见初始化陷阱。

4.1 probe函数的标准骨架

一个典型的PCI驱动probe函数遵循“启用设备 → 获取资源 → 注册子系统”的三段式结构。下面给出一个标准实现模板：

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

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

    // 2. 启用PCI设备
    ret = pci_enable_device(pdev);
    if (ret)
        goto err_free;

    // 3. 请求MMIO区域
    ret = pcim_iomap_regions(pdev, BIT(0), "my_driver");
    if (ret)
        goto err_disable;
    dev->regs = pcim_iomap_table(pdev)[0];

    // 4. 设置DMA掩码
    ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(64));
    if (ret) {
        ret = dma_set_mask(&pdev->dev, DMA_BIT_MASK(32));
        if (ret)
            goto err_disable;
    }

    // 5. 设置总线主控
    pci_set_master(pdev);

    // 6. 申请中断
    ret = request_irq(pdev->irq, my_isr, IRQF_SHARED, "my_driver", dev);
    if (ret)
        goto err_disable;

    // 7. 注册设备到内核框架（如misc、net、char等）
    // ...

    dev_info(&pdev->dev, "device probed successfully\n");
    return 0;

err_disable:
    pci_disable_device(pdev);
err_free:
    kfree(dev);
    return ret;
}

4.2 资源申请：IO端口、MMIO与中断

PCI设备通过基地址寄存器（BAR）暴露资源。每个BAR可映射为IO端口空间或内存映射IO（MMIO）空间。驱动必须正确识别BAR类型并申请对应资源。

4.2.1 中断资源申请

PCI中断分为INTx（传统中断线）和MSI/MSI-X（消息 signaled 中断）。现代驱动应优先使用MSI-X，以获得更好的中断亲和性和性能。

// MSI-X 申请示例
int nr_vectors = pci_alloc_irq_vectors(pdev, 1, 4, PCI_IRQ_MSIX);
if (nr_vectors < 0) {
    // 回退到MSI
    nr_vectors = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_MSI);
}
if (nr_vectors < 0) {
    // 回退到INTx
    nr_vectors = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_INTX);
}

for (i = 0; i < nr_vectors; i++) {
    ret = request_irq(pci_irq_vector(pdev, i), my_isr, 0,
                      "my_driver", &dev->irq_ctx[i]);
}

4.3 pci_enable_device 与 pci_set_master

这两个函数是PCI初始化的基石，但它们的职责和调用时机常被混淆。下面通过对比表格厘清差异：

4.3.1 pci_enable_device 的深层行为

调用pci_enable_device时，PCI核心执行以下操作：

1. 检查设备是否已被启用（引用计数防止重复调用）。
2. 将设备电源状态从D3hot恢复至D0（如果之前被挂起）。
3. 分配所有BAR对应的IO/内存地址空间（由BIOS或firmware分配）。
4. 设置PCI命令寄存器：启用IO空间（bit 0）和内存空间（bit 1），但不设置Bus Master位。

4.3.2 pci_set_master 与 DMA 的关系

pci_set_master设置命令寄存器的Bus Master Enable (BME)位（bit 2）。只有设置此位后，设备才能作为总线主控发起DMA传输。未设置BME时，设备只能作为目标响应CPU的MMIO访问。

// pci_set_master 的内核实现（简化）
void pci_set_master(struct pci_dev *dev)
{
    u16 cmd;
    pci_read_config_word(dev, PCI_COMMAND, &cmd);
    cmd |= PCI_COMMAND_MASTER;   // 设置 bit 2
    pci_write_config_word(dev, PCI_COMMAND, cmd);
    dev->is_busmaster = 1;
}

4.4 PCI初始化流程SVG架构图

下图展示了从probe入口到设备完全初始化的完整流程，包含资源申请、中断注册和Master使能的时序关系：

4.5 常见初始化陷阱与排查方法

• 陷阱1：忘记调用pci_enable_device — 直接访问BAR返回全F（0xFFFFFFFF），因为设备未激活。排查：检查/sys/bus/pci/devices/<BDF>/enable是否为1。
• 陷阱2：pci_set_master调用过早 — 在DMA掩码设置之前使能Master，可能导致设备使用错误的地址宽度。排查：确认dma_set_mask返回0后再调用pci_set_master。
• 陷阱3：中断申请时flags错误 — INTx使用IRQF_SHARED，MSI/MSI-X不能使用。排查：cat /proc/interrupts查看中断触发类型。
• 陷阱4：资源泄漏 — probe失败时未释放已申请的BAR或中断。排查：使用pcim_*托管API自动管理生命周期。

4.6 资源申请与释放对照表

为确保初始化与清理的对称性，下表列出每项资源的申请与释放API配对：