X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=ui%2Fcommon%2Fdiscover-client.h;h=719f72ab15e6d99dd00b26aedd4223c50c8724a2;hp=3ce745b2a84a8caebd8b68197b331b4399d3a3b3;hb=1bb9ed4ffbbd1b186b30702d3c4104d5597d0888;hpb=32e6a41f33e5576716b351bd473a27939fe94fa1

diff --git a/ui/common/discover-client.h b/ui/common/discover-client.h
index 3ce745b..719f72a 100644
--- a/ui/common/discover-client.h
+++ b/ui/common/discover-client.h
@@ -2,18 +2,26 @@
 #define _DISCOVER_CLIENT_H
 
 #include <pb-protocol/pb-protocol.h>
-#include "ui/common/device.h"
 
 struct discover_client;
 
+/**
+ * struct discover_client_ops - Application supplied client info.
+ * @add_device: PB_PROTOCOL_ACTION_ADD event callback.
+ * @remove_device: PB_PROTOCOL_ACTION_REMOVE event callback.
+ * @cb_arg: Client managed convenience variable passed to callbacks.
+ */
+
 struct discover_client_ops {
-	int	(*add_device)(struct device *);
-	void	(*remove_device)(char *);
+	int (*add_device)(const struct device *device, void *arg);
+	void (*remove_device)(const struct device *device, void *arg);
+	void *cb_arg;
 };
 
-struct discover_client *discover_client_init(struct discover_client_ops *ops);
+struct discover_client *discover_client_init(
+	const struct discover_client_ops *ops);
 
-int discover_client_get_fd(struct discover_client *client);
+int discover_client_get_fd(const struct discover_client *client);
 
 void discover_client_destroy(struct discover_client *client);
 
@@ -26,4 +34,21 @@ void discover_client_destroy(struct discover_client *client);
  */
 int discover_client_process(struct discover_client *client);
 
+/**
+ * Get the number of devices that the discover client has stored. This
+ * is the set of devices that have been added and not removed
+ *
+ * @param client The discover client
+ * @return	 The number of devices that have been added.
+ */
+int discover_client_device_count(struct discover_client *client);
+
+/**
+ * Get the device at a specific index.
+ * @param client A pointer to the discover client
+ * @param index  The index of the device to retrieve
+ * @return	 The device at the specified index, or NULL if none exists
+ */
+struct device *discover_client_get_device(struct discover_client *client,
+		int index);
 #endif