From 0af8d0aac77f4df4bc7dadbcdea5d9a16f5f3e45 Mon Sep 17 00:00:00 2001 From: Weijie Gao Date: Mon, 25 Jul 2022 10:44:57 +0800 Subject: [PATCH 43/71] cmd: add nmbm command Add nmbm command for debugging, data operations and image-booting support Signed-off-by: Weijie Gao --- cmd/Kconfig | 6 + cmd/Makefile | 1 + cmd/nmbm.c | 327 +++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 334 insertions(+) create mode 100644 cmd/nmbm.c --- a/cmd/Kconfig +++ b/cmd/Kconfig @@ -1465,6 +1465,12 @@ config CMD_NAND_TORTURE endif # CMD_NAND +config CMD_NMBM + depends on NMBM_MTD + bool "nmbm" + help + NAND mapping block management (NMBM) utility + config CMD_NVME bool "nvme" depends on NVME --- a/cmd/Makefile +++ b/cmd/Makefile @@ -127,6 +127,7 @@ obj-y += legacy-mtd-utils.o endif obj-$(CONFIG_CMD_MUX) += mux.o obj-$(CONFIG_CMD_NAND) += nand.o +obj-$(CONFIG_CMD_NMBM) += nmbm.o obj-$(CONFIG_CMD_NET) += net.o obj-$(CONFIG_ENV_SUPPORT) += nvedit.o obj-$(CONFIG_CMD_NVEDIT_EFI) += nvedit_efi.o --- /dev/null +++ b/cmd/nmbm.c @@ -0,0 +1,327 @@ +// SPDX-License-Identifier: GPL-2.0 +/* + * Copyright (C) 2020 MediaTek Inc. All Rights Reserved. + * + * Author: Weijie Gao + */ + +#include +#include +#include +#include +#include +#include + +#include + +static int nmbm_parse_offset_size(struct mtd_info *mtd, char *off_str, + char *size_str, uint64_t *off, + uint64_t *size) +{ + char *end; + + *off = simple_strtoull(off_str, &end, 16); + if (end == off_str) { + printf("Error: offset '%s' is invalid\n", off_str); + return -EINVAL; + } + + if (*off >= mtd->size) { + printf("Error: offset '0x%llx' is beyond the end of device\n", + *off); + return -EINVAL; + } + + *size = simple_strtoull(size_str, &end, 16); + if (end == off_str) { + printf("Error: size '%s' is invalid\n", off_str); + return -EINVAL; + } + + if (*off + *size > mtd->size) { + printf("Error: size '0x%llx' is too large\n", *size); + return -EINVAL; + } + + return 0; +} + +static int do_nmbm_erase(struct mtd_info *mtd, uint64_t offset, uint64_t size) +{ + struct erase_info ei; + int ret; + + memset(&ei, 0, sizeof(ei)); + + ei.mtd = mtd; + ei.addr = offset; + ei.len = size; + + printf("Erasing from 0x%llx, size 0x%llx ...\n", offset, size); + + ret = mtd_erase(mtd, &ei); + + if (!ret) { + printf("Succeeded\n"); + return CMD_RET_SUCCESS; + } + + printf("Failed at 0x%llx\n", ei.fail_addr); + + return CMD_RET_FAILURE; +} + +static int do_nmbm_rw(int read, struct mtd_info *mtd, uintptr_t addr, + uint64_t offset, size_t size) +{ + size_t retlen; + int ret; + + printf("%s 0x%llx, size 0x%zx\n", read ? "Reading from" : "Writing to", + offset, size); + + if (read) + ret = mtd_read(mtd, offset, size, &retlen, (void *)addr); + else + ret = mtd_write(mtd, offset, size, &retlen, (void *)addr); + + if (!ret) { + printf("Succeeded\n"); + return CMD_RET_SUCCESS; + } + + printf("Failed at 0x%llx\n", offset + retlen); + + return CMD_RET_FAILURE; +} + +static int do_nmbm_mtd_boot(struct cmd_tbl *cmdtp, struct mtd_info *mtd, + int argc, char *const argv[]) +{ + bool print_image_contents = true; + uintptr_t loadaddr = image_load_addr; + char *end, *image_name; + const char *ep; + size_t retlen; + uint32_t size; + uint64_t off; + int ret; + +#if defined(CONFIG_CMD_MTDPARTS) + struct mtd_device *partdev; + struct mtd_info *partmtd; + struct part_info *part; + u8 pnum; +#endif + + ep = env_get("autostart"); + + if (ep && !strcmp(ep, "yes")) + print_image_contents = false; + + if (argc == 2) { + loadaddr = simple_strtoul(argv[0], &end, 0); + if (*end || end == argv[0]) { + printf("'%s' is not a valid address\n", argv[0]); + return CMD_RET_FAILURE; + } + + argc--; + argv++; + } + + off = simple_strtoull(argv[0], &end, 0); + if (*end || end == argv[0]) { +#if defined(CONFIG_CMD_MTDPARTS) + ret = mtdparts_init(); + if (ret) + return CMD_RET_FAILURE; + + ret = find_dev_and_part(argv[0], &partdev, &pnum, &part); + if (ret) + return CMD_RET_FAILURE; + + if (partdev->id->type != MTD_DEV_TYPE_NMBM) { + printf("'%s' is not a NMBM device partition\n", + argv[0]); + return CMD_RET_FAILURE; + } + + partmtd = nmbm_mtd_get_upper_by_index(partdev->id->num); + + if (partmtd != mtd) { + printf("'%s' does not belong to this device\n", + argv[0]); + return CMD_RET_FAILURE; + } + + off = part->offset; +#else + printf("'%s' is not a valid offset\n", argv[0]); + return CMD_RET_FAILURE; +#endif + } + + ret = mtd_read(mtd, off, sizeof(struct legacy_img_hdr), &retlen, + (void *)loadaddr); + if (ret || retlen != sizeof(struct legacy_img_hdr)) { + printf("Failed to read NMBM at offset 0x%08llx\n", off); + return CMD_RET_FAILURE; + } + + switch (genimg_get_format((void *)loadaddr)) { +#if defined(CONFIG_LEGACY_IMAGE_FORMAT) + case IMAGE_FORMAT_LEGACY: + size = image_get_image_size((struct legacy_img_hdr *)loadaddr); + image_name = "legacy"; + break; +#endif +#if defined(CONFIG_FIT) + case IMAGE_FORMAT_FIT: + size = fit_get_size((const void *)loadaddr); + image_name = "FIT"; + break; +#endif + default: + printf("Error: no Image found at offset 0x%08llx\n", off); + return CMD_RET_FAILURE; + } + + printf("Loading %s image at offset 0x%llx to memory 0x%08lx, size 0x%x ...\n", + image_name, off, loadaddr, size); + + ret = mtd_read(mtd, off, size, &retlen, (void *)loadaddr); + if (ret || retlen != size) { + printf("Error: Failed to load image at offset 0x%08llx\n", + off + retlen); + return CMD_RET_FAILURE; + } + + switch (genimg_get_format((void *)loadaddr)) { +#if defined(CONFIG_LEGACY_IMAGE_FORMAT) + case IMAGE_FORMAT_LEGACY: + if (print_image_contents) + image_print_contents((void *)loadaddr); + break; +#endif +#if defined(CONFIG_FIT) + case IMAGE_FORMAT_FIT: + if (print_image_contents) + fit_print_contents((void *)loadaddr); + break; +#endif + } + + image_load_addr = loadaddr; + + return bootm_maybe_autostart(cmdtp, "nmbm"); +} + +static int do_nmbm(struct cmd_tbl *cmdtp, int flag, int argc, + char *const argv[]) +{ + struct mtd_info *mtd; + uint64_t offset, size; + char *end; + uintptr_t addr; + int ret, all = 0; + + if (argc == 1) + return CMD_RET_USAGE; + + if (!strcmp(argv[1], "list")) { + nmbm_mtd_list_devices(); + return CMD_RET_SUCCESS; + } + + if (argc < 3) + return CMD_RET_USAGE; + + if (!strcmp(argv[2], "info")) + return !!nmbm_mtd_print_info(argv[1]); + + if (!strcmp(argv[2], "state")) + return !!nmbm_mtd_print_states(argv[1]); + + if (!strcmp(argv[2], "bad")) + return !!nmbm_mtd_print_bad_blocks(argv[1]); + + if (!strcmp(argv[2], "mapping")) { + if (argc >= 4) { + if (!strcmp(argv[3], "all")) + all = 1; + } + + return nmbm_mtd_print_mappings(argv[1], all); + } + + if (argc < 4) + return CMD_RET_USAGE; + + mtd = get_mtd_device_nm(argv[1]); + if (IS_ERR(mtd)) { + printf("Error: NMBM device '%s' not found\n", argv[1]); + return CMD_RET_FAILURE; + } + + if (mtd->type != MTD_DEV_TYPE_NMBM) { + printf("Error: '%s' is not a NMBM device\n", argv[1]); + return CMD_RET_FAILURE; + } + + if (!strcmp(argv[2], "boot")) + return do_nmbm_mtd_boot(cmdtp, mtd, argc - 3, argv + 3); + + if (argc < 5) + return CMD_RET_USAGE; + + if (!strcmp(argv[2], "erase")) { + ret = nmbm_parse_offset_size(mtd, argv[3], argv[4], &offset, + &size); + if (ret) + return CMD_RET_FAILURE; + + return do_nmbm_erase(mtd, offset, size); + } + + if (argc < 6) + return CMD_RET_USAGE; + + ret = nmbm_parse_offset_size(mtd, argv[4], argv[5], &offset, &size); + if (ret) + return CMD_RET_FAILURE; + + if (size > SIZE_MAX) { + printf("Error: size 0x%llx is too large\n", size); + return -EINVAL; + } + + addr = simple_strtoul(argv[3], &end, 16); + if (end == argv[3]) { + printf("Error: addr '%s' is invalid\n", argv[3]); + return -EINVAL; + } + + if (!strcmp(argv[2], "read")) + return do_nmbm_rw(1, mtd, addr, offset, (size_t)size); + + if (!strcmp(argv[2], "write")) + return do_nmbm_rw(0, mtd, addr, offset, (size_t)size); + + return CMD_RET_USAGE; +} + +U_BOOT_CMD( + nmbm, CONFIG_SYS_MAXARGS, 0, do_nmbm, + "NMBM utility commands", + "\n" + "nmbm list - List NMBM devices\n" + "nmbm info - Display NMBM information\n" + "nmbm state - Display block states\n" + "nmbm bad - Display bad blocks\n" + "nmbm boot - Boot from NMBM\n" + "nmbm mapping [all] - Display block mapping\n" + "nmbm erase - Erase blocks\n" + "nmbm read - Read data\n" + "nmbm write - Write data\n" +);