| /* |
| * (C) Copyright 2003 |
| * Kyle Harris, kharris@nexus-tech.net |
| * |
| * See file CREDITS for list of people who contributed to this |
| * project. |
| * |
| * This program is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU General Public License as |
| * published by the Free Software Foundation; either version 2 of |
| * the License, or (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, |
| * MA 02111-1307 USA |
| */ |
| |
| #include <common.h> |
| #include <malloc.h> |
| #include <command.h> |
| #include <linux/ctype.h> |
| #include <mmc.h> |
| #include "../drivers/mmc/mmc_private.h" |
| #include <partition_table.h> |
| #include <emmc_partitions.h> |
| #include <asm/arch/cpu_sdio.h> |
| #include <asm/arch/sd_emmc.h> |
| #include <linux/sizes.h> |
| #include <asm/arch-g12a/cpu_id.h> |
| #include <amlogic/aml_mmc.h> |
| |
| /* info system. */ |
| #define dtb_err(fmt, ...) printf( "%s()-%d: " fmt , \ |
| __func__, __LINE__, ##__VA_ARGS__) |
| |
| #define dtb_wrn(fmt, ...) printf( "%s()-%d: " fmt , \ |
| __func__, __LINE__, ##__VA_ARGS__) |
| |
| /* for detail debug info */ |
| #define dtb_info(fmt, ...) printf( "%s()-%d: " fmt , \ |
| __func__, __LINE__, ##__VA_ARGS__) |
| |
| #define fb_err(fmt, ...) printf("%s()-%d: " fmt , \ |
| __func__, __LINE__, ##__VA_ARGS__) |
| #define CONFIG_SYS_MMC_ENV_DEV 1 |
| struct aml_dtb_rsv { |
| u8 data[DTB_BLK_SIZE*DTB_BLK_CNT - 4*sizeof(u32)]; |
| u32 magic; |
| u32 version; |
| u32 timestamp; |
| u32 checksum; |
| }; |
| |
| struct aml_dtb_info { |
| u32 stamp[2]; |
| u8 valid[2]; |
| }; |
| |
| #define stamp_after(a,b) ((int)(b) - (int)(a) < 0) |
| /* glb dtb infos */ |
| static struct aml_dtb_info dtb_infos = {{0, 0}, {0, 0}}; |
| |
| #define CONFIG_SECURITYKEY |
| |
| #if !defined(CONFIG_SYS_MMC_BOOT_DEV) |
| #define CONFIG_SYS_MMC_BOOT_DEV (CONFIG_SYS_MMC_ENV_DEV) |
| #endif |
| |
| #define GXB_START_BLK 0 |
| #define GXL_START_BLK 1 |
| |
| /* max 2MB for emmc in blks */ |
| #define UBOOT_SIZE (0x1000) |
| |
| |
| extern int mmc_key_write(unsigned char *buf, unsigned int size, uint32_t *actual_lenth); |
| extern int mmc_key_read(unsigned char *buf, unsigned int size, uint32_t *actual_lenth); |
| |
| int info_disprotect = 0; |
| |
| |
| int mmc_read_status(struct mmc *mmc, int timeout) |
| { |
| struct mmc_cmd cmd; |
| int err, retries = 5; |
| int status; |
| |
| cmd.cmdidx = MMC_CMD_SEND_STATUS; |
| cmd.resp_type = MMC_RSP_R1; |
| if (!mmc_host_is_spi(mmc)) |
| cmd.cmdarg = mmc->rca << 16; |
| do { |
| err = mmc_send_cmd(mmc, &cmd, NULL); |
| if (!err) { |
| if ((cmd.response[0] & MMC_STATUS_RDY_FOR_DATA) && |
| (cmd.response[0] & MMC_STATUS_CURR_STATE) != |
| MMC_STATE_PRG) |
| break; |
| else if (cmd.response[0] & MMC_STATUS_MASK) { |
| printf("Status Error: 0x%08X\n", |
| cmd.response[0]); |
| return -18; |
| } |
| } else if (--retries < 0) |
| return err; |
| |
| udelay(1000); |
| |
| } while (timeout--); |
| |
| status = (cmd.response[0] & MMC_STATUS_CURR_STATE) >> 9; |
| printf("CURR STATE:%d, status = 0x%x\n", status, cmd.response[0]); |
| |
| if (timeout <= 0) { |
| printf("read status Timeout waiting card ready\n"); |
| return -19; |
| } |
| if (cmd.response[0] & MMC_STATUS_SWITCH_ERROR) { |
| printf("mmc status swwitch error status =0x%x\n", status); |
| return -21; |
| } |
| return 0; |
| } |
| |
| bool emmckey_is_protected (struct mmc *mmc) |
| { |
| #ifdef CONFIG_STORE_COMPATIBLE |
| #ifdef CONFIG_SECURITYKEY |
| if (info_disprotect & DISPROTECT_KEY) { |
| printf("%s(): disprotect\n", __func__); |
| return 0; |
| }else{ |
| printf("%s(): protect\n", __func__); |
| return 1; |
| } |
| #else |
| return 0; |
| #endif |
| #else |
| #ifdef CONFIG_SECURITYKEY |
| //return mmc->key_protect; |
| return 0; /* fixme, */ |
| #else |
| return 0; |
| #endif |
| #endif |
| } |
| |
| unsigned emmc_cur_partition = 0; |
| |
| |
| static int get_off_size(struct mmc * mmc, char * name, uint64_t offset, uint64_t size, u64 * blk, u64 * cnt, u64 * sz_byte) |
| { |
| struct partitions *part_info = NULL; |
| uint64_t off = 0; |
| int blk_shift = 0; |
| |
| blk_shift = ffs(mmc->read_bl_len) - 1; |
| // printf("blk_shift:%d , off:0x%llx , size:0x%llx.\n ",blk_shift,off,size ); |
| part_info = find_mmc_partition_by_name(name); |
| if (part_info == NULL) { |
| printf("get partition info failed !!\n"); |
| return -1; |
| } |
| off = part_info->offset + offset; |
| |
| // printf("part_info->offset:0x%llx , off:0x%llx , size:0x%llx.\n",part_info->offset ,off,size); |
| |
| *blk = off >> blk_shift ; |
| *cnt = size >> blk_shift ; |
| *sz_byte = size - ((*cnt) << blk_shift) ; |
| |
| // printf("get_partition_off_size : blk:0x%llx , cnt:0x%llx.\n",*blk,*cnt); |
| return 0; |
| } |
| |
| static int get_partition_size(unsigned char* name, uint64_t* addr) |
| { |
| struct partitions *part_info = NULL; |
| part_info = find_mmc_partition_by_name((char *)name); |
| if (part_info == NULL) { |
| printf("get partition info failed !!\n"); |
| return -1; |
| } |
| |
| *addr = part_info->size >> 9; // unit: 512 bytes |
| return 0; |
| } |
| |
| static inline int isstring(char *p) |
| { |
| char *endptr = p; |
| while (*endptr != '\0') { |
| if (!(((*endptr >= '0') && (*endptr <= '9')) |
| || ((*endptr >= 'a') && (*endptr <= 'f')) |
| || ((*endptr >= 'A') && (*endptr <= 'F')) |
| || (*endptr == 'x') || (*endptr == 'X'))) |
| return 1; |
| endptr++; |
| } |
| |
| return 0; |
| } |
| |
| /* |
| erase bootloader on user/boot0/boot1 which indicate by map. |
| bit 0: user |
| bit 1: boot0 |
| bit 2: boot1 |
| */ |
| int amlmmc_erase_bootloader(int dev, int map) |
| { |
| int ret = 0, i, count = 3, nonius = MMC_PART_USER; |
| int blk_shift; |
| unsigned long n; |
| char *partname[3] = {"user", "boot0", "boot1"}; |
| cpu_id_t cpu_id = get_cpu_id(); |
| struct mmc *mmc = find_mmc_device(dev); |
| |
| #if defined(CONFIG_AML_GPT) |
| nonius = MMC_PART_BOOT0; |
| #endif |
| |
| /* do nothing */ |
| if (0 == map) |
| goto _out; |
| |
| if (!mmc) { |
| printf("%s() %d: not valid emmc %d\n", __func__, __LINE__, dev); |
| ret = -1; |
| goto _out; |
| } |
| /* make sure mmc is initilized! */ |
| ret = mmc_init(mmc); |
| if (ret) { |
| printf("%s() %d: emmc %d init %d\n", __func__, __LINE__, dev, ret); |
| ret = -2; |
| goto _out; |
| } |
| |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| /* erase bootloader in user/boot0/boot1 */ |
| for (i = nonius; i < count; i++) { |
| if (map & (0x1 << i)) { |
| if (!blk_select_hwpart_devnum(IF_TYPE_MMC, 1, i)) { |
| lbaint_t start = 0, blkcnt; |
| |
| blkcnt = mmc->capacity >> blk_shift; |
| if (0 == i) { |
| struct partitions *part_info; |
| /* get info by partition */ |
| part_info = find_mmc_partition_by_name(MMC_BOOT_NAME); |
| if (part_info == NULL) { |
| printf("%s() %d: error!!\n", __func__, __LINE__); |
| /* fixme, do somthing! */ |
| continue; |
| } else { |
| start = part_info->offset>> blk_shift; |
| blkcnt = part_info->size>> blk_shift; |
| if (cpu_id.family_id >= MESON_CPU_MAJOR_ID_GXL) { |
| start = GXL_START_BLK; |
| blkcnt -= GXL_START_BLK; |
| } |
| } |
| } |
| /* some customer may use boot1 higher 2M as private data. */ |
| #ifdef CONFIG_EMMC_BOOT1_TOUCH_REGION |
| if (2 == i && CONFIG_EMMC_BOOT1_TOUCH_REGION <= mmc->capacity) { |
| blkcnt = CONFIG_EMMC_BOOT1_TOUCH_REGION >> blk_shift; |
| } |
| #endif/* CONFIG_EMMC_BOOT1_TOUCH_REGION */ |
| #ifdef CONFIG_AML_GPT |
| if (i == 0) |
| continue; |
| #endif |
| printf("Erasing blocks " LBAFU " to " LBAFU " @ %s\n", |
| start, blkcnt, partname[i]); |
| n = blk_derase(mmc_get_blk_desc(mmc), start, blkcnt); |
| if (n != 0) { |
| printf("mmc erase %s failed\n", partname[i]); |
| ret = -3; |
| break; |
| } |
| } else |
| printf("%s() %d: switch dev %d to %s fail\n", |
| __func__, __LINE__, dev, partname[i]); |
| } |
| } |
| /* try to switch back to user. */ |
| blk_select_hwpart(mmc->dev, 0); |
| |
| _out: |
| return ret; |
| } |
| |
| /* |
| write bootloader on user/boot0/boot1 which indicate by map. |
| bit 0: user |
| bit 1: boot0 |
| bit 2: boot1 |
| */ |
| int amlmmc_write_bootloader(int dev, int map, unsigned int size, const void *src) |
| { |
| int ret = 0, i, count = 3, nonius = MMC_PART_USER; |
| unsigned long n; |
| char *partname[3] = {"user", "boot0", "boot1"}; |
| struct mmc *mmc = find_mmc_device(dev); |
| lbaint_t start = GXB_START_BLK, blkcnt; |
| cpu_id_t cpu_id = get_cpu_id(); |
| |
| #if defined(CONFIG_AML_GPT) |
| nonius = MMC_PART_BOOT0; |
| #endif |
| |
| /* do nothing */ |
| if (0 == map) |
| goto _out; |
| |
| if (!mmc) { |
| printf("%s() %d: not valid emmc %d\n", __func__, __LINE__, dev); |
| ret = -1; |
| goto _out; |
| } |
| /* make sure mmc is initilized! */ |
| ret = mmc_init(mmc); |
| if (ret) { |
| printf("%s() %d: emmc %d init %d\n", __func__, __LINE__, dev, ret); |
| ret = -2; |
| goto _out; |
| } |
| |
| if (cpu_id.family_id >= MESON_CPU_MAJOR_ID_GXL) |
| start = GXL_START_BLK; |
| blkcnt = (size + mmc->read_bl_len - 1) / mmc->read_bl_len; |
| |
| /* erase bootloader in user/boot0/boot1 */ |
| for (i = nonius; i < count; i++) { |
| if (map & (0x1 << i)) { |
| if (!blk_select_hwpart(mmc->dev,i)) { |
| /* some customer may use boot1 higher 2M as private data. */ |
| #ifdef CONFIG_EMMC_BOOT1_TOUCH_REGION |
| if (2 == i && CONFIG_EMMC_BOOT1_TOUCH_REGION <= size) { |
| printf("%s(), size %d exceeds TOUCH_REGION %d, skip\n", |
| __func__, size, CONFIG_EMMC_BOOT1_TOUCH_REGION); |
| break; |
| } |
| #endif /* CONFIG_EMMC_BOOT1_TOUCH_REGION */ |
| #ifdef CONFIG_AML_GPT |
| if (i == 0) |
| continue; |
| #endif |
| printf("Wrting blocks " LBAFU " to " LBAFU " @ %s\n", |
| start, blkcnt, partname[i]); |
| n = blk_dwrite(mmc_get_blk_desc(mmc), start, blkcnt, src); |
| if (n != blkcnt) { |
| printf("mmc write %s failed\n", partname[i]); |
| ret = -3; |
| break; |
| } |
| } else |
| printf("%s() %d: switch dev %d to %s fail\n", |
| __func__, __LINE__, dev, partname[i]); |
| } |
| } |
| /* try to switch back to user. */ |
| blk_select_hwpart(mmc->dev, 0); |
| |
| _out: |
| return ret; |
| } |
| |
| static int amlmmc_erase_in_dev(int argc, char *const argv[]) |
| { |
| int dev = 0; |
| u64 cnt = 0, blk = 0, n = 0; |
| struct mmc *mmc; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| blk = simple_strtoull(argv[3], NULL, 16); |
| cnt = simple_strtoull(argv[4], NULL, 16); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| printf("MMC erase: dev # %d, start_erase_address(in block) # %#llx,\ |
| several blocks # %lld will be erased ...\n ", |
| dev, blk, cnt); |
| |
| mmc_init(mmc); |
| |
| if (cnt != 0) |
| n = blk_derase(mmc_get_blk_desc(mmc), blk, cnt); |
| |
| printf("dev # %d, %s, several blocks erased %s\n", |
| dev, " ", (n == 0) ? "OK" : "ERROR"); |
| |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_in_card(int argc, char *const argv[]) |
| { |
| int dev = 0; |
| u64 cnt = 0, blk = 0, n = 0; |
| /*sz_byte =0;*/ |
| char *name = NULL; |
| u64 offset_addr = 0, size = 0; |
| struct mmc *mmc; |
| int tmp_shift; |
| |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name (name); |
| offset_addr = simple_strtoull(argv[3], NULL, 16); |
| size = simple_strtoull(argv[4], NULL, 16); |
| mmc = find_mmc_device(dev); |
| |
| tmp_shift = ffs(mmc->read_bl_len) -1; |
| cnt = size >> tmp_shift; |
| blk = offset_addr >> tmp_shift; |
| /* sz_byte = size - (cnt<<tmp_shift); */ |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| printf("MMC erase: dev # %d, start_erase_address(in block) # %#llx,\ |
| several blocks # %lld will be erased ...\n ", |
| dev, blk, cnt); |
| |
| mmc_init(mmc); |
| |
| if (cnt != 0) |
| n = blk_derase(mmc_get_blk_desc(mmc), blk, cnt); |
| |
| printf("dev # %d, %s, several blocks erased %s\n", |
| dev, argv[2], (n == 0) ? "OK" : "ERROR"); |
| |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_in_part(int argc, char *const argv[]) |
| { |
| int dev = 0; |
| u64 cnt = 0, blk = 0, n = 0, sz_byte =0; |
| char *name = NULL; |
| u64 offset_addr = 0, size = 0; |
| struct mmc *mmc; |
| struct partitions *part_info; |
| |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name (name); |
| offset_addr = simple_strtoull(argv[3], NULL, 16); |
| size = simple_strtoull(argv[4], NULL, 16); |
| part_info = find_mmc_partition_by_name(name); |
| mmc = find_mmc_device(dev); |
| |
| if (offset_addr >= part_info->size) { |
| printf("Start address out #%s# partition'address region,(addr_byte < 0x%llx)\n", |
| name, part_info->size); |
| return 1; |
| } |
| if ((offset_addr+size) > part_info->size) { |
| printf("End address exceeds #%s# partition,(offset = 0x%llx,size = 0x%llx)\n", |
| name, part_info->offset,part_info->size); |
| return 1; |
| } |
| get_off_size(mmc, name, offset_addr, size, &blk, &cnt, &sz_byte); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| |
| if (!mmc) |
| return 1; |
| |
| printf("MMC erase: dev # %d, start_erase_address(in block) # %#llx,\ |
| several blocks # %lld will be erased ...\n ", |
| dev, blk, cnt); |
| |
| mmc_init(mmc); |
| |
| if (cnt != 0) |
| n = blk_derase(mmc_get_blk_desc(mmc), blk, cnt); |
| |
| printf("dev # %d, %s, several blocks erased %s\n", |
| dev, argv[2], (n == 0) ? "OK" : "ERROR"); |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_by_add(int argc, char *const argv[]) |
| { |
| int ret = 0; |
| |
| if (argc != 5) |
| return CMD_RET_USAGE; |
| |
| if (isdigit(argv[2][0])) |
| ret = amlmmc_erase_in_dev(argc, argv); |
| else if (strcmp(argv[2], "card") == 0) |
| ret = amlmmc_erase_in_card(argc, argv); |
| else if (isstring(argv[2])) |
| ret = amlmmc_erase_in_part(argc, argv); |
| |
| return ret; |
| } |
| |
| static int amlmmc_erase_non_loader(int argc, char *const argv[]) |
| { |
| int dev; |
| u32 n = 0; |
| int blk_shift; |
| u64 blk = 0, start_blk = 0; |
| struct partitions *part_info; |
| struct mmc *mmc; |
| |
| dev = 1; |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| mmc_init(mmc); |
| |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| part_info = find_mmc_partition_by_name(MMC_BOOT_NAME); |
| |
| if (part_info == NULL) { |
| start_blk = 0; |
| printf("no uboot partition for eMMC boot, just erase from 0\n"); |
| } |
| else |
| start_blk = (part_info->offset + part_info->size) >> blk_shift; |
| |
| if (emmckey_is_protected(mmc)) { |
| part_info = find_mmc_partition_by_name(MMC_RESERVED_NAME); |
| if (part_info == NULL) { |
| return 1; |
| } |
| blk = part_info->offset; |
| // it means: there should be other partitions before reserve-partition. |
| if (blk > 0) |
| blk -= PARTITION_RESERVED; |
| blk >>= blk_shift; |
| blk -= start_blk; |
| // (1) erase all the area before reserve-partition |
| if (blk > 0) |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, blk); |
| if (n == 0) { // not error |
| // (2) erase all the area after reserve-partition |
| start_blk = (part_info->offset + part_info->size + PARTITION_RESERVED) |
| >> blk_shift; |
| u64 erase_cnt = (mmc->capacity >> blk_shift) - start_blk; |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, erase_cnt); |
| } |
| } else { |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, 0); // erase the whole card |
| } |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_single_part(int argc, char *const argv[]) |
| { |
| char *name = NULL; |
| int dev; |
| u32 n = 0; |
| int blk_shift; |
| u64 cnt = 0, blk = 0; |
| struct partitions *part_info; |
| struct mmc *mmc; |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name(name); |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| mmc_init(mmc); |
| |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| if (emmckey_is_protected(mmc) |
| && (strncmp(name, MMC_RESERVED_NAME, sizeof(MMC_RESERVED_NAME)) == 0x00)) { |
| printf("\"%s-partition\" is been protecting and should no be erased!\n", |
| MMC_RESERVED_NAME); |
| return 1; |
| } |
| |
| part_info = find_mmc_partition_by_name(name); |
| if (part_info == NULL) { |
| return 1; |
| } |
| |
| blk = part_info->offset >> blk_shift; |
| if (emmc_cur_partition && !strncmp(name, "bootloader", strlen("bootloader"))) |
| cnt = 0; |
| else |
| cnt = part_info->size >> blk_shift; |
| n = blk_derase(mmc_get_blk_desc(mmc), blk, cnt); |
| |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_whole(int argc, char *const argv[]) |
| { |
| int dev; |
| u32 n = 0; |
| int blk_shift; |
| //u64 cnt = 0, |
| u64 blk = 0, start_blk = 0; |
| struct partitions *part_info; |
| struct mmc *mmc; |
| int map; |
| |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| mmc_init(mmc); |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| start_blk = 0; |
| |
| if (emmckey_is_protected(mmc)) { |
| part_info = find_mmc_partition_by_name(MMC_RESERVED_NAME); |
| if (part_info == NULL) { |
| return 1; |
| } |
| blk = part_info->offset; |
| // it means: there should be other partitions before reserve-partition. |
| if (blk > 0) |
| blk -= PARTITION_RESERVED; |
| blk >>= blk_shift; |
| blk -= start_blk; |
| // (1) erase all the area before reserve-partition |
| if (blk > 0) |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, blk); |
| if (n == 0) { // not error |
| // (2) erase all the area after reserve-partition |
| start_blk = (part_info->offset + part_info->size + PARTITION_RESERVED) |
| >> blk_shift; |
| u64 erase_cnt = (mmc->capacity >> blk_shift) - start_blk; |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, erase_cnt); |
| } |
| } else { |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, 0); // erase the whole card |
| } |
| map = AML_BL_BOOT; |
| if (n == 0) |
| n = amlmmc_erase_bootloader(dev, map); |
| if (n) |
| printf("erase bootloader in boot partition failed\n"); |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_non_cache(int arc, char *const argv[]) |
| { |
| int dev; |
| u32 n = 0; |
| int blk_shift; |
| u64 blk = 0, start_blk = 0; |
| struct partitions *part_info; |
| struct mmc *mmc; |
| int map; |
| |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| mmc_init(mmc); |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| if (emmckey_is_protected(mmc)) { |
| part_info = find_mmc_partition_by_name(MMC_RESERVED_NAME); |
| if (part_info == NULL) { |
| return 1; |
| } |
| |
| blk = part_info->offset; |
| // it means: there should be other partitions before reserve-partition. |
| if (blk > 0) { |
| blk -= PARTITION_RESERVED; |
| } |
| blk >>= blk_shift; |
| blk -= start_blk; |
| // (1) erase all the area before reserve-partition |
| if (blk > 0) { |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, blk); |
| // printf("(1) erase blk: 0 --> %llx %s\n", blk, (n == 0) ? "OK" : "ERROR"); |
| } |
| if (n == 0) { // not error |
| // (2) erase all the area after reserve-partition |
| part_info = find_mmc_partition_by_name(MMC_CACHE_NAME); |
| if (part_info == NULL) { |
| return 1; |
| } |
| start_blk = (part_info->offset + part_info->size + PARTITION_RESERVED) |
| >> blk_shift; |
| u64 erase_cnt = (mmc->capacity >> blk_shift) - start_blk; |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, erase_cnt); |
| } |
| } else { |
| n = blk_derase(mmc_get_blk_desc(mmc), start_blk, 0); // erase the whole card |
| } |
| map = AML_BL_BOOT; |
| if (n == 0) { |
| n = amlmmc_erase_bootloader(dev, map); |
| if (n) |
| printf("erase bootloader in boot partition failed\n"); |
| } |
| return (n == 0) ? 0 : 1; |
| } |
| |
| static int amlmmc_erase_dev(int argc, char *const argv[]) |
| { |
| return amlmmc_erase_whole(argc, argv); |
| } |
| |
| static int amlmmc_erase_allbootloader(int argc, char*const argv[]) |
| { |
| int map; |
| int rc; |
| int dev; |
| map = AML_BL_ALL; |
| |
| #if defined(CONFIG_AML_GPT) |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| #else |
| name = "bootloader"; |
| dev = find_dev_num_by_partition_name(name); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| #endif |
| rc = amlmmc_erase_bootloader(dev, map); |
| return rc; |
| } |
| |
| static int amlmmc_erase_by_part(int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| |
| if (argc != 3) |
| return ret; |
| |
| if (isdigit(argv[2][0])) |
| ret = amlmmc_erase_dev(argc, argv); |
| else if (strcmp(argv[2], "whole") == 0) |
| ret = amlmmc_erase_whole(argc, argv); |
| else if (strcmp(argv[2], "non_cache") == 0) |
| ret = amlmmc_erase_non_cache(argc, argv); |
| else if (strcmp(argv[2], "non_loader") == 0) |
| ret = amlmmc_erase_non_loader(argc, argv); |
| else if (strcmp(argv[2], "allbootloader") == 0) |
| ret = amlmmc_erase_allbootloader(argc, argv); |
| else |
| ret = amlmmc_erase_single_part(argc, argv); |
| return ret; |
| } |
| |
| static int do_amlmmc_erase(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| |
| if (argc == 3) |
| ret = amlmmc_erase_by_part(argc, argv); |
| else if (argc == 5) |
| ret = amlmmc_erase_by_add(argc, argv); |
| |
| return ret; |
| } |
| |
| static int amlmmc_write_in_part(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| u64 cnt = 0, n = 0, blk = 0, sz_byte = 0; |
| char *name = NULL; |
| u64 offset = 0, size = 0; |
| cpu_id_t cpu_id = get_cpu_id(); |
| struct mmc *mmc; |
| |
| name = argv[2]; |
| if (strcmp(name, "bootloader") == 0) |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| else |
| dev = find_dev_num_by_partition_name (name); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| offset = simple_strtoull(argv[4], NULL, 16); |
| size = simple_strtoull(argv[5], NULL, 16); |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| if (strcmp(name, "bootloader") == 0) { |
| cnt = UBOOT_SIZE; |
| if (cpu_id.family_id >= MESON_CPU_MAJOR_ID_GXL) { |
| blk = GXL_START_BLK; |
| cnt -= GXL_START_BLK; |
| } |
| else |
| blk = GXB_START_BLK; |
| } else |
| get_off_size(mmc, name, offset, size, &blk, &cnt, &sz_byte); |
| |
| mmc_init(mmc); |
| n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| //write sz_byte bytes |
| if ((n == cnt) && (sz_byte != 0)) { |
| // printf("sz_byte=%#llx bytes\n",sz_byte); |
| void *addr_tmp = malloc(mmc->write_bl_len); |
| void *addr_byte = (void*)(addr+cnt*(mmc->write_bl_len)); |
| ulong start_blk = blk+cnt; |
| |
| if (addr_tmp == NULL) { |
| printf("mmc write: malloc fail\n"); |
| return 1; |
| } |
| |
| if (blk_dread(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // read 1 block |
| free(addr_tmp); |
| printf("mmc read 1 block fail\n"); |
| return 1; |
| } |
| |
| memcpy(addr_tmp, addr_byte, sz_byte); |
| if (blk_dwrite(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // write 1 block |
| free(addr_tmp); |
| printf("mmc write 1 block fail\n"); |
| return 1; |
| } |
| free(addr_tmp); |
| } |
| //printf("%#llx blocks , %#llx bytes written: %s\n", n, sz_byte, (n==cnt) ? "OK" : "ERROR"); |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int amlmmc_write_in_card(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| u64 cnt = 0, n = 0, blk = 0, sz_byte = 0; |
| char *name = NULL; |
| u64 offset = 0, size = 0; |
| struct mmc *mmc; |
| |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name (name); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| offset = simple_strtoull(argv[4], NULL, 16); |
| size = simple_strtoull(argv[5], NULL, 16); |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| int blk_shift = ffs( mmc->read_bl_len) -1; |
| cnt = size >> blk_shift; |
| blk = offset >> blk_shift; |
| sz_byte = size - (cnt<<blk_shift); |
| mmc_init(mmc); |
| |
| n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| |
| //write sz_byte bytes |
| if ((n == cnt) && (sz_byte != 0)) { |
| // printf("sz_byte=%#llx bytes\n",sz_byte); |
| void *addr_tmp = malloc(mmc->write_bl_len); |
| void *addr_byte = (void*)(addr+cnt*(mmc->write_bl_len)); |
| ulong start_blk = blk+cnt; |
| |
| if (addr_tmp == NULL) { |
| printf("mmc write: malloc fail\n"); |
| return 1; |
| } |
| |
| if (blk_dread(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // read 1 block |
| free(addr_tmp); |
| printf("mmc read 1 block fail\n"); |
| return 1; |
| } |
| |
| memcpy(addr_tmp, addr_byte, sz_byte); |
| if (blk_dwrite(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // write 1 block |
| free(addr_tmp); |
| printf("mmc write 1 block fail\n"); |
| return 1; |
| } |
| free(addr_tmp); |
| } |
| //printf("%#llx blocks , %#llx bytes written: %s\n", n, sz_byte, (n==cnt) ? "OK" : "ERROR"); |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int amlmmc_write_in_dev(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| u64 cnt = 0, n = 0, blk = 0; |
| struct mmc *mmc; |
| dev = simple_strtoul(argv[2], NULL, 10); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| blk = simple_strtoull(argv[4], NULL, 16); |
| cnt = simple_strtoull(argv[5], NULL, 16); |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| //printf("MMC write: dev # %d, block # %#llx, count # %#llx ... ", |
| //dev, blk, cnt); |
| |
| mmc_init(mmc); |
| |
| n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int do_amlmmc_write(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = 0; |
| if (argc != 6) |
| return CMD_RET_USAGE; |
| |
| if (isdigit(argv[2][0])) |
| ret = amlmmc_write_in_dev(argc, argv); |
| else if (strcmp(argv[2], "card") == 0) |
| ret = amlmmc_write_in_card(argc, argv); |
| else if (isstring(argv[2])) |
| ret = amlmmc_write_in_part(argc, argv); |
| |
| return ret; |
| } |
| |
| static int amlmmc_read_in_dev(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| u64 cnt =0, n = 0, blk = 0; |
| struct mmc *mmc; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| blk = simple_strtoull(argv[4], NULL, 16); |
| cnt = simple_strtoull(argv[5], NULL, 16); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int amlmmc_read_in_card(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| //u32 flag =0; |
| u64 cnt =0, n = 0, blk = 0, sz_byte = 0; |
| char *name = NULL; |
| u64 offset = 0, size = 0; |
| int blk_shift; |
| struct mmc *mmc; |
| void *addr_tmp; |
| void *addr_byte; |
| ulong start_blk; |
| |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name (name); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| size = simple_strtoull(argv[5], NULL, 16); |
| offset = simple_strtoull(argv[4], NULL, 16); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| blk_shift = ffs( mmc->read_bl_len) - 1; |
| cnt = size >> blk_shift; |
| blk = offset >> blk_shift; |
| sz_byte = size - (cnt<<blk_shift); |
| |
| mmc_init(mmc); |
| n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| |
| //read sz_byte bytes |
| if ((n == cnt) && (sz_byte != 0)) { |
| addr_tmp = malloc(mmc->read_bl_len); |
| addr_byte = (void *)(addr+cnt*(mmc->read_bl_len)); |
| start_blk = blk+cnt; |
| if (addr_tmp == NULL) { |
| printf("mmc read: malloc fail\n"); |
| return 1; |
| } |
| if (blk_dread(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // read 1 block |
| free(addr_tmp); |
| printf("mmc read 1 block fail\n"); |
| return 1; |
| } |
| memcpy(addr_byte, addr_tmp, sz_byte); |
| free(addr_tmp); |
| } |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int amlmmc_read_in_part(int argc, char *const argv[]) |
| { |
| int dev; |
| void *addr = NULL; |
| u64 cnt = 0, n = 0, blk = 0, sz_byte = 0; |
| char *name = NULL; |
| u64 offset = 0, size = 0; |
| struct mmc *mmc; |
| void *addr_tmp; |
| void *addr_byte; |
| ulong start_blk; |
| |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name (name); |
| addr = (void *)simple_strtoul(argv[3], NULL, 16); |
| offset = simple_strtoull(argv[4], NULL, 16); |
| size = simple_strtoull(argv[5], NULL, 16); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| get_off_size(mmc, name, offset, size, &blk, &cnt, &sz_byte); |
| mmc_init(mmc); |
| n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| |
| //read sz_byte bytes |
| if ((n == cnt) && (sz_byte != 0)) { |
| /*printf("sz_byte=%#llx bytes\n",sz_byte);*/ |
| addr_tmp = malloc(mmc->read_bl_len); |
| addr_byte = (void *)(addr+cnt*(mmc->read_bl_len)); |
| start_blk = blk+cnt; |
| |
| if (addr_tmp == NULL) { |
| printf("mmc read: malloc fail\n"); |
| return 1; |
| } |
| |
| if (blk_dread(mmc_get_blk_desc(mmc), start_blk, 1, addr_tmp) != 1) { // read 1 block |
| free(addr_tmp); |
| printf("mmc read 1 block fail\n"); |
| return 1; |
| } |
| |
| memcpy(addr_byte, addr_tmp, sz_byte); |
| free(addr_tmp); |
| } |
| return (n == cnt) ? 0 : 1; |
| } |
| |
| static int do_amlmmc_read(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = 0; |
| if (argc != 6) |
| return CMD_RET_USAGE; |
| |
| if (isdigit(argv[2][0])) |
| ret = amlmmc_read_in_dev(argc, argv); |
| else if (strcmp(argv[2], "card") == 0) |
| ret = amlmmc_read_in_card(argc, argv); |
| else if (isstring(argv[2])) |
| ret = amlmmc_read_in_part(argc, argv); |
| |
| return ret; |
| } |
| |
| static int do_amlmmc_env(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| printf("herh\n"); |
| env_relocate(); |
| return 0; |
| } |
| |
| static int do_amlmmc_list(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| print_mmc_devices('\n'); |
| return 0; |
| } |
| |
| static int do_amlmmc_size(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| char *name; |
| uint64_t* addr = NULL; |
| int dev; |
| struct mmc *mmc = NULL; |
| |
| if (argc != 4) |
| return CMD_RET_USAGE; |
| |
| name = argv[2]; |
| addr = (uint64_t *)simple_strtoul(argv[3], NULL, 16); |
| if (!strcmp(name, "wholeDev")) { |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| mmc = find_mmc_device(dev); |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| mmc_init(mmc); |
| |
| *addr = mmc->capacity >> 9; // unit: 512 bytes |
| return 0; |
| } |
| return get_partition_size((unsigned char *)name, addr); |
| } |
| |
| static int amlmmc_get_ext_csd(int argc, char *const argv[]) |
| { |
| int ret= 0; |
| u8 ext_csd[512] = {0}; |
| int dev, byte; |
| struct mmc *mmc; |
| |
| if (argc != 4) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| byte = simple_strtoul(argv[3], NULL, 10); |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| mmc_init(mmc); |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| printf("read EXT_CSD byte[%d] val[0x%x] %s\n", |
| byte, ext_csd[byte], (ret == 0) ? "ok" : "fail"); |
| ret = ret || ret; |
| return ret; |
| } |
| |
| static int amlmmc_set_ext_csd(int argc, char *const argv[]) |
| { |
| int ret = 0; |
| int dev, byte; |
| struct mmc *mmc; |
| int val; |
| |
| if (argc != 5) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| byte = simple_strtoul(argv[3], NULL, 10); |
| val = simple_strtoul(argv[4], NULL, 16); |
| if ((byte > 191) || (byte < 0)) { |
| printf("byte is not able to write!\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| |
| mmc_init(mmc); |
| |
| ret = mmc_set_ext_csd(mmc, byte, val); |
| printf("write EXT_CSD byte[%d] val[0x%x] %s\n", |
| byte, val, (ret == 0) ? "ok" : "fail"); |
| ret =ret || ret; |
| return ret; |
| } |
| |
| static int do_amlmmc_ext_csd(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| |
| if (argc == 4) |
| ret = amlmmc_get_ext_csd(argc,argv); |
| else if (argc == 5) |
| ret = amlmmc_set_ext_csd(argc,argv); |
| |
| return ret; |
| } |
| |
| static int do_amlmmc_switch(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int rc = 0; |
| int dev; |
| struct mmc *mmc; |
| |
| if (argc != 4) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| |
| mmc_init(mmc); |
| printf("mmc switch to "); |
| |
| if (strcmp(argv[3], "boot0") == 0) { |
| rc = mmc_switch_part(mmc, 1); |
| if (rc == 0) { |
| emmc_cur_partition = 1; |
| printf("boot0 success\n"); |
| } else { |
| printf("boot0 failed\n"); |
| } |
| } |
| else if(strcmp(argv[3], "boot1") == 0) { |
| rc = mmc_switch_part(mmc, 2); |
| if (rc == 0) { |
| emmc_cur_partition = 2; |
| printf("boot1 success\n"); |
| } else { |
| printf("boot1 failed\n"); |
| } |
| } |
| else if(strcmp(argv[3], "user") == 0) { |
| rc = mmc_switch_part(mmc, 0); |
| if (rc == 0) { |
| emmc_cur_partition = 0; |
| printf("user success\n"); |
| } else { |
| printf("user failed\n"); |
| } |
| } |
| #ifdef CONFIG_SUPPORT_EMMC_RPMB |
| else if(strcmp(argv[3], "rpmb") == 0) { |
| rc = mmc_switch_part(mmc, 3); |
| if (rc == 0) { |
| emmc_cur_partition = 3; |
| printf("rpmb success\n"); |
| } else { |
| printf("rpmb failed\n"); |
| } |
| } |
| #endif |
| else |
| printf("%s failed\n", argv[3]); |
| return rc; |
| } |
| |
| static int do_amlmmc_controller(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| /*int dev; |
| struct mmc *mmc; |
| struct aml_card_sd_info *aml_priv; |
| struct sd_emmc_global_regs *sd_emmc_reg; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| aml_priv = mmc->priv; |
| sd_emmc_reg = aml_priv->sd_emmc_reg; |
| |
| printf("sd_emmc_reg->gclock = 0x%x\n", sd_emmc_reg->gclock); |
| printf("sd_emmc_reg->gdelay = 0x%x\n", sd_emmc_reg->gdelay); |
| printf("sd_emmc_reg->gadjust = 0x%x\n", sd_emmc_reg->gadjust); |
| printf("sd_emmc_reg->gcalout = 0x%x\n", sd_emmc_reg->gcalout); |
| if (!mmc->has_init) { |
| printf("mmc dev %d has not been initialed\n", dev); |
| return 1; |
| } |
| printf("sd_emmc_reg->gstart = 0x%x\n", sd_emmc_reg->gstart); |
| printf("sd_emmc_reg->gcfg = 0x%x\n", sd_emmc_reg->gcfg); |
| printf("sd_emmc_reg->gstatus = 0x%x\n", sd_emmc_reg->gstatus); |
| printf("sd_emmc_reg->girq_en = 0x%x\n", sd_emmc_reg->girq_en); |
| printf("sd_emmc_reg->gcmd_cfg = 0x%x\n", sd_emmc_reg->gcmd_cfg); |
| printf("sd_emmc_reg->gcmd_arg = 0x%x\n", sd_emmc_reg->gcmd_arg); |
| printf("sd_emmc_reg->gcmd_dat = 0x%x\n", sd_emmc_reg->gcmd_dat); |
| printf("sd_emmc_reg->gcmd_rsp0 = 0x%x\n", sd_emmc_reg->gcmd_rsp0); |
| printf("sd_emmc_reg->gcmd_rsp1 = 0x%x\n", sd_emmc_reg->gcmd_rsp1); |
| printf("sd_emmc_reg->gcmd_rsp2 = 0x%x\n", sd_emmc_reg->gcmd_rsp2); |
| printf("sd_emmc_reg->gcmd_rsp3 = 0x%x\n", sd_emmc_reg->gcmd_rsp3);*/ |
| return 0; |
| } |
| |
| static int do_amlmmc_response(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| /*int dev; |
| struct mmc *mmc; |
| struct aml_card_sd_info *aml_priv; |
| struct sd_emmc_global_regs *sd_emmc_reg; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| if (!mmc->has_init) { |
| printf("mmc dev %d has not been initialed\n", dev); |
| return 1; |
| } |
| |
| aml_priv = mmc->priv; |
| sd_emmc_reg = aml_priv->sd_emmc_reg; |
| |
| printf("last cmd = %d, response0 = 0x%x\n", |
| (sd_emmc_reg->gcmd_cfg & 0x3f), sd_emmc_reg->gcmd_rsp0);*/ |
| return 0; |
| } |
| |
| static int do_amlmmc_status(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int rc = 0; |
| int dev; |
| struct mmc *mmc; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| if (!mmc->has_init) { |
| printf("mmc dev %d has not been initialed\n", dev); |
| return 1; |
| } |
| rc = mmc_read_status(mmc, 1000); |
| if (rc) |
| return 1; |
| else |
| return 0; |
| } |
| |
| static int do_amlmmc_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
| { |
| int dev; |
| struct blk_desc *mmc_dev; |
| struct mmc *mmc; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| mmc_init(mmc); |
| mmc_dev = mmc_get_blk_desc(mmc); |
| if (mmc_dev != NULL && |
| mmc_dev->type != DEV_TYPE_UNKNOWN) { |
| part_print(mmc_dev); |
| return 0; |
| } |
| puts("get mmc type error!\n"); |
| return 1; |
| } |
| |
| static int do_amlmmc_rescan(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
| { |
| int dev; |
| struct mmc *mmc; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| |
| if (dev < 0) { |
| printf("Cannot find dev.\n"); |
| return 1; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| mmc->has_init =0; |
| return mmc_init(mmc); |
| } |
| |
| #ifdef CONFIG_SECURITYKEY |
| static int do_amlmmc_key(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| struct mmc *mmc; |
| int dev; |
| |
| //char *name = "logo"; |
| dev = CONFIG_SYS_MMC_BOOT_DEV; |
| mmc = find_mmc_device(dev); |
| if (!mmc) { |
| printf("device %d is invalid\n",dev); |
| return 1; |
| } |
| //mmc->key_protect = 0; |
| #ifdef CONFIG_STORE_COMPATIBLE |
| info_disprotect |= DISPROTECT_KEY; //disprotect |
| printf("emmc disprotect key\n"); |
| #endif |
| return 0; |
| } |
| #endif |
| |
| static int set_write_prot(struct mmc *mmc, u64 start) |
| { |
| struct mmc_cmd cmd; |
| int err; |
| |
| cmd.cmdidx = MMC_CMD_SET_WRITE_PROTECT; |
| cmd.cmdarg = start; |
| cmd.resp_type = MMC_RSP_R1b; |
| |
| err = mmc_send_cmd(mmc, &cmd, NULL); |
| if (err) |
| goto err_out; |
| |
| return 0; |
| |
| err_out: |
| puts("Failed: mmc write protect failed\n"); |
| return err; |
| } |
| |
| static int set_us_wp_en(struct mmc *mmc, u8 *ext_csd, u8 wp_enable_type) |
| { |
| u8 index = EXT_CSD_USER_WP; |
| u8 user_wp = ext_csd[index]; |
| user_wp = user_wp & (~WP_ENABLE_MASK); |
| user_wp = user_wp | wp_enable_type; |
| int err = 0; |
| err = mmc_set_ext_csd(mmc, index, user_wp); |
| if (err) |
| printf("Failed: set write protect enable failed\n"); |
| return err; |
| } |
| |
| static int mmc_set_us_perm_wp_dis(struct mmc *mmc, u8 *ext_csd) |
| { |
| u8 usr_wp = ext_csd[EXT_CSD_USER_WP]; |
| u8 perm_disable_bit = US_PERM_WP_DIS_BIT; |
| |
| int err; |
| if (usr_wp & perm_disable_bit) |
| return 0; |
| |
| usr_wp = usr_wp | perm_disable_bit; |
| err = mmc_set_ext_csd(mmc, EXT_CSD_USER_WP, usr_wp); |
| if (err) { |
| printf("Failed: set permanent write protect disable failed\n"); |
| return 1; |
| } |
| return 0; |
| } |
| |
| static int mmc_is_us_pwr_wp_dis(u8 user_wp) |
| { |
| return user_wp & US_PWR_WP_DIS_BIT; |
| } |
| |
| static int mmc_is_us_perm_wp_dis(u8 user_wp) |
| { |
| return user_wp & US_PERM_WP_DIS_BIT; |
| } |
| |
| static int check_wp_type(u8 *addr, u8 wp_type, u64 set_protect_cnt) |
| { |
| u8 type_mask = WP_TYPE_MASK; |
| u64 cnt = set_protect_cnt; |
| u8 times = 0; |
| u8 index = 7; |
| u8 cur_group_wp_type = addr[index]; |
| |
| while (cnt != 0) { |
| if (wp_type != ((type_mask)&(cur_group_wp_type))) { |
| return 1; |
| } |
| if (times == 3) { |
| times = 0; |
| index--; |
| cur_group_wp_type = addr[index]; |
| } |
| else { |
| cur_group_wp_type = cur_group_wp_type >> 2; |
| times++; |
| } |
| cnt--; |
| } |
| return 0; |
| } |
| |
| static int set_register_to_temporary(struct mmc *mmc, u8 *ext_csd) |
| { |
| int err; |
| u8 wp_enable_type = WP_TEMPORARY_EN_BIT; |
| err = set_us_wp_en(mmc, ext_csd, wp_enable_type); |
| if (err) |
| printf("Failed: set temporary write protect failed\n"); |
| return err; |
| } |
| |
| static int set_register_to_pwr(struct mmc *mmc, u8 *ext_csd) |
| { |
| int err; |
| u8 user_wp = ext_csd[EXT_CSD_USER_WP]; |
| u8 wp_enable_type = WP_POWER_ON_EN_BIT; |
| if (mmc_is_us_pwr_wp_dis(user_wp)) { |
| printf("Failed: power on protection had been disabled\n"); |
| return 1; |
| } |
| |
| err = mmc_set_us_perm_wp_dis(mmc, ext_csd); |
| if (err) { |
| printf("Failed: set permanent protection diable failed\n"); |
| return 1; |
| } |
| |
| err = set_us_wp_en(mmc, ext_csd, wp_enable_type); |
| if (err) { |
| printf("Failed: set power on write protect enable failed\n"); |
| return 1; |
| } |
| return 0; |
| } |
| |
| static int set_register_to_perm(struct mmc *mmc, u8 *ext_csd) |
| { |
| int err; |
| u8 wp_enable_type = WP_PERM_EN_BIT; |
| u8 user_wp = ext_csd[EXT_CSD_USER_WP]; |
| |
| if (mmc_is_us_perm_wp_dis(user_wp)) { |
| printf("Failed: Permanent protection had been disabled\n"); |
| return 1; |
| } |
| |
| err = set_us_wp_en(mmc, ext_csd, wp_enable_type); |
| if (err) { |
| printf("Failed: set permanent write protect enable failed\n"); |
| return 1; |
| } |
| return 0; |
| } |
| |
| static int set_wp_register(struct mmc *mmc, u8 *ext_csd, u8 wp_type) |
| { |
| int ret = 1; |
| if (wp_type == WP_POWER_ON_TYPE) |
| ret = set_register_to_pwr(mmc, ext_csd); |
| else if (wp_type == WP_PERMANENT_TYPE) |
| ret = set_register_to_perm(mmc, ext_csd); |
| else if (wp_type == WP_TEMPORARY_TYPE) |
| ret = set_register_to_temporary(mmc, ext_csd); |
| return ret; |
| } |
| |
| static u64 write_protect_group_size(struct mmc *mmc, u8 *ext_csd) |
| { |
| int erase_group_def = ext_csd[EXT_CSD_ERASE_GROUP_DEF]; |
| u64 write_protect_group_size; |
| int wp_grp_size = mmc->csd[2] & WP_GRP_SIZE_MASK; |
| int hc_wp_grp_size = ext_csd[EXT_CSD_HC_WP_GRP_SIZE]; |
| |
| if (erase_group_def == 0) |
| write_protect_group_size = (wp_grp_size + 1) * mmc->erase_grp_size; |
| else |
| write_protect_group_size = hc_wp_grp_size * mmc->erase_grp_size; |
| |
| return write_protect_group_size; |
| } |
| |
| int is_write_protect_valid(u8 *ext_csd) |
| { |
| u8 class_6_ctrl = ext_csd[EXT_CSD_CLASS_6_CTRL]; |
| if (class_6_ctrl == 0) |
| return 1; |
| return 0; |
| } |
| |
| static int compute_write_protect_range(struct mmc *mmc, char *name, |
| u8 *ext_csd, u64 *wp_grp_size_addr, u64 *start_addr, u64 *end) |
| { |
| int blk_shift; |
| struct partitions *part_info; |
| u64 cnt; |
| u64 start = *start_addr; |
| u64 align_start = *start_addr; |
| u64 wp_grp_size = *wp_grp_size_addr; |
| u64 group_num ; |
| u64 partition_end; |
| |
| wp_grp_size = write_protect_group_size(mmc, ext_csd); |
| |
| blk_shift = ffs(mmc->read_bl_len) -1; |
| |
| part_info = find_mmc_partition_by_name(name); |
| if (part_info == NULL) |
| return 1; |
| |
| start = part_info->offset >> blk_shift; |
| printf("start:%d wp_grp_size:%d",(int)start,(int)wp_grp_size); |
| if ((start % wp_grp_size)) { |
| align_start = (start + wp_grp_size - 1) / wp_grp_size * wp_grp_size; |
| printf("Caution! The partition start address isn't' aligned" |
| "to group size\n" |
| "the start address is change from 0x%llx to 0x%llx\n", |
| start, align_start); |
| } else { |
| align_start = start; |
| } |
| if (emmc_cur_partition && !strncmp(name, "bootloader", strlen("bootloader"))) |
| cnt =0; |
| else |
| cnt = part_info->size >> blk_shift; |
| if (cnt < wp_grp_size) { |
| printf("Caution: The partition size is 0x%llx sector smaller than " |
| "the group size 0x%llx sector, \n" |
| "so the partition can't be protect\n", cnt, wp_grp_size); |
| return 1; |
| } |
| |
| *start_addr = align_start; |
| *wp_grp_size_addr = wp_grp_size; |
| partition_end = start + cnt - 1; |
| group_num = (cnt - (align_start - start)) / wp_grp_size; |
| *end = align_start + group_num * wp_grp_size - 1; |
| |
| if (partition_end != *end) { |
| printf("Caution! The boundary of partition isn't aligned with write " |
| "protected group,\n" |
| "so the write protected boundry of the " |
| "partition is 0x%llx, rather than 0x%llx\n", |
| *end, partition_end); |
| } |
| |
| printf("write_protect group size is 0x%llx sector\n", wp_grp_size); |
| printf("The %s partition write protect group number is %lld\n", name, group_num); |
| #ifdef WP_DEBUG |
| printf("the start address is 0x%llx, group size is 0x%llx, end is 0x%llx\n", |
| *start_addr, *wp_grp_size_addr, *end); |
| #endif |
| return 0; |
| } |
| |
| static int send_wp_prot_type(struct mmc *mmc, void *dst, u64 blk) |
| { |
| struct mmc_cmd cmd; |
| int err; |
| struct mmc_data data; |
| |
| cmd.cmdidx = MMC_CMD_SEND_WRITE_PROT_TYPE; |
| cmd.cmdarg = blk; |
| cmd.resp_type = MMC_RSP_R1; |
| |
| data.dest = dst; |
| data.blocks = 1; |
| data.blocksize = 8; |
| data.flags = MMC_DATA_READ; |
| |
| err = mmc_send_cmd(mmc, &cmd, &data); |
| if (err) |
| goto err_out; |
| |
| return 0; |
| |
| err_out: |
| puts("Failed: mmc send write protect type failed\n"); |
| return err; |
| } |
| |
| static int is_wp_set_failed(struct mmc *mmc, u8 wp_type, u64 start, u64 group_cnt) |
| { |
| u8 *addr = NULL; |
| u8 err = 0; |
| |
| addr = malloc(sizeof(u64)); |
| if (addr == NULL) { |
| printf("Failed: malloc failed\n"); |
| return 1; |
| } |
| |
| err = send_wp_prot_type(mmc, addr, start); |
| if (err) |
| goto err_out; |
| |
| #ifdef WP_DEBUG |
| int i; |
| for (i = 0; i < 8; i++) |
| printf("write_protect status is %x\n", ((u8 *)addr)[i]); |
| #endif |
| if (check_wp_type(addr, wp_type, group_cnt)) { |
| printf("Failed: Write Protection set failed\n"); |
| goto err_out; |
| } |
| return 0; |
| |
| err_out: |
| free(addr); |
| return 1; |
| } |
| |
| static int send_part_wp_type(struct mmc *mmc, char *name) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u64 wp_grp_size, start, part_end; |
| u64 group_start; |
| void *addr = NULL; |
| int i; |
| int ret; |
| |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| if (ret) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| addr = malloc(sizeof(u64)); |
| if (addr == NULL) { |
| printf("Failed: malloc failed\n"); |
| return 1; |
| } |
| |
| err = compute_write_protect_range(mmc, name, ext_csd, |
| &wp_grp_size, &start, &part_end); |
| if (err) |
| return 1; |
| |
| group_start = start; |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = send_wp_prot_type(mmc, addr, group_start); |
| if (err) |
| return 1; |
| printf("The write protect type for the 32 groups after 0x%llx is: \n0x", |
| group_start); |
| for (i = 0; i < 8; i++) |
| printf("%02x", ((u8*)addr)[i]); |
| printf("\n"); |
| group_start += 32 * wp_grp_size; |
| } |
| return 0; |
| } |
| |
| static int send_add_wp_type(struct mmc *mmc, u64 start, u64 cnt) |
| { |
| u8 ext_csd[512] = {0}; |
| u64 wp_grp_size = 0; |
| u64 part_end = 0; |
| u64 group_start; |
| u64 mmc_boundary; |
| int blk_shift; |
| void *addr = NULL; |
| int i; |
| int ret; |
| |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| if (ret) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| addr = malloc(sizeof(u64)); |
| if (addr == NULL) { |
| printf("Failed: malloc failed\n"); |
| return 1; |
| } |
| |
| wp_grp_size = write_protect_group_size(mmc, ext_csd); |
| |
| if ((start % wp_grp_size)) { |
| group_start = (start + wp_grp_size - 1) / wp_grp_size * wp_grp_size; |
| printf("Caution! The partition start address isn't' aligned" |
| "to group size\n" |
| "the start address is change from 0x%llx to 0x%llx\n", |
| start, group_start); |
| part_end = group_start + (cnt - 1) * wp_grp_size - 1; |
| printf("The write protect group number is 0x%llx, rather than 0x%lld\n", |
| cnt - 1, cnt); |
| } else { |
| group_start = start; |
| part_end = group_start + cnt * wp_grp_size - 1; |
| } |
| |
| blk_shift = ffs(mmc->read_bl_len) - 1; |
| mmc_boundary = mmc->capacity>>blk_shift; |
| |
| if ((part_end + 1) > mmc_boundary) { |
| printf("Error: the operation cross the boundary of mmc\n"); |
| return 1; |
| } |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| ret = send_wp_prot_type(mmc, addr, group_start); |
| if (ret) |
| return 1; |
| printf("The write protect type for the 32 groups after 0x%llx is: \n0x", |
| group_start); |
| for (i = 0; i < 8; i++) |
| printf("%02x", ((u8*)addr)[i]); |
| printf("\n"); |
| group_start += 32 * wp_grp_size; |
| } |
| return 0; |
| } |
| |
| static int set_part_write_protect(struct mmc *mmc, u8 wp_type, char *name) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u8 group_num = 32; |
| u64 wp_grp_size, start, part_end; |
| u64 group_start; |
| u64 check_group_start; |
| u64 set_protect_cnt = 0; |
| |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| if (err) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| err = compute_write_protect_range(mmc, name, ext_csd, |
| &wp_grp_size, &start, &part_end); |
| if (err) |
| return 1; |
| |
| group_start = start; |
| err = set_wp_register(mmc, ext_csd, wp_type); |
| if (err) |
| return 1; |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = set_write_prot(mmc, group_start); |
| if (err) |
| return 1; |
| group_start += wp_grp_size; |
| set_protect_cnt++; |
| //check write protect type every 32 group |
| if (set_protect_cnt % 32 == 0) { |
| check_group_start = group_start - group_num * wp_grp_size; |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| } |
| |
| group_num = set_protect_cnt % 32; |
| check_group_start = group_start - group_num * wp_grp_size; |
| |
| if (group_num) { |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| |
| return 0; |
| } |
| |
| static int set_add_write_protect(struct mmc *mmc, u8 wp_type, u64 start, u64 cnt) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| int group_num = 32; |
| u64 wp_grp_size, part_end; |
| u64 group_start; |
| u64 check_group_start; |
| u64 set_protect_cnt = 0; |
| u64 mmc_boundary = 0; |
| int blk_shift; |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| |
| if (err) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| if (err) |
| return 1; |
| |
| wp_grp_size = write_protect_group_size(mmc, ext_csd); |
| |
| if ((start % wp_grp_size)) { |
| group_start = (start + wp_grp_size - 1) / wp_grp_size * wp_grp_size; |
| printf("Caution! The partition start address isn't' aligned" |
| "to group size\n" |
| "the start address is change from 0x%llx to 0x%llx\n", |
| start, group_start); |
| part_end = group_start + (cnt - 1) * wp_grp_size - 1; |
| printf("The write protect group number is 0x%llx, rather than 0x%lld\n", |
| cnt - 1, cnt); |
| } else { |
| group_start = start; |
| part_end = group_start + cnt * wp_grp_size - 1; |
| } |
| |
| blk_shift = ffs(mmc->read_bl_len) - 1; |
| mmc_boundary = mmc->capacity>>blk_shift; |
| |
| if ((part_end + 1) > mmc_boundary) { |
| printf("Error: the operation cross the boundary of mmc\n"); |
| return 1; |
| } |
| |
| err = set_wp_register(mmc, ext_csd, wp_type); |
| if (err) |
| return 1; |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = set_write_prot(mmc, group_start); |
| if (err) |
| return 1; |
| group_start += wp_grp_size; |
| set_protect_cnt++; |
| //check write protect type every 32 group |
| if (set_protect_cnt % 32 == 0) { |
| check_group_start = group_start - group_num * wp_grp_size; |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| } |
| |
| group_num = set_protect_cnt % 32; |
| check_group_start = group_start - group_num * wp_grp_size; |
| |
| if (group_num) { |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| |
| return 0; |
| } |
| |
| static int do_amlmmc_write_protect(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| struct mmc *mmc; |
| int dev = 1; |
| char *name = NULL; |
| char *wp_type_str = NULL; |
| u8 write_protect_type; |
| u64 start, cnt; |
| |
| if (argc > 5 || argc < 4) |
| return ret; |
| if (argc == 4) { |
| name = argv[2]; |
| wp_type_str = argv[3]; |
| dev = find_dev_num_by_partition_name(name); |
| if (dev < 0) { |
| printf("Error: Cannot find dev.\n"); |
| return CMD_RET_USAGE; |
| } |
| } else { |
| start = simple_strtoull(argv[2], NULL, 16); |
| cnt = simple_strtoull(argv[3], NULL, 0); |
| wp_type_str = argv[4]; |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (IS_SD(mmc)) { |
| mmc = find_mmc_device(~dev); |
| if (IS_SD(mmc)) { |
| printf("SD card can not be write protect\n"); |
| return 1; |
| } |
| } |
| |
| if (!mmc) |
| return 1; |
| |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| |
| if (strcmp(wp_type_str, "temporary") == 0) |
| write_protect_type = WP_TEMPORARY_TYPE; |
| else if (strcmp(wp_type_str, "power_on") == 0 ) |
| write_protect_type = WP_POWER_ON_TYPE; |
| else if (strcmp(wp_type_str, "permanent") == 0) |
| write_protect_type = WP_PERMANENT_TYPE; |
| else |
| return ret; |
| |
| if (argc == 4) |
| ret = set_part_write_protect(mmc, write_protect_type, name); |
| else |
| ret = set_add_write_protect(mmc, write_protect_type, start, cnt); |
| |
| return ret; |
| } |
| |
| static int clear_write_prot_per_group(struct mmc *mmc, u64 blk) |
| { |
| struct mmc_cmd cmd; |
| int err; |
| |
| cmd.cmdidx = MMC_CMD_CLR_WRITE_PROT; |
| cmd.cmdarg = blk; |
| cmd.resp_type = MMC_RSP_R1b; |
| |
| err = mmc_send_cmd(mmc, &cmd, NULL); |
| |
| return err; |
| } |
| |
| static int set_part_clear_wp(struct mmc *mmc, char *name) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u8 group_num = 32; |
| u64 wp_grp_size, start, part_end; |
| u64 group_start; |
| u64 check_group_start; |
| u64 set_protect_cnt = 0; |
| u8 wp_type = WP_CLEAR_TYPE; |
| |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| if (err) { |
| printf("get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("CLASS_6_CTRL isn't '0' write protect process is invalid\n"); |
| return 1; |
| } |
| |
| err = compute_write_protect_range(mmc, name, ext_csd, |
| &wp_grp_size, &start, &part_end); |
| if (err) |
| return 1; |
| |
| group_start = start; |
| /* |
| if (!is_wp_type_temporary(ext_csd)) { |
| printf("The write protect can't be clear\n"); |
| return 1; |
| } |
| */ |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = clear_write_prot_per_group(mmc, group_start); |
| if (err) { |
| printf("Error: The write protect can't be clear\n"); |
| return 1; |
| } |
| group_start += wp_grp_size; |
| set_protect_cnt++; |
| //check write protect type every 32 group |
| if (set_protect_cnt % 32 == 0) { |
| check_group_start = group_start - group_num * wp_grp_size; |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| } |
| |
| group_num = set_protect_cnt % 32; |
| check_group_start = group_start - group_num * wp_grp_size; |
| |
| if (group_num) { |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| |
| return 0; |
| } |
| |
| static int set_add_clear_wp(struct mmc *mmc, u64 start, u64 cnt) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u8 group_num = 32; |
| u64 wp_grp_size, part_end; |
| u64 group_start; |
| u64 check_group_start; |
| u64 set_protect_cnt = 0; |
| u8 wp_type = WP_CLEAR_TYPE; |
| int blk_shift; |
| u64 mmc_boundary; |
| |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| if (err) { |
| printf("get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("CLASS_6_CTRL isn't '0' write protect process is invalid\n"); |
| return 1; |
| } |
| |
| wp_grp_size = write_protect_group_size(mmc, ext_csd); |
| |
| if ((start % wp_grp_size)) { |
| group_start = (start + wp_grp_size - 1) / wp_grp_size * wp_grp_size; |
| printf("Caution! The partition start address isn't' aligned" |
| "to group size\n" |
| "the start address is change from 0x%llx to 0x%llx\n", |
| start, group_start); |
| part_end = group_start + (cnt - 1) * wp_grp_size - 1; |
| printf("The write protect group number is 0x%llx, rather than 0x%lld\n", |
| cnt - 1, cnt); |
| } else { |
| group_start = start; |
| part_end = group_start + cnt * wp_grp_size - 1; |
| } |
| |
| blk_shift = ffs(mmc->read_bl_len) - 1; |
| mmc_boundary = mmc->capacity>>blk_shift; |
| |
| if ((part_end + 1) > mmc_boundary) { |
| printf("Error: the operation cross the boundary of mmc\n"); |
| return 1; |
| } |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = clear_write_prot_per_group(mmc, group_start); |
| if (err) { |
| printf("Error: The write protect can't be clear\n"); |
| return 1; |
| } |
| group_start += wp_grp_size; |
| set_protect_cnt++; |
| //check write protect type every 32 group |
| if (set_protect_cnt % 32 == 0) { |
| check_group_start = group_start - group_num * wp_grp_size; |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| } |
| |
| group_num = set_protect_cnt % 32; |
| check_group_start = group_start - group_num * wp_grp_size; |
| |
| if (group_num) { |
| err = is_wp_set_failed(mmc, wp_type, check_group_start, group_num); |
| if (err) |
| return 1; |
| } |
| |
| return 0; |
| } |
| |
| static int do_amlmmc_clear_wp(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| struct mmc *mmc; |
| int dev = 1; |
| char *name = NULL; |
| u64 start, cnt; |
| |
| if (argc < 3 || argc > 4) |
| return ret; |
| |
| if (argc == 3) { |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name(name); |
| if (dev < 0) { |
| printf("Error: Cannot find dev.\n"); |
| return CMD_RET_USAGE; |
| } |
| } else { |
| start = simple_strtoull(argv[2], NULL, 16); |
| cnt = simple_strtoull(argv[3], NULL, 10); |
| } |
| |
| mmc = find_mmc_device(dev); |
| if (!mmc) |
| return 1; |
| |
| if (IS_SD(mmc)) { |
| mmc = find_mmc_device(~dev); |
| if (IS_SD(mmc)) { |
| printf("SD card can not be write protect\n"); |
| return 1; |
| } |
| } |
| |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| |
| if (argc == 3) |
| ret = set_part_clear_wp(mmc, name); |
| else |
| ret = set_add_clear_wp(mmc, start, cnt); |
| |
| return ret; |
| } |
| |
| static int send_write_prot_status_group(struct mmc *mmc, u64 blk) |
| { |
| struct mmc_cmd cmd; |
| struct mmc_data data; |
| int err; |
| void *addr = NULL; |
| int i = 0; |
| addr = malloc(4*sizeof(u8)); |
| |
| if (addr == NULL) { |
| printf("Failed: malloc failed\n"); |
| return 1; |
| } |
| |
| cmd.cmdidx = MMC_CMD_SEND_WRITE_PROT; |
| cmd.cmdarg = blk; |
| cmd.resp_type = MMC_RSP_R1; |
| |
| data.dest = addr; |
| data.blocks = 1; |
| data.blocksize = 4; |
| data.flags = MMC_DATA_READ; |
| |
| err = mmc_send_cmd(mmc, &cmd, &data); |
| if (err) |
| goto err_out; |
| |
| printf("The write protect type for the 32 groups after 0x%llx is:\n0x", |
| blk); |
| for (i = 0 ; i < 4; i++) |
| printf("%02x", ((u8 *)addr)[i]); |
| printf("\n"); |
| |
| free(addr); |
| return 0; |
| err_out: |
| free(addr); |
| return 1; |
| } |
| |
| static int send_part_wp_status(struct mmc *mmc, char *name) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u64 wp_grp_size, start, part_end; |
| u64 group_start; |
| |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| if (err) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| err = compute_write_protect_range(mmc, name, ext_csd, |
| &wp_grp_size, &start, &part_end); |
| if (err) |
| return 1; |
| |
| group_start = start; |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = send_write_prot_status_group(mmc, group_start); |
| if (err) |
| return 1; |
| group_start += 32 * wp_grp_size; |
| } |
| |
| return 0; |
| } |
| |
| static int send_add_wp_status(struct mmc *mmc, u64 start, u64 cnt) |
| { |
| int err = 0; |
| u8 ext_csd[512] = {0}; |
| u64 wp_grp_size, part_end; |
| u64 group_start; |
| int blk_shift; |
| u64 mmc_boundary; |
| err = mmc_get_ext_csd(mmc, ext_csd); |
| if (err) { |
| printf("Failed: get ext_csd failed\n"); |
| return 1; |
| } |
| |
| if (!is_write_protect_valid(ext_csd)) { |
| printf("Failed: CLASS_6_CTRL isn't '0' " |
| "write protect process is invalid\n"); |
| return 1; |
| } |
| |
| wp_grp_size = write_protect_group_size(mmc, ext_csd); |
| |
| if ((start % wp_grp_size)) { |
| group_start = (start + wp_grp_size - 1) / wp_grp_size * wp_grp_size; |
| printf("Caution! The partition start address isn't' aligned" |
| "to group size\n" |
| "the start address is change from 0x%llx to 0x%llx\n", |
| start, group_start); |
| part_end = group_start + (cnt - 1) * wp_grp_size - 1; |
| printf("The write protect group number is 0x%llx, rather than 0x%lld\n", |
| cnt - 1, cnt); |
| } else { |
| group_start = start; |
| part_end = group_start + cnt * wp_grp_size - 1; |
| } |
| |
| blk_shift = ffs(mmc->read_bl_len) - 1; |
| mmc_boundary = mmc->capacity>>blk_shift; |
| |
| if ((part_end + 1) > mmc_boundary) { |
| printf("Error: the operation cross the boundary of mmc\n"); |
| return 1; |
| } |
| |
| while ((group_start + wp_grp_size - 1) <= part_end) { |
| err = send_write_prot_status_group(mmc, group_start); |
| if (err) |
| return 1; |
| group_start += 32 * wp_grp_size; |
| } |
| |
| return 0; |
| } |
| |
| static int do_amlmmc_send_wp_status(cmd_tbl_t *cmdtp, |
| int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| struct mmc *mmc; |
| int dev = 1; |
| char *name = NULL; |
| u64 start, cnt; |
| |
| if (argc < 3 || argc > 4) |
| return ret; |
| |
| if (argc == 3) { |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name(name); |
| if (dev < 0) { |
| printf("Error: Cannot find dev.\n"); |
| return 1; |
| } |
| } else { |
| start = simple_strtoull(argv[2], NULL, 16); |
| cnt = simple_strtoull(argv[3], NULL, 0); |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (IS_SD(mmc)) { |
| mmc = find_mmc_device(~dev); |
| if (IS_SD(mmc)) { |
| printf("SD card can not be write protect\n"); |
| return 1; |
| } |
| } |
| |
| if (!mmc) |
| return 1; |
| |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| |
| if (argc == 3) |
| ret = send_part_wp_status(mmc, name); |
| else |
| ret = send_add_wp_status(mmc, start, cnt); |
| |
| if (ret) { |
| printf("Failed: send partition write protect status failed\n"); |
| } |
| |
| return ret; |
| } |
| |
| static int do_amlmmc_send_wp_type(cmd_tbl_t *cmdtp, |
| int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| struct mmc *mmc; |
| int dev = 1; |
| char *name = NULL; |
| u64 start, cnt; |
| |
| if (argc < 3 || argc > 4) |
| return ret; |
| |
| if (argc == 3) { |
| name = argv[2]; |
| dev = find_dev_num_by_partition_name(name); |
| if (dev < 0) { |
| printf("Error: Cannot find dev.\n"); |
| return 1; |
| } |
| } else { |
| start = simple_strtoull(argv[2], NULL, 16); |
| cnt = simple_strtoull(argv[3], NULL, 0); |
| } |
| |
| mmc = find_mmc_device(dev); |
| |
| if (!mmc) |
| return 1; |
| |
| if (IS_SD(mmc)) { |
| mmc = find_mmc_device(~dev); |
| if (IS_SD(mmc)) { |
| printf("SD card can not be write protect\n"); |
| return 1; |
| } |
| } |
| |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| if (argc == 3) |
| ret = send_part_wp_type(mmc, name); |
| else |
| ret = send_add_wp_type(mmc, start, cnt); |
| |
| if (ret) { |
| printf("Failed: send parittion write protect type failed\n"); |
| } |
| |
| return ret; |
| } |
| |
| static int set_driver_strength(struct mmc *mmc, int strength) |
| { |
| int ret = 0; |
| u8 ext_csd[512] = {0}; |
| u8 strength_type = 0; |
| u8 driver_strength; |
| u8 hs_timing = 0; |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| if (ret) { |
| printf("get ext_csd failed\n"); |
| return ret; |
| } |
| strength_type = 1 << strength; |
| driver_strength = ext_csd[EXT_CSD_DRIVER_STRENGTH]; |
| if (0 == (strength_type & driver_strength)) { |
| printf("Failed: This device didn't support strength type %d\n", strength); |
| return 1; |
| } |
| |
| hs_timing = ext_csd[EXT_CSD_HS_TIMING]; |
| if ((hs_timing >> 4) > 0) { |
| printf("Failed: The driver strength has been set already, \ |
| please reset the device\n"); |
| return 1; |
| } |
| |
| hs_timing = hs_timing | (strength << 4); |
| |
| ret = mmc_set_ext_csd(mmc, EXT_CSD_HS_TIMING, hs_timing); |
| if (ret) { |
| printf("set ext_csd hs_timing field failed\n"); |
| return ret; |
| } |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| if (ret) { |
| printf("get ext_csd failed\n"); |
| return ret; |
| } |
| printf("The ext_csd[HS_TIMING] has been set to 0x%x\n", |
| ext_csd[EXT_CSD_HS_TIMING]); |
| return ret; |
| } |
| |
| static int get_driver_strength(struct mmc *mmc) |
| { |
| int ret = 0; |
| u8 ext_csd[512] = {0}; |
| u8 support_ds_type = 0; |
| u8 cur_driver_strength; |
| u8 hs_timing = 0; |
| ret = mmc_get_ext_csd(mmc, ext_csd); |
| if (ret) { |
| printf("get ext_csd failed\n"); |
| return ret; |
| } |
| |
| support_ds_type = ext_csd[EXT_CSD_DRIVER_STRENGTH]; |
| |
| hs_timing = ext_csd[EXT_CSD_HS_TIMING]; |
| cur_driver_strength = hs_timing >> 4; |
| |
| printf("current strength type is: "); |
| int strength_type = 0; |
| while (support_ds_type) { |
| if (support_ds_type & 1) { |
| if (cur_driver_strength == strength_type) |
| printf("[%d] ", strength_type); |
| else |
| printf("%d ", strength_type); |
| } |
| strength_type++; |
| support_ds_type = support_ds_type >> 1; |
| } |
| printf("\n"); |
| return ret; |
| } |
| |
| static int amlmmc_set_driver_strength(int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| int dev, strength; |
| struct mmc *mmc; |
| |
| if (argc != 4) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| strength = simple_strtoul(argv[3], NULL, 10); |
| mmc = find_mmc_device(dev); |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| |
| ret = set_driver_strength(mmc, strength); |
| |
| return ret; |
| } |
| |
| static int amlmmc_get_driver_strength(int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| int dev; |
| struct mmc *mmc; |
| |
| if (argc != 3) |
| return CMD_RET_USAGE; |
| |
| dev = simple_strtoul(argv[2], NULL, 10); |
| mmc = find_mmc_device(dev); |
| if (!mmc) { |
| puts("no mmc devices available\n"); |
| return 1; |
| } |
| mmc_init(mmc); |
| if (!mmc) |
| return 1; |
| |
| ret = get_driver_strength(mmc); |
| |
| return ret; |
| } |
| |
| static int do_amlmmc_driver_strength(cmd_tbl_t *cmdtp, |
| int flag, int argc, char *const argv[]) |
| { |
| int ret = CMD_RET_USAGE; |
| |
| if (argc == 3) |
| ret = amlmmc_get_driver_strength(argc,argv); |
| else if (argc == 4) |
| ret = amlmmc_set_driver_strength(argc,argv); |
| |
| return ret; |
| } |
| |
| static cmd_tbl_t cmd_amlmmc[] = { |
| U_BOOT_CMD_MKENT(read, 6, 0, do_amlmmc_read, "", ""), |
| U_BOOT_CMD_MKENT(write, 6, 0, do_amlmmc_write, "", ""), |
| U_BOOT_CMD_MKENT(erase, 5, 0, do_amlmmc_erase, "", ""), |
| U_BOOT_CMD_MKENT(rescan, 3, 0, do_amlmmc_rescan, "", ""), |
| U_BOOT_CMD_MKENT(part, 3, 0, do_amlmmc_part, "", ""), |
| U_BOOT_CMD_MKENT(list, 2, 0, do_amlmmc_list, "", ""), |
| U_BOOT_CMD_MKENT(switch, 4, 0, do_amlmmc_switch, "", ""), |
| U_BOOT_CMD_MKENT(status, 3, 0, do_amlmmc_status, "", ""), |
| U_BOOT_CMD_MKENT(ext_csd, 5, 0, do_amlmmc_ext_csd, "", ""), |
| U_BOOT_CMD_MKENT(response, 3, 0, do_amlmmc_response, "", ""), |
| U_BOOT_CMD_MKENT(controller, 3, 0, do_amlmmc_controller, "", ""), |
| U_BOOT_CMD_MKENT(size, 4, 0, do_amlmmc_size, "", ""), |
| U_BOOT_CMD_MKENT(env, 2, 0, do_amlmmc_env, "", ""), |
| U_BOOT_CMD_MKENT(write_protect, 5, 0, do_amlmmc_write_protect, "", ""), |
| U_BOOT_CMD_MKENT(send_wp_status, 4, 0, do_amlmmc_send_wp_status, "", ""), |
| U_BOOT_CMD_MKENT(send_wp_type, 4, 0, do_amlmmc_send_wp_type, "", ""), |
| U_BOOT_CMD_MKENT(clear_wp, 4, 0, do_amlmmc_clear_wp, "", ""), |
| U_BOOT_CMD_MKENT(ds, 4, 0, do_amlmmc_driver_strength, "", ""), |
| #ifdef CONFIG_SECURITYKEY |
| U_BOOT_CMD_MKENT(key, 2, 0, do_amlmmc_key, "", ""), |
| #endif |
| }; |
| |
| static int do_amlmmcops(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
| { |
| cmd_tbl_t *cp; |
| |
| cp = find_cmd_tbl(argv[1], cmd_amlmmc, ARRAY_SIZE(cmd_amlmmc)); |
| |
| if (cp == NULL || argc > cp->maxargs) |
| return CMD_RET_USAGE; |
| |
| if (flag == CMD_FLAG_REPEAT && !cp->repeatable) |
| return CMD_RET_SUCCESS; |
| |
| return cp->cmd(cmdtp, flag, argc, argv); |
| } |
| |
| U_BOOT_CMD( |
| amlmmc, 6, 1, do_amlmmcops, |
| "AMLMMC sub system", |
| "read <partition_name> ram_addr addr_byte# cnt_byte\n" |
| "amlmmc write <partition_name> ram_addr addr_byte# cnt_byte\n" |
| "amlmmc erase <partition_name> addr_byte# cnt_byte\n" |
| "amlmmc erase <partition_name>/<device num>\n" |
| "amlmmc rescan <device_num>\n" |
| "amlmmc part <device_num> - show partition infomation of mmc\n" |
| "amlmmc list - lists available devices\n" |
| "amlmmc env - display env partition offset\n" |
| "amlmmc switch <device_num> <part name> - part name : boot0, boot1, user\n" |
| "amlmmc status <device_num> - read sd/emmc device status\n" |
| "amlmmc ext_csd <device_num> <byte> - read sd/emmc device EXT_CSD [byte]\n" |
| "amlmmc ext_csd <device_num> <byte> <value> - write sd/emmc device EXT_CSD [byte] value\n" |
| "amlmmc response <device_num> - read sd/emmc last command response\n" |
| "amlmmc controller <device_num> - read sd/emmc controller register\n" |
| "amlmmc write_protect <partition_name> <write_protect_type>\n" |
| " - set write protect on partition through power_on or temporary\n" |
| "amlmmc write_protect <addr_base16> <cnt_base10> <write_protect_type>\n" |
| " - set write protect on specified address through power_on or temporary\n" |
| "amlmmc send_wp_status <partition_name> send protect status of partition\n" |
| "amlmmc send_wp_status <addr_base16> <cnt_base10> send protect status on specified address\n" |
| "amlmmc send_wp_type <partition_name> send protect type of partition\n" |
| "amlmmc send_wp_type <addr_base16> <cnt_base10> send protect type on specified address\n" |
| "amlmmc clear_wp <partition_name> clear write protect of partition\n" |
| "amlmmc clear_wp <addr_base16> <cnt_base10> clear write protect on specified addresst\n" |
| "amlmmc ds <dev_num> <val> set driver strength\n" |
| #ifdef CONFIG_SECURITYKEY |
| "amlmmc key - disprotect key partition\n" |
| #endif |
| ); |
| |
| /* dtb read&write operation with backup updates */ |
| static u32 _calc_dtb_checksum(struct aml_dtb_rsv * dtb) |
| { |
| int i = 0; |
| int size = sizeof(struct aml_dtb_rsv) - sizeof(u32); |
| u32 * buffer; |
| u32 checksum = 0; |
| |
| if ((u64)dtb % 4 != 0) { |
| BUG(); |
| } |
| |
| size = size >> 2; |
| buffer = (u32*) dtb; |
| while (i < size) |
| checksum += buffer[i++]; |
| |
| return checksum; |
| } |
| |
| static int _verify_dtb_checksum(struct aml_dtb_rsv * dtb) |
| { |
| u32 checksum; |
| |
| checksum = _calc_dtb_checksum(dtb); |
| dtb_info("calc %x, store %x\n", checksum, dtb->checksum); |
| |
| return !(checksum == dtb->checksum); |
| } |
| |
| static int _dtb_read(struct mmc *mmc, u64 blk, u64 cnt, void * addr) |
| { |
| int dev = EMMC_DTB_DEV; |
| u64 n; |
| n = blk_dread(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| if (n != cnt) { |
| dtb_err("%s: dev # %d, block # %#llx, count # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| } |
| |
| return (n != cnt); |
| } |
| |
| static int _dtb_write(struct mmc *mmc, u64 blk, u64 cnt, void * addr) |
| { |
| int dev = EMMC_DTB_DEV; |
| u64 n; |
| n = blk_dwrite(mmc_get_blk_desc(mmc), blk, cnt, addr); |
| if (n != cnt) { |
| dtb_err("%s: dev # %d, block # %#llx, count # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| } |
| |
| return (n != cnt); |
| } |
| |
| static struct mmc *_dtb_init(void) |
| { |
| struct mmc *mmc = find_mmc_device(EMMC_DTB_DEV); |
| if (!mmc) { |
| dtb_err("not find mmc\n"); |
| return NULL; |
| } |
| |
| if (mmc_init(mmc)) { |
| dtb_err("mmc init failed\n"); |
| return NULL; |
| } |
| return mmc; |
| } |
| |
| static int dtb_read_shortcut(struct mmc * mmc, void *addr) |
| { |
| u64 blk, cnt, dtb_glb_offset; |
| int dev = EMMC_DTB_DEV; |
| struct aml_dtb_info *info = &dtb_infos; |
| struct partitions * part = NULL; |
| struct virtual_partition *vpart = NULL; |
| vpart = aml_get_virtual_partition_by_name(MMC_DTB_NAME); |
| part = aml_get_partition_by_name(MMC_RESERVED_NAME); |
| dtb_glb_offset = part->offset + vpart->offset; |
| /* short cut */ |
| if (info->valid[0]) { |
| dtb_info("short cut in...\n"); |
| blk = dtb_glb_offset / mmc->read_bl_len; |
| cnt = vpart->size / mmc->read_bl_len; |
| if (_dtb_read(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| /*try dtb2 if it's valid */ |
| if (info->valid[1]) { |
| blk = (dtb_glb_offset + vpart->size) / mmc->read_bl_len; |
| cnt = vpart->size / mmc->read_bl_len; |
| if (_dtb_read(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx, cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| return -1; |
| } |
| } |
| } |
| return 0; |
| } |
| return -2; |
| } |
| |
| static int update_dtb_info(struct mmc *mmc, void *addr) |
| { |
| int ret = 0, dev = EMMC_DTB_DEV; |
| u64 blk, cnt, dtb_glb_offset; |
| struct aml_dtb_rsv * dtb = (struct aml_dtb_rsv *) addr; |
| struct aml_dtb_info *info = &dtb_infos; |
| int cpy = 1, valid = 0; |
| struct partitions * part = NULL; |
| struct virtual_partition *vpart = NULL; |
| vpart = aml_get_virtual_partition_by_name(MMC_DTB_NAME); |
| part = aml_get_partition_by_name(MMC_RESERVED_NAME); |
| dtb_glb_offset = part->offset + vpart->offset; |
| |
| while (cpy >= 0) { |
| blk = (dtb_glb_offset + cpy * (vpart->size)) / mmc->read_bl_len; |
| cnt = vpart->size / mmc->read_bl_len; |
| ret = _dtb_read(mmc, blk, cnt, addr); |
| if (ret) { |
| dtb_err("%s: dev # %d, block # %#llx, cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| } else { |
| ret = _verify_dtb_checksum(dtb); |
| /* check magic avoid whole 0 issue */ |
| if (!ret && (dtb->magic != 0)) { |
| info->stamp[cpy] = dtb->timestamp; |
| info->valid[cpy] = 1; |
| } |
| else |
| dtb_wrn("cpy %d is not valid\n", cpy); |
| } |
| valid += info->valid[cpy]; |
| cpy --; |
| } |
| return valid; |
| } |
| |
| static int update_invalid_dtb(struct mmc *mmc, void *addr) |
| { |
| int ret = 0, dev = EMMC_DTB_DEV; |
| u64 blk, cnt, dtb_glb_offset; |
| struct aml_dtb_rsv * dtb = (struct aml_dtb_rsv *) addr; |
| struct aml_dtb_info *info = &dtb_infos; |
| struct partitions * part = NULL; |
| struct virtual_partition *vpart = NULL; |
| vpart = aml_get_virtual_partition_by_name(MMC_DTB_NAME); |
| part = aml_get_partition_by_name(MMC_RESERVED_NAME); |
| dtb_glb_offset = part->offset + vpart->offset; |
| cnt = vpart->size / mmc->read_bl_len; |
| |
| if (info->valid[1]) { |
| blk = (dtb_glb_offset + vpart->size) / mmc->read_bl_len; |
| if (_dtb_read(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -2; |
| } |
| /* fixme, update the invalid one - dtb1 */ |
| blk = (dtb_glb_offset) / mmc->read_bl_len; |
| if (_dtb_write(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -4; |
| } |
| info->valid[0] = 1; |
| info->stamp[0] = dtb->timestamp; |
| ret = 0; |
| } else { |
| dtb_info("update dtb2"); |
| blk = (dtb_glb_offset + vpart->size) / mmc->read_bl_len; |
| if (_dtb_write(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -2; |
| } |
| info->valid[1] = 1; |
| info->stamp[1] = dtb->timestamp; |
| } |
| return ret; |
| } |
| |
| int update_old_dtb(struct mmc *mmc, void *addr) |
| { |
| int ret = 0, dev = EMMC_DTB_DEV; |
| u64 blk, cnt, dtb_glb_offset; |
| struct aml_dtb_rsv * dtb = (struct aml_dtb_rsv *) addr; |
| struct aml_dtb_info *info = &dtb_infos; |
| struct partitions * part = NULL; |
| struct virtual_partition *vpart = NULL; |
| vpart = aml_get_virtual_partition_by_name(MMC_DTB_NAME); |
| part = aml_get_partition_by_name(MMC_RESERVED_NAME); |
| dtb_glb_offset = part->offset + vpart->offset; |
| cnt = vpart->size / mmc->read_bl_len; |
| if (stamp_after(info->stamp[1], info->stamp[0])) { |
| blk = (dtb_glb_offset + vpart->size) / mmc->read_bl_len; |
| if (_dtb_read(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -3; |
| } |
| /*update dtb1*/ |
| blk = dtb_glb_offset / mmc->read_bl_len; |
| if (_dtb_write(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -3; |
| } |
| info->stamp[0] = dtb->timestamp; |
| ret = 0; |
| } else if (stamp_after(info->stamp[0], info->stamp[1])) { |
| /*update dtb2*/ |
| blk = (dtb_glb_offset + vpart->size) / mmc->read_bl_len; |
| if (_dtb_write(mmc, blk, cnt, addr)) { |
| dtb_err("%s: dev # %d, block # %#llx,cnt # %#llx ERROR!\n", |
| __func__, dev, blk, cnt); |
| ret = -3; |
| } |
| info->stamp[ |