Update Snowbridge versions and prep for publishing (#2894)

- updates snowbridge crates to `0.9.0` - updates Cargo.toml files in preparation for publishing to crates.io - adds Kusama and Polkadot Snowbridge runtime config crates - moves runtime tests from the Snowbridge subtree into the bridge hub tests dir --------- Co-authored-by: claravanstaden <Cats 4 life!> Co-authored-by: Ron <yrong1997@gmail.com>
2026-05-31 01:41:03 +00:00 · 2024-01-10 18:51:27 +02:00
parent 6a80c10a6f
commit bab0348372
88 changed files with 579 additions and 490 deletions
@@ -0,0 +1,166 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: 2023 Snowfork <hello@snowfork.com>
+use super::*;
+mod util;
+
+use crate::Pallet as EthereumBeaconClient;
+use frame_benchmarking::v2::*;
+use frame_system::RawOrigin;
+
+#[cfg(feature = "beacon-spec-minimal")]
+mod fixtures_minimal;
+#[cfg(feature = "beacon-spec-minimal")]
+use fixtures_minimal::*;
+
+#[cfg(not(feature = "beacon-spec-minimal"))]
+mod fixtures_mainnet;
+#[cfg(not(feature = "beacon-spec-minimal"))]
+use fixtures_mainnet::*;
+
+use primitives::{
+	fast_aggregate_verify, prepare_aggregate_pubkey, prepare_aggregate_signature,
+	verify_merkle_branch,
+};
+use util::*;
+
+#[benchmarks]
+mod benchmarks {
+	use super::*;
+
+	#[benchmark]
+	fn force_checkpoint() -> Result<(), BenchmarkError> {
+		let checkpoint_update = make_checkpoint();
+		let block_root: H256 = checkpoint_update.header.hash_tree_root().unwrap();
+
+		#[extrinsic_call]
+		_(RawOrigin::Root, Box::new(*checkpoint_update));
+
+		assert!(<LatestFinalizedBlockRoot<T>>::get() == block_root);
+		assert!(<FinalizedBeaconState<T>>::get(block_root).is_some());
+
+		Ok(())
+	}
+
+	#[benchmark]
+	fn submit() -> Result<(), BenchmarkError> {
+		let caller: T::AccountId = whitelisted_caller();
+		let checkpoint_update = make_checkpoint();
+		let finalized_header_update = make_finalized_header_update();
+		let block_root: H256 = finalized_header_update.finalized_header.hash_tree_root().unwrap();
+		EthereumBeaconClient::<T>::process_checkpoint_update(&checkpoint_update)?;
+
+		#[extrinsic_call]
+		submit(RawOrigin::Signed(caller.clone()), Box::new(*finalized_header_update));
+
+		assert!(<LatestFinalizedBlockRoot<T>>::get() == block_root);
+		assert!(<FinalizedBeaconState<T>>::get(block_root).is_some());
+
+		Ok(())
+	}
+
+	#[benchmark]
+	fn submit_with_sync_committee() -> Result<(), BenchmarkError> {
+		let caller: T::AccountId = whitelisted_caller();
+		let checkpoint_update = make_checkpoint();
+		let sync_committee_update = make_sync_committee_update();
+		EthereumBeaconClient::<T>::process_checkpoint_update(&checkpoint_update)?;
+
+		#[extrinsic_call]
+		submit(RawOrigin::Signed(caller.clone()), Box::new(*sync_committee_update));
+
+		assert!(<NextSyncCommittee<T>>::exists());
+
+		Ok(())
+	}
+
+	#[benchmark]
+	fn submit_execution_header() -> Result<(), BenchmarkError> {
+		let caller: T::AccountId = whitelisted_caller();
+		let checkpoint_update = make_checkpoint();
+		let finalized_header_update = make_finalized_header_update();
+		let execution_header_update = make_execution_header_update();
+		let execution_header_hash = execution_header_update.execution_header.block_hash;
+		EthereumBeaconClient::<T>::process_checkpoint_update(&checkpoint_update)?;
+		EthereumBeaconClient::<T>::process_update(&finalized_header_update)?;
+
+		#[extrinsic_call]
+		_(RawOrigin::Signed(caller.clone()), Box::new(*execution_header_update));
+
+		assert!(<ExecutionHeaders<T>>::contains_key(execution_header_hash));
+
+		Ok(())
+	}
+
+	#[benchmark(extra)]
+	fn bls_fast_aggregate_verify_pre_aggregated() -> Result<(), BenchmarkError> {
+		EthereumBeaconClient::<T>::process_checkpoint_update(&make_checkpoint())?;
+		let update = make_sync_committee_update();
+		let participant_pubkeys = participant_pubkeys::<T>(&update)?;
+		let signing_root = signing_root::<T>(&update)?;
+		let agg_sig =
+			prepare_aggregate_signature(&update.sync_aggregate.sync_committee_signature).unwrap();
+		let agg_pub_key = prepare_aggregate_pubkey(&participant_pubkeys).unwrap();
+
+		#[block]
+		{
+			agg_sig.fast_aggregate_verify_pre_aggregated(signing_root.as_bytes(), &agg_pub_key);
+		}
+
+		Ok(())
+	}
+
+	#[benchmark(extra)]
+	fn bls_fast_aggregate_verify() -> Result<(), BenchmarkError> {
+		EthereumBeaconClient::<T>::process_checkpoint_update(&make_checkpoint())?;
+		let update = make_sync_committee_update();
+		let current_sync_committee = <CurrentSyncCommittee<T>>::get();
+		let absent_pubkeys = absent_pubkeys::<T>(&update)?;
+		let signing_root = signing_root::<T>(&update)?;
+
+		#[block]
+		{
+			fast_aggregate_verify(
+				&current_sync_committee.aggregate_pubkey,
+				&absent_pubkeys,
+				signing_root,
+				&update.sync_aggregate.sync_committee_signature,
+			)
+			.unwrap();
+		}
+
+		Ok(())
+	}
+
+	#[benchmark(extra)]
+	fn verify_merkle_proof() -> Result<(), BenchmarkError> {
+		EthereumBeaconClient::<T>::process_checkpoint_update(&make_checkpoint())?;
+		let update = make_sync_committee_update();
+		let block_root: H256 = update.finalized_header.hash_tree_root().unwrap();
+
+		#[block]
+		{
+			verify_merkle_branch(
+				block_root,
+				&update.finality_branch,
+				config::FINALIZED_ROOT_SUBTREE_INDEX,
+				config::FINALIZED_ROOT_DEPTH,
+				update.attested_header.state_root,
+			);
+		}
+
+		Ok(())
+	}
+
+	#[cfg(feature = "beacon-spec-minimal")]
+	impl_benchmark_test_suite!(
+		EthereumBeaconClient,
+		crate::mock::minimal::new_tester(),
+		crate::mock::minimal::Test
+	);
+	#[cfg(not(feature = "beacon-spec-minimal"))]
+	impl_benchmark_test_suite!(
+		EthereumBeaconClient,
+		crate::mock::mainnet::new_tester(),
+		crate::mock::mainnet::Test
+	);
+}