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

Signed-off-by: Steve Rae <s...@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

_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
http://lists.denx.de/mailman/listinfo/u-boot

Reply via email to