thunderbolt: Generalize port finding routines to support all port types

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.westerberg@linux.intel.com>

diff --git a/drivers/thunderbolt/switch.c b/drivers/thunderbolt/switch.c
index 00aec2124f79ff1807b801b7863ebcab82cc002a..028e9a2933826a4b1d7a546e681b36369d4d81cf 100644 (file)
--- a/drivers/thunderbolt/switch.c
+++ b/drivers/thunderbolt/switch.c
@@ -730,6 +730,22 @@ struct tb_port *tb_next_port_on_path(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 fb01396a62a9d8fda1d3a9ae3736438662e51939..903922a16d64967a184c3663e0f19b9c7c328f73 100644 (file)
--- a/drivers/thunderbolt/tb.c
+++ b/drivers/thunderbolt/tb.c
@@ -180,40 +180,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];
        }
@@ -255,7 +254,7 @@ static struct tb_port *tb_find_pcie_down(struct tb_switch *sw,
        }
 
 out:
-       return tb_find_unused_down_port(sw);
+       return tb_find_unused_port(sw, TB_TYPE_PCIE_DOWN);
 }
 
 static int tb_tunnel_pci(struct tb *tb, struct tb_switch *sw)
@@ -265,7 +264,7 @@ static int tb_tunnel_pci(struct tb *tb, struct tb_switch *sw)
        struct tb_switch *parent_sw;
        struct tb_tunnel *tunnel;
 
-       up = tb_find_pci_up_port(sw);
+       up = tb_find_port(sw, TB_TYPE_PCIE_UP);
        if (!up)
                return 0;
 

diff --git a/drivers/thunderbolt/tb.h b/drivers/thunderbolt/tb.h
index 260360ec59a9d3f483fc2065e0acd887ef7934e3..e6040c9f68fab043e8b02b7678ed301cd841ca86 100644 (file)
--- a/drivers/thunderbolt/tb.h
+++ b/drivers/thunderbolt/tb.h
@@ -576,6 +576,7 @@ struct tb_port *tb_next_port_on_path(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);