common: spl: ab: add ab interface to get slot info
Change-Id: Iffd16dfcf9014b024e522639cbcd851f002b742e Signed-off-by: Jason Zhu <jason.zhu@rock-chips.com>
This commit is contained in:
Jason Zhu
Kever Yang
parent
6bccab8707
commit
0cc16201bc
|
|
@ -785,6 +785,12 @@ config SPL_OPTEE
|
|||
OP-TEE is an open source Trusted OS which is loaded by SPL.
|
||||
More detail at: https://github.com/OP-TEE/optee_os
|
||||
|
||||
config SPL_AB
|
||||
bool "Support AB system boot"
|
||||
depends on SPL
|
||||
help
|
||||
Enable this config to support AB system boot.
|
||||
|
||||
config SPL_LOAD_RKFW
|
||||
bool "SPL support load rockchip firmware images"
|
||||
depends on SPL
|
||||
|
|
|
|||
|
|
@ -34,6 +34,7 @@ obj-$(CONFIG_$(SPL_TPL_)MMC_SUPPORT) += spl_mmc.o
|
|||
obj-$(CONFIG_$(SPL_TPL_)ATF) += spl_atf.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)OPTEE) += spl_optee.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)LOAD_RKFW) += spl_rkfw.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)AB) += spl_ab.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)USB_SUPPORT) += spl_usb.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)FAT_SUPPORT) += spl_fat.o
|
||||
obj-$(CONFIG_$(SPL_TPL_)EXT_SUPPORT) += spl_ext.o
|
||||
|
|
|
|||
|
|
@ -0,0 +1,267 @@
|
|||
// SPDX-License-Identifier: GPL-2.0
|
||||
/*
|
||||
* Copyright (c) 2019 Fuzhou Rockchip Electronics Co., Ltd
|
||||
*/
|
||||
|
||||
#include <common.h>
|
||||
#include <blk.h>
|
||||
#include <spl_ab.h>
|
||||
|
||||
int safe_memcmp(const void *s1, const void *s2, size_t n)
|
||||
{
|
||||
const unsigned char *us1 = s1;
|
||||
const unsigned char *us2 = s2;
|
||||
int result = 0;
|
||||
|
||||
if (0 == n)
|
||||
return 0;
|
||||
|
||||
/*
|
||||
* Code snippet without data-dependent branch due to Nate Lawson
|
||||
* (nate@root.org) of Root Labs.
|
||||
*/
|
||||
while (n--)
|
||||
result |= *us1++ ^ *us2++;
|
||||
|
||||
return result != 0;
|
||||
}
|
||||
|
||||
static uint32_t htobe32(uint32_t in)
|
||||
{
|
||||
union {
|
||||
uint32_t word;
|
||||
uint8_t bytes[4];
|
||||
} ret;
|
||||
|
||||
ret.bytes[0] = (in >> 24) & 0xff;
|
||||
ret.bytes[1] = (in >> 16) & 0xff;
|
||||
ret.bytes[2] = (in >> 8) & 0xff;
|
||||
ret.bytes[3] = in & 0xff;
|
||||
|
||||
return ret.word;
|
||||
}
|
||||
|
||||
static uint32_t be32toh(uint32_t in)
|
||||
{
|
||||
uint8_t *d = (uint8_t *)∈
|
||||
uint32_t ret;
|
||||
|
||||
ret = ((uint32_t)d[0]) << 24;
|
||||
ret |= ((uint32_t)d[1]) << 16;
|
||||
ret |= ((uint32_t)d[2]) << 8;
|
||||
ret |= ((uint32_t)d[3]);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static bool spl_ab_data_verify_and_byteswap(const AvbABData *src,
|
||||
AvbABData *dest)
|
||||
{
|
||||
/* Ensure magic is correct. */
|
||||
if (safe_memcmp(src->magic, AVB_AB_MAGIC, AVB_AB_MAGIC_LEN) != 0) {
|
||||
printf("Magic is incorrect.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
memcpy(dest, src, sizeof(AvbABData));
|
||||
dest->crc32 = be32toh(dest->crc32);
|
||||
|
||||
/* Ensure we don't attempt to access any fields if the major version
|
||||
* is not supported.
|
||||
*/
|
||||
if (dest->version_major > AVB_AB_MAJOR_VERSION) {
|
||||
printf("No support for given major version.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
/* Bail if CRC32 doesn't match. */
|
||||
if (dest->crc32 !=
|
||||
crc32(0, (const uint8_t *)dest,
|
||||
sizeof(AvbABData) - sizeof(uint32_t))) {
|
||||
printf("CRC32 does not match.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static void spl_ab_data_update_crc_and_byteswap(const AvbABData *src,
|
||||
AvbABData *dest)
|
||||
{
|
||||
memcpy(dest, src, sizeof(AvbABData));
|
||||
dest->crc32 = htobe32(crc32(0, (const uint8_t *)dest,
|
||||
sizeof(AvbABData) - sizeof(uint32_t)));
|
||||
}
|
||||
|
||||
static void spl_ab_data_init(AvbABData *data)
|
||||
{
|
||||
memset(data, '\0', sizeof(AvbABData));
|
||||
memcpy(data->magic, AVB_AB_MAGIC, AVB_AB_MAGIC_LEN);
|
||||
data->version_major = AVB_AB_MAJOR_VERSION;
|
||||
data->version_minor = AVB_AB_MINOR_VERSION;
|
||||
data->last_boot = 0;
|
||||
data->slots[0].priority = AVB_AB_MAX_PRIORITY;
|
||||
data->slots[0].tries_remaining = AVB_AB_MAX_TRIES_REMAINING;
|
||||
data->slots[0].successful_boot = 0;
|
||||
data->slots[1].priority = AVB_AB_MAX_PRIORITY - 1;
|
||||
data->slots[1].tries_remaining = AVB_AB_MAX_TRIES_REMAINING;
|
||||
data->slots[1].successful_boot = 0;
|
||||
}
|
||||
|
||||
static int spl_read_ab_metadata(struct blk_desc *dev_desc, AvbABData *ab_data,
|
||||
char *partition)
|
||||
{
|
||||
disk_partition_t part_info;
|
||||
char temp[512];
|
||||
int ret;
|
||||
|
||||
if (!dev_desc || !partition || !ab_data)
|
||||
return -EFAULT;
|
||||
|
||||
if (part_get_info_by_name(dev_desc, partition, &part_info) < 0)
|
||||
return -ENOENT;
|
||||
|
||||
ret = blk_dread(dev_desc, part_info.start + AB_METADATA_OFFSET, 1,
|
||||
temp);
|
||||
if (ret != 1)
|
||||
return -ENODEV;
|
||||
|
||||
if (sizeof(AvbABData) > 512)
|
||||
return -ENOMEM;
|
||||
|
||||
memcpy(ab_data, temp, sizeof(AvbABData));
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int spl_write_ab_metadata(struct blk_desc *dev_desc, AvbABData *ab_data,
|
||||
char *partition)
|
||||
{
|
||||
disk_partition_t part_info;
|
||||
char temp[512];
|
||||
int ret;
|
||||
|
||||
if (!dev_desc || !partition || !ab_data)
|
||||
return -EFAULT;
|
||||
|
||||
if (sizeof(AvbABData) > 512)
|
||||
return -ENOMEM;
|
||||
|
||||
memcpy(temp, ab_data, sizeof(AvbABData));
|
||||
if (part_get_info_by_name(dev_desc, partition, &part_info) < 0)
|
||||
return -ENOENT;
|
||||
|
||||
ret = blk_dwrite(dev_desc, part_info.start + AB_METADATA_OFFSET, 1,
|
||||
temp);
|
||||
if (ret != 1)
|
||||
return -ENODEV;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int spl_ab_data_write(struct blk_desc *dev_desc, AvbABData *ab_data,
|
||||
char *partition)
|
||||
{
|
||||
AvbABData serialized;
|
||||
|
||||
spl_ab_data_update_crc_and_byteswap(ab_data, &serialized);
|
||||
|
||||
return spl_write_ab_metadata(dev_desc, &serialized, partition);
|
||||
}
|
||||
|
||||
static int spl_ab_data_read(struct blk_desc *dev_desc, AvbABData *ab_data,
|
||||
char *partition)
|
||||
{
|
||||
int ret;
|
||||
AvbABData serialized;
|
||||
|
||||
ret = spl_read_ab_metadata(dev_desc, &serialized, partition);
|
||||
if (ret)
|
||||
return ret;
|
||||
|
||||
if (!spl_ab_data_verify_and_byteswap(&serialized, ab_data)) {
|
||||
printf("Error validating A/B metadata from disk. "
|
||||
"Resetting and writing new A/B metadata to disk.\n");
|
||||
spl_ab_data_init(ab_data);
|
||||
spl_ab_data_write(dev_desc, ab_data, partition);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static bool spl_slot_is_bootable(AvbABSlotData *slot)
|
||||
{
|
||||
return slot->priority > 0 &&
|
||||
(slot->successful_boot || (slot->tries_remaining > 0));
|
||||
}
|
||||
|
||||
static int spl_get_lastboot(AvbABData *ab_data)
|
||||
{
|
||||
return ab_data->last_boot;
|
||||
}
|
||||
|
||||
int spl_get_current_slot(struct blk_desc *dev_desc, char *partition, char *slot)
|
||||
{
|
||||
size_t slot_index_to_boot;
|
||||
AvbABData ab_data;
|
||||
int ret;
|
||||
|
||||
ret = spl_ab_data_read(dev_desc, &ab_data, partition);
|
||||
if (ret)
|
||||
return ret;
|
||||
|
||||
if (spl_slot_is_bootable(&ab_data.slots[0]) &&
|
||||
spl_slot_is_bootable(&ab_data.slots[1])) {
|
||||
if (ab_data.slots[1].priority > ab_data.slots[0].priority)
|
||||
slot_index_to_boot = 1;
|
||||
else
|
||||
slot_index_to_boot = 0;
|
||||
} else if (spl_slot_is_bootable(&ab_data.slots[0])) {
|
||||
slot_index_to_boot = 0;
|
||||
} else if (spl_slot_is_bootable(&ab_data.slots[1])) {
|
||||
slot_index_to_boot = 1;
|
||||
} else {
|
||||
printf("No bootable slots found, use lastboot.\n");
|
||||
if (spl_get_lastboot(&ab_data) == 0) {
|
||||
memcpy(slot, "_a", 2);
|
||||
goto out;
|
||||
} else if (spl_get_lastboot(&ab_data) == 1) {
|
||||
memcpy(slot, "_b", 2);
|
||||
goto out;
|
||||
} else {
|
||||
return -ENODEV;
|
||||
}
|
||||
}
|
||||
|
||||
if (slot_index_to_boot == 0)
|
||||
memcpy(slot, "_a", 2);
|
||||
else if (slot_index_to_boot == 1)
|
||||
memcpy(slot, "_b", 2);
|
||||
|
||||
out:
|
||||
return 0;
|
||||
}
|
||||
|
||||
int spl_get_partitions_sector(struct blk_desc *dev_desc, char *partition,
|
||||
u32 *sectors)
|
||||
{
|
||||
disk_partition_t part_info;
|
||||
char part[10] = {0};
|
||||
char slot[3] = {0};
|
||||
|
||||
if (!partition || !sectors)
|
||||
return -EFAULT;
|
||||
|
||||
spl_get_current_slot(dev_desc, "misc", slot);
|
||||
if (strlen(partition) > 8)
|
||||
return -ENOMEM;
|
||||
|
||||
strcat(part, partition);
|
||||
strcat(part, slot);
|
||||
if (part_get_info_by_name(dev_desc, part, &part_info) < 0)
|
||||
return -ENODEV;
|
||||
|
||||
*sectors = part_info.start;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
|
@ -0,0 +1,36 @@
|
|||
/* SPDX-License-Identifier: GPL-2.0+ */
|
||||
/*
|
||||
* (C) Copyright 2019 Rockchip Electronics Co., Ltd
|
||||
*/
|
||||
|
||||
#ifndef _SPL_AB_H_
|
||||
#define _SPL_AB_H_
|
||||
|
||||
#include <android_avb/libavb_ab.h>
|
||||
#include <android_avb/avb_ab_flow.h>
|
||||
|
||||
#define AB_METADATA_OFFSET 4
|
||||
|
||||
/*
|
||||
* spl_get_current_slot
|
||||
*
|
||||
* @dev_desc: block description
|
||||
* @partition: partition name
|
||||
* @slot: A/B slot
|
||||
*
|
||||
* return: 0 success, others fail.
|
||||
*/
|
||||
int spl_get_current_slot(struct blk_desc *dev_desc, char *partition,
|
||||
char *slot);
|
||||
|
||||
/*
|
||||
* spl_get_partitions_sector
|
||||
*
|
||||
* @dev_desc: block description
|
||||
* @partition: partition name
|
||||
* @sectors: firmware load address
|
||||
*/
|
||||
int spl_get_partitions_sector(struct blk_desc *dev_desc, char *partition,
|
||||
u32 *sectors);
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue