1*897a1d94STom Rini // SPDX-License-Identifier: MIT 2d8f9d2afSIgor Opaniuk /* 3d8f9d2afSIgor Opaniuk * Copyright (C) 2016 The Android Open Source Project 4d8f9d2afSIgor Opaniuk */ 5d8f9d2afSIgor Opaniuk 6d8f9d2afSIgor Opaniuk #include "avb_chain_partition_descriptor.h" 7d8f9d2afSIgor Opaniuk #include "avb_util.h" 8d8f9d2afSIgor Opaniuk avb_chain_partition_descriptor_validate_and_byteswap(const AvbChainPartitionDescriptor * src,AvbChainPartitionDescriptor * dest)9d8f9d2afSIgor Opaniukbool avb_chain_partition_descriptor_validate_and_byteswap( 10d8f9d2afSIgor Opaniuk const AvbChainPartitionDescriptor* src, AvbChainPartitionDescriptor* dest) { 11d8f9d2afSIgor Opaniuk uint64_t expected_size; 12d8f9d2afSIgor Opaniuk 13d8f9d2afSIgor Opaniuk avb_memcpy(dest, src, sizeof(AvbChainPartitionDescriptor)); 14d8f9d2afSIgor Opaniuk 15d8f9d2afSIgor Opaniuk if (!avb_descriptor_validate_and_byteswap((const AvbDescriptor*)src, 16d8f9d2afSIgor Opaniuk (AvbDescriptor*)dest)) 17d8f9d2afSIgor Opaniuk return false; 18d8f9d2afSIgor Opaniuk 19d8f9d2afSIgor Opaniuk if (dest->parent_descriptor.tag != AVB_DESCRIPTOR_TAG_CHAIN_PARTITION) { 20d8f9d2afSIgor Opaniuk avb_error("Invalid tag for chain partition descriptor.\n"); 21d8f9d2afSIgor Opaniuk return false; 22d8f9d2afSIgor Opaniuk } 23d8f9d2afSIgor Opaniuk 24d8f9d2afSIgor Opaniuk dest->rollback_index_location = avb_be32toh(dest->rollback_index_location); 25d8f9d2afSIgor Opaniuk dest->partition_name_len = avb_be32toh(dest->partition_name_len); 26d8f9d2afSIgor Opaniuk dest->public_key_len = avb_be32toh(dest->public_key_len); 27d8f9d2afSIgor Opaniuk 28d8f9d2afSIgor Opaniuk if (dest->rollback_index_location < 1) { 29d8f9d2afSIgor Opaniuk avb_error("Invalid rollback index location value.\n"); 30d8f9d2afSIgor Opaniuk return false; 31d8f9d2afSIgor Opaniuk } 32d8f9d2afSIgor Opaniuk 33d8f9d2afSIgor Opaniuk /* Check that partition_name and public_key are fully contained. */ 34d8f9d2afSIgor Opaniuk expected_size = sizeof(AvbChainPartitionDescriptor) - sizeof(AvbDescriptor); 35d8f9d2afSIgor Opaniuk if (!avb_safe_add_to(&expected_size, dest->partition_name_len) || 36d8f9d2afSIgor Opaniuk !avb_safe_add_to(&expected_size, dest->public_key_len)) { 37d8f9d2afSIgor Opaniuk avb_error("Overflow while adding up sizes.\n"); 38d8f9d2afSIgor Opaniuk return false; 39d8f9d2afSIgor Opaniuk } 40d8f9d2afSIgor Opaniuk if (expected_size > dest->parent_descriptor.num_bytes_following) { 41d8f9d2afSIgor Opaniuk avb_error("Descriptor payload size overflow.\n"); 42d8f9d2afSIgor Opaniuk return false; 43d8f9d2afSIgor Opaniuk } 44d8f9d2afSIgor Opaniuk return true; 45d8f9d2afSIgor Opaniuk } 46