*/
 
 #include <linux/usb/role.h>
+#include <linux/property.h>
 #include <linux/device.h>
 #include <linux/module.h>
 #include <linux/mutex.h>
 }
 EXPORT_SYMBOL_GPL(usb_role_switch_get_role);
 
-static int __switch_match(struct device *dev, const void *name)
+static int switch_fwnode_match(struct device *dev, const void *fwnode)
+{
+       return dev_fwnode(dev) == fwnode;
+}
+
+static int switch_name_match(struct device *dev, const void *name)
 {
        return !strcmp((const char *)name, dev_name(dev));
 }
 {
        struct device *dev;
 
-       dev = class_find_device(role_class, NULL, con->endpoint[ep],
-                               __switch_match);
+       if (con->fwnode) {
+               if (!fwnode_property_present(con->fwnode, con->id))
+                       return NULL;
+
+               dev = class_find_device(role_class, NULL, con->fwnode,
+                                       switch_fwnode_match);
+       } else {
+               dev = class_find_device(role_class, NULL, con->endpoint[ep],
+                                       switch_name_match);
+       }
 
        return dev ? to_role_switch(dev) : ERR_PTR(-EPROBE_DEFER);
 }
        sw->get = desc->get;
 
        sw->dev.parent = parent;
+       sw->dev.fwnode = desc->fwnode;
        sw->dev.class = role_class;
        sw->dev.type = &usb_role_dev_type;
        dev_set_name(&sw->dev, "%s-role-switch", dev_name(parent));
 
 
 /**
  * struct usb_role_switch_desc - USB Role Switch Descriptor
+ * @fwnode: The device node to be associated with the role switch
  * @usb2_port: Optional reference to the host controller port device (USB2)
  * @usb3_port: Optional reference to the host controller port device (USB3)
  * @udc: Optional reference to the peripheral controller device
  * usb_role_switch_register() before registering the switch.
  */
 struct usb_role_switch_desc {
+       struct fwnode_handle *fwnode;
        struct device *usb2_port;
        struct device *usb3_port;
        struct device *udc;