3 files changed, 34 insertions, 9 deletions

diff --git a/drivers/pci/msi.c b/drivers/pci/msi.c
index e3a05cc9a595..8c61304cbb37 100644
--- a/drivers/pci/msi.c
+++ b/drivers/pci/msi.c
@@ -571,10 +571,9 @@ int pci_enable_msi(struct pci_dev* dev)
 }
 EXPORT_SYMBOL(pci_enable_msi);
 
-void pci_disable_msi(struct pci_dev* dev)
+void pci_msi_shutdown(struct pci_dev* dev)
 {
         struct msi_desc *entry;
-        int default_irq;
 
         if (!pci_msi_enable || !dev || !dev->msi_enabled)
                 return;
@@ -590,15 +589,26 @@ void pci_disable_msi(struct pci_dev* dev)
                 u32 mask = entry->msi_attrib.maskbits_mask;
                 msi_set_mask_bits(dev->irq, mask, ~mask);
         }
-        if (!entry->dev || entry->msi_attrib.type != PCI_CAP_ID_MSI) {
+        if (!entry->dev || entry->msi_attrib.type != PCI_CAP_ID_MSI)
                 return;
-        }
-
-        default_irq = entry->msi_attrib.default_irq;
-        msi_free_irqs(dev);
 
         /* Restore dev->irq to its default pin-assertion irq */
-        dev->irq = default_irq;
+        dev->irq = entry->msi_attrib.default_irq;
+}
+void pci_disable_msi(struct pci_dev* dev)
+{
+        struct msi_desc *entry;
+
+        if (!pci_msi_enable || !dev || !dev->msi_enabled)
+                return;
+
+        pci_msi_shutdown(dev);
+
+        entry = list_entry(dev->msi_list.next, struct msi_desc, list);
+        if (!entry->dev || entry->msi_attrib.type != PCI_CAP_ID_MSI)
+                return;
+
+        msi_free_irqs(dev);
 }
 EXPORT_SYMBOL(pci_disable_msi);
 
@@ -691,7 +701,7 @@ static void msix_free_all_irqs(struct pci_dev *dev)
         msi_free_irqs(dev);
 }
 
-void pci_disable_msix(struct pci_dev* dev)
+void pci_msix_shutdown(struct pci_dev* dev)
 {
         if (!pci_msi_enable || !dev || !dev->msix_enabled)
                 return;
@@ -699,6 +709,13 @@ void pci_disable_msix(struct pci_dev* dev)
         msix_set_enable(dev, 0);
         pci_intx_for_msi(dev, 1);
         dev->msix_enabled = 0;
+}
+void pci_disable_msix(struct pci_dev* dev)
+{
+        if (!pci_msi_enable || !dev || !dev->msix_enabled)
+                return;
+
+        pci_msix_shutdown(dev);
 
         msix_free_all_irqs(dev);
 }
diff --git a/drivers/pci/pci-driver.c b/drivers/pci/pci-driver.c
index e8d94fafc280..72cf61ed8f96 100644
--- a/drivers/pci/pci-driver.c
+++ b/drivers/pci/pci-driver.c
@@ -360,6 +360,8 @@ static void pci_device_shutdown(struct device *dev)
 
         if (drv && drv->shutdown)
                 drv->shutdown(pci_dev);
+        pci_msi_shutdown(pci_dev);
+        pci_msix_shutdown(pci_dev);
 }
 
 /**
diff --git a/include/linux/pci.h b/include/linux/pci.h
index 7a0770d4c4e2..e09c57e9c373 100644
--- a/include/linux/pci.h
+++ b/include/linux/pci.h
@@ -701,6 +701,8 @@ static inline int pci_enable_msi(struct pci_dev *dev)
         return -1;
 }
 
+static inline void pci_msi_shutdown(struct pci_dev *dev)
+{ }
 static inline void pci_disable_msi(struct pci_dev *dev)
 { }
 
@@ -710,6 +712,8 @@ static inline int pci_enable_msix(struct pci_dev *dev,
         return -1;
 }
 
+static inline void pci_msix_shutdown(struct pci_dev *dev)
+{ }
 static inline void pci_disable_msix(struct pci_dev *dev)
 { }
 
@@ -720,9 +724,11 @@ static inline void pci_restore_msi_state(struct pci_dev *dev)
 { }
 #else
 extern int pci_enable_msi(struct pci_dev *dev);
+extern void pci_msi_shutdown(struct pci_dev *dev);
 extern void pci_disable_msi(struct pci_dev *dev);
 extern int pci_enable_msix(struct pci_dev *dev,
         struct msix_entry *entries, int nvec);
+extern void pci_msix_shutdown(struct pci_dev *dev);
 extern void pci_disable_msix(struct pci_dev *dev);
 extern void msi_remove_pci_irq_vectors(struct pci_dev *dev);
 extern void pci_restore_msi_state(struct pci_dev *dev);