We will be needing these routines to find Display Port adapters as well
so modify them to take port type as the second parameter.

Signed-off-by: Mika Westerberg <mika.westerb...@linux.intel.com>
---
 drivers/thunderbolt/switch.c | 16 ++++++++++++++++
 drivers/thunderbolt/tb.c     | 35 +++++++++++++++++------------------
 drivers/thunderbolt/tb.h     |  1 +
 3 files changed, 34 insertions(+), 18 deletions(-)

diff --git a/drivers/thunderbolt/switch.c b/drivers/thunderbolt/switch.c
index c6cb6c44a571..29bf9119e0ae 100644
--- a/drivers/thunderbolt/switch.c
+++ b/drivers/thunderbolt/switch.c
@@ -743,6 +743,22 @@ struct tb_port *tb_port_get_next(struct tb_port *start, 
struct tb_port *end,
        return next;
 }
 
+/**
+ * tb_port_is_enabled() - Is the adapter port enabled
+ * @port: Port to check
+ */
+bool tb_port_is_enabled(struct tb_port *port)
+{
+       switch (port->config.type) {
+       case TB_TYPE_PCIE_UP:
+       case TB_TYPE_PCIE_DOWN:
+               return tb_pci_port_is_enabled(port);
+
+       default:
+               return false;
+       }
+}
+
 /**
  * tb_pci_port_is_enabled() - Is the PCIe adapter port enabled
  * @port: PCIe port to check
diff --git a/drivers/thunderbolt/tb.c b/drivers/thunderbolt/tb.c
index a450bebfeb92..98c993259759 100644
--- a/drivers/thunderbolt/tb.c
+++ b/drivers/thunderbolt/tb.c
@@ -188,40 +188,39 @@ static void tb_free_unplugged_children(struct tb_switch 
*sw)
        }
 }
 
-
 /**
- * find_pci_up_port() - return the first PCIe up port on @sw or NULL
+ * tb_find_port() - return the first port of @type on @sw or NULL
+ * @sw: Switch to find the port from
+ * @type: Port type to look for
  */
-static struct tb_port *tb_find_pci_up_port(struct tb_switch *sw)
+static struct tb_port *tb_find_port(struct tb_switch *sw,
+                                   enum tb_port_type type)
 {
        int i;
        for (i = 1; i <= sw->config.max_port_number; i++)
-               if (sw->ports[i].config.type == TB_TYPE_PCIE_UP)
+               if (sw->ports[i].config.type == type)
                        return &sw->ports[i];
        return NULL;
 }
 
 /**
- * find_unused_down_port() - return the first inactive PCIe down port on @sw
+ * tb_find_unused_port() - return the first inactive port on @sw
+ * @sw: Switch to find the port on
+ * @type: Port type to look for
  */
-static struct tb_port *tb_find_unused_down_port(struct tb_switch *sw)
+static struct tb_port *tb_find_unused_port(struct tb_switch *sw,
+                                          enum tb_port_type type)
 {
        int i;
-       int cap;
-       int res;
-       int data;
+
        for (i = 1; i <= sw->config.max_port_number; i++) {
                if (tb_is_upstream_port(&sw->ports[i]))
                        continue;
-               if (sw->ports[i].config.type != TB_TYPE_PCIE_DOWN)
-                       continue;
-               cap = sw->ports[i].cap_adap;
-               if (!cap)
+               if (sw->ports[i].config.type != type)
                        continue;
-               res = tb_port_read(&sw->ports[i], &data, TB_CFG_PORT, cap, 1);
-               if (res < 0)
+               if (!sw->ports[i].cap_adap)
                        continue;
-               if (data & 0x80000000)
+               if (tb_port_is_enabled(&sw->ports[i]))
                        continue;
                return &sw->ports[i];
        }
@@ -235,7 +234,7 @@ static int tb_tunnel_pci(struct tb *tb, struct tb_switch 
*sw)
        struct tb_port *up, *down;
        struct tb_tunnel *tunnel;
 
-       up = tb_find_pci_up_port(sw);
+       up = tb_find_port(sw, TB_TYPE_PCIE_UP);
        if (!up)
                return 0;
 
@@ -246,7 +245,7 @@ static int tb_tunnel_pci(struct tb *tb, struct tb_switch 
*sw)
        down = NULL;
        parent_sw = tb_to_switch(sw->dev.parent);
        while (parent_sw) {
-               down = tb_find_unused_down_port(parent_sw);
+               down = tb_find_unused_port(parent_sw, TB_TYPE_PCIE_DOWN);
                if (down)
                        break;
                parent_sw = tb_to_switch(parent_sw->dev.parent);
diff --git a/drivers/thunderbolt/tb.h b/drivers/thunderbolt/tb.h
index 8906ee0a8a6a..d16a67898a34 100644
--- a/drivers/thunderbolt/tb.h
+++ b/drivers/thunderbolt/tb.h
@@ -462,6 +462,7 @@ struct tb_port *tb_port_get_next(struct tb_port *start, 
struct tb_port *end,
 
 int tb_switch_find_vse_cap(struct tb_switch *sw, enum tb_switch_vse_cap vsec);
 int tb_port_find_cap(struct tb_port *port, enum tb_port_cap cap);
+bool tb_port_is_enabled(struct tb_port *port);
 
 bool tb_pci_port_is_enabled(struct tb_port *port);
 int tb_pci_port_enable(struct tb_port *port, bool enable);
-- 
2.20.1

Reply via email to