powerpc/eeh: Remove PE at appropriate time

During PCI hotplug and EEH recovery, the PE hierarchy tree might be
changed due to the PCI topology changes. At later point when the
PCI device is added, the PE will be created dynamically again.

The patch introduces new function to remove EEH devices from the
associated PE. That also can cause that the parent PE is removed
from the PE tree if the parent PE doesn't include valid EEH devices
and child PEs.

Signed-off-by: Gavin Shan <shangw@linux.vnet.ibm.com>
Signed-off-by: Benjamin Herrenschmidt <benh@kernel.crashing.org>

2 files changed, 48 insertions, 0 deletions

diff --git a/arch/powerpc/platforms/pseries/eeh.c b/arch/powerpc/platforms/pseries/eeh.c
index 8f2149061846..b60863b184c3 100644
--- a/arch/powerpc/platforms/pseries/eeh.c
+++ b/arch/powerpc/platforms/pseries/eeh.c
@@ -1156,6 +1156,7 @@ static void eeh_remove_device(struct pci_dev *dev)
         dev->dev.archdata.edev = NULL;
         pci_dev_put(dev);
 
+        eeh_rmv_from_parent_pe(edev);
         pci_addr_cache_remove_device(dev);
         eeh_sysfs_remove_device(dev);
 }
diff --git a/arch/powerpc/platforms/pseries/eeh_pe.c b/arch/powerpc/platforms/pseries/eeh_pe.c
index 1d632739d28e..e941e6315fa6 100644
--- a/arch/powerpc/platforms/pseries/eeh_pe.c
+++ b/arch/powerpc/platforms/pseries/eeh_pe.c
@@ -341,3 +341,50 @@ int eeh_add_to_parent_pe(struct eeh_dev *edev)
 
         return 0;
 }
+
+/**
+ * eeh_rmv_from_parent_pe - Remove one EEH device from the associated PE
+ * @edev: EEH device
+ *
+ * The PE hierarchy tree might be changed when doing PCI hotplug.
+ * Also, the PCI devices or buses could be removed from the system
+ * during EEH recovery. So we have to call the function remove the
+ * corresponding PE accordingly if necessary.
+ */
+int eeh_rmv_from_parent_pe(struct eeh_dev *edev)
+{
+        struct eeh_pe *pe, *parent;
+
+        if (!edev->pe) {
+                pr_warning("%s: No PE found for EEH device %s\n",
+                        __func__, edev->dn->full_name);
+                return -EEXIST;
+        }
+
+        /* Remove the EEH device */
+        pe = edev->pe;
+        edev->pe = NULL;
+        list_del(&edev->list);
+
+        /*
+         * Check if the parent PE includes any EEH devices.
+         * If not, we should delete that. Also, we should
+         * delete the parent PE if it doesn't have associated
+         * child PEs and EEH devices.
+         */
+        while (1) {
+                parent = pe->parent;
+                if (pe->type == EEH_PE_PHB)
+                        break;
+
+                if (list_empty(&pe->edevs) &&
+                    list_empty(&pe->child_list)) {
+                        list_del(&pe->child);
+                        kfree(pe);
+                }
+
+                pe = parent;
+        }
+
+        return 0;
+}