X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=lib%2Fpb-protocol%2Fpb-protocol.c;h=ded35eb5a1f5888b01374ac028fe9af2d68f2d0b;hp=6f278c89ac00e24185b8d8071152b3825076b413;hb=bd06734362bb727b09b943688d9b69aa0a84590d;hpb=29fbf1a71d8ac7deace11064fff48e7c2c67c178

diff --git a/lib/pb-protocol/pb-protocol.c b/lib/pb-protocol/pb-protocol.c
index 6f278c8..ded35eb 100644
--- a/lib/pb-protocol/pb-protocol.c
+++ b/lib/pb-protocol/pb-protocol.c
@@ -38,8 +38,38 @@
  * action = 0x2: device remove message
  *  payload:
  *   4-byte len, id
+ *
+ * action = 0x3: boot
+ *  payload:
+ *   4-byte len, boot option id
+ *   4-byte len, boot_image_file
+ *   4-byte len, initrd_file
+ *   4-byte len, boot_args
+ *
  */
 
+void pb_protocol_dump_device(const struct device *dev, const char *text,
+	FILE *stream)
+{
+	struct boot_option *opt;
+
+	fprintf(stream, "%snew dev:\n", text);
+	fprintf(stream, "%s\tid:   %s\n", text, dev->id);
+	fprintf(stream, "%s\tname: %s\n", text, dev->name);
+	fprintf(stream, "%s\tdesc: %s\n", text, dev->description);
+	fprintf(stream, "%s\ticon: %s\n", text, dev->icon_file);
+	fprintf(stream, "%s\tboot options:\n", text);
+	list_for_each_entry(&dev->boot_options, opt, list) {
+		fprintf(stream, "%s\t\tid:   %s\n", text, opt->id);
+		fprintf(stream, "%s\t\tname: %s\n", text, opt->name);
+		fprintf(stream, "%s\t\tdesc: %s\n", text, opt->description);
+		fprintf(stream, "%s\t\ticon: %s\n", text, opt->icon_file);
+		fprintf(stream, "%s\t\tboot: %s\n", text, opt->boot_image_file);
+		fprintf(stream, "%s\t\tinit: %s\n", text, opt->initrd_file);
+		fprintf(stream, "%s\t\targs: %s\n", text, opt->boot_args);
+	}
+}
+
 int pb_protocol_device_cmp(const struct device *a, const struct device *b)
 {
 	return !strcmp(a->id, b->id);
@@ -131,63 +161,81 @@ static int optional_strlen(const char *str)
 
 int pb_protocol_device_len(const struct device *dev)
 {
-	struct boot_option *opt;
-	int len;
-
-	len =	4 + optional_strlen(dev->id) +
+	return	4 + optional_strlen(dev->id) +
 		4 + optional_strlen(dev->name) +
 		4 + optional_strlen(dev->description) +
-		4 + optional_strlen(dev->icon_file) +
-		4;
+		4 + optional_strlen(dev->icon_file);
+}
 
-	list_for_each_entry(&dev->boot_options, opt, list) {
-		len +=	4 + optional_strlen(opt->id) +
-			4 + optional_strlen(opt->name) +
-			4 + optional_strlen(opt->description) +
-			4 + optional_strlen(opt->icon_file) +
-			4 + optional_strlen(opt->boot_image_file) +
-			4 + optional_strlen(opt->initrd_file) +
-			4 + optional_strlen(opt->boot_args);
-	}
+int pb_protocol_boot_option_len(const struct boot_option *opt)
+{
 
-	return len;
+	return	4 + optional_strlen(opt->device_id) +
+		4 + optional_strlen(opt->id) +
+		4 + optional_strlen(opt->name) +
+		4 + optional_strlen(opt->description) +
+		4 + optional_strlen(opt->icon_file) +
+		4 + optional_strlen(opt->boot_image_file) +
+		4 + optional_strlen(opt->initrd_file) +
+		4 + optional_strlen(opt->boot_args);
 }
 
-int pb_protocol_serialise_device(const struct device *dev, char *buf, int buf_len)
+int pb_protocol_boot_len(const struct boot_command *boot)
 {
-	struct boot_option *opt;
-	uint32_t n;
-	char *pos;
+	return  4 + optional_strlen(boot->option_id) +
+		4 + optional_strlen(boot->boot_image_file) +
+		4 + optional_strlen(boot->initrd_file) +
+		4 + optional_strlen(boot->boot_args);
+}
 
-	pos = buf;
+int pb_protocol_serialise_device(const struct device *dev,
+		char *buf, int buf_len)
+{
+	char *pos = buf;
 
-	/* construct payload into buffer */
 	pos += pb_protocol_serialise_string(pos, dev->id);
 	pos += pb_protocol_serialise_string(pos, dev->name);
 	pos += pb_protocol_serialise_string(pos, dev->description);
 	pos += pb_protocol_serialise_string(pos, dev->icon_file);
 
-	/* write option count */
-	n = 0;
+	assert(pos <= buf + buf_len);
+	(void)buf_len;
 
-	list_for_each_entry(&dev->boot_options, opt, list)
-		n++;
+	return 0;
+}
 
-	*(uint32_t *)pos = __cpu_to_be32(n);
-	pos += sizeof(uint32_t);
+int pb_protocol_serialise_boot_option(const struct boot_option *opt,
+		char *buf, int buf_len)
+{
+	char *pos = buf;
 
-	/* write each option */
-	list_for_each_entry(&dev->boot_options, opt, list) {
-		pos += pb_protocol_serialise_string(pos, opt->id);
-		pos += pb_protocol_serialise_string(pos, opt->name);
-		pos += pb_protocol_serialise_string(pos, opt->description);
-		pos += pb_protocol_serialise_string(pos, opt->icon_file);
-		pos += pb_protocol_serialise_string(pos, opt->boot_image_file);
-		pos += pb_protocol_serialise_string(pos, opt->initrd_file);
-		pos += pb_protocol_serialise_string(pos, opt->boot_args);
-	}
+	pos += pb_protocol_serialise_string(pos, opt->device_id);
+	pos += pb_protocol_serialise_string(pos, opt->id);
+	pos += pb_protocol_serialise_string(pos, opt->name);
+	pos += pb_protocol_serialise_string(pos, opt->description);
+	pos += pb_protocol_serialise_string(pos, opt->icon_file);
+	pos += pb_protocol_serialise_string(pos, opt->boot_image_file);
+	pos += pb_protocol_serialise_string(pos, opt->initrd_file);
+	pos += pb_protocol_serialise_string(pos, opt->boot_args);
 
 	assert(pos <= buf + buf_len);
+	(void)buf_len;
+
+	return 0;
+}
+
+int pb_protocol_serialise_boot_command(const struct boot_command *boot,
+		char *buf, int buf_len)
+{
+	char *pos = buf;
+
+	pos += pb_protocol_serialise_string(pos, boot->option_id);
+	pos += pb_protocol_serialise_string(pos, boot->boot_image_file);
+	pos += pb_protocol_serialise_string(pos, boot->initrd_file);
+	pos += pb_protocol_serialise_string(pos, boot->boot_args);
+
+	assert(pos <= buf + buf_len);
+	(void)buf_len;
 
 	return 0;
 }
@@ -283,67 +331,98 @@ struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd)
 }
 
 
