#include <asm/pci-bridge.h>
 #include <asm/pSeries_reconfig.h>
 #include <asm/ppc-pci.h>
+#include <asm/firmware.h>
 
 /*
  * Traverse_func that inits the PCI fields of the device node.
                pdn->busno = (regs[0] >> 16) & 0xff;
                pdn->devfn = (regs[0] >> 8) & 0xff;
        }
+       if (firmware_has_feature(FW_FEATURE_ISERIES)) {
+               u32 *busp = (u32 *)get_property(dn, "linux,subbus", NULL);
+               if (busp)
+                       pdn->bussubno = *busp;
+       }
 
        pdn->pci_ext_config_space = (type && *type == 1);
        return NULL;
 
 void iSeries_pcibios_init(void)
 {
        struct pci_controller *phb;
-       struct device_node *node;
-       struct device_node *dn;
+       struct device_node *root = of_find_node_by_path("/");
+       struct device_node *node = NULL;
 
-       for_each_node_by_type(node, "pci") {
+       if (root == NULL) {
+               printk(KERN_CRIT "iSeries_pcibios_init: can't find root "
+                               "of device tree\n");
+               return;
+       }
+       while ((node = of_get_next_child(root, node)) != NULL) {
                HvBusNumber bus;
                u32 *busp;
 
+               if ((node->type == NULL) || (strcmp(node->type, "pci") != 0))
+                       continue;
+
                busp = (u32 *)get_property(node, "bus-range", NULL);
                if (busp == NULL)
                        continue;
                phb->first_busno = bus;
                phb->last_busno = bus;
                phb->ops = &iSeries_pci_ops;
+       }
 
-               /* Find and connect the devices. */
-               for (dn = NULL; (dn = of_get_next_child(node, dn)) != NULL;) {
-                       struct pci_dn *pdn;
-                       u32 *reg;
-
-                       reg = (u32 *)get_property(dn, "reg", NULL);
-                       if (reg == NULL) {
-                               printk(KERN_DEBUG "no reg property!\n");
-                               continue;
-                       }
-                       busp = (u32 *)get_property(dn, "linux,subbus", NULL);
-                       if (busp == NULL) {
-                               printk(KERN_DEBUG "no subbus property!\n");
-                               continue;
-                       }
+       of_node_put(root);
 
-                       pdn = kzalloc(sizeof(*pdn), GFP_KERNEL);
-                       if (pdn == NULL)
-                               return;
-                       dn->data = pdn;
-                       pdn->node = dn;
-                       pdn->busno = bus;
-                       pdn->devfn = (reg[0] >> 8) & 0xff;
-                       pdn->bussubno = *busp;
-               }
-       }
+       pci_devs_phb_init();
 }
 
 /*
 
                                snprintf(buf, sizeof(buf), "pci@%x,%d",
                                                PCI_SLOT(devfn), function);
                        dt_start_node(dt, buf);
-                       reg[0] = (bus << 18) | (devfn << 8);
+                       reg[0] = (bus << 16) | (devfn << 8);
                        reg[1] = 0;
                        reg[2] = 0;
                        reg[3] = 0;