MSI: Consolidate BUG_ON()s.

[firefly-linux-kernel-4.4.55.git] / drivers / pci / msi.c

diff --git a/drivers/pci/msi.c b/drivers/pci/msi.c
index 91fed543fc84b8e88f25271174b6c1656293635f..38e4c807f80f22d3f102fe47434f9f80da8b97c6 100644 (file)
--- a/drivers/pci/msi.c
+++ b/drivers/pci/msi.c
@@ -573,8 +573,6 @@ void pci_disable_msi(struct pci_dev* dev)
                return;
        }
 
-       BUG_ON(irq_has_action(dev->first_msi_irq));
-
        default_irq = entry->msi_attrib.default_irq;
        msi_free_irq(dev, dev->first_msi_irq);
 
@@ -590,6 +588,8 @@ static int msi_free_irq(struct pci_dev* dev, int irq)
        int head, entry_nr, type;
        void __iomem *base;
 
+       BUG_ON(irq_has_action(irq));
+
        entry = get_irq_msi(irq);
        if (!entry || entry->dev != dev) {
                return -EINVAL;
@@ -674,10 +674,24 @@ int pci_enable_msix(struct pci_dev* dev, struct msix_entry *entries, int nvec)
        return status;
 }
 
-void pci_disable_msix(struct pci_dev* dev)
+static void msix_free_all_irqs(struct pci_dev *dev)
 {
        int irq, head, tail = 0;
 
+       irq = head = dev->first_msi_irq;
+       while (head != tail) {
+               tail = get_irq_msi(irq)->link.tail;
+
+               if (irq != head)
+                       msi_free_irq(dev, irq);
+               irq = tail;
+       }
+       msi_free_irq(dev, irq);
+       dev->first_msi_irq = 0;
+}
+
+void pci_disable_msix(struct pci_dev* dev)
+{
        if (!pci_msi_enable)
                return;
        if (!dev)
@@ -690,18 +704,7 @@ void pci_disable_msix(struct pci_dev* dev)
        pci_intx(dev, 1);               /* enable intx */
        dev->msix_enabled = 0;
 
-       irq = head = dev->first_msi_irq;
-       while (head != tail) {
-               tail = get_irq_msi(irq)->link.tail;
-
-               BUG_ON(irq_has_action(irq));
-
-               if (irq != head)        /* Release MSI-X irq */
-                       msi_free_irq(dev, irq);
-               irq = tail;
-       }
-       msi_free_irq(dev, irq);
-       dev->first_msi_irq = 0;
+       msix_free_all_irqs(dev);
 }
 
 /**
@@ -718,27 +721,11 @@ void msi_remove_pci_irq_vectors(struct pci_dev* dev)
        if (!pci_msi_enable || !dev)
                return;
 
-       if (dev->msi_enabled) {
-               BUG_ON(irq_has_action(dev->first_msi_irq));
+       if (dev->msi_enabled)
                msi_free_irq(dev, dev->first_msi_irq);
-       }
-       if (dev->msix_enabled) {
-               int irq, head, tail = 0;
-               void __iomem *base = NULL;
 
-               irq = head = dev->first_msi_irq;
-               while (head != tail) {
-                       tail = get_irq_msi(irq)->link.tail;
-                       base = get_irq_msi(irq)->mask_base;
-
-                       BUG_ON(irq_has_action(irq));
-
-                       if (irq != head) /* Release MSI-X irq */
-                               msi_free_irq(dev, irq);
-                       irq = tail;
-               }
-               msi_free_irq(dev, irq);
-       }
+       if (dev->msix_enabled)
+               msix_free_all_irqs(dev);
 }
 
 void pci_no_msi(void)