#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 pb_boot_data {
+ char *image;
+ char *initrd;
+ char *dtb;
+ char *args;
+};
+
/**
* struct discover_client_ops - Application supplied client info.
- * @add_device: PB_PROTOCOL_ACTION_ADD event callback.
- * @remove_device: PB_PROTOCOL_ACTION_REMOVE event callback.
+ * @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.
+ *
+ * The discover client holds talloc references to the devices (and the
+ * devices' boot options), so callbacks may store boot options and devices
+ * as long as the client remains allocated.
+ *
+ * The boot_status and system_info structs are allocated by the client,
+ * and will be free()ed after the callback is invoked. If the callback
+ * stores these structures for usage beyond the duration of the callback,
+ * it must talloc_steal() them.
*/
struct discover_client_ops {
- int (*add_device)(struct device *device, void *arg);
- void (*remove_device)(char *dev_id, void *arg);
+ 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 (*update_sysinfo)(struct system_info *sysinfo, void *arg);
+ void (*update_config)(struct config *sysinfo, void *arg);
void *cb_arg;
};
-struct discover_client *discover_client_init(struct discover_client_ops *ops);
-
-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);
/**
- * 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_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);
+
+/* Tell the discover server to cancel the default boot option, if any
*/
-int discover_client_process(struct discover_client *client);
+int discover_client_cancel_default(struct discover_client *client);
+/* Send new configuration data to the discover server */
+int discover_client_send_config(struct discover_client *client,
+ struct config *config);
#endif