discover: parse boot message from incoming ACTION_BOOT messages

Add a function in the protocol code to deserialise a boot message, and
use it to extract a boot_command in the discover server.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

2 files changed, 34 insertions, 0 deletions

diff --git a/lib/pb-protocol/pb-protocol.c b/lib/pb-protocol/pb-protocol.c
index 4e0af87..31637c6 100644
--- a/lib/pb-protocol/pb-protocol.c
+++ b/lib/pb-protocol/pb-protocol.c
@@ -404,3 +404,34 @@ out_err:
 	talloc_free(dev);
 	return NULL;
 }
+
+struct boot_command *pb_protocol_deserialise_boot_command(void *ctx,
+		const struct pb_protocol_message *message)
+{
+	struct boot_command *cmd;
+	const char *pos;
+	unsigned int len;
+
+	len = message->payload_len;
+	pos = message->payload;
+
+	cmd = talloc(ctx, struct boot_command);
+
+	if (read_string(cmd, &pos, &len, &cmd->option_id))
+		goto out_err;
+
+	if (read_string(cmd, &pos, &len, &cmd->boot_image_file))
+		goto out_err;
+
+	if (read_string(cmd, &pos, &len, &cmd->initrd_file))
+		goto out_err;
+
+	if (read_string(cmd, &pos, &len, &cmd->boot_args))
+		goto out_err;
+
+	return cmd;
+
+out_err:
+	talloc_free(cmd);
+	return NULL;
+}
diff --git a/lib/pb-protocol/pb-protocol.h b/lib/pb-protocol/pb-protocol.h
index 2ec264c..6068f05 100644
--- a/lib/pb-protocol/pb-protocol.h
+++ b/lib/pb-protocol/pb-protocol.h
@@ -50,4 +50,7 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd);
 struct device *pb_protocol_deserialise_device(void *ctx,
 		const struct pb_protocol_message *message);
 
+struct boot_command *pb_protocol_deserialise_boot_command(void *ctx,
+		const struct pb_protocol_message *message);
+
 #endif /* _PB_PROTOCOL_H */