common: android: clean up code

Signed-off-by: Joseph Chen <chenjh@rock-chips.com>
Change-Id: I98dfd54edd813f892c5f2717a694b2424b8ecded
This commit is contained in:
Joseph Chen 2020-09-05 14:53:25 +08:00 committed by Jianhong Chen
parent f870210e35
commit 69b1ad4693
2 changed files with 69 additions and 83 deletions

View File

@ -846,11 +846,10 @@ __weak int board_select_fdt_index(ulong dt_table_hdr)
static int android_get_dtbo(ulong *fdt_dtbo, static int android_get_dtbo(ulong *fdt_dtbo,
const struct andr_img_hdr *hdr, const struct andr_img_hdr *hdr,
int *index, int boot_mode) int *index, const char *part_dtbo)
{ {
struct dt_table_header *dt_hdr = NULL; struct dt_table_header *dt_hdr = NULL;
struct blk_desc *dev_desc; struct blk_desc *dev_desc;
const char *part_name;
disk_partition_t part_info; disk_partition_t part_info;
u32 blk_offset, blk_cnt; u32 blk_offset, blk_cnt;
void *buf; void *buf;
@ -859,49 +858,34 @@ static int android_get_dtbo(ulong *fdt_dtbo,
int e_idx; int e_idx;
int ret; int ret;
/* Get partition according to boot mode */
if (boot_mode == BOOT_MODE_RECOVERY)
part_name = PART_RECOVERY;
else
part_name = PART_DTBO;
/* Get partition info */ /* Get partition info */
dev_desc = rockchip_get_bootdev(); dev_desc = rockchip_get_bootdev();
if (!dev_desc) { if (!dev_desc)
printf("%s: dev_desc is NULL!\n", __func__);
return -ENODEV; return -ENODEV;
}
ret = part_get_info_by_name(dev_desc, part_name, &part_info); ret = part_get_info_by_name(dev_desc, part_dtbo, &part_info);
if (ret < 0) { if (ret < 0) {
printf("%s: failed to get %s part info, ret=%d\n", printf("No %s partition, ret=%d\n", part_dtbo, ret);
__func__, part_name, ret);
return ret; return ret;
} }
/* Check dt table header */ /* Check dt table header */
if (!strcmp(part_name, PART_RECOVERY)) if (!strcmp(part_dtbo, PART_RECOVERY))
blk_offset = part_info.start + blk_offset = part_info.start +
(hdr->recovery_dtbo_offset / part_info.blksz); (hdr->recovery_dtbo_offset / part_info.blksz);
else else
blk_offset = part_info.start; blk_offset = part_info.start;
dt_hdr = memalign(ARCH_DMA_MINALIGN, part_info.blksz); dt_hdr = memalign(ARCH_DMA_MINALIGN, part_info.blksz);
if (!dt_hdr) { if (!dt_hdr)
printf("%s: out of memory for dt header!\n", __func__);
return -ENOMEM; return -ENOMEM;
}
ret = blk_dread(dev_desc, blk_offset, 1, dt_hdr); ret = blk_dread(dev_desc, blk_offset, 1, dt_hdr);
if (ret != 1) { if (ret != 1)
printf("%s: failed to read dt table header\n",
__func__);
goto out1; goto out1;
}
if (!android_dt_check_header((ulong)dt_hdr)) { if (!android_dt_check_header((ulong)dt_hdr)) {
printf("%s: Error: invalid dt table header: 0x%x\n", printf("DTBO: invalid dt table header: 0x%x\n", dt_hdr->magic);
__func__, dt_hdr->magic);
ret = -EINVAL; ret = -EINVAL;
goto out1; goto out1;
} }
@ -915,17 +899,13 @@ static int android_get_dtbo(ulong *fdt_dtbo,
/* Read all DT Image */ /* Read all DT Image */
buf = memalign(ARCH_DMA_MINALIGN, part_info.blksz * blk_cnt); buf = memalign(ARCH_DMA_MINALIGN, part_info.blksz * blk_cnt);
if (!buf) { if (!buf) {
printf("%s: out of memory for %s part!\n", __func__, part_name);
ret = -ENOMEM; ret = -ENOMEM;
goto out1; goto out1;
} }
ret = blk_dread(dev_desc, blk_offset, blk_cnt, buf); ret = blk_dread(dev_desc, blk_offset, blk_cnt, buf);
if (ret != blk_cnt) { if (ret != blk_cnt)
printf("%s: failed to read dtbo, blk_cnt=%d, ret=%d\n",
__func__, blk_cnt, ret);
goto out2; goto out2;
}
e_idx = board_select_fdt_index((ulong)buf); e_idx = board_select_fdt_index((ulong)buf);
if (e_idx < 0) { if (e_idx < 0) {
@ -948,7 +928,7 @@ static int android_get_dtbo(ulong *fdt_dtbo,
free(dt_hdr); free(dt_hdr);
debug("ANDROID: Loading dt entry to 0x%lx size 0x%x idx %d from \"%s\" part\n", debug("ANDROID: Loading dt entry to 0x%lx size 0x%x idx %d from \"%s\" part\n",
e_addr, e_size, e_idx, part_name); e_addr, e_size, e_idx, part_dtbo);
return 0; return 0;
@ -964,38 +944,31 @@ int android_fdt_overlay_apply(void *fdt_addr)
{ {
struct andr_img_hdr *hdr; struct andr_img_hdr *hdr;
struct blk_desc *dev_desc; struct blk_desc *dev_desc;
const char *part_name; const char *part_boot;
disk_partition_t part_info; disk_partition_t part_info;
char *part_dtbo;
char buf[32] = {0}; char buf[32] = {0};
u32 blk_cnt; u32 blk_cnt;
ulong fdt_dtbo = -1; ulong fdt_dtbo = -1;
int boot_mode;
int index = -1; int index = -1;
int ret; int ret;
boot_mode = rockchip_get_boot_mode(); if (IS_ENABLED(CONFIG_ANDROID_AB) ||
#ifdef CONFIG_ANDROID_AB (rockchip_get_boot_mode() != BOOT_MODE_RECOVERY)) {
if (boot_mode == BOOT_MODE_RECOVERY) part_boot = PART_BOOT;
boot_mode = BOOT_MODE_NORMAL; part_dtbo = PART_DTBO;
#endif } else {
if (boot_mode == BOOT_MODE_RECOVERY) part_boot = PART_RECOVERY;
part_name = PART_RECOVERY; part_dtbo = PART_RECOVERY;
else }
part_name = PART_BOOT;
/* Get partition info */
dev_desc = rockchip_get_bootdev(); dev_desc = rockchip_get_bootdev();
if (!dev_desc) { if (!dev_desc)
printf("%s: dev_desc is NULL!\n", __func__);
return -ENODEV; return -ENODEV;
}
ret = part_get_info_by_name(dev_desc, part_name, &part_info); ret = part_get_info_by_name(dev_desc, part_boot, &part_info);
if (ret < 0) { if (ret < 0)
printf("%s: failed to get %s part info, ret=%d\n",
__func__, part_name, ret);
return ret; return ret;
}
blk_cnt = DIV_ROUND_UP(sizeof(*hdr), part_info.blksz); blk_cnt = DIV_ROUND_UP(sizeof(*hdr), part_info.blksz);
hdr = memalign(ARCH_DMA_MINALIGN, part_info.blksz * blk_cnt); hdr = memalign(ARCH_DMA_MINALIGN, part_info.blksz * blk_cnt);
@ -1006,7 +979,7 @@ int android_fdt_overlay_apply(void *fdt_addr)
ret = blk_dread(dev_desc, part_info.start, blk_cnt, hdr); ret = blk_dread(dev_desc, part_info.start, blk_cnt, hdr);
if (ret != blk_cnt) { if (ret != blk_cnt) {
printf("%s: failed to read %s hdr!\n", __func__, part_name); printf("%s: failed to read %s hdr!\n", __func__, part_boot);
goto out; goto out;
} }
@ -1024,7 +997,7 @@ int android_fdt_overlay_apply(void *fdt_addr)
goto out; goto out;
} }
ret = android_get_dtbo(&fdt_dtbo, (void *)hdr, &index, boot_mode); ret = android_get_dtbo(&fdt_dtbo, (void *)hdr, &index, part_dtbo);
if (!ret) { if (!ret) {
phys_size_t fdt_size; phys_size_t fdt_size;
/* Must incease size before overlay */ /* Must incease size before overlay */

View File

@ -28,7 +28,7 @@
DECLARE_GLOBAL_DATA_PTR; DECLARE_GLOBAL_DATA_PTR;
#define ANDROID_IMAGE_DEFAULT_KERNEL_ADDR 0x10008000 #define ANDROID_IMAGE_DEFAULT_KERNEL_ADDR 0x10008000
#define ANDROID_ARG_FDT_FILENAME "rk-kernel.dtb" #define ANDROID_ARG_FDT_FILENAME "rk-kernel.dtb"
#define ANDROID_Q_VER 10 #define ANDROID_Q_VER 10
/* Defined by rockchip legacy mkboot tool(SDK version < 8.1) */ /* Defined by rockchip legacy mkboot tool(SDK version < 8.1) */
@ -47,15 +47,15 @@ u32 android_bcb_msg_sector_offset(void)
{ {
/* /*
* Rockchip platforms defines BCB message at the 16KB offset of * Rockchip platforms defines BCB message at the 16KB offset of
* misc partition while the Google defines it at 0x0 offset. * misc partition while the Google defines it at 0x00 offset.
* *
* From Android-Q, the 0x0 offset is mandary on Google VTS, so that * From Android-Q, the 0x00 offset is mandary on Google VTS, so that
* this is a compatibility according to android image 'os_version'. * this is a compatibility according to android image 'os_version'.
*/ */
#ifdef CONFIG_RKIMG_BOOTLOADER #ifdef CONFIG_RKIMG_BOOTLOADER
return (android_image_major_version() >= ANDROID_Q_VER) ? 0x0 : 0x20; return (android_image_major_version() >= ANDROID_Q_VER) ? 0x00 : 0x20;
#else #else
return 0x0; return 0x00;
#endif #endif
} }
@ -296,7 +296,7 @@ typedef enum {
IMG_MAX, IMG_MAX,
} img_t; } img_t;
static int image_read(img_t img, struct andr_img_hdr *hdr, static int image_load(img_t img, struct andr_img_hdr *hdr,
ulong blkstart, void *ram_base, ulong blkstart, void *ram_base,
struct udevice *crypto) struct udevice *crypto)
{ {
@ -422,7 +422,7 @@ static int android_image_separate(struct andr_img_hdr *hdr,
return -EINVAL; return -EINVAL;
} }
/* set for image_read(IMG_KERNEL, ...) */ /* set for image_load(IMG_KERNEL, ...) */
env_set_hex("android_addr_r", (ulong)load_address); env_set_hex("android_addr_r", (ulong)load_address);
bstart = part ? part->start : 0; bstart = part ? part->start : 0;
@ -457,20 +457,20 @@ static int android_image_separate(struct andr_img_hdr *hdr,
crypto_sha_init(dev, &ctx); crypto_sha_init(dev, &ctx);
/* load, never change order ! */ /* load, never change order ! */
if (image_read(IMG_RK_DTB, hdr, bstart, ram_base, NULL)) if (image_load(IMG_RK_DTB, hdr, bstart, ram_base, NULL))
return -1; return -1;
if (image_read(IMG_KERNEL, hdr, bstart, ram_base, dev)) if (image_load(IMG_KERNEL, hdr, bstart, ram_base, dev))
return -1; return -1;
if (image_read(IMG_RAMDISK, hdr, bstart, ram_base, dev)) if (image_load(IMG_RAMDISK, hdr, bstart, ram_base, dev))
return -1; return -1;
if (image_read(IMG_SECOND, hdr, bstart, ram_base, dev)) if (image_load(IMG_SECOND, hdr, bstart, ram_base, dev))
return -1; return -1;
if (hdr->header_version > 0) { if (hdr->header_version > 0) {
if (image_read(IMG_RECOVERY_DTBO, hdr, bstart, ram_base, dev)) if (image_load(IMG_RECOVERY_DTBO, hdr, bstart, ram_base, dev))
return -1; return -1;
} }
if (hdr->header_version > 1) { if (hdr->header_version > 1) {
if (image_read(IMG_DTB, hdr, bstart, ram_base, dev)) if (image_load(IMG_DTB, hdr, bstart, ram_base, dev))
return -1; return -1;
} }
@ -484,20 +484,20 @@ static int android_image_separate(struct andr_img_hdr *hdr,
} }
#else /* !(CONFIG_DM_CRYPTO && CONFIG_ANDROID_BOOT_IMAGE_HASH) */ #else /* !(CONFIG_DM_CRYPTO && CONFIG_ANDROID_BOOT_IMAGE_HASH) */
if (image_read(IMG_RK_DTB, hdr, bstart, ram_base, NULL)) if (image_load(IMG_RK_DTB, hdr, bstart, ram_base, NULL))
return -1; return -1;
if (image_read(IMG_KERNEL, hdr, bstart, ram_base, NULL)) if (image_load(IMG_KERNEL, hdr, bstart, ram_base, NULL))
return -1; return -1;
if (image_read(IMG_RAMDISK, hdr, bstart, ram_base, NULL)) if (image_load(IMG_RAMDISK, hdr, bstart, ram_base, NULL))
return -1; return -1;
if (image_read(IMG_SECOND, hdr, bstart, ram_base, NULL)) if (image_load(IMG_SECOND, hdr, bstart, ram_base, NULL))
return -1; return -1;
if (hdr->header_version > 0) { if (hdr->header_version > 0) {
if (image_read(IMG_RECOVERY_DTBO, hdr, bstart, ram_base, NULL)) if (image_load(IMG_RECOVERY_DTBO, hdr, bstart, ram_base, NULL))
return -1; return -1;
} }
if (hdr->header_version > 1) { if (hdr->header_version > 1) {
if (image_read(IMG_DTB, hdr, bstart, ram_base, NULL)) if (image_load(IMG_DTB, hdr, bstart, ram_base, NULL))
return -1; return -1;
} }
#endif #endif
@ -508,17 +508,12 @@ static int android_image_separate(struct andr_img_hdr *hdr,
return 0; return 0;
} }
/* static ulong android_image_get_comp_addr(struct andr_img_hdr *hdr, int comp)
* 'boot_android' cmd use "kernel_addr_r" as default load address !
* We update it according to compress type and "kernel_addr_c/r".
*/
int android_image_parse_comp(struct andr_img_hdr *hdr, ulong *load_addr)
{ {
ulong kernel_addr_c; ulong kernel_addr_c;
int comp; ulong load_addr = 0;
kernel_addr_c = env_get_ulong("kernel_addr_c", 16, 0); kernel_addr_c = env_get_ulong("kernel_addr_c", 16, 0);
comp = android_image_parse_kernel_comp(hdr);
#ifdef CONFIG_ARM64 #ifdef CONFIG_ARM64
/* /*
@ -538,7 +533,7 @@ int android_image_parse_comp(struct andr_img_hdr *hdr, ulong *load_addr)
env_set_hex("kernel_addr_c", comp_addr); env_set_hex("kernel_addr_c", comp_addr);
} }
*load_addr = comp_addr - hdr->page_size; load_addr = comp_addr - hdr->page_size;
} }
#else #else
/* /*
@ -555,22 +550,40 @@ int android_image_parse_comp(struct andr_img_hdr *hdr, ulong *load_addr)
/* input load_addr is for Image, nothing to do */ /* input load_addr is for Image, nothing to do */
} else { } else {
/* input load_addr is for lz4/zImage, set default addr for Image */ /* input load_addr is for lz4/zImage, set default addr for Image */
*load_addr = CONFIG_SYS_SDRAM_BASE + 0x8000; load_addr = CONFIG_SYS_SDRAM_BASE + 0x8000;
env_set_hex("kernel_addr_r", *load_addr); env_set_hex("kernel_addr_r", load_addr);
*load_addr -= hdr->page_size; load_addr -= hdr->page_size;
} }
} else { } else {
if (kernel_addr_c) { if (kernel_addr_c) {
/* input load_addr is for Image, so use another for lz4/zImage */ /* input load_addr is for Image, so use another for lz4/zImage */
*load_addr = kernel_addr_c - hdr->page_size; load_addr = kernel_addr_c - hdr->page_size;
} else { } else {
/* input load_addr is for lz4/zImage, nothing to do */ /* input load_addr is for lz4/zImage, nothing to do */
} }
} }
#endif #endif
return load_addr;
}
/*
* 'boot_android' cmd use "kernel_addr_r" as default load address !
* We update it according to compress type and "kernel_addr_c/r".
*/
int android_image_parse_comp(struct andr_img_hdr *hdr, ulong *load_addr)
{
ulong new_load_addr;
int comp;
comp = android_image_parse_kernel_comp(hdr);
env_set_ulong("os_comp", comp); env_set_ulong("os_comp", comp);
new_load_addr = android_image_get_comp_addr(hdr, comp);
if (new_load_addr != 0)
*load_addr = new_load_addr;
return comp; return comp;
} }