404 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			404 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			C
		
	
	
	
// SPDX-License-Identifier: GPL-2.0+
 | 
						|
/*
 | 
						|
 * EFI Firmware management protocol
 | 
						|
 *
 | 
						|
 *  Copyright (c) 2020 Linaro Limited
 | 
						|
 *			Author: AKASHI Takahiro
 | 
						|
 */
 | 
						|
 | 
						|
#include <common.h>
 | 
						|
#include <charset.h>
 | 
						|
#include <dfu.h>
 | 
						|
#include <efi_loader.h>
 | 
						|
#include <image.h>
 | 
						|
#include <linux/list.h>
 | 
						|
 | 
						|
/* Place holder; not supported */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_get_image_unsupported(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	u8 image_index,
 | 
						|
	void *image,
 | 
						|
	efi_uintn_t *image_size)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %d %p %p\n", this, image_index, image, image_size);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_UNSUPPORTED);
 | 
						|
}
 | 
						|
 | 
						|
/* Place holder; not supported */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_check_image_unsupported(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	u8 image_index,
 | 
						|
	const void *image,
 | 
						|
	efi_uintn_t *image_size,
 | 
						|
	u32 *image_updatable)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %d %p %p %p\n", this, image_index, image, image_size,
 | 
						|
		  image_updatable);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_UNSUPPORTED);
 | 
						|
}
 | 
						|
 | 
						|
/* Place holder; not supported */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_get_package_info_unsupported(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	u32 *package_version,
 | 
						|
	u16 **package_version_name,
 | 
						|
	u32 *package_version_name_maxlen,
 | 
						|
	u64 *attributes_supported,
 | 
						|
	u64 *attributes_setting)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %p %p %p %p %p\n", this, package_version,
 | 
						|
		  package_version_name, package_version_name_maxlen,
 | 
						|
		  attributes_supported, attributes_setting);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_UNSUPPORTED);
 | 
						|
}
 | 
						|
 | 
						|
/* Place holder; not supported */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_set_package_info_unsupported(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	const void *image,
 | 
						|
	efi_uintn_t *image_size,
 | 
						|
	const void *vendor_code,
 | 
						|
	u32 package_version,
 | 
						|
	const u16 *package_version_name)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %p %p %p %x %p\n", this, image, image_size, vendor_code,
 | 
						|
		  package_version, package_version_name);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_UNSUPPORTED);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * efi_get_dfu_info - return information about the current firmware image
 | 
						|
 * @this:			Protocol instance
 | 
						|
 * @image_info_size:		Size of @image_info
 | 
						|
 * @image_info:			Image information
 | 
						|
 * @descriptor_version:		Pointer to version number
 | 
						|
 * @descriptor_count:		Pointer to number of descriptors
 | 
						|
 * @descriptor_size:		Pointer to descriptor size
 | 
						|
 * package_version:		Package version
 | 
						|
 * package_version_name:	Package version's name
 | 
						|
 * image_type:			Image type GUID
 | 
						|
 *
 | 
						|
 * Return information bout the current firmware image in @image_info.
 | 
						|
 * @image_info will consist of a number of descriptors.
 | 
						|
 * Each descriptor will be created based on "dfu_alt_info" variable.
 | 
						|
 *
 | 
						|
 * Return		status code
 | 
						|
 */
 | 
						|
