Pass cb_arg to discover_client_init
[petitboot] / ui / common / discover-client.h
index 132da02c30a3187fe65234a2a8fc485593a06e90..fdcf93054bb1a0b8e716f263ab4ba41b176843a0 100644 (file)
@@ -14,12 +14,12 @@ struct discover_client;
 
 struct discover_client_ops {
        int (*add_device)(const struct device *device, void *arg);
-       void (*remove_device)(const char *dev_id, void *arg);
+       void (*remove_device)(const struct device *device, void *arg);
        void *cb_arg;
 };
 
 struct discover_client *discover_client_init(
-       const struct discover_client_ops *ops);
+       const struct discover_client_ops *ops, void *cb_arg);
 
 int discover_client_get_fd(const struct discover_client *client);
 
@@ -34,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