[PATCH v5 1/4] pci: layerscape: move per-pci device fdt fixup in a function

[laurentiu . tudor]

From: Laurentiu Tudor <laurentiu.tu...@nxp.com>

Move the pci device related fdt fixup in a function in order to
re-use it in a following patch. While at it, improve the error
handling.

Signed-off-by: Laurentiu Tudor <laurentiu.tu...@nxp.com>
---
 drivers/pci/pcie_layerscape_fixup.c | 60 ++++++++++++++++-------------
 1 file changed, 34 insertions(+), 26 deletions(-)

diff --git a/drivers/pci/pcie_layerscape_fixup.c 
b/drivers/pci/pcie_layerscape_fixup.c
index 8315b0b590..7c55e3d249 100644
--- a/drivers/pci/pcie_layerscape_fixup.c
+++ b/drivers/pci/pcie_layerscape_fixup.c
@@ -174,12 +174,41 @@ static void fdt_pcie_set_iommu_map_entry_ls(void *blob,
        }
 }
 
+static int fdt_fixup_pcie_device_ls(void *blob, pci_dev_t bdf,
+                                   struct ls_pcie_rc *pcie_rc)
+{
+       int streamid, index;
+
+       streamid = pcie_next_streamid(pcie_rc->stream_id_cur,
+                                     pcie_rc->pcie->idx);
+       if (streamid < 0) {
+               printf("ERROR: out of stream ids for BDF %d.%d.%d\n",
+                      PCI_BUS(bdf), PCI_DEV(bdf), PCI_FUNC(bdf));
+               return -ENOENT;
+       }
+       pcie_rc->stream_id_cur++;
+
+       index = ls_pcie_next_lut_index(pcie_rc);
+       if (index < 0) {
+               printf("ERROR: out of LUT indexes for BDF %d.%d.%d\n",
+                      PCI_BUS(bdf), PCI_DEV(bdf), PCI_FUNC(bdf));
+               return -ENOENT;
+       }
+
+       /* map PCI b.d.f to streamID in LUT */
+       ls_pcie_lut_set_mapping(pcie_rc, index, bdf >> 8, streamid);
+       /* update msi-map in device tree */
+       fdt_pcie_set_msi_map_entry_ls(blob, pcie_rc, bdf >> 8, streamid);
+       /* update iommu-map in device tree */
+       fdt_pcie_set_iommu_map_entry_ls(blob, pcie_rc, bdf >> 8, streamid);
+
+       return 0;
+}
+
 static void fdt_fixup_pcie_ls(void *blob)
 {
        struct udevice *dev, *bus;
        struct ls_pcie_rc *pcie_rc;
-       int streamid;
-       int index;
        pci_dev_t bdf;
 
        /* Scan all known buses */
@@ -190,32 +219,11 @@ static void fdt_fixup_pcie_ls(void *blob)
                        bus = bus->parent;
                pcie_rc = dev_get_priv(bus);
 
-               streamid = pcie_next_streamid(pcie_rc->stream_id_cur,
-                                             pcie_rc->pcie->idx);
-               if (streamid < 0) {
-                       debug("ERROR: no stream ids free\n");
-                       continue;
-               } else {
-                       pcie_rc->stream_id_cur++;
-               }
-
-               index = ls_pcie_next_lut_index(pcie_rc);
-               if (index < 0) {
-                       debug("ERROR: no LUT indexes free\n");
-                       continue;
-               }
-
                /* the DT fixup must be relative to the hose first_busno */
                bdf = dm_pci_get_bdf(dev) - PCI_BDF(bus->seq, 0, 0);
-               /* map PCI b.d.f to streamID in LUT */
-               ls_pcie_lut_set_mapping(pcie_rc, index, bdf >> 8,
-                                       streamid);
-               /* update msi-map in device tree */
-               fdt_pcie_set_msi_map_entry_ls(blob, pcie_rc, bdf >> 8,
-                                             streamid);
-               /* update iommu-map in device tree */
-               fdt_pcie_set_iommu_map_entry_ls(blob, pcie_rc, bdf >> 8,
-                                               streamid);
+
+               if (fdt_fixup_pcie_device_ls(blob, bdf, pcie_rc) < 0)
+                       break;
        }
        pcie_board_fix_fdt(blob);
 }
-- 
2.17.1