PCI: hv: Rework MSI handling

[platform/kernel/linux-starfive.git] / drivers / pci / controller / pci-hyperv.c

diff --git a/drivers/pci/controller/pci-hyperv.c b/drivers/pci/controller/pci-hyperv.c
index 6733cb1..45b6ff8 100644 (file)
--- a/drivers/pci/controller/pci-hyperv.c
+++ b/drivers/pci/controller/pci-hyperv.c
@@ -3445,18 +3445,23 @@ static int hv_pci_suspend(struct hv_device *hdev)
 
 static int hv_pci_restore_msi_msg(struct pci_dev *pdev, void *arg)
 {
-       struct msi_desc *entry;
        struct irq_data *irq_data;
+       struct msi_desc *entry;
+       int ret = 0;
 
-       for_each_pci_msi_entry(entry, pdev) {
+       msi_lock_descs(&pdev->dev);
+       msi_for_each_desc(entry, &pdev->dev, MSI_DESC_ASSOCIATED) {
                irq_data = irq_get_irq_data(entry->irq);
-               if (WARN_ON_ONCE(!irq_data))
-                       return -EINVAL;
+               if (WARN_ON_ONCE(!irq_data)) {
+                       ret = -EINVAL;
+                       break;
+               }
 
                hv_compose_msi_msg(irq_data, &entry->msg);
        }
+       msi_unlock_descs(&pdev->dev);
 
-       return 0;
+       return ret;
 }
 
 /*