diff mbox series

[v3,10/19] android: boot: update android_image_get_data to support v3, v4

Message ID 20230205235021.355410-11-souajih@baylibre.com
State Accepted
Commit 1115027d2f75ade805f0d4d44a6a2c21e74ea6b2
Delegated to: Tom Rini
Headers show
Series Support android boot image v3/v4 | expand

Commit Message

Safae Ouajih Feb. 5, 2023, 11:50 p.m. UTC
Since boot image header version 3 and 4 introduced vendor boot image,
use the following functions to fill the generic android
structure : andr_image_data:

 - android_boot_image_v3_v4_parse_hdr()
 - android_vendor_boot_image_v3_v4_parse_hdr()

Update android_image_get_data() to support v3 and v4

Signed-off-by: Safae Ouajih <souajih@baylibre.com>
Reviewed-by: Simon Glass <sjg@chromium.org>
---
 boot/image-android.c    | 80 +++++++++++++++++++++++++++++++++++++++--
 include/android_image.h |  3 ++
 include/image.h         | 11 ++++++
 3 files changed, 91 insertions(+), 3 deletions(-)

Comments

Mattijs Korpershoek Feb. 9, 2023, 2:32 p.m. UTC | #1
On Mon, Feb 06, 2023 at 00:50, Safae Ouajih <souajih@baylibre.com> wrote:

> Since boot image header version 3 and 4 introduced vendor boot image,
> use the following functions to fill the generic android
> structure : andr_image_data:
>
>  - android_boot_image_v3_v4_parse_hdr()
>  - android_vendor_boot_image_v3_v4_parse_hdr()
>
> Update android_image_get_data() to support v3 and v4
>
> Signed-off-by: Safae Ouajih <souajih@baylibre.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>

Reviewed-by: Mattijs Korpershoek <mkorpershoek@baylibre.com>

