1 files changed, 121 insertions, 0 deletions
diff --git a/common/cmd_blkdevparts.c b/common/cmd_blkdevparts.c
new file mode 100644
index 000000000..10b34e29d
--- /dev/null
+++ b/common/cmd_blkdevparts.c
@@ -0,0 +1,121 @@
+/*
+ * Copyright (C) ST-Ericsson SA 2010
+ * Author: Peter Nessrup <peter.nessrup@stericsson.com> for
+ * ST-Ericsson.
+ * License terms: GNU General Public License (GPL), version 2.
+ */
+
+/*
+ * Support for setting blkdevparts environment variable, to be passed to
+ * the Kernel on the kernel command line
+ */
+#include <common.h>
+#include <part.h>
+#include <malloc.h>
+
+#if !defined(CONFIG_TOC_PARTITION)
+#error TOC partition support must be selected
+#endif
+
+#define MMC_BLKDEV_ID			"mmcblk"
+#define MAX_BLKDEVPARTS_MAXLEN		256
+#define MAX_PARTITION_MAXLEN		20
+
+int do_blkdevparts(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
+{
+	disk_partition_t part_info;
+	block_dev_desc_t *dev_desc;
+	char *ep;
+	int part = 1;	/* Start with partition 1 */
+	char *blkdevparts;
+	char blkdev_buf[MAX_BLKDEVPARTS_MAXLEN];
+	char part_buf[MAX_PARTITION_MAXLEN];
+	int dev;
+	int m, n;
+	const char *env_p;
+
+	debug("do_blkdevparts\n");
+	memset(blkdev_buf, 0, sizeof(blkdev_buf));
+
+	if (argc < 3) {
+		cmd_usage(cmdtp);
+		return 1;
+	}
+
+	dev = (int) simple_strtoul(argv[2], &ep, 16);
+	dev_desc = get_dev(argv[1], dev);
+
+	if (dev_desc == NULL) {
+		printf("Invalid device\n");
+		return 1;
+	}
+
+	if (dev_desc->part_type != PART_TYPE_TOC) {
+		printf("Device is not TOC partitioned!\n");
+		return 1;
+	}
+
+	/* Only support sd/mmc interface */
+	switch (dev_desc->if_type) {
+	case IF_TYPE_MMC:
+	case IF_TYPE_SD:
+		n = sprintf(blkdev_buf, "%s", MMC_BLKDEV_ID);
+		break;
+	default:
+		printf("Interface is not supported: %s\n", argv[1]);
+		return 1;
+	}
+
+	/* Add the device */
+	n += sprintf(&blkdev_buf[n], "%d:", dev);
+
+	while (get_partition_info(dev_desc, part, &part_info) == 0) {
+		memset(part_buf, 0, sizeof(part_buf));
+		m = 0;
+		/*
+		 * Build up the bootargs environment
+		 * variable with blkdev info
+		 */
+		debug("Partition found - part: %d!\n", part);
+		/* Partition found */
+		if (part > 1)
+			/* Don't add a comma before the first partition */
+			m = sprintf(part_buf, "%s", ",");
+
+		sprintf(&part_buf[m], "%lu@%lu",
+			part_info.size, part_info.start);
+
+		/* Append the partition to the blkdev string */
+		if ((n + sizeof(part_buf)) <= sizeof(blkdev_buf))
+			n += sprintf(&blkdev_buf[n], "%s", part_buf);
+		else {
+			printf("No space to add partition to blkdevparts\n");
+			return 1;
+		}
+		part++;
+	};
+	debug("No more partitions\n");
+
+	/* Check if blkdevparts env var already existing */
+	env_p = getenv("blkdevparts");
+	if (env_p) {
+		debug("Appending to existing blkdevparts env var\n");
+		/* Allocate a buffer incl the added ;*/
+		blkdevparts = malloc(strlen(env_p) + strlen(blkdev_buf) + 1);
+		sprintf(blkdevparts, "%s;%s", env_p, blkdev_buf);
+		setenv("blkdevparts", blkdevparts);
+		/* free up the buffer */
+		free(blkdevparts);
+	} else
+		setenv("blkdevparts", blkdev_buf);
+
+	return 0;
+}
+
+U_BOOT_CMD(
+	blkdevparts,	3,	0,	do_blkdevparts,
+	"assembles partition info for kernel cmd line",
+	"<interface> <dev>\n"
+	"    - assembles the blkdevparts env var with partition information\n"
+	"      for kernel cmd line, read from the <dev> TOC on <interface>"
+);