mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-11 15:21:08 +00:00
984 lines
32 KiB
Rust
984 lines
32 KiB
Rust
// Copyright 2019-2023 Parity Technologies (UK) Ltd.
|
|
// This file is dual-licensed as Apache-2.0 or GPL-3.0.
|
|
// see LICENSE for license details.
|
|
|
|
//! Utility functions for metadata validation.
|
|
|
|
use std::collections::HashSet;
|
|
|
|
use frame_metadata::v15::{
|
|
ExtrinsicMetadata, PalletMetadata, RuntimeApiMetadata, RuntimeApiMethodMetadata,
|
|
RuntimeMetadataV15, StorageEntryMetadata, StorageEntryType,
|
|
};
|
|
use scale_info::{form::PortableForm, Field, PortableRegistry, TypeDef, Variant};
|
|
|
|
/// Start with a predefined hashing value for the pallets.
|
|
const MAGIC_PALLET_VALUE: &[u8] = &[19];
|
|
|
|
/// Predefined value to be returned when we already visited a type.
|
|
const MAGIC_RECURSIVE_TYPE_VALUE: &[u8] = &[123];
|
|
|
|
// The number of bytes our `hash` function produces.
|
|
const HASH_LEN: usize = 32;
|
|
|
|
/// Internal byte representation for various metadata types utilized for
|
|
/// generating deterministic hashes between different rust versions.
|
|
#[repr(u8)]
|
|
enum TypeBeingHashed {
|
|
Composite,
|
|
Variant,
|
|
Sequence,
|
|
Array,
|
|
Tuple,
|
|
Primitive,
|
|
Compact,
|
|
BitSequence,
|
|
}
|
|
|
|
/// Hashing function utilized internally.
|
|
fn hash(data: &[u8]) -> [u8; 32] {
|
|
sp_core_hashing::twox_256(data)
|
|
}
|
|
|
|
/// XOR two hashes together. If we have two pseudorandom hashes, then this will
|
|
/// lead to another pseudorandom value. If there is potentially some pattern to
|
|
/// the hashes we are xoring (eg we might be xoring the same hashes a few times),
|
|
/// prefer `concat_and_hash` to give us stronger pseudorandomness guarantees.
|
|
fn xor(a: [u8; 32], b: [u8; 32]) -> [u8; 32] {
|
|
let mut out = [0u8; 32];
|
|
for (idx, (a, b)) in a.into_iter().zip(b).enumerate() {
|
|
out[idx] = a ^ b;
|
|
}
|
|
out
|
|
}
|
|
|
|
/// Combine two hashes or hash-like sets of bytes together into a single hash.
|
|
/// `xor` is OK for one-off combinations of bytes, but if we are merging
|
|
/// potentially identical hashes, this is a safer way to ensure the result is
|
|
/// unique.
|
|
fn concat_and_hash(a: [u8; 32], b: [u8; 32]) -> [u8; 32] {
|
|
let mut out = [0u8; HASH_LEN * 2];
|
|
out[0..HASH_LEN].copy_from_slice(&a[..]);
|
|
out[HASH_LEN..].copy_from_slice(&b[..]);
|
|
hash(&out)
|
|
}
|
|
|
|
/// Obtain the hash representation of a `scale_info::Field`.
|
|
fn get_field_hash(
|
|
registry: &PortableRegistry,
|
|
field: &Field<PortableForm>,
|
|
visited_ids: &mut HashSet<u32>,
|
|
) -> [u8; 32] {
|
|
let mut bytes = get_type_hash(registry, field.ty.id, visited_ids);
|
|
|
|
// XOR name and field name with the type hash if they exist
|
|
if let Some(name) = &field.name {
|
|
bytes = xor(bytes, hash(name.as_bytes()));
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash representation of a `scale_info::Variant`.
|
|
fn get_variant_hash(
|
|
registry: &PortableRegistry,
|
|
var: &Variant<PortableForm>,
|
|
visited_ids: &mut HashSet<u32>,
|
|
) -> [u8; 32] {
|
|
// Merge our hashes of the name and each field together using xor.
|
|
let mut bytes = hash(var.name.as_bytes());
|
|
for field in &var.fields {
|
|
bytes = concat_and_hash(bytes, get_field_hash(registry, field, visited_ids))
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash representation of a `scale_info::TypeDef`.
|
|
fn get_type_def_hash(
|
|
registry: &PortableRegistry,
|
|
ty_def: &TypeDef<PortableForm>,
|
|
visited_ids: &mut HashSet<u32>,
|
|
) -> [u8; 32] {
|
|
match ty_def {
|
|
TypeDef::Composite(composite) => {
|
|
let mut bytes = hash(&[TypeBeingHashed::Composite as u8]);
|
|
for field in &composite.fields {
|
|
bytes = concat_and_hash(bytes, get_field_hash(registry, field, visited_ids));
|
|
}
|
|
bytes
|
|
}
|
|
TypeDef::Variant(variant) => {
|
|
let mut bytes = hash(&[TypeBeingHashed::Variant as u8]);
|
|
for var in &variant.variants {
|
|
bytes = concat_and_hash(bytes, get_variant_hash(registry, var, visited_ids));
|
|
}
|
|
bytes
|
|
}
|
|
TypeDef::Sequence(sequence) => {
|
|
let bytes = hash(&[TypeBeingHashed::Sequence as u8]);
|
|
xor(
|
|
bytes,
|
|
get_type_hash(registry, sequence.type_param.id, visited_ids),
|
|
)
|
|
}
|
|
TypeDef::Array(array) => {
|
|
// Take length into account; different length must lead to different hash.
|
|
let len_bytes = array.len.to_be_bytes();
|
|
let bytes = hash(&[
|
|
TypeBeingHashed::Array as u8,
|
|
len_bytes[0],
|
|
len_bytes[1],
|
|
len_bytes[2],
|
|
len_bytes[3],
|
|
]);
|
|
xor(
|
|
bytes,
|
|
get_type_hash(registry, array.type_param.id, visited_ids),
|
|
)
|
|
}
|
|
TypeDef::Tuple(tuple) => {
|
|
let mut bytes = hash(&[TypeBeingHashed::Tuple as u8]);
|
|
for field in &tuple.fields {
|
|
bytes = concat_and_hash(bytes, get_type_hash(registry, field.id, visited_ids));
|
|
}
|
|
bytes
|
|
}
|
|
TypeDef::Primitive(primitive) => {
|
|
// Cloning the 'primitive' type should essentially be a copy.
|
|
hash(&[TypeBeingHashed::Primitive as u8, primitive.clone() as u8])
|
|
}
|
|
TypeDef::Compact(compact) => {
|
|
let bytes = hash(&[TypeBeingHashed::Compact as u8]);
|
|
xor(
|
|
bytes,
|
|
get_type_hash(registry, compact.type_param.id, visited_ids),
|
|
)
|
|
}
|
|
TypeDef::BitSequence(bitseq) => {
|
|
let mut bytes = hash(&[TypeBeingHashed::BitSequence as u8]);
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, bitseq.bit_order_type.id, visited_ids),
|
|
);
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, bitseq.bit_store_type.id, visited_ids),
|
|
);
|
|
bytes
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Obtain the hash representation of a `scale_info::Type` identified by id.
|
|
fn get_type_hash(registry: &PortableRegistry, id: u32, visited_ids: &mut HashSet<u32>) -> [u8; 32] {
|
|
// Guard against recursive types and return a fixed arbitrary hash
|
|
if !visited_ids.insert(id) {
|
|
return hash(MAGIC_RECURSIVE_TYPE_VALUE);
|
|
}
|
|
|
|
let ty = registry
|
|
.resolve(id)
|
|
.expect("Type ID provided by the metadata is registered; qed");
|
|
get_type_def_hash(registry, &ty.type_def, visited_ids)
|
|
}
|
|
|
|
/// Obtain the hash representation of a `frame_metadata::v15::ExtrinsicMetadata`.
|
|
fn get_extrinsic_hash(
|
|
registry: &PortableRegistry,
|
|
extrinsic: &ExtrinsicMetadata<PortableForm>,
|
|
) -> [u8; 32] {
|
|
let mut visited_ids = HashSet::<u32>::new();
|
|
|
|
let mut bytes = get_type_hash(registry, extrinsic.ty.id, &mut visited_ids);
|
|
|
|
bytes = xor(bytes, hash(&[extrinsic.version]));
|
|
for signed_extension in extrinsic.signed_extensions.iter() {
|
|
let mut ext_bytes = hash(signed_extension.identifier.as_bytes());
|
|
ext_bytes = xor(
|
|
ext_bytes,
|
|
get_type_hash(registry, signed_extension.ty.id, &mut visited_ids),
|
|
);
|
|
ext_bytes = xor(
|
|
ext_bytes,
|
|
get_type_hash(
|
|
registry,
|
|
signed_extension.additional_signed.id,
|
|
&mut visited_ids,
|
|
),
|
|
);
|
|
bytes = concat_and_hash(bytes, ext_bytes);
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Get the hash corresponding to a single storage entry.
|
|
fn get_storage_entry_hash(
|
|
registry: &PortableRegistry,
|
|
entry: &StorageEntryMetadata<PortableForm>,
|
|
visited_ids: &mut HashSet<u32>,
|
|
) -> [u8; 32] {
|
|
let mut bytes = hash(entry.name.as_bytes());
|
|
// Cloning 'entry.modifier' should essentially be a copy.
|
|
bytes = xor(bytes, hash(&[entry.modifier.clone() as u8]));
|
|
bytes = xor(bytes, hash(&entry.default));
|
|
|
|
match &entry.ty {
|
|
StorageEntryType::Plain(ty) => {
|
|
bytes = xor(bytes, get_type_hash(registry, ty.id, visited_ids));
|
|
}
|
|
StorageEntryType::Map {
|
|
hashers,
|
|
key,
|
|
value,
|
|
} => {
|
|
for hasher in hashers {
|
|
// Cloning the hasher should essentially be a copy.
|
|
bytes = concat_and_hash(bytes, [hasher.clone() as u8; 32]);
|
|
}
|
|
bytes = xor(bytes, get_type_hash(registry, key.id, visited_ids));
|
|
bytes = xor(bytes, get_type_hash(registry, value.id, visited_ids));
|
|
}
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash for a specific storage item, or an error if it's not found.
|
|
pub fn get_storage_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
pallet_name: &str,
|
|
storage_name: &str,
|
|
) -> Result<[u8; 32], NotFound> {
|
|
let pallet = metadata
|
|
.pallets
|
|
.iter()
|
|
.find(|p| p.name == pallet_name)
|
|
.ok_or(NotFound::Root)?;
|
|
|
|
let storage = pallet.storage.as_ref().ok_or(NotFound::Item)?;
|
|
|
|
let entry = storage
|
|
.entries
|
|
.iter()
|
|
.find(|s| s.name == storage_name)
|
|
.ok_or(NotFound::Item)?;
|
|
|
|
let hash = get_storage_entry_hash(&metadata.types, entry, &mut HashSet::new());
|
|
Ok(hash)
|
|
}
|
|
|
|
/// Obtain the hash for a specific constant, or an error if it's not found.
|
|
pub fn get_constant_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
pallet_name: &str,
|
|
constant_name: &str,
|
|
) -> Result<[u8; 32], NotFound> {
|
|
let pallet = metadata
|
|
.pallets
|
|
.iter()
|
|
.find(|p| p.name == pallet_name)
|
|
.ok_or(NotFound::Root)?;
|
|
|
|
let constant = pallet
|
|
.constants
|
|
.iter()
|
|
.find(|c| c.name == constant_name)
|
|
.ok_or(NotFound::Item)?;
|
|
|
|
// We only need to check that the type of the constant asked for matches.
|
|
let bytes = get_type_hash(&metadata.types, constant.ty.id, &mut HashSet::new());
|
|
Ok(bytes)
|
|
}
|
|
|
|
/// Obtain the hash for a specific call, or an error if it's not found.
|
|
pub fn get_call_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
pallet_name: &str,
|
|
call_name: &str,
|
|
) -> Result<[u8; 32], NotFound> {
|
|
let pallet = metadata
|
|
.pallets
|
|
.iter()
|
|
.find(|p| p.name == pallet_name)
|
|
.ok_or(NotFound::Root)?;
|
|
|
|
let call_id = pallet.calls.as_ref().ok_or(NotFound::Item)?.ty.id;
|
|
|
|
let call_ty = metadata.types.resolve(call_id).ok_or(NotFound::Item)?;
|
|
|
|
let call_variants = match &call_ty.type_def {
|
|
TypeDef::Variant(variant) => &variant.variants,
|
|
_ => return Err(NotFound::Item),
|
|
};
|
|
|
|
let variant = call_variants
|
|
.iter()
|
|
.find(|v| v.name == call_name)
|
|
.ok_or(NotFound::Item)?;
|
|
|
|
// hash the specific variant representing the call we are interested in.
|
|
let hash = get_variant_hash(&metadata.types, variant, &mut HashSet::new());
|
|
Ok(hash)
|
|
}
|
|
|
|
fn get_runtime_method_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
trait_metadata: &RuntimeApiMetadata<PortableForm>,
|
|
method_metadata: &RuntimeApiMethodMetadata<PortableForm>,
|
|
visited_ids: &mut HashSet<u32>,
|
|
) -> [u8; 32] {
|
|
// The trait name is part of the runtime API call that is being
|
|
// generated for this method. Therefore the trait name is strongly
|
|
// connected to the method in the same way as a parameter is
|
|
// to the method.
|
|
let mut bytes = hash(trait_metadata.name.as_bytes());
|
|
bytes = xor(bytes, hash(method_metadata.name.as_bytes()));
|
|
|
|
for input in &method_metadata.inputs {
|
|
bytes = xor(bytes, hash(input.name.as_bytes()));
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(&metadata.types, input.ty.id, visited_ids),
|
|
);
|
|
}
|
|
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(&metadata.types, method_metadata.output.id, visited_ids),
|
|
);
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash of a specific runtime trait.
|
|
pub fn get_runtime_trait_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
trait_metadata: &RuntimeApiMetadata<PortableForm>,
|
|
) -> [u8; 32] {
|
|
// Start out with any hash, the trait name is already part of the
|
|
// runtime method hash.
|
|
let mut bytes = hash(trait_metadata.name.as_bytes());
|
|
let mut visited_ids = HashSet::new();
|
|
|
|
let mut methods: Vec<_> = trait_metadata
|
|
.methods
|
|
.iter()
|
|
.map(|method_metadata| {
|
|
let bytes = get_runtime_method_hash(
|
|
metadata,
|
|
trait_metadata,
|
|
method_metadata,
|
|
&mut visited_ids,
|
|
);
|
|
(&*method_metadata.name, bytes)
|
|
})
|
|
.collect();
|
|
|
|
// Sort by method name to create a deterministic representation of the underlying metadata.
|
|
methods.sort_by_key(|&(name, _hash)| name);
|
|
|
|
// Note: Hash already takes into account the method name.
|
|
for (_, hash) in methods {
|
|
bytes = xor(bytes, hash);
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash of a specific runtime API function, or an error if it's not found.
|
|
pub fn get_runtime_api_hash(
|
|
metadata: &RuntimeMetadataV15,
|
|
trait_name: &str,
|
|
method_name: &str,
|
|
) -> Result<[u8; 32], NotFound> {
|
|
let trait_metadata = metadata
|
|
.apis
|
|
.iter()
|
|
.find(|m| m.name == trait_name)
|
|
.ok_or(NotFound::Root)?;
|
|
|
|
let method_metadata = trait_metadata
|
|
.methods
|
|
.iter()
|
|
.find(|m| m.name == method_name)
|
|
.ok_or(NotFound::Item)?;
|
|
|
|
Ok(get_runtime_method_hash(
|
|
metadata,
|
|
trait_metadata,
|
|
method_metadata,
|
|
&mut HashSet::new(),
|
|
))
|
|
}
|
|
|
|
/// Obtain the hash representation of a `frame_metadata::v15::PalletMetadata`.
|
|
pub fn get_pallet_hash(
|
|
registry: &PortableRegistry,
|
|
pallet: &PalletMetadata<PortableForm>,
|
|
) -> [u8; 32] {
|
|
// The pallet could potentially be empty and not contain any calls, events and so on.
|
|
// Use a magic (arbitrary) value as a base for hashing.
|
|
let mut bytes = hash(MAGIC_PALLET_VALUE);
|
|
let mut visited_ids = HashSet::<u32>::new();
|
|
|
|
if let Some(calls) = &pallet.calls {
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, calls.ty.id, &mut visited_ids),
|
|
);
|
|
}
|
|
if let Some(ref event) = pallet.event {
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, event.ty.id, &mut visited_ids),
|
|
);
|
|
}
|
|
for constant in pallet.constants.iter() {
|
|
bytes = xor(bytes, hash(constant.name.as_bytes()));
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, constant.ty.id, &mut visited_ids),
|
|
);
|
|
}
|
|
if let Some(ref error) = pallet.error {
|
|
bytes = xor(
|
|
bytes,
|
|
get_type_hash(registry, error.ty.id, &mut visited_ids),
|
|
);
|
|
}
|
|
if let Some(ref storage) = pallet.storage {
|
|
bytes = xor(bytes, hash(storage.prefix.as_bytes()));
|
|
for entry in storage.entries.iter() {
|
|
bytes = concat_and_hash(
|
|
bytes,
|
|
get_storage_entry_hash(registry, entry, &mut visited_ids),
|
|
);
|
|
}
|
|
}
|
|
|
|
bytes
|
|
}
|
|
|
|
/// Obtain the hash representation of a `frame_metadata::v15::RuntimeMetadataV15`.
|
|
pub fn get_metadata_hash(metadata: &RuntimeMetadataV15) -> [u8; 32] {
|
|
// The number of metadata components, other than variable number of pallets that produce a unique hash.
|
|
const STATIC_METADATA_COMPONENTS: usize = 2;
|
|
|
|
// Collect all pairs of (pallet name, pallet hash).
|
|
let mut pallets: Vec<(&str, [u8; 32])> = metadata
|
|
.pallets
|
|
.iter()
|
|
.map(|pallet| {
|
|
let hash = get_pallet_hash(&metadata.types, pallet);
|
|
(&*pallet.name, hash)
|
|
})
|
|
.collect();
|
|
|
|
// Sort by pallet name to create a deterministic representation of the underlying metadata.
|
|
pallets.sort_by_key(|&(name, _hash)| name);
|
|
|
|
// Note: pallet name is excluded from hashing.
|
|
// The number of hashes that we take into account, each having a `HASH_LEN` output.
|
|
let metadata_components = pallets.len() + STATIC_METADATA_COMPONENTS;
|
|
let mut bytes = Vec::with_capacity(metadata_components * HASH_LEN);
|
|
for (_, hash) in pallets.iter() {
|
|
bytes.extend(hash)
|
|
}
|
|
|
|
bytes.extend(get_extrinsic_hash(&metadata.types, &metadata.extrinsic));
|
|
|
|
let mut visited_ids = HashSet::<u32>::new();
|
|
bytes.extend(get_type_hash(
|
|
&metadata.types,
|
|
metadata.ty.id,
|
|
&mut visited_ids,
|
|
));
|
|
|
|
let mut apis: Vec<_> = metadata
|
|
.apis
|
|
.iter()
|
|
.map(|api| (&*api.name, get_runtime_trait_hash(metadata, api)))
|
|
.collect();
|
|
|
|
// Sort the runtime APIs by trait name to provide a deterministic output.
|
|
apis.sort_by_key(|&(name, _hash)| name);
|
|
|
|
for (_, hash) in apis.iter() {
|
|
bytes.extend(hash)
|
|
}
|
|
|
|
hash(&bytes)
|
|
}
|
|
|
|
/// Obtain the hash representation of a `frame_metadata::v15::RuntimeMetadataV15`
|
|
/// hashing only the provided pallets.
|
|
///
|
|
/// **Note:** This is similar to `get_metadata_hash`, but performs hashing only of the provided
|
|
/// pallets if they exist. There are cases where the runtime metadata contains a subset of
|
|
/// the pallets from the static metadata. In those cases, the static API can communicate
|
|
/// properly with the subset of pallets from the runtime node.
|
|
pub fn get_metadata_per_pallet_hash<T: AsRef<str>>(
|
|
metadata: &RuntimeMetadataV15,
|
|
pallets: &[T],
|
|
) -> [u8; 32] {
|
|
// Collect all pairs of (pallet name, pallet hash).
|
|
let mut pallets_hashed: Vec<(&str, [u8; 32])> = metadata
|
|
.pallets
|
|
.iter()
|
|
.filter_map(|pallet| {
|
|
// Make sure to filter just the pallets we are interested in.
|
|
let in_pallet = pallets
|
|
.iter()
|
|
.any(|pallet_ref| pallet_ref.as_ref() == pallet.name);
|
|
if in_pallet {
|
|
let hash = get_pallet_hash(&metadata.types, pallet);
|
|
Some((&*pallet.name, hash))
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
.collect();
|
|
|
|
// Sort by pallet name to create a deterministic representation of the underlying metadata.
|
|
pallets_hashed.sort_by_key(|&(name, _hash)| name);
|
|
|
|
// Note: pallet name is excluded from hashing.
|
|
// We are only hashing the hashes of the pallets.
|
|
let mut bytes = Vec::with_capacity(pallets_hashed.len() * HASH_LEN);
|
|
for (_, hash) in pallets_hashed.iter() {
|
|
bytes.extend(hash)
|
|
}
|
|
|
|
hash(&bytes)
|
|
}
|
|
|
|
/// An error returned if we attempt to get the hash for a specific call, constant,
|
|
/// storage or runtime API function does not exist.
|
|
///
|
|
/// The location of the specific item (call, constant, storage or runtime API function)
|
|
/// is stored with two indirections:
|
|
/// - Root
|
|
/// The root location of the item. For calls, constants, storage this represents the
|
|
/// pallet name. While for runtime API function this represents the trait name.
|
|
/// - Item
|
|
/// The actual item. For calls, constants, storage this represents the actual name.
|
|
/// While for runtime API functions this represents the method name.
|
|
#[derive(Clone, Debug)]
|
|
pub enum NotFound {
|
|
/// The root location of the item cannot be found.
|
|
/// - pallet name: for calls, constants, storage
|
|
/// - trait name: for runtime API functions
|
|
Root,
|
|
/// The actual item name cannot be found.
|
|
Item,
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use bitvec::{order::Lsb0, vec::BitVec};
|
|
use frame_metadata::v15::{
|
|
ExtrinsicMetadata, PalletCallMetadata, PalletConstantMetadata, PalletErrorMetadata,
|
|
PalletEventMetadata, PalletMetadata, PalletStorageMetadata, RuntimeMetadataV15,
|
|
StorageEntryMetadata, StorageEntryModifier,
|
|
};
|
|
use scale_info::meta_type;
|
|
|
|
use super::*;
|
|
|
|
// Define recursive types.
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct A {
|
|
pub b: Box<B>,
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct B {
|
|
pub a: Box<A>,
|
|
}
|
|
|
|
// Define TypeDef supported types.
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
// TypeDef::Composite with TypeDef::Array with Typedef::Primitive.
|
|
struct AccountId32([u8; 32]);
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
// TypeDef::Variant.
|
|
enum DigestItem {
|
|
PreRuntime(
|
|
// TypeDef::Array with primitive.
|
|
[::core::primitive::u8; 4usize],
|
|
// TypeDef::Sequence.
|
|
::std::vec::Vec<::core::primitive::u8>,
|
|
),
|
|
Other(::std::vec::Vec<::core::primitive::u8>),
|
|
// Nested TypeDef::Tuple.
|
|
RuntimeEnvironmentUpdated(((i8, i16), (u32, u64))),
|
|
// TypeDef::Compact.
|
|
Index(#[codec(compact)] ::core::primitive::u8),
|
|
// TypeDef::BitSequence.
|
|
BitSeq(BitVec<u8, Lsb0>),
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
// Ensure recursive types and TypeDef variants are captured.
|
|
struct MetadataTestType {
|
|
recursive: A,
|
|
composite: AccountId32,
|
|
type_def: DigestItem,
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
// Simulate a PalletCallMetadata.
|
|
enum Call {
|
|
#[codec(index = 0)]
|
|
FillBlock { ratio: AccountId32 },
|
|
#[codec(index = 1)]
|
|
Remark { remark: DigestItem },
|
|
}
|
|
|
|
fn build_default_extrinsic() -> ExtrinsicMetadata {
|
|
ExtrinsicMetadata {
|
|
ty: meta_type::<()>(),
|
|
version: 0,
|
|
signed_extensions: vec![],
|
|
}
|
|
}
|
|
|
|
fn default_pallet() -> PalletMetadata {
|
|
PalletMetadata {
|
|
name: "Test",
|
|
storage: None,
|
|
calls: None,
|
|
event: None,
|
|
constants: vec![],
|
|
error: None,
|
|
index: 0,
|
|
docs: vec![],
|
|
}
|
|
}
|
|
|
|
fn build_default_pallets() -> Vec<PalletMetadata> {
|
|
vec![
|
|
PalletMetadata {
|
|
name: "First",
|
|
calls: Some(PalletCallMetadata {
|
|
ty: meta_type::<MetadataTestType>(),
|
|
}),
|
|
..default_pallet()
|
|
},
|
|
PalletMetadata {
|
|
name: "Second",
|
|
index: 1,
|
|
calls: Some(PalletCallMetadata {
|
|
ty: meta_type::<(DigestItem, AccountId32, A)>(),
|
|
}),
|
|
..default_pallet()
|
|
},
|
|
]
|
|
}
|
|
|
|
fn pallets_to_metadata(pallets: Vec<PalletMetadata>) -> RuntimeMetadataV15 {
|
|
RuntimeMetadataV15::new(
|
|
pallets,
|
|
build_default_extrinsic(),
|
|
meta_type::<()>(),
|
|
vec![],
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn different_pallet_index() {
|
|
let pallets = build_default_pallets();
|
|
let mut pallets_swap = pallets.clone();
|
|
|
|
let metadata = pallets_to_metadata(pallets);
|
|
|
|
// Change the order in which pallets are registered.
|
|
pallets_swap.swap(0, 1);
|
|
pallets_swap[0].index = 0;
|
|
pallets_swap[1].index = 1;
|
|
let metadata_swap = pallets_to_metadata(pallets_swap);
|
|
|
|
let hash = get_metadata_hash(&metadata);
|
|
let hash_swap = get_metadata_hash(&metadata_swap);
|
|
|
|
// Changing pallet order must still result in a deterministic unique hash.
|
|
assert_eq!(hash, hash_swap);
|
|
}
|
|
|
|
#[test]
|
|
fn recursive_type() {
|
|
let mut pallet = default_pallet();
|
|
pallet.calls = Some(PalletCallMetadata {
|
|
ty: meta_type::<A>(),
|
|
});
|
|
let metadata = pallets_to_metadata(vec![pallet]);
|
|
|
|
// Check hashing algorithm finishes on a recursive type.
|
|
get_metadata_hash(&metadata);
|
|
}
|
|
|
|
#[test]
|
|
/// Ensure correctness of hashing when parsing the `metadata.types`.
|
|
///
|
|
/// Having a recursive structure `A: { B }` and `B: { A }` registered in different order
|
|
/// `types: { { id: 0, A }, { id: 1, B } }` and `types: { { id: 0, B }, { id: 1, A } }`
|
|
/// must produce the same deterministic hashing value.
|
|
fn recursive_types_different_order() {
|
|
let mut pallets = build_default_pallets();
|
|
pallets[0].calls = Some(PalletCallMetadata {
|
|
ty: meta_type::<A>(),
|
|
});
|
|
pallets[1].calls = Some(PalletCallMetadata {
|
|
ty: meta_type::<B>(),
|
|
});
|
|
pallets[1].index = 1;
|
|
let mut pallets_swap = pallets.clone();
|
|
let metadata = pallets_to_metadata(pallets);
|
|
|
|
pallets_swap.swap(0, 1);
|
|
pallets_swap[0].index = 0;
|
|
pallets_swap[1].index = 1;
|
|
let metadata_swap = pallets_to_metadata(pallets_swap);
|
|
|
|
let hash = get_metadata_hash(&metadata);
|
|
let hash_swap = get_metadata_hash(&metadata_swap);
|
|
|
|
// Changing pallet order must still result in a deterministic unique hash.
|
|
assert_eq!(hash, hash_swap);
|
|
}
|
|
|
|
#[test]
|
|
fn pallet_hash_correctness() {
|
|
let compare_pallets_hash = |lhs: &PalletMetadata, rhs: &PalletMetadata| {
|
|
let metadata = pallets_to_metadata(vec![lhs.clone()]);
|
|
let hash = get_metadata_hash(&metadata);
|
|
|
|
let metadata = pallets_to_metadata(vec![rhs.clone()]);
|
|
let new_hash = get_metadata_hash(&metadata);
|
|
|
|
assert_ne!(hash, new_hash);
|
|
};
|
|
|
|
// Build metadata progressively from an empty pallet to a fully populated pallet.
|
|
let mut pallet = default_pallet();
|
|
let pallet_lhs = pallet.clone();
|
|
pallet.storage = Some(PalletStorageMetadata {
|
|
prefix: "Storage",
|
|
entries: vec![StorageEntryMetadata {
|
|
name: "BlockWeight",
|
|
modifier: StorageEntryModifier::Default,
|
|
ty: StorageEntryType::Plain(meta_type::<u8>()),
|
|
default: vec![],
|
|
docs: vec![],
|
|
}],
|
|
});
|
|
compare_pallets_hash(&pallet_lhs, &pallet);
|
|
|
|
let pallet_lhs = pallet.clone();
|
|
// Calls are similar to:
|
|
//
|
|
// ```
|
|
// pub enum Call {
|
|
// call_name_01 { arg01: type },
|
|
// call_name_02 { arg01: type, arg02: type }
|
|
// }
|
|
// ```
|
|
pallet.calls = Some(PalletCallMetadata {
|
|
ty: meta_type::<Call>(),
|
|
});
|
|
compare_pallets_hash(&pallet_lhs, &pallet);
|
|
|
|
let pallet_lhs = pallet.clone();
|
|
// Events are similar to Calls.
|
|
pallet.event = Some(PalletEventMetadata {
|
|
ty: meta_type::<Call>(),
|
|
});
|
|
compare_pallets_hash(&pallet_lhs, &pallet);
|
|
|
|
let pallet_lhs = pallet.clone();
|
|
pallet.constants = vec![PalletConstantMetadata {
|
|
name: "BlockHashCount",
|
|
ty: meta_type::<u64>(),
|
|
value: vec![96u8, 0, 0, 0],
|
|
docs: vec![],
|
|
}];
|
|
compare_pallets_hash(&pallet_lhs, &pallet);
|
|
|
|
let pallet_lhs = pallet.clone();
|
|
pallet.error = Some(PalletErrorMetadata {
|
|
ty: meta_type::<MetadataTestType>(),
|
|
});
|
|
compare_pallets_hash(&pallet_lhs, &pallet);
|
|
}
|
|
|
|
#[test]
|
|
fn metadata_per_pallet_hash_correctness() {
|
|
let pallets = build_default_pallets();
|
|
|
|
// Build metadata with just the first pallet.
|
|
let metadata_one = pallets_to_metadata(vec![pallets[0].clone()]);
|
|
// Build metadata with both pallets.
|
|
let metadata_both = pallets_to_metadata(pallets);
|
|
|
|
// Hashing will ignore any non-existant pallet and return the same result.
|
|
let hash = get_metadata_per_pallet_hash(&metadata_one, &["First", "Second"]);
|
|
let hash_rhs = get_metadata_per_pallet_hash(&metadata_one, &["First"]);
|
|
assert_eq!(hash, hash_rhs, "hashing should ignore non-existant pallets");
|
|
|
|
// Hashing one pallet from metadata with 2 pallets inserted will ignore the second pallet.
|
|
let hash_second = get_metadata_per_pallet_hash(&metadata_both, &["First"]);
|
|
assert_eq!(
|
|
hash_second, hash,
|
|
"hashing one pallet should ignore the others"
|
|
);
|
|
|
|
// Check hashing with all pallets.
|
|
let hash_second = get_metadata_per_pallet_hash(&metadata_both, &["First", "Second"]);
|
|
assert_ne!(
|
|
hash_second, hash,
|
|
"hashing both pallets should produce a different result from hashing just one pallet"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn field_semantic_changes() {
|
|
// Get a hash representation of the provided meta type,
|
|
// inserted in the context of pallet metadata call.
|
|
let to_hash = |meta_ty| {
|
|
let pallet = PalletMetadata {
|
|
calls: Some(PalletCallMetadata { ty: meta_ty }),
|
|
..default_pallet()
|
|
};
|
|
let metadata = pallets_to_metadata(vec![pallet]);
|
|
get_metadata_hash(&metadata)
|
|
};
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNotNamedA {
|
|
First(u8),
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNotNamedB {
|
|
First(u8),
|
|
}
|
|
// Semantic changes apply only to field names.
|
|
// This is considered to be a good tradeoff in hashing performance, as refactoring
|
|
// a structure / enum 's name is less likely to cause a breaking change.
|
|
// Even if the enums have different names, 'EnumFieldNotNamedA' and 'EnumFieldNotNamedB',
|
|
// they are equal in meaning (i.e, both contain `First(u8)`).
|
|
assert_eq!(
|
|
to_hash(meta_type::<EnumFieldNotNamedA>()),
|
|
to_hash(meta_type::<EnumFieldNotNamedB>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructFieldNotNamedA([u8; 32]);
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructFieldNotNamedSecondB([u8; 32]);
|
|
// Similarly to enums, semantic changes apply only inside the structure fields.
|
|
assert_eq!(
|
|
to_hash(meta_type::<StructFieldNotNamedA>()),
|
|
to_hash(meta_type::<StructFieldNotNamedSecondB>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNotNamed {
|
|
First(u8),
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNotNamedSecond {
|
|
Second(u8),
|
|
}
|
|
// The enums are binary compatible, they contain a different semantic meaning:
|
|
// `First(u8)` and `Second(u8)`.
|
|
assert_ne!(
|
|
to_hash(meta_type::<EnumFieldNotNamed>()),
|
|
to_hash(meta_type::<EnumFieldNotNamedSecond>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNamed {
|
|
First { a: u8 },
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldNamedSecond {
|
|
First { b: u8 },
|
|
}
|
|
// Named fields contain a different semantic meaning ('a' and 'b').
|
|
assert_ne!(
|
|
to_hash(meta_type::<EnumFieldNamed>()),
|
|
to_hash(meta_type::<EnumFieldNamedSecond>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructFieldNamed {
|
|
a: u32,
|
|
}
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructFieldNamedSecond {
|
|
b: u32,
|
|
}
|
|
// Similar to enums, struct fields contain a different semantic meaning ('a' and 'b').
|
|
assert_ne!(
|
|
to_hash(meta_type::<StructFieldNamed>()),
|
|
to_hash(meta_type::<StructFieldNamedSecond>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumField {
|
|
First,
|
|
// Field is unnamed, but has type name `u8`.
|
|
Second(u8),
|
|
// File is named and has type name `u8`.
|
|
Third { named: u8 },
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
enum EnumFieldSwap {
|
|
Second(u8),
|
|
First,
|
|
Third { named: u8 },
|
|
}
|
|
// Swapping the registration order should also be taken into account.
|
|
assert_ne!(
|
|
to_hash(meta_type::<EnumField>()),
|
|
to_hash(meta_type::<EnumFieldSwap>())
|
|
);
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructField {
|
|
a: u32,
|
|
b: u32,
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(scale_info::TypeInfo)]
|
|
struct StructFieldSwap {
|
|
b: u32,
|
|
a: u32,
|
|
}
|
|
assert_ne!(
|
|
to_hash(meta_type::<StructField>()),
|
|
to_hash(meta_type::<StructFieldSwap>())
|
|
);
|
|
}
|
|
}
|