lib/waiter: Add timeout waiters
[petitboot] / ui / common / discover-client.h
index 50bea6cbe085db39026eb3cbad9142d8093c12a6..2245a65141330883383af48f9e448fa4e4a0e256 100644 (file)
@@ -1,28 +1,62 @@
 #ifndef _DISCOVER_CLIENT_H
 #define _DISCOVER_CLIENT_H
 
 #ifndef _DISCOVER_CLIENT_H
 #define _DISCOVER_CLIENT_H
 
-#include <pb-protocol/pb-protocol.h>
+#include <types/types.h>
+#include <waiter/waiter.h>
 
 struct discover_client;
 
 
 struct discover_client;
 
-struct discover_client_ops {
-       int     (*add_device)(struct device *);
-       void    (*remove_device)(char *);
+struct pb_boot_data {
+       char *image;
+       char *initrd;
+       char *args;
 };
 
 };
 
-struct discover_client *discover_client_init(struct discover_client_ops *ops);
+/**
+ * struct discover_client_ops - Application supplied client info.
+ * @device_add: PB_PROTOCOL_ACTION_ADD event callback.
+ * @device_remove: PB_PROTOCOL_ACTION_REMOVE event callback.
+ * @cb_arg: Client managed convenience variable passed to callbacks.
+ */
+
+struct discover_client_ops {
+       int (*device_add)(struct device *device, void *arg);
+       int (*boot_option_add)(struct device *dev, struct boot_option *option,
+                       void *arg);
+       void (*device_remove)(struct device *device, void *arg);
+       void (*update_status)(struct boot_status *status, void *arg);
+       void *cb_arg;
+};
 
 
-int discover_client_get_fd(struct discover_client *client);
+struct discover_client *discover_client_init(struct waitset *waitset,
+       const struct discover_client_ops *ops, void *cb_arg);
 
 void discover_client_destroy(struct discover_client *client);
 
 /**
 
 void discover_client_destroy(struct discover_client *client);
 
 /**
- * Process data from the server.
+ * Get the number of devices that the discover client has stored. This
+ * is the set of devices that have been added and not removed
  *
  *
- * Will read from the client socket, and call add_device on any discovered
- * devices.
- * 
+ * @param client The discover client
+ * @return      The number of devices that have been added.
  */
  */
-int discover_client_process(struct discover_client *client);
+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);
+
+/* Tell the discover server to boot an image
+ * @param client A pointer to the discover client
+ * @param boot_command The command to boot
+ */
+int discover_client_boot(struct discover_client *client,
+               const struct device *device,
+               const struct boot_option *boot_option,
+               const struct pb_boot_data *data);
 #endif
 #endif