greybus: bundle: Initialize all bundles on link-up

An interface can have 1 or more bundles. On link-up event, we must initialize
all the bundles associated with the interface.

Signed-off-by: Viresh Kumar <viresh.kumar@linaro.org>
Signed-off-by: Greg Kroah-Hartman <gregkh@google.com>
diff --git a/drivers/staging/greybus/ap.c b/drivers/staging/greybus/ap.c
index 132ecb4..1dc1339 100644
--- a/drivers/staging/greybus/ap.c
+++ b/drivers/staging/greybus/ap.c
@@ -146,9 +146,7 @@
 				management->link_up.interface_id);
 			return;
 		}
-		ret = gb_bundle_init(intf,
-				management->link_up.interface_id,
-				management->link_up.device_id);
+		ret = gb_bundles_init(intf, management->link_up.device_id);
 		if (ret) {
 			dev_err(hd->parent,
 				"error %d initializing bundles for interface %hhu\n",
diff --git a/drivers/staging/greybus/bundle.c b/drivers/staging/greybus/bundle.c
index 47a3413..9691978 100644
--- a/drivers/staging/greybus/bundle.c
+++ b/drivers/staging/greybus/bundle.c
@@ -146,16 +146,11 @@
 	}
 }
 
-int gb_bundle_init(struct gb_interface *intf, u8 bundle_id, u8 device_id)
+int gb_bundle_init(struct gb_bundle *bundle, u8 device_id)
 {
-	struct gb_bundle *bundle;
+	struct gb_interface *intf = bundle->intf;
 	int ret;
 
-	bundle = gb_bundle_find(intf, bundle_id);
-	if (!bundle) {
-		dev_err(intf->hd->parent, "bundle %hhu not found\n", bundle_id);
-		return -ENOENT;
-	}
 	bundle->device_id = device_id;
 
 	ret = svc_set_route_send(bundle, intf->hd);
@@ -175,6 +170,24 @@
 	return 0;
 }
 
+int gb_bundles_init(struct gb_interface *intf, u8 device_id)
+{
+	struct gb_bundle *bundle;
+	int ret = 0;
+
+	list_for_each_entry(bundle, &intf->bundles, links) {
+		ret = gb_bundle_init(bundle, device_id);
+		if (ret) {
+			dev_err(intf->hd->parent,
+				"Failed to initialize bundle %hhu\n",
+				bundle->id);
+			break;
+		}
+	}
+
+	return ret;
+}
+
 struct gb_bundle *gb_bundle_find(struct gb_interface *intf, u8 bundle_id)
 {
 	struct gb_bundle *bundle;
diff --git a/drivers/staging/greybus/bundle.h b/drivers/staging/greybus/bundle.h
index 385c90a..2948098 100644
--- a/drivers/staging/greybus/bundle.h
+++ b/drivers/staging/greybus/bundle.h
@@ -31,7 +31,8 @@
 struct gb_bundle *gb_bundle_create(struct gb_interface *intf, u8 bundle_id,
 				   u8 class_type);
 void gb_bundle_destroy(struct gb_interface *intf);
-int gb_bundle_init(struct gb_interface *intf, u8 module_id, u8 device_id);
+int gb_bundle_init(struct gb_bundle *bundle, u8 device_id);
+int gb_bundles_init(struct gb_interface *intf, u8 device_id);
 
 struct gb_bundle *gb_bundle_find(struct gb_interface *intf, u8 bundle_id);
 void gb_bundle_bind_protocols(void);