static efi_status_t efi_get_dfu_info(
 | 
						|
	efi_uintn_t *image_info_size,
 | 
						|
	struct efi_firmware_image_descriptor *image_info,
 | 
						|
	u32 *descriptor_version,
 | 
						|
	u8 *descriptor_count,
 | 
						|
	efi_uintn_t *descriptor_size,
 | 
						|
	u32 *package_version,
 | 
						|
	u16 **package_version_name,
 | 
						|
	const efi_guid_t *image_type)
 | 
						|
{
 | 
						|
	struct dfu_entity *dfu;
 | 
						|
	size_t names_len, total_size;
 | 
						|
	int dfu_num, i;
 | 
						|
	u16 *name, *next;
 | 
						|
 | 
						|
	dfu_init_env_entities(NULL, NULL);
 | 
						|
 | 
						|
	names_len = 0;
 | 
						|
	dfu_num = 0;
 | 
						|
	list_for_each_entry(dfu, &dfu_list, list) {
 | 
						|
		names_len += (utf8_utf16_strlen(dfu->name) + 1) * 2;
 | 
						|
		dfu_num++;
 | 
						|
	}
 | 
						|
	if (!dfu_num) {
 | 
						|
		log_warning("Probably dfu_alt_info not defined\n");
 | 
						|
		*image_info_size = 0;
 | 
						|
		dfu_free_entities();
 | 
						|
 | 
						|
		return EFI_SUCCESS;
 | 
						|
	}
 | 
						|
 | 
						|
	total_size = sizeof(*image_info) * dfu_num + names_len;
 | 
						|
	/*
 | 
						|
	 * we will assume that sizeof(*image_info) * dfu_name
 | 
						|
	 * is, at least, a multiple of 2. So the start address for
 | 
						|
	 * image_id_name would be aligned with 2 bytes.
 | 
						|
	 */
 | 
						|
	if (*image_info_size < total_size) {
 | 
						|
		*image_info_size = total_size;
 | 
						|
		dfu_free_entities();
 | 
						|
 | 
						|
		return EFI_BUFFER_TOO_SMALL;
 | 
						|
	}
 | 
						|
	*image_info_size = total_size;
 | 
						|
 | 
						|
	*descriptor_version = EFI_FIRMWARE_IMAGE_DESCRIPTOR_VERSION;
 | 
						|
	*descriptor_count = dfu_num;
 | 
						|
	*descriptor_size = sizeof(*image_info);
 | 
						|
	*package_version = 0xffffffff; /* not supported */
 | 
						|
	*package_version_name = NULL; /* not supported */
 | 
						|
 | 
						|
	/* DFU alt number should correspond to image_index */
 | 
						|
	i = 0;
 | 
						|
	/* Name area starts just after descriptors */
 | 
						|
	name = (u16 *)((u8 *)image_info + sizeof(*image_info) * dfu_num);
 | 
						|
	next = name;
 | 
						|
	list_for_each_entry(dfu, &dfu_list, list) {
 | 
						|
		image_info[i].image_index = dfu->alt + 1;
 | 
						|
		image_info[i].image_type_id = *image_type;
 | 
						|
		image_info[i].image_id = dfu->alt;
 | 
						|
 | 
						|
		/* copy the DFU entity name */
 | 
						|
		utf8_utf16_strcpy(&next, dfu->name);
 | 
						|
		image_info[i].image_id_name = name;
 | 
						|
		name = ++next;
 | 
						|
 | 
						|
		image_info[i].version = 0; /* not supported */
 | 
						|
		image_info[i].version_name = NULL; /* not supported */
 | 
						|
		image_info[i].size = 0;
 | 
						|
		image_info[i].attributes_supported =
 | 
						|
				IMAGE_ATTRIBUTE_IMAGE_UPDATABLE;
 | 
						|
		image_info[i].attributes_setting =
 | 
						|
				IMAGE_ATTRIBUTE_IMAGE_UPDATABLE;
 | 
						|
		image_info[i].lowest_supported_image_version = 0;
 | 
						|
		image_info[i].last_attempt_version = 0;
 | 
						|
		image_info[i].last_attempt_status = LAST_ATTEMPT_STATUS_SUCCESS;
 | 
						|
		image_info[i].hardware_instance = 1;
 | 
						|
		image_info[i].dependencies = NULL;
 | 
						|
 | 
						|
		i++;
 | 
						|
	}
 | 
						|
 | 
						|
	dfu_free_entities();
 | 
						|
 | 
						|
	return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef CONFIG_EFI_CAPSULE_FIRMWARE_FIT
 | 
						|
/*
 | 
						|
 * This FIRMWARE_MANAGEMENT_PROTOCOL driver provides a firmware update
 | 
						|
 * method with existing FIT image format, and handles
 | 
						|
 *   - multiple regions of firmware via DFU
 | 
						|
 * but doesn't support
 | 
						|
 *   - versioning of firmware image
 | 
						|
 *   - package information
 | 
						|
 */
 | 
						|
const efi_guid_t efi_firmware_image_type_uboot_fit =
 | 
						|
	EFI_FIRMWARE_IMAGE_TYPE_UBOOT_FIT_GUID;
 | 
						|
 | 
						|
/**
 | 
						|
 * efi_firmware_fit_get_image_info - return information about the current
 | 
						|
 *				     firmware image
 | 
						|
 * @this:			Protocol instance
 | 
						|
 * @image_info_size:		Size of @image_info
 | 
						|
 * @image_info:			Image information
 | 
						|
 * @descriptor_version:		Pointer to version number
 | 
						|
 * @descriptor_count:		Pointer to number of descriptors
 | 
						|
 * @descriptor_size:		Pointer to descriptor size
 | 
						|
 * package_version:		Package version
 | 
						|
 * package_version_name:	Package version's name
 | 
						|
 *
 | 
						|
 * Return information bout the current firmware image in @image_info.
 | 
						|
 * @image_info will consist of a number of descriptors.
 | 
						|
 * Each descriptor will be created based on "dfu_alt_info" variable.
 | 
						|
 *
 | 
						|
 * Return		status code
 | 
						|
 */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_fit_get_image_info(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	efi_uintn_t *image_info_size,
 | 
						|
	struct efi_firmware_image_descriptor *image_info,
 | 
						|
	u32 *descriptor_version,
 | 
						|
	u8 *descriptor_count,
 | 
						|
	efi_uintn_t *descriptor_size,
 | 
						|
	u32 *package_version,
 | 
						|
	u16 **package_version_name)
 | 
						|
{
 | 
						|
	efi_status_t ret;
 | 
						|
 | 
						|
	EFI_ENTRY("%p %p %p %p %p %p %p %p\n", this,
 | 
						|
		  image_info_size, image_info,
 | 
						|
		  descriptor_version, descriptor_count, descriptor_size,
 | 
						|
		  package_version, package_version_name);
 | 
						|
 | 
						|
	if (!image_info_size)
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	if (*image_info_size &&
 | 
						|
	    (!image_info || !descriptor_version || !descriptor_count ||
 | 
						|
	     !descriptor_size || !package_version || !package_version_name))
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	ret = efi_get_dfu_info(image_info_size, image_info,
 | 
						|
			       descriptor_version, descriptor_count,
 | 
						|
			       descriptor_size,
 | 
						|
			       package_version, package_version_name,
 | 
						|
			       &efi_firmware_image_type_uboot_fit);
 | 
						|
 | 
						|
	return EFI_EXIT(ret);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * efi_firmware_fit_set_image - update the firmware image
 | 
						|
 * @this:		Protocol instance
 | 
						|
 * @image_index:	Image index number
 | 
						|
 * @image:		New image
 | 
						|
 * @image_size:		Size of new image
 | 
						|
 * @vendor_code:	Vendor-specific update policy
 | 
						|
 * @progress:		Function to report the progress of update
 | 
						|
 * @abort_reason:	Pointer to string of abort reason
 | 
						|
 *
 | 
						|
 * Update the firmware to new image, using dfu. The new image should
 | 
						|
 * have FIT image format commonly used in U-Boot.
 | 
						|
 * @vendor_code, @progress and @abort_reason are not supported.
 | 
						|
 *
 | 
						|
 * Return:		status code
 | 
						|
 */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_fit_set_image(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	u8 image_index,
 | 
						|
	const void *image,
 | 
						|
	efi_uintn_t image_size,
 | 
						|
	const void *vendor_code,
 | 
						|
	efi_status_t (*progress)(efi_uintn_t completion),
 | 
						|
	u16 **abort_reason)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %d %p %ld %p %p %p\n", this, image_index, image,
 | 
						|
		  image_size, vendor_code, progress, abort_reason);
 | 
						|
 | 
						|
	if (!image || image_index != 1)
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	if (fit_update(image))
 | 
						|
		return EFI_EXIT(EFI_DEVICE_ERROR);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_SUCCESS);
 | 
						|
}
 | 
						|
 | 
						|
const struct efi_firmware_management_protocol efi_fmp_fit = {
 | 
						|
	.get_image_info = efi_firmware_fit_get_image_info,
 | 
						|
	.get_image = efi_firmware_get_image_unsupported,
 | 
						|
	.set_image = efi_firmware_fit_set_image,
 | 
						|
	.check_image = efi_firmware_check_image_unsupported,
 | 
						|
	.get_package_info = efi_firmware_get_package_info_unsupported,
 | 
						|
	.set_package_info = efi_firmware_set_package_info_unsupported,
 | 
						|
};
 | 
						|
#endif /* CONFIG_EFI_CAPSULE_FIRMWARE_FIT */
 | 
						|
 | 
						|
#ifdef CONFIG_EFI_CAPSULE_FIRMWARE_RAW
 | 
						|
/*
 | 
						|
 * This FIRMWARE_MANAGEMENT_PROTOCOL driver provides a firmware update
 | 
						|
 * method with raw data.
 | 
						|
 */
 | 
						|
const efi_guid_t efi_firmware_image_type_uboot_raw =
 | 
						|
	EFI_FIRMWARE_IMAGE_TYPE_UBOOT_RAW_GUID;
 | 
						|
 | 
						|
/**
 | 
						|
 * efi_firmware_raw_get_image_info - return information about the current
 | 
						|
				     firmware image
 | 
						|
 * @this:			Protocol instance
 | 
						|
 * @image_info_size:		Size of @image_info
 | 
						|
 * @image_info:			Image information
 | 
						|
 * @descriptor_version:		Pointer to version number
 | 
						|
 * @descriptor_count:		Pointer to number of descriptors
 | 
						|
 * @descriptor_size:		Pointer to descriptor size
 | 
						|
 * package_version:		Package version
 | 
						|
 * package_version_name:	Package version's name
 | 
						|
 *
 | 
						|
 * Return information bout the current firmware image in @image_info.
 | 
						|
 * @image_info will consist of a number of descriptors.
 | 
						|
 * Each descriptor will be created based on "dfu_alt_info" variable.
 | 
						|
 *
 | 
						|
 * Return		status code
 | 
						|
 */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_raw_get_image_info(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	efi_uintn_t *image_info_size,
 | 
						|
	struct efi_firmware_image_descriptor *image_info,
 | 
						|
	u32 *descriptor_version,
 | 
						|
	u8 *descriptor_count,
 | 
						|
	efi_uintn_t *descriptor_size,
 | 
						|
	u32 *package_version,
 | 
						|
	u16 **package_version_name)
 | 
						|
{
 | 
						|
	efi_status_t ret = EFI_SUCCESS;
 | 
						|
 | 
						|
	EFI_ENTRY("%p %p %p %p %p %p %p %p\n", this,
 | 
						|
		  image_info_size, image_info,
 | 
						|
		  descriptor_version, descriptor_count, descriptor_size,
 | 
						|
		  package_version, package_version_name);
 | 
						|
 | 
						|
	if (!image_info_size)
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	if (*image_info_size &&
 | 
						|
	    (!image_info || !descriptor_version || !descriptor_count ||
 | 
						|
	     !descriptor_size || !package_version || !package_version_name))
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	ret = efi_get_dfu_info(image_info_size, image_info,
 | 
						|
			       descriptor_version, descriptor_count,
 | 
						|
			       descriptor_size,
 | 
						|
			       package_version, package_version_name,
 | 
						|
			       &efi_firmware_image_type_uboot_raw);
 | 
						|
 | 
						|
	return EFI_EXIT(ret);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * efi_firmware_raw_set_image - update the firmware image
 | 
						|
 * @this:		Protocol instance
 | 
						|
 * @image_index:	Image index number
 | 
						|
 * @image:		New image
 | 
						|
 * @image_size:		Size of new image
 | 
						|
 * @vendor_code:	Vendor-specific update policy
 | 
						|
 * @progress:		Function to report the progress of update
 | 
						|
 * @abort_reason:	Pointer to string of abort reason
 | 
						|
 *
 | 
						|
 * Update the firmware to new image, using dfu. The new image should
 | 
						|
 * be a single raw image.
 | 
						|
 * @vendor_code, @progress and @abort_reason are not supported.
 | 
						|
 *
 | 
						|
 * Return:		status code
 | 
						|
 */
 | 
						|
static
 | 
						|
efi_status_t EFIAPI efi_firmware_raw_set_image(
 | 
						|
	struct efi_firmware_management_protocol *this,
 | 
						|
	u8 image_index,
 | 
						|
	const void *image,
 | 
						|
	efi_uintn_t image_size,
 | 
						|
	const void *vendor_code,
 | 
						|
	efi_status_t (*progress)(efi_uintn_t completion),
 | 
						|
	u16 **abort_reason)
 | 
						|
{
 | 
						|
	EFI_ENTRY("%p %d %p %ld %p %p %p\n", this, image_index, image,
 | 
						|
		  image_size, vendor_code, progress, abort_reason);
 | 
						|
 | 
						|
	if (!image)
 | 
						|
		return EFI_EXIT(EFI_INVALID_PARAMETER);
 | 
						|
 | 
						|
	if (dfu_write_by_alt(image_index - 1, (void *)image, image_size,
 | 
						|
			     NULL, NULL))
 | 
						|
		return EFI_EXIT(EFI_DEVICE_ERROR);
 | 
						|
 | 
						|
	return EFI_EXIT(EFI_SUCCESS);
 | 
						|
}
 | 
						|
 | 
						|
const struct efi_firmware_management_protocol efi_fmp_raw = {
 | 
						|
	.get_image_info = efi_firmware_raw_get_image_info,
 | 
						|
	.get_image = efi_firmware_get_image_unsupported,
 | 
						|
	.set_image = efi_firmware_raw_set_image,
 | 
						|
	.check_image = efi_firmware_check_image_unsupported,
 | 
						|
	.get_package_info = efi_firmware_get_package_info_unsupported,
 | 
						|
	.set_package_info = efi_firmware_set_package_info_unsupported,
 | 
						|
};
 | 
						|
#endif /* CONFIG_EFI_CAPSULE_FIRMWARE_RAW */
 |