1 /* 2 * (C) Copyright 2017 Rockchip Electronics Co., Ltd 3 * 4 * SPDX-License-Identifier: GPL-2.0+ 5 */ 6 7 #include <common.h> 8 #include <attestation_key.h> 9 #include <bootm.h> 10 #include <boot_rkimg.h> 11 #include <keymaster.h> 12 #include <console.h> 13 #include <image.h> 14 #include <malloc.h> 15 #include <sysmem.h> 16 #include <linux/libfdt.h> 17 #include <asm/arch/hotkey.h> 18 #include <asm/arch/resource_img.h> 19 #include <asm/arch/boot_mode.h> 20 21 DECLARE_GLOBAL_DATA_PTR; 22 23 #ifdef CONFIG_ROCKCHIP_CRC 24 #define tole(x) cpu_to_le32(x) 25 26 /* Table of CRC-32's of all single-byte values (made by make_crc_table) */ 27 static const uint32_t crc_table[256] = { 28 tole(0x00000000L), tole(0x04c10db7L), tole(0x09821b6eL), tole(0x0d4316d9L), 29 tole(0x130436dcL), tole(0x17c53b6bL), tole(0x1a862db2L), tole(0x1e472005L), 30 tole(0x26086db8L), tole(0x22c9600fL), tole(0x2f8a76d6L), tole(0x2b4b7b61L), 31 tole(0x350c5b64L), tole(0x31cd56d3L), tole(0x3c8e400aL), tole(0x384f4dbdL), 32 tole(0x4c10db70L), tole(0x48d1d6c7L), tole(0x4592c01eL), tole(0x4153cda9L), 33 tole(0x5f14edacL), tole(0x5bd5e01bL), tole(0x5696f6c2L), tole(0x5257fb75L), 34 tole(0x6a18b6c8L), tole(0x6ed9bb7fL), tole(0x639aada6L), tole(0x675ba011L), 35 tole(0x791c8014L), tole(0x7ddd8da3L), tole(0x709e9b7aL), tole(0x745f96cdL), 36 tole(0x9821b6e0L), tole(0x9ce0bb57L), tole(0x91a3ad8eL), tole(0x9562a039L), 37 tole(0x8b25803cL), tole(0x8fe48d8bL), tole(0x82a79b52L), tole(0x866696e5L), 38 tole(0xbe29db58L), tole(0xbae8d6efL), tole(0xb7abc036L), tole(0xb36acd81L), 39 tole(0xad2ded84L), tole(0xa9ece033L), tole(0xa4aff6eaL), tole(0xa06efb5dL), 40 tole(0xd4316d90L), tole(0xd0f06027L), tole(0xddb376feL), tole(0xd9727b49L), 41 tole(0xc7355b4cL), tole(0xc3f456fbL), tole(0xceb74022L), tole(0xca764d95L), 42 tole(0xf2390028L), tole(0xf6f80d9fL), tole(0xfbbb1b46L), tole(0xff7a16f1L), 43 tole(0xe13d36f4L), tole(0xe5fc3b43L), tole(0xe8bf2d9aL), tole(0xec7e202dL), 44 tole(0x34826077L), tole(0x30436dc0L), tole(0x3d007b19L), tole(0x39c176aeL), 45 tole(0x278656abL), tole(0x23475b1cL), tole(0x2e044dc5L), tole(0x2ac54072L), 46 tole(0x128a0dcfL), tole(0x164b0078L), tole(0x1b0816a1L), tole(0x1fc91b16L), 47 tole(0x018e3b13L), tole(0x054f36a4L), tole(0x080c207dL), tole(0x0ccd2dcaL), 48 tole(0x7892bb07L), tole(0x7c53b6b0L), tole(0x7110a069L), tole(0x75d1addeL), 49 tole(0x6b968ddbL), tole(0x6f57806cL), tole(0x621496b5L), tole(0x66d59b02L), 50 tole(0x5e9ad6bfL), tole(0x5a5bdb08L), tole(0x5718cdd1L), tole(0x53d9c066L), 51 tole(0x4d9ee063L), tole(0x495fedd4L), tole(0x441cfb0dL), tole(0x40ddf6baL), 52 tole(0xaca3d697L), tole(0xa862db20L), tole(0xa521cdf9L), tole(0xa1e0c04eL), 53 tole(0xbfa7e04bL), tole(0xbb66edfcL), tole(0xb625fb25L), tole(0xb2e4f692L), 54 tole(0x8aabbb2fL), tole(0x8e6ab698L), tole(0x8329a041L), tole(0x87e8adf6L), 55 tole(0x99af8df3L), tole(0x9d6e8044L), tole(0x902d969dL), tole(0x94ec9b2aL), 56 tole(0xe0b30de7L), tole(0xe4720050L), tole(0xe9311689L), tole(0xedf01b3eL), 57 tole(0xf3b73b3bL), tole(0xf776368cL), tole(0xfa352055L), tole(0xfef42de2L), 58 tole(0xc6bb605fL), tole(0xc27a6de8L), tole(0xcf397b31L), tole(0xcbf87686L), 59 tole(0xd5bf5683L), tole(0xd17e5b34L), tole(0xdc3d4dedL), tole(0xd8fc405aL), 60 tole(0x6904c0eeL), tole(0x6dc5cd59L), tole(0x6086db80L), tole(0x6447d637L), 61 tole(0x7a00f632L), tole(0x7ec1fb85L), tole(0x7382ed5cL), tole(0x7743e0ebL), 62 tole(0x4f0cad56L), tole(0x4bcda0e1L), tole(0x468eb638L), tole(0x424fbb8fL), 63 tole(0x5c089b8aL), tole(0x58c9963dL), tole(0x558a80e4L), tole(0x514b8d53L), 64 tole(0x25141b9eL), tole(0x21d51629L), tole(0x2c9600f0L), tole(0x28570d47L), 65 tole(0x36102d42L), tole(0x32d120f5L), tole(0x3f92362cL), tole(0x3b533b9bL), 66 tole(0x031c7626L), tole(0x07dd7b91L), tole(0x0a9e6d48L), tole(0x0e5f60ffL), 67 tole(0x101840faL), tole(0x14d94d4dL), tole(0x199a5b94L), tole(0x1d5b5623L), 68 tole(0xf125760eL), tole(0xf5e47bb9L), tole(0xf8a76d60L), tole(0xfc6660d7L), 69 tole(0xe22140d2L), tole(0xe6e04d65L), tole(0xeba35bbcL), tole(0xef62560bL), 70 tole(0xd72d1bb6L), tole(0xd3ec1601L), tole(0xdeaf00d8L), tole(0xda6e0d6fL), 71 tole(0xc4292d6aL), tole(0xc0e820ddL), tole(0xcdab3604L), tole(0xc96a3bb3L), 72 tole(0xbd35ad7eL), tole(0xb9f4a0c9L), tole(0xb4b7b610L), tole(0xb076bba7L), 73 tole(0xae319ba2L), tole(0xaaf09615L), tole(0xa7b380ccL), tole(0xa3728d7bL), 74 tole(0x9b3dc0c6L), tole(0x9ffccd71L), tole(0x92bfdba8L), tole(0x967ed61fL), 75 tole(0x8839f61aL), tole(0x8cf8fbadL), tole(0x81bbed74L), tole(0x857ae0c3L), 76 tole(0x5d86a099L), tole(0x5947ad2eL), tole(0x5404bbf7L), tole(0x50c5b640L), 77 tole(0x4e829645L), tole(0x4a439bf2L), tole(0x47008d2bL), tole(0x43c1809cL), 78 tole(0x7b8ecd21L), tole(0x7f4fc096L), tole(0x720cd64fL), tole(0x76cddbf8L), 79 tole(0x688afbfdL), tole(0x6c4bf64aL), tole(0x6108e093L), tole(0x65c9ed24L), 80 tole(0x11967be9L), tole(0x1557765eL), tole(0x18146087L), tole(0x1cd56d30L), 81 tole(0x02924d35L), tole(0x06534082L), tole(0x0b10565bL), tole(0x0fd15becL), 82 tole(0x379e1651L), tole(0x335f1be6L), tole(0x3e1c0d3fL), tole(0x3add0088L), 83 tole(0x249a208dL), tole(0x205b2d3aL), tole(0x2d183be3L), tole(0x29d93654L), 84 tole(0xc5a71679L), tole(0xc1661bceL), tole(0xcc250d17L), tole(0xc8e400a0L), 85 tole(0xd6a320a5L), tole(0xd2622d12L), tole(0xdf213bcbL), tole(0xdbe0367cL), 86 tole(0xe3af7bc1L), tole(0xe76e7676L), tole(0xea2d60afL), tole(0xeeec6d18L), 87 tole(0xf0ab4d1dL), tole(0xf46a40aaL), tole(0xf9295673L), tole(0xfde85bc4L), 88 tole(0x89b7cd09L), tole(0x8d76c0beL), tole(0x8035d667L), tole(0x84f4dbd0L), 89 tole(0x9ab3fbd5L), tole(0x9e72f662L), tole(0x9331e0bbL), tole(0x97f0ed0cL), 90 tole(0xafbfa0b1L), tole(0xab7ead06L), tole(0xa63dbbdfL), tole(0xa2fcb668L), 91 tole(0xbcbb966dL), tole(0xb87a9bdaL), tole(0xb5398d03L), tole(0xb1f880b4L) 92 }; 93 94 #define DO_CRC(x) crc = tab[((crc >> 24) ^ (x)) & 255] ^ (crc << 8) 95 96 static uint32_t crc32_rk(uint32_t crc, const unsigned char *s, uint32_t len) 97 { 98 const uint32_t *tab; 99 100 tab = crc_table; 101 crc = cpu_to_le32(crc); 102 103 do { 104 DO_CRC(*s++); 105 } while (--len); 106 107 return le32_to_cpu(crc); 108 } 109 110 #undef DO_CRC 111 112 static u32 crc32_verify(unsigned char *data, u32 size) 113 { 114 u32 crc_check = 0, crc_calc = 0; 115 int i = 0; 116 117 if (size <= 4) 118 return 0; 119 120 size -= 4; 121 for (i = 3; i >= 0; i--) 122 crc_check = (crc_check << 8) + (*(data + size + i)); 123 124 crc_calc = crc32_rk(0, data, size); 125 126 return (crc_calc == crc_check) ? crc_check : 0; 127 } 128 #endif 129 130 #if !defined(CONFIG_ARM64) 131 #ifdef CONFIG_LMB 132 static void boot_start_lmb(bootm_headers_t *images) 133 { 134 lmb_init(&images->lmb); 135 #ifdef CONFIG_NR_DRAM_BANKS 136 int i; 137 138 for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) { 139 lmb_add(&images->lmb, gd->bd->bi_dram[i].start, 140 gd->bd->bi_dram[i].size); 141 } 142 #else 143 ulong mem_start; 144 phys_size_t mem_size; 145 146 mem_start = env_get_bootm_low(); 147 mem_size = env_get_bootm_size(); 148 lmb_add(&images->lmb, (phys_addr_t)mem_start, mem_size); 149 #endif 150 arch_lmb_reserve(&images->lmb); 151 board_lmb_reserve(&images->lmb); 152 } 153 #else 154 static inline void boot_start_lmb(bootm_headers_t *images) { } 155 #endif 156 157 static void boot_lmb_init(bootm_headers_t *images) 158 { 159 boot_start_lmb(images); 160 images->state = BOOTM_STATE_OS_GO; 161 } 162 #endif 163 164 /* 165 * non-OTA packaged kernel.img & boot.img return the image size on success, 166 * and a negative value on error. 167 */ 168 static int read_rockchip_image(struct blk_desc *dev_desc, 169 disk_partition_t *part, void *dst) 170 { 171 struct rockchip_image *img; 172 int header_len = 8; 173 int cnt, ret; 174 #ifdef CONFIG_ROCKCHIP_CRC 175 u32 crc32; 176 #endif 177 178 img = memalign(ARCH_DMA_MINALIGN, RK_BLK_SIZE); 179 if (!img) 180 return -ENOMEM; 181 182 /* read first block with header imformation */ 183 ret = blk_dread(dev_desc, part->start, 1, img); 184 if (ret != 1) { 185 ret = -EIO; 186 goto err; 187 } 188 189 if (img->tag != TAG_KERNEL) { 190 printf("Invalid %s image tag(0x%x)\n", part->name, img->tag); 191 ret = -EINVAL; 192 goto err; 193 } 194 195 /* 196 * read the rest blks 197 * total size = image size + 8 bytes header + 4 bytes crc32 198 */ 199 cnt = DIV_ROUND_UP(img->size + 8 + 4, RK_BLK_SIZE); 200 if (!sysmem_alloc_base_by_name((const char *)part->name, 201 (phys_addr_t)dst, 202 cnt * dev_desc->blksz)) { 203 ret = -ENXIO; 204 goto err; 205 } 206 207 memcpy(dst, img->image, RK_BLK_SIZE - header_len); 208 ret = blk_dread(dev_desc, part->start + 1, cnt - 1, 209 dst + RK_BLK_SIZE - header_len); 210 if (ret != (cnt - 1)) { 211 printf("Failed to read %s part, ret=%d\n", part->name, ret); 212 ret = -EIO; 213 } else { 214 ret = img->size; 215 } 216 217 #ifdef CONFIG_ROCKCHIP_CRC 218 printf("%s image rk crc32 verify... ", part->name); 219 crc32 = crc32_verify((uchar *)(ulong)dst, img->size + 4); 220 if (!crc32) { 221 printf("fail!\n"); 222 ret = -EINVAL; 223 } else { 224 printf("okay.\n"); 225 } 226 #endif 227 228 err: 229 free(img); 230 return ret; 231 } 232 233 static int boot_rockchip_image(struct blk_desc *dev_desc, 234 disk_partition_t *boot_part) 235 { 236 disk_partition_t kernel_part; 237 ulong ramdisk_addr_r; 238 ulong kernel_addr_r; 239 ulong fdt_addr_r; 240 int ramdisk_size; 241 int kernel_size; 242 int fdt_size; 243 int ret; 244 245 printf("\n## Booting Rockchip Format Image\n"); 246 247 ramdisk_addr_r = env_get_ulong("ramdisk_addr_r", 16, 0); 248 kernel_addr_r = env_get_ulong("kernel_addr_r", 16, 0); 249 fdt_addr_r = env_get_ulong("fdt_addr_r", 16, 0); 250 251 ret = part_get_info_by_name(dev_desc, PART_KERNEL, &kernel_part); 252 if (ret < 0) { 253 printf("Could not find kernel partition, ret=%d\n", ret); 254 return -EINVAL; 255 } 256 257 kernel_size = read_rockchip_image(dev_desc, &kernel_part, 258 (void *)kernel_addr_r); 259 if (kernel_size < 0) { 260 printf("Failed to read kernel image, ret=%d\n", ret); 261 return -EINVAL; 262 } 263 264 ramdisk_size = read_rockchip_image(dev_desc, boot_part, 265 (void *)ramdisk_addr_r); 266 if (ramdisk_size < 0) 267 ramdisk_size = 0; 268 269 if (gd->fdt_blob != (void *)fdt_addr_r) { 270 fdt_size = rockchip_read_dtb_file((void *)fdt_addr_r); 271 if (fdt_size < 0) { 272 printf("Failed to read fdt, ret=%d\n", fdt_size); 273 return -EINVAL; 274 } 275 } 276 277 env_set("bootm-no-reloc", "y"); 278 279 printf("fdt @ 0x%08lx (0x%08x)\n", fdt_addr_r, fdt_totalsize(fdt_addr_r)); 280 printf("kernel @ 0x%08lx (0x%08x)\n", kernel_addr_r, kernel_size); 281 printf("ramdisk @ 0x%08lx (0x%08x)\n", ramdisk_addr_r, ramdisk_size); 282 283 #if defined(CONFIG_ARM64) 284 char cmdbuf[64]; 285 286 snprintf(cmdbuf, 64, "booti 0x%lx 0x%lx:0x%x 0x%lx", 287 kernel_addr_r, ramdisk_addr_r, ramdisk_size, fdt_addr_r); 288 run_command(cmdbuf, 0); 289 #else 290 /* We asume it's always zImage on 32-bit platform */ 291 ulong kaddr_c = env_get_ulong("kernel_addr_c", 16, 0); 292 ulong kaddr_r, kaddr, ksize; 293 294 if (kernel_addr_r && !kaddr_c) { 295 kaddr_c = kernel_addr_r; 296 kaddr_r = CONFIG_SYS_SDRAM_BASE; 297 } 298 299 if (!sysmem_free((phys_addr_t)kaddr_c)) { 300 kaddr = kaddr_r; 301 ksize = kernel_size * 100 / 45 ; /* Ratio: 45% */ 302 ksize = ALIGN(ksize, dev_desc->blksz); 303 if (!sysmem_alloc_base(MEM_UNCOMP_KERNEL, 304 (phys_addr_t)kaddr, ksize)) 305 return -ENOMEM; 306 } 307 308 boot_lmb_init(&images); 309 images.ep = kernel_addr_r; 310 images.initrd_start = ramdisk_addr_r; 311 images.initrd_end = ramdisk_addr_r + ramdisk_size; 312 images.ft_addr = (void *)fdt_addr_r; 313 images.ft_len = fdt_totalsize(fdt_addr_r); 314 do_bootm_linux(0, 0, NULL, &images); 315 #endif 316 317 return 0; 318 } 319 320 static int do_boot_rockchip(cmd_tbl_t *cmdtp, int flag, 321 int argc, char *const argv[]) 322 { 323 char *part_name = PART_BOOT; 324 struct blk_desc *dev_desc; 325 disk_partition_t part; 326 int ret; 327 328 dev_desc = rockchip_get_bootdev(); 329 if (!dev_desc) { 330 printf("dev_desc is NULL!\n"); 331 return CMD_RET_FAILURE; 332 } 333 334 #ifdef CONFIG_ANDROID_KEYMASTER_CA 335 /* load attestation key from misc partition. */ 336 ret = part_get_info_by_name(dev_desc, PART_MISC, &part); 337 if (ret < 0) 338 printf("Could not find misc partition\n"); 339 else 340 load_attestation_key(dev_desc, &part); 341 #endif 342 343 #ifdef CONFIG_FASTBOOT_OEM_UNLOCK 344 /* read oem unlock status and attach to bootargs */ 345 char oem_unlock[30] = {0}; 346 TEEC_Result result; 347 uint8_t unlock = 0; 348 349 result = trusty_read_oem_unlock(&unlock); 350 if (result) { 351 printf("Failed to read oem unlock status, ret=%d\n", result); 352 } else { 353 snprintf(oem_unlock, sizeof(oem_unlock), 354 "androidboot.oem_unlocked=%d", unlock); 355 env_update("bootargs", oem_unlock); 356 } 357 #endif 358 if (rockchip_get_boot_mode() == BOOT_MODE_RECOVERY) 359 part_name = PART_RECOVERY; 360 361 ret = part_get_info_by_name(dev_desc, part_name, &part); 362 if (ret < 0) { 363 printf("Could not find %s part\n", part.name); 364 return CMD_RET_FAILURE; 365 } 366 367 return boot_rockchip_image(dev_desc, &part) ? CMD_RET_FAILURE : 0; 368 } 369 370 U_BOOT_CMD( 371 bootrkp, 1, 1, do_boot_rockchip, 372 "Boot Linux Image from rockchip image type", 373 "kernel.img: zImage/Image\n" 374 "boot.img: ramdisk\n" 375 "resource.img: dtb, u-boot logo, kernel logo" 376 ); 377