- add support for 'fastboot flash' command for eMMC devices
Signed-off-by: Steve Rae srae@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);