xref: /rk3399_rockchip-uboot/common/image-android.c (revision e1b9a8426d639e558da1cc707c505dfbe8e57007)
1 /*
2  * Copyright (c) 2011 Sebastian Andrzej Siewior <bigeasy@linutronix.de>
3  *
4  * SPDX-License-Identifier:	GPL-2.0+
5  */
6 
7 #include <common.h>
8 #include <image.h>
9 #include <android_image.h>
10 #include <android_bootloader.h>
11 #include <malloc.h>
12 #include <mapmem.h>
13 #include <errno.h>
14 #include <boot_rkimg.h>
15 #include <crypto.h>
16 #include <sysmem.h>
17 #include <u-boot/sha1.h>
18 #ifdef CONFIG_RKIMG_BOOTLOADER
19 #include <asm/arch/resource_img.h>
20 #endif
21 #ifdef CONFIG_RK_AVB_LIBAVB_USER
22 #include <android_avb/avb_slot_verify.h>
23 #include <android_avb/avb_ops_user.h>
24 #include <android_avb/rk_avb_ops_user.h>
25 #endif
26 #include <optee_include/OpteeClientInterface.h>
27 
28 DECLARE_GLOBAL_DATA_PTR;
29 
30 #define ANDROID_IMAGE_DEFAULT_KERNEL_ADDR	0x10008000
31 #define ANDROID_ARG_FDT_FILENAME "rk-kernel.dtb"
32 #define ANDROID_Q_VER				10
33 
34 /* Defined by rockchip legacy mkboot tool(SDK version < 8.1) */
35 #define ANDROID_ROCKCHIP_LEGACY_PAGE_SIZE	0x4000
36 
37 static char andr_tmp_str[ANDR_BOOT_ARGS_SIZE + 1];
38 static u32 android_kernel_comp_type = IH_COMP_NONE;
39 
40 u32 android_image_major_version(void)
41 {
42 	/* MSB 7-bits */
43 	return gd->bd->bi_andr_version >> 25;
44 }
45 
46 u32 android_bcb_msg_sector_offset(void)
47 {
48 	/*
49 	 * Rockchip platforms defines BCB message at the 16KB offset of
50 	 * misc partition while the Google defines it at 0x0 offset.
51 	 *
52 	 * From Android-Q, the 0x0 offset is mandary on Google VTS, so that
53 	 * this is a compatibility according to android image 'os_version'.
54 	 */
55 #ifdef CONFIG_RKIMG_BOOTLOADER
56 	return (android_image_major_version() >= ANDROID_Q_VER) ? 0x0 : 0x20;
57 #else
58 	return 0x0;
59 #endif
60 }
61 
62 static ulong android_image_get_kernel_addr(const struct andr_img_hdr *hdr)
63 {
64 	/*
65 	 * All the Android tools that generate a boot.img use this
66 	 * address as the default.
67 	 *
68 	 * Even though it doesn't really make a lot of sense, and it
69 	 * might be valid on some platforms, we treat that address as
70 	 * the default value for this field, and try to execute the
71 	 * kernel in place in such a case.
72 	 *
73 	 * Otherwise, we will return the actual value set by the user.
74 	 */
75 	if (hdr->kernel_addr == ANDROID_IMAGE_DEFAULT_KERNEL_ADDR)
76 		return (ulong)hdr + hdr->page_size;
77 
78 #ifdef CONFIG_ARCH_ROCKCHIP
79 	/*
80 	 * If kernel is compressed, kernel_addr is set as decompressed address
81 	 * after compressed being loaded to ram, so let's use it.
82 	 */
83 	if (android_kernel_comp_type != IH_COMP_NONE &&
84 	    android_kernel_comp_type != IH_COMP_ZIMAGE)
85 		return hdr->kernel_addr;
86 
87 	/*
88 	 * Compatble with rockchip legacy packing with kernel/ramdisk/second
89 	 * address base from 0x60000000(SDK versiont < 8.1), these are invalid
90 	 * address, so we calc it by real size.
91 	 */
92 	return (ulong)hdr + hdr->page_size;
93 #else
94 	return hdr->kernel_addr;
95 #endif
96 
97 }
98 
99 void android_image_set_comp(struct andr_img_hdr *hdr, u32 comp)
100 {
101 	android_kernel_comp_type = comp;
102 }
103 
104 u32 android_image_get_comp(const struct andr_img_hdr *hdr)
105 {
106 	return android_kernel_comp_type;
107 }
108 
109 int android_image_parse_kernel_comp(const struct andr_img_hdr *hdr)
110 {
111 	ulong kaddr = android_image_get_kernel_addr(hdr);
112 	return bootm_parse_comp((const unsigned char *)kaddr);
113 }
114 
115 /**
116  * android_image_get_kernel() - processes kernel part of Android boot images
117  * @hdr:	Pointer to image header, which is at the start
118  *			of the image.
119  * @verify:	Checksum verification flag. Currently unimplemented.
120  * @os_data:	Pointer to a ulong variable, will hold os data start
121  *			address.
122  * @os_len:	Pointer to a ulong variable, will hold os data length.
123  *
124  * This function returns the os image's start address and length. Also,
125  * it appends the kernel command line to the bootargs env variable.
126  *
127  * Return: Zero, os start address and length on success,
128  *		otherwise on failure.
129  */
130 int android_image_get_kernel(const struct andr_img_hdr *hdr, int verify,
131 			     ulong *os_data, ulong *os_len)
132 {
133 	u32 kernel_addr = android_image_get_kernel_addr(hdr);
134 
135 	/*
136 	 * Not all Android tools use the id field for signing the image with
137 	 * sha1 (or anything) so we don't check it. It is not obvious that the
138 	 * string is null terminated so we take care of this.
139 	 */
140 	strncpy(andr_tmp_str, hdr->name, ANDR_BOOT_NAME_SIZE);
141 	andr_tmp_str[ANDR_BOOT_NAME_SIZE] = '\0';
142 	if (strlen(andr_tmp_str))
143 		printf("Android's image name: %s\n", andr_tmp_str);
144 
145 	printf("Kernel load addr 0x%08x size %u KiB\n",
146 	       kernel_addr, DIV_ROUND_UP(hdr->kernel_size, 1024));
147 
148 	int len = 0;
149 	if (*hdr->cmdline) {
150 		debug("Kernel command line: %s\n", hdr->cmdline);
151 		len += strlen(hdr->cmdline);
152 	}
153 
154 	char *bootargs = env_get("bootargs");
155 	if (bootargs)
156 		len += strlen(bootargs);
157 
158 	char *newbootargs = malloc(len + 2);
159 	if (!newbootargs) {
160 		puts("Error: malloc in android_image_get_kernel failed!\n");
161 		return -ENOMEM;
162 	}
163 	*newbootargs = '\0';
164 
165 	if (bootargs) {
166 		strcpy(newbootargs, bootargs);
167 		strcat(newbootargs, " ");
168 	}
169 	if (*hdr->cmdline)
170 		strcat(newbootargs, hdr->cmdline);
171 
172 	env_set("bootargs", newbootargs);
173 
174 	if (os_data) {
175 		*os_data = (ulong)hdr;
176 		*os_data += hdr->page_size;
177 	}
178 	if (os_len)
179 		*os_len = hdr->kernel_size;
180 	return 0;
181 }
182 
183 int android_image_check_header(const struct andr_img_hdr *hdr)
184 {
185 	return memcmp(ANDR_BOOT_MAGIC, hdr->magic, ANDR_BOOT_MAGIC_SIZE);
186 }
187 
188 ulong android_image_get_end(const struct andr_img_hdr *hdr)
189 {
190 	ulong end;
191 	/*
192 	 * The header takes a full page, the remaining components are aligned
193 	 * on page boundary
194 	 */
195 	end = (ulong)hdr;
196 	end += hdr->page_size;
197 	end += ALIGN(hdr->kernel_size, hdr->page_size);
198 	end += ALIGN(hdr->ramdisk_size, hdr->page_size);
199 	end += ALIGN(hdr->second_size, hdr->page_size);
200 
201 	if (hdr->header_version >= 1)
202 		end += ALIGN(hdr->recovery_dtbo_size, hdr->page_size);
203 
204 	return end;
205 }
206 
207 u32 android_image_get_ksize(const struct andr_img_hdr *hdr)
208 {
209 	return hdr->kernel_size;
210 }
211 
212 void android_image_set_kload(struct andr_img_hdr *hdr, u32 load_address)
213 {
214 	hdr->kernel_addr = load_address;
215 }
216 
217 ulong android_image_get_kload(const struct andr_img_hdr *hdr)
218 {
219 	return android_image_get_kernel_addr(hdr);
220 }
221 
222 int android_image_get_ramdisk(const struct andr_img_hdr *hdr,
223 			      ulong *rd_data, ulong *rd_len)
224 {
225 	ulong ramdisk_addr_r;
226 
227 	if (!hdr->ramdisk_size) {
228 		*rd_data = *rd_len = 0;
229 		return -1;
230 	}
231 
232 	/* Have been loaded by android_image_load_separate() on ramdisk_addr_r */
233 	ramdisk_addr_r = env_get_ulong("ramdisk_addr_r", 16, 0);
234 	if (!ramdisk_addr_r) {
235 		printf("No Found Ramdisk Load Address.\n");
236 		return -1;
237 	}
238 
239 	*rd_data = ramdisk_addr_r;
240 	*rd_len = hdr->ramdisk_size;
241 
242 	printf("RAM disk load addr 0x%08lx size %u KiB\n",
243 	       *rd_data, DIV_ROUND_UP(hdr->ramdisk_size, 1024));
244 
245 	return 0;
246 }
247 
248 int android_image_get_fdt(const struct andr_img_hdr *hdr,
249 			      ulong *rd_data)
250 {
251 	ulong fdt_addr_r;
252 
253 	if (!hdr->second_size) {
254 		*rd_data = 0;
255 		return -1;
256 	}
257 
258 	/* Have been loaded by android_image_load_separate() on fdt_addr_r */
259 	fdt_addr_r = env_get_ulong("fdt_addr_r", 16, 0);
260 	if (!fdt_addr_r) {
261 		printf("No Found FDT Load Address.\n");
262 		return -1;
263 	}
264 
265 	*rd_data = fdt_addr_r;
266 
267 	debug("FDT load addr 0x%08x size %u KiB\n",
268 	      hdr->second_addr, DIV_ROUND_UP(hdr->second_size, 1024));
269 
270 	return 0;
271 }
272 
273 #ifdef CONFIG_ANDROID_BOOT_IMAGE_HASH
274 static void print_hash(const char *label, u8 *hash, int len)
275 {
276 	int i;
277 
278 	printf("%s:\n    0x", label ? : "Hash");
279 	for (i = 0; i < len; i++)
280 		printf("%02x", hash[i]);
281 	printf("\n");
282 }
283 
284 /*
285  * This is only for Non-AVB image, because AVB image is verified by AVB bootflow.
286  * The kernel/ramdisk/second address should be the real address in hdr before
287  * calling this function.
288  *
289  * mkbootimg tool always use SHA1 for images.
290  */
291 static int android_image_hash_verify(struct andr_img_hdr *hdr)
292 {
293 	u8 hash[20];
294 
295 #ifdef DEBUG
296 	android_print_contents(hdr);
297 #endif
298 
299 	if (hdr->kernel_addr == ANDROID_IMAGE_DEFAULT_KERNEL_ADDR) {
300 		printf("No real image address in android hdr\n");
301 		return -EINVAL;
302 	}
303 
304 #ifdef CONFIG_DM_CRYPTO
305 	struct udevice *dev;
306 	sha_context ctx;
307 
308 	dev = crypto_get_device(CRYPTO_SHA1);
309 	if (!dev) {
310 		printf("Can't find crypto device for SHA1 capability\n");
311 		return -ENODEV;
312 	}
313 
314 	ctx.algo = CRYPTO_SHA1;
315 	ctx.length = hdr->kernel_size + sizeof(hdr->kernel_size) +
316 		     hdr->ramdisk_size + sizeof(hdr->ramdisk_size) +
317 		     hdr->second_size + sizeof(hdr->second_size);
318 #ifdef CONFIG_HASH_ROCKCHIP_LEGACY
319 	ctx.length += sizeof(hdr->tags_addr) + sizeof(hdr->page_size) +
320 		      sizeof(hdr->unused) + sizeof(hdr->name) +
321 		      sizeof(hdr->cmdline);
322 #endif
323 
324 	crypto_sha_init(dev, &ctx);
325 
326 	crypto_sha_update(dev, (u32 *)(ulong)hdr->kernel_addr,
327 			  hdr->kernel_size);
328 	crypto_sha_update(dev, (u32 *)&hdr->kernel_size,
329 			  sizeof(hdr->kernel_size));
330 	crypto_sha_update(dev, (u32 *)(ulong)hdr->ramdisk_addr,
331 			  hdr->ramdisk_size);
332 	crypto_sha_update(dev, (u32 *)&hdr->ramdisk_size,
333 			  sizeof(hdr->ramdisk_size));
334 	crypto_sha_update(dev, (u32 *)(ulong)hdr->second_addr,
335 			  hdr->second_size);
336 	crypto_sha_update(dev, (u32 *)&hdr->second_size,
337 			  sizeof(hdr->second_size));
338 #ifdef CONFIG_HASH_ROCKCHIP_LEGACY
339 	crypto_sha_update(dev, (u32 *)&hdr->tags_addr, sizeof(hdr->tags_addr));
340 	crypto_sha_update(dev, (u32 *)&hdr->page_size, sizeof(hdr->page_size));
341 	crypto_sha_update(dev, (u32 *)&hdr->header_version,
342 			  sizeof(hdr->header_version));
343 	crypto_sha_update(dev, (u32 *)&hdr->os_version, sizeof(hdr->os_version));
344 	crypto_sha_update(dev, (u32 *)&hdr->name, sizeof(hdr->name));
345 	crypto_sha_update(dev, (u32 *)&hdr->cmdline, sizeof(hdr->cmdline));
346 #endif
347 
348 	crypto_sha_final(dev, &ctx, hash);
349 
350 #elif CONFIG_SHA1
351 	sha1_context ctx;
352 
353 	sha1_starts(&ctx);
354 	sha1_update(&ctx, (u8 *)(ulong)hdr->kernel_addr, hdr->kernel_size);
355 	sha1_update(&ctx, (u8 *)&hdr->kernel_size, sizeof(hdr->kernel_size));
356 	sha1_update(&ctx, (u8 *)(ulong)hdr->ramdisk_addr, hdr->ramdisk_size);
357 	sha1_update(&ctx, (u8 *)&hdr->ramdisk_size, sizeof(hdr->ramdisk_size));
358 	sha1_update(&ctx, (u8 *)(ulong)hdr->second_addr, hdr->second_size);
359 	sha1_update(&ctx, (u8 *)&hdr->second_size, sizeof(hdr->second_size));
360 #ifdef CONFIG_HASH_ROCKCHIP_LEGACY
361 	sha1_update(&ctx, (u8 *)&hdr->tags_addr, sizeof(hdr->tags_addr));
362 	sha1_update(&ctx, (u8 *)&hdr->page_size, sizeof(hdr->page_size));
363 	sha1_update(&ctx, (u8 *)&hdr->header_version,
364 		    sizeof(hdr->header_version));
365 	sha1_update(&ctx, (u8 *)&hdr->os_version, sizeof(hdr->os_version));
366 	sha1_update(&ctx, (u8 *)&hdr->name, sizeof(hdr->name));
367 	sha1_update(&ctx, (u8 *)&hdr->cmdline, sizeof(hdr->cmdline));
368 #endif
369 
370 	sha1_finish(&ctx, hash);
371 #endif	/* CONFIG_SHA1 */
372 
373 	if (memcmp(hash, hdr->id, 20)) {
374 		print_hash("SHA1 from image header", (u8 *)hdr->id, 20);
375 		print_hash("SHA1 real", (u8 *)hash, 20);
376 		return -EBADFD;
377 	}
378 
379 	return 0;
380 }
381 #endif
382 
383 static int android_image_separate(struct andr_img_hdr *hdr,
384 				  const disk_partition_t *part,
385 				  void *load_address,
386 				  void *ram_base)
387 {
388 	struct blk_desc *dev_desc = rockchip_get_bootdev();
389 	ulong ramdisk_addr_r = env_get_ulong("ramdisk_addr_r", 16, 0);
390 	char *fdt_high = env_get("fdt_high");
391 	char *ramdisk_high = env_get("initrd_high");
392 	ulong blk_start, blk_cnt, size;
393 	ulong start, second_addr_r = 0;
394 	int ret;
395 
396 	if (android_image_check_header(hdr)) {
397 		printf("Bad android image header\n");
398 		return -EINVAL;
399 	}
400 
401 	if (hdr->kernel_size) {
402 		size = hdr->kernel_size + hdr->page_size;
403 		blk_cnt = DIV_ROUND_UP(size, dev_desc->blksz);
404 		if (!sysmem_alloc_base(MEMBLK_ID_KERNEL,
405 				       (phys_addr_t)load_address,
406 				       blk_cnt * dev_desc->blksz))
407 			return -ENXIO;
408 
409 		if (ram_base) {
410 			start = (ulong)ram_base;
411 			memcpy((char *)load_address, (char *)start, size);
412 		} else {
413 			blk_start = part->start;
414 			ret = blk_dread(dev_desc, blk_start,
415 					blk_cnt, load_address);
416 			if (ret != blk_cnt) {
417 				printf("%s: read kernel failed, ret=%d\n",
418 				      __func__, ret);
419 				return -1;
420 			}
421 		}
422 	}
423 
424 	if (hdr->ramdisk_size) {
425 		size = hdr->page_size + ALIGN(hdr->kernel_size, hdr->page_size);
426 		blk_cnt = DIV_ROUND_UP(hdr->ramdisk_size, dev_desc->blksz);
427 		if (!sysmem_alloc_base(MEMBLK_ID_RAMDISK,
428 				       ramdisk_addr_r,
429 				       blk_cnt * dev_desc->blksz))
430 			return -ENXIO;
431 		if (ram_base) {
432 			start = (unsigned long)ram_base;
433 			start += hdr->page_size;
434 			start += ALIGN(hdr->kernel_size, hdr->page_size);
435 			memcpy((char *)ramdisk_addr_r,
436 			       (char *)start, hdr->ramdisk_size);
437 		} else {
438 			blk_start = part->start +
439 				DIV_ROUND_UP(size, dev_desc->blksz);
440 			ret = blk_dread(dev_desc, blk_start,
441 					blk_cnt, (void *)ramdisk_addr_r);
442 			if (ret != blk_cnt) {
443 				printf("%s: read ramdisk failed, ret=%d\n",
444 				      __func__, ret);
445 				return -1;
446 			}
447 		}
448 	}
449 
450 	/*
451 	 * Load dtb file by rockchip_read_dtb_file() which support pack
452 	 * dtb in second position or resource file.
453 	 */
454 #ifdef CONFIG_RKIMG_BOOTLOADER
455 	ulong fdt_addr_r = env_get_ulong("fdt_addr_r", 16, 0);
456 
457 	if (hdr->second_size && (gd->fdt_blob != (void *)fdt_addr_r)) {
458 		ulong fdt_size;
459 
460 		fdt_size = rockchip_read_dtb_file((void *)fdt_addr_r);
461 		if (fdt_size < 0) {
462 			printf("%s: read fdt failed\n", __func__);
463 			return ret;
464 		}
465 	}
466 #endif
467 
468 #ifdef CONFIG_ANDROID_BOOT_IMAGE_HASH
469 	if (hdr->second_size) {
470 		ulong blk_start, blk_cnt;
471 
472 		/* Just for image data hash calculation */
473 		second_addr_r = (ulong)malloc(hdr->second_size);
474 		if (!second_addr_r)
475 			return -ENOMEM;
476 
477 		size = hdr->page_size +
478 		       ALIGN(hdr->kernel_size, hdr->page_size) +
479 		       ALIGN(hdr->ramdisk_size, hdr->page_size);
480 		blk_cnt = DIV_ROUND_UP(hdr->second_size, dev_desc->blksz);
481 
482 		if (ram_base) {
483 			start = (unsigned long)ram_base;
484 			start += hdr->page_size;
485 			start += ALIGN(hdr->kernel_size, hdr->page_size);
486 			start += ALIGN(hdr->ramdisk_size, hdr->page_size);
487 			memcpy((char *)second_addr_r,
488 			       (char *)start, hdr->second_size);
489 		} else {
490 			blk_start = part->start +
491 					DIV_ROUND_UP(size, dev_desc->blksz);
492 			ret = blk_dread(dev_desc, blk_start, blk_cnt,
493 					(void *)second_addr_r);
494 			if (ret != blk_cnt) {
495 				printf("%s: read second pos failed, ret=%d\n",
496 				       __func__, ret);
497 				return -1;
498 			}
499 		}
500 	}
501 #endif
502 
503 	/*
504 	 * Update hdr with real image address.
505 	 *
506 	 * kernel_addr depends on load_address can handle both the
507 	 * compressed and no-compressed kernel position.
508 	 */
509 	hdr->kernel_addr = (ulong)load_address + hdr->page_size;
510 	hdr->second_addr = second_addr_r;
511 	hdr->ramdisk_addr = ramdisk_addr_r;
512 
513 	/*
514 	 * Since images are loaded separate, fdt/ramdisk relocation
515 	 * can be disabled, it saves boot time.
516 	 */
517 	if (!fdt_high) {
518 		env_set_hex("fdt_high", -1UL);
519 		printf("Fdt ");
520 	}
521 	if (!ramdisk_high) {
522 		env_set_hex("initrd_high", -1UL);
523 		printf("Ramdisk ");
524 	}
525 	if (!fdt_high || !ramdisk_high)
526 		printf("skip relocation\n");
527 
528 	return 0;
529 }
530 
531 /*
532  * 'boot_android' cmd use "kernel_addr_r" as default load address !
533  * We update it according to compress type and "kernel_addr_c/r".
534  */
535 int android_image_parse_comp(struct andr_img_hdr *hdr, ulong *load_addr)
536 {
537 	ulong kernel_addr_c;
538 	int comp;
539 
540 	kernel_addr_c = env_get_ulong("kernel_addr_c", 16, 0);
541 	comp = android_image_parse_kernel_comp(hdr);
542 
543 #ifdef CONFIG_ARM64
544 	/*
545 	 * On 64-bit kernel, assuming use IMAGE by default.
546 	 *
547 	 * kernel_addr_c is for LZ4-IMAGE but maybe not defined.
548 	 * kernel_addr_r is for IMAGE.
549 	 */
550 	if (comp != IH_COMP_NONE) {
551 		ulong comp_addr;
552 
553 		if (kernel_addr_c) {
554 			comp_addr = kernel_addr_c;
555 		} else {
556 			printf("Warn: No \"kernel_addr_c\"\n");
557 			comp_addr = CONFIG_SYS_SDRAM_BASE + 0x2000000;/* 32M */
558 			env_set_ulong("kernel_addr_c", comp_addr);
559 		}
560 
561 		*load_addr = comp_addr - hdr->page_size;
562 	}
563 #else
564 	/*
565 	 * On 32-bit kernel, assuming use zImage by default.
566 	 *
567 	 * kernel_addr_c is for LZ4/zImage but maybe not defined.
568 	 * kernel_addr_r is for zImage when kernel_addr_c is not defined.
569 	 * kernel_addr_r is for IMAGE when kernel_addr_c is defined.
570 	 */
571 	if (comp == IH_COMP_NONE) {
572 		if (kernel_addr_c)
573 			*load_addr = env_get_ulong("kernel_addr_r", 16, 0);
574 		else
575 			*load_addr = CONFIG_SYS_SDRAM_BASE + 0x8000;
576 	} else {
577 		if (kernel_addr_c)
578 			*load_addr = kernel_addr_c - hdr->page_size;
579 	}
580 #endif
581 
582 	env_set_ulong("os_comp", comp);
583 	return comp;
584 }
585 
586 void android_image_set_decomp(struct andr_img_hdr *hdr, int comp)
587 {
588 	ulong kernel_addr_r;
589 
590 	/* zImage handles decompress itself */
591 	if (comp != IH_COMP_NONE && comp != IH_COMP_ZIMAGE) {
592 		kernel_addr_r = env_get_ulong("kernel_addr_r", 16, 0x02080000);
593 		android_image_set_kload(hdr, kernel_addr_r);
594 		android_image_set_comp(hdr, comp);
595 	} else {
596 		android_image_set_comp(hdr, IH_COMP_NONE);
597 	}
598 }
599 
600 static int android_image_load_separate(struct andr_img_hdr *hdr,
601 				       const disk_partition_t *part,
602 				       void *load_addr)
603 {
604 	return android_image_separate(hdr, part, load_addr, NULL);
605 }
606 
607 int android_image_memcpy_separate(struct andr_img_hdr *hdr, ulong *load_addr)
608 {
609 	ulong comp_addr = *load_addr;
610 	int comp;
611 
612 	comp = android_image_parse_comp(hdr, &comp_addr);
613 	if (android_image_separate(hdr, NULL, (void *)comp_addr, hdr))
614 		return -1;
615 
616 	*load_addr = comp_addr;
617 	android_image_set_decomp((void *)comp_addr, comp);
618 
619 	return 0;
620 }
621 
622 long android_image_load(struct blk_desc *dev_desc,
623 			const disk_partition_t *part_info,
624 			unsigned long load_address,
625 			unsigned long max_size) {
626 	struct andr_img_hdr *hdr;
627 	u32 blksz = dev_desc->blksz;
628 	u32 pszcnt, hdrcnt, kercnt;
629 	int comp, ret;
630 
631 	if (max_size < part_info->blksz)
632 		return -1;
633 
634 	/*
635 	 * read Android image header and leave enough space for page_size align
636 	 * and kernel image header(1 block maybe enough).
637 	 *
638 	 * ANDROID_ROCKCHIP_LEGACY_PAGE_SIZE is defined by rockchip legacy
639 	 * mkboot tool(SDK version < 8.1) and larger than Google defined.
640 	 *
641 	 * To compatible this, we malloc enough buffer but only read android
642 	 * header and kernel image(1 block) from storage(ignore page size).
643 	 */
644 	kercnt = 1;
645 	hdrcnt = DIV_ROUND_UP(sizeof(*hdr), blksz);
646 	pszcnt = DIV_ROUND_UP(ANDROID_ROCKCHIP_LEGACY_PAGE_SIZE, blksz);
647 
648 	hdr = memalign(ARCH_DMA_MINALIGN, (hdrcnt + pszcnt + kercnt) * blksz);
649 	if (!hdr) {
650 		printf("No memory\n");
651 		return -1;
652 	}
653 
654 	if (blk_dread(dev_desc, part_info->start, hdrcnt, hdr) != hdrcnt) {
655 		printf("Failed to read image header\n");
656 		goto fail;
657 	}
658 
659 	if (android_image_check_header(hdr) != 0) {
660 		printf("** Invalid Android Image header **\n");
661 		goto fail;
662 	}
663 
664 	/*
665 	 * Update and skip pszcnt(hdr is included) according to hdr->page_size,
666 	 * reading kernel image for compress validation.
667 	 */
668 	pszcnt = DIV_ROUND_UP(hdr->page_size, blksz);
669 	if (blk_dread(dev_desc, part_info->start + pszcnt, kercnt,
670 		      (void *)((ulong)hdr + hdr->page_size)) != kercnt) {
671 		printf("Failed to read kernel header\n");
672 		goto fail;
673 	}
674 
675 	load_address -= hdr->page_size;
676 
677 	/* Let's load kernel now ! */
678 	comp = android_image_parse_comp(hdr, &load_address);
679 	ret = android_image_load_separate(hdr, part_info, (void *)load_address);
680 	if (ret) {
681 		printf("Failed to load android image\n");
682 		goto fail;
683 	}
684 	android_image_set_decomp((void *)load_address, comp);
685 
686 	debug("Loading Android Image to 0x%08lx\n", load_address);
687 
688 	/* Verify image hash */
689 #ifdef CONFIG_ANDROID_BOOT_IMAGE_HASH
690 	if (android_image_hash_verify(hdr)) {
691 		printf("HASH: Image verify failed!\n");
692 		return -EBADFD;
693 	}
694 
695 	printf("HASH: Image verify OK\n");
696 #endif
697 
698 	free(hdr);
699 	return load_address;
700 
701 fail:
702 	free(hdr);
703 	return -1;
704 }
705 
706 #if !defined(CONFIG_SPL_BUILD)
707 /**
708  * android_print_contents - prints out the contents of the Android format image
709  * @hdr: pointer to the Android format image header
710  *
711  * android_print_contents() formats a multi line Android image contents
712  * description.
713  * The routine prints out Android image properties
714  *
715  * returns:
716  *     no returned results
717  */
718 void android_print_contents(const struct andr_img_hdr *hdr)
719 {
720 	const char * const p = IMAGE_INDENT_STRING;
721 	/* os_version = ver << 11 | lvl */
722 	u32 os_ver = hdr->os_version >> 11;
723 	u32 os_lvl = hdr->os_version & ((1U << 11) - 1);
724 	u32 header_version = hdr->header_version;
725 
726 	printf("%skernel size:      %x\n", p, hdr->kernel_size);
727 	printf("%skernel address:   %x\n", p, hdr->kernel_addr);
728 	printf("%sramdisk size:     %x\n", p, hdr->ramdisk_size);
729 	printf("%sramdisk addrress: %x\n", p, hdr->ramdisk_addr);
730 	printf("%ssecond size:      %x\n", p, hdr->second_size);
731 	printf("%ssecond address:   %x\n", p, hdr->second_addr);
732 	printf("%stags address:     %x\n", p, hdr->tags_addr);
733 	printf("%spage size:        %x\n", p, hdr->page_size);
734 	printf("%sheader_version:   %x\n", p, header_version);
735 	/* ver = A << 14 | B << 7 | C         (7 bits for each of A, B, C)
736 	 * lvl = ((Y - 2000) & 127) << 4 | M  (7 bits for Y, 4 bits for M) */
737 	printf("%sos_version:       %x (ver: %u.%u.%u, level: %u.%u)\n",
738 	       p, hdr->os_version,
739 	       (os_ver >> 7) & 0x7F, (os_ver >> 14) & 0x7F, os_ver & 0x7F,
740 	       (os_lvl >> 4) + 2000, os_lvl & 0x0F);
741 	printf("%sname:             %s\n", p, hdr->name);
742 	printf("%scmdline:          %s\n", p, hdr->cmdline);
743 
744 	if (header_version >= 1) {
745 		printf("%srecovery dtbo size:    %x\n", p, hdr->recovery_dtbo_size);
746 		printf("%srecovery dtbo offset:  %llx\n", p, hdr->recovery_dtbo_offset);
747 		printf("%sheader size:           %x\n", p, hdr->header_size);
748 	}
749 
750 	if (header_version >= 2) {
751 		printf("%sdtb size:              %x\n", p, hdr->dtb_size);
752 		printf("%sdtb addr:              %llx\n", p, hdr->dtb_addr);
753 	}
754 }
755 #endif
756