> ---
>  boot/image-android.c    | 80 +++++++++++++++++++++++++++++++++++++++--
>  include/android_image.h |  3 ++
>  include/image.h         | 11 ++++++
>  3 files changed, 91 insertions(+), 3 deletions(-)
>
> diff --git a/boot/image-android.c b/boot/image-android.c
> index f16eebff49..712d437766 100644
> --- a/boot/image-android.c
> +++ b/boot/image-android.c
> @@ -18,6 +18,65 @@
>  
>  static char andr_tmp_str[ANDR_BOOT_ARGS_SIZE + 1];
>  
> +static void android_boot_image_v3_v4_parse_hdr(const struct andr_boot_img_hdr_v3 *hdr,
> +					       struct andr_image_data *data)
> +{
> +	ulong end;
> +
> +	data->kcmdline = hdr->cmdline;
> +	data->header_version = hdr->header_version;
> +
> +	/*
> +	 * The header takes a full page, the remaining components are aligned
> +	 * on page boundary.
> +	 */
> +	end = (ulong)hdr;
> +	end += ANDR_GKI_PAGE_SIZE;
> +	data->kernel_ptr = end;
> +	data->kernel_size = hdr->kernel_size;
> +	end += ALIGN(hdr->kernel_size, ANDR_GKI_PAGE_SIZE);
> +	data->ramdisk_size = hdr->ramdisk_size;
> +	data->boot_ramdisk_size = hdr->ramdisk_size;
> +	end += ALIGN(hdr->ramdisk_size, ANDR_GKI_PAGE_SIZE);
> +
> +	if (hdr->header_version > 3)
> +		end += ALIGN(hdr->signature_size, ANDR_GKI_PAGE_SIZE);
> +
> +	data->boot_img_total_size = end - (ulong)hdr;
> +}
> +
> +static void android_vendor_boot_image_v3_v4_parse_hdr(const struct andr_vnd_boot_img_hdr
> +						      *hdr, struct andr_image_data *data)
> +{
> +	ulong end;
> +
> +	/*
> +	 * The header takes a full page, the remaining components are aligned
> +	 * on page boundary.
> +	 */
> +	data->tags_addr = hdr->tags_addr;
> +	data->image_name = hdr->name;
> +	data->kernel_addr = hdr->kernel_addr;
> +	data->ramdisk_addr = hdr->ramdisk_addr;
> +	data->dtb_load_addr = hdr->dtb_addr;
> +	end = (ulong)hdr;
> +	end += hdr->page_size;
> +	if (hdr->vendor_ramdisk_size) {
> +		data->vendor_ramdisk_ptr = end;
> +		data->vendor_ramdisk_size = hdr->vendor_ramdisk_size;
> +		data->ramdisk_size += hdr->vendor_ramdisk_size;
> +		end += ALIGN(hdr->vendor_ramdisk_size, hdr->page_size);
> +	}
> +
> +	data->dtb_ptr = end;
> +	data->dtb_size = hdr->dtb_size;
> +
> +	end += ALIGN(hdr->dtb_size, hdr->page_size);
> +	end += ALIGN(hdr->vendor_ramdisk_table_size, hdr->page_size);
> +	end += ALIGN(hdr->bootconfig_size, hdr->page_size);
> +	data->vendor_boot_img_total_size = end - (ulong)hdr;
> +}
> +
>  static void android_boot_image_v0_v1_v2_parse_hdr(const struct andr_boot_img_hdr_v0 *hdr,
>  						  struct andr_image_data *data)
>  {
> @@ -79,10 +138,20 @@ bool android_image_get_data(const void *boot_hdr, const void *vendor_boot_hdr,
>  		return false;
>  	}
>  
> -	if (((struct andr_boot_img_hdr_v0 *)boot_hdr)->header_version > 2)
> -		printf("Only boot image header version 2 and below are supported\n");
> -	else
> +	if (((struct andr_boot_img_hdr_v0 *)boot_hdr)->header_version > 2) {
> +		if (!vendor_boot_hdr) {
> +			printf("For boot header v3+ vendor boot image has to be provided\n");
> +			return false;
> +		}
> +		if (!is_android_vendor_boot_image_header(vendor_boot_hdr)) {
> +			printf("Incorrect vendor boot image header\n");
> +			return false;
> +		}
> +		android_boot_image_v3_v4_parse_hdr(boot_hdr, data);
> +		android_vendor_boot_image_v3_v4_parse_hdr(vendor_boot_hdr, data);
> +	} else {
>  		android_boot_image_v0_v1_v2_parse_hdr(boot_hdr, data);
> +	}
>  
>  	return true;
>  }
> @@ -200,6 +269,11 @@ int android_image_get_kernel(const struct andr_boot_img_hdr_v0 *hdr,
>  	return 0;
>  }
>  
> +bool is_android_vendor_boot_image_header(const void *vendor_boot_img)
> +{
> +	return !memcmp(VENDOR_BOOT_MAGIC, vendor_boot_img, ANDR_VENDOR_BOOT_MAGIC_SIZE);
> +}
> +
>  bool is_android_boot_image_header(const struct andr_boot_img_hdr_v0 *hdr)
>  {
>  	return !memcmp(ANDR_BOOT_MAGIC, hdr, ANDR_BOOT_MAGIC_SIZE);
> diff --git a/include/android_image.h b/include/android_image.h
> index fb3a9d9858..99e7803508 100644
> --- a/include/android_image.h
> +++ b/include/android_image.h
> @@ -323,6 +323,8 @@ struct andr_image_data {
>  	ulong kernel_ptr;  /* kernel address */
>  	u32 kernel_size;  /* size in bytes */
>  	u32 ramdisk_size;  /* size in bytes */
> +	ulong vendor_ramdisk_ptr;  /* vendor ramdisk address */
> +	u32 vendor_ramdisk_size;  /* vendor ramdisk size*/
>  	u32 boot_ramdisk_size;  /* size in bytes */
>  	ulong second_ptr;  /* secondary bootloader address */
>  	u32 second_size;  /* secondary bootloader size */
> @@ -342,6 +344,7 @@ struct andr_image_data {
>  	ulong tags_addr;  /* physical addr for kernel tags */
>  	u32 header_version;  /* version of the boot image header */
>  	u32 boot_img_total_size;  /* boot image size */
> +	u32 vendor_boot_img_total_size;  /* vendor boot image size */
>  };
>  
>  #endif
> diff --git a/include/image.h b/include/image.h
> index f84c03f08f..c2e751c136 100644
> --- a/include/image.h
> +++ b/include/image.h
> @@ -1876,6 +1876,17 @@ bool android_image_print_dtb_contents(ulong hdr_addr);
>   */
>  bool is_android_boot_image_header(const struct andr_boot_img_hdr_v0 *hdr);
>  
> +/**
> + * is_android_vendor_boot_image_header() - Check the magic of vendor boot image
> + *
> + * This checks the header of Android vendor boot image and verifies the magic
> + * is "VNDRBOOT"
> + *
> + * @vendor_boot_img: Pointer to boot image
> + * Return: non-zero if the magic is correct, zero otherwise
> + */
> +bool is_android_vendor_boot_image_header(const void *vendor_boot_img);
> +
>  /**
>   * board_fit_config_name_match() - Check for a matching board name
>   *
> -- 
> 2.34.1
diff mbox series

Patch

diff --git a/boot/image-android.c b/boot/image-android.c
index f16eebff49..712d437766 100644
--- a/boot/image-android.c
+++ b/boot/image-android.c
@@ -18,6 +18,65 @@ 
 
 static char andr_tmp_str[ANDR_BOOT_ARGS_SIZE + 1];
 
+static void android_boot_image_v3_v4_parse_hdr(const struct andr_boot_img_hdr_v3 *hdr,
+					       struct andr_image_data *data)
+{
+	ulong end;
+
+	data->kcmdline = hdr->cmdline;
+	data->header_version = hdr->header_version;
+
+	/*
+	 * The header takes a full page, the remaining components are aligned
+	 * on page boundary.
+	 */
+	end = (ulong)hdr;
+	end += ANDR_GKI_PAGE_SIZE;
+	data->kernel_ptr = end;
+	data->kernel_size = hdr->kernel_size;
+	end += ALIGN(hdr->kernel_size, ANDR_GKI_PAGE_SIZE);
+	data->ramdisk_size = hdr->ramdisk_size;
+	data->boot_ramdisk_size = hdr->ramdisk_size;
+	end += ALIGN(hdr->ramdisk_size, ANDR_GKI_PAGE_SIZE);
+
+	if (hdr->header_version > 3)
+		end += ALIGN(hdr->signature_size, ANDR_GKI_PAGE_SIZE);
+
+	data->boot_img_total_size = end - (ulong)hdr;
+}
+
+static void android_vendor_boot_image_v3_v4_parse_hdr(const struct andr_vnd_boot_img_hdr
+						      *hdr, struct andr_image_data *data)
+{
+	ulong end;
+
+	/*
+	 * The header takes a full page, the remaining components are aligned
+	 * on page boundary.
+	 */
+	data->tags_addr = hdr->tags_addr;
+	data->image_name = hdr->name;
+	data->kernel_addr = hdr->kernel_addr;
+	data->ramdisk_addr = hdr->ramdisk_addr;
+	data->dtb_load_addr = hdr->dtb_addr;
+	end = (ulong)hdr;
+	end += hdr->page_size;
+	if (hdr->vendor_ramdisk_size) {
+		data->vendor_ramdisk_ptr = end;
+		data->vendor_ramdisk_size = hdr->vendor_ramdisk_size;
+		data->ramdisk_size += hdr->vendor_ramdisk_size;
+		end += ALIGN(hdr->vendor_ramdisk_size, hdr->page_size);
+	}
+
+	data->dtb_ptr = end;
+	data->dtb_size = hdr->dtb_size;
+
+	end += ALIGN(hdr->dtb_size, hdr->page_size);
+	end += ALIGN(hdr->vendor_ramdisk_table_size, hdr->page_size);
+	end += ALIGN(hdr->bootconfig_size, hdr->page_size);
+	data->vendor_boot_img_total_size = end - (ulong)hdr;
+}
+
 static void android_boot_image_v0_v1_v2_parse_hdr(const struct andr_boot_img_hdr_v0 *hdr,
 						  struct andr_image_data *data)
 {
@@ -79,10 +138,20 @@  bool android_image_get_data(const void *boot_hdr, const void *vendor_boot_hdr,
 		return false;
 	}
 
-	if (((struct andr_boot_img_hdr_v0 *)boot_hdr)->header_version > 2)
-		printf("Only boot image header version 2 and below are supported\n");
-	else
+	if (((struct andr_boot_img_hdr_v0 *)boot_hdr)->header_version > 2) {
+		if (!vendor_boot_hdr) {
+			printf("For boot header v3+ vendor boot image has to be provided\n");
+			return false;
+		}
+		if (!is_android_vendor_boot_image_header(vendor_boot_hdr)) {
+			printf("Incorrect vendor boot image header\n");
+			return false;
+		}
+		android_boot_image_v3_v4_parse_hdr(boot_hdr, data);
+		android_vendor_boot_image_v3_v4_parse_hdr(vendor_boot_hdr, data);
+	} else {
 		android_boot_image_v0_v1_v2_parse_hdr(boot_hdr, data);
+	}
 
 	return true;
 }
@@ -200,6 +269,11 @@  int android_image_get_kernel(const struct andr_boot_img_hdr_v0 *hdr,
 	return 0;
 }
 
+bool is_android_vendor_boot_image_header(const void *vendor_boot_img)
+{
+	return !memcmp(VENDOR_BOOT_MAGIC, vendor_boot_img, ANDR_VENDOR_BOOT_MAGIC_SIZE);
+}
+
 bool is_android_boot_image_header(const struct andr_boot_img_hdr_v0 *hdr)
 {
 	return !memcmp(ANDR_BOOT_MAGIC, hdr, ANDR_BOOT_MAGIC_SIZE);
diff --git a/include/android_image.h b/include/android_image.h
index fb3a9d9858..99e7803508 100644
--- a/include/android_image.h
+++ b/include/android_image.h
@@ -323,6 +323,8 @@  struct andr_image_data {
 	ulong kernel_ptr;  /* kernel address */
 	u32 kernel_size;  /* size in bytes */
 	u32 ramdisk_size;  /* size in bytes */
+	ulong vendor_ramdisk_ptr;  /* vendor ramdisk address */
+	u32 vendor_ramdisk_size;  /* vendor ramdisk size*/
 	u32 boot_ramdisk_size;  /* size in bytes */
 	ulong second_ptr;  /* secondary bootloader address */
 	u32 second_size;  /* secondary bootloader size */
@@ -342,6 +344,7 @@  struct andr_image_data {
 	ulong tags_addr;  /* physical addr for kernel tags */
 	u32 header_version;  /* version of the boot image header */
 	u32 boot_img_total_size;  /* boot image size */
+	u32 vendor_boot_img_total_size;  /* vendor boot image size */
 };
 
 #endif
diff --git a/include/image.h b/include/image.h
index f84c03f08f..c2e751c136 100644
--- a/include/image.h
+++ b/include/image.h
@@ -1876,6 +1876,17 @@  bool android_image_print_dtb_contents(ulong hdr_addr);
  */
 bool is_android_boot_image_header(const struct andr_boot_img_hdr_v0 *hdr);
 
+/**
+ * is_android_vendor_boot_image_header() - Check the magic of vendor boot image
+ *
+ * This checks the header of Android vendor boot image and verifies the magic
+ * is "VNDRBOOT"
+ *
+ * @vendor_boot_img: Pointer to boot image
+ * Return: non-zero if the magic is correct, zero otherwise
+ */
+bool is_android_vendor_boot_image_header(const void *vendor_boot_img);
+
 /**
  * board_fit_config_name_match() - Check for a matching board name
  *