[U-Boot] [RFC PATCH 2/3] usb/gadget: fastboot: add eMMC support for flash command

[Steve Rae]

- add support for 'fastboot flash' command for eMMC devices

Signed-off-by: Steve Rae <srae at broadcom.com>
---
I suspect that the "sparse image" handling (ie. the "while (remaining_chunks)" loop)
has been implemented elsewhere -- I need help finding the original code to determine
any licensing issues....
Thanks, Steve

 common/Makefile  |   5 ++
 common/fb_mmc.c  | 144 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 include/fb_mmc.h |   8 ++++
 3 files changed, 157 insertions(+)
 create mode 100644 common/fb_mmc.c
 create mode 100644 include/fb_mmc.h

diff --git a/common/Makefile b/common/Makefile
index f6cd980..c825c2c 100644
--- a/common/Makefile
+++ b/common/Makefile
@@ -264,4 +264,9 @@ obj-$(CONFIG_FIT_SIGNATURE) += image-sig.o
 obj-y += memsize.o
 obj-y += stdio.o
 
+# This option is not just y/n - it can have a numeric value
+ifdef CONFIG_FASTBOOT_FLASH_MMC_DEV
+obj-y += fb_mmc.o
+endif
+
 CFLAGS_env_embedded.o := -Wa,--no-warn -DENV_CRC=$(shell tools/envcrc 2>/dev/null)
diff --git a/common/fb_mmc.c b/common/fb_mmc.c
new file mode 100644
index 0000000..de6a4c7
--- /dev/null
+++ b/common/fb_mmc.c
@@ -0,0 +1,144 @@
+/*
+ * Copyright TODO
+ *
+ * SPDX-License-Identifier:	GPL-2.0+
+ */
+
+#include <common.h>
+#include <fb_mmc.h>
+#include <mmc.h>
+#include <sparse_format.h>
+
+void fb_mmc_flash_write(const char *cmd, void *download_buffer,
+			unsigned int download_bytes, char *response)
+{
+	int ret;
+	block_dev_desc_t *mmc_dev;
+	disk_partition_t info;
+	lbaint_t blk;
+	lbaint_t blkcnt;
+	ulong blks;
+	sparse_header_t *s_header = (sparse_header_t *)download_buffer;
+	chunk_header_t *c_header;
+	void *buffer;
+	uint32_t blk_sz;
+	uint32_t remaining_chunks;
+	uint32_t bytes_written = 0;
+
+	mmc_dev = mmc_get_dev(CONFIG_FASTBOOT_FLASH_MMC_DEV);
+	if (!mmc_dev || mmc_dev->type == DEV_TYPE_UNKNOWN) {
+		printf("%s: invalid mmc device\n", __func__);
+		strcpy(response, "FAILinvalid mmc device");
+		return;
+	}
+
+	ret = get_partition_info_efi_by_name(mmc_dev, cmd, &info);
+	if (ret) {
+		printf("%s: cannot find partition: '%s'\n", __func__, cmd);
+		strcpy(response, "FAILcannot find partition");
+		return;
+	}
+
+	if ((le32_to_cpu(s_header->magic) == SPARSE_HEADER_MAGIC) &&
+	    (le16_to_cpu(s_header->major_version) == 1)) {
+		/* sparse image */
+
+		blk_sz = le32_to_cpu(s_header->blk_sz);
+
+		/* verify s_header->blk_sz is exact multiple of info.blksz */
+		if (blk_sz != (blk_sz & ~(info.blksz - 1))) {
+			printf("%s: Sparse image block size issue [%u]\n",
+			       __func__, blk_sz);
+			strcpy(response, "FAILsparse image block size issue");
+			return;
+		}
+
+		if ((le32_to_cpu(s_header->total_blks) * blk_sz) >
+		    (info.size * info.blksz)) {
+			printf("%s: Sparse image is too large for the partition\n",
+			       __func__);
+			strcpy(response, "FAILsparse image is too large");
+			return;
+		}
+
+		printf("Flashing Sparse Image\n");
+
+		blk = info.start;
+		remaining_chunks = le32_to_cpu(s_header->total_chunks);
+		c_header = (chunk_header_t *)(download_buffer +
+		    le16_to_cpu(s_header->file_hdr_sz));
+		while (remaining_chunks) {
+			switch (le16_to_cpu(c_header->chunk_type)) {
+			case CHUNK_TYPE_RAW:
+				blkcnt =
+				    (le32_to_cpu(c_header->chunk_sz) * blk_sz) /
+				    info.blksz;
+				buffer =
+				    (void *)c_header +
+				    le16_to_cpu(s_header->chunk_hdr_sz);
+
+				blks = mmc_dev->block_write(mmc_dev->dev, blk,
+							    blkcnt, buffer);
+				if (blks != blkcnt) {
+					printf("Write failed %lu\n", blks);
+					strcpy(response,
+					       "FAILmmc write failure");
+					return;
+				}
+
+				bytes_written += blkcnt * info.blksz;
+				break;
+
+			case CHUNK_TYPE_FILL:
+			case CHUNK_TYPE_DONT_CARE:
+			case CHUNK_TYPE_CRC32:
+				/* do nothing */
+				break;
+
+			default:
+				/* error */
+				printf("Unknown chunk type\n");
+				strcpy(response,
+				       "FAILunknown chunk type in sparse image");
+				return;
+			}
+
+			blk += (le32_to_cpu(c_header->chunk_sz) * blk_sz) /
+			    info.blksz;
+			c_header = (chunk_header_t *)((void *)c_header +
+			    le32_to_cpu(c_header->total_sz));
+			remaining_chunks--;
+		}
+
+		printf("........ wrote %u bytes to '%s'\n", bytes_written, cmd);
+	} else {
+		/* raw image */
+
+		/* determine number of blocks to write */
+		blkcnt =
+		    ((download_bytes + (info.blksz - 1)) & ~(info.blksz - 1));
+		blkcnt = blkcnt / info.blksz;
+
+		if (blkcnt > info.size) {
+			printf("%s: too large for partition: '%s'\n",
+			       __func__, cmd);
+			strcpy(response, "FAILtoo large for partition");
+			return;
+		}
+
+		printf("Flashing Raw Image\n");
+
+		blks = mmc_dev->block_write(mmc_dev->dev, info.start, blkcnt,
+					    download_buffer);
+		if (blks != blkcnt) {
+			printf("%s: failed writing to mmc device %d\n",
+			       __func__, mmc_dev->dev);
+			strcpy(response, "FAILfailed writing to mmc device");
+			return;
+		}
+
+		printf("........ wrote %lu bytes to '%s'\n",
+		       blkcnt * info.blksz, cmd);
+	}
+	strcpy(response, "OKAY");
+}
diff --git a/include/fb_mmc.h b/include/fb_mmc.h
new file mode 100644
index 0000000..1ad1d13
--- /dev/null
+++ b/include/fb_mmc.h
@@ -0,0 +1,8 @@
+/*
+ * Copyright 2014 Broadcom Corporation.
+ *
+ * SPDX-License-Identifier:	GPL-2.0+
+ */
+
+void fb_mmc_flash_write(const char *cmd, void *download_buffer,
+			unsigned int download_bytes, char *response);
-- 
1.8.5