-struct device *pb_protocol_deserialise_device(void *ctx,
+int pb_protocol_deserialise_device(struct device *dev,
 		const struct pb_protocol_message *message)
 {
-	struct device *dev;
-	const char *pos;
-	int i, n_options;
 	unsigned int len;
+	const char *pos;
+	int rc = -1;
 
 	len = message->payload_len;
 	pos = message->payload;
 
-	dev = talloc(ctx, struct device);
-
 	if (read_string(dev, &pos, &len, &dev->id))
-		goto out_err;
+		goto out;
 
 	if (read_string(dev, &pos, &len, &dev->name))
-		goto out_err;
+		goto out;
 
 	if (read_string(dev, &pos, &len, &dev->description))
-		goto out_err;
+		goto out;
 
 	if (read_string(dev, &pos, &len, &dev->icon_file))
-		goto out_err;
+		goto out;
 
-	n_options = __be32_to_cpu(*(uint32_t *)pos);
-	pos += sizeof(uint32_t);
+	rc = 0;
 
-	list_init(&dev->boot_options);
-
-	for (i = 0; i < n_options; i++) {
-		struct boot_option *opt;
-
-		opt = talloc(dev, struct boot_option);
-
-		if (read_string(opt, &pos, &len, &opt->id))
-			goto out_err;
-		if (read_string(opt, &pos, &len, &opt->name))
-			goto out_err;
-		if (read_string(opt, &pos, &len,
-					&opt->description))
-			goto out_err;
-		if (read_string(opt, &pos, &len,
-					&opt->icon_file))
-			goto out_err;
-		if (read_string(opt, &pos, &len,
-					&opt->boot_image_file))
-			goto out_err;
-		if (read_string(opt, &pos, &len,
-					&opt->initrd_file))
-			goto out_err;
-		if (read_string(opt, &pos, &len,
-					&opt->boot_args))
-			goto out_err;
-
-		list_add(&dev->boot_options, &opt->list);
-	}
+out:
+	return rc;
+}
+
+int pb_protocol_deserialise_boot_option(struct boot_option *opt,
+		const struct pb_protocol_message *message)
+{
+	unsigned int len;
+	const char *pos;
+	int rc = -1;
+
+	len = message->payload_len;
+	pos = message->payload;
+
+	if (read_string(opt, &pos, &len, &opt->device_id))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->id))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->name))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->description))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->icon_file))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->boot_image_file))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->initrd_file))
+		goto out;
+
+	if (read_string(opt, &pos, &len, &opt->boot_args))
+		goto out;
+
+	rc = 0;
+
+out:
+	return rc;
+}
+
+int pb_protocol_deserialise_boot_command(struct boot_command *cmd,
+		const struct pb_protocol_message *message)
+{
+	unsigned int len;
+	const char *pos;
+	int rc = -1;
+
+	len = message->payload_len;
+	pos = message->payload;
+
+	if (read_string(cmd, &pos, &len, &cmd->option_id))
+		goto out;
+
+	if (read_string(cmd, &pos, &len, &cmd->boot_image_file))
+		goto out;
+
+	if (read_string(cmd, &pos, &len, &cmd->initrd_file))
+		goto out;
+
+	if (read_string(cmd, &pos, &len, &cmd->boot_args))
+		goto out;
 
-	return dev;
+	rc = 0;
 
-out_err:
-	talloc_free(dev);
-	return NULL;
+out:
+	return rc;
 }